asm_traits.hpp

// FEAT3: Finite Element Analysis Toolbox, Version 3
// Copyright (C) 2010 by Stefan Turek & the FEAT group
// FEAT3 is released under the GNU General Public License version 3,
// see the file 'copyright.txt' in the top level directory for details.
 
#pragma once
 
// includes, FEAT
#include <kernel/assembly/base.hpp>
#include <kernel/cubature/dynamic_factory.hpp>
 
namespace FEAT
{
  namespace Assembly
  {
    namespace Intern
    {
      template<typename Space_, typename DataType_>
      struct EvalPolicyFetcher
      {
        typedef typename Space_::TrafoType TrafoType;
        typedef typename TrafoType::ShapeType ShapeType;
        typedef typename TrafoType::template Evaluator<ShapeType, DataType_>::Type TrafoEvalType;
        typedef typename TrafoEvalType::EvalPolicy EvalPolicy;
      };
 
      template<typename TrafoEvaluator_>
      class CubatureTraits
      {
      public:
        typedef typename TrafoEvaluator_::EvalPolicy EvalPolicy;
 
        typedef typename EvalPolicy::DataType DataType;
 
        typedef typename EvalPolicy::ShapeType ShapeType;
 
        typedef typename EvalPolicy::DomainPointType DomainPointType;
 
        typedef Cubature::Rule<ShapeType, DataType, DataType, DomainPointType> RuleType;
      };
    }
 
    template<
      typename DataType_,
      typename Space_,
      TrafoTags trafo_config_,
      SpaceTags space_config_>
    class AsmTraits1 :
      public Intern::EvalPolicyFetcher<Space_, DataType_>::EvalPolicy
    {
    public:
      typedef DataType_ DataType;
      typedef Space_ SpaceType;
      typedef SpaceType TestSpaceType;
      typedef SpaceType TrialSpaceType;
      typedef SpaceType MultSpaceType;
 
      typedef typename SpaceType::TrafoType TrafoType;
      typedef typename TrafoType::ShapeType ShapeType;
      typedef typename TrafoType::MeshType MeshType;
 
      typedef typename TrafoType::template Evaluator<ShapeType, DataType>::Type TrafoEvaluator;
 
      typedef typename TrafoEvaluator::CellIterator CellIterator;
 
      typedef typename SpaceType::template Evaluator<TrafoEvaluator>::Type SpaceEvaluator;
      typedef SpaceEvaluator TestEvaluator;
      typedef SpaceEvaluator TrialEvaluator;
      typedef SpaceEvaluator MultEvaluator;
 
      typedef typename SpaceEvaluator::SpaceEvalTraits SpaceEvalTraits;
      typedef SpaceEvalTraits TestEvalTraits;
      typedef SpaceEvalTraits TrialEvalTraits;
      typedef SpaceEvalTraits MultEvalTraits;
 
      // define test- and trial-space configs
      static constexpr SpaceTags space_config = space_config_;
      static constexpr SpaceTags test_config = space_config_;
      static constexpr SpaceTags trial_config = space_config_;
      static constexpr SpaceTags mult_config = space_config_;
 
      // now fetch the trafo config from the space
      typedef typename SpaceEvaluator::template ConfigTraits<space_config> SpaceConfigTraits;
      static constexpr TrafoTags space_trafo_config = SpaceConfigTraits::trafo_config;
      static constexpr TrafoTags test_trafo_config = SpaceConfigTraits::trafo_config;
      static constexpr TrafoTags trial_trafo_config = SpaceConfigTraits::trafo_config;
      static constexpr TrafoTags mult_trafo_config = SpaceConfigTraits::trafo_config;
 
      static constexpr TrafoTags trafo_config = trafo_config_ | space_trafo_config | TrafoTags::jac_det;
 
      typedef typename TrafoEvaluator::template ConfigTraits<trafo_config>::EvalDataType TrafoEvalData;
      typedef TrafoEvalData TrafoData;
 
      typedef typename SpaceEvaluator::template ConfigTraits<space_config>::EvalDataType SpaceEvalData;
      typedef SpaceEvalData TestEvalData;
      typedef SpaceEvalData TrialEvalData;
      typedef SpaceEvalData MultEvalData;
 
      typedef typename SpaceEvalData::BasisDataType SpaceBasisData;
      typedef SpaceBasisData BasisData;
      typedef SpaceBasisData TestBasisData;
      typedef SpaceBasisData TrialBasisData;
      typedef SpaceBasisData MultBasisData;
 
      typedef typename SpaceType::DofMappingType DofMapping;
      typedef DofMapping TestDofMapping;
      typedef DofMapping TrialDofMapping;
      typedef DofMapping MultDofMapping;
 
      static constexpr int domain_dim = TrafoEvaluator::domain_dim;
      static constexpr int image_dim = TrafoEvaluator::image_dim;
 
      static constexpr int max_local_dofs       = SpaceEvaluator::max_local_dofs;
      static constexpr int max_local_test_dofs  = TestEvaluator::max_local_dofs;
      static constexpr int max_local_trial_dofs = TrialEvaluator::max_local_dofs;
      static constexpr int max_local_mult_dofs  = MultEvaluator::max_local_dofs;
 
      template<typename Value_>
      using TLocalVector      = Tiny::Vector<Value_, max_local_dofs>;
      template<typename Value_>
      using TLocalTestVector  = Tiny::Vector<Value_, max_local_test_dofs>;
      template<typename Value_>
      using TLocalTrialVector = Tiny::Vector<Value_, max_local_trial_dofs>;
      template<typename Value_>
      using TLocalMultVector  = Tiny::Vector<Value_, max_local_mult_dofs>;
 
      template<typename Value_>
      using TLocalMatrix = Tiny::Matrix<Value_, max_local_test_dofs, max_local_trial_dofs>;
 
      typedef typename Intern::CubatureTraits<TrafoEvaluator>::RuleType CubatureRuleType;
    }; // class AsmTraits1
 
    template<
      typename DataType_,
      typename TestSpace_,
      typename TrialSpace_,
      TrafoTags trafo_config_,
      SpaceTags test_config_,
      SpaceTags trial_config_>
    class AsmTraits2 :
      public Intern::EvalPolicyFetcher<TestSpace_, DataType_>::EvalPolicy
    {
    public:
      typedef DataType_ DataType;
      typedef TestSpace_ TestSpaceType;
      typedef TrialSpace_ TrialSpaceType;
      typedef TrialSpaceType MultSpaceType;
 
      typedef typename TestSpaceType::TrafoType TrafoType;
      typedef typename TrafoType::ShapeType ShapeType;
      typedef typename TrafoType::MeshType MeshType;
 
      typedef typename TrafoType::template Evaluator<ShapeType, DataType>::Type TrafoEvaluator;
 
      typedef typename TrafoEvaluator::CellIterator CellIterator;
 
      typedef typename TestSpaceType::template Evaluator<TrafoEvaluator>::Type TestEvaluator;
      typedef typename TrialSpaceType::template Evaluator<TrafoEvaluator>::Type TrialEvaluator;
      typedef TrialEvaluator MultEvaluator;
 
      typedef typename TestEvaluator::SpaceEvalTraits TestEvalTraits;
      typedef typename TrialEvaluator::SpaceEvalTraits TrialEvalTraits;
      typedef TrialEvalTraits MultEvalTraits;
 
      // define test- and trial-space configs
      static constexpr SpaceTags test_config = test_config_;
      static constexpr SpaceTags trial_config = trial_config_;
      static constexpr SpaceTags mult_config = trial_config_;
 
      // now fetch the trafo configs from the spaces
      typedef typename TestEvaluator::template ConfigTraits<test_config> TestConfigTraits;
      typedef typename TrialEvaluator::template ConfigTraits<trial_config> TrialConfigTraits;
      static constexpr TrafoTags test_trafo_config = TestConfigTraits::trafo_config;
      static constexpr TrafoTags trial_trafo_config = TrialConfigTraits::trafo_config;
      static constexpr TrafoTags mult_trafo_config = TrialConfigTraits::trafo_config;
 
      static constexpr TrafoTags trafo_config = trafo_config_ | test_trafo_config | trial_trafo_config | TrafoTags::jac_det;
 
      typedef typename TrafoEvaluator::template ConfigTraits<trafo_config>::EvalDataType TrafoEvalData;
      typedef TrafoEvalData TrafoData;
 
      typedef typename TestEvaluator::template ConfigTraits<test_config>::EvalDataType TestEvalData;
      typedef typename TrialEvaluator::template ConfigTraits<trial_config>::EvalDataType TrialEvalData;
      typedef TrialEvalData MultEvalData;
 
      typedef typename TestEvalData::BasisDataType TestBasisData;
      typedef typename TrialEvalData::BasisDataType TrialBasisData;
      typedef TrialBasisData MultBasisData;
 
      typedef typename TestSpaceType::DofMappingType TestDofMapping;
      typedef typename TrialSpaceType::DofMappingType TrialDofMapping;
      typedef TrialDofMapping MultDofMapping;
 
      static constexpr int domain_dim = TrafoEvaluator::domain_dim;
      static constexpr int image_dim = TrafoEvaluator::image_dim;
 
      static constexpr int max_local_test_dofs  = TestEvaluator::max_local_dofs;
      static constexpr int max_local_trial_dofs = TrialEvaluator::max_local_dofs;
      static constexpr int max_local_mult_dofs  = MultEvaluator::max_local_dofs;
 
      template<typename Value_>
      using TLocalTestVector  = Tiny::Vector<Value_, max_local_test_dofs>;
      template<typename Value_>
      using TLocalTrialVector = Tiny::Vector<Value_, max_local_trial_dofs>;
      template<typename Value_>
      using TLocalMultVector  = Tiny::Vector<Value_, max_local_mult_dofs>;
 
      template<typename Value_>
      using TLocalMatrix = Tiny::Matrix<Value_, max_local_test_dofs, max_local_trial_dofs>;
 
      typedef typename Intern::CubatureTraits<TrafoEvaluator>::RuleType CubatureRuleType;
    }; // class AsmTraits2
 
    template<
      typename DataType_,
      typename TestSpace_,
      typename TrialSpace_,
      typename MultSpace_,
      TrafoTags trafo_config_,
      SpaceTags test_config_,
      SpaceTags trial_config_,
      SpaceTags mult_config_>
    class AsmTraits3 :
      public Intern::EvalPolicyFetcher<TestSpace_, DataType_>::EvalPolicy
    {
    public:
      typedef DataType_ DataType;
      typedef TestSpace_ TestSpaceType;
      typedef TrialSpace_ TrialSpaceType;
      typedef MultSpace_ MultSpaceType;
 
      typedef typename TestSpaceType::TrafoType TrafoType;
      typedef typename TrafoType::ShapeType ShapeType;
      typedef typename TrafoType::MeshType MeshType;
 
      typedef typename TrafoType::template Evaluator<ShapeType, DataType>::Type TrafoEvaluator;
 
      typedef typename TrafoEvaluator::CellIterator CellIterator;
 
      typedef typename TestSpaceType::template Evaluator<TrafoEvaluator>::Type TestEvaluator;
      typedef typename TrialSpaceType::template Evaluator<TrafoEvaluator>::Type TrialEvaluator;
      typedef typename MultSpaceType::template Evaluator<TrafoEvaluator>::Type MultEvaluator;
 
      typedef typename TestEvaluator::SpaceEvalTraits TestEvalTraits;
      typedef typename TrialEvaluator::SpaceEvalTraits TrialEvalTraits;
      typedef typename MultEvaluator::SpaceEvalTraits MultEvalTraits;
 
      // define test- and trial-space configs
      static constexpr SpaceTags test_config = test_config_;
      static constexpr SpaceTags trial_config = trial_config_;
      static constexpr SpaceTags mult_config = mult_config_;
 
      // now fetch the trafo configs from the spaces
      typedef typename TestEvaluator::template ConfigTraits<test_config> TestConfigTraits;
      typedef typename TrialEvaluator::template ConfigTraits<trial_config> TrialConfigTraits;
      typedef typename MultEvaluator::template ConfigTraits<mult_config> MultConfigTraits;
      static constexpr TrafoTags test_trafo_config = TestConfigTraits::trafo_config;
      static constexpr TrafoTags trial_trafo_config = TrialConfigTraits::trafo_config;
      static constexpr TrafoTags mult_trafo_config = MultConfigTraits::trafo_config;
 
      static constexpr TrafoTags trafo_config = trafo_config_ | test_trafo_config | trial_trafo_config | mult_trafo_config | TrafoTags::jac_det;
 
      typedef typename TrafoEvaluator::template ConfigTraits<trafo_config>::EvalDataType TrafoEvalData;
      typedef TrafoEvalData TrafoData;
 
      typedef typename TestEvaluator::template ConfigTraits<test_config>::EvalDataType TestEvalData;
      typedef typename TrialEvaluator::template ConfigTraits<trial_config>::EvalDataType TrialEvalData;
      typedef typename MultEvaluator::template ConfigTraits<trial_config>::EvalDataType MultEvalData;
 
      typedef typename TestEvalData::BasisDataType TestBasisData;
      typedef typename TrialEvalData::BasisDataType TrialBasisData;
      typedef typename MultEvalData::BasisDataType MultBasisData;
 
      typedef typename TestSpaceType::DofMappingType TestDofMapping;
      typedef typename TrialSpaceType::DofMappingType TrialDofMapping;
      typedef typename MultSpaceType::DofMappingType MultDofMapping;
 
      static constexpr int domain_dim = TrafoEvaluator::domain_dim;
      static constexpr int image_dim = TrafoEvaluator::image_dim;
 
      static constexpr int max_local_test_dofs  = TestEvaluator::max_local_dofs;
      static constexpr int max_local_trial_dofs = TrialEvaluator::max_local_dofs;
      static constexpr int max_local_mult_dofs  = MultEvaluator::max_local_dofs;
 
      template<typename Value_>
      using TLocalTestVector  = Tiny::Vector<Value_, max_local_test_dofs>;
      template<typename Value_>
      using TLocalTrialVector = Tiny::Vector<Value_, max_local_trial_dofs>;
      template<typename Value_>
      using TLocalMultVector  = Tiny::Vector<Value_, max_local_mult_dofs>;
 
      template<typename Value_>
      using TLocalMatrix = Tiny::Matrix<Value_, max_local_test_dofs, max_local_trial_dofs>;
 
      typedef typename Intern::CubatureTraits<TrafoEvaluator>::RuleType CubatureRuleType;
    }; // class AsmTraits3
  } // namespace Assembly
} // namespace FEAT