Fix for MCTruth label checks

Author: nburmaso

Detectors/AOD/src/AODProducerWorkflowSpec.cxx

@@ -534,16 +534,21 @@ void AODProducerWorkflowDPL::run(ProcessingContext& pc)
   //   bit 13 -- ITS and TPC labels are not equal
   //   bit 14 -- isNoise() == true
   //   bit 15 -- isFake() == true
+  // labelID = std::numeric_limits<uint32_t>::max() -- label is not set
+
+  uint32_t labelID;
+  uint32_t labelITS;
+  uint32_t labelTPC;
+  uint16_t labelMask;
 
   if (mFillTracksITS) {
     fillTracksTable(tracksITS, vCollRefsITS, tracksCursor, o2::vertexing::GIndex::Source::ITS); // fTrackType = 1
     for (auto& mcTruthITS : tracksITSMCTruth) {
-      uint32_t labelID = std::numeric_limits<uint32_t>::max();
+      labelID = std::numeric_limits<uint32_t>::max();
       // TODO:
       // fill label mask
-      uint16_t labelMask = 0;
-      int nEventsITS = mcReader.getNEvents(mcTruthITS.getSourceID());
-      if (mcTruthITS.getEventID() < mcReader.getNEvents(mcTruthITS.getSourceID())) {
+      labelMask = 0;
+      if (mcTruthITS.isValid()) {
         labelID = mIDsToIndex.at(std::make_tuple(mcTruthITS.getSourceID(), mcTruthITS.getEventID(), mcTruthITS.getTrackID()));
       }
       if (mcTruthITS.isFake()) {
@@ -561,12 +566,11 @@ void AODProducerWorkflowDPL::run(ProcessingContext& pc)
   if (mFillTracksTPC) {
     fillTracksTable(tracksTPC, vCollRefsTPC, tracksCursor, o2::vertexing::GIndex::Source::TPC); // fTrackType = 2
     for (auto& mcTruthTPC : tracksTPCMCTruth) {
-      uint32_t labelID = std::numeric_limits<uint32_t>::max();
+      labelID = std::numeric_limits<uint32_t>::max();
       // TODO:
       // fill label mask
-      uint16_t labelMask = 0;
-      int nEventsTPC = mcReader.getNEvents(mcTruthTPC.getSourceID());
-      if (mcTruthTPC.getEventID() < nEventsTPC) {
+      labelMask = 0;
+      if (mcTruthTPC.isValid()) {
         labelID = mIDsToIndex.at(std::make_tuple(mcTruthTPC.getSourceID(), mcTruthTPC.getEventID(), mcTruthTPC.getTrackID()));
       }
       if (mcTruthTPC.isFake()) {
@@ -586,16 +590,14 @@ void AODProducerWorkflowDPL::run(ProcessingContext& pc)
     for (int i = 0; i < tracksITSTPC.size(); i++) {
       auto& mcTruthITS = tracksITSTPC_ITSMC[i];
       auto& mcTruthTPC = tracksITSTPC_TPCMC[i];
-      uint32_t labelID = std::numeric_limits<uint32_t>::max();
-      uint32_t labelITS = std::numeric_limits<uint32_t>::max();
-      uint32_t labelTPC = std::numeric_limits<uint32_t>::max();
+      labelID = std::numeric_limits<uint32_t>::max();
+      labelITS = std::numeric_limits<uint32_t>::max();
+      labelTPC = std::numeric_limits<uint32_t>::max();
       // TODO:
       // fill label mask
       // currently using label mask to indicate labelITS != labelTPC
-      uint16_t labelMask = 0;
-      int nEventsITS = mcReader.getNEvents(mcTruthITS.getSourceID());
-      int nEventsTPC = mcReader.getNEvents(mcTruthTPC.getSourceID());
-      if (mcTruthITS.getEventID() < nEventsITS && mcTruthTPC.getEventID() < nEventsTPC) {
+      labelMask = 0;
+      if (mcTruthITS.isValid() && mcTruthTPC.isValid()) {
         labelITS = mIDsToIndex.at(std::make_tuple(mcTruthITS.getSourceID(), mcTruthITS.getEventID(), mcTruthITS.getTrackID()));
         labelTPC = mIDsToIndex.at(std::make_tuple(mcTruthTPC.getSourceID(), mcTruthTPC.getEventID(), mcTruthTPC.getTrackID()));
         labelID = labelITS;