/* -*- mode: C++; indent-tabs-mode: nil; -*-

  *

  * This file is a part of LEMON, a generic C++ optimization library.

  *

  * Copyright (C) 2003-2008

  * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport

  * (Egervary Research Group on Combinatorial Optimization, EGRES).

  *

  * Permission to use, modify and distribute this software is granted

  * provided that this copyright notice appears in all copies. For

  * precise terms see the accompanying LICENSE file.

  *

  * This software is provided "AS IS" with no warranty of any kind,

  * express or implied, and with no claim as to its suitability for any

  * purpose.

  *

  */

 #include <lemon/lp_clp.h>

 #include <coin/ClpSimplex.hpp>

 namespace lemon {

   LpClp::LpClp() {

     _prob = new ClpSimplex();

     _init_temporals();

     messageLevel(MESSAGE_NO_OUTPUT);

   }

   LpClp::LpClp(const LpClp& other) {

     _prob = new ClpSimplex(*other._prob);

     rows = other.rows;

     cols = other.cols;

     _init_temporals();

     messageLevel(MESSAGE_NO_OUTPUT);

   }

   LpClp::~LpClp() {

     delete _prob;

     _clear_temporals();

   }

   void LpClp::_init_temporals() {

     _primal_ray = 0;

     _dual_ray = 0;

   }

   void LpClp::_clear_temporals() {

     if (_primal_ray) {

       delete[] _primal_ray;

       _primal_ray = 0;

     }

     if (_dual_ray) {

       delete[] _dual_ray;

       _dual_ray = 0;

     }

   }

   LpClp* LpClp::_newSolver() const {

     LpClp* newlp = new LpClp;

     return newlp;

   }

   LpClp* LpClp::_cloneSolver() const {

     LpClp* copylp = new LpClp(*this);

     return copylp;

   }

   const char* LpClp::_solverName() const { return "LpClp"; }

   int LpClp::_addCol() {

     _prob->addColumn(0, 0, 0, -COIN_DBL_MAX, COIN_DBL_MAX, 0.0);

     return _prob->numberColumns() - 1;

   }

   int LpClp::_addRow() {

     _prob->addRow(0, 0, 0, -COIN_DBL_MAX, COIN_DBL_MAX);

     return _prob->numberRows() - 1;

   }

   void LpClp::_eraseCol(int c) {

     _col_names_ref.erase(_prob->getColumnName(c));

     _prob->deleteColumns(1, &c);

   }

   void LpClp::_eraseRow(int r) {

     _row_names_ref.erase(_prob->getRowName(r));

     _prob->deleteRows(1, &r);

   }

   void LpClp::_eraseColId(int i) {

     cols.eraseIndex(i);

     cols.shiftIndices(i);

   }

   void LpClp::_eraseRowId(int i) {

     rows.eraseIndex(i);

     rows.shiftIndices(i);

   }

   void LpClp::_getColName(int c, std::string& name) const {

     name = _prob->getColumnName(c);

   }

   void LpClp::_setColName(int c, const std::string& name) {

     _prob->setColumnName(c, const_cast<std::string&>(name));

     _col_names_ref[name] = c;

   }

   int LpClp::_colByName(const std::string& name) const {

     std::map<std::string, int>::const_iterator it = _col_names_ref.find(name);

     return it != _col_names_ref.end() ? it->second : -1;

   }

   void LpClp::_getRowName(int r, std::string& name) const {

     name = _prob->getRowName(r);

   }

   void LpClp::_setRowName(int r, const std::string& name) {

     _prob->setRowName(r, const_cast<std::string&>(name));

     _row_names_ref[name] = r;

   }

   int LpClp::_rowByName(const std::string& name) const {

     std::map<std::string, int>::const_iterator it = _row_names_ref.find(name);

     return it != _row_names_ref.end() ? it->second : -1;

   }

   void LpClp::_setRowCoeffs(int ix, ExprIterator b, ExprIterator e) {

     std::map<int, Value> coeffs;

     int n = _prob->clpMatrix()->getNumCols();

     const int* indices = _prob->clpMatrix()->getIndices();

     const double* elements = _prob->clpMatrix()->getElements();

     for (int i = 0; i < n; ++i) {

       CoinBigIndex begin = _prob->clpMatrix()->getVectorStarts()[i];

       CoinBigIndex end = begin + _prob->clpMatrix()->getVectorLengths()[i];

       const int* it = std::lower_bound(indices + begin, indices + end, ix);

       if (it != indices + end && *it == ix && elements[it - indices] != 0.0) {

         coeffs[i] = 0.0;

       }

     }

     for (ExprIterator it = b; it != e; ++it) {

       coeffs[it->first] = it->second;

     }

     for (std::map<int, Value>::iterator it = coeffs.begin();

          it != coeffs.end(); ++it) {

       _prob->modifyCoefficient(ix, it->first, it->second);

     }

   }

   void LpClp::_getRowCoeffs(int ix, InsertIterator b) const {

     int n = _prob->clpMatrix()->getNumCols();

     const int* indices = _prob->clpMatrix()->getIndices();

     const double* elements = _prob->clpMatrix()->getElements();

     for (int i = 0; i < n; ++i) {

       CoinBigIndex begin = _prob->clpMatrix()->getVectorStarts()[i];

       CoinBigIndex end = begin + _prob->clpMatrix()->getVectorLengths()[i];

       const int* it = std::lower_bound(indices + begin, indices + end, ix);

       if (it != indices + end && *it == ix) {

         *b = std::make_pair(i, elements[it - indices]);

       }

     }

   }

   void LpClp::_setColCoeffs(int ix, ExprIterator b, ExprIterator e) {

     std::map<int, Value> coeffs;

     CoinBigIndex begin = _prob->clpMatrix()->getVectorStarts()[ix];

     CoinBigIndex end = begin + _prob->clpMatrix()->getVectorLengths()[ix];

     const int* indices = _prob->clpMatrix()->getIndices();

     const double* elements = _prob->clpMatrix()->getElements();

     for (CoinBigIndex i = begin; i != end; ++i) {

       if (elements[i] != 0.0) {

         coeffs[indices[i]] = 0.0;

       }

     }

     for (ExprIterator it = b; it != e; ++it) {

       coeffs[it->first] = it->second;

     }

     for (std::map<int, Value>::iterator it = coeffs.begin();

          it != coeffs.end(); ++it) {

       _prob->modifyCoefficient(it->first, ix, it->second);

     }

   }

   void LpClp::_getColCoeffs(int ix, InsertIterator b) const {

     CoinBigIndex begin = _prob->clpMatrix()->getVectorStarts()[ix];

     CoinBigIndex end = begin + _prob->clpMatrix()->getVectorLengths()[ix];

     const int* indices = _prob->clpMatrix()->getIndices();

     const double* elements = _prob->clpMatrix()->getElements();

     for (CoinBigIndex i = begin; i != end; ++i) {

       *b = std::make_pair(indices[i], elements[i]);

       ++b;

     }

   }

   void LpClp::_setCoeff(int ix, int jx, Value value) {

     _prob->modifyCoefficient(ix, jx, value);

   }

   LpClp::Value LpClp::_getCoeff(int ix, int jx) const {

     CoinBigIndex begin = _prob->clpMatrix()->getVectorStarts()[ix];

     CoinBigIndex end = begin + _prob->clpMatrix()->getVectorLengths()[ix];

     const int* indices = _prob->clpMatrix()->getIndices();

     const double* elements = _prob->clpMatrix()->getElements();

     const int* it = std::lower_bound(indices + begin, indices + end, jx);

     if (it != indices + end && *it == jx) {

       return elements[it - indices];

     } else {

       return 0.0;

     }

   }

   void LpClp::_setColLowerBound(int i, Value lo) {

     _prob->setColumnLower(i, lo == - INF ? - COIN_DBL_MAX : lo);

   }

   LpClp::Value LpClp::_getColLowerBound(int i) const {

     double val = _prob->getColLower()[i];

     return val == - COIN_DBL_MAX ? - INF : val;

   }

   void LpClp::_setColUpperBound(int i, Value up) {

     _prob->setColumnUpper(i, up == INF ? COIN_DBL_MAX : up);

   }

   LpClp::Value LpClp::_getColUpperBound(int i) const {

     double val = _prob->getColUpper()[i];

     return val == COIN_DBL_MAX ? INF : val;

   }

   void LpClp::_setRowLowerBound(int i, Value lo) {

     _prob->setRowLower(i, lo == - INF ? - COIN_DBL_MAX : lo);

   }

   LpClp::Value LpClp::_getRowLowerBound(int i) const {

     double val = _prob->getRowLower()[i];

     return val == - COIN_DBL_MAX ? - INF : val;

   }

   void LpClp::_setRowUpperBound(int i, Value up) {

     _prob->setRowUpper(i, up == INF ? COIN_DBL_MAX : up);

   }

   LpClp::Value LpClp::_getRowUpperBound(int i) const {

     double val = _prob->getRowUpper()[i];

     return val == COIN_DBL_MAX ? INF : val;

   }

   void LpClp::_setObjCoeffs(ExprIterator b, ExprIterator e) {

     int num = _prob->clpMatrix()->getNumCols();

     for (int i = 0; i < num; ++i) {

       _prob->setObjectiveCoefficient(i, 0.0);

     }

     for (ExprIterator it = b; it != e; ++it) {

       _prob->setObjectiveCoefficient(it->first, it->second);

     }

   }

   void LpClp::_getObjCoeffs(InsertIterator b) const {

     int num = _prob->clpMatrix()->getNumCols();

     for (int i = 0; i < num; ++i) {

       Value coef = _prob->getObjCoefficients()[i];

       if (coef != 0.0) {

         *b = std::make_pair(i, coef);

         ++b;

       }

     }

   }

   void LpClp::_setObjCoeff(int i, Value obj_coef) {

     _prob->setObjectiveCoefficient(i, obj_coef);

   }

   LpClp::Value LpClp::_getObjCoeff(int i) const {

     return _prob->getObjCoefficients()[i];

   }

   LpClp::SolveExitStatus LpClp::_solve() {

     return _prob->primal() >= 0 ? SOLVED : UNSOLVED;

   }

   LpClp::SolveExitStatus LpClp::solvePrimal() {

     return _prob->primal() >= 0 ? SOLVED : UNSOLVED;

   }

   LpClp::SolveExitStatus LpClp::solveDual() {

     return _prob->dual() >= 0 ? SOLVED : UNSOLVED;

   }

   LpClp::SolveExitStatus LpClp::solveBarrier() {

     return _prob->barrier() >= 0 ? SOLVED : UNSOLVED;

   }

   LpClp::Value LpClp::_getPrimal(int i) const {

     return _prob->primalColumnSolution()[i];

   }

   LpClp::Value LpClp::_getPrimalValue() const {

     return _prob->objectiveValue();

   }

   LpClp::Value LpClp::_getDual(int i) const {

     return _prob->dualRowSolution()[i];

   }

   LpClp::Value LpClp::_getPrimalRay(int i) const {

     if (!_primal_ray) {

       _primal_ray = _prob->unboundedRay();

       LEMON_ASSERT(_primal_ray != 0, "Primal ray is not provided");

     }

     return _primal_ray[i];

   }

   LpClp::Value LpClp::_getDualRay(int i) const {

     if (!_dual_ray) {

       _dual_ray = _prob->infeasibilityRay();

       LEMON_ASSERT(_dual_ray != 0, "Dual ray is not provided");

     }

     return _dual_ray[i];

   }

   LpClp::VarStatus LpClp::_getColStatus(int i) const {

     switch (_prob->getColumnStatus(i)) {

     case ClpSimplex::basic:

       return BASIC;

     case ClpSimplex::isFree:

       return FREE;

     case ClpSimplex::atUpperBound:

       return UPPER;

     case ClpSimplex::atLowerBound:

       return LOWER;

     case ClpSimplex::isFixed:

       return FIXED;

     case ClpSimplex::superBasic:

       return FREE;

     default:

       LEMON_ASSERT(false, "Wrong column status");

       return VarStatus();

     }

   }

   LpClp::VarStatus LpClp::_getRowStatus(int i) const {

     switch (_prob->getColumnStatus(i)) {

     case ClpSimplex::basic:

       return BASIC;

     case ClpSimplex::isFree:

       return FREE;

     case ClpSimplex::atUpperBound:

       return UPPER;

     case ClpSimplex::atLowerBound:

       return LOWER;

     case ClpSimplex::isFixed:

       return FIXED;

     case ClpSimplex::superBasic:

       return FREE;

     default:

       LEMON_ASSERT(false, "Wrong row status");

       return VarStatus();

     }

   }

   LpClp::ProblemType LpClp::_getPrimalType() const {

     if (_prob->isProvenOptimal()) {

       return OPTIMAL;

     } else if (_prob->isProvenPrimalInfeasible()) {

       return INFEASIBLE;

     } else if (_prob->isProvenDualInfeasible()) {

       return UNBOUNDED;

     } else {

       return UNDEFINED;

     }

   }

   LpClp::ProblemType LpClp::_getDualType() const {

     if (_prob->isProvenOptimal()) {

       return OPTIMAL;

     } else if (_prob->isProvenDualInfeasible()) {

       return INFEASIBLE;

     } else if (_prob->isProvenPrimalInfeasible()) {

       return INFEASIBLE;

     } else {

       return UNDEFINED;

     }

   }

   void LpClp::_setSense(LpClp::Sense sense) {

     switch (sense) {

     case MIN:

       _prob->setOptimizationDirection(1);

       break;

     case MAX:

       _prob->setOptimizationDirection(-1);

       break;

     }

   }

   LpClp::Sense LpClp::_getSense() const {

     double dir = _prob->optimizationDirection();

     if (dir > 0.0) {

       return MIN;

     } else {

       return MAX;

     }

   }

   void LpClp::_clear() {

     delete _prob;

     _prob = new ClpSimplex();

     rows.clear();

     cols.clear();

     _col_names_ref.clear();

     _clear_temporals();

   }

   void LpClp::messageLevel(MessageLevel m) {

     _prob->setLogLevel(static_cast<int>(m));

   }

 } //END OF NAMESPACE LEMON