Update to 2.0.0 tree from current Fremantle build
[opencv] / src / highgui / utils.cpp
diff --git a/src/highgui/utils.cpp b/src/highgui/utils.cpp
new file mode 100644 (file)
index 0000000..c52d20e
--- /dev/null
@@ -0,0 +1,635 @@
+/*M///////////////////////////////////////////////////////////////////////////////////////
+//
+//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
+//
+//  By downloading, copying, installing or using the software you agree to this license.
+//  If you do not agree to this license, do not download, install,
+//  copy or use the software.
+//
+//
+//                        Intel License Agreement
+//                For Open Source Computer Vision Library
+//
+// Copyright (C) 2000, Intel Corporation, all rights reserved.
+// Third party copyrights are property of their respective owners.
+//
+// Redistribution and use in source and binary forms, with or without modification,
+// are permitted provided that the following conditions are met:
+//
+//   * Redistribution's of source code must retain the above copyright notice,
+//     this list of conditions and the following disclaimer.
+//
+//   * Redistribution's in binary form must reproduce the above copyright notice,
+//     this list of conditions and the following disclaimer in the documentation
+//     and/or other materials provided with the distribution.
+//
+//   * The name of Intel Corporation may not be used to endorse or promote products
+//     derived from this software without specific prior written permission.
+//
+// This software is provided by the copyright holders and contributors "as is" and
+// any express or implied warranties, including, but not limited to, the implied
+// warranties of merchantability and fitness for a particular purpose are disclaimed.
+// In no event shall the Intel Corporation or contributors be liable for any direct,
+// indirect, incidental, special, exemplary, or consequential damages
+// (including, but not limited to, procurement of substitute goods or services;
+// loss of use, data, or profits; or business interruption) however caused
+// and on any theory of liability, whether in contract, strict liability,
+// or tort (including negligence or otherwise) arising in any way out of
+// the use of this software, even if advised of the possibility of such damage.
+//
+//M*/
+
+#include "_highgui.h"
+#include "utils.h"
+
+#define  SCALE  14
+#define  cR  (int)(0.299*(1 << SCALE) + 0.5)
+#define  cG  (int)(0.587*(1 << SCALE) + 0.5)
+#define  cB  ((1 << SCALE) - cR - cG)
+
+void icvCvt_BGR2Gray_8u_C3C1R( const uchar* rgb, int rgb_step,
+                               uchar* gray, int gray_step,
+                               CvSize size, int _swap_rb )
+{
+    int i;
+    int swap_rb = _swap_rb ? 2 : 0;
+    for( ; size.height--; gray += gray_step )
+    {
+        for( i = 0; i < size.width; i++, rgb += 3 )
+        {
+            int t = descale( rgb[swap_rb]*cB + rgb[1]*cG + rgb[swap_rb^2]*cR, SCALE );
+            gray[i] = (uchar)t;
+        }
+
+        rgb += rgb_step - size.width*3;
+    }
+}
+
+
+void icvCvt_BGR2Gray_16u_C3C1R( const ushort* rgb, int rgb_step,
+                                ushort* gray, int gray_step,
+                                CvSize size, int _swap_rb )
+{
+    int i;
+    int swap_rb = _swap_rb ? 2 : 0;
+    for( ; size.height--; gray += gray_step )
+    {
+        for( i = 0; i < size.width; i++, rgb += 3 )
+        {
+            int t = descale( rgb[swap_rb]*cB + rgb[1]*cG + rgb[swap_rb^2]*cR, SCALE );
+            gray[i] = (ushort)t;
+        }
+
+        rgb += rgb_step - size.width*3;
+    }
+}
+
+
+void icvCvt_BGRA2Gray_8u_C4C1R( const uchar* rgba, int rgba_step,
+                                uchar* gray, int gray_step,
+                                CvSize size, int _swap_rb )
+{
+    int i;
+    int swap_rb = _swap_rb ? 2 : 0;
+    for( ; size.height--; gray += gray_step )
+    {
+        for( i = 0; i < size.width; i++, rgba += 4 )
+        {
+            int t = descale( rgba[swap_rb]*cB + rgba[1]*cG + rgba[swap_rb^2]*cR, SCALE );
+            gray[i] = (uchar)t;
+        }
+
+        rgba += rgba_step - size.width*4;
+    }
+}
+
+
+void icvCvt_Gray2BGR_8u_C1C3R( const uchar* gray, int gray_step,
+                               uchar* bgr, int bgr_step, CvSize size )
+{
+    int i;
+    for( ; size.height--; gray += gray_step )
+    {
+        for( i = 0; i < size.width; i++, bgr += 3 )
+        {
+            bgr[0] = bgr[1] = bgr[2] = gray[i];
+        }
+        bgr += bgr_step - size.width*3;
+    }
+}
+
+
+void icvCvt_BGRA2BGR_8u_C4C3R( const uchar* bgra, int bgra_step,
+                               uchar* bgr, int bgr_step,
+                               CvSize size, int _swap_rb )
+{
+    int i;
+    int swap_rb = _swap_rb ? 2 : 0;
+    for( ; size.height--; )
+    {
+        for( i = 0; i < size.width; i++, bgr += 3, bgra += 4 )
+        {
+            uchar t0 = bgra[swap_rb], t1 = bgra[1];
+            bgr[0] = t0; bgr[1] = t1;
+            t0 = bgra[swap_rb^2]; bgr[2] = t0;
+        }
+        bgr += bgr_step - size.width*3;
+        bgra += bgra_step - size.width*4;
+    }
+}
+
+
+void icvCvt_BGRA2RGBA_8u_C4R( const uchar* bgra, int bgra_step,
+                              uchar* rgba, int rgba_step, CvSize size )
+{
+    int i;
+    for( ; size.height--; )
+    {
+        for( i = 0; i < size.width; i++, bgra += 4, rgba += 4 )
+        {
+            uchar t0 = bgra[0], t1 = bgra[1];
+            uchar t2 = bgra[2], t3 = bgra[3];
+            rgba[0] = t2; rgba[1] = t1;
+            rgba[2] = t0; rgba[3] = t3;
+        }
+        bgra += bgra_step - size.width*4;
+        rgba += rgba_step - size.width*4;
+    }
+}
+
+
+void icvCvt_BGR2RGB_8u_C3R( const uchar* bgr, int bgr_step,
+                            uchar* rgb, int rgb_step, CvSize size )
+{
+    int i;
+    for( ; size.height--; )
+    {
+        for( i = 0; i < size.width; i++, bgr += 3, rgb += 3 )
+        {
+            uchar t0 = bgr[0], t1 = bgr[1], t2 = bgr[2];
+            rgb[2] = t0; rgb[1] = t1; rgb[0] = t2;
+        }
+        bgr += bgr_step - size.width*3;
+        rgb += rgb_step - size.width*3;
+    }
+}
+
+
+void icvCvt_BGR2RGB_16u_C3R( const ushort* bgr, int bgr_step,
+                             ushort* rgb, int rgb_step, CvSize size )
+{
+    int i;
+    for( ; size.height--; )
+    {
+        for( i = 0; i < size.width; i++, bgr += 3, rgb += 3 )
+        {
+            ushort t0 = bgr[0], t1 = bgr[1], t2 = bgr[2];
+            rgb[2] = t0; rgb[1] = t1; rgb[0] = t2;
+        }
+        bgr += bgr_step - size.width*3;
+        rgb += rgb_step - size.width*3;
+    }
+}
+
+
+typedef unsigned short ushort;
+
+void icvCvt_BGR5552Gray_8u_C2C1R( const uchar* bgr555, int bgr555_step,
+                                  uchar* gray, int gray_step, CvSize size )
+{
+    int i;
+    for( ; size.height--; gray += gray_step, bgr555 += bgr555_step )
+    {
+        for( i = 0; i < size.width; i++ )
+        {
+            int t = descale( ((((ushort*)bgr555)[i] << 3) & 0xf8)*cB +
+                             ((((ushort*)bgr555)[i] >> 2) & 0xf8)*cG +
+                             ((((ushort*)bgr555)[i] >> 7) & 0xf8)*cR, SCALE );
+            gray[i] = (uchar)t;
+        }
+    }
+}
+
+
+void icvCvt_BGR5652Gray_8u_C2C1R( const uchar* bgr565, int bgr565_step,
+                                  uchar* gray, int gray_step, CvSize size )
+{
+    int i;
+    for( ; size.height--; gray += gray_step, bgr565 += bgr565_step )
+    {
+        for( i = 0; i < size.width; i++ )
+        {
+            int t = descale( ((((ushort*)bgr565)[i] << 3) & 0xf8)*cB +
+                             ((((ushort*)bgr565)[i] >> 3) & 0xfc)*cG +
+                             ((((ushort*)bgr565)[i] >> 8) & 0xf8)*cR, SCALE );
+            gray[i] = (uchar)t;
+        }
+    }
+}
+
+
+void icvCvt_BGR5552BGR_8u_C2C3R( const uchar* bgr555, int bgr555_step,
+                                 uchar* bgr, int bgr_step, CvSize size )
+{
+    int i;
+    for( ; size.height--; bgr555 += bgr555_step )
+    {
+        for( i = 0; i < size.width; i++, bgr += 3 )
+        {
+            int t0 = (((ushort*)bgr555)[i] << 3) & 0xf8;
+            int t1 = (((ushort*)bgr555)[i] >> 2) & 0xf8;
+            int t2 = (((ushort*)bgr555)[i] >> 7) & 0xf8;
+            bgr[0] = (uchar)t0; bgr[1] = (uchar)t1; bgr[2] = (uchar)t2;
+        }
+        bgr += bgr_step - size.width*3;
+    }
+}
+
+
+void icvCvt_BGR5652BGR_8u_C2C3R( const uchar* bgr565, int bgr565_step,
+                                 uchar* bgr, int bgr_step, CvSize size )
+{
+    int i;
+    for( ; size.height--; bgr565 += bgr565_step )
+    {
+        for( i = 0; i < size.width; i++, bgr += 3 )
+        {
+            int t0 = (((ushort*)bgr565)[i] << 3) & 0xf8;
+            int t1 = (((ushort*)bgr565)[i] >> 3) & 0xfc;
+            int t2 = (((ushort*)bgr565)[i] >> 8) & 0xf8;
+            bgr[0] = (uchar)t0; bgr[1] = (uchar)t1; bgr[2] = (uchar)t2;
+        }
+        bgr += bgr_step - size.width*3;
+    }
+}
+
+
+void icvCvt_CMYK2BGR_8u_C4C3R( const uchar* cmyk, int cmyk_step,
+                               uchar* bgr, int bgr_step, CvSize size )
+{
+    int i;
+    for( ; size.height--; )
+    {
+        for( i = 0; i < size.width; i++, bgr += 3, cmyk += 4 )
+        {
+            int c = cmyk[0], m = cmyk[1], y = cmyk[2], k = cmyk[3];
+            c = k - ((255 - c)*k>>8);
+            m = k - ((255 - m)*k>>8);
+            y = k - ((255 - y)*k>>8);
+            bgr[2] = (uchar)c; bgr[1] = (uchar)m; bgr[0] = (uchar)y;
+        }
+        bgr += bgr_step - size.width*3;
+        cmyk += cmyk_step - size.width*4;
+    }
+}
+
+
+void icvCvt_CMYK2Gray_8u_C4C1R( const uchar* cmyk, int cmyk_step,
+                                uchar* gray, int gray_step, CvSize size )
+{
+    int i;
+    for( ; size.height--; )
+    {
+        for( i = 0; i < size.width; i++, cmyk += 4 )
+        {
+            int c = cmyk[0], m = cmyk[1], y = cmyk[2], k = cmyk[3];
+            c = k - ((255 - c)*k>>8);
+            m = k - ((255 - m)*k>>8);
+            y = k - ((255 - y)*k>>8);
+            int t = descale( y*cB + m*cG + c*cR, SCALE );
+            gray[i] = (uchar)t;
+        }
+        gray += gray_step;
+        cmyk += cmyk_step - size.width*4;
+    }
+}
+
+
+void CvtPaletteToGray( const PaletteEntry* palette, uchar* grayPalette, int entries )
+{
+    int i;
+    for( i = 0; i < entries; i++ )
+    {
+        icvCvt_BGR2Gray_8u_C3C1R( (uchar*)(palette + i), 0, grayPalette + i, 0, cvSize(1,1) );
+    }
+}
+
+
+void  FillGrayPalette( PaletteEntry* palette, int bpp, bool negative )
+{
+    int i, length = 1 << bpp;
+    int xor_mask = negative ? 255 : 0;
+
+    for( i = 0; i < length; i++ )
+    {
+        int val = (i * 255/(length - 1)) ^ xor_mask;
+        palette[i].b = palette[i].g = palette[i].r = (uchar)val;
+        palette[i].a = 0;
+    }
+}
+
+
+bool  IsColorPalette( PaletteEntry* palette, int bpp )
+{
+    int i, length = 1 << bpp;
+
+    for( i = 0; i < length; i++ )
+    {
+        if( palette[i].b != palette[i].g ||
+            palette[i].b != palette[i].r )
+            return true;
+    }
+
+    return false;
+}
+
+
+uchar* FillUniColor( uchar* data, uchar*& line_end,
+                     int step, int width3,
+                     int& y, int height,
+                     int count3, PaletteEntry clr )
+{
+    do
+    {
+        uchar* end = data + count3;
+
+        if( end > line_end )
+            end = line_end;
+
+        count3 -= (int)(end - data);
+
+        for( ; data < end; data += 3 )
+        {
+            WRITE_PIX( data, clr );
+        }
+
+        if( data >= line_end )
+        {
+            line_end += step;
+            data = line_end - width3;
+            if( ++y >= height  ) break;
+        }
+    }
+    while( count3 > 0 );
+
+    return data;
+}
+
+
+uchar* FillUniGray( uchar* data, uchar*& line_end,
+                    int step, int width,
+                    int& y, int height,
+                    int count, uchar clr )
+{
+    do
+    {
+        uchar* end = data + count;
+
+        if( end > line_end )
+            end = line_end;
+
+        count -= (int)(end - data);
+
+        for( ; data < end; data++ )
+        {
+            *data = clr;
+        }
+
+        if( data >= line_end )
+        {
+            line_end += step;
+            data = line_end - width;
+            if( ++y >= height  ) break;
+        }
+    }
+    while( count > 0 );
+
+    return data;
+}
+
+
+uchar* FillColorRow8( uchar* data, uchar* indices, int len, PaletteEntry* palette )
+{
+    uchar* end = data + len*3;
+    while( (data += 3) < end )
+    {
+        *((PaletteEntry*)(data-3)) = palette[*indices++];
+    }
+    PaletteEntry clr = palette[indices[0]];
+    WRITE_PIX( data - 3, clr );
+    return data;
+}
+
+
+uchar* FillGrayRow8( uchar* data, uchar* indices, int len, uchar* palette )
+{
+    int i;
+    for( i = 0; i < len; i++ )
+    {
+        data[i] = palette[indices[i]];
+    }
+    return data + len;
+}
+
+
+uchar* FillColorRow4( uchar* data, uchar* indices, int len, PaletteEntry* palette )
+{
+    uchar* end = data + len*3;
+
+    while( (data += 6) < end )
+    {
+        int idx = *indices++;
+        *((PaletteEntry*)(data-6)) = palette[idx >> 4];
+        *((PaletteEntry*)(data-3)) = palette[idx & 15];
+    }
+
+    int idx = indices[0];
+    PaletteEntry clr = palette[idx >> 4];
+    WRITE_PIX( data - 6, clr );
+
+    if( data == end )
+    {
+        clr = palette[idx & 15];
+        WRITE_PIX( data - 3, clr );
+    }
+    return end;
+}
+
+
+uchar* FillGrayRow4( uchar* data, uchar* indices, int len, uchar* palette )
+{
+    uchar* end = data + len;
+    while( (data += 2) < end )
+    {
+        int idx = *indices++;
+        data[-2] = palette[idx >> 4];
+        data[-1] = palette[idx & 15];
+    }
+
+    int idx = indices[0];
+    uchar clr = palette[idx >> 4];
+    data[-2] = clr;
+
+    if( data == end )
+    {
+        clr = palette[idx & 15];
+        data[-1] = clr;
+    }
+    return end;
+}
+
+
+uchar* FillColorRow1( uchar* data, uchar* indices, int len, PaletteEntry* palette )
+{
+    uchar* end = data + len*3;
+
+    while( (data += 24) < end )
+    {
+        int idx = *indices++;
+        *((PaletteEntry*)(data - 24)) = palette[(idx & 128) != 0];
+        *((PaletteEntry*)(data - 21)) = palette[(idx & 64) != 0];
+        *((PaletteEntry*)(data - 18)) = palette[(idx & 32) != 0];
+        *((PaletteEntry*)(data - 15)) = palette[(idx & 16) != 0];
+        *((PaletteEntry*)(data - 12)) = palette[(idx & 8) != 0];
+        *((PaletteEntry*)(data - 9)) = palette[(idx & 4) != 0];
+        *((PaletteEntry*)(data - 6)) = palette[(idx & 2) != 0];
+        *((PaletteEntry*)(data - 3)) = palette[(idx & 1) != 0];
+    }
+
+    int idx = indices[0] << 24;
+    for( data -= 24; data < end; data += 3, idx += idx )
+    {
+        PaletteEntry clr = palette[idx < 0];
+        WRITE_PIX( data, clr );
+    }
+
+    return data;
+}
+
+
+uchar* FillGrayRow1( uchar* data, uchar* indices, int len, uchar* palette )
+{
+    uchar* end = data + len;
+
+    while( (data += 8) < end )
+    {
+        int idx = *indices++;
+        *((uchar*)(data - 8)) = palette[(idx & 128) != 0];
+        *((uchar*)(data - 7)) = palette[(idx & 64) != 0];
+        *((uchar*)(data - 6)) = palette[(idx & 32) != 0];
+        *((uchar*)(data - 5)) = palette[(idx & 16) != 0];
+        *((uchar*)(data - 4)) = palette[(idx & 8) != 0];
+        *((uchar*)(data - 3)) = palette[(idx & 4) != 0];
+        *((uchar*)(data - 2)) = palette[(idx & 2) != 0];
+        *((uchar*)(data - 1)) = palette[(idx & 1) != 0];
+    }
+
+    int idx = indices[0] << 24;
+    for( data -= 8; data < end; data++, idx += idx )
+    {
+        data[0] = palette[idx < 0];
+    }
+
+    return data;
+}
+
+
+CV_IMPL void
+cvConvertImage( const CvArr* srcarr, CvArr* dstarr, int flags )
+{
+    CvMat* temp = 0;
+
+    CV_FUNCNAME( "cvConvertImage" );
+
+    __BEGIN__;
+
+    CvMat srcstub, *src;
+    CvMat dststub, *dst;
+    int src_cn, dst_cn, swap_rb = flags & CV_CVTIMG_SWAP_RB;
+
+    CV_CALL( src = cvGetMat( srcarr, &srcstub ));
+    CV_CALL( dst = cvGetMat( dstarr, &dststub ));
+
+    src_cn = CV_MAT_CN( src->type );
+    dst_cn = CV_MAT_CN( dst->type );
+
+    if( src_cn != 1 && src_cn != 3 && src_cn != 4 )
+        CV_ERROR( CV_BadNumChannels, "Source image must have 1, 3 or 4 channels" );
+
+    if( CV_MAT_DEPTH( dst->type ) != CV_8U )
+        CV_ERROR( CV_BadDepth, "Destination image must be 8u" );
+
+    if( CV_MAT_CN(dst->type) != 1 && CV_MAT_CN(dst->type) != 3 )
+        CV_ERROR( CV_BadNumChannels, "Destination image must have 1 or 3 channels" );
+
+    if( !CV_ARE_DEPTHS_EQ( src, dst ))
+    {
+        int src_depth = CV_MAT_DEPTH(src->type);
+        double scale = src_depth <= CV_8S ? 1 : src_depth <= CV_32S ? 1./256 : 255;
+        double shift = src_depth == CV_8S || src_depth == CV_16S ? 128 : 0;
+
+        if( !CV_ARE_CNS_EQ( src, dst ))
+        {
+            temp = cvCreateMat( src->height, src->width,
+                (src->type & CV_MAT_CN_MASK)|(dst->type & CV_MAT_DEPTH_MASK));
+            cvConvertScale( src, temp, scale, shift );
+            src = temp;
+        }
+        else
+        {
+            cvConvertScale( src, dst, scale, shift );
+            src = dst;
+        }
+    }
+
+    if( src_cn != dst_cn || (src_cn == 3 && swap_rb) )
+    {
+        uchar *s = src->data.ptr, *d = dst->data.ptr;
+        int s_step = src->step, d_step = dst->step;
+        int code = src_cn*10 + dst_cn;
+        CvSize size = { src->cols, src->rows };
+
+        if( CV_IS_MAT_CONT(src->type & dst->type) )
+        {
+            size.width *= size.height;
+            size.height = 1;
+            s_step = d_step = CV_STUB_STEP;
+        }
+
+        switch( code )
+        {
+        case 13:
+            icvCvt_Gray2BGR_8u_C1C3R( s, s_step, d, d_step, size );
+            break;
+        case 31:
+            icvCvt_BGR2Gray_8u_C3C1R( s, s_step, d, d_step, size, swap_rb );
+            break;
+        case 33:
+            assert( swap_rb );
+            icvCvt_RGB2BGR_8u_C3R( s, s_step, d, d_step, size );
+            break;
+        case 41:
+            icvCvt_BGRA2Gray_8u_C4C1R( s, s_step, d, d_step, size, swap_rb );
+            break;
+        case 43:
+            icvCvt_BGRA2BGR_8u_C4C3R( s, s_step, d, d_step, size, swap_rb );
+            break;
+        default:
+            CV_ERROR( CV_StsUnsupportedFormat, "Unsupported combination of input/output formats" );
+        }
+        src = dst;
+    }
+
+    if( flags & CV_CVTIMG_FLIP )
+    {
+        CV_CALL( cvFlip( src, dst, 0 ));
+    }
+    else if( src != dst )
+    {
+        CV_CALL( cvCopy( src, dst ));
+    }
+
+    __END__;
+
+    cvReleaseMat( &temp );
+}