Andrey Kamaev
13 years ago
parent
00fbf58902
commit
ae5649d3f7
7 changed files with 0 additions and 481 deletions
@ -1,396 +0,0 @@ |
||||
/*
|
||||
* grfmt_imageio.cpp |
||||
* |
||||
* |
||||
* Created by Morgan Conbere on 5/17/07. |
||||
* |
||||
*/ |
||||
|
||||
#include "precomp.hpp" |
||||
|
||||
#ifdef HAVE_IMAGEIO |
||||
|
||||
#include "grfmt_imageio.hpp" |
||||
|
||||
namespace cv |
||||
{ |
||||
|
||||
/////////////////////// ImageIODecoder ///////////////////
|
||||
|
||||
ImageIODecoder::ImageIODecoder() |
||||
{ |
||||
imageRef = NULL; |
||||
} |
||||
|
||||
ImageIODecoder::~ImageIODecoder() |
||||
{ |
||||
close(); |
||||
} |
||||
|
||||
|
||||
void ImageIODecoder::close() |
||||
{ |
||||
CGImageRelease( imageRef ); |
||||
imageRef = NULL; |
||||
} |
||||
|
||||
|
||||
size_t ImageIODecoder::signatureLength() const |
||||
{ |
||||
return 12; |
||||
} |
||||
|
||||
bool ImageIODecoder::checkSignature( const string& signature ) const |
||||
{ |
||||
// TODO: implement real signature check
|
||||
return true; |
||||
} |
||||
|
||||
ImageDecoder ImageIODecoder::newDecoder() const |
||||
{ |
||||
return new ImageIODecoder; |
||||
} |
||||
|
||||
bool ImageIODecoder::readHeader() |
||||
{ |
||||
CFURLRef imageURLRef; |
||||
CGImageSourceRef sourceRef; |
||||
// diciu, if ReadHeader is called twice in a row make sure to release the previously allocated imageRef
|
||||
if (imageRef != NULL) |
||||
CGImageRelease(imageRef); |
||||
imageRef = NULL; |
||||
|
||||
imageURLRef = CFURLCreateFromFileSystemRepresentation( NULL, |
||||
(const UInt8*)m_filename.c_str(), m_filename.size(), false ); |
||||
|
||||
sourceRef = CGImageSourceCreateWithURL( imageURLRef, NULL ); |
||||
CFRelease( imageURLRef ); |
||||
if ( !sourceRef ) |
||||
return false; |
||||
|
||||
imageRef = CGImageSourceCreateImageAtIndex( sourceRef, 0, NULL ); |
||||
CFRelease( sourceRef ); |
||||
if( !imageRef ) |
||||
return false; |
||||
|
||||
m_width = CGImageGetWidth( imageRef ); |
||||
m_height = CGImageGetHeight( imageRef ); |
||||
|
||||
CGColorSpaceRef colorSpace = CGImageGetColorSpace( imageRef ); |
||||
if( !colorSpace ) |
||||
return false; |
||||
|
||||
m_type = CGColorSpaceGetNumberOfComponents( colorSpace ) > 1 ? CV_8UC3 : CV_8UC1; |
||||
|
||||
return true; |
||||
} |
||||
|
||||
|
||||
bool ImageIODecoder::readData( Mat& img ) |
||||
{ |
||||
uchar* data = img.data; |
||||
int step = img.step; |
||||
bool color = img.channels() > 1; |
||||
int bpp; // Bytes per pixel
|
||||
int bit_depth = 8; |
||||
|
||||
// Get Height, Width, and color information
|
||||
if( !readHeader() ) |
||||
return false; |
||||
|
||||
CGContextRef context = NULL; // The bitmap context
|
||||
CGColorSpaceRef colorSpace = NULL; |
||||
uchar* bitmap = NULL; |
||||
CGImageAlphaInfo alphaInfo; |
||||
|
||||
// CoreGraphics will take care of converting to grayscale and back as long as the
|
||||
// appropriate colorspace is set
|
||||
if( color == CV_LOAD_IMAGE_GRAYSCALE ) |
||||
{ |
||||
colorSpace = CGColorSpaceCreateDeviceGray(); |
||||
bpp = 1; |
||||
alphaInfo = kCGImageAlphaNone; |
||||
} |
||||
else if( color == CV_LOAD_IMAGE_COLOR ) |
||||
{ |
||||
#if TARGET_OS_IPHONE || TARGET_IPHONE_SIMULATOR |
||||
colorSpace = CGColorSpaceCreateDeviceRGB(); |
||||
#else |
||||
colorSpace = CGColorSpaceCreateWithName( kCGColorSpaceGenericRGBLinear ); |
||||
#endif |
||||
bpp = 4; /* CG only has 8 and 32 bit color spaces, so we waste a byte */ |
||||
alphaInfo = kCGImageAlphaNoneSkipLast; |
||||
} |
||||
if( !colorSpace ) |
||||
return false; |
||||
|
||||
bitmap = (uchar*)malloc( bpp * m_height * m_width ); |
||||
if( !bitmap ) |
||||
{ |
||||
CGColorSpaceRelease( colorSpace ); |
||||
return false; |
||||
} |
||||
|
||||
context = CGBitmapContextCreate( (void *)bitmap, |
||||
m_width, /* width */ |
||||
m_height, /* height */ |
||||
bit_depth, /* bit depth */ |
||||
bpp * m_width, /* bytes per row */ |
||||
colorSpace, /* color space */ |
||||
alphaInfo); |
||||
|
||||
CGColorSpaceRelease( colorSpace ); |
||||
if( !context ) |
||||
{ |
||||
free( bitmap ); |
||||
return false; |
||||
} |
||||
|
||||
// Copy the image data into the bitmap region
|
||||
CGRect rect = {{0,0},{m_width,m_height}}; |
||||
CGContextDrawImage( context, rect, imageRef ); |
||||
|
||||
uchar* bitdata = (uchar*)CGBitmapContextGetData( context ); |
||||
if( !bitdata ) |
||||
{ |
||||
free( bitmap); |
||||
CGContextRelease( context ); |
||||
return false; |
||||
} |
||||
|
||||
// Move the bitmap (in RGB) into data (in BGR)
|
||||
int bitmapIndex = 0; |
||||
|
||||
if( color == CV_LOAD_IMAGE_COLOR ) |
||||
{ |
||||
uchar * base = data; |
||||
|
||||
for (int y = 0; y < m_height; y++) |
||||
{ |
||||
uchar * line = base + y * step; |
||||
|
||||
for (int x = 0; x < m_width; x++) |
||||
{ |
||||
// Blue channel
|
||||
line[0] = bitdata[bitmapIndex + 2]; |
||||
// Green channel
|
||||
line[1] = bitdata[bitmapIndex + 1]; |
||||
// Red channel
|
||||
line[2] = bitdata[bitmapIndex + 0]; |
||||
|
||||
line += 3; |
||||
bitmapIndex += bpp; |
||||
} |
||||
} |
||||
} |
||||
else if( color == CV_LOAD_IMAGE_GRAYSCALE ) |
||||
{ |
||||
for (int y = 0; y < m_height; y++) |
||||
memcpy (data + y * step, bitmap + y * m_width, m_width); |
||||
} |
||||
|
||||
free( bitmap ); |
||||
CGContextRelease( context ); |
||||
return true; |
||||
} |
||||
|
||||
|
||||
/////////////////////// ImageIOEncoder ///////////////////
|
||||
|
||||
ImageIOEncoder::ImageIOEncoder() |
||||
{ |
||||
m_description = "Apple ImageIO (*.bmp;*.dib;*.exr;*.jpeg;*.jpg;*.jpe;*.jp2;*.pdf;*.png;*.tiff;*.tif)"; |
||||
} |
||||
|
||||
|
||||
ImageIOEncoder::~ImageIOEncoder() |
||||
{ |
||||
} |
||||
|
||||
|
||||
ImageEncoder ImageIOEncoder::newEncoder() const |
||||
{ |
||||
return new ImageIOEncoder; |
||||
} |
||||
|
||||
static |
||||
CFStringRef FilenameToUTI( const char* filename ) |
||||
{ |
||||
const char* ext = filename; |
||||
char* ext_buf; |
||||
int i; |
||||
CFStringRef imageUTI = NULL; |
||||
|
||||
for(;;) |
||||
{ |
||||
const char* temp = strchr( ext + 1, '.' ); |
||||
if( !temp ) break; |
||||
ext = temp; |
||||
} |
||||
|
||||
if(!ext) |
||||
return NULL; |
||||
|
||||
ext_buf = (char*)malloc(strlen(ext)+1); |
||||
for(i = 0; ext[i] != '\0'; i++) |
||||
ext_buf[i] = (char)tolower(ext[i]); |
||||
ext_buf[i] = '\0'; |
||||
ext = ext_buf; |
||||
|
||||
if( !strcmp(ext, ".bmp") || !strcmp(ext, ".dib") ) |
||||
imageUTI = CFSTR( "com.microsoft.bmp" ); |
||||
else if( !strcmp(ext, ".exr") ) |
||||
imageUTI = CFSTR( "com.ilm.openexr-image" ); |
||||
else if( !strcmp(ext, ".jpeg") || !strcmp(ext, ".jpg") || !strcmp(ext, ".jpe") ) |
||||
imageUTI = CFSTR( "public.jpeg" ); |
||||
else if( !strcmp(ext, ".jp2") ) |
||||
imageUTI = CFSTR( "public.jpeg-2000" ); |
||||
else if( !strcmp(ext, ".pdf") ) |
||||
imageUTI = CFSTR( "com.adobe.pdf" ); |
||||
else if( !strcmp(ext, ".png") ) |
||||
imageUTI = CFSTR( "public.png" ); |
||||
else if( !strcmp(ext, ".tiff") || !strcmp(ext, ".tif") ) |
||||
imageUTI = CFSTR( "public.tiff" ); |
||||
|
||||
free(ext_buf); |
||||
|
||||
return imageUTI; |
||||
} |
||||
|
||||
|
||||
bool ImageIOEncoder::write( const Mat& img, const vector<int>& params ) |
||||
{ |
||||
int width = img.cols, height = img.rows; |
||||
int _channels = img.channels(); |
||||
const uchar* data = img.data; |
||||
int step = img.step; |
||||
|
||||
// Determine the appropriate UTI based on the filename extension
|
||||
CFStringRef imageUTI = FilenameToUTI( m_filename.c_str() ); |
||||
|
||||
// Determine the Bytes Per Pixel
|
||||
int bpp = (_channels == 1) ? 1 : 4; |
||||
|
||||
// Write the data into a bitmap context
|
||||
CGContextRef context; |
||||
CGColorSpaceRef colorSpace; |
||||
uchar* bitmapData = NULL; |
||||
|
||||
if( bpp == 1 ) { |
||||
#if TARGET_OS_IPHONE || TARGET_IPHONE_SIMULATOR |
||||
colorSpace = CGColorSpaceCreateDeviceGray(); |
||||
#else |
||||
colorSpace = CGColorSpaceCreateWithName( kCGColorSpaceGenericGray ); |
||||
#endif |
||||
} |
||||
else if( bpp == 4 ) { |
||||
#if TARGET_OS_IPHONE || TARGET_IPHONE_SIMULATOR |
||||
colorSpace = CGColorSpaceCreateDeviceRGB(); |
||||
#else |
||||
colorSpace = CGColorSpaceCreateWithName( kCGColorSpaceGenericRGBLinear ); |
||||
#endif |
||||
} |
||||
if( !colorSpace ) |
||||
return false; |
||||
|
||||
bitmapData = (uchar*)malloc( bpp * height * width ); |
||||
if( !bitmapData ) |
||||
{ |
||||
CGColorSpaceRelease( colorSpace ); |
||||
return false; |
||||
} |
||||
|
||||
context = CGBitmapContextCreate( bitmapData, |
||||
width, |
||||
height, |
||||
8, |
||||
bpp * width, |
||||
colorSpace, |
||||
(bpp == 1) ? kCGImageAlphaNone : |
||||
kCGImageAlphaNoneSkipLast ); |
||||
CGColorSpaceRelease( colorSpace ); |
||||
if( !context ) |
||||
{ |
||||
free( bitmapData ); |
||||
return false; |
||||
} |
||||
|
||||
// Copy pixel information from data into bitmapData
|
||||
if (bpp == 4) |
||||
{ |
||||
int bitmapIndex = 0; |
||||
const uchar * base = data; |
||||
|
||||
for (int y = 0; y < height; y++) |
||||
{ |
||||
const uchar * line = base + y * step; |
||||
|
||||
for (int x = 0; x < width; x++) |
||||
{ |
||||
// Blue channel
|
||||
bitmapData[bitmapIndex + 2] = line[0]; |
||||
// Green channel
|
||||
bitmapData[bitmapIndex + 1] = line[1]; |
||||
// Red channel
|
||||
bitmapData[bitmapIndex + 0] = line[2]; |
||||
|
||||
line += 3; |
||||
bitmapIndex += bpp; |
||||
} |
||||
} |
||||
} |
||||
else if (bpp == 1) |
||||
{ |
||||
for (int y = 0; y < height; y++) |
||||
memcpy (bitmapData + y * width, data + y * step, width); |
||||
} |
||||
|
||||
// Turn the bitmap context into an imageRef
|
||||
CGImageRef imageRef = CGBitmapContextCreateImage( context ); |
||||
CGContextRelease( context ); |
||||
if( !imageRef ) |
||||
{ |
||||
free( bitmapData ); |
||||
return false; |
||||
} |
||||
|
||||
// Write the imageRef to a file based on the UTI
|
||||
CFURLRef imageURLRef = CFURLCreateFromFileSystemRepresentation( NULL, |
||||
(const UInt8*)m_filename.c_str(), m_filename.size(), false ); |
||||
if( !imageURLRef ) |
||||
{ |
||||
CGImageRelease( imageRef ); |
||||
free( bitmapData ); |
||||
return false; |
||||
} |
||||
|
||||
CGImageDestinationRef destRef = CGImageDestinationCreateWithURL( imageURLRef, |
||||
imageUTI, |
||||
1, |
||||
NULL); |
||||
CFRelease( imageURLRef ); |
||||
if( !destRef ) |
||||
{ |
||||
CGImageRelease( imageRef ); |
||||
free( bitmapData ); |
||||
fprintf(stderr, "!destRef\n"); |
||||
return false; |
||||
} |
||||
|
||||
CGImageDestinationAddImage(destRef, imageRef, NULL); |
||||
if( !CGImageDestinationFinalize(destRef) ) |
||||
{ |
||||
fprintf(stderr, "Finalize failed\n"); |
||||
return false; |
||||
} |
||||
|
||||
CFRelease( destRef ); |
||||
CGImageRelease( imageRef ); |
||||
free( bitmapData ); |
||||
|
||||
return true; |
||||
} |
||||
|
||||
} |
||||
|
||||
#endif /* HAVE_IMAGEIO */ |
||||
@ -1,67 +0,0 @@ |
||||
/*
|
||||
* grfmt_imageio.h |
||||
* |
||||
* |
||||
* Created by Morgan Conbere on 5/17/07. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef _GRFMT_IMAGEIO_H_ |
||||
#define _GRFMT_IMAGEIO_H_ |
||||
|
||||
#ifdef HAVE_IMAGEIO |
||||
|
||||
#include "grfmt_base.hpp" |
||||
#include <TargetConditionals.h> |
||||
|
||||
#if TARGET_OS_IPHONE || TARGET_IPHONE_SIMULATOR |
||||
|
||||
#include <MobileCoreServices/MobileCoreServices.h> |
||||
#include <ImageIO/ImageIO.h> |
||||
|
||||
#else |
||||
|
||||
#include <ApplicationServices/ApplicationServices.h> |
||||
|
||||
#endif |
||||
|
||||
namespace cv |
||||
{ |
||||
|
||||
class ImageIODecoder : public BaseImageDecoder |
||||
{ |
||||
public: |
||||
|
||||
ImageIODecoder(); |
||||
~ImageIODecoder(); |
||||
|
||||
bool readData( Mat& img ); |
||||
bool readHeader(); |
||||
void close(); |
||||
|
||||
size_t signatureLength() const; |
||||
bool checkSignature( const string& signature ) const; |
||||
|
||||
ImageDecoder newDecoder() const; |
||||
|
||||
protected: |
||||
|
||||
CGImageRef imageRef; |
||||
}; |
||||
|
||||
class ImageIOEncoder : public BaseImageEncoder |
||||
{ |
||||
public: |
||||
ImageIOEncoder(); |
||||
~ImageIOEncoder(); |
||||
|
||||
bool write( const Mat& img, const vector<int>& params ); |
||||
|
||||
ImageEncoder newEncoder() const; |
||||
}; |
||||
|
||||
} |
||||
|
||||
#endif/*HAVE_IMAGEIO*/ |
||||
|
||||
#endif/*_GRFMT_IMAGEIO_H_*/ |
||||
Loading…
Reference in new issue