From 05418e3c5bd85e753e188eedcf0aafedfad61611 Mon Sep 17 00:00:00 2001
From: Rohaan Deb <67634136+RD0407@users.noreply.github.com>
Date: Tue, 10 Dec 2024 19:10:54 +0100
Subject: [PATCH] Update spectraTOF.cxx

---
 PWGLF/Tasks/Nuspex/spectraTOF.cxx | 54 +++++++++++++++----------------
 1 file changed, 27 insertions(+), 27 deletions(-)

diff --git a/PWGLF/Tasks/Nuspex/spectraTOF.cxx b/PWGLF/Tasks/Nuspex/spectraTOF.cxx
index 35c36100e46..ad94a217cdf 100644
--- a/PWGLF/Tasks/Nuspex/spectraTOF.cxx
+++ b/PWGLF/Tasks/Nuspex/spectraTOF.cxx
@@ -1329,7 +1329,7 @@ struct tofSpectra {
   using TrackCandidates = soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA,
                                     aod::pidEvTimeFlags, aod::TrackSelection, aod::TOFSignal>;
 
-    void processOccupancy(CollisionCandidates::iterator const& collision,
+      void processOccupancy(CollisionCandidates::iterator const& collision,
                         soa::Join<TrackCandidates,
                                   aod::pidTPCFullPi, aod::pidTPCFullKa, aod::pidTPCFullPr,
                                   aod::pidTOFFullPi, aod::pidTOFFullKa, aod::pidTOFFullPr> const& tracks)
@@ -1341,11 +1341,11 @@ struct tofSpectra {
     const float multiplicity = getMultiplicity(collision);
     histos.fill(HIST("nsigmatpc/test_occupancy/Mult_vs_Occupancy"), multiplicity, occupancy);
     for (const auto& track : tracks) {
-      if (!isTrackSelected<true>(track, collision)) {
+      if(!isTrackSelected<true>(track, collision)) {
         continue;
       }
       const auto& nsigmaTPCPi = o2::aod::pidutils::tpcNSigma<2>(track);
-      const auto& nsigmaTPCKa = o2::aod::pidutils::tpcNSigma<3>(track);
+      const auto& nsigmaTPCKa = track.tpcNSigmaKa();//o2::aod::pidutils::tpcNSigma<3>(track);
       const auto& nsigmaTPCPr = o2::aod::pidutils::tpcNSigma<4>(track);
       const auto& nsigmaTOFPi = o2::aod::pidutils::tofNSigma<2>(track);
       const auto& nsigmaTOFKa = o2::aod::pidutils::tofNSigma<3>(track);
@@ -1357,48 +1357,48 @@ struct tofSpectra {
         bool isTOFPion = fabs(nsigmaTOFPi) < 10;
         bool isTOFKaon = fabs(nsigmaTOFKa) < 10;
         bool isTOFProton = fabs(nsigmaTOFPr) < 10;
-        if (isTPCPion) {
-        if(std::abs(track.rapidity(PID::getMass(2)) >= trkselOptions.cfgCutY)) continue;
-            if (track.sign() > 0) {
+        if(isTPCPion) {
+        if(std::abs(track.rapidity(PID::getMass(2)) >= trkselOptions.cfgCutY)){continue;}
+            if(track.sign() > 0) {
                 histos.fill(HIST("nsigmatpc/test_occupancy/pos/pi"), track.pt(), nsigmaTPCPi, multiplicity, occupancy);
             } else {
                 histos.fill(HIST("nsigmatpc/test_occupancy/neg/pi"), track.pt(), nsigmaTPCPi, multiplicity, occupancy);
             }
-        } else if (isTPCKaon) {
-        if(std::abs(track.rapidity(PID::getMass(3)) >= trkselOptions.cfgCutY)) continue;
-            if (track.sign() > 0) {
+        }else if(isTPCKaon) {
+        if(std::abs(track.rapidity(PID::getMass(3)) >= trkselOptions.cfgCutY)){continue;}
+            if(track.sign() > 0) {
                 histos.fill(HIST("nsigmatpc/test_occupancy/pos/ka"), track.pt(), nsigmaTPCKa, multiplicity, occupancy);
-            } else {
+            }else{
                 histos.fill(HIST("nsigmatpc/test_occupancy/neg/ka"), track.pt(), nsigmaTPCKa, multiplicity, occupancy);
             }
-        }  else if (isTPCProton) {
-        if(std::abs(track.rapidity(PID::getMass(4)) >=trkselOptions.cfgCutY)) continue;
-            if (track.sign() > 0) {
+        }else if(isTPCProton) {
+        if(std::abs(track.rapidity(PID::getMass(4)) >=trkselOptions.cfgCutY)){continue;}
+            if(track.sign() > 0) {
                 histos.fill(HIST("nsigmatpc/test_occupancy/pos/pr"), track.pt(), nsigmaTPCPr, multiplicity, occupancy);
-            } else {
+            }else{
                 histos.fill(HIST("nsigmatpc/test_occupancy/neg/pr"), track.pt(), nsigmaTPCPr, multiplicity, occupancy);
             }
           }
-      if (track.hasTOF()) {
-      if (isTOFPion) {
-        if(std::abs(track.rapidity(PID::getMass(2)) >= trkselOptions.cfgCutY)) continue;
-            if (track.sign() > 0) {
+      if(track.hasTOF()) {
+      if(isTOFPion) {
+        if(std::abs(track.rapidity(PID::getMass(2)) >= trkselOptions.cfgCutY)){continue;}
+            if(track.sign() > 0) {
                 histos.fill(HIST("nsigmatof/test_occupancy/pos/pi"), track.pt(), nsigmaTOFPi, multiplicity, occupancy);
-            } else {
+            }else {
                 histos.fill(HIST("nsigmatof/test_occupancy/neg/pi"), track.pt(), nsigmaTOFPi, multiplicity, occupancy);
             }
-        } else if (isTOFKaon) {
-        if(std::abs(track.rapidity(PID::getMass(3)) >= trkselOptions.cfgCutY)) continue;
-            if (track.sign() > 0) {
+        }else if (isTOFKaon) {
+        if(std::abs(track.rapidity(PID::getMass(3)) >= trkselOptions.cfgCutY)){continue;}
+            if(track.sign() > 0) {
                 histos.fill(HIST("nsigmatof/test_occupancy/pos/ka"), track.pt(), nsigmaTOFKa, multiplicity, occupancy);
-            } else {
+            }else {
                 histos.fill(HIST("nsigmatof/test_occupancy/neg/ka"), track.pt(), nsigmaTOFKa, multiplicity, occupancy);
             }
-        }  else if (isTOFProton) {
-        if(std::abs(track.rapidity(PID::getMass(4)) >= trkselOptions.cfgCutY)) continue;
-            if (track.sign() > 0) {
+        }else if (isTOFProton) {
+        if(std::abs(track.rapidity(PID::getMass(4)) >= trkselOptions.cfgCutY)){continue;}
+            if(track.sign() > 0) {
                 histos.fill(HIST("nsigmatof/test_occupancy/pos/pr"), track.pt(), nsigmaTOFPr, multiplicity, occupancy);
-            } else {
+            }else {
                 histos.fill(HIST("nsigmatof/test_occupancy/neg/pr"), track.pt(), nsigmaTOFPr, multiplicity, occupancy);
             }
           }