Sacado_UQ_PCE_Traits.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                           Stokhos Package
//
// Copyright 2009 NTESS and the Stokhos contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef SACADO_UQ_PCE_TRAITS_HPP
#define SACADO_UQ_PCE_TRAITS_HPP

#include "Sacado_Traits.hpp"
#include "Sacado_mpl_apply.hpp"

// Forward declarations
namespace Sacado {
  namespace UQ {
    template <typename S> class PCE;
  }
}

namespace Sacado {

  SACADO_AD_PROMOTE_SPEC( UQ, PCE )

  
  template <typename S>
  struct ScalarType< UQ::PCE<S> > {
    typedef typename ScalarType<typename UQ::PCE<S>::value_type>::type type;
  };

  template <typename S>
  struct ValueType< UQ::PCE<S> > {
    typedef typename UQ::PCE<S>::value_type type;
  };

  template <typename S>
  struct IsADType< UQ::PCE<S> > {
    static const bool value = true;
  };

  template <typename S>
  struct IsScalarType< UQ::PCE<S> > {
    static const bool value = false;
  };

  template <typename S>
  struct Value< UQ::PCE<S> > {
    typedef typename ValueType< UQ::PCE<S> >::type value_type;
    KOKKOS_INLINE_FUNCTION
    static const value_type& eval(const UQ::PCE<S>& x) {
      return x.val(); }
  };

  template <typename S>
  struct ScalarValue< UQ::PCE<S> > {
    typedef typename ValueType< UQ::PCE<S> >::type value_type;
    typedef typename ScalarType< UQ::PCE<S> >::type scalar_type;
    KOKKOS_INLINE_FUNCTION
    static const scalar_type& eval(const UQ::PCE<S>& x) {
      return ScalarValue<value_type>::eval(x.val()); }
  };

  template <typename S>
  struct StringName< UQ::PCE<S> > {
    static std::string eval() {
      return std::string("Sacado::UQ::PCE< ") +
        StringName<S>::eval() + " >"; }
  };

  template <typename S>
  struct IsEqual< UQ::PCE<S> > {
    KOKKOS_INLINE_FUNCTION
    static bool eval(const UQ::PCE<S>& x,
                     const UQ::PCE<S>& y) {
      return x.isEqualTo(y);
    }
  };

  template <typename S>
  struct IsStaticallySized< UQ::PCE<S> > {
    static const bool value = S::is_static;
  };

} // namespace Sacado

// Define Teuchos traits classes
// Note Stokhos has required dependency on all Teuchos sub-packages
#include "Teuchos_PromotionTraits.hpp"
#include "Teuchos_ScalarTraits.hpp"
#include "Sacado_UQ_PCE_ScalarTraitsImp.hpp"
#include "Teuchos_SerializationTraits.hpp"

namespace Teuchos {

  template <typename S>
  struct PromotionTraits< Sacado::UQ::PCE<S>,
                          Sacado::UQ::PCE<S> > {
    typedef typename Sacado::Promote< Sacado::UQ::PCE<S>,
                                      Sacado::UQ::PCE<S> >::type
    promote;
  };

  template <typename S, typename R>
  struct PromotionTraits< Sacado::UQ::PCE<S>, R > {
    typedef typename Sacado::Promote< Sacado::UQ::PCE<S>, R >::type
    promote;
  };

  template <typename L, typename S>
  struct PromotionTraits< L, Sacado::UQ::PCE<S> > {
  public:
    typedef typename Sacado::Promote< L, Sacado::UQ::PCE<S> >::type
    promote;
  };

  template <typename S>
  struct ScalarTraits< Sacado::UQ::PCE<S> > :
    public Sacado::UQ::PCEScalarTraitsImp< Sacado::UQ::PCE<S> > {};

  template <typename TypeTo, typename S>
  struct ValueTypeConversionTraits< TypeTo, Sacado::UQ::PCE<S> > :
    public Sacado::UQ::PCEValueTypeConversionTraitsImp< TypeTo,
                                                        Sacado::UQ::PCE<S> > {};

  template <typename Ordinal, typename S>
  struct SerializationTraits<Ordinal, Sacado::UQ::PCE<S> > :
    public Sacado::UQ::PCESerializationTraitsImp< Ordinal,
                                                  Sacado::UQ::PCE<S> >
  {};

  template <typename Ordinal, typename S>
  struct ValueTypeSerializer<Ordinal, Sacado::UQ::PCE<S> > :
    public Sacado::UQ::PCESerializerImp< Ordinal,
                                         Sacado::UQ::PCE<S>,
                                         ValueTypeSerializer<Ordinal,typename S::value_type> >
  {
    typedef Sacado::UQ::PCE<S> PCEType;
    typedef ValueTypeSerializer<Ordinal,typename S::value_type> ValueSerializer;
    typedef Sacado::UQ::PCESerializerImp< Ordinal,PCEType,ValueSerializer> Base;
    typedef typename Base::cijk_type cijk_type;
    ValueTypeSerializer(const cijk_type& cijk,
                        const Teuchos::RCP<const ValueSerializer>& vs) :
      Base(cijk,vs) {}
  };

// Should also do TypeTo, and TypeTo,TypeFrom as UQ::PCE, but the real way
// to fix is to make sure it is never called at all (requires fixing
// magnitudeType)

}

#endif // SACADO_UQ_PCE_TRAITS_HPP