ios像素点颜色取样

时间:2023-03-08 21:11:03
  1. 一、像素点颜色取样
  2. + (UIColor*) getPixelColorAtLocation:(CGPoint)point inImage:(UIImage *)image {
  3. UIColor* color = nil;
  4. CGImageRef inImage = image.CGImage;
  5. CGContextRef cgctx = [DetectColour createARGBBitmapContextFromImage:
  6. inImage];
  7. if (cgctx == NULL) { return nil; /* error */ }
  8. size_t w = CGImageGetWidth(inImage);
  9. size_t h = CGImageGetHeight(inImage);
  10. CGRect rect = {{0,0},{w,h}};
  11. CGContextDrawImage(cgctx, rect, inImage);
  12. unsigned char* data = CGBitmapContextGetData (cgctx);
  13. if (data != NULL) {
  14. int offset = 4*((w*round(point.y))+round(point.x));
  15. int alpha =  data[offset];
  16. int red = data[offset+1];
  17. int green = data[offset+2];
  18. int blue = data[offset+3];
  19. NSLog(@"offset: %i colors: RGB A %i %i %i  %i",offset,red,green,
  20. blue,alpha);
  21. NSLog(@"x:%f y:%f", point.x, point.y);
  22. color = [UIColor colorWithRed:(red/255.0f) green:(green/255.0f) blue:
  23. (blue/255.0f) alpha:(alpha/255.0f)];
  24. }
  25. CGContextRelease(cgctx);
  26. if (data) { free(data); }
  27. return color;
  28. }
  29. + (CGContextRef) createARGBBitmapContextFromImage:(CGImageRef) inImage {
  30. CGContextRef    context = NULL;
  31. CGColorSpaceRef colorSpace;
  32. void *          bitmapData;
  33. int             bitmapByteCount;
  34. int             bitmapBytesPerRow;
  35. size_t pixelsWide = CGImageGetWidth(inImage);
  36. size_t pixelsHigh = CGImageGetHeight(inImage);
  37. bitmapBytesPerRow   = (pixelsWide * 4);
  38. bitmapByteCount     = (bitmapBytesPerRow * pixelsHigh);
  39. colorSpace = CGColorSpaceCreateDeviceRGB();
  40. if (colorSpace == NULL)
  41. {
  42. fprintf(stderr, "Error allocating color space\n");
  43. return NULL;
  44. }
  45. bitmapData = malloc( bitmapByteCount );
  46. if (bitmapData == NULL)
  47. {
  48. fprintf (stderr, "Memory not allocated!");
  49. CGColorSpaceRelease( colorSpace );
  50. return NULL;
  51. }
  52. context = CGBitmapContextCreate (bitmapData,
  53. pixelsWide,
  54. pixelsHigh,
  55. 8,
  56. bitmapBytesPerRow,
  57. colorSpace,
  58. kCGImageAlphaPremultipliedFirst);
  59. if (context == NULL)
  60. {
  61. free (bitmapData);
  62. fprintf (stderr, "Context not created!");
  63. }
  64. CGColorSpaceRelease( colorSpace );
  65. return context;
  66. }

二、颜色匹配计算
一是精确的匹配,通过生成两个CGColor,利用CGColorEqualToColor函数进行对比即可,如下:

  1. //生成采样对照颜色(黑色)
  2. UIColor* sampleColor = [UIColor colorWithRed:(0/255.0f) green:(0/255.0f)
  3. blue:(0/255.0f) alpha:(255/255.0f)];
  4. //比较
  5. if (CGColorEqualToColor(detectedColor.CGColor, sampleColor.CGColor)) {
  6. //处理
  7. }

二是模糊的匹配,在RGB色彩空间下, 每个像素点的颜色有四个通道(R, G, B, Alpha),比较
两个颜色是否一致的方法,可以通过比较其向量的方式来进行,如下:

  1. //获取颜色分量
  2. const CGFloat* components = CGColorGetComponents(detectedColor.CGColor);
  3. CGFloat Red, Green, Blue, Alpha;
  4. Alpha = components[0];
  5. Red = components[0+1];
  6. ....
  7. //向量比较
  8. float difference = pow( pow((red1 - red2), 2) + pow((green1 - green2), 2) +
  9. pow((blue1 - blue2), 2), 0.5 );

至于阀值的设定,就需要大家根据实际情况来调整了,一般可设置为85,如果严格一点,则小一点;宽松一点,则可以设大一点。

相关文章