EnumerationInference.h

Go to the documentation of this file.

#pragma once

#include <queue>
#include <thread>

#include "BasicEnumeration.h"
#include "Control.h"
#include "SampleStreams.h"

#include "ThreadedInferenceInterface.h"

template<typename HYP, typename Grammar_t, typename Enumerator>
class EnumerationInference : public ThreadedInferenceInterface<HYP> {

public:
    
    Grammar_t* grammar;
    typename HYP::data_t* data;
        
    EnumerationInference(Grammar_t* g, typename HYP::data_t* d) : 
        grammar(g), data(d) {
    }
    
    generator<HYP&> run_thread(Control& ctl) override {
    
        Enumerator ge(this->grammar);
        
        ctl.start();
        while(ctl.running()) {
            enumerationidx_t nxt = (enumerationidx_t) this->next_index();
            auto n = ge.toNode(nxt, grammar->start());
            //print(n.string());
            
            MyHypothesis h(n); 
            h.compute_posterior(*data);

            co_yield h;

            ++FleetStatistics::enumeration_steps;
        }
    }
};