product_element_matrix_provider_builder.h

#ifndef PROJECTS_DPG_PRODUCT_ELEMENT_MATRIX_PROVIDER_BUILDER
#define PROJECTS_DPG_PRODUCT_ELEMENT_MATRIX_PROVIDER_BUILDER
 
#include "dpg.h"
#include "loc_comp_dpg.h"
#include "product_element_matrix_provider.h"
 
namespace projects::dpg {
template <typename SCALAR>
class ProductElementMatrixProviderBuilder {
 public:
  ProductElementMatrixProviderBuilder(
      const ProductElementMatrixProviderBuilder&) = delete;
  ProductElementMatrixProviderBuilder(
      ProductElementMatrixProviderBuilder&&) noexcept = delete;
  ProductElementMatrixProviderBuilder& operator=(
      const ProductElementMatrixProviderBuilder&) = delete;
  ProductElementMatrixProviderBuilder& operator=(
      ProductElementMatrixProviderBuilder&&) = delete;
 
  ProductElementMatrixProviderBuilder(
      std::shared_ptr<ProductUniformFESpace<SCALAR>> fe_space_trial,
      std::shared_ptr<ProductUniformFESpace<SCALAR>> fe_space_test)
      : fe_space_trial_(std::move(fe_space_trial)),
        fe_space_test_(std::move(fe_space_test)) {}
 
  template <typename DIFF_COEFF>
  ProductElementMatrixProviderBuilder& AddDiffusionElementMatrixProvider(
      size_type trial_component, size_type test_component, DIFF_COEFF alpha);
 
  template <typename CONVECTION_COEFF_1, typename CONVECTION_COEFF_2>
  ProductElementMatrixProviderBuilder& AddConvectionElementMatrixProvider(
      size_type trial_component, size_type test_component,
      CONVECTION_COEFF_1 beta_1, CONVECTION_COEFF_2 beta_2);
 
  template <typename REACTION_COEFF>
  ProductElementMatrixProviderBuilder& AddReactionElementMatrixProvider(
      size_type trial_component, size_type test_component,
      REACTION_COEFF gamma);
 
  template <typename DIFF_COEFF>
  ProductElementMatrixProviderBuilder& AddFluxElementMatrixProvider(
      size_type trial_component, size_type test_component, DIFF_COEFF alpha);
 
  template <typename COEFF>
  ProductElementMatrixProviderBuilder& AddTraceElementMatrixProvider(
      size_type trial_component, size_type test_component, COEFF beta);
 
  std::shared_ptr<ProductElementMatrixProvider<SCALAR>> Build();
 
  ~ProductElementMatrixProviderBuilder() = default;
 
 private:
  std::shared_ptr<ProductUniformFESpace<SCALAR>> fe_space_trial_;
  std::shared_ptr<ProductUniformFESpace<SCALAR>> fe_space_test_;
  std::vector<std::shared_ptr<SubElementMatrixProvider<SCALAR>>> subproviders_;
};
 
template <typename SCALAR>
template <typename DIFF_COEFF>
ProductElementMatrixProviderBuilder<SCALAR>&
ProductElementMatrixProviderBuilder<SCALAR>::AddDiffusionElementMatrixProvider(
    size_type trial_component, size_type test_component, DIFF_COEFF alpha) {
  subproviders_.push_back(
      std::make_shared<DiffusionElementMatrixProvider<SCALAR, DIFF_COEFF>>(
          fe_space_trial_, fe_space_test_, trial_component, test_component,
          alpha));
 
  return *this;
}
 
template <typename SCALAR>
template <typename CONVECTION_COEFF_1, typename CONVECTION_COEFF_2>
ProductElementMatrixProviderBuilder<SCALAR>&
ProductElementMatrixProviderBuilder<SCALAR>::AddConvectionElementMatrixProvider(
    size_type trial_component, size_type test_component,
    CONVECTION_COEFF_1 beta_1, CONVECTION_COEFF_2 beta_2) {
  subproviders_.push_back(std::make_shared<ConvectionElementMatrixProvider<
                              SCALAR, CONVECTION_COEFF_1, CONVECTION_COEFF_2>>(
      fe_space_trial_, fe_space_test_, trial_component, test_component, beta_1,
      beta_2));
  return *this;
}
 
template <typename SCALAR>
template <typename REACTION_COEFF>
ProductElementMatrixProviderBuilder<SCALAR>&
ProductElementMatrixProviderBuilder<SCALAR>::AddReactionElementMatrixProvider(
    size_type trial_component, size_type test_component, REACTION_COEFF gamma) {
  subproviders_.push_back(
      std::make_shared<ReactionElementMatrixProvider<SCALAR, REACTION_COEFF>>(
          fe_space_trial_, fe_space_test_, trial_component, test_component,
          gamma));
  return *this;
}
 
template <typename SCALAR>
template <typename DIFF_COEFF>
ProductElementMatrixProviderBuilder<SCALAR>&
ProductElementMatrixProviderBuilder<SCALAR>::AddFluxElementMatrixProvider(
    size_type trial_component, size_type test_component, DIFF_COEFF alpha) {
  subproviders_.push_back(
      std::make_shared<FluxElementMatrixProvider<SCALAR, DIFF_COEFF>>(
          fe_space_trial_, fe_space_test_, trial_component, test_component,
          alpha));
  return *this;
}
 
template <typename SCALAR>
template <typename COEFF>
ProductElementMatrixProviderBuilder<SCALAR>&
ProductElementMatrixProviderBuilder<SCALAR>::AddTraceElementMatrixProvider(
    size_type trial_component, size_type test_component, COEFF beta) {
  subproviders_.push_back(
      std::make_shared<TraceElementMatrixProvider<SCALAR, COEFF>>(
          fe_space_trial_, fe_space_test_, trial_component, test_component,
          beta));
  return *this;
}
 
template <typename SCALAR>
std::shared_ptr<ProductElementMatrixProvider<SCALAR>>
ProductElementMatrixProviderBuilder<SCALAR>::Build() {
  auto provider = std::make_shared<ProductElementMatrixProvider<SCALAR>>(
      fe_space_trial_, fe_space_test_, subproviders_);
  // clear supplied information
  subproviders_ = {};
  return provider;
}
 
}  // namespace projects::dpg
 
#endif  // PROJECTS_DPG_PRODUCT_ELEMENT_MATRIX_PROVIDER_BUILDER