coeff.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: MIT
// SPDX-FileCopyrightText: 2023 NeoN authors
#pragma once
 
#include "NeoN/core/primitives/scalar.hpp"
#include "NeoN/core/vector.hpp"
 
namespace NeoN::dsl
{
 
class Coeff
{
 
public:
 
    Coeff();
 
    Coeff(scalar value);
 
    Coeff(scalar coeff, const Vector<scalar>& field);
 
    Coeff(const Vector<scalar>& field);
 
    KOKKOS_INLINE_FUNCTION
    scalar operator[](const localIdx i) const { return (hasView_) ? view_[i] * coeff_ : coeff_; }
 
    bool hasSpan();
 
    View<const scalar> view();
 
    Coeff& operator*=(scalar rhs);
 
 
    Coeff& operator*=(const Coeff& rhs);
 
 
private:
 
    scalar coeff_;
 
    View<const scalar> view_;
 
    bool hasView_;
};
 
 
[[nodiscard]] inline Coeff operator*(const Coeff& lhs, const Coeff& rhs)
{
    Coeff result = lhs;
    result *= rhs;
    return result;
}
 
namespace detail
{
/* @brief function to force evaluation to a field, the field will be resized to hold either a
 * single value or the full field
 *
 * @param field to store the result
 */
void toVector(Coeff& coeff, Vector<scalar>& rhs);
 
} // namespace detail
 
} // namespace NeoN::dsl