DependencyMap.h

#pragma once
 
#include <map>
#include <set>
#include <string>
 
namespace Belle2 {
  class Module;
 
  class DependencyMap {
  public:
    enum EEntryType {
      c_Input, 
      c_OptionalInput, 
      c_Output, 
      c_NEntryTypes 
    };
 
    struct ModuleInfo {
      std::set<std::string> entries[c_NEntryTypes]; 
      std::set<std::string> relations[c_NEntryTypes]; 
      void addEntry(const std::string& name, EEntryType type, bool isRelation);
    };
 
    static std::string getModuleID(const Module& mod);
 
    bool isUsedAs(const std::string& branchName, EEntryType type) const;
 
    void setModule(const Module& mod) { m_currentModule = getModuleID(mod); }
 
    ModuleInfo& getCurrentModuleInfo() { return m_moduleInfo[m_currentModule]; }
 
    const std::map<std::string, ModuleInfo>& getModuleInfoMap() const { return m_moduleInfo; }
 
    void clear() { m_moduleInfo.clear(); m_currentModule.clear(); }
 
  private:
    std::map<std::string, ModuleInfo> m_moduleInfo;
 
    std::string m_currentModule;
  };
}