Hypothesis.h

Go to the documentation of this file.

#ifndef HYPOTHESIS_H_
#define HYPOTHESIS_H_

#include <string>
#include <list>

namespace mhf {

class AssignmentSet;
class SemanticObject;
class Assignment;
class ProbabilityModel;
class AssignmentMatrix;

class Hypothesis {

public:
        /* CONSTRUCTORS / DESTRUCTORS */

    // Constructor
    Hypothesis(const double& global_timestamp, double probability = -1);

    // Destructor
    virtual ~Hypothesis();

    Hypothesis* clone() const;


        /* GETTERS */

    const AssignmentSet* getAssignments() const;

    const Hypothesis* getBestLeaf() const;

    std::list<Hypothesis*>& getChildHypotheses();

    int getHeight() const;

    // returns the number of objects in the hypothesis
    int getNumObjects() const;

    // returns the list of objects contained in this hypothesis
    const std::list<SemanticObject*>& getObjects() const;

    //double getObjectProbability(MHTObject* obj) const;

    const Hypothesis* getParent() const;

    // returns probability of this hypothesis
    double getProbability() const;

    double getTimestamp() const;

    AssignmentMatrix* getAssignmentMatrix() const;

        /* SETTERS */

    void setAssignments(AssignmentSet* assignments);

    void setInactive();

    void setProbability(double prob);

    //void setObjectProbability(MHTObject* obj, double p);

        /* HYPOTHESIS MODIFIERS */

    void addChildHypothesis(Hypothesis* h);

    // Add object to the object list
    void addObject(SemanticObject* obj);

    void clearAssignmentSet();

    void clear();

    void addPotentialAssignment(Assignment* assignment);

    void applyAssignments();

        /* TREE UPDATE METHODS  */

    void findActiveLeafs(std::list<Hypothesis*>& active_leafs);

    double calculateBranchProbabilities();

    int calculateHeigth();

    Hypothesis* determineBestLeaf();


        /* TREE CLEAR / DELETE METHODS */

    void clearInactive();

    void deleteChildren();

    Hypothesis* deleteSinglePaths();

protected:

    double probability_;

    double timestamp_;

    std::list<SemanticObject*> objects_;

    Hypothesis* parent_;

    std::list<Hypothesis*> children_;

    AssignmentSet* assignment_set_;

    AssignmentMatrix* assignment_matrix_;

    Hypothesis* best_leaf_;

    int height_;

    bool is_active_leaf_;

};

}
#endif /* HYPOTHESIS_H_ */