优化一些细节

banchichen · Jul 4, 2018 · 38e315e · 38e315e
1 parent 5f6b34f
commit 38e315e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/TZImagePickerController/TZImagePickerController/TZPhotoPreviewController.m b/TZImagePickerController/TZImagePickerController/TZPhotoPreviewController.m
@@ -345,6 +345,12 @@ - (void)select:(UIButton *)selectButton {
 - (void)backButtonClick {
     if (self.navigationController.childViewControllers.count < 2) {
         [self.navigationController dismissViewControllerAnimated:YES completion:nil];
+        if ([self.navigationController isKindOfClass: [TZImagePickerController class]]) {
+            TZImagePickerController *nav = (TZImagePickerController *)self.navigationController;
+            if (nav.imagePickerControllerDidCancelHandle) {
+                nav.imagePickerControllerDidCancelHandle();
+            }
+        }
         return;
     }
     [self.navigationController popViewControllerAnimated:YES];