StandaloneL1TopoHistSvc.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
#include "L1TopoCoreSim/StandaloneL1TopoHistSvc.h"
#include "TFile.h"
#include "TH1.h"
#include "TH2.h"
 
#include <iostream>
#include <map>
 
#include "TrigConfBase/TrigConfMessaging.h"
 
 
class StandaloneL1TopoHistSvc::StandaloneL1TopoHistSvcImpl : public TrigConf::TrigConfMessaging
{
public:
 
   StandaloneL1TopoHistSvcImpl() :
      TrigConfMessaging("StandaloneL1TopoHistSvc")
   {}
 
   ~StandaloneL1TopoHistSvcImpl() {
      for( auto h : m_hists1D ) {
         delete h.second;
      }
      for( auto h : m_hists2D ) {
         delete h.second;
      }
   }
 
   void registerHist(TH1 * h) {
      if(h != nullptr) {
         TRG_MSG_DEBUG("Registering histogram " << h->GetName());
         const std::string key = h->GetName();
         if( m_hists1D.find(key)  == end(m_hists1D) ) {
            m_hists1D[key] = h;
         } else {
            TRG_MSG_WARNING("StandaloneL1TopoHistSvc::registerHist: a histogram " << key << " exists already. Will keep the first one and delete the newly requested.");
            delete h;
         }
      }
   }
 
   void registerHist(TH2 * h) {
      if(h != nullptr) {
         TRG_MSG_DEBUG("Registering histogram " << h->GetName());
         const std::string key = h->GetName();
         if( m_hists2D.find(key)  == end(m_hists2D) ) {
            m_hists2D[key] = h;
         } else {
            TRG_MSG_WARNING("StandaloneL1TopoHistSvc::registerHist: a histogram " << key << " exists already. Will keep the first one and delete the newly requested.");
            delete h;
         }
      }
   }
 
   TH1 * findHist(const std::string & histName) {
      auto colPos = histName.find_first_of('/');
      std::string realhistName = histName.substr(colPos+1);
      auto h = m_hists1D.find(realhistName);
      if( h == end(m_hists1D) ) {
         return nullptr;
      } else {
         return h->second;
      }
   }
 
   void fillHist1D(const std::string & histName,double x) {
     auto h = m_hists1D.find(histName);
     if( h == end(m_hists1D) ) {
       TRG_MSG_WARNING("No histogram found for " << histName);
     } else 
       h->second->Fill(x);
   }
 
   void fillHist2D(const std::string & histName,double x,double y) {
     auto h = m_hists2D.find(histName);
     if( h == end(m_hists2D) ) {
       TRG_MSG_WARNING("No histogram found for " << histName);
     } else 
       h->second->Fill(x,y);
   }
 
   void setBaseDir(const std::string & baseDir) {
      m_baseDir = baseDir;
   }
 
   void save() {
 
      std::string filename = "L1Topo.root";
      std::string basepath = "";
 
      std::string opt = "RECREATE";
 
      auto colPos = m_baseDir.find_last_of(':');
      if( colPos != std::string::npos ) {
         filename = m_baseDir.substr(0, colPos);
         basepath = m_baseDir.substr(colPos+1);
      } else {
         basepath = m_baseDir;
      }
 
 
      TFile * f = TFile::Open(filename.c_str(),opt.c_str());
      for( auto h : m_hists1D ) {
 
         std::string fullName(h.second->GetName());
         std::string path(basepath);
 
         auto slashPos = fullName.find_last_of('/');
         if(slashPos != std::string::npos) {
            if(path!="")
               path += "/";
            path += fullName.substr(0,slashPos);
            // set the name
            h.second->SetName( fullName.substr(slashPos+1).c_str() );
         }
 
         const char* dir = path.c_str();
         if( ! f->GetDirectory(dir)) {
            f->mkdir(dir);
         }
         f->cd(dir);
         h.second->Write();
      }
      for( auto h : m_hists2D ) {
 
         std::string fullName(h.second->GetName());
         std::string path(basepath);
 
         auto slashPos = fullName.find_last_of('/');
         if(slashPos != std::string::npos) {
            if(path!="")
               path += "/";
            path += fullName.substr(0,slashPos);
            // set the name
            h.second->SetName( fullName.substr(slashPos+1).c_str() );
         }
 
         const char* dir = path.c_str();
         if( ! f->GetDirectory(dir)) {
            f->mkdir(dir);
         }
         f->cd(dir);
         h.second->Write();
      }
      f->Write();
      f->Close();
      TRG_MSG_INFO("Wrote " << m_hists1D.size()+m_hists2D.size() << " l1topo algorithm histograms to file " << filename);
   }
 
private:
 
   std::map<std::string,TH1*>   m_hists1D;
   std::map<std::string,TH2*>   m_hists2D;
   std::string             m_baseDir {""};
 
};
 
 
 
 
// the interface class
 
StandaloneL1TopoHistSvc::StandaloneL1TopoHistSvc() : 
   m_impl(new StandaloneL1TopoHistSvc::StandaloneL1TopoHistSvcImpl())
{}
 
StandaloneL1TopoHistSvc::~StandaloneL1TopoHistSvc()
{}
 
// forwarding the calls from the interface class
 
void
StandaloneL1TopoHistSvc::registerHist(TH1 * h) {
   //cout << "Called StandaloneL1TopoHistSvc::registerHist(" << h->GetName() << ")" << endl;
   m_impl->registerHist(h);
}
 
void
StandaloneL1TopoHistSvc::registerHist(TH2 * h) {
   //cout << "Called StandaloneL1TopoHistSvc::registerHist(" << h->GetName() << ")" << endl;
   m_impl->registerHist(h);
}
 
TH1 * 
StandaloneL1TopoHistSvc::findHist(const std::string & histName) {
   //cout << "Called StandaloneL1TopoHistSvc::findHist(" << histName << ")" << endl;
   return m_impl->findHist(histName);
}
 
void StandaloneL1TopoHistSvc::fillHist1D(const std::string & histName, double x) {
  m_impl->fillHist1D(histName,x);
}
 
void StandaloneL1TopoHistSvc::fillHist2D(const std::string & histName, double x, double y) {
  m_impl->fillHist2D(histName,x,y);
}
 
void
StandaloneL1TopoHistSvc::setBaseDir(const std::string & baseDir) {
   m_impl->setBaseDir(baseDir);
}
 
 
void
StandaloneL1TopoHistSvc::save() {
   m_impl->save();
}