Cmesh_generate_centroid_integrated.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_INTEGRATED
#define CMESH_GENERATE_CENTROID_INTEGRATED

#include "Cmesh_generate_centroid.h"

template<typename T,typename Timg>
class Cmesh_generate_centroid_integrated : public Cmesh_generate_centroid<T,Timg>
{
  
  public:
    
  int m_mode_num;

  Cmesh_generate_centroid_integrated() : Cmesh_generate_centroid<T,Timg>()
  {
    this->class_name = "Cmesh_generate_centroid_integrated";
    
  }
  
  virtual int get_attributes(CParameterNetCDF &fp)
  {
    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_mode_num = this->m_dof;

  }

  virtual int cropList(const Cimage<Timg> &oImage, const int &elem, CImg<Timg> &list, std::vector<T> &coord0) const
  {//extract sub_box from global image one using node and element table
  
    std::vector<T> coord1;
    coord0.assign(3,0);coord1.assign(3,0);
    int msk(1);
    double unmasked(num_mod()*num_mod());

    cimglist_for(this->m_node, dim)
    {//find corners positions
      
      coord0[dim] = this->m_node[dim][this->m_connect[elem].front()] - this->m_win[dim]/2 + 1;
      coord1[dim] = this->m_node[dim][this->m_connect[elem].front()] - this->m_win[dim]/2 + this->m_win[dim] - 1;
      
      if (coord0[dim] < 1)
      {
        coord0[dim] = 1;
      }
      
      if (coord1[dim] >= oImage.cur_dim(dim)-1)
      {
        coord1[dim] = oImage.cur_dim(dim) - 2;
      }
      
    }

    if (!oImage.m_curMask.is_empty())
    {//number of unmasked pixels within element
      unmasked = oImage.m_curMask.get_crop(coord0[0],coord0[1],coord0[2],0,coord1[0],coord1[1],coord1[2],0,0).sum();
    }
    
    
    if (unmasked >= num_mod()*2)
    {list.assign(coord1[0]-coord0[0]+1, coord1[1]-coord0[1]+1, coord1[2]-coord0[2]+1, 1, 0);}
    else
    {msk = 0;}

    return msk;
    
  }
  
  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 1; break;
      default:
      {
        std::cerr<<this->class_name<<"::"<<__func__<<": error: id="<<i<<" which is over [0-3]\n"<<std::flush;
        return 0;
      }
    }
  
  }
 
  virtual int num_mod() const
  {
    
    return m_mode_num;
    
  }
 
  
};

#endif