JSONIO.cxx

Go to the documentation of this file.

/*
 * Project: RooFit
 * Authors:
 *   Carsten D. Burgard, DESY/ATLAS, Dec 2021
 *
 * Copyright (c) 2022, CERN
 *
 * Redistribution and use in source and binary forms,
 * with or without modification, are permitted according to the terms
 * listed in LICENSE (http://roofit.sourceforge.net/license.txt)
 */
 
#include <RooFitHS3/JSONIO.h>
 
#include <RooFit/Detail/JSONInterface.h>
 
#include <RooAbsPdf.h>
 
#include <TClass.h>
 
#include <fstream>
#include <iostream>
#include <sstream>
 
// Include raw strings with initial export and import keys in JSON
#include "RooFitHS3_wsexportkeys.cxx"
#include "RooFitHS3_wsfactoryexpressions.cxx"
 
namespace RooFit::JSONIO {
 
void setupExportKeys()
{
   static bool isAlreadySetup = false;
   if (isAlreadySetup) {
      return;
   }
 
   isAlreadySetup = true;
 
   std::stringstream exportkeys;
   exportkeys << RooFitHS3_wsexportkeys;
   loadExportKeys(exportkeys);
}
 
void setupFactoryExpressions()
{
   static bool isAlreadySetup = false;
   if (isAlreadySetup) {
      return;
   }
 
   isAlreadySetup = true;
 
   std::stringstream factoryexpressions;
   factoryexpressions << RooFitHS3_wsfactoryexpressions;
   loadFactoryExpressions(factoryexpressions);
}
 
ImportMap &importers()
{
   static ImportMap _importers;
   return _importers;
}
 
namespace {
 
auto &exportersToAdd()
{
   static std::map<const std::string, std::vector<std::unique_ptr<const Exporter>>> toAdd;
   return toAdd;
}
 
ExportMap &exportersImpl()
{
   static ExportMap _exporters;
   return _exporters;
}
 
} // namespace
 
ExportMap &exporters()
{
   // If there are exporters to be added, do this now.
   for (auto &item : exportersToAdd()) {
      TClass *klass = TClass::GetClass(item.first.c_str());
      auto &exporters = exportersImpl()[klass]; // registered exporters so far
      auto &toAdd = item.second;                // exporters to add
 
      // Find the nullptr separator
      auto nullIt = std::find(toAdd.begin(), toAdd.end(), nullptr);
 
      if (nullIt == toAdd.end()) {
         throw std::runtime_error("toAdd does not contain nullptr separator");
      }
 
      // Move elements after nullptr to the back
      exporters.insert(exporters.end(), std::make_move_iterator(nullIt + 1), std::make_move_iterator(toAdd.end()));
 
      // Move elements before nullptr to the front
      exporters.insert(exporters.begin(), std::make_move_iterator(toAdd.begin()), std::make_move_iterator(nullIt));
 
      toAdd.clear();
   }
   exportersToAdd().clear();
   return exportersImpl();
}
 
ImportExpressionMap &importExpressions()
{
   setupFactoryExpressions();
   static ImportExpressionMap _factoryExpressions;
   return _factoryExpressions;
}
 
ExportKeysMap &exportKeys()
{
   setupExportKeys();
   static ExportKeysMap _exportKeys;
   return _exportKeys;
}
 
bool registerImporter(const std::string &key, std::unique_ptr<const Importer> f, bool topPriority)
{
   auto &vec = importers()[key];
   vec.insert(topPriority ? vec.begin() : vec.end(), std::move(f));
   return true;
}
 
bool registerExporter(const TClass *key, std::unique_ptr<const Exporter> f, bool topPriority)
{
   auto &vec = exporters()[key];
   vec.insert(topPriority ? vec.begin() : vec.end(), std::move(f));
   return true;
}
 
bool registerExporter(const std::string &key, std::unique_ptr<const Exporter> f, bool topPriority)
{
   auto &vec = exportersToAdd()[key];
   // The vector starts out with just a nullptr separator. Elements before it
   // will be added to the top of the exporter queue, and the elements after
   // get appended to the end.
   if (vec.empty()) {
      vec.emplace_back(nullptr);
   }
   vec.insert(topPriority ? vec.begin() : vec.end(), std::move(f));
   return true;
}
 
namespace {
 
template <class Map>
int removeByTypeName(Map &map, const std::string &needle)
{
   int n = 0;
   for (auto &element : map) {
      for (size_t i = element.second.size(); i > 0; --i) {
         auto *imp = element.second[i - 1].get();
         std::string name(typeid(*imp).name());
         if (name.find(needle) != std::string::npos) {
            element.second.erase(element.second.begin() + i - 1);
            ++n;
         }
      }
   }
   return n;
}
 
template <class Map, class KeyPrinter>
void printByTypeName(Map &map, KeyPrinter printKey)
{
   for (const auto &x : map) {
      for (const auto &ePtr : x.second) {
         // Passing *e directory to typeid results in clang warnings.
         auto const &e = *ePtr;
         std::cout << printKey(x.first) << "\t" << typeid(e).name() << std::endl;
      }
   }
}
 
} // namespace
 
int removeImporters(const std::string &needle)
{
   return removeByTypeName(importers(), needle);
}
 
int removeExporters(const std::string &needle)
{
   return removeByTypeName(exporters(), needle);
}
 
void printImporters()
{
   printByTypeName(importers(), [](auto const &key) { return key; });
}
void printExporters()
{
   printByTypeName(exporters(), [](auto const &key) { return key->GetName(); });
}
 
void loadFactoryExpressions(const std::string &fname)
{
   // load a yml file defining the factory expressions
   std::ifstream infile(fname);
   if (!infile.is_open()) {
      std::cerr << "unable to read file '" << fname << "'" << std::endl;
      return;
   }
   loadFactoryExpressions(infile);
}
 
void loadFactoryExpressions(std::istream &is)
{
   auto &factoryExpressions = RooFit::JSONIO::importExpressions();
 
   std::unique_ptr<RooFit::Detail::JSONTree> tree = RooFit::Detail::JSONTree::create(is);
   const RooFit::Detail::JSONNode &n = tree->rootnode();
   for (const auto &cl : n.children()) {
      std::string key = cl.key();
      if (!cl.has_child("class")) {
         std::cerr << "error for entry '" << key << "': 'class' key is required!" << std::endl;
         continue;
      }
      std::string classname(cl["class"].val());
      TClass *c = TClass::GetClass(classname.c_str());
      if (!c) {
         std::cerr << "unable to find class " << classname << ", skipping." << std::endl;
         continue;
      }
      RooFit::JSONIO::ImportExpression ex;
      ex.tclass = c;
      if (!cl.has_child("arguments")) {
         std::cerr << "class " << classname << " seems to have no arguments attached, skipping" << std::endl;
         continue;
      }
      for (const auto &arg : cl["arguments"].children()) {
         ex.arguments.push_back(arg.val());
      }
      factoryExpressions[key] = ex;
   }
}
 
void clearFactoryExpressions()
{
   // clear all factory expressions
   RooFit::JSONIO::importExpressions().clear();
}
 
void printFactoryExpressions()
{
   // print all factory expressions
   for (auto it : RooFit::JSONIO::importExpressions()) {
      std::cout << it.first;
      std::cout << " " << it.second.tclass->GetName();
      for (auto v : it.second.arguments) {
         std::cout << " " << v;
      }
      std::cout << std::endl;
   }
}
 
///////////////////////////////////////////////////////////////////////////////////////////////////////
// RooProxy-based export handling
///////////////////////////////////////////////////////////////////////////////////////////////////////
 
void loadExportKeys(const std::string &fname)
{
   // load a yml file defining the export keys
   std::ifstream infile(fname);
   if (!infile.is_open()) {
      std::cerr << "unable to read file '" << fname << "'" << std::endl;
      return;
   }
   loadExportKeys(infile);
}
 
void loadExportKeys(std::istream &is)
{
   auto &exportKeys = RooFit::JSONIO::exportKeys();
 
   std::unique_ptr<RooFit::Detail::JSONTree> tree = RooFit::Detail::JSONTree::create(is);
   const RooFit::Detail::JSONNode &n = tree->rootnode();
   for (const auto &cl : n.children()) {
      std::string classname = cl.key();
      TClass *c = TClass::GetClass(classname.c_str());
      if (!c) {
         std::cerr << "unable to find class " << classname << ", skipping." << std::endl;
         continue;
      }
      RooFit::JSONIO::ExportKeys ex;
      auto *type = cl.find("type");
      auto *proxies = cl.find("proxies");
      if (!type) {
         std::cerr << "class " << classname << "has not type key set, skipping" << std::endl;
         continue;
      }
      if (!proxies) {
         std::cerr << "class " << classname << "has no proxies identified, skipping" << std::endl;
         continue;
      }
      ex.type = type->val();
      for (const auto &k : proxies->children()) {
         ex.proxies[k.key()] = k.val();
      }
      exportKeys[c] = ex;
   }
}
 
void clearExportKeys()
{
   // clear all export keys
   RooFit::JSONIO::exportKeys().clear();
}
 
void printExportKeys()
{
   // print all export keys
   for (const auto &it : RooFit::JSONIO::exportKeys()) {
      std::cout << it.first->GetName() << ": " << it.second.type;
      for (const auto &kv : it.second.proxies) {
         std::cout << " " << kv.first << "=" << kv.second;
      }
      std::cout << std::endl;
   }
}
 
} // namespace RooFit::JSONIO