Obtenir des pixels et des couleurs de NSImage

Question

J'ai créé un objet NSImage et, idéalement, je voudrais déterminer le nombre de pixels de chaque couleur qu'il contient. Est-ce possible?

Answer 1

Je propose de créer votre propre bitmap context, l'enveloppant dans un graphics context et du paramètre que le contexte actuel, telling the image to draw itself, puis accéder aux données de pixels derrière le contexte bitmap directement.

Ce sera plus de code, mais vous permettra d'économiser à la fois un voyage à travers une représentation TIFF et la création de milliers ou de millions d'objets NSColor. Si vous travaillez avec des images de taille appréciable, ces dépenses s'additionneront rapidement.

Answer 2

Obtenez un NSBitmapImageRep à partir de votre NSImage. Ensuite, vous pouvez avoir accès aux pixels.

NSImage* img = ...; 
NSBitmapImageRep* raw_img = [NSBitmapImageRep imageRepWithData:[img TIFFRepresentation]]; 
NSColor* color = [raw_img colorAtX:0 y:0]; 

Answer 3

Recherchez "histogramme" dans la documentation Core Image.

Answer 4

Ce code rend le NSImage dans un CGBitmapContext:

- (void)updateImageData { 

    if (!_image) 
     return; 

    // Dimensions - source image determines context size 

    NSSize imageSize = _image.size; 
    NSRect imageRect = NSMakeRect(0, 0, imageSize.width, imageSize.height); 

    // Create a context to hold the image data 

    CGColorSpaceRef colorSpace = CGColorSpaceCreateWithName(kCGColorSpaceGenericRGB); 

    CGContextRef ctx = CGBitmapContextCreate(NULL, 
              imageSize.width, 
              imageSize.height, 
              8, 
              0, 
              colorSpace, 
              kCGImageAlphaPremultipliedLast); 

    // Wrap graphics context 

    NSGraphicsContext* gctx = [NSGraphicsContext graphicsContextWithCGContext:ctx flipped:NO]; 

    // Make our bitmap context current and render the NSImage into it 

    [NSGraphicsContext setCurrentContext:gctx]; 
    [_image drawInRect:imageRect]; 

    // Calculate the histogram 

    [self computeHistogramFromBitmap:ctx]; 

    // Clean up 

    [NSGraphicsContext setCurrentContext:nil]; 
    CGContextRelease(ctx); 
    CGColorSpaceRelease(colorSpace); 
} 

Compte tenu d'un contexte bitmap, nous pouvons accéder aux données d'images brutes directement, et calculer les histogrammes pour chaque canal de couleur:

- (void)computeHistogramFromBitmap:(CGContextRef)bitmap { 

    // NB: Assumes RGBA 8bpp 

    size_t width = CGBitmapContextGetWidth(bitmap); 
    size_t height = CGBitmapContextGetHeight(bitmap); 

    uint32_t* pixel = (uint32_t*)CGBitmapContextGetData(bitmap); 

    for (unsigned y = 0; y < height; y++) 
    { 
     for (unsigned x = 0; x < width; x++) 
     { 
      uint32_t rgba = *pixel; 

      // Extract colour components 
      uint8_t red = (rgba & 0x000000ff) >> 0; 
      uint8_t green = (rgba & 0x0000ff00) >> 8; 
      uint8_t blue = (rgba & 0x00ff0000) >> 16; 

      // Accumulate each colour 
      _histogram[kRedChannel][red]++; 
      _histogram[kGreenChannel][green]++; 
      _histogram[kBlueChannel][blue]++; 

      // Next pixel! 
      pixel++; 
     } 
    } 
} 

@end 

J'ai publié un projet complet, un exemple d'application Cocoa, qui inclut ce qui précède.

• https://github.com/gavinb/CocoaImageHistogram.git

Answer 5

En utilisant colorAtX avec NSBitmapImageRep ne conduit pas toujours à la bonne couleur exacte.

j'ai réussi à obtenir la couleur correcte avec ce code simple:

[yourImage lockFocus]; // yourImage is just your NSImage variable 
NSColor *pixelColor = NSReadPixel(NSMakePoint(1, 1)); // Or another point 
[yourImage unlockFocus];