Graph.hpp

#pragma once

#include "Export.hpp"

#include "GraphTypes.hpp"
#include "Events.hpp"

#include "ShowcaseTypes.hpp"

#include "GraphGenerators.hpp"

#include "ValueStream.hpp"
#include "GeneratorValueStream.hpp"
#include "ValueStreamChannels.hpp"

#include <folly/futures/Future.h>
#include <folly/Executor.h>

namespace Showcase
{
  typedef ValueStreamPtr<NodeEvent> NodeEventStreamPtr;
  typedef ValueStreamPtr<PropertyEvent> PropertyEventStreamPtr;

  typedef ShowcaseFactoryPtr<GraphGenerator> GraphGenFactoryPtr;
  typedef ShowcaseFactoryPtr<PropertyGenerator> PropGenFactoryPtr;


  class SHOWCASE_API Graph : public std::enable_shared_from_this<Graph>
  {
  public:
    Graph(std::shared_ptr<folly::Executor> executor);

    ~Graph();

    NodeEventStreamPtr children(const GraphNode& node, std::shared_ptr<folly::Executor> executor = nullptr) const;
    PropertyEventStreamPtr properties(const GraphNode& node, std::shared_ptr<folly::Executor> executor = nullptr) const;
    PropertyEventStreamPtr properties(const GraphEdge& edge, std::shared_ptr<folly::Executor> executor = nullptr) const;

    // ------------ Initializing the graph for queries ---------------

    FactoryKey<GraphGenerator>
    registerFactory(const QString& name, GraphGenFactoryPtr factory);

    FactoryKey<PropertyGenerator>
    registerFactory(const QString& name, PropGenFactoryPtr factory);

    bool unregisterFactory(FactoryKey<GraphGenerator> key);
    bool unregisterFactory(FactoryKey<PropertyGenerator> key);

    std::shared_ptr<folly::Executor> executor();

  private:
    std::shared_ptr<GeneratorValueStream<PropertyGenerator>>
    properties(const PropertyEvent& event) const;

    void initializeEventStream(std::shared_ptr<GeneratorValueStream<GraphGenerator>> stream) const;
    void initializeEventStream(std::shared_ptr<GeneratorValueStream<PropertyGenerator>> stream) const;

    std::shared_ptr<folly::Executor> m_executor;

    FactoryMap<GraphGenerator> m_graphFactories;
    FactoryMap<PropertyGenerator> m_propertyFactories;
  };
  typedef std::shared_ptr<Graph> GraphPtr;

}