From cc28b60365f276e21c8f7ef601ac2ac47b6682e4 Mon Sep 17 00:00:00 2001
From: Robinpreet Dhillon <rdhill13@narval2.narval.calcul.quebec>
Date: Wed, 30 Mar 2022 15:26:56 -0400
Subject: [PATCH] added shaper function

---
 .../imaging_shower_ML/options/imaging_ml_data.py       | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/benchmarks/imaging_shower_ML/options/imaging_ml_data.py b/benchmarks/imaging_shower_ML/options/imaging_ml_data.py
index 4b8188c9..fbdcf932 100644
--- a/benchmarks/imaging_shower_ML/options/imaging_ml_data.py
+++ b/benchmarks/imaging_shower_ML/options/imaging_ml_data.py
@@ -12,7 +12,7 @@ from Configurables import Jug__Reco__CalorimeterHitsMerger as CalHitsMerger
 from Configurables import Jug__Reco__ImagingPixelMerger as MLDataMerger
 from Configurables import Jug__Reco__ImagingPixelDataSorter as MLDataSorter
 from Configurables import Jug__Reco__ImagingPixelDataCombiner as MLDataCombiner
-
+from Configurables import Jug__Reco__ImagingPixelDataShaper as MLDataShaper
 
 # input arguments through environment variables
 kwargs = dict()
@@ -123,7 +123,11 @@ becal_combiner = MLDataCombiner('becal_combiner',
         layerIncrement=100,
         rule=kwargs['combine'])
 
-        
+
+# shape layers
+becal_shaper = MLDataShaper('becal_combiner',
+        inputHits=becal_combiner.outputHits)
+
 
 podout.outputCommands = [
 #        'keep *',
@@ -138,7 +142,7 @@ ApplicationMgr(
     TopAlg=[podin,
             becal_img_digi, becal_img_reco, becal_img_merger, becal_img_sorter,
             becal_scfi_digi, becal_scfi_reco, becal_scfi_merger, becal_scfi_sorter,
-            becal_combiner,
+            becal_combiner, becal_shaper,
             podout],
     EvtSel='NONE',
     EvtMax=kwargs['nev'],
-- 
GitLab