提交历史代码

lib_face/src/main/java/com/sw/plate/utils/arcface/faceserver/FaceServer.java

@@ -621,4 +621,10 @@ public class FaceServer {
         rect.inset(-padding, -padding);
         return rect;
     }
+
+    public void registerFace(Context context, FaceEngine frEngine) {
+        List<FaceEntity> faceEntityList = FaceDatabase.getInstance(context).faceDao().getAllFaces();
+        registerFaceFeatureInfoListFromDb(frEngine, faceEntityList);
+    }
+
 }

lib_face/src/main/java/com/sw/plate/utils/arcface/viewmodel/RecognizeViewModel.java

@@ -347,7 +347,14 @@ public class RecognizeViewModel extends ViewModel implements RecognizeCallback {
     }
 
     public void refreshFaceList() {
-        FaceServer.getInstance().initFaceList(App.getContext(), frEngine, faceCount -> loadFaceList = true, true);
+        refreshFaceList(true);
+    }
+    public void refreshFaceList(boolean recognize) {
+        loadFaceList = false;
+        FaceServer.getInstance().initFaceList(App.getContext(), frEngine, faceCount -> {
+            Log.i(TAG, "unInitEngine: faceCount=" + faceCount);
+            loadFaceList = true;
+        }, recognize);
     }
 
     /**
@@ -596,7 +603,7 @@ public class RecognizeViewModel extends ViewModel implements RecognizeCallback {
             if (recognizeStatus != null) {
                 if (recognizeStatus == RequestFeatureStatus.FAILED) {
                     color = RecognizeColor.COLOR_FAILED;
-                }else if (recognizeStatus == RequestFeatureStatus.SUCCEED) {
+                } else if (recognizeStatus == RequestFeatureStatus.SUCCEED) {
                     color = RecognizeColor.COLOR_SUCCESS;
                 } else if (recognizeStatus == RequestFeatureStatus.TO_RETRY) {
                     color = RecognizeColor.COLOR_UNKNOWN;
@@ -674,4 +681,12 @@ public class RecognizeViewModel extends ViewModel implements RecognizeCallback {
         String[] size = ConfigUtil.getPreviewSize(App.getContext()).split("x");
         return new Point(Integer.parseInt(size[0]), Integer.parseInt(size[1]));
     }
+
+    public void registerFace() {
+        FaceServer.getInstance().registerFace(App.getContext(), frEngine);
+    }
+
+    public void clearAllFaces() {
+        FaceServer.getInstance().clearAllFaces();
+    }
 }