KokkosMaterialPropertyValueDecl.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#pragma once

#include "KokkosArray.h"

#define usingKokkosMaterialPropertyValueBaseMembers(T, dimension)                                  \
  using MaterialPropertyValueBase<T, dimension>::_qp;                                              \
  using MaterialPropertyValueBase<T, dimension>::_data;                                            \
  using MaterialPropertyValueBase<T, dimension>::_value

namespace Moose
{
namespace Kokkos
{

template <typename T, unsigned int dimension>
class MaterialProperty;

class Datum;

template <typename T, unsigned int dimension>
class MaterialPropertyValueBase
{
public:
  KOKKOS_FUNCTION MaterialPropertyValueBase(const MaterialProperty<T, dimension> & property,
                                            const Datum & datum,
                                            const unsigned int qp);

  KOKKOS_FUNCTION dof_id_type n(unsigned int dim) const { return _data->n(dim); }

protected:
  const dof_id_type _qp;
  Array<T, dimension + 1> const * _data;
  const T & _value;
};

template <typename T, unsigned int dimension>
class MaterialPropertyValue;

template <typename T>
class MaterialPropertyValue<T, 0> : public MaterialPropertyValueBase<T, 0>
{
  usingKokkosMaterialPropertyValueBaseMembers(T, 0);

public:
  KOKKOS_FUNCTION
  MaterialPropertyValue(const MaterialProperty<T, 0> & property,
                        const Datum & datum,
                        unsigned int qp);

  KOKKOS_FUNCTION operator const T &() const { return _data ? (*_data)(_qp) : _value; }
  KOKKOS_FUNCTION auto & operator=(const T & value);
  KOKKOS_FUNCTION auto & operator=(const MaterialPropertyValue<T, 0> & value);
};

template <typename T>
class MaterialPropertyValue<T, 1> : public MaterialPropertyValueBase<T, 1>
{
  usingKokkosMaterialPropertyValueBaseMembers(T, 1);

public:
  KOKKOS_FUNCTION
  MaterialPropertyValue(const MaterialProperty<T, 1> & property,
                        const Datum & datum,
                        unsigned int qp);

  KOKKOS_FUNCTION T & operator()(unsigned int i0) { return (*_data)(i0, _qp); }
  KOKKOS_FUNCTION const T & operator()(unsigned int i0) const
  {
    return _data ? (*_data)(i0, _qp) : _value;
  }
};

template <typename T>
class MaterialPropertyValue<T, 2> : public MaterialPropertyValueBase<T, 2>
{
  usingKokkosMaterialPropertyValueBaseMembers(T, 2);

public:
  KOKKOS_FUNCTION
  MaterialPropertyValue(const MaterialProperty<T, 2> & property,
                        const Datum & datum,
                        unsigned int qp);

  KOKKOS_FUNCTION T & operator()(unsigned int i0, unsigned int i1) { return (*_data)(i0, i1, _qp); }
  KOKKOS_FUNCTION const T & operator()(unsigned int i0, unsigned int i1) const
  {
    return _data ? (*_data)(i0, i1, _qp) : _value;
  }
};

template <typename T>
class MaterialPropertyValue<T, 3> : public MaterialPropertyValueBase<T, 3>
{
  usingKokkosMaterialPropertyValueBaseMembers(T, 3);

public:
  KOKKOS_FUNCTION
  MaterialPropertyValue(const MaterialProperty<T, 3> & property,
                        const Datum & datum,
                        unsigned int qp);

  KOKKOS_FUNCTION T & operator()(unsigned int i0, unsigned int i1, unsigned int i2)
  {
    return (*_data)(i0, i1, i2, _qp);
  }
  KOKKOS_FUNCTION const T & operator()(unsigned int i0, unsigned int i1, unsigned int i2) const
  {
    return _data ? (*_data)(i0, i1, i2, _qp) : _value;
  }
};

template <typename T>
class MaterialPropertyValue<T, 4> : public MaterialPropertyValueBase<T, 4>
{
  usingKokkosMaterialPropertyValueBaseMembers(T, 4);

public:
  KOKKOS_FUNCTION
  MaterialPropertyValue(const MaterialProperty<T, 4> & property,
                        const Datum & datum,
                        unsigned int qp);

  KOKKOS_FUNCTION T & operator()(unsigned int i0, unsigned int i1, unsigned int i2, unsigned int i3)
  {
    return (*_data)(i0, i1, i2, i3, _qp);
  }
  KOKKOS_FUNCTION const T &
  operator()(unsigned int i0, unsigned int i1, unsigned int i2, unsigned int i3) const
  {
    return _data ? (*_data)(i0, i1, i2, i3, _qp) : _value;
  }
};

} // namespace Kokkos
} // namespace Moose