Cpeak_barycenter.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_BARYCENTER
#define CPEAK_BARYCENTER

#include "Cpeak_subPix.h"


template<typename T> 
class Cpeak_barycenter : public Cpeak_subPix<T>
{
  
  public:
    
    std::string class_name;
    
    Cpeak_barycenter() : Cpeak_subPix<T>()
    {// constructor
      
      class_name = "Cpeak_barycenter";
      
    }
    
    virtual void sub_pixel(CImg<T> &img)
    {
      
      std::vector<float> mass;
      float sum(0), neighb(5);
      
      for (int i=0; i<this->m_pos.size(); i++)
      {//initialization of directionnal mass
        mass.push_back(0);
      }
      
      CImg<T> imgC;
      switch(this->m_pos.size())
      {
        case 1: imgC = img.get_crop(this->m_pos[0]-neighb,0,0,0,this->m_pos[0]+neighb,0,0,0, false); break;
        case 2: imgC = img.get_crop(this->m_pos[0]-neighb,this->m_pos[1]-neighb,0,0,this->m_pos[0]+neighb,this->m_pos[1]+neighb,0,0, false); break;
        case 3: imgC = img.get_crop(this->m_pos[0]-neighb,this->m_pos[1]-neighb,this->m_pos[2]-neighb,0,this->m_pos[0]+neighb,this->m_pos[1]+neighb,this->m_pos[2]+neighb,0, false); break;
        case 4: imgC = img.get_crop(this->m_pos[0]-neighb,this->m_pos[1]-neighb,this->m_pos[2]-neighb,this->m_pos[3]-neighb,this->m_pos[0]+neighb,this->m_pos[1]+neighb,this->m_pos[2]+neighb,this->m_pos[3]+neighb, false); break;
        default:
        {
          std::cerr<<class_name<<"::"<<__func__<<": error: id="<<this->m_pos.size()<<" is unknown as CImg dimensions\n"<<std::flush;break;
        }
      }

      cimg_forXYZC(imgC, x, y, z, c)
      {
        mass[0]+=imgC(x,y,z,c)*(x-neighb);
        if(mass.size()>1){mass[1]+=imgC(x,y,z,c)*(y-neighb);}
        if(mass.size()>2){mass[2]+=imgC(x,y,z,c)*(z-neighb);}           
        if(mass.size()>3){mass[3]+=imgC(x,y,z,c)*(c-neighb);}   
      }
      
      sum = imgC.sum();
      
      for (int i =0; i<this->m_pos.size(); i++)
      {
        this->m_pos[i] = this->m_pos[i] + mass[i]/sum;
      }
      
    }
    
};

#endif