File indexing completed on 2024-04-06 12:30:39
0001 #ifndef SimGeneral_PileupInformation_h
0002 #define SimGeneral_PileupInformation_h
0003
0004 #include <map>
0005
0006 #include "DataFormats/Common/interface/Handle.h"
0007
0008 #include "FWCore/Framework/interface/stream/EDProducer.h"
0009 #include "FWCore/Framework/interface/Event.h"
0010 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0011
0012
0013
0014 #include "SimDataFormats/CrossingFrame/interface/CrossingFrame.h"
0015 #include "SimDataFormats/CrossingFrame/interface/MixCollection.h"
0016 #include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
0017 #include "SimDataFormats/Track/interface/SimTrack.h"
0018 #include "SimDataFormats/Vertex/interface/SimVertex.h"
0019 #include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
0020 #include "SimDataFormats/TrackingAnalysis/interface/TrackingVertexContainer.h"
0021 #include "SimDataFormats/TrackingHit/interface/PSimHit.h"
0022 #include "SimDataFormats/PileupSummaryInfo/interface/PileupSummaryInfo.h"
0023 #include "SimDataFormats/PileupSummaryInfo/interface/PileupMixingContent.h"
0024 #include "SimDataFormats/PileupSummaryInfo/interface/PileupVertexContent.h"
0025
0026 #include "SimGeneral/TrackingAnalysis/interface/EncodedTruthId.h"
0027
0028 class PileupInformation : public edm::stream::EDProducer<> {
0029 public:
0030 explicit PileupInformation(const edm::ParameterSet &);
0031
0032 private:
0033 void produce(edm::Event &, const edm::EventSetup &) override;
0034
0035 edm::ParameterSet conf_;
0036
0037 typedef std::map<EncodedEventId, unsigned int> EncodedEventIdToIndex;
0038 typedef std::map<int, int> myindex;
0039 myindex event_index_;
0040
0041 std::vector<float> zpositions;
0042 std::vector<float> sumpT_lowpT;
0043 std::vector<float> sumpT_highpT;
0044 std::vector<int> ntrks_lowpT;
0045 std::vector<int> ntrks_highpT;
0046
0047 double distanceCut_;
0048 double volumeRadius_;
0049 double volumeZ_;
0050 double pTcut_1_;
0051 double pTcut_2_;
0052 bool isPreMixed_;
0053
0054 edm::EDGetTokenT<TrackingParticleCollection> trackingTruthT_;
0055 edm::EDGetTokenT<TrackingVertexCollection> trackingTruthV_;
0056 edm::EDGetTokenT<PileupMixingContent> PileupInfoLabel_;
0057 edm::EDGetTokenT<int> bunchSpacingToken_;
0058 edm::EDGetTokenT<std::vector<PileupSummaryInfo> > pileupSummaryToken_;
0059 edm::EDGetTokenT<PileupVertexContent> PileupVtxLabel_;
0060
0061 bool LookAtTrackingTruth_;
0062
0063 bool saveVtxTimes_;
0064
0065 std::string MessageCategory_;
0066
0067
0068
0069 };
0070
0071 #endif