codac2_analytic_variables.h

Go to the documentation of this file.

 
#pragma once
 
#include <iostream>
#include "codac2_AnalyticExpr.h"
#include "codac2_VarBase.h"
#include "codac2_component.h"
 
namespace codac2
{
  template<typename T>
  struct AnalyticExprWrapper;
 
  template<typename T>
  class AnalyticVarExpr : public AnalyticExpr<T>, public VarBase
  {
    public:
 
      explicit AnalyticVarExpr(const std::string& name)
        : VarBase(name)
      { }
 
      virtual const ExprID& unique_id() const
      {
        return AnalyticExpr<T>::unique_id();
      }
 
      T fwd_eval(ValuesMap& v, [[maybe_unused]] Index total_input_size, [[maybe_unused]] bool natural_eval) const
      {
        return AnalyticExpr<T>::value(v);
      }
      
      void bwd_eval([[maybe_unused]] ValuesMap& v) const
      { }
 
      void replace_arg([[maybe_unused]] const ExprID& old_arg_id, [[maybe_unused]] const std::shared_ptr<ExprBase>& new_expr)
      { }
 
      virtual bool belongs_to_args_list(const FunctionArgsList& args) const
      {
        for(const auto& xi : args)
          if(xi->unique_id() == this->unique_id())
            return true;
        return false;
      }
 
      virtual std::string str(bool in_parentheses = false) const
      {
        return in_parentheses ? "(" + _name + ")" : _name;
      }
 
      virtual bool is_str_leaf() const
      {
        return true;
      }
  };
 
  class ScalarVar : public AnalyticVarExpr<ScalarType>
  {
    public:
 
      explicit ScalarVar(const std::string& name = "?");
 
      std::shared_ptr<VarBase> arg_copy() const;
      std::shared_ptr<ExprBase> copy() const;
      Index size() const;
      std::pair<Index,Index> output_shape() const;
 
      AnalyticExprWrapper<ScalarType> operator-() const;
  };
 
  class VectorVar : public AnalyticVarExpr<VectorType>
  {
    public:
 
      explicit VectorVar(Index n, const std::string& name = "?");
 
      std::shared_ptr<VarBase> arg_copy() const;
      std::shared_ptr<ExprBase> copy() const;
      Index size() const;
      std::pair<Index,Index> output_shape() const;
 
      AnalyticExprWrapper<ScalarType> operator[](Index i) const;
      AnalyticExprWrapper<VectorType> subvector(Index i, Index j) const;
 
    protected:
 
      Index _n;
  };
 
  class MatrixVar : public AnalyticVarExpr<MatrixType>
  {
    public:
 
      explicit MatrixVar(Index r, Index c, const std::string& name = "?");
 
      std::shared_ptr<VarBase> arg_copy() const;
      std::shared_ptr<ExprBase> copy() const;
      Index size() const;
      Index rows() const;
      Index cols() const;
      std::pair<Index,Index> output_shape() const;
 
      AnalyticExprWrapper<ScalarType> operator()(Index i, Index j) const;
      //AnalyticExprWrapper<VectorType> col(Index i) const;
 
    protected:
 
      Index _r, _c;
  };
}