chore(face-detector): make face detection stoppage smoother

react-native-camera · Feb 19, 2018 · 3b3c38d · 3b3c38d
1 parent 19b958d
commit 3b3c38d
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/ios/FaceDetector/RNFaceDetectorManager.m b/ios/FaceDetector/RNFaceDetectorManager.m
@@ -68,9 +68,13 @@ - (void)setIsEnabled:(id)json
         _faceDetecting = newFaceDetecting;
         [self _runBlockIfQueueIsPresent:^{
             if ([self isDetectingFaces]) {
-                [self tryEnablingFaceDetection];
+                if (_dataOutput) {
+                    [self _setConnectionsEnabled:true];
+                } else {
+                    [self tryEnablingFaceDetection];
+                }
             } else {
-                [self stopFaceDetection];
+                [self _setConnectionsEnabled:false];
             }
         }];
     }
@@ -155,6 +159,16 @@ - (void)stopFaceDetection
 
 # pragma mark Private API
 
+- (void)_setConnectionsEnabled:(BOOL)enabled
+{
+    if (!_dataOutput) {
+        return;
+    }
+    for (AVCaptureConnection *connection in _dataOutput.connections) {
+        connection.enabled = enabled;
+    }
+}
+
 - (void)_resetFaceDetector
 {
     [self stopFaceDetection];