Поиск типа изображения из NSData или UIImage
Я загружаю изображение с URL, предоставленного третьей стороной. В URL-адресе отсутствует расширение файла (или имя файла) (так как это скрытый URL-адрес). Я могу взять данные из этого (в форме NSData) и загрузить их в UIImage и отобразить его нормально.
Я хочу сохранить эти данные в файл. Тем не менее, я не знаю, в каком формате данные (PNG, JPG, BMP)? Я предполагаю, что это JPG (так как это изображение из Интернета), но есть ли программный способ узнать наверняка? Я осмотрел Stackru и документацию и не смог ничего найти.
ТИА.
Изменить: мне действительно нужно расширение файла? Я сохраняю его во внешнем хранилище (Amazon S3), но, учитывая, что он всегда будет использоваться в контексте iOS или браузера (оба из них кажутся хорошими в интерпретации данных без расширения), возможно, это не проблема,
9 ответов
Если у вас есть 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;
}
Улучшение ответа WL. Вот гораздо более расширенный и точный способ предсказать тип 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(СДП)image/iff(МФЛ)image/webp(WebP)image/vnd.microsoft.icon(ICO)image/tiff(TIF, TIFF)image/png(PNG)image/jp2(JP2)
К сожалению, не существует простого способа получить такую информацию от UIImage экземпляр, потому что его инкапсулированные данные растрового изображения не могут быть доступны.
Решением @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
}
}
Если вы извлекаете изображение из URL, то, вероятно, вы можете проверить заголовки HTTP-ответа. Ли Content-Type заголовок содержать что нибудь полезное? (Я полагаю, что так будет, поскольку браузер, вероятно, сможет правильно отображать изображение, и он сможет это сделать, только если тип содержимого установлен правильно)
Версия 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
}
}
Альтернативой принятого ответа является проверка UTI изображения с image I/O frameWork, Вы можете добиться изображения типа формы UTI. попробуй это:
CGImageSourceRef imgSrc = CGImageSourceCreateWithData((CFDataRef)data, NULL);
NSString *uti = (NSString*)CGImageSourceGetType(imgSrc);
NSLog(@"%@",uti);
Например, UTI изображения GIF - это "com.compuserve.gif", а UTI изображения PNG - "public.png". Но вы не можете получить UTI из изображения, которое image I/O frameWork не распознается
Чтобы получить тип изображения из UIImage вы можете получить идентификатор типа (UTI) из базовых данных изображения Quartz:
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://stackru.com/images/673d858dae84d6cc6c2a964c015f8d4a15825ba4.jpg")!
imageURL.asyncTypeIdentifier { typeIdentifier, error in
guard let typeIdentifier = typeIdentifier, error == nil else { return }
print("typeIdentifier:", typeIdentifier)
}
Мое улучшенное решение на основе @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
Я сделал библиотеку, чтобы проверить тип изображения NSData:
Реализуйте проверку подписи для каждого известного формата изображения. Вот быстрая функция 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;
}
Если это действительно имеет значение для вас, я думаю, вам придется изучить этот поток. JPEG начнется с байтов FF D8. PNG начнется с 89 50 4E 47 0D 0A 1A 0A. Я не знаю, есть ли у BMP похожий заголовок, но я не думаю, что вы слишком вероятно столкнетесь с такими в Интернете в 2010 году.
Но имеет ли это для вас значение? Разве вы не можете просто воспринимать это как неизвестное изображение и позволить Cocoa Touch делать всю работу?