Merge pull request #39 from key4hep/anaUpdate

update of analysis suite and implementation of corrections (pre-commit errors) of the -help branch merger

Author: dirkzerwas

k4GeneratorsConfig/CMakeLists.txt

@@ -58,6 +58,7 @@ set(xsection_sources
   ${CMAKE_CURRENT_SOURCE_DIR}/src/xsection.cxx
   ${CMAKE_CURRENT_SOURCE_DIR}/src/xsectionCollection.cxx
   ${CMAKE_CURRENT_SOURCE_DIR}/src/xsection2Root.cxx
+  ${CMAKE_CURRENT_SOURCE_DIR}/src/differential.cxx
 )
 
 add_executable(xsectionSummary ${xsection_sources})

k4GeneratorsConfig/src/differential.cxx

@@ -0,0 +1,88 @@
+#include "differential.h"
+#include <iostream>
+#include <memory>
+
+#include "TFile.h"
+#include "TKey.h"
+#include "TROOT.h"
+
+k4GeneratorsConfig::differential::differential()
+    : m_sqrts(0.), m_generator(""), m_process(""), m_file(""), m_isValid(false) {}
+k4GeneratorsConfig::differential::differential(const differential& theOriginal) {
+  if (this != &theOriginal) {
+    m_sqrts = theOriginal.m_sqrts;
+    m_generator = theOriginal.m_generator;
+    m_process = theOriginal.m_process;
+    m_file = theOriginal.m_file;
+    m_isValid = theOriginal.m_isValid;
+    for (auto histo : theOriginal.m_listOfHists) {
+      m_listOfHists.push_back(new TH1D(*histo));
+    }
+  }
+}
+k4GeneratorsConfig::differential& k4GeneratorsConfig::differential::operator=(const differential& theOriginal) {
+  if (this != &theOriginal) {
+    m_sqrts = theOriginal.m_sqrts;
+    m_generator = theOriginal.m_generator;
+    m_process = theOriginal.m_process;
+    m_file = theOriginal.m_file;
+    m_isValid = theOriginal.m_isValid;
+    for (auto hist : m_listOfHists) {
+      delete hist;
+    }
+    m_listOfHists.clear();
+    for (auto hist : theOriginal.m_listOfHists) {
+      m_listOfHists.push_back(new TH1D(*hist));
+    }
+  }
+
+  return *this;
+}
+k4GeneratorsConfig::differential::~differential() {}
+bool k4GeneratorsConfig::differential::processFile() {
+
+  // open the root file
+  std::unique_ptr<TFile> theFile(TFile::Open(m_file.c_str()));
+  if (!theFile || theFile->IsZombie())
+    return false;
+
+  // retrieve the RunInfo for the weight names, there should only be 1 entry per Run
+  TIter keyList(theFile->GetListOfKeys());
+  TKey* key;
+  while ((key = (TKey*)keyList())) {
+    TClass* cl = gROOT->GetClass(key->GetClassName());
+    if (cl->InheritsFrom("TH1")) {
+      m_listOfHists.push_back(new TH1D(*(TH1D*)key->ReadObj()));
+      // turn ownership over to differential:
+      m_listOfHists[m_listOfHists.size() - 1]->SetDirectory(0);
+    }
+  }
+
+  return true;
+}
+void k4GeneratorsConfig::differential::setSQRTS(double sqrts) { m_sqrts = sqrts; }
+void k4GeneratorsConfig::differential::setGenerator(std::string gen) { m_generator = gen; }
+void k4GeneratorsConfig::differential::setProcess(std::string proc) { m_process = proc; }
+void k4GeneratorsConfig::differential::setFile(std::string file) {
+  m_file = file;
+  m_isValid = processFile();
+}
+double k4GeneratorsConfig::differential::SQRTS() { return m_sqrts; }
+std::string k4GeneratorsConfig::differential::Generator() { return m_generator; }
+std::string k4GeneratorsConfig::differential::Process() { return m_process; }
+std::string k4GeneratorsConfig::differential::File() { return m_file; }
+bool k4GeneratorsConfig::differential::isValid() { return m_isValid; }
+TH1D* k4GeneratorsConfig::differential::TH1DHisto(unsigned int iHisto) {
+  return m_listOfHists.size() > 0 ? m_listOfHists[iHisto] : 0;
+}
+unsigned int k4GeneratorsConfig::differential::NbOf1DHistos() { return m_listOfHists.size(); }
+void k4GeneratorsConfig::differential::Print() {
+  std::cout << std::endl;
+  std::cout << "Differential object summary:" << std::endl;
+  std::cout << "File          : " << m_file << std::endl;
+  std::cout << "Process       : " << m_process << std::endl;
+  std::cout << "SQRTS         : " << m_sqrts << std::endl;
+  std::cout << "Generator     : " << m_generator << std::endl;
+  std::cout << "differential valid: " << m_isValid << std::endl;
+  std::cout << std::endl;
+}

k4GeneratorsConfig/src/differential.h

@@ -0,0 +1,43 @@
+#ifndef K4GENERATORSCONFIG_DIFFERENTIAL_H
+#define K4GENERATORSCONFIG_DIFFERENTIAL_H
+
+#include <string>
+
+#include "TH1D.h"
+
+namespace k4GeneratorsConfig {
+class differential {
+public:
+  differential();
+  differential(const differential&);
+  differential& operator=(const differential&);
+  ~differential();
+
+  bool processFile();
+
+  void setSQRTS(double);
+  void setGenerator(std::string);
+  void setProcess(std::string);
+  void setFile(std::string);
+
+  double SQRTS();
+  std::string Generator();
+  std::string Process();
+  std::string File();
+  bool isValid();
+  TH1D* TH1DHisto(unsigned int);
+  unsigned int NbOf1DHistos();
+
+  void Print();
+
+private:
+  double m_sqrts;
+  std::string m_generator;
+  std::string m_process;
+  std::string m_file;
+  bool m_isValid;
+  std::vector<TH1D*> m_listOfHists;
+};
+} // namespace k4GeneratorsConfig
+
+#endif