Replace hardcoded value with the number of samples from EcalDataFrame

cms-patatrack · Jul 21, 2020 · d368e22 · d368e22
1 parent 4a55c17
commit d368e22
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/EventFilter/EcalRawToDigi/plugins/EcalCPUDigisProducer.cc b/EventFilter/EcalRawToDigi/plugins/EcalCPUDigisProducer.cc
@@ -3,6 +3,7 @@
 #include "CUDADataFormats/EcalDigi/interface/DigisCollection.h"
 #include "CondFormats/DataRecord/interface/EcalMappingElectronicsRcd.h"
 #include "DataFormats/EcalDetId/interface/EcalDetIdCollections.h"
+#include "DataFormats/EcalDigi/interface/EcalDataFrame.h"
 #include "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
 #include "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
 #include "DataFormats/FEDRawData/interface/FEDRawDataCollection.h"
@@ -134,11 +135,10 @@ void EcalCPUDigisProducer::acquire(edm::Event const& event,
   auto const& eedigis = ctx.get(eedigisProduct);
 
   // resize tmp buffers
-  // FIXME remove hardcoded values
+  dataebtmp.resize(ebdigis.size * EcalDataFrame::MAXSAMPLES);
+  dataeetmp.resize(eedigis.size * EcalDataFrame::MAXSAMPLES);
   idsebtmp.resize(ebdigis.size);
-  dataebtmp.resize(ebdigis.size * 10);
   idseetmp.resize(eedigis.size);
-  dataeetmp.resize(eedigis.size * 10);
 
   // enqeue transfers
   cudaCheck(cudaMemcpyAsync(