TrackManager.hh

Go to the documentation of this file.

#ifndef SLIC_TRACKMANAGER_HH_
#define SLIC_TRACKMANAGER_HH_ 1
 
// SLIC
#include "Module.hh"
#include "TrackSummary.hh"
#include "Singleton.hh"
 
// LCIO
#include "EVENT/LCEvent.h"
 
// Geant4
#include "G4Event.hh"
 
using EVENT::LCEvent;
 
namespace slic {
 
class TrackManager: public Singleton<TrackManager>, public Module {
 
    public:
 
        TrackManager() :
                Module("TrackManager") {
            _trackSummaries = new TrackSummaryVector;
            _trackSummaryMap = new TrackSummaryMap;
        }
 
        virtual ~TrackManager() {
            delete _trackSummaries;
            delete _trackSummaryMap;
        }
 
    public:
 
        TrackSummaryVector* getTrackSummaries() {
            return _trackSummaries;
        }
 
        void saveTrackSummaries(const G4Event* event, LCEvent* lcevent);
 
        void reset() {
            /* Clear the vector. */
            for (TrackSummaryVector::iterator it = _trackSummaries->begin(); it != _trackSummaries->end(); it++) {
                delete *it;
            }
            _trackSummaries->clear();
 
            /* Clear the map. */
            _trackSummaryMap->clear();
        }
 
        TrackSummaryMap* getTrackSummaryMap() {
            return _trackSummaryMap;
        }
 
        void registerTrackSummary(TrackSummary* trackSummary) {
            (*_trackSummaryMap)[trackSummary->getTrackID()] = trackSummary;
        }
 
        TrackSummary* findTrackSummary(G4int trackID) {
            return (*_trackSummaryMap)[trackID];
        }
 
    private:
 
        TrackSummary* findFirstSavedAncestor(TrackSummary* trackSummary);
 
    private:
 
        TrackSummaryVector* _trackSummaries;
        TrackSummaryMap* _trackSummaryMap;
};
 
}
 
#endif