Add UParT pT regression nodes in NanoAOD

Author: nurfikri89

PhysicsTools/NanoAOD/python/jetsAK8_cff.py

@@ -253,6 +253,12 @@ def nanoAOD_addDeepInfoAK8(process, addDeepBTag, addDeepBoostedJet, addDeepDoubl
     variables = cms.PSet(P4Vars,
         btagDeepFlavB = Var("bDiscriminator('pfDeepFlavourJetTags:probb')+bDiscriminator('pfDeepFlavourJetTags:probbb')+bDiscriminator('pfDeepFlavourJetTags:problepb')",float,doc="DeepJet b+bb+lepb tag discriminator",precision=10),
         btagUParTAK4B = Var("?bDiscriminator('pfUnifiedParticleTransformerAK4DiscriminatorsJetTags:BvsAll')>0?bDiscriminator('pfUnifiedParticleTransformerAK4DiscriminatorsJetTags:BvsAll'):-1",float,precision=10,doc="UnifiedParT b vs. udscg"),
+        UParTAK4RegPtRawCorr = Var("?bDiscriminator('pfUnifiedParticleTransformerAK4JetTags:ptcorr')>0?bDiscriminator('pfUnifiedParticleTransformerAK4JetTags:ptcorr'):-1",float,precision=10,doc="UnifiedParT universal flavor-aware visible pT regression (no neutrinos), correction relative to raw jet pT"),
+        UParTAK4RegPtRawCorrNeutrino = Var("?bDiscriminator('pfUnifiedParticleTransformerAK4JetTags:ptnu')>0?bDiscriminator('pfUnifiedParticleTransformerAK4JetTags:ptnu'):-1",float,precision=10,doc="UnifiedParT universal flavor-aware pT regression neutrino correction, relative to visible. Correction relative to raw jet pT"),
+        UParTAK4RegPtRawRes = Var("?(bDiscriminator('pfUnifiedParticleTransformerAK4JetTags:ptreshigh')+bDiscriminator('pfUnifiedParticleTransformerAK4JetTags:ptreslow'))>0?0.5*(bDiscriminator('pfUnifiedParticleTransformerAK4JetTags:ptreshigh')-bDiscriminator('pfUnifiedParticleTransformerAK4JetTags:ptreslow')):-1",float,precision=10,doc="UnifiedParT universal flavor-aware jet pT resolution estimator, (q84 - q16)/2"),
+        UParTAK4V1RegPtRawCorr = Var("?bDiscriminator('pfUnifiedParticleTransformerAK4V1JetTags:ptcorr')>0?bDiscriminator('pfUnifiedParticleTransformerAK4V1JetTags:ptcorr'):-1",float,precision=10,doc="UnifiedParT V1 universal flavor-aware visible pT regression (no neutrinos), correction relative to raw jet pT"),
+        UParTAK4V1RegPtRawCorrNeutrino = Var("?bDiscriminator('pfUnifiedParticleTransformerAK4V1JetTags:ptnu')>0?bDiscriminator('pfUnifiedParticleTransformerAK4V1JetTags:ptnu'):-1",float,precision=10,doc="UnifiedParT V1 universal flavor-aware pT regression neutrino correction, relative to visible. Correction relative to raw jet pT"),
+        UParTAK4V1RegPtRawRes = Var("?(bDiscriminator('pfUnifiedParticleTransformerAK4V1JetTags:ptreshigh')+bDiscriminator('pfUnifiedParticleTransformerAK4V1JetTags:ptreslow'))>0?0.5*(bDiscriminator('pfUnifiedParticleTransformerAK4V1JetTags:ptreshigh')-bDiscriminator('pfUnifiedParticleTransformerAK4V1JetTags:ptreslow')):-1",float,precision=10,doc="UnifiedParT V1 universal flavor-aware jet pT resolution estimator, (q84 - q16)/2"),
         rawFactor = Var("1.-jecFactor('Uncorrected')",float,doc="1 - Factor to get back to raw pT",precision=6),
         area = Var("jetArea()", float, doc="jet catchment area, for JECs",precision=10),
         tau1 = Var("userFloat('NjettinessAK8Subjets:tau1')",float, doc="Nsubjettiness (1 axis)",precision=10),
@@ -272,12 +278,24 @@ def nanoAOD_addDeepInfoAK8(process, addDeepBTag, addDeepBoostedJet, addDeepDoubl
 (run2_nanoAOD_106Xv2).toModify(
     subJetTable.variables,
     area = None,
+    UParTAK4RegPtRawCorr = None,
+    UParTAK4RegPtRawCorrNeutrino = None,
+    UParTAK4RegPtRawRes = None,
+    UParTAK4V1RegPtRawCorr = None,
+    UParTAK4V1RegPtRawCorrNeutrino = None,
+    UParTAK4V1RegPtRawRes = None,
 )
 
 run3_nanoAOD_pre142X.toModify(
     subJetTable.variables,
     btagDeepFlavB = None,
     btagUParTAK4B = None,
+    UParTAK4RegPtRawCorr = None,
+    UParTAK4RegPtRawCorrNeutrino = None,
+    UParTAK4RegPtRawRes = None,
+    UParTAK4V1RegPtRawCorr = None,
+    UParTAK4V1RegPtRawCorrNeutrino = None,
+    UParTAK4V1RegPtRawRes = None,
     btagDeepB = Var("bDiscriminator('pfDeepCSVJetTags:probb')+bDiscriminator('pfDeepCSVJetTags:probbb')",float,doc="DeepCSV b+bb tag discriminator",precision=10),
 )
 

PhysicsTools/NanoAOD/python/nanoDQM_cff.py

@@ -215,7 +215,7 @@ def _match(name):
 
 _SubJet_pre142X_plots = cms.VPSet()
 for plot in nanoDQM.vplots.SubJet.plots:
-    if 'btagDeepFlavB' not in plot.name.value() and 'btagUParTAK4B' not in plot.name.value():
+    if ('btagDeepFlavB' not in plot.name.value()) and ('btagUParTAK4B' not in plot.name.value()) and ('UParTAK4Reg' not in plot.name.value()):
         _SubJet_pre142X_plots.append(plot)
 _SubJet_pre142X_plots.extend([
     Plot1D('btagDeepB', 'btagDeepB', 20, -1, 1, 'Deep B+BB btag discriminator'),

PhysicsTools/NanoAOD/python/nanoDQM_cfi.py

@@ -877,6 +877,12 @@
                 Count1D('_size', 9, -0.5, 8.5, 'slimmedJetsAK8PFPuppiSoftDropPacked::SubJets, i.e. soft-drop subjets for ak8 fat jets for boosted'),
                 Plot1D('btagDeepFlavB', 'btagDeepFlavB', 20, -1, 1, 'DeepJet b+bb+lepb tag discriminator'),
                 Plot1D('btagUParTAK4B', 'btagUParTAK4B', 20, -1, 1, 'UnifiedParT b vs. udscg'),
+                Plot1D('UParTAK4RegPtRawCorr', 'UParTAK4RegPtRawCorr', 20, 0, 2, 'UnifiedParT universal flavor-aware visible pT regression (no neutrinos), correction relative to raw jet pT'),
+                Plot1D('UParTAK4RegPtRawCorrNeutrino', 'UParTAK4RegPtRawCorrNeutrino', 20, 0, 2, 'UnifiedParT universal flavor-aware pT regression neutrino correction, relative to visible. Correction relative to raw jet pT'),
+                Plot1D('UParTAK4RegPtRawRes', 'UParTAK4RegPtRawRes', 20, 0, 1, 'UnifiedParT universal flavor-aware jet pT resolution estimator, (q84 - q16)/2'),
+                Plot1D('UParTAK4V1RegPtRawCorr', 'UParTAK4V1RegPtRawCorr', 20, 0, 2, 'UnifiedParT V1 universal flavor-aware visible pT regression (no neutrinos), correction relative to raw jet pT'),
+                Plot1D('UParTAK4V1RegPtRawCorrNeutrino', 'UParTAK4V1RegPtRawCorrNeutrino', 20, 0, 2, 'UnifiedParT V1 universal flavor-aware pT regression neutrino correction, relative to visible. Correction relative to raw jet pT'),
+                Plot1D('UParTAK4V1RegPtRawRes', 'UParTAK4V1RegPtRawRes', 20, 0, 1, 'UnifiedParT V1 universal flavor-aware jet pT resolution estimator, (q84 - q16)/2'),
                 Plot1D('eta', 'eta', 20, -4, 4, 'eta'),
                 Plot1D('hadronFlavour', 'hadronFlavour', 6, -0.5, 5.5, 'flavour from hadron ghost clustering'),
                 Plot1D('mass', 'mass', 20, -200, 200, 'mass'),