#include "test_tools.h"

 #ifdef HAVE_CONFIG_H

 #include <config.h>

 #endif

 #ifdef HAVE_CPLEX

 #include <lemon/mip_cplex.h>

 #endif

 #ifdef HAVE_GLPK

 #include <lemon/mip_glpk.h>

 #endif

 using namespace lemon;

 void solveAndCheck(MipSolverBase& lp, MipSolverBase::SolutionStatus stat, 

 		   double exp_opt) {

   using std::string;

   lp.solve();

   //int decimal,sign;

   std::ostringstream buf;

   buf << "Primalstatus should be: " << int(stat)<<" and it is "<<int(lp.primalStatus());

   //  itoa(stat,buf1, 10);

   check(lp.mipStatus()==stat, buf.str());

   if (stat ==  MipSolverBase::OPTIMAL) {

     std::ostringstream buf;

     buf << "Wrong optimal value: the right optimum is " << exp_opt; 

     check(std::abs(lp.primalValue()-exp_opt) < 1e-3, buf.str());

     //+ecvt(exp_opt,2)

   }

 }

 void aTest(MipSolverBase& mip)

 {

  //The following example is very simple

   typedef MipSolverBase::Row Row;

   typedef MipSolverBase::Col Col;

   Col x1 = mip.addCol();

   Col x2 = mip.addCol();

   //Objective function

   mip.setObj(x1);

   mip.max();

   //Unconstrained optimization

   mip.solve();

   //Check it out!

   //Constraints

   mip.addRow(2*x1+x2 <=2);  

   mip.addRow(x1-2*x2 <=0);  

   //Nonnegativity of the variable x1

   mip.colLowerBound(x1, 0);

   //Maximization of x1

   //over the triangle with vertices (0,0),(4/5,2/5),(0,2)

   double expected_opt=4.0/5.0;

   solveAndCheck(mip, MipSolverBase::OPTIMAL, expected_opt);

   //Restrict x2 to integer

   mip.colType(x2,MipSolverBase::LEMON_INTEGER);  

   expected_opt=1.0/2.0;

   solveAndCheck(mip, MipSolverBase::OPTIMAL, expected_opt);

   //Restrict both to integer

   mip.colType(x1,MipSolverBase::LEMON_INTEGER);  

   expected_opt=0;

   solveAndCheck(mip, MipSolverBase::OPTIMAL, expected_opt);

 }

 int main() 

 {

 #ifdef HAVE_GLPK

   MipGlpk mip1;

   aTest(mip1);

 #endif

 #ifdef HAVE_CPLEX

   MipCplex mip2;

   aTest(mip2);

 #endif

   return 0;

 }