SimpleAssembler.hpp

// This file is part of the imaging2 class library.
//
// University of Innsbruck, Infmath Imaging, 2009.
// http://infmath.uibk.ac.at
//
// All rights reserved.


#ifndef FEM_SIMPLEASSEMBLER_H
#define FEM_SIMPLEASSEMBLER_H

#include <fem/Grid.hpp>
#include <fem/FemKernel.hpp>
#include <fem/Assembler.hpp>
#include <fem/equation/SimpleEquationAdaptor.hpp>




namespace imaging
{
  class SimpleAssembler
  {
    Assembler _assembler;
    
  public:

    template<class fem_types, class simple_equation_t>
    void assemble(const simple_equation_t & simple_equation, const Grid<fem_types> & grid,
                  ublas::compressed_matrix<float_t> & stiffness_matrix,
                  ublas::vector<float_t> & force_vector) const;
    
    template<class fem_types, class simple_equation_t>
    void assemble_stiffness_matrix(const simple_equation_t & simple_equation, const Grid<fem_types> & grid,
                  ublas::compressed_matrix<float_t> & stiffness_matrix) const;
                  
    template<class fem_types, class simple_equation_t>
    void assemble_force_vector(const simple_equation_t & simple_equation, const Grid<fem_types> & grid,
                  ublas::vector<float_t> & force_vector) const;

  }
  ;

  template<class fem_types, class simple_equation_t>
  void SimpleAssembler::assemble(const simple_equation_t & simple_equation, const Grid<fem_types> & grid,
                                       ublas::compressed_matrix<float_t> & stiffness_matrix,
                                       ublas::vector<float_t> & force_vector) const
  {
    SimpleEquationAdaptor<simple_equation_t> adaptor(simple_equation);
    _assembler.assemble(adaptor, grid, stiffness_matrix, force_vector);
  }

  template<class fem_types, class simple_equation_t>
  void SimpleAssembler::assemble_stiffness_matrix(const simple_equation_t & simple_equation,
                                      const Grid<fem_types> & grid,
                                      ublas::compressed_matrix<float_t> & stiffness_matrix) const
  {
    SimpleEquationAdaptor<simple_equation_t> adaptor(simple_equation);
    _assembler.assemble_stiffness_matrix(adaptor, grid, stiffness_matrix);
  }

  template<class fem_types, class simple_equation_t>
  void SimpleAssembler::assemble_force_vector(const simple_equation_t & simple_equation,
                                      const Grid<fem_types> & grid,
                                      ublas::vector<float_t> & force_vector) const
  {
    SimpleEquationAdaptor<simple_equation_t> adaptor(simple_equation);
    _assembler.assemble_force_vector(adaptor, grid, force_vector);
  }
}


#endif

---