#include "StdAfx.h"

#include "CvvImage.h"



//////////////////////////////////////////////////////////////////////

// Construction/Destruction

//////////////////////////////////////////////////////////////////////



CV_INLINE RECT NormalizeRect( RECT r );

CV_INLINE RECT NormalizeRect( RECT r )

{

   int t;



   if( r.left > r.right )

   {

      t = r.left;

      r.left = r.right;

      r.right = t;

   }



   if( r.top > r.bottom )

   {

      t = r.top;

      r.top = r.bottom;

      r.bottom = t;

   }



   return r;

}



CV_INLINE CvRect RectToCvRect( RECT sr );

CV_INLINE CvRect RectToCvRect( RECT sr )

{

   sr = NormalizeRect( sr );

   return cvRect( sr.left, sr.top, sr.right - sr.left, sr.bottom - sr.top );

}



CV_INLINE RECT CvRectToRect( CvRect sr );

CV_INLINE RECT CvRectToRect( CvRect sr )

{

   RECT dr;

   dr.left = sr.x;

   dr.top = sr.y;

   dr.right = sr.x + sr.width;

   dr.bottom = sr.y + sr.height;



   return dr;

}



CV_INLINE IplROI RectToROI( RECT r );

CV_INLINE IplROI RectToROI( RECT r )

{

   IplROI roi;

   r = NormalizeRect( r );

   roi.xOffset = r.left;

   roi.yOffset = r.top;

   roi.width = r.right - r.left;

   roi.height = r.bottom - r.top;

   roi.coi = 0;



   return roi;

}



void  FillBitmapInfo( BITMAPINFO* bmi, int width, int height, int bpp, int origin )

{

   assert( bmi && width >= 0 && height >= 0 && (bpp == 8 || bpp == 24 || bpp == 32));



   BITMAPINFOHEADER* bmih = &(bmi->bmiHeader);



   memset( bmih, 0, sizeof(*bmih));

   bmih->biSize = sizeof(BITMAPINFOHEADER);

   bmih->biWidth = width;

   bmih->biHeight = origin ? abs(height) : -abs(height);

   bmih->biPlanes = 1;

   bmih->biBitCount = (unsigned short)bpp;

   bmih->biCompression = BI_RGB;



   if( bpp == 8 )

   {

      RGBQUAD* palette = bmi->bmiColors;

      int i;

      for( i = 0; i < 256; i++ )

      {

         palette[i].rgbBlue = palette[i].rgbGreen = palette[i].rgbRed = (BYTE)i;

         palette[i].rgbReserved = 0;

      }

   }

}



CvvImage::CvvImage()

{

   m_img = 0;

}



void CvvImage::Destroy()

{

   cvReleaseImage( &m_img );

}



CvvImage::~CvvImage()

{

   Destroy();

}



bool  CvvImage::Create( int w, int h, int bpp, int origin )

{

   const unsigned max_img_size = 10000;



   if( (bpp != 8 && bpp != 24 && bpp != 32) ||

      (unsigned)w >=  max_img_size || (unsigned)h >= max_img_size ||

      (origin != IPL_ORIGIN_TL && origin != IPL_ORIGIN_BL))

   {

      assert(0); // most probably, it is a programming error

      return false;

   }



   if( !m_img || Bpp() != bpp || m_img->width != w || m_img->height != h )

   {

      if( m_img && m_img->nSize == sizeof(IplImage))

         Destroy();



      /* prepare IPL header */

      m_img = cvCreateImage( cvSize( w, h ), IPL_DEPTH_8U, bpp/8 );

   }



   if( m_img )

      m_img->origin = origin == 0 ? IPL_ORIGIN_TL : IPL_ORIGIN_BL;



   return m_img != 0;

}



void  CvvImage::CopyOf( CvvImage& image, int desired_color )

{

   IplImage* img = image.GetImage();

   if( img )

   {

      CopyOf( img, desired_color );

   }

}





#define HG_IS_IMAGE(img)                                                  \

   ((img) != 0 && ((const IplImage*)(img))->nSize == sizeof(IplImage) && \

   ((IplImage*)img)->imageData != 0)





void  CvvImage::CopyOf( IplImage* img, int desired_color )

{

   if( HG_IS_IMAGE(img) )

   {

      int color = desired_color;

      CvSize size = cvGetSize( img ); 



      if( color < 0 )

         color = img->nChannels > 1;



      if( Create( size.width, size.height,

         (!color ? 1 : img->nChannels > 1 ? img->nChannels : 3)*8,

         img->origin ))

      {

         cvConvertImage( img, m_img, 0 );

      }

   }

}





bool  CvvImage::Load( const char* filename, int desired_color )

{

   IplImage* img = cvLoadImage( filename, desired_color );

   if( !img )

      return false;



   CopyOf( img, desired_color );

   cvReleaseImage( &img );



   return true;

}





bool  CvvImage::LoadRect( const char* filename,

                   int desired_color, CvRect r )

{

   if( r.width < 0 || r.height < 0 ) return false;



   IplImage* img = cvLoadImage( filename, desired_color );

   if( !img )

      return false;



   if( r.width == 0 || r.height == 0 )

   {

      r.width = img->width;

      r.height = img->height;

      r.x = r.y = 0;

   }



   if( r.x > img->width || r.y > img->height ||

      r.x + r.width < 0 || r.y + r.height < 0 )

   {

      cvReleaseImage( &img );

      return false;

   }



   /* truncate r to source image */

   if( r.x < 0 )

   {

      r.width += r.x;

      r.x = 0;

   }

   if( r.y < 0 )

   {

      r.height += r.y;

      r.y = 0;

   }



   if( r.x + r.width > img->width )

      r.width = img->width - r.x;



   if( r.y + r.height > img->height )

      r.height = img->height - r.y;



   cvSetImageROI( img, r );

   CopyOf( img, desired_color );



   cvReleaseImage( &img );

   return true;

}





bool  CvvImage::Save( const char* filename )

{

   if( !m_img )

      return false;

   cvSaveImage( filename, m_img );

   return true;

}





void  CvvImage::Show( const char* window )

{

   if( m_img )

      cvShowImage( window, m_img );

}





void  CvvImage::Show( HDC dc, int x, int y, int w, int h, int from_x, int from_y )

{

   if( m_img && m_img->depth == IPL_DEPTH_8U )

   {

      uchar buffer[sizeof(BITMAPINFOHEADER) + 1024];

      BITMAPINFO* bmi = (BITMAPINFO*)buffer;

      int bmp_w = m_img->width, bmp_h = m_img->height;



      FillBitmapInfo( bmi, bmp_w, bmp_h, Bpp(), m_img->origin );



      from_x = MIN( MAX( from_x, 0 ), bmp_w - 1 );

      from_y = MIN( MAX( from_y, 0 ), bmp_h - 1 );



      int sw = MAX( MIN( bmp_w - from_x, w ), 0 );

      int sh = MAX( MIN( bmp_h - from_y, h ), 0 );



      SetDIBitsToDevice(

         dc, x, y, sw, sh, from_x, from_y, from_y, sh,

         m_img->imageData + from_y*m_img->widthStep,

         bmi, DIB_RGB_COLORS );

   }

}





void  CvvImage::DrawToHDC( HDC hDCDst, RECT* pDstRect ) 

{

   if( pDstRect && m_img && m_img->depth == IPL_DEPTH_8U && m_img->imageData )

   {

      uchar buffer[sizeof(BITMAPINFOHEADER) + 1024];

      BITMAPINFO* bmi = (BITMAPINFO*)buffer;

      int bmp_w = m_img->width, bmp_h = m_img->height;



      CvRect roi = cvGetImageROI( m_img );

      CvRect dst = RectToCvRect( *pDstRect );



      if( roi.width == dst.width && roi.height == dst.height )

      {

         Show( hDCDst, dst.x, dst.y, dst.width, dst.height, roi.x, roi.y );

         return;

      }



      if( roi.width > dst.width )

      {

         SetStretchBltMode(

            hDCDst,           // handle to device context

            HALFTONE );

      }

      else

      {

         SetStretchBltMode(

            hDCDst,           // handle to device context

            COLORONCOLOR );

      }



      FillBitmapInfo( bmi, bmp_w, bmp_h, Bpp(), m_img->origin );



      ::StretchDIBits(

         hDCDst,

         dst.x, dst.y, dst.width, dst.height,

         roi.x, roi.y, roi.width, roi.height,

         m_img->imageData, bmi, DIB_RGB_COLORS, SRCCOPY );

   }

}





void  CvvImage::Fill( int color )

{

   cvSet( m_img, cvScalar(color&255,(color>>8)&255,(color>>16)&255,(color>>24)&255) );

}