modules/highgui/src/ios_conversions.mm

   1 /*M///////////////////////////////////////////////////////////////////////////////////////
   2 //
   3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
   4 //
   5 //  By downloading, copying, installing or using the software you agree to this license.
   6 //  If you do not agree to this license, do not download, install,
   7 //  copy or use the software.
   8 //
   9 //
  10 //                          License Agreement
  11 //                For Open Source Computer Vision Library
  12 //
  13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
  14 // Copyright (C) 2009, Willow Garage Inc., all rights reserved.
  15 // Third party copyrights are property of their respective owners.
  16 //
  17 // Redistribution and use in source and binary forms, with or without modification,
  18 // are permitted provided that the following conditions are met:
  19 //
  20 //   * Redistribution's of source code must retain the above copyright notice,
  21 //     this list of conditions and the following disclaimer.
  22 //
  23 //   * Redistribution's in binary form must reproduce the above copyright notice,
  24 //     this list of conditions and the following disclaimer in the documentation
  25 //     and/or other materials provided with the distribution.
  26 //
  27 //   * The name of the copyright holders may not be used to endorse or promote products
  28 //     derived from this software without specific prior written permission.
  29 //
  30 // This software is provided by the copyright holders and contributors "as is" and
  31 // any express or implied warranties, including, but not limited to, the implied
  32 // warranties of merchantability and fitness for a particular purpose are disclaimed.
  33 // In no event shall the Intel Corporation or contributors be liable for any direct,
  34 // indirect, incidental, special, exemplary, or consequential damages
  35 // (including, but not limited to, procurement of substitute goods or services;
  36 // loss of use, data, or profits; or business interruption) however caused
  37 // and on any theory of liability, whether in contract, strict liability,
  38 // or tort (including negligence or otherwise) arising in any way out of
  39 // the use of this software, even if advised of the possibility of such damage.
  40 //
  41 //M*/
  42
  43 #import "opencv2/highgui/cap_ios.h"
  44 #include "precomp.hpp"
  45
  46 UIImage* MatToUIImage(const cv::Mat& image) {
  47
  48     NSData *data = [NSData dataWithBytes:image.data
  49                                   length:image.elemSize()*image.total()];
  50
  51     CGColorSpaceRef colorSpace;
  52
  53     if (image.elemSize() == 1) {
  54         colorSpace = CGColorSpaceCreateDeviceGray();
  55     } else {
  56         colorSpace = CGColorSpaceCreateDeviceRGB();
  57     }
  58
  59     CGDataProviderRef provider =
  60             CGDataProviderCreateWithCFData((__bridge CFDataRef)data);
  61
  62     // Creating CGImage from cv::Mat
  63     CGImageRef imageRef = CGImageCreate(image.cols,
  64                                         image.rows,
  65                                         8,
  66                                         8 * image.elemSize(),
  67                                         image.step.p[0],
  68                                         colorSpace,
  69                                         kCGImageAlphaNone|
  70                                         kCGBitmapByteOrderDefault,
  71                                         provider,
  72                                         NULL,
  73                                         false,
  74                                         kCGRenderingIntentDefault
  75                                         );
  76
  77
  78     // Getting UIImage from CGImage
  79     UIImage *finalImage = [UIImage imageWithCGImage:imageRef];
  80     CGImageRelease(imageRef);
  81     CGDataProviderRelease(provider);
  82     CGColorSpaceRelease(colorSpace);
  83
  84     return finalImage;
  85 }
  86
  87 void UIImageToMat(const UIImage* image,
  88                          cv::Mat& m, bool alphaExist) {
  89     CGColorSpaceRef colorSpace = CGImageGetColorSpace(image.CGImage);
  90     CGFloat cols = image.size.width, rows = image.size.height;
  91     CGContextRef contextRef;
  92     CGBitmapInfo bitmapInfo = kCGImageAlphaPremultipliedLast;
  93     if (CGColorSpaceGetModel(colorSpace) == 0)
  94     {
  95         m.create(rows, cols, CV_8UC1); // 8 bits per component, 1 channel
  96         bitmapInfo = kCGImageAlphaNone;
  97         if (!alphaExist)
  98             bitmapInfo = kCGImageAlphaNone;
  99         contextRef = CGBitmapContextCreate(m.data, m.cols, m.rows, 8,
 100                                            m.step[0], colorSpace,
 101                                            bitmapInfo);
 102     }
 103     else
 104     {
 105         m.create(rows, cols, CV_8UC4); // 8 bits per component, 4 channels
 106         if (!alphaExist)
 107             bitmapInfo = kCGImageAlphaNoneSkipLast |
 108                                 kCGBitmapByteOrderDefault;
 109         contextRef = CGBitmapContextCreate(m.data, m.cols, m.rows, 8,
 110                                            m.step[0], colorSpace,
 111                                            bitmapInfo);
 112     }
 113     CGContextDrawImage(contextRef, CGRectMake(0, 0, cols, rows),
 114                        image.CGImage);
 115     CGContextRelease(contextRef);
 116     CGColorSpaceRelease(colorSpace);
 117 }