Added a new property for single selection image size.

YangMingShan/YMSPhotoPicker/Public/YMSPhotoPickerViewController.h

@@ -37,6 +37,11 @@
  */
 @property (nonatomic, readonly) YMSPhotoPickerTheme *theme;
 
+/**
+ *  @brief Use this property to customize the returned item type for single selection. YES for UIImage, NO for PHAsset. Default value is YES.
+ */
+@property (nonatomic, assign) BOOL shouldReturnImageForSingleSelection;
+
 @end
 
 /**

YangMingShan/YMSPhotoPicker/Public/YMSPhotoPickerViewController.m

@@ -61,6 +61,7 @@ - (instancetype)init
     if (self) {
         self.selectedPhotos = [NSMutableArray array];
         self.numberOfPhotoToSelect = 1;
+        self.shouldReturnImageForSingleSelection = YES;
     }
     return self;
 }
@@ -214,26 +215,32 @@ - (void)collectionView:(UICollectionView *)collectionView didSelectItemAtIndexPa
         [self yms_presentCameraCaptureViewWithDelegate:self];
     }
     else if (NO == self.allowsMultipleSelection) {
-
-        PHFetchResult *fetchResult = self.currentCollectionItem[@"assets"];
-        PHAsset *asset = fetchResult[indexPath.item-1];
-
-        // Prepare the options to pass when fetching the live photo.
-        PHImageRequestOptions *options = [[PHImageRequestOptions alloc] init];
-        options.deliveryMode = PHImageRequestOptionsDeliveryModeHighQualityFormat;
-        options.networkAccessAllowed = YES;
-        options.resizeMode = PHImageRequestOptionsResizeModeExact;
-
-        CGSize targetSize = CGSizeMake(asset.pixelWidth, asset.pixelHeight);
-
-        [self.imageManager requestImageForAsset:asset targetSize:targetSize contentMode:PHImageContentModeAspectFill options:options resultHandler:^(UIImage *image, NSDictionary *info) {
-            if (image && [self.delegate respondsToSelector:@selector(photoPickerViewController:didFinishPickingImage:)]) {
-                [self.delegate photoPickerViewController:self didFinishPickingImage:[self yms_orientationNormalizedImage:image]];
-            }
-            else {
-                [self dismiss:nil];
-            }
-        }];
+        if (NO == self.shouldReturnImageForSingleSelection) {
+            PHFetchResult *fetchResult = self.currentCollectionItem[@"assets"];
+            PHAsset *asset = fetchResult[indexPath.item-1];
+            [self.selectedPhotos addObject:asset];
+            [self finishPickingPhotos:nil];
+        } else {
+            PHFetchResult *fetchResult = self.currentCollectionItem[@"assets"];
+            PHAsset *asset = fetchResult[indexPath.item-1];
+
+            // Prepare the options to pass when fetching the live photo.
+            PHImageRequestOptions *options = [[PHImageRequestOptions alloc] init];
+            options.deliveryMode = PHImageRequestOptionsDeliveryModeHighQualityFormat;
+            options.networkAccessAllowed = YES;
+            options.resizeMode = PHImageRequestOptionsResizeModeExact;
+
+            CGSize targetSize = CGSizeMake(asset.pixelWidth, asset.pixelHeight);
+
+            [self.imageManager requestImageForAsset:asset targetSize:targetSize contentMode:PHImageContentModeAspectFill options:options resultHandler:^(UIImage *image, NSDictionary *info) {
+                if (image && [self.delegate respondsToSelector:@selector(photoPickerViewController:didFinishPickingImage:)]) {
+                    [self.delegate photoPickerViewController:self didFinishPickingImage:[self yms_orientationNormalizedImage:image]];
+                }
+                else {
+                    [self dismiss:nil];
+                }
+            }];
+        }
     }
     else {
         PHFetchResult *fetchResult = self.currentCollectionItem[@"assets"];