LcioHitsCollectionBuilder.hh

Go to the documentation of this file.

#ifndef SLIC_LCIOHITSCOLLECTIONBUILDER_HH_
#define SLIC_LCIOHITSCOLLECTIONBUILDER_HH_ 1
 
// SLIC
#include "Module.hh"
 
// LCDD
#include "lcdd/detectors/SensitiveDetector.hh"
#include "lcdd/hits/CalorimeterHit.hh"
#include "lcdd/hits/TrackerHit.hh"
 
// LCIO
#include "EVENT/LCEvent.h"
#include "IMPL/LCFlagImpl.h"
#include "IMPL/LCEventImpl.h"
#include "IMPL/LCCollectionVec.h"
#include "IMPL/SimCalorimeterHitImpl.h"
#include "IMPL/SimTrackerHitImpl.h"
 
// Geant4
#include "G4Event.hh"
 
namespace slic {
 
class LcioMcpManager;
 
class LcioHitsCollectionBuilder: public Module {
 
    public:
 
        LcioHitsCollectionBuilder();
 
        virtual ~LcioHitsCollectionBuilder();
 
        EVENT::LCEvent* createHitCollectionsFromEvent(const G4Event* event, EVENT::LCEvent* lcioEvent);
 
        void setLongFlag(bool longFlag = true);
 
        void setPDGFlag(bool pdgFlag = true);
 
        void setStoreMomentum(bool storeMomentum = true) {
            m_storeMomentum = storeMomentum;
        }
 
        bool getStoreMomentum() {
            return m_storeMomentum;
        }
 
    private:
 
        /*
         * Create the output LCIO hit collections from the current Geant4 event,
         * using the instance variables of this class.
         */
        void createHitCollections();
 
        IMPL::LCCollectionVec* createCollectionVec(G4VHitsCollection* hitCollection, SensitiveDetector::EType sensitiveDetectorType);
 
        IMPL::LCCollectionVec* createTrackerCollectionVec(G4VHitsCollection* hitCollection);
 
        IMPL::LCCollectionVec* createCalorimeterCollectionVec(G4VHitsCollection* hitCollection);
 
        void saveHits(CalorimeterHitsCollection* calorimeterHits, IMPL::LCCollectionVec* collection);
 
        void saveHits(TrackerHitsCollection* trackerHits, IMPL::LCCollectionVec* collection);
 
        IMPL::SimCalorimeterHitImpl* createHit(CalorimeterHit* calorimeterHit);
 
        IMPL::SimTrackerHitImpl* createHit(TrackerHit* trackerHit);
 
        void addParticleContributions(CalorimeterHit* calorimeterHit, IMPL::SimCalorimeterHitImpl* simCalorimeterHit);
 
        void setFlagDefaults();
 
        void combineMcpHitContribs(const HitContributionList& allContributions, HitContributionList& combinedContributions);
 
        HitContribution* findHitContribution(int trackID, HitContributionList& contributions);
 
        void setEndcapFlag(SensitiveDetector* sensitiveDetector);
 
        static bool containsCollection(EVENT::LCEvent* event, const std::string& name);
 
    private:
 
        EVENT::LCEvent* m_currentLCEvent;
 
        const G4Event* m_currentG4Event;
 
        IMPL::LCFlagImpl m_trkCollFlag;
        IMPL::LCFlagImpl m_calCollFlag;
 
        bool m_storeMomentum;
};
}
 
#endif