#include "selecteur_scenario_1thermique.h"

#include <sstream>

#include <indices.h>
#include <cinstance.h>
#include <cinstancereduite.h>
#include <cplan.h>

Selecteur_Scenario_1Thermique::Selecteur_Scenario_1Thermique()
 : Selecteur()
{
}


Selecteur_Scenario_1Thermique::~Selecteur_Scenario_1Thermique()
{
}


void Selecteur_Scenario_1Thermique::setproblemdata()
{
    /* Création des vecteurs de contraintes de clique */
    m_cliques.clear();
    ajoute_contraintes ( C14, C21 );

    /* définition des zones */
    for (scenario_t s=(scenario_t)0; s<S(); ++s)
    {
        m_cplex.addLigne( string("production=demande scenario ")+s, H());
        m_cplex.addColonne( string("production thermique scenario ")+s, H());
    }
    m_cplex.addColonne( "colonnes nucléaires",      CPlan::getNbPlans() );
    m_cplex.addLigne( "unicité colonne nucléaire",  I()                             );
    m_cplex.addLigne( "contraintes de clique",      m_cliques.size()                );

    /// l'objectif est de minimiser une combinaison linéaire des éléments de zobj (en ligne)
    ///     donc de trouver des valeurs pour NUMCOLS variables
    // coeff: coût des colonnes nucléaires (centrale nucléaire)
    // variables associées : binaires (0 ou 1)
    m_cplex.setColonne( "colonnes nucléaires" );
    for ( size_t p=0; p<CPlan::getNbPlans(); p++)
        m_cplex.addToObj( p,
                          0,
                          1,
                          CPlexModel::Binary,
                          CPlan::getPlan(p)->m_cout.v );

    // coeff: coût de la production thermique unitaire par semaine / nombre de scénarios
    // variables associées : énergie produite, continue (entre pmin et pmax)
    for (scenario_t s=(scenario_t)0; s<S(); ++s)
    {
        m_cplex.setColonne( string("production thermique scenario ")+s );
        for ( week_t h=(week_t) 0; h<H(); ++h)
            m_cplex.addToObj( h,
                                ThMin(s,h).v,
                                    ThMax(s,h).v,
                                        CPlexModel::Continuous,
                                        ThCout(s,h).v / S() );
    }

    /// remplissage de Z (contraintes "horizontales")

    /// satisfaction de la demande
    for (scenario_t s=(scenario_t)0; s<S(); ++s)
    {
        m_cplex.setLigne( string("production=demande scenario ")+s);

        // partie droite : demande, opérateur =
        for (week_t h=(week_t) 0; h<H(); ++h)
            m_cplex.addToRHS( h, CPlexModel::Equals, EDEM(s,h).v );

        // production des centrales nucléaires par semaine
        m_cplex.setColonne( "colonnes nucléaires" );
        for (size_t p=0; p<CPlan::getNbPlans(); p++)
            for (week_t h=(week_t) 0; h<H(); ++h)
                m_cplex.addToZ( h, p, CPlan::getPlan(p)->m_colonne[h].v );

        // production des centrales thermiques par semaine (matrices identité)
        m_cplex.setColonne( string("production thermique scenario ")+s );
        for (week_t h=(week_t)0; h<H(); ++h)
            m_cplex.addToZ( h, h, 1 );
    }
    /// sélection de la colonne nucléaire
    m_cplex.setLigne( "unicité colonne nucléaire" );

    // partie droite : 1, opérateur =
    for (powerplants2_t i=(powerplants2_t) 0; i<I(); ++i)
        m_cplex.addToRHS( i, CPlexModel::Equals, 1 );
    // partie gauche : 1 sur la centrale nucléaire correspondant à chaque colonne
    m_cplex.setColonne( "colonnes nucléaires" );
    for (size_t p=0; p<CPlan::getNbPlans(); p++)
        m_cplex.addToZ( CPlan::getPlan(p)->m_i, p, 1 );

    /// contraintes de cliques
    m_cplex.setLigne( "contraintes de clique" );
    m_cplex.setColonne( "colonnes nucléaires" );

    // partie droite : partie droite de la contrainte de clique
    // partie gauche : partie gauche de la contrainte de clique
    size_t c=0 ;
    for (std::map<std::vector<double>, double>::iterator it=m_cliques.begin();
         it!=m_cliques.end(); c++, it++)
    {
        m_cplex.addToRHS( c, CPlexModel::LowerThan, it->second );
        for (size_t p=0; p<CPlan::getNbPlans(); p++)
            m_cplex.addToZ( c, p, it->first[p] );
    }
}