DataFlowVisualization.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/datastore/DependencyMap.h>
 
#include <iosfwd>
 
namespace Belle2 {
  class Module;
  class Path;
 
  class DataFlowVisualization {
  public:
    explicit DataFlowVisualization(const DependencyMap* dependencyMap);
 
    void visualizePath(const std::string& filename, const Path& path);
 
    static void executeModuleAndCreateIOPlot(const std::string& module);
  private:
    void generateModulePlot(std::ofstream& file, const Module& mod, bool steeringFileFlow = false);
 
    static void plotPath(std::ofstream& file, const Path& path, const std::string& pathName = "");
 
    bool checkArrayUnknown(const std::string& name, const DependencyMap::ModuleInfo& info);
 
    const DependencyMap* m_map;
 
    std::set<std::string> m_allInputs; 
    std::set<std::string> m_allOutputs; 
    std::set<std::string> m_unknownArrays; 
    std::string m_fillcolor[DependencyMap::c_NEntryTypes]; 
    std::string m_arrowcolor[DependencyMap::c_NEntryTypes]; 
  };
}