Merge pull request cms-sw#44660 from AlexDeMoor/from-CMSSW_14_0_X_202…

…4-04-07-2300 [14_0_X] Introduce Unified Particle Transformer AK4 jet tagger (Backport)
stahlleiton · Aug 5, 2024 · 311adb9 · 311adb9
1 parent 85b53af
commit 311adb9
Show file tree

Hide file tree

Showing 28 changed files with 2,260 additions and 40 deletions.
diff --git a/DataFormats/BTauReco/interface/ChargedCandidateFeatures.h b/DataFormats/BTauReco/interface/ChargedCandidateFeatures.h
@@ -22,7 +22,6 @@ namespace btagbtvdeep {
     float btagPf_trackSip3dSig;
     float btagPf_trackSip2dVal;
     float btagPf_trackSip2dSig;
-
     float btagPf_trackJetDistVal;
 
     float drsubjet1;
@@ -39,6 +38,15 @@ namespace btagbtvdeep {
     float drminsv;
     float distminsv;
 
+    float charge;
+    float btagPf_trackDecayLen;
+    float HadFrac;
+    float CaloFrac;
+    float pdgID;
+    float lostInnerHits;
+    float numberOfPixelHits;
+    float numberOfStripHits;
+
     float pt;
     float px;
     float py;

diff --git a/DataFormats/BTauReco/interface/LostTracksFeatures.h b/DataFormats/BTauReco/interface/LostTracksFeatures.h
@@ -0,0 +1,34 @@
+#ifndef DataFormats_BTauReco_LostTracksFeatures_h
+#define DataFormats_BTauReco_LostTracksFeatures_h
+
+namespace btagbtvdeep {
+
+  class LostTracksFeatures {
+  public:
+    float btagPf_trackEtaRel;
+    float btagPf_trackPtRel;
+    float btagPf_trackPPar;
+    float btagPf_trackDeltaR;
+    float btagPf_trackPParRatio;
+    float btagPf_trackSip2dVal;
+    float btagPf_trackSip2dSig;
+    float btagPf_trackSip3dVal;
+    float btagPf_trackSip3dSig;
+    float btagPf_trackJetDistVal;
+    float drminsv;
+    float charge;
+    float puppiw;
+    float chi2;
+    float quality;
+    float lostInnerHits;
+    float numberOfPixelHits;
+    float numberOfStripHits;
+    float pt;
+    float eta;
+    float phi;
+    float e;
+  };
+
+}  // namespace btagbtvdeep
+
+#endif  //DataFormats_BTauReco_LostTracksFeatures_h
diff --git a/DataFormats/BTauReco/interface/UnifiedParticleTransformerAK4Features.h b/DataFormats/BTauReco/interface/UnifiedParticleTransformerAK4Features.h
@@ -0,0 +1,25 @@
+#ifndef DataFormats_BTauReco_UnifiedParticleTransformerAK4Features_h
+#define DataFormats_BTauReco_UnifiedParticleTransformerAK4Features_h
+
+#include "DataFormats/BTauReco/interface/SecondaryVertexFeatures.h"
+#include "DataFormats/BTauReco/interface/NeutralCandidateFeatures.h"
+#include "DataFormats/BTauReco/interface/ChargedCandidateFeatures.h"
+#include "DataFormats/BTauReco/interface/LostTracksFeatures.h"
+
+#include <vector>
+
+namespace btagbtvdeep {
+
+  class UnifiedParticleTransformerAK4Features {
+  public:
+    bool is_filled = true;
+    std::vector<SecondaryVertexFeatures> sv_features;
+
+    std::vector<NeutralCandidateFeatures> n_pf_features;
+    std::vector<ChargedCandidateFeatures> c_pf_features;
+    std::vector<LostTracksFeatures> lt_features;
+  };
+
+}  // namespace btagbtvdeep
+
+#endif  //DataFormats_BTauReco_UnifiedParticleTransformerAK4Features_h
diff --git a/DataFormats/BTauReco/interface/UnifiedParticleTransformerAK4TagInfo.h b/DataFormats/BTauReco/interface/UnifiedParticleTransformerAK4TagInfo.h
@@ -0,0 +1,15 @@
+#ifndef DataFormats_BTauReco_UnifiedParticleTransformerAK4TagInfo_h
+#define DataFormats_BTauReco_UnifiedParticleTransformerAK4TagInfo_h
+
+#include "DataFormats/BTauReco/interface/UnifiedParticleTransformerAK4Features.h"
+#include "DataFormats/BTauReco/interface/FeaturesTagInfo.h"
+
+namespace reco {
+
+  typedef FeaturesTagInfo<btagbtvdeep::UnifiedParticleTransformerAK4Features> UnifiedParticleTransformerAK4TagInfo;
+
+  DECLARE_EDM_REFS(UnifiedParticleTransformerAK4TagInfo)
+
+}  // namespace reco
+
+#endif  // DataFormats_BTauReco_UnifiedParticleTransformerAK4TagInfo_h
diff --git a/DataFormats/BTauReco/src/classes.h b/DataFormats/BTauReco/src/classes.h
@@ -65,6 +65,8 @@
 #include "DataFormats/BTauReco/interface/DeepDoubleXTagInfo.h"
 #include "DataFormats/BTauReco/interface/ParticleTransformerAK4Features.h"
 #include "DataFormats/BTauReco/interface/ParticleTransformerAK4TagInfo.h"
+#include "DataFormats/BTauReco/interface/UnifiedParticleTransformerAK4Features.h"
+#include "DataFormats/BTauReco/interface/UnifiedParticleTransformerAK4TagInfo.h"
 #include "DataFormats/BTauReco/interface/DeepBoostedJetTagInfo.h"
 #include "DataFormats/BTauReco/interface/PixelClusterTagInfo.h"
 

diff --git a/DataFormats/BTauReco/src/classes_def.xml b/DataFormats/BTauReco/src/classes_def.xml
@@ -329,6 +329,8 @@
   <class name="edm::reftobase::RefHolder<reco::DeepFlavourTagInfoRef>" />
   <class name="edm::reftobase::Holder<reco::BaseTagInfo, reco::ParticleTransformerAK4TagInfoRef>" />
   <class name="edm::reftobase::RefHolder<reco::ParticleTransformerAK4TagInfoRef>" />
+  <class name="edm::reftobase::Holder<reco::BaseTagInfo, reco::UnifiedParticleTransformerAK4TagInfoRef>" />
+  <class name="edm::reftobase::RefHolder<reco::UnifiedParticleTransformerAK4TagInfoRef>" />  
   <class name="edm::reftobase::Holder<reco::BaseTagInfo, reco::CombinedTauTagInfoRef>" />
   <class name="edm::reftobase::RefHolder<reco::CombinedTauTagInfoRef>" />
   <class name="edm::reftobase::Holder<reco::BaseTagInfo, reco::IsolatedTauTagInfoRef>" />
@@ -366,6 +368,8 @@
   <class name="edm::reftobase::RefHolder<reco::DeepFlavourTagInfoFwdRef>" />
   <class name="edm::reftobase::Holder<reco::BaseTagInfo, reco::ParticleTransformerAK4TagInfoFwdRef>" />
   <class name="edm::reftobase::RefHolder<reco::ParticleTransformerAK4TagInfoFwdRef>" />
+  <class name="edm::reftobase::Holder<reco::BaseTagInfo, reco::UnifiedParticleTransformerAK4TagInfoFwdRef>" />
+  <class name="edm::reftobase::RefHolder<reco::UnifiedParticleTransformerAK4TagInfoFwdRef>" />
   <class name="edm::reftobase::Holder<reco::BaseTagInfo, reco::CombinedTauTagInfoFwdRef>" />
   <class name="edm::reftobase::RefHolder<reco::CombinedTauTagInfoFwdRef>" />
   <class name="edm::reftobase::Holder<reco::BaseTagInfo, reco::IsolatedTauTagInfoFwdRef>" />
@@ -427,7 +431,8 @@
    <version ClassVersion="4" checksum="3657612864"/>
     <version ClassVersion="3" checksum="938442516"/>
   </class>
-  <class name="btagbtvdeep::ChargedCandidateFeatures" ClassVersion="6">
+  <class name="btagbtvdeep::ChargedCandidateFeatures" ClassVersion="7">
+   <version ClassVersion="7" checksum="2760889180"/>
    <version ClassVersion="6" checksum="3294985591"/>
    <version ClassVersion="5" checksum="3385708923"/>
     <version ClassVersion="4" checksum="2173116472"/>
@@ -467,6 +472,14 @@
   <class name="reco::ParticleTransformerAK4TagInfoRefProd"/>
   <class name="reco::ParticleTransformerAK4TagInfoRefVector"/>
   <class name="edm::Wrapper<reco::ParticleTransformerAK4TagInfoCollection>"/>
+
+  <class name="reco::UnifiedParticleTransformerAK4TagInfo"/>
+  <class name="reco::UnifiedParticleTransformerAK4TagInfoCollection"/>
+  <class name="reco::UnifiedParticleTransformerAK4TagInfoRef"/>
+  <class name="reco::UnifiedParticleTransformerAK4TagInfoFwdRef"/>
+  <class name="reco::UnifiedParticleTransformerAK4TagInfoRefProd"/>
+  <class name="reco::UnifiedParticleTransformerAK4TagInfoRefVector"/>
+  <class name="edm::Wrapper<reco::UnifiedParticleTransformerAK4TagInfoCollection>"/>
 
   <class name="reco::DeepDoubleXTagInfo"/>
   <class name="reco::DeepDoubleXTagInfoCollection"/>
@@ -477,6 +490,7 @@
   <class name="edm::Wrapper<reco::DeepDoubleXTagInfoCollection>"/>
 
   <class name="btagbtvdeep::DeepBoostedJetFeatures"/>
+  <class name="btagbtvdeep::LostTracksFeatures"/>
   <class name="reco::DeepBoostedJetTagInfo"/>
   <class name="reco::DeepBoostedJetTagInfoCollection"/>
   <class name="reco::DeepBoostedJetTagInfoRef"/>