image.cpp

   1 #include "image.hpp"
   2 #include <ImageIO/ImageIO.h>
   3
   4 Image::Image(const std::string &path) {
   5         CFStringRef str = CFStringCreateWithCString(NULL, path.c_str(), kCFStringEncodingUTF8);
   6         CFURLRef url = CFURLCreateWithFileSystemPath(NULL, str, kCFURLPOSIXPathStyle, false);
   7         CGImageSourceRef source = CGImageSourceCreateWithURL(url, NULL);
   8         CGImageRef ref = CGImageSourceCreateImageAtIndex(source, 0, NULL);
   9         initWithImageRef(ref);
  10         CGImageRelease(ref);
  11 }
  12
  13 Image::Image(const unsigned char *data, size_t length, const char *formatHint) {
  14         CGDataProviderRef dpRef = CGDataProviderCreateWithData(NULL, data, length, NULL);
  15         CGImageRef ref;
  16         if (strcmp("png", formatHint) == 0)
  17                 ref = CGImageCreateWithPNGDataProvider(dpRef, NULL, false, kCGRenderingIntentDefault);
  18         else if (strcmp("jpg", formatHint) == 0)
  19                 ref = CGImageCreateWithJPEGDataProvider(dpRef, NULL, false, kCGRenderingIntentDefault);
  20         else
  21                 abort();
  22         initWithImageRef(ref);
  23         CGImageRelease(ref);
  24 }
  25
  26 void Image::initWithImageRef(CGImageRef ref) {
  27         _width = CGImageGetWidth(ref), _height = CGImageGetHeight(ref);
  28         info = CGImageGetBitmapInfo(ref);
  29         alphaInfo = CGImageGetAlphaInfo(ref);
  30         colorSpace = CGImageGetColorSpace(ref);
  31         bitsPerComponent = CGImageGetBitsPerComponent(ref);
  32
  33         dataRef = CGDataProviderCopyData(CGImageGetDataProvider(ref));
  34 }
  35
  36 unsigned char *Image::data() const {
  37         return (unsigned char*) CFDataGetBytePtr(dataRef);
  38 }
  39
  40 GLfloat Image::width() const { return _width; }
  41 GLfloat Image::height() const { return _height; }
  42
  43 // TODO: Properly implement this for both internal format + format
  44 GLenum Image::format() const {
  45         if (CGColorSpaceGetModel(colorSpace) == kCGColorSpaceModelMonochrome) {
  46                 return GL_DEPTH_COMPONENT;
  47         }
  48         if (alphaInfo == kCGImageAlphaNone) {
  49                 return GL_RGB;
  50         }
  51         return GL_RGBA;
  52 }
  53
  54 GLint Image::internalFormat() const {
  55         switch (format()) {
  56                 case GL_DEPTH_COMPONENT: return GL_DEPTH_COMPONENT;
  57                 case GL_RGB: return GL_RGB;
  58                 default: return GL_RGBA;
  59         }
  60 }
  61
  62 GLenum Image::type() const {
  63         bool isFloat = info & kCGBitmapFloatComponents;
  64         if (isFloat) return GL_FLOAT;
  65         return GL_UNSIGNED_BYTE;
  66
  67         //TODO:
  68         /* switch (bitsPerComponent) { */
  69         /*      case 1: return GL_UNSIGNED_BYTE; */
  70         /*      case 16: return GL_UNSIGNED_SHORT; */
  71         /*  case 24: */
  72 }
  73
  74 Image::~Image() { CFRelease(dataRef); }