Cpeak_fit.h

Go to the documentation of this file.

/**********************************************************************
 * Copyright (C) 2012, The YaDICs Project Developers. 
 * See the COPYRIGHT file at the top-level directory of this distribution ./COPYRIGHT.
 * See ./COPYING file for copying and redistribution conditions.
 * 
 * This file is part of YaDICs.
 * YaDICs is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * YaDICs is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with YaDICs.  If not, see <http://www.gnu.org/licenses/>.
 * 
 * Information about how to use the software are provided at http://yadic.univ-lille1.fr/
 **********************************************************************/


#ifndef CPEAK_FIT
#define CPEAK_FIT

#include "Cpeak_subPix.h"

template<typename T, typename Timg>
class Cpeak_fit : public Cpeak_subPix<T,Timg>
{
  
  public:
    
    std::string class_name;
    std::vector<Timg> m_parameters;
    T m_neighb, m_CornerX, m_CornerY;
    
    
    Cpeak_fit() : Cpeak_subPix<T,Timg>(), m_neighb(20), m_CornerX(0), m_CornerY(0)
    {// constructor
      
      m_parameters.clear();
      
      class_name = "Cpeak_fit";
      
    }
     
    virtual void sub_pixel(CImg<Timg> &img)
    {// method sub_pixel 
      
      CImg<Timg> imgC;
      
      imgC = img.get_crop(this->m_x0-m_neighb,this->m_y0-m_neighb,this->m_x0+m_neighb,this->m_y0+m_neighb, "false");
      
      peak_fit(imgC);
      
      set_position();
      
    } 
    
   
    void ssd(CImg<Timg> &img)
    {
      
      CImg<Timg> n(6,1,1,1,1,0,0,0,0,0), b(1,6,1,1,0);
      CImg<Timg> nnt(6,6,1,1,0);
      
      cimg_forXY(img,x,y){
        
        n(1)=x;
        n(2)=y;
        n(3)=x*x;
        n(4)=y*y;
        n(5)=x*y;
        nnt+=matrixProduct(n.get_transpose(),n);
        b+=(n.get_transpose())*img(x,y);
        
      };

      b.solve(nnt);
      
      for (int i=0; i<b.height(); i++)
      {
        
        m_parameters.push_back(b(i));
        
      };
      
    }// method ssd
         
    void set_position()
    {
      
      //corner position of the ssd box
      m_CornerX = this->m_x0-m_neighb;
      m_CornerY = this->m_y0-m_neighb;
      
      //floating position within the box
      T X1(-2*(m_parameters[1]*m_parameters[4]+m_parameters[2]*m_parameters[5])/(4*m_parameters[3]*m_parameters[4]-m_parameters[5]*m_parameters[5])),
      Y1((-m_parameters[1]*m_parameters[5]+2*m_parameters[2]*m_parameters[3])/(m_parameters[5]*m_parameters[5]-4*m_parameters[4]*m_parameters[3]));
      
      //sub-pixel peak position
      (*this).m_x0 = m_CornerX+X1;
      (*this).m_y0 = m_CornerY+Y1;
      
      if(m_CornerX<0){m_CornerX=0;};
      if(m_CornerY<0){m_CornerY=0;};
      
    }
    
    virtual void peak_fit(CImg<Timg> &img) = 0;
   
};

#endif