File indexing completed on 2022-07-08 01:45:58
0001 #ifndef HeavyFlavorAnalysis_SpecificDecay_BPHWriteSpecificDecay_h
0002 #define HeavyFlavorAnalysis_SpecificDecay_BPHWriteSpecificDecay_h
0003
0004 #include "HeavyFlavorAnalysis/RecoDecay/interface/BPHAnalyzerTokenWrapper.h"
0005
0006 #include "FWCore/Framework/interface/Event.h"
0007 #include "FWCore/Framework/interface/EventSetup.h"
0008 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0009 #include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
0010
0011 #include "HeavyFlavorAnalysis/RecoDecay/interface/BPHPlusMinusCandidate.h"
0012 #include "HeavyFlavorAnalysis/RecoDecay/interface/BPHRecoCandidate.h"
0013 #include "HeavyFlavorAnalysis/RecoDecay/interface/BPHTrackReference.h"
0014
0015 #include "DataFormats/Candidate/interface/VertexCompositeCandidate.h"
0016 #include "HeavyFlavorAnalysis/RecoDecay/interface/BPHVertexCompositePtrCandidate.h"
0017
0018 #include "DataFormats/Common/interface/Ref.h"
0019 #include "DataFormats/TrackReco/interface/Track.h"
0020 #include "DataFormats/PatCandidates/interface/Muon.h"
0021 #include "DataFormats/PatCandidates/interface/GenericParticle.h"
0022 #include "DataFormats/PatCandidates/interface/CompositeCandidate.h"
0023
0024 #include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicParticle.h"
0025 #include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicVertex.h"
0026
0027 #include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
0028 #include "TrackingTools/PatternTools/interface/TwoTrackMinimumDistance.h"
0029 #include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
0030 #include "TrackingTools/TransientTrack/interface/TransientTrack.h"
0031 #include "TrackingTools/Records/interface/TransientTrackRecord.h"
0032
0033 #include <string>
0034 #include <vector>
0035 #include <map>
0036
0037 class TH1F;
0038 class BPHRecoCandidate;
0039
0040 class BPHWriteSpecificDecay : public BPHAnalyzerWrapper<BPHModuleWrapper::stream_producer> {
0041 public:
0042 explicit BPHWriteSpecificDecay(const edm::ParameterSet& ps);
0043 ~BPHWriteSpecificDecay() override = default;
0044
0045 static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
0046
0047 void produce(edm::Event& ev, const edm::EventSetup& es) override;
0048 virtual void fill(edm::Event& ev, const BPHEventSetupWrapper& es);
0049
0050 private:
0051 std::string pVertexLabel;
0052 std::string patMuonLabel;
0053 std::string ccCandsLabel;
0054 std::string pfCandsLabel;
0055 std::string pcCandsLabel;
0056 std::string gpCandsLabel;
0057 std::string k0CandsLabel;
0058 std::string l0CandsLabel;
0059 std::string kSCandsLabel;
0060 std::string lSCandsLabel;
0061
0062
0063 BPHESTokenWrapper<MagneticField, IdealMagneticFieldRecord> magFieldToken;
0064 BPHESTokenWrapper<TransientTrackBuilder, TransientTrackRecord> ttBToken;
0065 BPHTokenWrapper<std::vector<reco::Vertex> > pVertexToken;
0066 BPHTokenWrapper<pat::MuonCollection> patMuonToken;
0067 BPHTokenWrapper<std::vector<pat::CompositeCandidate> > ccCandsToken;
0068 BPHTokenWrapper<std::vector<reco::PFCandidate> > pfCandsToken;
0069 BPHTokenWrapper<std::vector<BPHTrackReference::candidate> > pcCandsToken;
0070 BPHTokenWrapper<std::vector<pat::GenericParticle> > gpCandsToken;
0071 BPHTokenWrapper<std::vector<reco::VertexCompositeCandidate> > k0CandsToken;
0072 BPHTokenWrapper<std::vector<reco::VertexCompositeCandidate> > l0CandsToken;
0073 BPHTokenWrapper<std::vector<reco::VertexCompositePtrCandidate> > kSCandsToken;
0074 BPHTokenWrapper<std::vector<reco::VertexCompositePtrCandidate> > lSCandsToken;
0075
0076 bool usePV;
0077 bool usePM;
0078 bool useCC;
0079 bool usePF;
0080 bool usePC;
0081 bool useGP;
0082 bool useK0;
0083 bool useL0;
0084 bool useKS;
0085 bool useLS;
0086
0087 std::string oniaName;
0088 std::string sdName;
0089 std::string ssName;
0090 std::string buName;
0091 std::string bpName;
0092 std::string bdName;
0093 std::string bsName;
0094 std::string k0Name;
0095 std::string l0Name;
0096 std::string b0Name;
0097 std::string lbName;
0098 std::string bcName;
0099 std::string psi2SName;
0100 std::string x3872Name;
0101
0102 enum recoType {
0103 Onia,
0104 Pmm,
0105 Psi1,
0106 Psi2,
0107 Ups,
0108 Ups1,
0109 Ups2,
0110 Ups3,
0111 Kx0,
0112 Pkk,
0113 Bu,
0114 Bp,
0115 Bd,
0116 Bs,
0117 K0s,
0118 Lambda0,
0119 B0,
0120 Lambdab,
0121 Bc,
0122 Psi2S,
0123 X3872
0124 };
0125 enum parType {
0126 ptMin,
0127 etaMax,
0128 mPsiMin,
0129 mPsiMax,
0130 mKx0Min,
0131 mKx0Max,
0132 mPhiMin,
0133 mPhiMax,
0134 mK0sMin,
0135 mK0sMax,
0136 mLambda0Min,
0137 mLambda0Max,
0138 massMin,
0139 massMax,
0140 probMin,
0141 mFitMin,
0142 mFitMax,
0143 constrMass,
0144 constrSigma,
0145 requireJPsi,
0146 constrMJPsi,
0147 constrMPsi2,
0148 writeCandidate
0149 };
0150 std::map<std::string, recoType> rMap;
0151 std::map<std::string, parType> pMap;
0152 std::map<std::string, parType> fMap;
0153 std::map<recoType, std::map<parType, double> > parMap;
0154
0155 bool recoOnia;
0156 bool recoKx0;
0157 bool recoPkk;
0158 bool recoBu;
0159 bool recoBp;
0160 bool recoBd;
0161 bool recoBs;
0162 bool recoK0s;
0163 bool recoLambda0;
0164 bool recoB0;
0165 bool recoLambdab;
0166 bool recoBc;
0167 bool recoPsi2S;
0168 bool recoX3872;
0169
0170 bool allKx0;
0171 bool allPkk;
0172 bool allK0s;
0173 bool allLambda0;
0174
0175 bool writeOnia;
0176 bool writeKx0;
0177 bool writePkk;
0178 bool writeBu;
0179 bool writeBp;
0180 bool writeBd;
0181 bool writeBs;
0182 bool writeK0s;
0183 bool writeLambda0;
0184 bool writeB0;
0185 bool writeLambdab;
0186 bool writeBc;
0187 bool writePsi2S;
0188 bool writeX3872;
0189
0190 bool writeVertex;
0191 bool writeMomentum;
0192
0193 std::vector<BPHPlusMinusConstCandPtr> lFull;
0194 std::vector<BPHPlusMinusConstCandPtr> lJPsi;
0195 std::vector<BPHRecoConstCandPtr> lSd;
0196 std::vector<BPHRecoConstCandPtr> lSs;
0197 std::vector<BPHRecoConstCandPtr> lBu;
0198 std::vector<BPHRecoConstCandPtr> lBp;
0199 std::vector<BPHRecoConstCandPtr> lBd;
0200 std::vector<BPHRecoConstCandPtr> lBs;
0201 std::vector<BPHPlusMinusConstCandPtr> lK0;
0202 std::vector<BPHPlusMinusConstCandPtr> lL0;
0203 std::vector<BPHRecoConstCandPtr> lB0;
0204 std::vector<BPHRecoConstCandPtr> lLb;
0205 std::vector<BPHRecoConstCandPtr> lBc;
0206 std::vector<BPHRecoConstCandPtr> lPsi2S;
0207 std::vector<BPHRecoConstCandPtr> lX3872;
0208
0209 std::map<const BPHRecoCandidate*, const BPHRecoCandidate*> jPsiOMap;
0210 std::map<const BPHRecoCandidate*, const BPHRecoCandidate*> daughMap;
0211 typedef edm::Ref<std::vector<reco::Vertex> > vertex_ref;
0212 std::map<const BPHRecoCandidate*, vertex_ref> pvRefMap;
0213 typedef edm::Ref<pat::CompositeCandidateCollection> compcc_ref;
0214 std::map<const BPHRecoCandidate*, compcc_ref> ccRefMap;
0215
0216 void setRecoParameters(const edm::ParameterSet& ps);
0217
0218 static void addTrackModes(const std::string& name, const BPHRecoCandidate& cand, std::string& modes, bool& count);
0219
0220 static void addTrackModes(const std::string& name, const BPHRecoCandidate& cand, pat::CompositeCandidate& cc);
0221
0222 template <class T>
0223 edm::OrphanHandle<pat::CompositeCandidateCollection> write(edm::Event& ev,
0224 const std::vector<T>& list,
0225 const std::string& name) {
0226 pat::CompositeCandidateCollection* ccList = new pat::CompositeCandidateCollection;
0227 int i;
0228 int n = list.size();
0229 std::map<const BPHRecoCandidate*, const BPHRecoCandidate*>::const_iterator dauIter;
0230 std::map<const BPHRecoCandidate*, const BPHRecoCandidate*>::const_iterator dauIend = daughMap.end();
0231 std::map<const BPHRecoCandidate*, const BPHRecoCandidate*>::const_iterator jpoIter;
0232 std::map<const BPHRecoCandidate*, const BPHRecoCandidate*>::const_iterator jpoIend = jPsiOMap.end();
0233 std::map<const BPHRecoCandidate*, vertex_ref>::const_iterator pvrIter;
0234 std::map<const BPHRecoCandidate*, vertex_ref>::const_iterator pvrIend = pvRefMap.end();
0235 std::map<const BPHRecoCandidate*, compcc_ref>::const_iterator ccrIter;
0236 std::map<const BPHRecoCandidate*, compcc_ref>::const_iterator ccrIend = ccRefMap.end();
0237 for (i = 0; i < n; ++i) {
0238 const T& ptr = list[i];
0239 ccList->push_back(ptr->composite());
0240 pat::CompositeCandidate& cc = ccList->back();
0241 std::string modes;
0242 bool count = false;
0243 addTrackModes("", *ptr, modes, count);
0244 cc.addUserData("trackModes", modes, true);
0245 addTrackModes("trackMode_", *ptr, cc);
0246 if ((pvrIter = pvRefMap.find(ptr.get())) != pvrIend)
0247 cc.addUserData("primaryVertex", pvrIter->second);
0248 const std::vector<std::string>& cNames = ptr->compNames();
0249 int j = 0;
0250 int m = cNames.size();
0251 while (j < m) {
0252 const std::string& compName = cNames[j++];
0253 const BPHRecoCandidate* cptr = ptr->getComp(compName).get();
0254 if ((ccrIter = ccRefMap.find(cptr)) == ccrIend) {
0255 if ((dauIter = daughMap.find(cptr)) != dauIend)
0256 cptr = dauIter->second;
0257 if ((jpoIter = jPsiOMap.find(cptr)) != jpoIend)
0258 cptr = jpoIter->second;
0259 }
0260 if ((ccrIter = ccRefMap.find(cptr)) != ccrIend) {
0261 compcc_ref cref = ccrIter->second;
0262 if (cref.isNonnull())
0263 cc.addUserData("refTo" + compName, cref);
0264 }
0265 }
0266 const BPHPlusMinusCandidate* pmp = dynamic_cast<const BPHPlusMinusCandidate*>(ptr.get());
0267 if (pmp != nullptr) {
0268 cc.addUserInt("cowboy", (pmp->isCowboy() ? +1 : -1));
0269
0270 }
0271 if (writeVertex)
0272 cc.addUserData("vertex", ptr->vertex());
0273 if (ptr->isEmpty())
0274 continue;
0275 if (writeVertex)
0276 cc.addUserData("fitVertex", reco::Vertex(*ptr->topDecayVertex()));
0277 if (ptr->isValidFit()) {
0278 const RefCountedKinematicParticle kinPart = ptr->topParticle();
0279 const KinematicState kinStat = kinPart->currentState();
0280 cc.addUserFloat("fitMass", kinStat.mass());
0281 if (writeMomentum)
0282 cc.addUserData("fitMomentum", kinStat.kinematicParameters().momentum());
0283 }
0284 }
0285 typedef std::unique_ptr<pat::CompositeCandidateCollection> ccc_pointer;
0286 edm::OrphanHandle<pat::CompositeCandidateCollection> ccHandle = ev.put(ccc_pointer(ccList), name);
0287 for (i = 0; i < n; ++i) {
0288 const BPHRecoCandidate* ptr = list[i].get();
0289 edm::Ref<pat::CompositeCandidateCollection> ccRef(ccHandle, i);
0290 ccRefMap[ptr] = ccRef;
0291 }
0292 return ccHandle;
0293 }
0294 };
0295
0296 #endif
