MatrixAssignment.h

Go to the documentation of this file.

/* Copyright 2017-2020 CNRS-AIST JRL and CNRS-UM LIRMM */
 
#pragma once
 
#include <tvm/api.h>
#include <tvm/defs.h>
 
#include <tvm/Variable.h> // Range
#include <tvm/scheme/internal/AssignmentTarget.h>
#include <tvm/scheme/internal/CompiledAssignmentWrapper.h>
 
#include <Eigen/Core>
 
namespace tvm::scheme::internal
{
class TVM_DLLAPI MatrixAssignment
{
public:
  using MatrixFunction = MatrixRef (AssignmentTarget::*)(int, int) const;
 
  CompiledAssignmentWrapper<Eigen::MatrixXd> assignment;
  Variable * x;
  Range colRange;
  MatrixFunction getTargetMatrix;
 
  void updateTarget(const AssignmentTarget & target);
  void updateMapping(const VariableVector & newVar, const AssignmentTarget & target, bool updateMatrixTarget);
};
 
class TVM_DLLAPI VectorSubstitutionAssignement
{
public:
  using VectorFunction = VectorRef (AssignmentTarget::*)() const;
 
  VectorSubstitutionAssignement(const CompiledAssignmentWrapper<Eigen::VectorXd> & a, VectorFunction getTarget)
  : assignment(a), getTargetVector(getTarget)
  {}
 
  CompiledAssignmentWrapper<Eigen::VectorXd> assignment; // The underlying assignment
  VectorFunction getTargetVector;                        // The way to retrieve the target vector part
};
 
class TVM_DLLAPI VectorAssignment : public VectorSubstitutionAssignement
{
public:
  using RHSFunction = const Eigen::VectorXd & (constraint::abstract::LinearConstraint::*)() const;
  using VectorSubstitutionAssignement::VectorFunction;
 
  VectorAssignment(const CompiledAssignmentWrapper<Eigen::VectorXd> & a,
                   bool useSource,
                   RHSFunction getSource,
                   VectorFunction getTarget)
  : VectorSubstitutionAssignement(a, getTarget), useSource(useSource), getSourceVector(getSource)
  {}
 
  bool useSource;              // Whether or not this assignment uses a source
  RHSFunction getSourceVector; // The way to retrieve the source vector.
};
} // namespace tvm::scheme::internal