discrete_problem_selective_assembler.h

#ifndef __H2D_DISCRETE_PROBLEM_SELECTIVE_ASSEMBLER_H
#define __H2D_DISCRETE_PROBLEM_SELECTIVE_ASSEMBLER_H

#include "hermes_common.h"
#include "forms.h"
#include "weakform/weakform.h"
#include "mesh/traverse.h"
#include "space/space.h"
#include "mixins2d.h"
#include "discrete_problem_helpers.h"

namespace Hermes
{
  namespace Hermes2D
  {
    class PrecalcShapeset;
    template<typename Scalar> class Solver;
    template<typename Scalar>
    class HERMES_API DiscreteProblemSelectiveAssembler :
      public Hermes::Mixins::TimeMeasurable,
      public Hermes::Hermes2D::Mixins::Parallel,
      public Hermes::Mixins::Loggable,
      public Hermes::Hermes2D::Mixins::DiscreteProblemRungeKutta<Scalar>,
      public Hermes::Hermes2D::Mixins::DiscreteProblemWeakForm < Scalar >
    {
    public:
      DiscreteProblemSelectiveAssembler();
      ~DiscreteProblemSelectiveAssembler();

      bool prepare_sparse_structure(SparseMatrix<Scalar>* mat, Vector<Scalar>* rhs, std::vector<SpaceSharedPtr<Scalar> > spaces, Traverse::State**& states, unsigned int& num_states);

      void set_spaces(std::vector<SpaceSharedPtr<Scalar> > spaces);

      void set_weak_formulation(WeakFormSharedPtr<Scalar> wf);

      bool form_to_be_assembled(MatrixForm<Scalar>* form, Traverse::State* current_state);
      bool form_to_be_assembled(MatrixFormVol<Scalar>* form, Traverse::State* current_state);
      bool form_to_be_assembled(MatrixFormSurf<Scalar>* form, Traverse::State* current_state);
      bool form_to_be_assembled(MatrixFormDG<Scalar>* form, Traverse::State* current_state);

      bool form_to_be_assembled(VectorForm<Scalar>* form, Traverse::State* current_state);
      bool form_to_be_assembled(VectorFormVol<Scalar>* form, Traverse::State* current_state);
      bool form_to_be_assembled(VectorFormSurf<Scalar>* form, Traverse::State* current_state);
      bool form_to_be_assembled(VectorFormDG<Scalar>* form, Traverse::State* current_state);

    protected:
      unsigned int spaces_size;

      int* sp_seq;

      bool matrix_structure_reusable;
      SparseMatrix<Scalar>* previous_mat;
      bool vector_structure_reusable;
      Vector<Scalar>* previous_rhs;

      friend class DiscreteProblem < Scalar > ;
      friend class DiscreteProblemIntegrationOrderCalculator < Scalar > ;
      friend class Solver < Scalar > ;
    };
  }
}
#endif