decisiontreemanager.h

Go to the documentation of this file.

#ifndef DECISIONTREEMANAGER_H
#define DECISIONTREEMANAGER_H

#include <commons_global.h>
#include <dtree/variables.h>

#include <string>
#include <vector>
#include <set>

#include <memory>

namespace displace {
namespace commons {
struct Scenario;
}
}

namespace dtree {
class DecisionTree;

class COMMONSSHARED_EXPORT DecisionTreeManager
{
private:
    DecisionTreeManager();

    ~DecisionTreeManager();
public:

    enum TreeType {
        InvalidTreeType = -1,   // this is a special code, do not change this value

        GoFishing, ChooseGround,
        StartFishing, ChangeGround, StopFishing,
        ChoosePort,

        SIZE
    };

    int readFromDirectory(std::string path);
    int readFromScenario (std::string path, displace::commons::Scenario scenario);
    std::shared_ptr<dtree::DecisionTree> tree (TreeType type);
    bool hasTree(TreeType type);
    bool hasTreeVariable (TreeType type, dtree::Variable variable) const;

    static DecisionTreeManager *manager() {
        if (mInstance == 0)
            mInstance = new DecisionTreeManager();
        return mInstance;
    }

    static std::string treeTypeCode(TreeType type);
    static TreeType treeTypeFromCode (std::string code);

protected:
    bool readFile(std::string filename);
    void readScenarioFileIfAvailable(std::string path, std::string scenarioname);

private:
    std::vector<std::shared_ptr<dtree::DecisionTree> > mTrees;
    std::vector<std::set<dtree::Variable> > mVariableDictionary;

    static DecisionTreeManager *mInstance;

    static const char * const mCodes[];
};

} // ns

#endif // DECISIONTREEMANAGER_H