scen-tree_struct.hpp

Go to the documentation of this file.

#ifndef SCENTREE_STRUCT_HPP
#define SCENTREE_STRUCT_HPP

#include <iostream>
#include <vector>
#include <cassert>

namespace FlopSmiEx {
using std::vector;
using std::cout;
using std::cerr;
using std::endl;

class ScenTreeStruct {
protected:
        int nmbScens;           
        vector<int> leaves;     

public:
        int nmbStages;          

        ScenTreeStruct(int const nScens, int const nStages = 0)
        : nmbScens(nScens), leaves(nScens, -1), nmbStages(nStages)
        {}
        virtual ~ScenTreeStruct() {}


        virtual int get_parent_node(int n) const = 0;

        int get_nmb_stages() const { return nmbStages; }

        int get_nmb_scens() const { return static_cast<int>(leaves.size()); }

        virtual double get_scen_prob(int const sc) const = 0;

        virtual int const * get_scen_nodes(int const sc) = 0;

        inline int const * get_core_scen() { return get_scen_nodes(0); }


        virtual int const * get_next_scen(int &scen, int &parentScen,
                                          int &branchStage, double &prob) = 0;
};


class BinTreeStruct : public ScenTreeStruct {
protected:
        int * scenNodeNmb; 
        int nextScen;      
        double scenProb;   

        int set_scen_nodes(int const sc);

public:

        BinTreeStruct(int const T);

        ~BinTreeStruct(){
                delete[] scenNodeNmb;
        }

        int get_parent_node(int n) const {
                return (n-1) / 2;    // This gives: get_parent_node(0) = 0
        }

        double get_scen_prob(int const sc) const { return scenProb; }

        int const * get_scen_nodes(int const sc) {
                set_scen_nodes(sc);
                return scenNodeNmb;
        }

        int const * get_next_scen(int &scen, int &parentScen,
                                  int &branchStage, double &prob);
};


class TwoStageTree : public ScenTreeStruct {
protected:
        int nmbScens;         
        vector<int> leaves;   
        vector<double> probs; 
        int scenNodeNmb[2];   
        int nextScen;         

        int set_scen_nodes(int const sc) {
                scenNodeNmb[1] = 1 + sc; // scenNodeNmb[0] == 0, the root
                return 1;
        }

public:
        int nmbStages;      

        TwoStageTree(int const nScens);

         ~TwoStageTree() {}

        int get_parent_node(int n) const { return 0; }

        void set_scen_prob(double * const pr);

        double get_scen_prob(int const sc) const { return probs[sc]; }

        int const * get_scen_nodes(int const sc) {
                set_scen_nodes(sc);
                return scenNodeNmb;
        }


        int const * get_next_scen(int &scen, int &parentScen,
                                  int &branchStage, double &prob);
};


} // namespace FlopSmiEx
#endif