Cmesh_generate_centroid.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 CMESH_GENERATE_CENTROID
#define CMESH_GENERATE_CENTROID

#include "Cmesh_generate_regularFem.h"

template<typename T,typename Timg>
class Cmesh_generate_centroid : public Cmesh_generate_regularFem<T,Timg>
{
  
  public:
    
  int m_peaks;

  Cmesh_generate_centroid() : Cmesh_generate_regularFem<T,Timg>()
  {// constructor
    
    this->class_name = "Cmesh_generate_centroid";
    
  }
  
  virtual int get_attributes(CParameterNetCDF &fp)
  {// correlation window size and the desired peak number loading
  
    // load components
    Cmesh_generate_regularFem<T,Timg>::get_attributes(fp);
     
    int error(0);
    std::string att;

    //m_win[x,y,z]
    att = "window";
    error = fp.loadAttribute(att,this->m_win);
    for (int i=this->m_win.size(); i<3; i++){this->m_win.push_back(1);} 
    if (this->m_verbose){printf("\tMesh ZOI : [%i %i %i]\n",this->m_win[0], this->m_win[1], this->m_win[2]);}
    
    //m_peaks
    att = "store_peak";
    error = fp.loadAttribute(att,m_peaks);
    if (this->m_verbose){printf("\tNumber of peak : %i\n",m_peaks);}

  }

  virtual int assign(const Cimage<Timg> &oImage)
  {// initialize attribute dimensions
  
    this->m_dof = 1;
    this->m_grid.assign(3,1);
    for (int i=0; i<this->num_var(); i++)
    {
      this->m_grid[i] = (int)ceil((float)oImage.cur_dim(i)/(float)this->m_pitch[i])+1;
      this->m_dof *= this->m_grid[i];
    }
    this->m_dof *= this->num_var();
    if (this->m_verbose){printf("\tDegree Of Freedom : %i\n",this->m_dof);}
    
    this->m_node.assign(this->num_var(), this->m_grid[0],this->m_grid[1],this->m_grid[2],1, 0);
    if (this->m_verbose){printf("\tNode table : [%i](%i,%i,%i)\n",this->m_node.size(), this->m_node[0].width(), this->m_node[0].height(), this->m_node[0].depth());}
    
    if (this->_3D)
    {
      this->m_connect.assign((this->m_grid[0]-1)*(this->m_grid[1]-1)*(this->m_grid[2]-1), 1,1,1,1, 0);
    }
    else
    {
      this->m_connect.assign((this->m_grid[0]-1)*(this->m_grid[1]-1), 1,1,1,1, 0);
    }
    if (this->m_verbose){printf("\tElement table [%i](%i)\n",this->m_connect.size(), this->m_connect[0].width());}
    
    this->m_numPix = this->m_win[0]*this->m_win[1]*this->m_win[2]*this->m_connect.size();
    
    return 0;
    
  }

  virtual void set_connectivity(const Cimage<Timg> &oImage, const CImgList<T> &nodes, CImgList<int> &elements)
  {// element and connectivity table definition
    
    #pragma omp for schedule(runtime)
    cimglist_for(elements, elem)
    {//fill element table

      elements[elem][0] = elem;

    }
    
  }

  virtual int grid_dims(const int &i) const
  {// return mesh grid dimensions
  
    switch(i)
    {
      case 0: return this->m_grid[0]; break;
      case 1: return this->m_grid[1]; break;
      case 2: return this->m_grid[2]; break;
      case 3: return m_peaks; break;
      default:
      {
        std::cerr<<this->class_name<<"::"<<__func__<<": error: id="<<i<<" which is over [0-3]\n"<<std::flush;
        return 0;
      }
    }
  
  }
 
 
};

#endif