iOS 通过图片文件头，获取图片类型，以及图片尺寸

有时候，我们需要获取一个图片的格式和大小，但是却不想从内存中把它的整个文件从内存中读取出来

因为读取整个文件的内存耗费还是有些的，所以这么做通常不可取。

实在需要读取那也没办法。

但是其实是可以从文件中读取出来。 原因是图片文件的格式标准。

这些标准的存在使得可以只读取部分字段，

就能获知图片的格式。

先说图片的格式：

图片的格式存在 图片文件的前8个字节中。

以PNG 为例

PNG的前八个字节依次为：0x89  0x50 0x4E   0x47 0x0D 0x0A 0x1A 0x0A

读取PNG 图片前8个字节，判断一下就可以知道是否是PNG 图片。

其他图片的格式类似。

再来说说图片的大小。

图片的大小也是存在相关字段中

还是以PNG 为例，PNG 的图片大小，存储在第9个字节开始的8个字节中

前四个字节表示宽度，后四个字节表示高度。

PNG文件存储的长宽信息存储较简单。

但是JPG文件就比较复杂了。

首先JPG文件存储的东西是分块存储的。

很多很多的块。

这些块的开头几个字节，表示该块包含哪些信息。

其中，如果某个块的开头0xffc0 或者是 0xffc1或者是 0xffc2

那么该块将会存储 图片文件的 尺寸信息，也就是长宽。

如果不是的话，那么就是别的块。

除了块的开头两个字节是固定的 0xffxx信息，块开头的第三第四个字节表示该块的的长度。

我们可以根据该块的大小信息快速跳过此块，查找下一个块，这样能够过滤不含有

图片尺寸的信息。

下面为部分代码。

//
//  HBImageTypeSizeUtill.m
//  Pods
//
//  Created by wangdan on 16/9/1.
//
//

#import "HBImageTypeSizeUtil.h"
#import <UIKit/UIKit.h>

@implementation HBImageTypeSizeUtil

+ (HBImageType)imageTypeOfFilePath:(NSString *)filePath
{

HBImageType type = HBImageTypeUnKnow;
if (filePath.length == 0) {
return HBImageTypeUnKnow;
}
if (![[NSFileManager defaultManager] fileExistsAtPath:filePath]) {
return HBImageTypeUnKnow;
}
NSFileHandle *fileHandler  = [NSFileHandle fileHandleForReadingAtPath:filePath];
if (!fileHandler) {
return HBImageTypeUnKnow;
}

NSData *headData = [fileHandler readDataOfLength:8];
if (!headData.length == 8) {
[fileHandler closeFile];
return HBImageTypeUnKnow;
}

Byte *bytesArray = (Byte*)[headData bytes];
if (bytesArray[0] == 0x89 &&
bytesArray[1] == 0x50 &&  // P
bytesArray[2] == 0x4E &&  // N
bytesArray[3] == 0x47 &&  // G
bytesArray[4] == 0x0D &&
bytesArray[5] == 0x0A &&
bytesArray[6] == 0x1A &&
bytesArray[7] == 0x0A)
{
type = HBImageTypePNG;
}
else if (bytesArray[0] == 0xFF && bytesArray[1] == 0xD8)
{
type = HBImageTypeJPG;
}
else if (bytesArray[0] == 0x47 &&    // G
bytesArray[1] == 0x49 &&    // I
bytesArray[2] == 0x46 &&    // F
bytesArray[3] == 0x38 &&    // 8
(bytesArray[4] == 0x39 ||   // 9
bytesArray[4] == 0x37) &&  // 7
bytesArray[5] == 0x61)      // a
{
type = HBImageTypeGIF;
}
else if (bytesArray[0] == 0x42 &&   //B
bytesArray[1] == 0x4D)      //M
{
type = HBImageTypeBMP;
}

[fileHandler closeFile];

return type;

}

+ (CGSize)imagSizeOfFilePath:(NSString *)filePath
{
CGSize finalSize = CGSizeZero;
HBImageType type = [self imageTypeOfFilePath:filePath];
switch (type) {
case HBImageTypeJPG:
{
return  [self jpgImageSizeWithFilePath:filePath];
}
break;
case HBImageTypePNG:
{
NSData *data = [self fileHeaderData:8 seek:16 filePath:filePath];
finalSize =  [self pngImageSizeWithHeaderData:data];
}
break;
case HBImageTypeBMP:
{
NSData *data = [self fileHeaderData:8 seek:18 filePath:filePath];
finalSize = [self bmpImageSizeWithHeaderData:data];
}
break;
case HBImageTypeGIF:
{
NSData *data = [self fileHeaderData:4 seek:6 filePath:filePath];
finalSize = [self gifImageSizeWithHeaderData:data];
}
break;
case HBImageTypeUnKnow:
{
UIImage *image = [UIImage imageWithContentsOfFile:filePath];
if (image) {
finalSize =  image.size;
}
else {
finalSize = CGSizeZero;
}
}
break;
default:
finalSize = CGSizeZero;
break;
}

return finalSize;
}

+ (NSData*)fileHeaderData:(NSInteger)length seek:(NSInteger)seek filePath:(NSString *)filePath {
if (![[NSFileManager defaultManager] fileExistsAtPath:filePath]) {
return nil;
}
NSDictionary *fileAttribute = [[NSFileManager defaultManager] attributesOfItemAtPath:filePath error:nil];
NSUInteger fileSize = [fileAttribute fileSize];
if (length + seek > fileSize) {
return nil;
}

NSFileHandle *fileHandler = [NSFileHandle fileHandleForReadingAtPath:filePath];
[fileHandler seekToFileOffset:seek];
NSData *data = [fileHandler readDataOfLength:length];
[fileHandler closeFile];
return data;
}

+ (CGSize)gifImageSizeWithHeaderData:(NSData *)data
{
if (data.length != 4) {
return CGSizeZero;
}
unsigned char w1 = 0, w2 = 0;
[data getBytes:&w1 range:NSMakeRange(0, 1)];
[data getBytes:&w2 range:NSMakeRange(1, 1)];
short w = w1 + (w2 << 8);
unsigned char h1 = 0, h2 = 0;
[data getBytes:&h1 range:NSMakeRange(2, 1)];
[data getBytes:&h2 range:NSMakeRange(3, 1)];
short h = h1 + (h2 << 8);
return CGSizeMake(w, h);
}

//JPG文件数据，分很多很多的数据段， 并且每个数据段都会以 0xFF开头
//找到一个数据断后，如果数据段的开头是0xffc0，那么该数据段将会存储 图片的尺寸信息
//否则0xffc0 后面紧跟的两个字段，存储的是当前这个数据段的长度，可跳过当前的数据段
//然后寻找下一个数据段，然后查看是否有图片尺寸信息
+ (CGSize)jpgImageSizeWithFilePath:(NSString *)filePath
{
if (!filePath.length) {
return CGSizeZero;
}
if (![[NSFileManager defaultManager] fileExistsAtPath:filePath]) {
return  CGSizeZero;
}

NSFileHandle *fileHandle = [NSFileHandle fileHandleForReadingAtPath:filePath];
NSUInteger offset = 2;
NSUInteger length = 0;
while (1) {
[fileHandle seekToFileOffset:offset];
length = 4;
NSData *data = [fileHandle readDataOfLength:length];
if (data.length != length) {
break;
}
offset += length;
int marker,code;
NSUInteger newLength;
unsigned char value1,value2,value3,value4;
[data getBytes:&value1 range:NSMakeRange(0, 1)];
[data getBytes:&value2 range:NSMakeRange(1, 1)];
[data getBytes:&value3 range:NSMakeRange(2, 1)];
[data getBytes:&value4 range:NSMakeRange(3, 1)];
marker = value1;
code = value2;
newLength = (value3 << 8) + value4;
if (marker != 0xff) {
[fileHandle closeFile];
return CGSizeZero;
}

if (code >= 0xc0 && code <= 0xc3) {
length = 5;
[fileHandle seekToFileOffset:offset];
NSData *data =[fileHandle readDataOfLength:length];
if (data.length != length) {
break;
}
Byte *bytesArray = (Byte*)[data bytes];
NSUInteger height = ((unsigned char)bytesArray[1] << 8) + (unsigned char)bytesArray[2];
NSUInteger width =  ((unsigned char)bytesArray[3] << 8) + (unsigned char)bytesArray[4];
[fileHandle closeFile];
return CGSizeMake(width, height);
}
else {
offset += newLength;
offset -=2;
}
}
[fileHandle closeFile];
UIImage *image = [UIImage imageWithContentsOfFile:filePath];
if (image) {
CGSizeMake((NSInteger)image.size.width, (NSInteger)image.size.height);
}
return CGSizeZero;

}

+ (CGSize)pngImageSizeWithHeaderData:(NSData *)data
{
if (data.length != 8) {
return CGSizeZero;
}
unsigned char w1 = 0, w2 = 0, w3 = 0, w4 = 0;
[data getBytes:&w1 range:NSMakeRange(0, 1)];
[data getBytes:&w2 range:NSMakeRange(1, 1)];
[data getBytes:&w3 range:NSMakeRange(2, 1)];
[data getBytes:&w4 range:NSMakeRange(3, 1)];
int w = (w1 << 24) + (w2 << 16) + (w3 << 8) + w4;

unsigned char h1 = 0, h2 = 0, h3 = 0, h4 = 0;
[data getBytes:&h1 range:NSMakeRange(4, 1)];
[data getBytes:&h2 range:NSMakeRange(5, 1)];
[data getBytes:&h3 range:NSMakeRange(6, 1)];
[data getBytes:&h4 range:NSMakeRange(7, 1)];
int h = (h1 << 24) + (h2 << 16) + (h3 << 8) + h4;
return CGSizeMake(w, h);
}

+ (CGSize)bmpImageSizeWithHeaderData:(NSData *)data {
if (data.length != 8) {
return CGSizeZero;
}
unsigned char w1 = 0, w2 = 0, w3 = 0, w4 = 0;
[data getBytes:&w1 range:NSMakeRange(0, 1)];
[data getBytes:&w2 range:NSMakeRange(1, 1)];
[data getBytes:&w3 range:NSMakeRange(2, 1)];
[data getBytes:&w4 range:NSMakeRange(3, 1)];
int w = w1 + (w2 << 8) + (w3 << 16) + (w4 << 24);
unsigned char h1 = 0, h2 = 0, h3 = 0, h4 = 0;
[data getBytes:&h1 range:NSMakeRange(4, 1)];
[data getBytes:&h2 range:NSMakeRange(5, 1)];
[data getBytes:&h3 range:NSMakeRange(6, 1)];
[data getBytes:&h4 range:NSMakeRange(7, 1)];
int h = h1 + (h2 << 8) + (h3 << 16) + (h4 << 24);
return CGSizeMake(w, h);
}

@end

完整的代码见下面的demo链接

demo

也可以参考我的github

地址