eval_data.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/base_header.hpp>
#include <kernel/space/base.hpp>
#ifndef __CUDA_ARCH__
#include <kernel/util/math.hpp>
#endif
 
namespace FEAT
{
  namespace Space
  {
    template<
      typename EvalTraits_,
      SpaceTags cfg_tags_>
    class BasisData
    {
    public:
      typedef typename EvalTraits_::DataType DataType;
 
      typename EvalTraits_::BasisHessianType hess;
      typename EvalTraits_::BasisGradientType grad;
      typename EvalTraits_::BasisValueType value;
 
      typename EvalTraits_::BasisReferenceHessianType ref_hess;
      typename EvalTraits_::BasisReferenceGradientType ref_grad;
      typename EvalTraits_::BasisReferenceValueType ref_value;
 
      // our space config
      static constexpr SpaceTags config = cfg_tags_;
 
#if defined(DEBUG) && !defined(__CUDA_ARCH__)
      BasisData()
      {
        // format all values to NaN
        const auto qnan = Math::nan<typename EvalTraits_::DataType>();
        hess = qnan;
        grad = qnan;
        value = qnan;
        ref_hess = qnan;
        ref_grad = qnan;
        ref_value = qnan;
      }
#endif // DEBUG
 
      void format()
      {
        hess = DataType(0);
        grad = DataType(0);
        value = DataType(0);
        ref_hess = DataType(0);
        ref_grad = DataType(0);
        ref_value = DataType(0);
      }
    }; // class BasisData<...>
 
    template<
      typename EvalTraits_,
      SpaceTags cfg_tags_>
    class BasisDataReduced
    {
    public:
      // union
      // {
        typename EvalTraits_::BasisGradientType grad;
        typename EvalTraits_::BasisReferenceGradientType ref_grad;
      // };
      union
      {
        typename EvalTraits_::BasisValueType value;
        typename EvalTraits_::BasisReferenceValueType ref_value = value;
      };
 
      // our space config
      static constexpr SpaceTags config = cfg_tags_;
 
      BasisDataReduced() : grad(), value() {}
 
    }; // class BasisData<...>
 
    template<
      typename EvalTraits_,
      SpaceTags cfg_tags_>
    class EvalData
    {
    public:
      static constexpr int max_local_dofs = EvalTraits_::max_local_dofs;
 
      typedef BasisData<EvalTraits_, cfg_tags_> BasisDataType;
 
      BasisDataType phi[max_local_dofs];
 
      // our space config
      static constexpr SpaceTags config = BasisDataType::config;
    }; // class EvalData<...>
 
    template<
      typename EvalTraits_,
      SpaceTags cfg_tags_>
    class EvalDataReduced
    {
    public:
      static constexpr int max_local_dofs = EvalTraits_::max_local_dofs;
 
      typedef BasisDataReduced<EvalTraits_, cfg_tags_> BasisDataType;
 
      typedef EvalTraits_ EvalTraits;
 
      BasisDataType phi[max_local_dofs];
 
      // our space config
      static constexpr SpaceTags config = BasisDataType::config;
    }; // class EvalData<...>
  } // namespace Space
} // namespace FEAT