Adding cut for removable realigned muons in table-maker-with-mc

Author: ppoava

PWGDQ/TableProducer/tableMakerMC_withAssoc.cxx

@@ -937,17 +937,24 @@ struct TableMakerMC {
         if constexpr ((static_cast<bool>(TMuonRealignFillMap)) && static_cast<int>(muon.trackType()) > 2) {
           // refill kinematic info and recalculate propagation in case of using realigned muons
           auto muonRealignSelected = muonsRealign.sliceBy(fwdtrackRealignPerMuon, assoc.fwdtrackId());
+          int realignRemoveFlag = 0;
           if (muonRealignSelected.size() == 1) {
             for (const auto& muonRealign : muonRealignSelected) {
               // refill muon information with realigned tracks
               VarManager::FillTrack<TMuonRealignFillMap>(muonRealign);
+              realignRemoveFlag = muonRealign.isRemovable();
 
               if (fConfigVariousOptions.fPropMuon) {
                 VarManager::FillPropagateMuon<TMuonRealignFillMap>(muonRealign, collision);
               }
 
               VarManager::FillTrackCollision<TMuonRealignFillMap>(muonRealign, collision);
             }
+
+            if (realignRemoveFlag) {
+              continue;
+            }
+
           } else {
             LOGF(fatal, "Inconsistent size of realigned muon track candidates.");
           }
@@ -1068,17 +1075,24 @@ struct TableMakerMC {
         if constexpr ((static_cast<bool>(TMuonRealignFillMap)) && static_cast<int>(muon.trackType()) > 2) {
           // refill kinematic info and recalculate propagation in case of using realigned muons
           auto muonRealignSelected = muonsRealign.sliceBy(fwdtrackRealignPerMuon, muon.globalIndex());
+          int realignRemoveFlag = 0;
           if (muonRealignSelected.size() == 1) {
             for (const auto& muonRealign : muonRealignSelected) {
               // refill muon information with realigned tracks
               VarManager::FillTrack<TMuonRealignFillMap>(muonRealign);
+              realignRemoveFlag = muonRealign.isRemovable();
 
               if (fConfigVariousOptions.fPropMuon) {
                 VarManager::FillPropagateMuon<TMuonRealignFillMap>(muonRealign, collision);
               }
 
               VarManager::FillTrackCollision<TMuonRealignFillMap>(muonRealign, collision);
             }
+
+            if (realignRemoveFlag) {
+              continue;
+            }
+
           } else {
             LOGF(fatal, "Inconsistent size of realigned muon track candidates.");
           }