////////////////////////////////////////////////////////
//
// GEM - Graphics Environment for Multimedia
//
// zmoelnig@iem.kug.ac.at
//
// Implementation file
//
//    Copyright (c) 1997-2000 Mark Danks.
//    Copyright (c) Günther Geiger.
//    Copyright (c) 2001-2002 IOhannes m zmoelnig. forum::für::umläute. IEM
//    Copyright (c) 2002 James Tittle & Chris Clepper
//    For information on usage and redistribution, and for a DISCLAIMER OF ALL
//    WARRANTIES, see the file, "GEM.LICENSE.TERMS" in this distribution.
//
/////////////////////////////////////////////////////////

#include "pix_opencv_hist_compare.h"

CPPEXTERN_NEW(pix_opencv_hist_compare)

/////////////////////////////////////////////////////////
//
// pix_opencv_hist_compare
//
/////////////////////////////////////////////////////////
// Constructor
//
/////////////////////////////////////////////////////////

pix_opencv_hist_compare :: pix_opencv_hist_compare()
{ 
  comp_xsize=320;
  comp_ysize=240;

  inlet_new(this->x_obj, &this->x_obj->ob_pd, gensym("float"), gensym("save"));
  m_dataout = outlet_new(this->x_obj, &s_anything);
  m_measureout = outlet_new(this->x_obj, &s_anything);

  save_now = 0;

  rgba = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
  rgb = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
  grey = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
  hsv = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3 );

  h_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
  s_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
  v_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
  planes[0] = h_plane;
  planes[1] = s_plane;
  h_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
  s_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
  v_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
  saved_planes[0] = h_saved_plane;
  saved_planes[1] = s_saved_plane;

  int h_bins = (int)(comp_xsize/10), s_bins = (int)(comp_ysize/10);
  {
     int    hist_size[]  = { h_bins, s_bins };
     float  h_ranges[]   = { 0, 180 };         // hue is [0,180]
     float  s_ranges[]   = { 0, 255 };
     float* ranges[]     = { h_ranges, s_ranges };
     hist = cvCreateHist(
            2,
            hist_size,
            CV_HIST_ARRAY,
            ranges,
            1
      );
      int n;
      for (n=0; n<MAX_HISTOGRAMS_TO_COMPARE; n++) {
          saved_hist[n] = cvCreateHist(
                  2,
                  hist_size,
                  CV_HIST_ARRAY,
                  ranges,
                  1
          );
      }
  }
  nbsaved=0;
}

/////////////////////////////////////////////////////////
// Destructor
//
/////////////////////////////////////////////////////////
pix_opencv_hist_compare :: ~pix_opencv_hist_compare()
{
    //Destroy cv_images to clean memory
    cvReleaseImage(&rgba);
    cvReleaseImage(&rgb);
    cvReleaseImage(&grey);
    cvReleaseImage(&h_plane);
    cvReleaseImage(&s_plane);
    cvReleaseImage(&v_plane);
    cvReleaseImage(&h_saved_plane);
    cvReleaseImage(&s_saved_plane);
    cvReleaseImage(&v_saved_plane);

}

/////////////////////////////////////////////////////////
// processImage
//
/////////////////////////////////////////////////////////
    	
void pix_opencv_hist_compare :: processImage(imageStruct &image)
{ 
  int h_bins = (int)(comp_xsize/10), s_bins = (int)(comp_ysize/10);

  if ((this->comp_xsize!=image.xsize)&&(this->comp_ysize!=image.ysize)) {

    this->comp_xsize=image.xsize;
    this->comp_ysize=image.ysize;

    //Destroy cv_images to clean memory
    cvReleaseImage(&rgba);
    cvReleaseImage(&rgb);
    cvReleaseImage(&grey);
    cvReleaseImage(&h_plane);
    cvReleaseImage(&s_plane);
    cvReleaseImage(&v_plane);
    cvReleaseImage(&h_saved_plane);
    cvReleaseImage(&s_saved_plane);
    cvReleaseImage(&v_saved_plane);

    //Create cv_images 
    rgba = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
    rgb = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
    grey = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
    hsv = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3 );
      
      
    h_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
    s_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
    v_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
    planes[0] = h_plane;
    planes[1] = s_plane;
    cvCvtPixToPlane( hsv, h_plane, s_plane, v_plane, 0 );
    h_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
    s_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
    v_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
    saved_planes[0] = h_saved_plane;
    saved_planes[1] = s_saved_plane;
      
    h_bins = (int)(comp_xsize/10);
    s_bins = (int)(comp_ysize/10);
    {
      int    hist_size[]  = { h_bins, s_bins };
      float  h_ranges[]   = { 0, 180 };         // hue is [0,180]
      float  s_ranges[]   = { 0, 255 };
      float* ranges[]     = { h_ranges, s_ranges };
      if ( hist ) { 
        cvReleaseHist( &hist );
        hist = NULL;
      }
      
      hist = cvCreateHist(
                  2,
                  hist_size,
                  CV_HIST_ARRAY,
                  ranges,
                  1
      );
      int n;
      for (n=0; n<MAX_HISTOGRAMS_TO_COMPARE; n++) {
          if ( saved_hist[n] ) {
            cvReleaseHist( &saved_hist[n] );
            saved_hist[n] = NULL;
          }
          
          saved_hist[n] = cvCreateHist(
                    2,
                    hist_size,
                    CV_HIST_ARRAY,
                    ranges,
                    1
          );
      }
    }
      
  }
  
  // no need to copy the image, just create valid header and point to image.data...
  IplImage* imgMat = cvCreateImageHeader( cv::Size(image.xsize, image.ysize), IPL_DEPTH_8U, image.csize);
  imgMat->imageData = (char*) image.data; 
  
  if ( image.csize == 1 ){ // gray
    cvCvtColor(imgMat, rgb, CV_GRAY2BGR);
  } else if ( image.csize == 4 ) { //RGBA
    cvCvtColor(imgMat, rgb, CV_BGRA2BGR);
  } else {
    error("support only RGBA or GRAY image");
    return;
  }

  // Convert to hsv
  cvCvtColor( rgb, hsv, CV_BGR2HSV );
  cvCvtPixToPlane( hsv, h_plane, s_plane, v_plane, 0 );
  
  // Build the histogram and compute its contents.
  if (save_now>=0) {
       post("saving histogram %d\n",save_now);
       cvCvtPixToPlane( hsv, h_saved_plane, s_saved_plane, v_saved_plane, 0 );
       cvCalcHist( saved_planes, saved_hist[save_now], 0, 0 ); //Compute histogram
       cvNormalizeHist( saved_hist[save_now], 1.0 );  //Normalize it
       save_now=-1;
       nbsaved++;
   }
   cvCalcHist( planes, hist, 0, 0 ); //Compute histogram
   cvNormalizeHist( hist, 1.0 );  //Normalize it

   if ( nbsaved > 0 ) {
      double* tato = new double[nbsaved];
      t_atom* datalist = new t_atom[nbsaved];
      int nearest = -1;
      double max  =  0;

      for ( int n=0; n<nbsaved; n++) {
        tato[n] = cvCompareHist(hist, saved_hist[n], CV_COMP_INTERSECT);
        SETFLOAT(&datalist[n], tato[n]);
        if (tato[n]>max) {
                max = tato[n];
                nearest = n;
        }
      }
      outlet_float(m_dataout, (float)nearest);
      outlet_list( m_measureout, 0, nbsaved , datalist );
      
      if ( tato ) delete[] tato; tato=NULL;
      if ( datalist ) delete[] datalist; datalist=NULL;
   } else {
      outlet_float(m_dataout, -1.0);
   }

   // Create an image to use to visualize our histogram.
   int scale = 10;
   // populate our visualization with little gray squares.
   float max_value = 0;
   cvGetMinMaxHistValue( hist, 0, &max_value, 0, 0 );

   int h = 0;
   int s = 0;

   for( h = 0; h < h_bins; h++ ) {
      for( s = 0; s < s_bins; s++ ) {
          float bin_val = cvQueryHistValue_2D( hist, h, s );
          int intensity = cvRound( bin_val * 255 / max_value );
          cvRectangle(
                imgMat,
                cvPoint( h*scale, s*scale ),
                cvPoint( (h+1)*scale - 1, (s+1)*scale - 1),
                CV_RGB(intensity,intensity,intensity), CV_FILLED, 8 , 0 );
        }
   }
}

/////////////////////////////////////////////////////////
// static member function
//
/////////////////////////////////////////////////////////

void pix_opencv_hist_compare :: obj_setupCallback(t_class *classPtr)
{
  class_addmethod(classPtr, (t_method)&pix_opencv_hist_compare::saveMessCallback,
                  gensym("save"), A_FLOAT, A_NULL);
}

void pix_opencv_hist_compare :: saveMess(float index)
{
    if (((int)index>=0)&&((int)index<MAX_HISTOGRAMS_TO_COMPARE)) save_now = (int)index;
}

void pix_opencv_hist_compare :: saveMessCallback(void *data, t_floatarg index)
{
  GetMyClass(data)->saveMess(index);
}