从NSData或UIImage中查找图像类型

时间:2010-11-10 17:39:32

标签: iphone ios uiimage nsdata persist

我正在从第三方提供的网址加载图片。 URL上没有文件扩展名(或文件名)(因为它是一个模糊的URL)。我可以从中获取数据(以NSData的形式)并将其加载到UIImage中并显示它。

我想将此数据保存到文件中。但是,我不知道数据的格式是什么(PNG,JPG,BMP)?我认为它是JPG(因为它是来自网络的图像)但是有一种程序化的方法可以找到它吗?我查看了StackOverflow和文档,但未能找到任何内容。

TIA。


编辑:我真的需要文件扩展名吗?我将它持久化到外部存储(Amazon S3),但考虑到它总是在iOS或浏览器的上下文中使用(两者在没有扩展的情况下解释数据似乎都很好)也许这不是问题

11 个答案:

答案 0 :(得分:140)

如果您有图像文件的NSData,那么您可以通过查看第一个字节来猜测内容类型:

+ (NSString *)contentTypeForImageData:(NSData *)data {
    uint8_t c;
    [data getBytes:&c length:1];

    switch (c) {
    case 0xFF:
        return @"image/jpeg";
    case 0x89:
        return @"image/png";
    case 0x47:
        return @"image/gif";
    case 0x49:
    case 0x4D:
        return @"image/tiff";
    }
    return nil;
}

答案 1 :(得分:24)

改进wl.'s answer,这是一种基于签名预测图像MIME类型的更加扩展和精确的方法。代码很大程度上受到了php ext/standard/image.c的启发。

- (NSString *)mimeTypeByGuessingFromData:(NSData *)data {

    char bytes[12] = {0};
    [data getBytes:&bytes length:12];

    const char bmp[2] = {'B', 'M'};
    const char gif[3] = {'G', 'I', 'F'};
    const char swf[3] = {'F', 'W', 'S'};
    const char swc[3] = {'C', 'W', 'S'};
    const char jpg[3] = {0xff, 0xd8, 0xff};
    const char psd[4] = {'8', 'B', 'P', 'S'};
    const char iff[4] = {'F', 'O', 'R', 'M'};
    const char webp[4] = {'R', 'I', 'F', 'F'};
    const char ico[4] = {0x00, 0x00, 0x01, 0x00};
    const char tif_ii[4] = {'I','I', 0x2A, 0x00};
    const char tif_mm[4] = {'M','M', 0x00, 0x2A};
    const char png[8] = {0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a};
    const char jp2[12] = {0x00, 0x00, 0x00, 0x0c, 0x6a, 0x50, 0x20, 0x20, 0x0d, 0x0a, 0x87, 0x0a};


    if (!memcmp(bytes, bmp, 2)) {
        return @"image/x-ms-bmp";
    } else if (!memcmp(bytes, gif, 3)) {
        return @"image/gif";
    } else if (!memcmp(bytes, jpg, 3)) {
        return @"image/jpeg";
    } else if (!memcmp(bytes, psd, 4)) {
        return @"image/psd";
    } else if (!memcmp(bytes, iff, 4)) {
        return @"image/iff";
    } else if (!memcmp(bytes, webp, 4)) {
        return @"image/webp";
    } else if (!memcmp(bytes, ico, 4)) {
        return @"image/vnd.microsoft.icon";
    } else if (!memcmp(bytes, tif_ii, 4) || !memcmp(bytes, tif_mm, 4)) {
        return @"image/tiff";
    } else if (!memcmp(bytes, png, 8)) {
        return @"image/png";
    } else if (!memcmp(bytes, jp2, 12)) {
        return @"image/jp2";
    }

    return @"application/octet-stream"; // default type

}

上述方法可识别以下图像类型:

  • image/x-ms-bmp(bmp)
  • image/gif(gif)
  • image/jpeg(jpg,jpeg)
  • image/psd(psd)
  • image/iff(iff)
  • image/webp(webp)
  • image/vnd.microsoft.icon(ico)
  • image/tiff(tif,tiff)
  • image/png(png)
  • image/jp2(jp2)

不幸的是,没有简单的方法可以从UIImage实例获取此类信息,因为无法访问其封装的位图数据。

答案 2 :(得分:8)

如果您要从URL检索图像,那么可能您可以检查HTTP响应标头。 Content-Type标头是否包含有用的内容? (我想它会因为浏览器可能能够正确显示图像,并且只有在适当设置内容类型时才能这样做)

答案 3 :(得分:8)

@Tai Le的Swift 3解决方案是将整个数据分配到字节数组中。如果图像较大,则可能导致崩溃。这个解决方案只分配单个字节:

import Foundation

public extension Data {
    var fileExtension: String {
        var values = [UInt8](repeating:0, count:1)
        self.copyBytes(to: &values, count: 1)

        let ext: String
        switch (values[0]) {
        case 0xFF:
            ext = ".jpg"
        case 0x89:
            ext = ".png"
        case 0x47:
            ext = ".gif"
        case 0x49, 0x4D :
            ext = ".tiff"
        default:
            ext = ".png"
        }
        return ext
    }
}

答案 4 :(得分:5)

Swift3版本:

let data: Data = UIImagePNGRepresentation(yourImage)!

extension Data {
    var format: String {
        let array = [UInt8](self)
        let ext: String
        switch (array[0]) {
        case 0xFF:
            ext = "jpg"
        case 0x89:
            ext = "png"
        case 0x47:
            ext = "gif"
        case 0x49, 0x4D :
            ext = "tiff"
        default:
            ext = "unknown"
        }
        return ext
    }
}

答案 5 :(得分:2)

要从UIImage获取图像类型,可以从基础Quartz图像数据获取类型标识符(UTI):

extension UIImage {
    var typeIdentifier: String? {
        cgImage?.utType as String?
    }
}

要从URL获取图像类型标识符,将取决于URL是否指向本地资源:

extension URL {
    // for local resources (fileURLs)
    var typeIdentifier: String? { (try? resourceValues(forKeys: [.typeIdentifierKey]))?.typeIdentifier }
    // for non local resources (web) you can get it asyncronously
    func asyncTypeIdentifier(completion: @escaping ((String?, Error?) -> Void)) {
        var request = URLRequest(url: self)
        request.httpMethod = "HEAD"
        URLSession.shared.dataTask(with: request) { _ , response , error in
            completion((response as? HTTPURLResponse)?.mimeType, error)
        }.resume()
    }
}

let imageURL = URL(string: "https://i.stack.imgur.com/varL9.jpg")!
imageURL.asyncTypeIdentifier { typeIdentifier, error in
    guard let typeIdentifier = typeIdentifier, error == nil else { return }
    print("typeIdentifier:", typeIdentifier)
}

答案 6 :(得分:1)

如果它对你真的很重要,我相信你必须检查字节流。 JPEG将以字节FF D8开头。 PNG将以89 50 4E 47 0D 0A 1A 0A开始。我不知道BMP是否有类似的标题,但我认为你不太可能在2010年遇到网络上的那些。

但这对你真的很重要吗?难道你不能把它当作一个未知的形象,并让Cocoa Touch做这项工作吗?

答案 7 :(得分:1)

接受答案的另一种方法是使用image I/O frameWork检查图像的UTI。您可以通过UTI实现图像类型。 试试这个:

CGImageSourceRef imgSrc = CGImageSourceCreateWithData((CFDataRef)data, NULL);
NSString *uti = (NSString*)CGImageSourceGetType(imgSrc);
NSLog(@"%@",uti);

例如,GIF图像的UTI是“com.compuserve.gif”,而PNG图像的UTI是“public.png”。但是您无法从image I/O frameWork无法识别的图像中获得UTI。 / p>

答案 8 :(得分:1)

我基于@ccoroom的改进解决方案

//  Data+ImageContentType.swift

import Foundation

extension Data {  
    enum ImageContentType: String {
        case jpg, png, gif, tiff, unknown

        var fileExtension: String {
            return self.rawValue
        }
    }

    var imageContentType: ImageContentType {

        var values = [UInt8](repeating: 0, count: 1)

        self.copyBytes(to: &values, count: 1)

        switch (values[0]) {
        case 0xFF:
            return .jpg
        case 0x89:
            return .png
        case 0x47:
           return .gif
        case 0x49, 0x4D :
           return .tiff
        default:
            return .unknown
        }
    }
}

一些用法示例:

//load some image
do {
    let imageData = try Data(contentsOf: URL(string: "https://myServer/images/test.jpg")!)
} catch {
    print("Unable to load image: \(error)")
}

//content type check
guard [Data.ImageContentType.jpg,
       Data.ImageContentType.png].contains(imageData.imageContentType) else {
    print("unsupported image type")
            return
        }

//set file extension
let image = "myImage.\(imageData.imageContentType.fileExtension)" //myImage.jpg

答案 9 :(得分:0)

对每种已知图像格式实施签名检查。这是一个快速的Objective-C函数,用于PNG数据:

// Verify that NSData contains PNG data by checking the signature

- (BOOL) isPNGData:(NSData*)data
{
  // Verify that the PNG file signature matches

  static const
  unsigned char   png_sign[8] = {137, 80, 78, 71, 13, 10, 26, 10};

  unsigned char   sig[8] = {0, 0, 0, 0, 0, 0, 0, 0};

  if ([data length] <= 8) {
    return FALSE;
  }

  [data getBytes:&sig length:8];

  BOOL same = (memcmp(sig, png_sign, 8) == 0);

  return same;
}

答案 10 :(得分:0)

我建立了一个库来检查NSData的图像类型:

https://github.com/sweetmandm/ImageFormatInspector