@@ -259,6 +259,13 @@ void runMassFitter(const std::string& configFileName)
259259 }
260260 };
261261
262+ if ((!isMc && enableRefl ) || isMc ) {
263+ checkVectorSizeMcHistograms (signalHistoName , promptHistoName , fdHistoName );
264+ }
265+ if (isMc && includeSecPeak ) {
266+ checkVectorSizeMcHistograms (signalSecPeakHistoName , promptSecPeakHistoName , fdSecPeakHistoName );
267+ }
268+
262269 for (int iSliceVar = 0 ; iSliceVar < nHistograms ; iSliceVar ++ ) {
263270 sliceVarLimits [iSliceVar ] = sliceVarMin [iSliceVar ];
264271
@@ -311,8 +318,6 @@ void runMassFitter(const std::string& configFileName)
311318 hMass [iSliceVar ] = getObjectWithNullPtrCheck < TH1 > (inputFile , inputHistoName [iSliceVar ]);
312319 if (enableRefl ) {
313320 hMassRefl [iSliceVar ] = getObjectWithNullPtrCheck < TH1 > (inputFileRefl , reflHistoName [iSliceVar ]);
314-
315- checkVectorSizeMcHistograms (signalHistoName , promptHistoName , fdHistoName );
316321 if (!signalHistoName .empty ()) {
317322 hMassSgn [iSliceVar ] = getObjectWithNullPtrCheck < TH1 > (inputFileRefl , signalHistoName [iSliceVar ]);
318323 } else {
@@ -321,15 +326,13 @@ void runMassFitter(const std::string& configFileName)
321326 }
322327 }
323328 } else {
324- checkVectorSizeMcHistograms (signalHistoName , promptHistoName , fdHistoName );
325329 if (!signalHistoName .empty ()) {
326330 hMass [iSliceVar ] = getObjectWithNullPtrCheck < TH1 > (inputFile , signalHistoName [iSliceVar ]);
327331 } else {
328332 hMass [iSliceVar ] = getObjectWithNullPtrCheck < TH1 > (inputFile , promptHistoName [iSliceVar ]);
329333 hMass [iSliceVar ]-> Add (getObjectWithNullPtrCheck < TH1 > (inputFile , fdHistoName [iSliceVar ]));
330334 }
331335 if (includeSecPeak ) {
332- checkVectorSizeMcHistograms (signalSecPeakHistoName , promptSecPeakHistoName , fdSecPeakHistoName );
333336 if (!signalHistoName .empty ()) {
334337 hMass [iSliceVar ]-> Add (getObjectWithNullPtrCheck < TH1 > (inputFile , signalSecPeakHistoName [iSliceVar ]));
335338 } else {
0 commit comments