eim_classes.h

Go to the documentation of this file.

#ifndef EIM_CLASSES_H
#define EIM_CLASSES_H

// libMesh includes
#include "libmesh/rb_eim_construction.h"
#include "libmesh/rb_eim_evaluation.h"

// Example includes
#include "assembly.h"

// C++ includes
#include <memory>

using libMesh::RBEIMConstruction;

// A simple subclass of RBEIMEvaluation. Overload
// evaluate_parametrized_function to define the
// function that we "empirically" interpolate.
class SimpleEIMEvaluation : public RBEIMEvaluation
{
public:

  SimpleEIMEvaluation(const libMesh::Parallel::Communicator & comm)
    : RBEIMEvaluation(comm)
  {
    set_parametrized_function(std::make_unique<Gxyz>());
  }

  virtual std::unique_ptr<RBTheta> build_eim_theta(unsigned int index)
  {
    return std::make_unique<ThetaEIM>(*this, index);
  }
};

// A simple subclass of RBEIMConstruction.
class SimpleEIMConstruction : public RBEIMConstruction
{
public:

  SimpleEIMConstruction (EquationSystems & es,
                         const std::string & name_in,
                         const unsigned int number_in)
    : RBEIMConstruction(es, name_in, number_in)
  {
  }

  virtual std::unique_ptr<ElemAssembly> build_eim_assembly(unsigned int index)
  {
    return std::make_unique<AssemblyEIM>(*this, index);
  }

  virtual void init_data()
  {
    this->add_variable ("eim_var", libMesh::FIRST);

    RBEIMConstruction::init_data();
  }

};

#endif