class_info.hpp

Go to the documentation of this file.

//
// DUNE DAQ modification notice:
// This file has been modified from the original ATLAS genconfig source for the DUNE DAQ project.
// Fork baseline commit: genconfig-03-10-00 (2021-04-26).
// Renamed since fork: yes (from src/class_info.h to apps/class_info.hpp).
//
 
#ifndef __OKSDALGEN_CLASS_INFO__
#define __OKSDALGEN_CLASS_INFO__
 
#include "oks/class.hpp"
 
#include <map>
#include <ostream>
#include <set>
#include <string>
 
namespace dunedaq {
namespace oksdalgen {
 
class ClassInfo
{
 
public:
  struct SortByName
  {
    bool operator()(const oks::OksClass* c1, const oks::OksClass* c2) const { return c1->get_name() < c2->get_name(); }
  };
 
  typedef std::map<const oks::OksClass*, ClassInfo, SortByName> Map;
 
  ClassInfo() {};
 
  ClassInfo(const std::string& cpp_ns_name, const std::string& dir_prefix)
    : p_namespace(cpp_ns_name)
    , p_include_prefix(dir_prefix) {};
 
  const std::string& get_namespace() const { return p_namespace; }
  const std::string& get_include_prefix() const { return p_include_prefix; }
 
private:
  std::string p_namespace;
  std::string p_include_prefix;
};
 
struct NameSpaceInfo
{
  std::set<std::string> m_classes;
  std::map<std::string, NameSpaceInfo> m_nested;
 
  bool empty() const { return (m_classes.empty() && m_nested.empty()); }
 
  void add(const std::string& ns_name, const std::string& class_name)
  {
    if (!ns_name.empty()) {
      std::string::size_type idx = ns_name.find_first_of(':');
      NameSpaceInfo& ns = m_nested[ns_name.substr(0, idx)];
      if (idx != std::string::npos)
        ns.add(ns_name.substr(ns_name.find_first_not_of(':', idx)), class_name);
      else
        ns.add("", class_name);
    } else
      m_classes.insert(class_name);
  }
 
  void print(std::ostream& s, int level) const
  {
    std::string dx(level * 2, ' ');
 
    for (const auto& x : m_nested) {
      s << dx << "namespace " << x.first << " {\n";
      x.second.print(s, level + 1);
      s << dx << "}\n";
    }
 
    for (const auto& x : m_classes)
      s << dx << "class " << x << ";\n";
  }
};
 
} // namespace oksdalgen
} // namespace dunedaq
#endif