CshapeFunction_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 CSHAPEFUNCTION_FACTORY
#define CSHAPEFUNCTION_FACTORY


//-----------------------------------------------------------------------
#include "CshapeFunction.h"
  #include "CshapeFunction_global.h"
    #include "CshapeFunction_global_rigidBody.h"
      #include "CshapeFunction_global_image.h"
      #include "CshapeFunction_global_blade.h"
      #include "CshapeFunction_global_homogeneous.h"
        #include "CshapeFunction_global_brasilian.h"  
  #include "CshapeFunction_FEM.h"
    #include "CshapeFunction_FEM_Q4.h"
      #include "CshapeFunction_FEM_C8.h"
//-----------------------------------------------------------------------

template<typename T,typename Timg>
class CshapeFunction_factory
{
  
  public:
    
  //attributes
  std::vector<std::string> m_infos;
  std::string class_name, m_variable, m_type;
  CParameterNetCDF m_paramFile;
  
  const char* m_paramPath;
  bool m_verbose;
  
  
  CshapeFunction_factory(std::string name)
  {//constructor
 
    class_name = "CshapeFunction_factory";
    m_variable = name;
    
    m_infos.push_back("rigid_body");
    m_infos.push_back("image");
    m_infos.push_back("blade");
    m_infos.push_back("homogeneous");
    m_infos.push_back("brasilian");
    m_infos.push_back("Q4");
    m_infos.push_back("C8");
    
  }

  CshapeFunction<T,Timg>* create(const int &number)
  {//create from "int"

    CshapeFunction<T,Timg>* pShape;
    
    switch(number)
    {
      case 0: pShape = new CshapeFunction_global_rigidBody<T,Timg>; break;
      case 1: pShape = new CshapeFunction_global_image<T,Timg>; break;
      case 2: pShape = new CshapeFunction_global_blade<T,Timg>; break;
      case 3: pShape = new CshapeFunction_global_homogeneous<T,Timg>;break;
      case 4: pShape = new CshapeFunction_global_brasilian<T,Timg>;break;
      case 5: pShape = new CshapeFunction_FEM_Q4<T,Timg>; break;
      case 6: pShape = new CshapeFunction_FEM_C8<T,Timg>; break;
      default:
      {
        std::cerr<<class_name<<" : "<<number<<" refers to an unknown or incompatible correlation class"<<std::flush;
        return NULL;
      }
    }
    
    if (m_verbose){printf("%s -> %s class loaded\n",class_name.c_str(), m_type.c_str());}
    
    pShape->m_verbose = m_verbose;// string correl-type as member
    pShape->m_paramPath = m_paramPath;// string correl-name as member
    
    pShape->m_type = m_type;// string correl-type as member
    pShape->m_name = m_variable;// string correl-name as member
    
    pShape->init(m_paramFile);//specific init

    
    return pShape;// return pointer to specific correlation method

  }
  
  CshapeFunction<T,Timg>* create(const std::string &type)
  {//create from "string"
  
    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 if(type==m_infos[5]) return create(5);
    else if(type==m_infos[6]) return create(6);
    else
    {
      std::cerr<<class_name<<"::"<<__func__<<": error: type = "<<type<<" is unknown.\n"<<std::flush;
      return NULL;
    }
    
  }
   
  CshapeFunction<T,Timg>* create(CParameterNetCDF &paramFile)
  {//create from "parameter file"
  
    m_paramFile = paramFile;

    int var(0);

    paramFile.loadVar(var,&m_variable);
    m_paramFile.loadAttribute("shape",m_type);
    
    return create(m_type);
    
  }
 
  CshapeFunction<T,Timg>* create(int argc, char *argv[])
  {//create from (argv,argc)

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

    CshapeFunction<T,Timg> *pShape;// 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

      if ( error = m_paramFile.setFile(&file) !=1 )
      {
        std::cerr<< "Error while loading"<<m_paramPath<<" parameter file"<<std::endl;
        return NULL;
      }

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

    
  }
  
  
  
  //other methods
  void commandLine(int argc, char *argv[])
  {// read the command 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