Add support for setting asset library group type.

dongle · Feb 5, 2013 · a37e558 · a37e558
1 parent 46c54d8
commit a37e558
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/DIYAssetPicker/DIYAssetPickerController.h b/DIYAssetPicker/DIYAssetPickerController.h
@@ -50,5 +50,6 @@ NSString *const DIYAssetPickerThumbnail;
 
 #pragma mark - Options
 @property (nonatomic, assign) DIYAssetPickerControllerAssetType           assetType;
+@property (nonatomic, assign) ALAssetsGroupType groupType;
 
 @end
diff --git a/DIYAssetPicker/DIYAssetPickerController.m b/DIYAssetPicker/DIYAssetPickerController.m
@@ -31,6 +31,7 @@ - (id)init
 {
     if (self = [super init]) {
         _assetType = DIYAssetPickerPhotoVideo;
+        _groupType = ALAssetsGroupSavedPhotos;
     }
 
     return self;
@@ -258,14 +259,21 @@ - (void)setAssetType:(DIYAssetPickerControllerAssetType)assetType
     [self.assetsTable reloadData];
 }
 
+- (void)setGroupType:(ALAssetsGroupType)groupType
+{
+    self->_groupType = groupType;
+    [self getAssetsArray];
+    [self.assetsTable reloadData];
+}
+
 #pragma mark - Utility
 
 - (void)getAssetsArray
 {
     [self.assetsArray removeAllObjects];
 
     [self.assetsLibrary
-     enumerateGroupsWithTypes:ALAssetsGroupSavedPhotos
+     enumerateGroupsWithTypes:self.groupType
      usingBlock:^(ALAssetsGroup *group, BOOL *stop) {
          if (self.assetType == DIYAssetPickerPhoto) {
              [group setAssetsFilter:[ALAssetsFilter allPhotos]];