TrackingTruthAccumulator.h

Go to the documentation of this file.

#ifndef TrackingAnalysis_TrackingTruthAccumulator_h
#define TrackingAnalysis_TrackingTruthAccumulator_h

#include "SimGeneral/MixingModule/interface/DigiAccumulatorMixMod.h"
#include "SimGeneral/TrackingAnalysis/interface/TrackingParticleSelector.h"
#include <memory> // required for std::auto_ptr
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingVertexContainer.h"


// Forward declarations
namespace edm
{
    class ParameterSet;
    class EDProducer;
    class Event;
    class EventSetup;
}
class PileUpEventPrincipal;
class PSimHit;



class TrackingTruthAccumulator : public DigiAccumulatorMixMod
{
public:
    explicit TrackingTruthAccumulator( const edm::ParameterSet& config, edm::EDProducer& mixMod );
private:
    virtual void initializeEvent( const edm::Event& event, const edm::EventSetup& setup );
    virtual void accumulate( const edm::Event& event, const edm::EventSetup& setup );
    virtual void accumulate( const PileUpEventPrincipal& event, const edm::EventSetup& setup );
    virtual void finalizeEvent( edm::Event& event, const edm::EventSetup& setup );

    template<class T> void accumulateEvent( const T& event, const edm::EventSetup& setup );

    template<class T> void fillSimHits( std::vector<const PSimHit*>& returnValue, const T& event, const edm::EventSetup& setup );

    const std::string messageCategory_; 

    const double volumeRadius_;
    const double volumeZ_;
    const bool ignoreTracksOutsideVolume_;

    const unsigned int maximumPreviousBunchCrossing_;
    const unsigned int maximumSubsequentBunchCrossing_;
    const bool createUnmergedCollection_;
    const bool createMergedCollection_;
    const bool addAncestors_;

    const bool removeDeadModules_;
    const edm::InputTag simTrackLabel_;
    const edm::InputTag simVertexLabel_;
    edm::ParameterSet simHitCollectionConfig_;
    edm::InputTag genParticleLabel_;

    bool selectorFlag_;
    TrackingParticleSelector selector_;
    bool chargedOnly_;
    bool signalOnly_;

    bool allowDifferentProcessTypeForDifferentDetectors_;
public:
    // These always go hand in hand, and I need to pass them around in the internal
    // functions, so I might as well package them up in a struct.
    struct OutputCollections
    {
        std::auto_ptr<TrackingParticleCollection> pTrackingParticles;
        std::auto_ptr<TrackingVertexCollection> pTrackingVertices;
        TrackingParticleRefProd refTrackingParticles;
        TrackingVertexRefProd refTrackingVertexes;
    };
private:
    OutputCollections unmergedOutput_;
    OutputCollections mergedOutput_;
};

#endif // end of "#ifndef TrackingAnalysis_TrackingTruthAccumulator_h"