Geant4Output2Podio.h

Go to the documentation of this file.

#ifndef DD4HEP_DDG4_GEANT4Output2Podio_H
#define DD4HEP_DDG4_GEANT4Output2Podio_H
 
#include "DDG4/Geant4OutputAction.h"
//#include "podio/GenericParameters.h"
#include "podio/CollectionBase.h"
#include "podio/CollectionIDTable.h"
 
#include "podio/EventStore.h"
#include "podio/ROOTWriter.h"
 
namespace dd4pod {
class Geant4Particle;
class Geant4ParticleCollection;
}
 
namespace dd4hep {
 
 
  class ComponentCast;
 
  namespace sim {
 
    class Geant4Particle;
 
    class Geant4Output2Podio: public Geant4OutputAction {
    protected:
     void
     ConstructParticle(dd4pod::Geant4ParticleCollection* col,
                       dd4hep::sim::Geant4Particle*      part);
 
     bool m_handleMCTruth;
 
     std::vector<std::string> m_disabledCollections;
 
     std::vector<std::string> m_enabledCollections;
 
     bool m_disableParticles = false;
 
     podio::EventStore  store;
     podio::ROOTWriter* writer = nullptr;
 
     std::map<std::string, podio::CollectionBase*>               m_col_map;
     std::vector<std::pair<std::string, podio::CollectionBase*>> m_collections;
     std::vector<std::pair<std::string, podio::CollectionBase*>>
                               m_readCollections;
     podio::CollectionIDTable* m_collectionIDs;
 
    public:
      Geant4Output2Podio(Geant4Context* context, const std::string& nam);
      virtual ~Geant4Output2Podio();
      
      //int fill(const std::string& nam, const ComponentCast& type, void* ptr);
 
      virtual void beginRun(const G4Run* run);
 
      virtual void endRun(const G4Run* run);
 
      virtual void saveRun(const G4Run*  run );
 
      virtual void saveCollection(OutputContext<G4Event>& ctxt, G4VHitsCollection* collection);
 
      virtual void saveEvent(OutputContext<G4Event>& ctxt);
 
      virtual void commit(OutputContext<G4Event>& ctxt);
    };
 
  }    // End namespace sim
}      // End namespace dd4hep
#endif // DD4HEP_DDG4_GEANT4Output2Podio_H