// ******************** FlopCpp **********************************************
// File: MP_constraint.cpp
//****************************************************************************

#include <iostream>
#include <sstream>
using std::cout;
using std::endl;

#include <OsiSolverInterface.hpp>
#include "MP_constraint.hpp"
#include "MP_expression.hpp"
#include "MP_model.hpp"
#include "MP_constant.hpp"
#include "MP_data.hpp"

using namespace flopc;


void MP_constraint::operator=(const Constraint &v) {
    left = v->left;
    right = v->right;
    sense = v->sense;
}

int MP_constraint::row_number() const {
    int i1 = S1.check(I1->evaluate());
    int i2 = S2.check(I2->evaluate());
    int i3 = S3.check(I3->evaluate());
    int i4 = S4.check(I4->evaluate());
    int i5 = S5.check(I5->evaluate());

    if (i1==outOfBound || i2==outOfBound || i3==outOfBound ||
        i4==outOfBound || i5==outOfBound) {
            return outOfBound;
    } else {
        return offset + f(I1->evaluate(),I2->evaluate(),I3->evaluate(),
            I4->evaluate(),I5->evaluate()); 
    }
}

double MP_constraint::price(int i1, int i2, int i3, int i4, int i5) const {
    return  M->Solver->getRowPrice()[offset + f(i1,i2,i3,i4,i5)];
}

MP_constraint::MP_constraint(
                             const MP_set_base &s1, 
                             const MP_set_base &s2, 
                             const MP_set_base &s3,
                             const MP_set_base &s4, 
                             const MP_set_base &s5) :
RowMajor(s1.size(),s2.size(),s3.size(),s4.size(),s5.size()),
M(MP_model::current_model),
offset(-1),
S1(s1),S2(s2),S3(s3),S4(s4),S5(s5),
I1(0),I2(0),I3(0),I4(0),I5(0)
{
    M->add(*this);
}

void MP_constraint::coefficients(vector<MP::Coef>& cfs) {
    MP::GenerateFunctor f(this, cfs);
    // f.setConstraint(this);

    vector<TerminalExpression*> v;
    if (left.isDefined() && right.isDefined()) {
        left->generate((S1(I1)*S2(I2)*S3(I3)*S4(I4)*S5(I5)).such_that(B),v,f,1.0);
        right->generate((S1(I1)*S2(I2)*S3(I3)*S4(I4)*S5(I5)).such_that(B),v,f,-1.0);
    } else {
        LOG(WARNING) << "FlopCpp Warning: Constraint declared but not defined." <<endl;
    }
    //TODO Delete pointer when in vector v TODO FIXME
}

void MP_constraint::insertVariables(set<MP_variable*>& v) {
    if (left.operator->()!=0) {
        left->insertVariables(v);
    }
    if (right.operator->()!=0) {
        right->insertVariables(v);
    }
}

void MP_constraint::insertRandomVariables(std::vector< std::set<RandomVariable*> >& v){
        if (left.operator->()!=0) {
        left->insertRandomVariables(v);
    }
    if (right.operator->()!=0) {
        right->insertRandomVariables(v);
    }
}


void MP_constraint::display(string s) const {
    cout<<s<<endl;
    if (offset >=0) {
        for (int i=offset; i<offset+size(); i++) {
            cout<<i<<"  "<<M->Solver->getRowLower()[i]<<"  "<<M->Solver->getRowActivity()[i]<<"  "<<M->Solver->getRowUpper()[i]<<"  "<<M->Solver->getRowPrice()[i]<<endl;
        }
    } else {
        cout<<"No solution available!"<<endl;
    }
}