Merge pull request #49218 from stahlleiton/DeDxMiniAOD_CMSSW_16_0_X

Reduce size of dEdx estimators in PbPb MiniAOD

Author: cmsbuild

PhysicsTools/PatAlgos/plugins/DeDxEstimatorRekeyer.cc

@@ -5,6 +5,7 @@
 #include "DataFormats/TrackReco/interface/DeDxHitInfo.h"
 #include "DataFormats/TrackReco/interface/TrackFwd.h"
 #include "DataFormats/PatCandidates/interface/PackedCandidate.h"
+#include "DataFormats/Math/interface/libminifloat.h"
 
 //
 // class declaration
@@ -101,8 +102,13 @@ void DeDxEstimatorRekeyer::produce(edm::StreamID, edm::Event& iEvent, const edm:
     // Loop over packed candidates
     for (const auto& h : pcTrkMap) {
       std::vector<reco::DeDxData> dedxEstimate(h.first->size());
-      for (const auto& p : h.second)
-        dedxEstimate[p.first.key()] = dedxEstimators[p.second];
+      for (const auto& p : h.second) {
+        const auto& dedx = dedxEstimators[p.second];
+        const auto val = MiniFloatConverter::reduceMantissaToNbitsRounding<12>(dedx.dEdx());
+        const auto err = MiniFloatConverter::reduceMantissaToNbitsRounding<8>(dedx.dEdxError());
+        dedxEstimate[p.first.key()] =
+            reco::DeDxData(val, err, dedx.numberOfSaturatedMeasurements(), dedx.numberOfMeasurements());
+      }
       filler.insert(h.first, dedxEstimate.begin(), dedxEstimate.end());
     }
     // Fill the value map and put it into the event

PhysicsTools/PatAlgos/python/slimming/MicroEventContent_cff.py

@@ -173,7 +173,7 @@
 from Configuration.Eras.Modifier_run3_upc_cff import run3_upc
 run3_upc.toModify(MicroEventContent, outputCommands = MicroEventContent.outputCommands + _upc_extraCommands)
 
-_dedx_extraCommands = ["keep recoDeDxDataedmValueMap_dedxEstimator_*_*"]
+_dedx_extraCommands = ["keep recoDeDxDataedmValueMap_dedxEstimator_dedxAllLikelihood_*"]
 from Configuration.Eras.Modifier_dedx_lfit_cff import dedx_lfit
 dedx_lfit.toModify(MicroEventContent, outputCommands = MicroEventContent.outputCommands + _dedx_extraCommands)