PileupInformation.h

Go to the documentation of this file.

#ifndef SimGeneral_PileupInformation_h
#define SimGeneral_PileupInformation_h

#include <map>

#include "DataFormats/Common/interface/Handle.h"

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

//#include "CommonTools/RecoAlgos/interface/TrackingParticleSelector.h"

#include "SimDataFormats/CrossingFrame/interface/CrossingFrame.h"
#include "SimDataFormats/CrossingFrame/interface/MixCollection.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
#include "SimDataFormats/Track/interface/SimTrack.h"
#include "SimDataFormats/Vertex/interface/SimVertex.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingVertexContainer.h"
#include "SimDataFormats/TrackingHit/interface/PSimHit.h"
#include "SimDataFormats/PileupSummaryInfo/interface/PileupSummaryInfo.h"
#include "SimDataFormats/PileupSummaryInfo/interface/PileupMixingContent.h"
#include "SimDataFormats/PileupSummaryInfo/interface/PileupVertexContent.h"

#include "SimGeneral/TrackingAnalysis/interface/EncodedTruthId.h"

class PileupInformation : public edm::stream::EDProducer<>
{

public:

    explicit PileupInformation( const edm::ParameterSet & );

private:

    void produce( edm::Event &, const edm::EventSetup & ) override;

    edm::ParameterSet conf_;

    typedef std::map<EncodedEventId, unsigned int> EncodedEventIdToIndex;
    typedef std::map< int, int > myindex;
    myindex event_index_;

    std::vector<float> zpositions;
    std::vector<float> sumpT_lowpT;
    std::vector<float> sumpT_highpT;
    std::vector<int> ntrks_lowpT;
    std::vector<int> ntrks_highpT;


    double                   distanceCut_;
    double                   volumeRadius_;
    double                   volumeZ_;
    double                   pTcut_1_;
    double                   pTcut_2_;
    bool isPreMixed_;


    edm::EDGetTokenT<TrackingParticleCollection>     trackingTruthT_;
    edm::EDGetTokenT<TrackingVertexCollection>     trackingTruthV_;
    edm::EDGetTokenT<PileupMixingContent>            PileupInfoLabel_;
    edm::EDGetTokenT<int> bunchSpacingToken_;
    edm::EDGetTokenT<std::vector<PileupSummaryInfo> > pileupSummaryToken_;
    edm::EDGetTokenT<PileupVertexContent>            PileupVtxLabel_;

    bool LookAtTrackingTruth_ ;

    bool saveVtxTimes_;

    std::string MessageCategory_;
    //std::string simHitLabel_;
    //std::unique_ptr<MixCollection<SimTrack> >   simTracks_;
    //std::unique_ptr<MixCollection<SimVertex> >  simVertexes_;



};


#endif