【iOS开发】图像处理之获取RGB

截取自雾霾检测机

-(float) returnGrayVale:(UInt8*)buffer withBytesPerRow:(size_t)bytesPerRow withX:(int)x withY:(int)y{

    UInt8*  tmp;

    tmp = buffer + y * bytesPerRow + x * 4; // RGBAの4つ値をもっているので、1ピクセルごとに*4してずらす

    

    // 获取图像的RGB

    UInt8 red,green,blue;

    red = *(tmp + 0);

    green = *(tmp + 1);

    blue = *(tmp + 2);

    

    //获取灰度值

    float grayValue = [self returnGrayValueWithRGB:red withGreen:green withBlue:blue];

    return grayValue;

}

-(float)ClearIndexBySobel:(UIImage*)sourceImage fromX:(int)Xpoint fromY:(int)Ypoint zoneWidth:(int)Width zoneHeihgt:(int)Height{

    

    int imageWidth = sourceImage.size.width;

    int imageHeight = sourceImage.size.height;

    

    //防止越界处理

    if (Xpoint<=0 || Xpoint>=imageWidth-1 || Ypoint<=0 || Ypoint>=imageHeight-1) {

        return -1;

    }

    

    

    //获取ref对象

    CGImageRef  imageRef;

    imageRef = sourceImage.CGImage;

    

    // 一行有多少个字节

    size_t                  bytesPerRow;

    bytesPerRow = CGImageGetBytesPerRow(imageRef);

    

    CGDataProviderRef   dataProvider;

    dataProvider = CGImageGetDataProvider(imageRef);

    

    CFDataRef   data;

    UInt8*      buffer;

    data = CGDataProviderCopyData(dataProvider);

    buffer = (UInt8*)CFDataGetBytePtr(data);

    //循环计算区域像素点获得总的梯度值

    float sum_grad = 0;

    int count = 0;

    for (int y = Ypoint; y < Ypoint + Height; y++) {

        for (int x = Xpoint; x < Xpoint + Width; x++) {

            

            //计算周边8个点的灰度值

            float gray_00 = [self returnGrayVale:buffer withBytesPerRow:bytesPerRow withX:x-1 withY:y-1];

            float gray_01 = [self returnGrayVale:buffer withBytesPerRow:bytesPerRow withX:x-1 withY:y];

            float gray_02 = [self returnGrayVale:buffer withBytesPerRow:bytesPerRow withX:x-1 withY:y+1];

            

            float gray_10 = [self returnGrayVale:buffer withBytesPerRow:bytesPerRow withX:x withY:y-1];

            float gray_12 = [self returnGrayVale:buffer withBytesPerRow:bytesPerRow withX:x withY:y+1];

            

            float gray_20 = [self returnGrayVale:buffer withBytesPerRow:bytesPerRow withX:x+1 withY:y-1];

            float gray_21 = [self returnGrayVale:buffer withBytesPerRow:bytesPerRow withX:x+1 withY:y];

            float gray_22 = [self returnGrayVale:buffer withBytesPerRow:bytesPerRow withX:x+1 withY:y+1];

            

            //计算梯度值

            float grad_x = gray_00-gray_02+2*gray_10-2*gray_12+gray_20-gray_22;

            float grad_y = gray_00+2*gray_01+gray_02-gray_20-2*gray_21-gray_22;

            

            float grad=sqrt(grad_x*grad_x+grad_y*grad_y);

            sum_grad +=grad;

            count++;

        }

    }

    

    if(count<=0){

        return 0;

    }

    

    float clearIndex = sum_grad / count;

    

    dataProvider = nil;

    data = nil;

    buffer = nil;

    

    return clearIndex;

}