@@ -544,8 +544,8 @@ class pidTPCModule
544544
545545 // faster counting
546546 for (const auto & track : tracks) {
547- if (track.hasTPC ()) {
548- if (track.collisionId () > -1 ) {
547+ if (track.hasTPC ()){
548+ if (track.collisionId ()> -1 ){
549549 pidmults[track.collisionId ()]++;
550550 }
551551 totalTPCtracks++;
@@ -725,8 +725,12 @@ class pidTPCModule
725725 products.tableTuneOnData (mcTunedTPCSignal);
726726 }
727727
728- auto makePidTablesDefault = [&trk, &tpcSignalToEvaluatePID, &cols, &pidmults, &network_prediction, &count_tracks, &tracksForNet_size, this ](const int flagFull, auto & tableFull, const int flagTiny, auto & tableTiny, const o2::track::PID::ID pid) {
729- makePidTables (flagFull, tableFull, flagTiny, tableTiny, pid, tpcSignalToEvaluatePID, trk, cols, pidmults[trk.collisionId ()], network_prediction, count_tracks, tracksForNet_size);
728+ int multTPCtoUse = 1 ;
729+ if (trk.has_collision ()){
730+ multTPCtoUse = pidmults[trk.collisionId ()];
731+ }
732+ auto makePidTablesDefault = [&trk, &tpcSignalToEvaluatePID, &cols, &multTPCtoUse, &network_prediction, &count_tracks, &tracksForNet_size, this ](const int flagFull, auto & tableFull, const int flagTiny, auto & tableTiny, const o2::track::PID::ID pid) {
733+ makePidTables (flagFull, tableFull, flagTiny, tableTiny, pid, tpcSignalToEvaluatePID, trk, cols, multTPCtoUse, network_prediction, count_tracks, tracksForNet_size);
730734 };
731735
732736 makePidTablesDefault (pidTPCopts.pidFullEl , products.tablePIDFullEl , pidTPCopts.pidTinyEl , products.tablePIDTinyEl , o2::track::PID::Electron);
0 commit comments