Cmesh_factory.h

/**********************************************************************
 * 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_FACTORY
#define CMESH_FACTORY


//-----------------------------------------------------------------------
#include "Cmesh.h"
  #include "Cmesh_global.h"
  #include "Cmesh_import.h"
    #include "Cmesh_import_image.h"
//     #include "Cmesh_import_T3.h"
  #include "Cmesh_generate.h"
    #include "Cmesh_generate_regularFem.h"
      #include "Cmesh_generate_centroid.h"
        #include "Cmesh_generate_centroid_integrated.h"
//-----------------------------------------------------------------------


template<typename T, typename Timg>
class Cmesh_factory
{
  
  public:
    
  //attributes
  std::string class_name, m_variable, m_type;
  std::vector<std::string> m_infos;
  CParameterNetCDF m_paramFile;
  
  const char* m_paramPath;
  bool m_verbose;
  
  long m_num_mode;
  

  Cmesh_factory(const std::string &name)
  {//constructor
 
    class_name="Mesh : factory";
        
    m_variable = name;
    
    m_infos.push_back("global");
    m_infos.push_back("generate_regularFem");
    m_infos.push_back("generate_centroid");
    m_infos.push_back("generate_centroid_integrated");
    m_infos.push_back("import_image");

  }

  Cmesh<T,Timg>* create(const int &number)
  {//create from int
 
    Cmesh<T,Timg>* pMesh;
    
    switch(number)
    {
      case 0:
        pMesh = new Cmesh_global<T,Timg>;
        pMesh->m_dof = m_num_mode; 
      break;
      case 1: pMesh = new Cmesh_generate_regularFem<T,Timg>; break;
      case 2: pMesh = new Cmesh_generate_centroid<T,Timg>; break;
      case 3:
        pMesh = new Cmesh_generate_centroid_integrated<T,Timg>; 
        pMesh->m_dof = m_num_mode; 
      break;
      case 4:
        pMesh = new Cmesh_import_image<T,Timg>; 
        pMesh->m_dof = m_num_mode; 
      break;
      default:
      {
        std::cerr<<class_name<<" : "<<number<<" refers to an unknown or incompatible mesh class"<<std::flush;
        return NULL;
      }
    }

    if (m_verbose){printf("%s -> %s class loaded\n",class_name.c_str(), m_type.c_str());}
    
    pMesh->m_verbose = m_verbose;// active or not the verbose mode
    pMesh->m_type = m_type;// string correl-type as member
    pMesh->m_name = m_variable;// string correl-name as member
    
    pMesh->init(m_paramFile);//specific init

    return pMesh;// return pointer to specific correlation method
    
  }
  
  Cmesh<T,Timg>* create(const std::string &type)
  {//create from string

    //if this function is used directly
    m_type = type;
  
    if(type==m_infos[0]) return create(0);
    else if(type==m_infos[1]) return create(1);
    else if(type==m_infos[2]) return create(2);
    else if(type==m_infos[3]) return create(3);
    else if(type==m_infos[4]) return create(4);
    else
    {
      std::cerr<<class_name<<" : "<<type<<" refers to an unknown or incompatible mesh class"<<std::flush;
      return NULL;
    }
    
  }
   
  Cmesh<T,Timg>* create(CParameterNetCDF &paramFile)
  {//create from CParameterNetCDF

    //if this function is used directly
    m_paramFile = paramFile;
  
    //find the variable name associated to the mesh attribute within correlation method
    std::string mesh_name;
    int var(0);
    
    paramFile.loadVar(var,&m_variable);
    paramFile.loadAttribute("mesh", mesh_name);
    
    if(m_verbose){printf("mesh name : %s\n", mesh_name.c_str());}
 
    if (m_variable.compare(0,3,"OFI") == 0)
    {
      std::vector<std::string> modes;
      m_paramFile.loadAttribute("modes", modes);
      m_num_mode += modes.size() - 1;
    }

    //find the mesh type
    m_variable = mesh_name;
    paramFile.loadVar(var,&m_variable);
    paramFile.loadAttribute("type", m_type);
    
    if(m_verbose){printf("mesh type : %s\n", m_type.c_str());}

    return create(m_type);
    
  }
 
  Cmesh<T,Timg>* create(int argc, char *argv[])
  {//create from (argv,argc)

    commandLine(argc,argv);//read command line argument

    Cmesh<T,Timg> *pMesh;// pointer toward a correlation class
    
    int error;
    
    if(m_paramPath!="true")
    {// parameter.nc location is provided
    
      NcFile file(m_paramPath,NcFile::ReadOnly);// open the file and keep the object in file variable

      NcToken name = "dim_mode";
      NcDim* dim = file.get_dim(name);
      m_num_mode = dim->size();
      
      if(m_verbose){std::cout<<"num modes : "<<m_num_mode<<"\n";}
      
      if ( error = m_paramFile.setFile(&file) !=1 )
      {
        std::cerr<< "Error while loading"<<m_paramPath<<" parameter file"<<std::endl;
        return NULL;
      }

      pMesh = create(m_paramFile);// associate pCorrel pointer to the appropriate type following parameter.nc file
      
      if (pMesh == NULL){return NULL;}
      
      return pMesh;
    
    }
    else
    {// m_paramPath = true
    
      std::cout<<" --mesh option : Command line\n";
      std::cout<<" --building!!\n";
      
      return NULL;
      
    }

    
  }

  
  //other methods
  void commandLine(int argc, char *argv[])
  {// get the comand line

    //set parameter file path
    m_paramPath = cimg_option("-p","./parameters.nc","parameter, mesh or true");
    //set verbose mode
    m_verbose = cimg_option("-v",false,0);
    
    if(m_verbose)
    {
      printf("Mode Verbose\n");
      printf("%s : %s\n",class_name.c_str(),m_paramPath);
    }
    
  }

  void stringSplit(std::string manyString, std::vector<std::string> value)
  {
    
    char *cstr,*ptr;
    char *split=(char*)(" ,;");
    
    cstr = new char [manyString.size()+1];
    strcpy(cstr,manyString.c_str());
    ptr=strtok(cstr,split);
    while (ptr!=NULL)
    {
      value.push_back(ptr);
      ptr=strtok(NULL,split);
    }
    delete[] cstr;
    
  }
  
  void info()
  {// print the whole set of available classes within the factory
    
    printf("\nclass %s\n",class_name.c_str());
    for (int i=0; i<m_infos.size(); i++)
    {
      
         printf("\t(%i) mode : %s\n",i,m_infos[i].c_str());
      
    }
    
  }
  
  
  
};

#endif