/**
* @file LFonHSG.cpp
* @brief Implementation of the LocalFeatureHSG.h. See description there.
* @author Eric Bach
* @date 26.10.2011
*/

/* LocalFeatureHSG Include */

#include "vislearning/features/localfeatures/LFonHSG.h"

using namespace OBJREC;
using namespace NICE;
using namespace std;

LFonHSG::LFonHSG ( const Config *conf, const string section )
{
  /** initialization **/
  this->lf = NULL;

  /** get parameters for the grid **/
  sampleScaling    = conf->gI ( section, "sample_scaling", 50 );
  scales   = conf->gS ( section, "scales"        , "1" );
  // the scales are seperated by '+', like in the Van de Sande implementation
  char separator = '+';

  /** get debuginformation **/
  debug    = conf->gB ( "debug", "show_log_HSG", false );

  /** generate the descriptor-instanz **/
  lf = GenericLocalFeatureSelection::selectLocalFeature ( conf, section );

  /** parse scales string **/
  debug && clog << "[log] LocalFeatureHSG::LocalFeatureHSG" << endl;
  debug && clog << "[log] try to parse the 'scales-string': " << scales << " -> ";

#ifdef NICE_USELIB_BOOST
  typedef tokenizer<boost::char_separator<char> > tokenizer;
  char_separator<char> sep ( separator );
  tokenizer tokens ( scales, sep );  // parse the string into tokens
  for ( tokenizer::iterator tok_iter = tokens.begin(); tok_iter != tokens.end(); ++tok_iter )
  {
    debug && clog << *tok_iter << " ";
    scalesV.push_back ( strToFloat ( *tok_iter ) );
  }
#else // standard
  vector<string> temp;
  StringTools::split ( scales, separator, temp );
  for ( vector<string>::const_iterator it = temp.begin(); it != temp.end(); ++it )
  {
    debug && clog << *it << " ";
    scalesV.push_back ( strToFloat ( *it ) );
  }
#endif
  debug && clog << endl;
}

LFonHSG::~LFonHSG()
{
  /** free memory of descriptors **/
}

void LFonHSG::getPositionsOnHSG ( const unsigned int imageWidth, const unsigned int imageHeight, VVector& positions ) const
{
  if ( sampleScaling < 1 ) {
    cerr << "[err] sample-scaling (" << sampleScaling << ") musst be larger the 0!" << endl;
    return;
  }

  debug && clog << "[log] LocalFeatureHSG::getPositionsOnHSG calculate ";

  bool oddRow = true;
  NICE::Vector pos ( 4 );

  /** we have to calculate the koo. for every different scale **/
  for ( vector <float>::const_iterator it = scalesV.begin(); it != scalesV.end(); ++it )
  {
    oddRow = true;

    for ( unsigned int j = sampleScaling; j <= ( imageHeight - sampleScaling ); j += sampleScaling )
    {
      for ( unsigned int i = ( oddRow ? sampleScaling + sampleScaling / 2 : sampleScaling ); i <= ( imageWidth - sampleScaling ); i += sampleScaling )
      {
        pos[ 0] = i;
        pos[ 1] = j;
        pos[ 2] = *it;
        pos[ 3] = 0;
        positions.push_back ( pos );
      }
      oddRow = !oddRow;
    }
  }
}

int LFonHSG::extractFeatures ( const NICE::ColorImage & cimg, VVector & features, VVector & positions ) const
{
  /** To get the keypoint descriptor, we need the positions of the keypoints. **/
  getPositionsOnHSG ( cimg.width(), cimg.height(), positions );

  /** calulate the descriptor-values **/
  return lf->getDescriptors ( cimg, positions, features );
}


int LFonHSG::extractFeatures ( const NICE::Image & img, VVector & features, VVector & positions ) const
{
  /** To get the keypoint descriptor, we need the positions of the keypoints. **/
  getPositionsOnHSG ( img.width(), img.height(), positions );

  /** calculate the descriptor-values **/
  return lf->getDescriptors ( img, positions, features );
}

void LFonHSG::visualizeFeatures ( NICE::Image & mark, const VVector & positions, size_t color ) const
{
  // TODO: Implementierung des gewaehlten Descriptortyps aufrufen.
}

float LFonHSG::strToFloat ( const string str ) const
{
  float temp;

  stringstream ss;
  ss << str;
  ss >> temp;

  return temp;
}