Geant4Output2ROOT.h

Go to the documentation of this file.

//==========================================================================
//  AIDA Detector description implementation 
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : M.Frank
//
//==========================================================================
#ifndef DDG4_GEANT4OUTPUT2ROOT_H
#define DDG4_GEANT4OUTPUT2ROOT_H
 
// Framework include files
#include <DDG4/Geant4OutputAction.h>
 
#include <memory>
 
class TFile;
class TTree;
class TBranch;
 
namespace dd4hep {
 
  class ComponentCast;
 
  namespace sim {
 
 
    class Geant4Output2ROOT: public Geant4OutputAction {
    protected:
      using Branches = std::map<std::string, TBranch*>;
      using Sections = std::map<std::string, TTree*>;
      Sections m_sections;
      Branches m_branches;
      std::unique_ptr<TFile> m_file;
      TTree* m_tree = nullptr;
      std::string m_section;
      std::vector<std::string> m_disabledCollections;
      bool  m_disableParticles = false;
      bool m_handleMCTruth = true;
      bool m_filesByRun = false;
 
    public:
      Geant4Output2ROOT(Geant4Context* context, const std::string& nam);
      ~Geant4Output2ROOT() override;
      TTree* section(const std::string& nam);
      int fill(const std::string& nam, const ComponentCast& type, void* ptr);
 
      virtual void closeOutput();
      void beginRun(const G4Run* run)  override;
      void saveCollection(OutputContext<G4Event>& ctxt, G4VHitsCollection* collection)  override;
      void saveEvent(OutputContext<G4Event>& ctxt)  override;
 
      void commit(OutputContext<G4Event>& ctxt)  override;
    };
 
  }    // End namespace sim
}      // End namespace dd4hep
#endif // DDG4_GEANT4OUTPUT2ROOT_H