Android虹软人脸识别sdk使用工具类

public class FaceUtil
{
private static final String TAG = FaceUtil.class.getSimpleName();
private static FaceUtil faceInstance = null;
public FaceDB mFaceDB;
private AFR_FSDKFace mAFR_FSDKFace;
private static Context mContext = null;

//face detect
private AFT_FSDKVersion version = new AFT_FSDKVersion();
private AFT_FSDKEngine engine = new AFT_FSDKEngine();
private ASAE_FSDKVersion mAgeVersion = new ASAE_FSDKVersion();
private ASAE_FSDKEngine mAgeEngine = new ASAE_FSDKEngine();
private ASGE_FSDKVersion mGenderVersion = new ASGE_FSDKVersion();
private ASGE_FSDKEngine mGenderEngine = new ASGE_FSDKEngine();
private List<AFT_FSDKFace> result = new ArrayList<>();
private List<ASAE_FSDKAge> ages = new ArrayList<>();
private List<ASGE_FSDKGender> genders = new ArrayList<>();
private AFT_FSDKFace mAFT_FSDKFace = null;
private  boolean openFace = false;

public static FaceUtil getInstance(Context context)
{
if(faceInstance == null)
{
faceInstance = new FaceUtil(context);
}
return faceInstance;
}

public FaceUtil(Context context)
{
mContext = context;
}

public void setFaceDB()
{
//face
mFaceDB = new FaceDB(mContext.getExternalCacheDir().getPath());
Log.d(TAG,"getExternalCacheDir : "+mContext.getExternalCacheDir().getPath());
}

public void initFaceDetect()
{
AFT_FSDKError err = engine.AFT_FSDK_InitialFaceEngine(FaceDB.appid, FaceDB.ft_key, AFT_FSDKEngine.AFT_OPF_0_HIGHER_EXT, 16, 5);
Log.d(TAG, "AFT_FSDK_InitialFaceEngine =" + err.getCode());
err = engine.AFT_FSDK_GetVersion(version);
Log.d(TAG, "AFT_FSDK_GetVersion:" + version.toString() + "," + err.getCode());

ASAE_FSDKError error = mAgeEngine.ASAE_FSDK_InitAgeEngine(FaceDB.appid, FaceDB.age_key);
Log.d(TAG, "ASAE_FSDK_InitAgeEngine =" + error.getCode());
error = mAgeEngine.ASAE_FSDK_GetVersion(mAgeVersion);
Log.d(TAG, "ASAE_FSDK_GetVersion:" + mAgeVersion.toString() + "," + error.getCode());

ASGE_FSDKError error1 = mGenderEngine.ASGE_FSDK_InitgGenderEngine(FaceDB.appid, FaceDB.gender_key);
Log.d(TAG, "ASGE_FSDK_InitgGenderEngine =" + error1.getCode());
error1 = mGenderEngine.ASGE_FSDK_GetVersion(mGenderVersion);
Log.d(TAG, "ASGE_FSDK_GetVersion:" + mGenderVersion.toString() + "," + error1.getCode());

}

public void deInitFaceDetect()
{
AFT_FSDKError err = engine.AFT_FSDK_UninitialFaceEngine();
Log.d(TAG, "AFT_FSDK_UninitialFaceEngine =" + err.getCode());

ASAE_FSDKError err1 = mAgeEngine.ASAE_FSDK_UninitAgeEngine();
Log.d(TAG, "ASAE_FSDK_UninitAgeEngine =" + err1.getCode());

ASGE_FSDKError err2 = mGenderEngine.ASGE_FSDK_UninitGenderEngine();
Log.d(TAG, "ASGE_FSDK_UninitGenderEngine =" + err2.getCode());
}

public boolean faceFeatureDetect(byte[] mYUVData,int width,int height)
{
boolean bDectectFace = false;
AFT_FSDKError err =engine.AFT_FSDK_FaceFeatureDetect(mYUVData, width, height, AFT_FSDKEngine.CP_PAF_NV21, result);
Log.d(TAG, "AFT_FSDK_FaceFeatureDetect =" + err.getCode());
Log.d(TAG, "Face=" + result.size());

if (!result.isEmpty())
{
for (AFT_FSDKFace face : result) {
Log.d(TAG, "Face:" + face.toString());
}
mAFT_FSDKFace = result.get(0).clone();

bDectectFace = true;
}
else
{
Log.d(TAG,"没有检测到人脸");
bDectectFace = false;
}
result.clear();
return bDectectFace;
}

public AFT_FSDKFace getmAFT_FSDKFace()
{
return mAFT_FSDKFace;
}

public String getAge(byte[] mYUVData,int width,int height,List<ASAE_FSDKFace> face1)
{
String age;
ASAE_FSDKError error1 = mAgeEngine.ASAE_FSDK_AgeEstimation_Image(mYUVData, width, height, AFT_FSDKEngine.CP_PAF_NV21, face1, ages);
Log.d("face", "ASAE_FSDK_AgeEstimation_Image:" + error1.getCode());
if(ages.size()== 0)
{
Log.d(TAG, "ages.size()== 0" );
age = "年龄未知";
}
else
{
Log.d(TAG, "age:" + ages.get(0).getAge());
age = ages.get(0).getAge() == 0 ? "年龄未知" : ages.get(0).getAge() + "岁";
}
return age;
}

public String getGender(byte[] mYUVData,int width,int height,List<ASGE_FSDKFace> face2)
{
String gender;
ASGE_FSDKError error2 = mGenderEngine.ASGE_FSDK_GenderEstimation_Image(mYUVData, width, height, AFT_FSDKEngine.CP_PAF_NV21, face2, genders);
Log.d("face", ",ASGE_FSDK_GenderEstimation_Image:" + error2.getCode());
if(ages.size()== 0)
{
Log.d(TAG, "ages.size()== 0" );
gender = "性别未知";
}
else
{
Log.d(TAG, "gender:" +  genders.get(0).getGender());
gender = genders.get(0).getGender() == -1 ? "性别未知" : (genders.get(0).getGender() == 0 ? "男" : "女");
}
return gender;
}

/**
* @param path
* @return
*/
public static Bitmap decodeImage(String path) {
Bitmap res;
try {
ExifInterface exif = new ExifInterface(path);
int orientation = exif.getAttributeInt(ExifInterface.TAG_ORIENTATION, ExifInterface.ORIENTATION_NORMAL);

BitmapFactory.Options op = new BitmapFactory.Options();
op.inSampleSize = 1;
op.inJustDecodeBounds = false;
//op.inMutable = true;
res = BitmapFactory.decodeFile(path, op);
//rotate and scale.
Matrix matrix = new Matrix();

if (orientation == ExifInterface.ORIENTATION_ROTATE_90) {
matrix.postRotate(90);
} else if (orientation == ExifInterface.ORIENTATION_ROTATE_180) {
matrix.postRotate(180);
} else if (orientation == ExifInterface.ORIENTATION_ROTATE_270) {
matrix.postRotate(270);
}

Bitmap temp = Bitmap.createBitmap(res, 0, 0, res.getWidth(), res.getHeight(), matrix, true);
Log.d(TAG, "check target Image:" + temp.getWidth() + "X" + temp.getHeight());

if (!temp.equals(res)) {
res.recycle();
}
return temp;
} catch (Exception e) {
e.printStackTrace();
}
return null;
}

public void bitmapDetect(String name,Bitmap mBitmap)
{
byte[] data = new byte[mBitmap.getWidth() * mBitmap.getHeight() * 3 / 2];
ImageConverter convert = new ImageConverter();
convert.initial(mBitmap.getWidth(), mBitmap.getHeight(), ImageConverter.CP_PAF_NV21);
if (convert.convert(mBitmap, data)) {
Log.d(TAG, "convert ok!");
}
convert.destroy();

AFD_FSDKEngine engine = new AFD_FSDKEngine();
AFD_FSDKVersion version = new AFD_FSDKVersion();
List<AFD_FSDKFace> result = new ArrayList<AFD_FSDKFace>();
AFD_FSDKError err = engine.AFD_FSDK_InitialFaceEngine(FaceDB.appid, FaceDB.fd_key, AFD_FSDKEngine.AFD_OPF_0_HIGHER_EXT, 16, 5);
Log.d(TAG, "AFD_FSDK_InitialFaceEngine = " + err.getCode());
if (err.getCode() != AFD_FSDKError.MOK) {
//            Message reg = Message.obtain();
//            reg.what = MSG_CODE;
//            reg.arg1 = MSG_EVENT_FD_ERROR;
//            reg.arg2 = err.getCode();
//            mUIHandler.sendMessage(reg);
Log.d(TAG, "MSG_EVENT_FD_ERROR");
ToastUtil.showToast("sdk error ：MSG_EVENT_FD_ERROR "+err.getCode());
}
err = engine.AFD_FSDK_GetVersion(version);
Log.d(TAG, "AFD_FSDK_GetVersion =" + version.toString() + ", " + err.getCode());
err  = engine.AFD_FSDK_StillImageFaceDetection(data, mBitmap.getWidth(), mBitmap.getHeight(), AFD_FSDKEngine.CP_PAF_NV21, result);
Log.d(TAG, "AFD_FSDK_StillImageFaceDetection =" + err.getCode() + "<" + result.size());

if (!result.isEmpty()) {
AFR_FSDKVersion version1 = new AFR_FSDKVersion();
AFR_FSDKEngine engine1 = new AFR_FSDKEngine();
AFR_FSDKFace result1 = new AFR_FSDKFace();
AFR_FSDKError error1 = engine1.AFR_FSDK_InitialEngine(FaceDB.appid, FaceDB.fr_key);
Log.d(TAG, "AFR_FSDK_InitialEngine = " + error1.getCode());
if (error1.getCode() != AFD_FSDKError.MOK) {
//                Message reg = Message.obtain();
//                reg.what = MSG_CODE;
//                reg.arg1 = MSG_EVENT_FR_ERROR;
//                reg.arg2 = error1.getCode();
//                mUIHandler.sendMessage(reg);
Log.d(TAG, "MSG_EVENT_FR_ERROR");
ToastUtil.showToast("sdk error ：MSG_EVENT_FR_ERROR "+error1.getCode());
}
error1 = engine1.AFR_FSDK_GetVersion(version1);
Log.d(TAG, "FR=" + version.toString() + "," + error1.getCode()); //(210, 178 - 478, 446), degree = 1　780, 2208 - 1942, 3370
error1 = engine1.AFR_FSDK_ExtractFRFeature(data, mBitmap.getWidth(), mBitmap.getHeight(), AFR_FSDKEngine.CP_PAF_NV21, new Rect(result.get(0).getRect()), result.get(0).getDegree(), result1);
Log.d(TAG, "Face=" + result1.getFeatureData()[0] + "," + result1.getFeatureData()[1] + "," + result1.getFeatureData()[2] + "," + error1.getCode());
if(error1.getCode() == error1.MOK) {
mAFR_FSDKFace = result1.clone();
//                int width = result.get(0).getRect().width();
//                int height = result.get(0).getRect().height();
//                Bitmap face_bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.RGB_565);
//                Canvas face_canvas = new Canvas(face_bitmap);
//                face_canvas.drawBitmap(mBitmap, result.get(0).getRect(), new Rect(0, 0, width, height), null);
//                Message reg = Message.obtain();
//                reg.what = MSG_CODE;
//                reg.arg1 = MSG_EVENT_REG;
//                reg.obj = face_bitmap;
//                mUIHandler.sendMessage(reg);
mFaceDB.addFace(name, mAFR_FSDKFace);
Log.d(TAG, "addFace MSG_EVENT_REG jxd");
ToastUtil.showToast(mContext.getString(R.string.register_face_success)+name);
} else {
//                Message reg = Message.obtain();
//                reg.what = MSG_CODE;
//                reg.arg1 = MSG_EVENT_NO_FEATURE;
//                mUIHandler.sendMessage(reg);
Log.d(TAG, "MSG_EVENT_NO_FEATURE");
ToastUtil.showToast(mContext.getString(R.string.no_feature));
}
error1 = engine1.AFR_FSDK_UninitialEngine();
Log.d(TAG, "AFR_FSDK_UninitialEngine : " + error1.getCode());
} else {
//            Message reg = Message.obtain();
//            reg.what = MSG_CODE;
//            reg.arg1 = MSG_EVENT_NO_FACE;
//            mUIHandler.sendMessage(reg);
Log.d(TAG, "MSG_EVENT_NO_FACE");
ToastUtil.showToast(mContext.getString(R.string.no_face));
}
err = engine.AFD_FSDK_UninitialFaceEngine();
Log.d(TAG, "AFD_FSDK_UninitialFaceEngine =" + err.getCode());
}

public static boolean isExternalStorageDocument(Uri uri) {
return "com.android.externalstorage.documents".equals(uri.getAuthority());
}

/**
* @param uri The Uri to check.
* @return Whether the Uri authority is DownloadsProvider.
*/
public static boolean isDownloadsDocument(Uri uri) {
return "com.android.providers.downloads.documents".equals(uri.getAuthority());
}

/**
* @param uri The Uri to check.
* @return Whether the Uri authority is MediaProvider.
*/
public static boolean isMediaDocument(Uri uri) {
return "com.android.providers.media.documents".equals(uri.getAuthority());
}

/**
* Get the value of the data column for this Uri. This is useful for
* MediaStore Uris, and other file-based ContentProviders.
*
* @param context The context.
* @param uri The Uri to query.
* @param selection (Optional) Filter used in the query.
* @param selectionArgs (Optional) Selection arguments used in the query.
* @return The value of the _data column, which is typically a file path.
*/
public static String getDataColumn(Context context, Uri uri, String selection,
String[] selectionArgs) {

Cursor cursor = null;
final String column = "_data";
final String[] projection = {
column
};

try {
cursor = context.getContentResolver().query(uri, projection, selection, selectionArgs,
null);
if (cursor != null && cursor.moveToFirst()) {
final int index = cursor.getColumnIndexOrThrow(column);
return cursor.getString(index);
}
} finally {
if (cursor != null)
cursor.close();
}
return null;
}

public String getPath(Uri uri) {
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.KITKAT) {
if (DocumentsContract.isDocumentUri(mContext, uri)) {
// ExternalStorageProvider
if (isExternalStorageDocument(uri)) {
final String docId = DocumentsContract.getDocumentId(uri);
final String[] split = docId.split(":");
final String type = split[0];

if ("primary".equalsIgnoreCase(type)) {
return Environment.getExternalStorageDirectory() + "/" + split[1];
}

// TODO handle non-primary volumes
} else if (isDownloadsDocument(uri)) {

final String id = DocumentsContract.getDocumentId(uri);
final Uri contentUri = ContentUris.withAppendedId(
Uri.parse("content://downloads/public_downloads"), Long.valueOf(id));

return getDataColumn(mContext, contentUri, null, null);
} else if (isMediaDocument(uri)) {
final String docId = DocumentsContract.getDocumentId(uri);
final String[] split = docId.split(":");
final String type = split[0];

Uri contentUri = null;
if ("image".equals(type)) {
contentUri = MediaStore.Images.Media.EXTERNAL_CONTENT_URI;
} else if ("video".equals(type)) {
contentUri = MediaStore.Video.Media.EXTERNAL_CONTENT_URI;
} else if ("audio".equals(type)) {
contentUri = MediaStore.Audio.Media.EXTERNAL_CONTENT_URI;
}

final String selection = "_id=?";
final String[] selectionArgs = new String[] {
split[1]
};

return getDataColumn(mContext, contentUri, selection, selectionArgs);
}
}
}
String[] proj = { MediaStore.Images.Media.DATA };
Cursor actualimagecursor = mContext.getContentResolver().query(uri, proj, null, null, null);
int actual_image_column_index = actualimagecursor.getColumnIndexOrThrow(MediaStore.Images.Media.DATA);
actualimagecursor.moveToFirst();
String img_path = actualimagecursor.getString(actual_image_column_index);
String end = img_path.substring(img_path.length() - 4);
if (0 != end.compareToIgnoreCase(".jpg") && 0 != end.compareToIgnoreCase(".png")) {
return null;
}
return img_path;
}

public void setFaceFunction()
{
openFace =  SharedPreferenceUtil.getBoolean(OPEN_FACE, Constant.FACE, false);
openFace = !openFace;
SharedPreferenceUtil.putBoolean(OPEN_FACE, Constant.FACE, openFace);
if(openFace)
{
ToastUtil.showToast(R.string.open_face);
}
else
{
ToastUtil.showToast(R.string.close_face);
}
}

public boolean getFaceIsOpen()
{
return openFace;
}