ElementFactory.hpp

Go to the documentation of this file.

// This file is part of GranOO, a workbench for FEM simulation.
//
// Author(s)    : - Jean-luc CHARLES I2M-DuMAS/ENSAM Talence France
//                  <jean-luc.charles@ensam.eu>
//                - Damien ANDRE     SPCTS/ENS Ceramique industrielle, Limoges France
//                  <damien.andre@unilim.fr>
//                - Jeremie GIRARDOT I2M-DuMAS/ENSAM Talence France
//                  <jeremie.girardot@ensam.eu>
//                - Cedric Hubert    LAMIH/UVHC Valenciennes France
//                  <cedric.hubert@univ-valenciennes.fr>
//                - Ivan IORDANOFF   I2M-DuMAS-MPI/ENSAM Talence France
//                  <ivan.iordanoff@ensam.eu>
//
// Copyright (C) 2008-2016 JL. CHARLES, D. ANDRE, I. IORDANOFF, J. GIRARDOT
//
//
//
//
//
// This program 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.
//
// This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 
#ifndef ELEMENTFACTORY_H
#define ELEMENTFACTORY_H
 
#include <stdio.h>
#include <fstream>
#include <map>
#include "GranOO3/FEM/Element.hpp"
 
// element declaration
#define DECLARE_ELEMENT(CLASS) \
public: \
  static const bool IsRegistered; \
  static CLASS *new_element(std::vector<unsigned int> connectivity) { return new CLASS(connectivity); };
 
// element registration
#define RECORD_ELEMENT(CLASS) \
  const bool CLASS::IsRegistered = ElementFactory::get().record<CLASS>();
 
namespace GranOO3 {
 
  namespace FEM {
 
    class ElementFactory {
 
    public:
      ElementFactory();
 
      ~ElementFactory();
 
      static ElementFactory &get();
 
      template<typename T> bool record() {
 
        // assert the element being registered is not already registered
        assert(_elements.find(T::element_name()) == _elements.end() && _elementsnode_number.find(T::element_name()) == _elementsnode_number.end() && "Cannot register an element class twice");
 
        // store a reference to the static constructor
        // T::new_element is declared through the DECLARE_ELEMENT(CLASS) preprocessor macro above
        _elements[T::element_name()] = &T::new_element;
 
        // store the number of nodes for this element
        _elementsnode_number[T::element_name()] = T::node_number;
 
        // return true, for fake...
        return true;
      }
 
      Element *new_element(std::string elementName, std::ifstream &stream);
 
      Element *new_element(std::string elementName, const std::vector<unsigned int> &connectivity);
 
    private:
      static ElementFactory *_factory;
 
      std::map<std::string, std::function<Element *(const std::vector<unsigned int>)> > _elements;
 
      std::map<std::string, unsigned int> _elementsnode_number;
    };
  }
}
 
#endif // ELEMENTFACTORY_H