Add restartVideoCapture method

smule · Oct 12, 2017 · c78b6ac · c78b6ac
1 parent fa22525
commit c78b6ac
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/Source/PBJVision.h b/Source/PBJVision.h
@@ -247,6 +247,7 @@ static CGFloat const PBJVideoBitRate1280x750 = 5000000 * 8;     // 40bmps
 - (void)pauseVideoCapture;
 - (void)resumeVideoCapture;
 - (void)endVideoCapture;
+- (void)restartVideoCapture;
 - (void)cancelVideoCapture;
 - (void)setPreviewFrameRate:(int)frameRate;
 

diff --git a/Source/PBJVision.m b/Source/PBJVision.m
@@ -233,6 +233,8 @@ @interface PBJVision () <
 @property (nonatomic, retain) __attribute__((NSObject)) CMFormatDescriptionRef outputVideoFormatDescription;
 @property (nonatomic, retain) __attribute__((NSObject)) CMFormatDescriptionRef outputAudioFormatDescription;
 
+@property (nonatomic) BOOL isRestartingVideoCapture;
+
 @end
 
 
@@ -2029,6 +2031,17 @@ - (void)endVideoCapture
     }];
 }
 
+- (void)restartVideoCapture
+{
+    if (self.isRestartingVideoCapture)
+    {
+        return;
+    }
+
+    self.isRestartingVideoCapture = YES;
+    [self endVideoCapture];
+}
+
 - (void)cancelVideoCapture
 {
     DLog(@"cancel video capture");
@@ -2673,7 +2686,11 @@ - (void)mediaWriterDidFinishRecording:(PBJMediaWriter *)mediaWriter
             }
         }];
 
-
+        if (self.isRestartingVideoCapture)
+        {
+            self.isRestartingVideoCapture = NO;
+            [self startVideoCapture];
+        }
 
     }];
 }