Cmesh.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
#define CMESH


//-----------------------------CIMG_NETCDF-------------------------------
#define cimg_plugin  "plugins/add_fileformat.h"
#define cimg_use_netcdf

#ifdef cimg_use_netcdf
 #include "../NetCDF.Tool/struct_parameter_NetCDF.h"
// #define     cimg_plugin1 "plugins/netcdf_file_format4CImg1.h"
// #define cimglist_plugin1 "plugins/netcdf_file_format4CImgList1.h"
 #define     cimg_plugin2 "plugins/netcdf_file_format4CImg2.h"
 #define cimglist_plugin2 "plugins/netcdf_file_format4CImgList2.h"
 #include "../CImg.Tool/CImg_NetCDF.h"
#endif
//-----------------------------CIMG_NETCDF-------------------------------

#include "Cmesh_factory.h"
#include "Cimage.h"
#include "CImg_math.h"
#include "textcolor.h"

template<typename T,typename Timg>
class Cmesh
{
  
  public:
    
  std::string class_name;
  std::string m_type, m_name;
  
  std::vector<std::string> m_componentName; //Composante names
  
  CImgList<T> m_node; // node table
  CImgList<int> m_connect; // connectivity table
  
  int _3D, m_threadNB, m_numPix, m_storeMesh;
  bool m_verbose;

  int m_dof;
  std::vector<int> m_pitch;
  std::vector<int> m_win;

    
  Cmesh()
  {// constructor 
  
      class_name = "mesh : mother";
      
      #pragma omp parallel
      {m_threadNB = omp_get_num_threads();}

  }  

  virtual int exec(const Cimage<Timg> &oImage)
  {//exec

    int error = assign(oImage);

    #pragma omp parallel
    {
      set_nodes(oImage, m_node); //generate node table
      set_connectivity(oImage, m_node, m_connect);// generate connectivity table
    }
    
    return error;
    
  }
  
  void init(CParameterNetCDF &fp)
  {// mesh initialization
    
    get_attributes(fp);

  }

  virtual int get_attributes(CParameterNetCDF &fp)
  {// get mesh attributes

    int error(0);
    std::string att;
    
    att = "component";
    error = fp.loadAttribute(att,m_componentName);
    if (m_componentName.size() == 3){_3D = 1;}else{_3D = 0;}
    
    if (this->m_verbose){std::cout<<"\tcomponents : [ ";for (int i=0; i<m_componentName.size(); i++){std::cout<<m_componentName[i]<<" ";}std::cout<<"]\n";}
    
  }

  virtual int cropList(const Cimage<Timg> &oImage, const int &elem, CImg<Timg> &list, std::vector<T> &coord0) const = 0;
  
  int num_node() const
  {// components dimension
  
    return m_node[0].size();
    
  }  
  
  int num_elem() const
  {// components dimension
  
    return m_connect.size();
    
  }  
  
  int num_var() const
  {
    
    return m_componentName.size();
    
  }
  
  virtual int num_mod() const
  {
    
    return m_connect[0].size();
    
  }

  virtual int grid_dims(const int &i) const = 0;
  virtual void set_nodes(const Cimage<Timg> &oImage, CImgList<T> &nodes) = 0;
  virtual void set_connectivity(const Cimage<Timg> &oImage, const CImgList<T> &nodes, CImgList<int> &elements) = 0;
  virtual int assign(const Cimage<Timg> &oImage) = 0;
 
};

#endif