File graph_var_view.h

File List > buffer > graph_var_view.h

Go to the documentation of this file

#pragma once
#include <muda/buffer/var_view.h>
#include <muda/compute_graph/compute_graph_var.h>

namespace muda
{
template <typename T>
class ComputeGraphVar<VarView<T>> : public ComputeGraphVarBase
{
  public:
    static_assert(!std::is_const_v<T>, "T must not be const");

    using VarType = VarView<T>;
    using ROView  = CVarView<T>;
    using RWView  = VarView<T>;

  protected:
    friend class ComputeGraph;
    friend class ComputeGraphVarManager;

    using ComputeGraphVarBase::ComputeGraphVarBase;

    ComputeGraphVar(ComputeGraphVarManager* var_manager, std::string_view name, VarId var_id) MUDA_NOEXCEPT
        : ComputeGraphVarBase(var_manager, name, var_id)
    {
    }

    ComputeGraphVar(ComputeGraphVarManager* var_manager,
                    std::string_view        name,
                    VarId                   var_id,
                    const RWView&           init_value) MUDA_NOEXCEPT
        : ComputeGraphVarBase(var_manager, name, var_id, true),
          m_value(init_value)
    {
    }

    virtual ~ComputeGraphVar() = default;

  public:
    ROView ceval() const { return _ceval(m_value); }
    RWView eval() { return _eval(m_value); }
    auto   cviewer() const { return ceval().cviewer(); };
    auto   viewer() { return eval().viewer(); };
    operator ROView() const { return ceval(); }
    operator RWView() { return eval(); }

    void                      update(const RWView& view);
    ComputeGraphVar<VarType>& operator=(const RWView& view);

  private:
    RWView m_value;
};

}  // namespace muda

#include "details/graph_var_view.inl"