AssignmentTarget.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/api.h>
#include <tvm/defs.h>
 
#include <tvm/constraint/enums.h>
 
#include <Eigen/Core>
 
#include <memory>
 
namespace tvm
{
 
namespace scheme
{
 
namespace internal
{
 
enum class TargetType
{
  Linear,
  Quadratic
};
 
class TVM_DLLAPI AssignmentTarget
{
public:
  AssignmentTarget(RangePtr range, MatrixRef A, constraint::Type ct);
  AssignmentTarget(RangePtr range, MatrixRef A, VectorRef b, constraint::Type ct, constraint::RHS cr);
 
  AssignmentTarget(RangePtr range, MatrixRef A, VectorRef l, VectorRef u, constraint::RHS cr);
 
  AssignmentTarget(RangePtr range, VectorRef l, VectorRef u);
 
  AssignmentTarget(RangePtr range, VectorRef lu, constraint::Type ct);
 
  AssignmentTarget(MatrixRef Q, VectorRef q, constraint::RHS cr);
 
  TargetType targetType() const;
  constraint::Type constraintType() const;
  constraint::RHS constraintRhs() const;
  int size() const;
  Range & range();
  const Range & range() const;
 
  MatrixRef A(int colStart, int colDim) const;
  MatrixRef Q() const;
  VectorRef l() const;
  VectorRef u() const;
  VectorRef b() const;
  VectorRef q() const;
 
  MatrixRef AFirstHalf(int colStart, int colDim) const;
  MatrixRef ASecondHalf(int colStart, int colDim) const;
  VectorRef bFirstHalf() const;
  VectorRef bSecondHalf() const;
  AssignmentTarget & setA(MatrixRef A);
  AssignmentTarget & setQ(MatrixRef Q);
  AssignmentTarget & setl(VectorRef l);
  AssignmentTarget & setu(VectorRef u);
  AssignmentTarget & setb(VectorRef b);
  AssignmentTarget & setq(VectorRef q);
 
  void changeData(MatrixRef AQ, VectorRef bq);
 
  void changeData(MatrixRef A, VectorRef l, VectorRef u);
 
  void changeData(VectorRef l, VectorRef u);
 
private:
  TargetType targetType_;
  constraint::Type cstrType_;
  constraint::RHS constraintRhs_;
  RangePtr range_;
  MatrixRef A_ = Eigen::Map<Eigen::MatrixXd>(nullptr, 0, 0);
  MatrixRef Q_ = Eigen::Map<Eigen::MatrixXd>(nullptr, 0, 0);
  VectorRef l_ = Eigen::Map<Eigen::VectorXd>(nullptr, 0);
  VectorRef u_ = Eigen::Map<Eigen::VectorXd>(nullptr, 0);
  VectorRef b_ = Eigen::Map<Eigen::VectorXd>(nullptr, 0);
  VectorRef q_ = Eigen::Map<Eigen::VectorXd>(nullptr, 0);
};
 
} // namespace internal
 
} // namespace scheme
 
} // namespace tvm