data_flow_model.h

#pragma once
 
#include <QString>
#include <QVariant>
#include <optional>
#include <unordered_map>
#include <vector>
 
#include "typing/type_reference.h"
 
namespace maki
{
 
using NodeId = QString;
using DataId = QString;

struct DataPort
{
  NodeId nodeId;
 
  QString eventName;
  QString argumentName;
 
  bool operator==(const DataPort& other) const
  {
    return nodeId == other.nodeId && eventName == other.eventName && argumentName == other.argumentName;
  }
};

enum class DataSourceKind
{
  Capability,
  MissionParameter
};

struct MissionValue
{
  QVariant value;
  bool hasValue = false;
};

struct DataEntry
{
  DataId id;
  QString name;
  koda::types::TypeReference type;
  DataSourceKind sourceKind = DataSourceKind::Capability;
 
  std::optional<DataPort> producer;
  std::vector<DataPort> consumers;
  std::optional<MissionValue> missionValue;
};
 
class DataFlowModel
{
public:
  DataFlowModel() = default;
 
  static DataFlowModel& instance();
  void clear();
 
  DataId createEntry(const QString& name, const koda::types::TypeReference& type);
  DataId createMissionParameter(const QString& name, const koda::types::TypeReference& type);
  bool removeEntry(const DataId& id);
 
  DataEntry* get(const DataId& id);
  const DataEntry* get(const DataId& id) const;
  const std::unordered_map<DataId, DataEntry>& entries() const;
 
  bool setProducer(const DataId& id, const DataPort& producer);
  bool addConsumer(const DataId& id, const DataPort& consumer);
  bool removeProducer(const DataId& id);
  bool removeConsumer(const DataId& id, const DataPort& consumer);
  void removeNode(const NodeId& nodeId);
 
  std::optional<DataId> entryProducedBy(const DataPort& producer) const;
  std::optional<DataId> entryConsumedBy(const DataPort& consumer) const;
  std::vector<DataId> entriesProducedBy(const NodeId& nodeId) const;
  std::vector<DataId> entriesConsumedBy(const NodeId& nodeId) const;
 
  bool setMissionValue(const DataId& id, const QVariant& value);
  bool clearMissionValue(const DataId& id);
  std::vector<DataId> missionParameters() const;
 
  bool hasProducer(const DataPort& port) const;
  bool hasConsumer(const DataPort& port) const;
  bool isUnused(const DataId& id) const;
 
private:
  DataId generateId() const;
 
private:
  std::unordered_map<DataId, DataEntry> mEntries;
};
 
}  // namespace maki