Filter capture media depending of filter.

Author: SergioEstevao

Pod/Classes/WPMediaCollectionViewController.m

@@ -559,9 +559,20 @@ - (BOOL)isMediaDeviceAvailable
 - (void)showMediaCaptureViewController
 {
     UIImagePickerController *imagePickerController = [[UIImagePickerController alloc] init];
-    imagePickerController.mediaTypes =
-        [UIImagePickerController availableMediaTypesForSourceType:
-                                     UIImagePickerControllerSourceTypeCamera];
+    NSMutableSet *mediaTypes = [NSMutableSet setWithArray:[UIImagePickerController availableMediaTypesForSourceType:
+                           UIImagePickerControllerSourceTypeCamera]];
+    switch (self.filter) {
+        case(WPMediaTypeImage): {
+            [mediaTypes intersectSet:[NSSet setWithArray:@[(__bridge NSString *)kUTTypeImage]]];
+        } break;
+        case(WPMediaTypeVideo): {
+            [mediaTypes intersectSet:[NSSet setWithArray:@[(__bridge NSString *)kUTTypeMovie]]];
+        } break;
+        default: {
+            //Don't intersect at all
+        }
+    }
+    imagePickerController.mediaTypes = [mediaTypes allObjects];
     imagePickerController.delegate = self;
     imagePickerController.sourceType = UIImagePickerControllerSourceTypeCamera;
     imagePickerController.modalTransitionStyle = UIModalTransitionStyleCoverVertical;