CMS 3D CMS Logo

PATPackedGenParticleProducer.cc
Go to the documentation of this file.
1 #include <string>
2 
21 
25 
32 #include "CLHEP/Vector/ThreeVector.h"
33 #include "CLHEP/Vector/LorentzVector.h"
34 #include "CLHEP/Matrix/Vector.h"
35 #include <string>
36 
37 namespace pat {
39  public:
42 
43  void produce(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
44 
45  private:
50  const double maxRapidity_;
51  };
52 } // namespace pat
53 
55  : Cands_(consumes<reco::GenParticleCollection>(iConfig.getParameter<edm::InputTag>("inputCollection"))),
56  GenOrigs_(consumes<reco::GenParticleCollection>(iConfig.getParameter<edm::InputTag>("inputOriginal"))),
57  Asso_(consumes<edm::Association<reco::GenParticleCollection>>(iConfig.getParameter<edm::InputTag>("map"))),
58  AssoOriginal_(consumes<edm::Association<reco::GenParticleCollection>>(
59  iConfig.getParameter<edm::InputTag>("inputCollection"))),
60  maxRapidity_(iConfig.getParameter<double>("maxRapidity")) {
61  produces<std::vector<pat::PackedGenParticle>>();
62  produces<edm::Association<std::vector<pat::PackedGenParticle>>>();
63 }
64 
66 
69  const edm::EventSetup& iSetup) const {
71  iEvent.getByToken(Cands_, cands);
72 
73  //from prunedGenParticlesWithStatusOne to prunedGenParticles
75  iEvent.getByToken(Asso_, asso);
76 
78  iEvent.getByToken(AssoOriginal_, assoOriginal);
79 
81  iEvent.getByToken(GenOrigs_, genOrigs);
82  std::vector<int> mapping(genOrigs->size(), -1);
83 
84  //invert the value map from Orig2New to New2Orig
85  std::map<edm::Ref<reco::GenParticleCollection>, edm::Ref<reco::GenParticleCollection>> reverseMap;
86  for (unsigned int ic = 0, nc = genOrigs->size(); ic < nc; ++ic) {
88  edm::Ref<reco::GenParticleCollection> newRef = (*assoOriginal)[originalRef];
89  reverseMap.insert(
91  }
92 
93  auto outPtrP = std::make_unique<std::vector<pat::PackedGenParticle>>();
94 
95  unsigned int packed = 0;
96  for (unsigned int ic = 0, nc = cands->size(); ic < nc; ++ic) {
97  const reco::GenParticle& cand = (*cands)[ic];
98  if (cand.status() == 1 && std::abs(cand.y()) < maxRapidity_) {
99  // Obtain original gen particle collection reference from input reference and map
101  edm::Ref<reco::GenParticleCollection> originalRef = reverseMap[inputRef];
102  edm::Ref<reco::GenParticleCollection> finalPrunedRef = (*asso)[inputRef];
103  mapping[originalRef.key()] = packed;
104  packed++;
105  if (finalPrunedRef.isNonnull()) { //this particle exists also in the final pruned
106  outPtrP->push_back(pat::PackedGenParticle(cand, finalPrunedRef));
107  } else {
108  if (cand.numberOfMothers() > 0) {
109  edm::Ref<reco::GenParticleCollection> newRef = (*asso)[cand.motherRef(0)];
110  outPtrP->push_back(pat::PackedGenParticle(cand, newRef));
111  } else {
113  }
114  }
115  }
116  }
117 
119 
120  auto gp2pgp = std::make_unique<edm::Association<std::vector<pat::PackedGenParticle>>>(oh);
121  edm::Association<std::vector<pat::PackedGenParticle>>::Filler gp2pgpFiller(*gp2pgp);
122  gp2pgpFiller.insert(genOrigs, mapping.begin(), mapping.end());
123  gp2pgpFiller.fill();
124  iEvent.put(std::move(gp2pgp));
125 }
126 
std::vector< GenParticle > GenParticleCollection
collection of GenParticles
PATPackedGenParticleProducer(const edm::ParameterSet &)
bool isNonnull() const
Checks for non-null.
Definition: Ref.h:232
key_type key() const
Accessor for product key.
Definition: Ref.h:244
Definition: HeavyIon.h:7
const edm::EDGetTokenT< reco::GenParticleCollection > Cands_
int iEvent
Definition: GenABIO.cc:224
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
#define DEFINE_FWK_MODULE(type)
Definition: MakerMacros.h:16
const edm::EDGetTokenT< edm::Association< reco::GenParticleCollection > > Asso_
fixed size matrix
HLT enums.
const edm::EDGetTokenT< reco::GenParticleCollection > GenOrigs_
void produce(edm::StreamID, edm::Event &, const edm::EventSetup &) const override
const edm::EDGetTokenT< edm::Association< reco::GenParticleCollection > > AssoOriginal_
def move(src, dest)
Definition: eostools.py:511