#include "selecteur_thermiques.h"

#include <sstream>

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

Selecteur_Thermiques::Selecteur_Thermiques()
 : Selecteur()
{
}


Selecteur_Thermiques::~Selecteur_Thermiques()
{
}


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

    /* définition des zones */
    m_cplex.addZoneVerticale( string("production=demande"), H());
    for (powerplants1_t j=(powerplants1_t)0; j<J(); ++j)
        m_cplex.addZoneHorizontale( string("production thermique centrale ")+j, H());

    m_cplex.addZoneHorizontale( "colonnes nucléaires",      CPlan::getNbPlans() );
    m_cplex.addZoneVerticale( "unicité colonne nucléaire",  I()                             );
    m_cplex.addZoneVerticale( "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.setZoneHorizontale( "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 / nbre de scenarios
    // variables associées : énergie produite, continue (entre pmin et pmax)
    for (powerplants1_t j=(powerplants1_t)0; j<J(); ++j)
    {
        m_cplex.setZoneHorizontale( string("production thermique centrale ")+j );
        for ( week_t h=(week_t) 0; h<H(); ++h)
            m_cplex.addToObj( h,
                                EMIN(j,h).v,
                                    EMAX(j,h).v,
                                        CPlexModel::Continuous,
                                        C(j,h).v );
    }

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

    /// satisfaction de la demande
    m_cplex.setZoneVerticale( string("production=demande"));

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

    // production des centrales nucléaires par semaine
    m_cplex.setZoneHorizontale( "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é)
    for (powerplants1_t j=(powerplants1_t)0; j<J(); ++j)
    {
        m_cplex.setZoneHorizontale( string("production thermique centrale ")+j );
        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.setZoneVerticale( "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.setZoneHorizontale( "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.setZoneVerticale( "contraintes de clique" );
    m_cplex.setZoneHorizontale( "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] );
    }
}