#include "cbendercplex.h"

#include "cplexmutable.h"
#include "cinstance.h"
#include "cinstancereduite.h"
#include "cplan.h"
#include "cdualcplexscenario.h"

#include <algorithm>

using namespace std;

CBenderCplex::CBenderCplex(bool internal_silent, bool silent, bool writeToFile)
{
    m_cplex = new CPlexMutable( "Benders", CPlexModel::Minimisation, CPlexModel::Linear,
                                internal_silent, silent, writeToFile, writeToFile );
    
    // dates de références initiales réparties sur la période totale
    for (powerplants2_t i; i<I(); ++i) {
        m_dates_ref.push_back(std::vector<double>(0));
        for (campaign_t k; k<K(); ++k)
            m_dates_ref[i].push_back((week_t)(k*H()/K()));
    }
    m_passe = 1;
    initCplex();
}

void CBenderCplex::initCplex()
{
    m_cplex->addColonne( "phi", S());
    for (powerplants2_t i; i<I(); ++i) {
        m_cplex->addColonne( string("abarre_i_")+i+"_k_", K());
        m_cplex->setColonne( string("abarre_i_")+i+"_k_");
        for (campaign_t k; k<K(); ++k) {
                m_cplex->addToObj(k, std::min(type2(i).TO(k), H()), type2(i).TA(k), CPlexModel::Continuous, 0);
        }
    }
    m_cplex->setColonne("phi");
    for (scenario_t s; s<S(); ++s)
        m_cplex->addToObj(s, 0, INFINI, CPlexModel::Continuous, 1.0/S());
}

void CBenderCplex::ajouteCoupe(scenario_t s, CDualCplexScenario* cplex)
{
    m_cplex->addLigne( string("passe_")+m_passe+"_s"+s, 1);
    m_cplex->setLigne( string("passe_")+m_passe+"_s"+s);
    m_cplex->setColonne("phi");
    m_cplex->addToZ(0,s,1);
    
    double z_moins_xi_atruc = cplex->getObjectiveValue();
    for (powerplants2_t i; i<I(); ++i)
    {
        m_cplex->setColonne(string("abarre_i_")+i+"_k_");
        pair<vector<double>,vector<double>> xis = cplex->getDualsDates(i);
        for (campaign_t k; k<K(); ++k)
        {
            double xi = (xis.second[k] - xis.first[k]);
            z_moins_xi_atruc += - xi * m_dates_ref[i][k];
            m_cplex->addToZ(0, k, -xi);
            cout << "i=" << i << ", k=" << k << ", abarétoile=" << m_dates_ref[i][k] << ", xi+=" << xis.first[k] << ", xi-=" << xis.second[k] << ", z=" << cplex->getObjectiveValue() << ", z_moins_xi_atruc=" << z_moins_xi_atruc << endl;
        }
    }
    m_cplex->addToRHS(0, CPlexModel::GreaterThan, z_moins_xi_atruc);
}

double CBenderCplex::resolution()
{
    ++ m_passe;
    double f = m_cplex->resolution();
    for (powerplants2_t i; i<I(); ++i) {
        vector<double> ai = m_cplex->getResult(string("abarre_i_")+i+"_k_");
        for (campaign_t k; k<K(); ++k)
            m_dates_ref[i][k] = round(ai[k]);
    }
    return f;
}