获取bundle目录下的所有图片文件名

　　今天在写代码时候，偶然发现自己忘记了一些oc的基础知识（这里指的是获取bundle目录下的所有图片），感到很不爽。在百度了几次，发现自己的领悟能力实在不行，感觉萌萌的::>_<::

好了，进入正题。怎么获取bundle目录下的所有文件图片呢？上代码（说的不好，还请大牛多多指正）：

//1   获取bundle目录文件路径
    NSString *bundleDir = [[NSBundle mainBundle] bundlePath];
    NSLog(@"bundleDir   %@", bundleDir);  //   文件路径名

//2   获取目录下的所有文件名
 NSFileManager *fm = [NSFileManager defaultManager];
    NSArray *files = [fm subpathsAtPath:bundleDir];
    NSLog(@"files   %@", files);

    NSMutableArray *array = [NSMutableArray array];  //  创建一个可变数组，接收取出的文件名

//3  开始判断，获取 图片
     for (NSString *imageName in files) {
//        判断字符串 是否以 “.png结尾”
          if ([imageName hasSuffix:@".png"]) {
              NSString *path = [[[NSBundle mainBundle] bundlePath]stringByAppendingPathComponent:imageName];
              [array addObject:path];
          }
          if ([imageName hasSuffix:@".jpg"]) {
              NSString *path = [[[NSBundle mainBundle] bundlePath]stringByAppendingPathComponent:imageName];
             [array addObject:path];
          }
      }
    NSLog(@"array content are %@", array);  // 结果