SmiSmpsIO.hpp

Go to the documentation of this file.

// Copyright (C) 2003, International Business Machines
// Corporation and others.  All Rights Reserved.

#ifndef SmiSmpsIO_HPP
#define SmiSmpsIO_HPP

#if defined(_MSC_VER)
// Turn off compiler warning about long names
#  pragma warning(disable:4786)
#endif

#include "CoinMpsIO.hpp"
#include "CoinFileIO.hpp"
#include "CoinFinite.hpp"
#include "SmiScnModel.hpp"
#include "SmiScnData.hpp"
#include "SmiScenarioTree.hpp"

#include <vector>
#include <map>
#include <string>

using namespace std;


// sections
enum SmiSectionType { SMI_NO_SECTION, SMI_NAME_SECTION,
                                          SMI_ENDATA_SECTION, 
                                        SMI_EOF_SECTION, 
                                          SMI_TIME_SECTION, 
                                        SMI_SCENARIOS_SECTION, 
                                        SMI_INDEPENDENT_SECTION,
                                          SMI_UNKNOWN_SECTION
};

enum SmiSmpsType { SMI_SC_CARD, SMI_BL_CARD, SMI_COLUMN_CARD, SMI_DISCRETE_DIST, SMI_SMPS_COMBINE_ADD, 
        SMI_SMPS_COMBINE_REPLACE, SMI_SMPS_COMBINE_UNKNOWN, SMI_UNKNOWN_MPS_TYPE,
        SMI_TIME_UNORDERED_CORE_TYPE, SMI_TIME_ORDERED_CORE_TYPE
};

class SmiSmpsCardReader:
public CoinMpsCardReader
{
public:
          SmiSectionType nextSmpsField (  );
          SmiSectionType whichSmpsSection(){return smiSection_;}
            
          SmiSmpsType whichSmpsType() {return smiSmpsType_;}
        
          inline const char *periodName (  ) const {return periodName_;}
          inline const char *scenarioNew (  ) const {return columnName_;}
          inline const char *scenarioAnc (  ) const {return rowName_;}

          inline double getProb(){ return prob_;}

          inline void setCoreCombineRule(SmiCoreCombineRule *r){combineRule_=r;combineRuleSet=true;}
          inline SmiCoreCombineRule *getCoreCombineRule() { return combineRule_;}

          SmiSmpsCardReader( CoinFileInput *input, CoinMpsIO * reader ):CoinMpsCardReader (input,reader ),
                combineRuleSet(false),prob_(0.0){}

          ~SmiSmpsCardReader(){}
private:
        char periodName_[COIN_MAX_FIELD_LENGTH];
        float fvalue_;
        SmiSectionType smiSection_;
        SmiSmpsType smiSmpsType_;
        SmiCoreCombineRule *combineRule_;
        bool combineRuleSet;
        double prob_;
};

class SmiSmpsIO: 
public CoinMpsIO
{
public:
        SmiCoreData * readTimeFile(SmiScnModel *smi,const char *c,const char *ext="time");
        int readStochFile(SmiScnModel *smi,SmiCoreData *core, const char *c,const char *ext="stoch");
    
        inline void setCoreCombineRule(SmiCoreCombineRule *r){combineRule_=r; combineRuleSet=true;}
        inline SmiCoreCombineRule *getCoreCombineRule() { return combineRule_;}
        inline int getNumStages(){ return nstag_;}
        inline int *getColumnStages(){ return cstag_;}
        inline int *getRowStages(){ return rstag_;}

    inline void setSolverInfinity(double solverInf) { solverInf_ = solverInf; }
    inline double getSolverInfinity() const { return solverInf_; }

        void writeSmps(const char* filename, bool winFileExtensions = false, bool strictFormat = true);
public:
        SmiSmpsIO():CoinMpsIO(),nstag_(0),cstag_(NULL),rstag_(NULL),solverInf_(COIN_DBL_MAX),iftime(false),ifstoch(false),smpsCardReader_(NULL),combineRule_(NULL),combineRuleSet(false),core(NULL),tree(NULL),periodMap_(),scenarioMap_() {}
    SmiSmpsIO(SmiCoreData * core, SmiScenarioTree<SmiScnNode *> * smiTree):CoinMpsIO(),nstag_(0),cstag_(NULL),rstag_(NULL),solverInf_(COIN_DBL_MAX),iftime(false),ifstoch(false),smpsCardReader_(NULL),combineRule_(NULL),combineRuleSet(false),core(core),tree(smiTree),periodMap_(),scenarioMap_() {}

    ~SmiSmpsIO(){delete [] cstag_;delete[] rstag_;delete smpsCardReader_;}
private:
    void writeCoreFile(const char* filename, const char* extension, const bool strictFormat);
    
    void writeTimeFile(const char* filename, const char* extension, const bool strictFormat);
    
    void writeStochFile(const char* filename, const char* extension, const bool strictFormat);

    void writeScenarioToStochFile(std::ostringstream& stream, SmiTreeNode<SmiScnNode *> * node, int scenario, bool strictFormat);
    
    std::string getModProblemName(); // get the (probably modified) problem name

        int nstag_;
        int *cstag_;
        int *rstag_;
    double solverInf_;

        bool iftime,ifstoch;
        SmiSmpsCardReader *smpsCardReader_;
        SmiCoreCombineRule *combineRule_;
        bool combineRuleSet;
        
        SmiCoreData * core;
        SmiScenarioTree<SmiScnNode *> * tree;

    typedef std::map<string,int> StringIntMap;
    StringIntMap periodMap_;
    StringIntMap scenarioMap_;

};









#endif //#define SmiSmpsIO_HPP