DataFlowVisualization.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2013  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Christian Pulvermacher                                   *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#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]; 
  };
}