QueryPipeline.h

Go to the documentation of this file.

/*
 * This file is part of KnowRob, please consult
 * https://github.com/knowrob/knowrob for license details.
 */
 
#ifndef KNOWROB_QUERY_PIPELINE_H
#define KNOWROB_QUERY_PIPELINE_H
 
#include "memory"
#include "vector"
#include "TokenStream.h"
#include "knowrob/formulas/DependencyGraph.h"
#include "knowrob/semweb/GraphPathQuery.h"
#include "ConjunctiveQuery.h"
#include "knowrob/reasoner/Computable.h"
 
namespace knowrob {
    class KnowledgeBase;
 
    class QueryPipeline {
    public:
        QueryPipeline(const std::shared_ptr<KnowledgeBase> &kb, const FormulaPtr &phi, const QueryContextPtr &ctx);
 
        QueryPipeline(const std::shared_ptr<KnowledgeBase> &kb, const ConjunctiveQueryPtr &conjunctiveQuery);
 
        ~QueryPipeline();
 
        void operator>>(const std::shared_ptr<TokenStream> &stage);
 
        void stopBuffering();
 
    protected:
        std::vector<std::shared_ptr<TokenStream>> initialStages_;
        std::shared_ptr<TokenBroadcaster> finalStage_;
        std::shared_ptr<TokenBuffer> bufferStage_;
 
        void addInitialStage(const std::shared_ptr<TokenStream> &stage);
 
        static std::vector<ComputablePtr> createComputationSequence(
                const std::shared_ptr<KnowledgeBase> &kb,
                const std::list<DependencyNodePtr> &dependencyGroup);
 
        static void createComputationPipeline(
                const std::shared_ptr<KnowledgeBase> &kb,
                std::vector<ComputablePtr> &computableLiterals,
                const std::shared_ptr<TokenBroadcaster> &pipelineInput,
                const std::shared_ptr<TokenBroadcaster> &pipelineOutput,
                const QueryContextPtr &ctx);
    };
 
    class AnswerBuffer_WithReference : public TokenBuffer {
    public:
        explicit AnswerBuffer_WithReference(const std::shared_ptr<QueryPipeline> &pipeline)
                : TokenBuffer(), pipeline_(pipeline) {}
 
    protected:
        std::shared_ptr<QueryPipeline> pipeline_;
    };
}
 
 
#endif //KNOWROB_QUERY_PIPELINE_H