Root_event.h

Go to the documentation of this file.

#ifndef ROOT_EVENT_H_
#define ROOT_EVENT_H_

#include "semantics/Syntactic_event.h"
#include "semantics/Event_view.h"
#include "semantics/Event_traits.h"
#include <boost/shared_ptr.hpp>
#ifdef USE_SEMANTICS
#include "semantics/Elaboration_tree.h"
#endif


namespace semantics
{

class Root_event : public Syntactic_event
{
public:
    boost::shared_ptr<S1_view> s1_view;
    boost::shared_ptr<S2_view> s2_view;

    /*------------------------------------------------------------
     * CONSTRUCTORS AND DESTRUCTOR
     *------------------------------------------------------------*/
    
#ifdef USE_SEMANTICS
    Root_event(
        const Node_data&                      node_data,
        const Semantic_data_base::Hash_pair&  sem_data,
        int                                   id,
        bool                                  learn = true);
#endif
    Root_event(
        const Node_data&                         node_data,
        int                                      id,
        bool                                     learn = true);

    Root_event(const Data_type& data, bool learn = false)
    : Syntactic_event(data, 0, learn)
    {
        update_event_views();
    }
    
    ~Root_event(){}

    /*------------------------------------------------------------
     * COMPUTATION
     *------------------------------------------------------------*/
    
    double log_probability(const bool& collins = false) const;

    /*------------------------------------------------------------
     * DISPLAY
     *------------------------------------------------------------*/

    void print(std::ostream& os) const;
    
    void print_with_links(std::ostream& os) const;

    void print_view_counts(std::ostream& os) const;

    /*------------------------------------------------------------
     * MANIPULATORS
     *------------------------------------------------------------*/

    void release_view_counts()
    {
        if(learn_ == true)
        {
            s1_view -> unlearn();
            s2_view -> unlearn();
            learn_ = false;
        }
    }

    void reacquire_view_counts()
    {
        if(learn_ == false)
        {
            s1_view -> learn();
            s2_view -> learn();
            learn_ = true;
        }
    }

    void resample_table_assignments()
    {
        s1_view -> resample_table_assignment();
        s2_view -> resample_table_assignment();
    }
    
    Event_ptr get_a_copy(bool learn = false) const
    {
        return Event_ptr(new Root_event(data_, learn));
    }

private:
    const Key_slots::Map& var_map() const
    {
        return S2_traits::variable_map;
    }
        
    void update_event_views()
    {
        s1_view =
            boost::make_shared<S1_view>(data_.begin() + 1, data_.end(), learn_);
        s2_view =
            boost::make_shared<S2_view>(data_.begin(), data_.end(), learn_);
    }

};

};

#endif