codac2_analytic_flat_input_layout.h

Go to the documentation of this file.

 
#pragma once
 
#include <typeindex>
#include <unordered_map>
 
#include "codac2_AnalyticExprWrapper.h"
#include "codac2_AnalyticType.h"
#include "codac2_ExprBase.h"
#include "codac2_FunctionArgsList.h"
 
namespace codac2
{
  class ScalarVar;
  class VectorVar;
  class MatrixVar;

  struct FlatInputBinding
  {
    std::type_index type = typeid(ScalarType); 
    Index offset = 0;                          
    Index rows = 1;                            
    Index cols = 1;                            

    bool is_scalar() const;

    bool is_vector() const;

    bool is_matrix() const;
  };

  class FlatInputLayout
  {
    public:

      explicit FlatInputLayout(const FunctionArgsList& args);

      Index size() const;

      bool same_domain_as(const FlatInputLayout& other) const;

      bool same_domain_as(const FunctionArgsList& other_args) const;

      Index flat_index_of(const ScalarVar& x) const;

      Index flat_index_of(const VectorVar& x, Index i) const;

      Index flat_index_of(const MatrixVar& x, Index i, Index j) const;

      bool flat_index_of(const ScalarExpr& x, Index& flat_index) const;

      const FlatInputBinding& binding_of(const ExprID& id) const;

      const FlatInputBinding* find_binding(const ExprID& id) const;
 
    private:
 
      std::unordered_map<Index,FlatInputBinding> _bindings; 
      Index _size = 0;                                      
  };
}