/* -*- C++ -*-
 * lemon/lp_cplex.cc - Part of LEMON, a generic C++ optimization library
 *
 * Copyright (C) 2005 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 <iostream>
#include<lemon/lp_cplex.h>

///\file
///\brief Implementation of the LEMON-CPLEX lp solver interface.
namespace lemon {
  
  LpCplex::LpCplex() : LpSolverBase() {

    //    env = CPXopenCPLEXdevelop(&status);     
    env = CPXopenCPLEX(&status);     
    lp = CPXcreateprob(env, &status, "LP problem");
    //CPXmsg (cpxresults, "Hello-bello\n");
  }
  
  LpCplex::~LpCplex() {
    CPXfreeprob(env,&lp); 
    CPXcloseCPLEX(&env); 
  }
  
  LpSolverBase &LpCplex::_newLp() 
  {
    //The first approach opens a new environment
    LpCplex* newlp=new LpCplex();
    return *newlp;
  }

  LpSolverBase &LpCplex::_copyLp() {
    //The first approach opens a new environment
    LpCplex* newlp=new LpCplex();
    //The routine CPXcloneprob can be used to create a new CPLEX problem 
    //object and copy all the problem data from an existing problem 
    //object to it. Solution and starting information is not copied.
    newlp->lp = CPXcloneprob (env, lp, &status);
    return *newlp;
  }

  int LpCplex::_addCol()
  {
    int i = CPXgetnumcols (env, lp);
    Value lb[1],ub[1];
    lb[0]=-INF;//-CPX_INFBOUND;
    ub[0]=INF;//CPX_INFBOUND;
    status = CPXnewcols (env, lp, 1, NULL, lb, ub, NULL, NULL);
    return i;
  }

  
  int LpCplex::_addRow() 
  {
    //We want a row that is not constrained
    char sense[1];
    sense[0]='L';//<= constraint
    Value rhs[1];
    rhs[0]=INF;
    int i = CPXgetnumrows (env, lp);
    status = CPXnewrows (env, lp, 1, rhs, sense, NULL, NULL);
    return i;
  }


  void LpCplex::_eraseCol(int i) {
    CPXdelcols (env, lp, i, i);
  }
  
  void LpCplex::_eraseRow(int i) {
    CPXdelrows (env, lp, i, i);
  }

  
  ///\warning Data at index 0 is ignored in the arrays.
  void LpCplex::_setRowCoeffs(int i, 
			      int length,
			      int  const * indices, 
			      Value  const * values )
  {
    int rowlist[length+1];
    int* p=rowlist;
    for (int k=1;k<=length;++k){
      rowlist[k]=i;
    }
    status = CPXchgcoeflist(env, lp, 
			    length, 
			    p+1, 
			    const_cast<int * >(indices+1), 
			    const_cast<Value * >(values+1));
  }
  
  void LpCplex::_setColCoeffs(int i, 
			      int length,
			      int  const * indices, 
			      Value  const * values)
  {
    int collist[length+1];
    int* p=collist;
    for (int k=1;k<=length;++k){
      collist[k]=i;
    }
    status = CPXchgcoeflist(env, lp, 
			    length, 
			    const_cast<int * >(indices+1), 
			    p+1, 
			    const_cast<Value * >(values+1));
  }
  
  void LpCplex::_setCoeff(int row, int col, Value value) 
  {
    CPXchgcoef (env, lp, row, col, value);
  }

  void LpCplex::_setColLowerBound(int i, Value value)
  {
    int indices[1];
    indices[0]=i;
    char lu[1];
    lu[0]='L';
    Value bd[1];
    bd[0]=value;
    status = CPXchgbds (env, lp, 1, indices, lu, bd);
 
  }
  
  void LpCplex::_setColUpperBound(int i, Value value)
  {
    int indices[1];
    indices[0]=i;
    char lu[1];
    lu[0]='U';
    Value bd[1];
    bd[0]=value;
    status = CPXchgbds (env, lp, 1, indices, lu, bd);
  }

  //This will be easier to implement
  void LpCplex::_setRowBounds(int i, Value lb, Value ub)
  {
    //Bad parameter
    if (lb==INF || ub==-INF) {
      //FIXME error
    }
    
    int cnt=1;
    int indices[1];
    indices[0]=i;
    char sense[1];

    if (lb==-INF){
      sense[0]='L';
      CPXchgsense (env, lp, cnt, indices, sense);
      CPXchgcoef (env, lp, i, -1, ub);
      
    }
    else{
      if (ub==INF){
	sense[0]='G';
	CPXchgsense (env, lp, cnt, indices, sense);
	CPXchgcoef (env, lp, i, -1, lb);
      }
      else{
	if (lb == ub){
	  sense[0]='E';
	  CPXchgsense (env, lp, cnt, indices, sense);
	  CPXchgcoef (env, lp, i, -1, lb);
	}
	else{
	  sense[0]='R';
	  CPXchgsense (env, lp, cnt, indices, sense);
	  CPXchgcoef (env, lp, i, -1, lb);
	  CPXchgcoef (env, lp, i, -2, ub-lb);	  
	}
      }
    }
  }

//   void LpCplex::_setRowLowerBound(int i, Value value)
//   {
//     //Not implemented, obsolete
//   }
  
//   void LpCplex::_setRowUpperBound(int i, Value value)
//   {
//     //Not implemented, obsolete
// //     //TODO Ezt kell meg megirni
// //     //type of the problem
// //     char sense[1];
// //     status = CPXgetsense (env, lp, sense, i, i);
// //     Value rhs[1];
// //     status = CPXgetrhs (env, lp, rhs, i, i);

// //     switch (sense[0]) {
// //     case 'L'://<= constraint
// //       break;
// //     case 'E'://= constraint
// //       break;
// //     case 'G'://>= constraint
// //       break;
// //     case 'R'://ranged constraint
// //       break;
// //     default: ;
// //       //FIXME error
// //     }

// //     status = CPXchgcoef (env, lp, i, -2, value_rng);
//   }
  
  void LpCplex::_setObjCoeff(int i, Value obj_coef)
  {
    CPXchgcoef (env, lp, -1, i, obj_coef);
  }

  void LpCplex::_clearObj()
  {
    for (int i=0;i< CPXgetnumcols (env, lp);++i){
      CPXchgcoef (env, lp, -1, i, 0);
    }
    
  }

  LpCplex::SolveExitStatus LpCplex::_solve()
  {
    
    status = CPXlpopt (env, lp);
    if (status == 0){
      return SOLVED;
    }
    else{
      return UNSOLVED;
    }
//     int i=  lpx_simplex(lp);
//     switch (i) {
//     case LPX_E_OK: 
//       return SOLVED;
//       break;
//     default:
//       return UNSOLVED;
//     }
  }

  LpCplex::SolutionStatus LpCplex::_getPrimalStatus()
  {
//7.5-os cplex statusai
// #define CPX_OPTIMAL                      1
// #define CPX_INFEASIBLE                   2
// #define CPX_UNBOUNDED                    3
// #define CPX_OBJ_LIM                      4
// #define CPX_IT_LIM_FEAS                  5
// #define CPX_IT_LIM_INFEAS                6
// #define CPX_TIME_LIM_FEAS                7
// #define CPX_TIME_LIM_INFEAS              8
// #define CPX_NUM_BEST_FEAS                9
// #define CPX_NUM_BEST_INFEAS             10
// #define CPX_OPTIMAL_INFEAS              11
// #define CPX_ABORT_FEAS                  12
// #define CPX_ABORT_INFEAS                13
// #define CPX_ABORT_DUAL_INFEAS           14
// #define CPX_ABORT_PRIM_INFEAS           15
// #define CPX_ABORT_PRIM_DUAL_INFEAS      16
// #define CPX_ABORT_PRIM_DUAL_FEAS        17
// #define CPX_ABORT_CROSSOVER             18
// #define CPX_INForUNBD                   19
// #define CPX_PIVOT                       20

//     Ezeket hova tegyem:
// ??case CPX_ABORT_DUAL_INFEAS           
// ??case CPX_ABORT_CROSSOVER             
// ??case CPX_INForUNBD                   
// ??case CPX_PIVOT                       

    int stat = CPXgetstat (env, lp);
    switch (stat) {
    case 0:
      return UNDEFINED; //Undefined
      break;      
    case CPX_OPTIMAL://Optimal
      return OPTIMAL;
      break;
    case CPX_UNBOUNDED://Unbounded
      return INFINITE;
      break;
    case CPX_INFEASIBLE://Infeasible 
    case CPX_IT_LIM_INFEAS:
    case CPX_TIME_LIM_INFEAS:
    case CPX_NUM_BEST_INFEAS:             
    case CPX_OPTIMAL_INFEAS:              
    case CPX_ABORT_INFEAS:                
    case CPX_ABORT_PRIM_INFEAS:           
    case CPX_ABORT_PRIM_DUAL_INFEAS:      
      return INFEASIBLE;
      break;
    case CPX_OBJ_LIM:                    
    case CPX_IT_LIM_FEAS:             
    case CPX_TIME_LIM_FEAS:                
    case CPX_NUM_BEST_FEAS:                
    case CPX_ABORT_FEAS:                  
    case CPX_ABORT_PRIM_DUAL_FEAS:        
      return FEASIBLE;
      break;
    default:
      return UNDEFINED; //Everything else comes here
      //FIXME error
    }


    //Nem tudom, hanyas cplex verzio statusai
// CPX_STAT_ABORT_DUAL_OBJ_LIM
// CPX_STAT_ABORT_IT_LIM
// CPX_STAT_ABORT_OBJ_LIM
// CPX_STAT_ABORT_PRIM_OBJ_LIM
// CPX_STAT_ABORT_TIME_LIM
// CPX_STAT_ABORT_USER
// CPX_STAT_FEASIBLE_RELAXED
// CPX_STAT_INFEASIBLE
// CPX_STAT_INForUNBD
// CPX_STAT_NUM_BEST
// CPX_STAT_OPTIMAL
// CPX_STAT_OPTIMAL_FACE_UNBOUNDED
// CPX_STAT_OPTIMAL_INFEAS
// CPX_STAT_OPTIMAL_RELAXED
// CPX_STAT_UNBOUNDED

//     int stat = CPXgetstat (env, lp);
//     switch (stat) {
//     case CPX_STAT_OPTIMAL://Optimal
//       return OPTIMAL;
//       break;
//     case CPX_STAT_INFEASIBLE://Infeasible 
//       return INFEASIBLE;
//       break;
//     case CPX_STAT_UNBOUNDED://Unbounded
//       return INFINITE;
//       break;
//     case CPX_STAT_NUM_BEST://Feasible
//       return FEASIBLE;
//       break;
//     default:
//       return UNDEFINED; //Everything else comes here
//       //FIXME error
//     }

  }

  LpCplex::Value LpCplex::_getPrimal(int i)
  {
    Value x;
    CPXgetx (env, lp, &x, i, i);
    return x;
  }
  
  LpCplex::Value LpCplex::_getPrimalValue()
  {
    Value objval;
    //method = CPXgetmethod (env, lp);
    status = CPXgetobjval (env, lp, &objval);
    return objval;
  }
  
 


  void LpCplex::_setMax()
  {
    CPXchgobjsen (env, lp, CPX_MAX);
   }
  void LpCplex::_setMin()
  {
    CPXchgobjsen (env, lp, CPX_MIN);
   }
  
} //namespace lemon