Как получить пиксельные данные из UIImage (Cocoa Touch) или CGImage (Core Graphics)? - PullRequest
Новая
       86

Как получить пиксельные данные из UIImage (Cocoa Touch) или CGImage (Core Graphics)?

194 голосов
/ 15 января 2009

У меня есть UIImage (Cocoa Touch). Исходя из этого, я счастлив получить CGImage или что-нибудь еще, что вы хотели бы, что доступно. Я хотел бы написать эту функцию: 

- (int)getRGBAFromImage:(UIImage *)image atX:(int)xx andY:(int)yy {
  // [...]
  // What do I want to read about to help
  // me fill in this bit, here?
  // [...]

  int result = (red << 24) | (green << 16) | (blue << 8) | alpha;
  return result;
}

Спасибо!

Ответы [ 9 ]

246 голосов
/ 12 августа 2009

К вашему сведению, я объединил ответ Керемка с моей первоначальной схемой, убрал опечатки, обобщил его, чтобы получить массив цветов, и получил все, что нужно скомпилировать. Вот результат: 

+ (NSArray*)getRGBAsFromImage:(UIImage*)image atX:(int)x andY:(int)y count:(int)count
{
    NSMutableArray *result = [NSMutableArray arrayWithCapacity:count];

    // First get the image into your data buffer
    CGImageRef imageRef = [image CGImage];
    NSUInteger width = CGImageGetWidth(imageRef);
    NSUInteger height = CGImageGetHeight(imageRef);
    CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
    unsigned char *rawData = (unsigned char*) calloc(height * width * 4, sizeof(unsigned char));
    NSUInteger bytesPerPixel = 4;
    NSUInteger bytesPerRow = bytesPerPixel * width;
    NSUInteger bitsPerComponent = 8;
    CGContextRef context = CGBitmapContextCreate(rawData, width, height,
                    bitsPerComponent, bytesPerRow, colorSpace,
                    kCGImageAlphaPremultipliedLast | kCGBitmapByteOrder32Big);
    CGColorSpaceRelease(colorSpace);

    CGContextDrawImage(context, CGRectMake(0, 0, width, height), imageRef);
    CGContextRelease(context);

    // Now your rawData contains the image data in the RGBA8888 pixel format.
    NSUInteger byteIndex = (bytesPerRow * y) + x * bytesPerPixel;
    for (int i = 0 ; i < count ; ++i)
    {
        CGFloat alpha = ((CGFloat) rawData[byteIndex + 3] ) / 255.0f;
        CGFloat red   = ((CGFloat) rawData[byteIndex]     ) / alpha;
        CGFloat green = ((CGFloat) rawData[byteIndex + 1] ) / alpha;
        CGFloat blue  = ((CGFloat) rawData[byteIndex + 2] ) / alpha;
        byteIndex += bytesPerPixel;

        UIColor *acolor = [UIColor colorWithRed:red green:green blue:blue alpha:alpha];
        [result addObject:acolor];
    }

  free(rawData);

  return result;
}
47 голосов
/ 16 января 2009

Один из способов сделать это - нарисовать изображение в растровом контексте, который поддерживается данным буфером для данного цветового пространства (в данном случае это RGB): (обратите внимание, что это скопирует данные изображения в этот буфер, поэтому вы хотите кэшировать его вместо выполнения этой операции каждый раз, когда вам нужно получить значения в пикселях)

См. Пример ниже: 

// First get the image into your data buffer
CGImageRef image = [myUIImage CGImage];
NSUInteger width = CGImageGetWidth(image);
NSUInteger height = CGImageGetHeight(image);
CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
unsigned char *rawData = malloc(height * width * 4);
NSUInteger bytesPerPixel = 4;
NSUInteger bytesPerRow = bytesPerPixel * width;
NSUInteger bitsPerComponent = 8;
CGContextRef context = CGBitmapContextCreate(rawData, width, height, bitsPerComponent, bytesPerRow, colorSpace, kCGImageAlphaPremultipliedLast | kCGBitmapByteOrder32Big);
CGColorSpaceRelease(colorSpace);

CGContextDrawImage(context, CGRectMake(0, 0, width, height));
CGContextRelease(context);

// Now your rawData contains the image data in the RGBA8888 pixel format.
int byteIndex = (bytesPerRow * yy) + xx * bytesPerPixel;
red = rawData[byteIndex];
green = rawData[byteIndex + 1];
blue = rawData[byteIndex + 2];
alpha = rawData[byteIndex + 3];
26 голосов
/ 08 августа 2009

Технические вопросы и ответы Apple QA1509 показывают следующий простой подход: 

CFDataRef CopyImagePixels(CGImageRef inImage)
{
    return CGDataProviderCopyData(CGImageGetDataProvider(inImage));
}

Используйте CFDataGetBytePtr, чтобы получить действительные байты (и различные CGImageGet* методы, чтобы понять, как их интерпретировать).

13 голосов
/ 16 января 2018

Не могу поверить, что здесь нет ни одного правильного ответа здесь. Нет необходимости выделять указатели, а неумноженные значения по-прежнему необходимо нормализовать. Чтобы перейти к погоне, вот верная версия для Swift 4. Для UIImage просто используйте .cgImage. 

extension CGImage {
    func colors(at: [CGPoint]) -> [UIColor]? {
        let colorSpace = CGColorSpaceCreateDeviceRGB()
        let bytesPerPixel = 4
        let bytesPerRow = bytesPerPixel * width
        let bitsPerComponent = 8
        let bitmapInfo: UInt32 = CGImageAlphaInfo.premultipliedLast.rawValue | CGBitmapInfo.byteOrder32Big.rawValue

        guard let context = CGContext(data: nil, width: width, height: height, bitsPerComponent: bitsPerComponent, bytesPerRow: bytesPerRow, space: colorSpace, bitmapInfo: bitmapInfo),
            let ptr = context.data?.assumingMemoryBound(to: UInt8.self) else {
            return nil
        }

        context.draw(self, in: CGRect(x: 0, y: 0, width: width, height: height))

        return at.map { p in
            let i = bytesPerRow * Int(p.y) + bytesPerPixel * Int(p.x)

            let a = CGFloat(ptr[i + 3]) / 255.0
            let r = (CGFloat(ptr[i]) / a) / 255.0
            let g = (CGFloat(ptr[i + 1]) / a) / 255.0
            let b = (CGFloat(ptr[i + 2]) / a) / 255.0

            return UIColor(red: r, green: g, blue: b, alpha: a)
        }
    }
}
9 голосов
/ 16 апреля 2011

Вот SO thread , где @Matt отображает только нужный пиксель в контекст 1x1, смещая изображение так, чтобы нужный пиксель совпал с одним пикселем в контексте.

9 голосов
/ 17 декабря 2010 
NSString * path = [[NSBundle mainBundle] pathForResource:@"filename" ofType:@"jpg"];
UIImage * img = [[UIImage alloc]initWithContentsOfFile:path];
CGImageRef image = [img CGImage];
CFDataRef data = CGDataProviderCopyData(CGImageGetDataProvider(image));
const unsigned char * buffer =  CFDataGetBytePtr(data);
7 голосов
/ 16 апреля 2015

UIImage является оберткой, байты которой CGImage или CIImage

Согласно справочнику Apple по на UIImage объект является неизменным, и у вас нет доступа к резервным байтам. Хотя верно то, что вы можете получить доступ к данным CGImage, если вы заполнили UIImage CGImage (явно или неявно), он вернет NULL, если UIImage поддерживается CIImage и наоборот. наоборот.

Объекты изображения не обеспечивают прямой доступ к своему базовому изображению данные. Тем не менее, вы можете получить данные изображения в других форматах для использовать в вашем приложении. В частности, вы можете использовать cgImage и ciImage свойства для получения версий изображения, которые совместимы с Core Graphics и Core Image соответственно. Вы также можете использовать UIImagePNGRepresentation ( :) и UIImageJPEGRepresentation (: _ :) функции для создания объекта NSData, содержащего данные изображения в в формате PNG или JPEG.

Общие хитрости, чтобы обойти эту проблему

Как указано, ваши варианты

  • UIImagePNGRпредставление или JPEG
  • Определите, содержит ли изображение вспомогательные данные CGImage или CIImage, и получите их там

Ни один из этих приемов не является особенно хорошим, если вы хотите выводить данные, которые не являются данными ARGB, PNG или JPEG, а данные еще не поддерживаются CIImage.

Моя рекомендация, попробуйте CIImage

При разработке вашего проекта для вас может иметь больше смысла вообще избегать UIImage и выбирать что-то другое. UIImage, как оболочка изображения Obj-C, часто поддерживается CGImage до такой степени, что мы принимаем это как должное. CIImage имеет тенденцию быть лучшим форматом-оберткой, поскольку вы можете использовать CIContext , чтобы получить нужный формат без необходимости знать, как он был создан. В вашем случае получение растрового изображения было бы вопросом вызова

- рендер: toBitmap: rowBytes: границы: формат: colorSpace:

В качестве дополнительного бонуса вы можете начать делать приятные манипуляции с изображением, цепляя фильтры на изображение. Это решает многие проблемы, когда изображение переворачивается вверх ногами или его нужно повернуть / масштабировать и т. Д.

6 голосов
/ 27 января 2017

Основываясь на ответе Оли и Алгала, вот обновленный ответ для Swift 3 

public func getRGBAs(fromImage image: UIImage, x: Int, y: Int, count: Int) -> [UIColor] {

var result = [UIColor]()

// First get the image into your data buffer
guard let cgImage = image.cgImage else {
    print("CGContext creation failed")
    return []
}

let width = cgImage.width
let height = cgImage.height
let colorSpace = CGColorSpaceCreateDeviceRGB()
let rawdata = calloc(height*width*4, MemoryLayout<CUnsignedChar>.size)
let bytesPerPixel = 4
let bytesPerRow = bytesPerPixel * width
let bitsPerComponent = 8
let bitmapInfo: UInt32 = CGImageAlphaInfo.premultipliedLast.rawValue | CGBitmapInfo.byteOrder32Big.rawValue

guard let context = CGContext(data: rawdata, width: width, height: height, bitsPerComponent: bitsPerComponent, bytesPerRow: bytesPerRow, space: colorSpace, bitmapInfo: bitmapInfo) else {
    print("CGContext creation failed")
    return result
}

context.draw(cgImage, in: CGRect(x: 0, y: 0, width: width, height: height))

// Now your rawData contains the image data in the RGBA8888 pixel format.
var byteIndex = bytesPerRow * y + bytesPerPixel * x

for _ in 0..<count {
    let alpha = CGFloat(rawdata!.load(fromByteOffset: byteIndex + 3, as: UInt8.self)) / 255.0
    let red = CGFloat(rawdata!.load(fromByteOffset: byteIndex, as: UInt8.self)) / alpha
    let green = CGFloat(rawdata!.load(fromByteOffset: byteIndex + 1, as: UInt8.self)) / alpha
    let blue = CGFloat(rawdata!.load(fromByteOffset: byteIndex + 2, as: UInt8.self)) / alpha
    byteIndex += bytesPerPixel

    let aColor = UIColor(red: red, green: green, blue: blue, alpha: alpha)
    result.append(aColor)
}

free(rawdata)

return result
}

2 голосов
/ 29 октября 2013

На основании разных ответов, но в основном это , это работает для того, что мне нужно: 

UIImage *image1 = ...; // The image from where you want a pixel data
int pixelX = ...; // The X coordinate of the pixel you want to retrieve
int pixelY = ...; // The Y coordinate of the pixel you want to retrieve

uint32_t pixel1; // Where the pixel data is to be stored
CGContextRef context1 = CGBitmapContextCreate(&pixel1, 1, 1, 8, 4, CGColorSpaceCreateDeviceRGB(), kCGImageAlphaNoneSkipFirst);
CGContextDrawImage(context1, CGRectMake(-pixelX, -pixelY, CGImageGetWidth(image1.CGImage), CGImageGetHeight(image1.CGImage)), image1.CGImage);
CGContextRelease(context1);

В результате этой строки у вас будет пиксель в формате AARRGGBB с альфа-значением, всегда установленным в FF в 4-байтовом целом числе без знака pixel1.

...