HypothesesTree.h

Go to the documentation of this file.

#ifndef HYPOTHESESTREE_H_
#define HYPOTHESESTREE_H_

#include "wire/core/datatypes.h"
#include <list>
#include <map>

namespace mhf {

class Evidence;
class SemanticObject;
class Hypothesis;
class AssignmentMatrix;
class KnowledgeDatabase;
class EvidenceSet;

class HypothesisTree {

public:

    /* CONSTRUCTORS / DESTRUCTORS */

    // Constructor
    HypothesisTree(int num_max_hyps, double max_min_prob_ratio);

    // Destructor
    virtual ~HypothesisTree();


    /* GETTERS */

    const std::list<SemanticObject*>& getMAPObjects() const;

    const std::list<SemanticObject*>& getAllObjects() const;

    const std::list<Hypothesis*>& getHypotheses() const;

    const Hypothesis& getMAPHypothesis() const;

    int getHeight() const;


    /* SETTERS */

    void addEvidence(const EvidenceSet& ev_set);


    /* PRINT METHODS */

    void showStatistics();

protected:

    Hypothesis* root_;

    std::list<Hypothesis*> leafs_;

    Hypothesis* MAP_hypothesis_;

    long n_updates_;

    double t_last_update_;

    int tree_height_;

    unsigned int num_max_hyps_;

    double max_min_prob_ratio_;

    void applyAssignments();

    //void determineMAPHypothesis();

    void expandTree(const EvidenceSet &ev_set);

    // Normalize the probabilities of all hypotheses in the tree
    void normalizeProbabilities();

    // Prune the tree
    void pruneTree(const Time& timestamp);

};

}
#endif /* HYPOTHESESTREE_H_ */