Add simple task for strangeness tracking (AliceO2Group#2426)

Author: qgp

PWGLF/Tasks/CMakeLists.txt

@@ -154,4 +154,10 @@ o2physics_add_dpl_workflow(cascpostprocessing
 o2physics_add_dpl_workflow(hstrangecorrelation
                     SOURCES hStrangeCorrelation.cxx
                     PUBLIC_LINK_LIBRARIES O2::Framework O2::DetectorsBase O2Physics::AnalysisCore
-                    COMPONENT_NAME Analysis)
+                    COMPONENT_NAME Analysis)
+
+o2physics_add_dpl_workflow(strangenesstracking
+                    SOURCES strangenesstracking.cxx
+                    PUBLIC_LINK_LIBRARIES O2::Framework O2::ReconstructionDataFormats O2Physics::AnalysisCore
+                    COMPONENT_NAME Analysis)
+

PWGLF/Tasks/strangenesstracking.cxx

@@ -0,0 +1,99 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+
+#include "Framework/AnalysisTask.h"
+#include "Framework/AnalysisDataModel.h"
+#include "Framework/ASoA.h"
+#include "Framework/runDataProcessing.h"
+#include "Common/DataModel/TrackSelectionTables.h"
+#include "Common/Core/trackUtilities.h"
+#include "ReconstructionDataFormats/DCA.h"
+
+using namespace o2;
+using namespace o2::framework;
+using namespace o2::framework::expressions;
+
+struct StrangenessTrackingTask {
+  using TracksExt = soa::Join<aod::TracksIU, aod::TracksCovIU, aod::TracksExtra, aod::McTrackLabels>;
+
+  Configurable<double> bz{"bz", 50., "magnetic field"};
+
+  OutputObj<TH1F> hDCA{"h_DCA"};
+
+  float Square(float x) { return x * x; }
+
+  void init(InitContext const&)
+  {
+    hDCA.setObject(new TH1F("h_dca", "DCA;DCA (cm)", 200, -2., 2.));
+  }
+
+  void processTrackedCascades(aod::Collision const& collision,
+                              aod::TrackedCascades const& trackedCascades, aod::Cascades const& cascades,
+                              aod::V0s const& v0s, TracksExt const& tracks, aod::McParticles const& mcParticles)
+  {
+    for (const auto& trackedCascade : trackedCascades) {
+      auto trackCovTrk = getTrackParCov(trackedCascade.track_as<TracksExt>());
+      auto primaryVertex = getPrimaryVertex(collision);
+      // auto covMatrixPV = primaryVertex.getCov();
+      o2::dataformats::DCA impactParameterTrk;
+      trackCovTrk.propagateToDCA(primaryVertex, bz, &impactParameterTrk);
+
+      hDCA->Fill(impactParameterTrk.getY());
+
+      const auto& casc = trackedCascade.cascade();
+      const auto& bachelor = casc.bachelor_as<TracksExt>();
+      const auto& v0 = casc.v0();
+      const auto& ptrack = v0.posTrack_as<TracksExt>();
+      const auto& ntrack = v0.negTrack_as<TracksExt>();
+
+      if (ptrack.mcParticle().has_mothers() && ntrack.mcParticle().has_mothers() &&
+          ptrack.mcParticle().mothersIds()[0] == ntrack.mcParticle().mothersIds()[0]) {
+        const auto v0part = ptrack.mcParticle().mothers_as<aod::McParticles>()[0];
+        if (v0part.has_mothers() && bachelor.mcParticle().has_mothers() &&
+            v0part.mothersIds()[0] == bachelor.mcParticle().mothersIds()[0])
+          LOG(debug) << "cascade with PDG code: " << v0part.mothers_as<aod::McParticles>()[0].pdgCode();
+      }
+    }
+  }
+  PROCESS_SWITCH(StrangenessTrackingTask, processTrackedCascades, "process tracked cascades", true);
+
+  void processTrackedV0s(aod::Collision const& collision,
+                         aod::TrackedV0s const& trackedV0s, aod::V0s const& v0s,
+                         TracksExt const& tracks, aod::McParticles const& mcParticles)
+  {
+    for (const auto& trackedV0 : trackedV0s) {
+      const auto& v0 = trackedV0.v0();
+      v0.posTrack();
+      v0.negTrack();
+    }
+  }
+  PROCESS_SWITCH(StrangenessTrackingTask, processTrackedV0s, "process tracked V0s", true);
+
+  void processTracked3Bodys(aod::Collision const& collision,
+                            aod::Tracked3Bodys const& tracked3Bodys, aod::Decay3Bodys const& decay3Bodys,
+                            TracksExt const& tracks, aod::McParticles const& mcParticles)
+  {
+    for (const auto& tracked3Body : tracked3Bodys) {
+      tracked3Body.itsTrack();
+      const auto& decay3Body = tracked3Body.decay3Body();
+      decay3Body.track0();
+      decay3Body.track1();
+      decay3Body.track2();
+    }
+  }
+  PROCESS_SWITCH(StrangenessTrackingTask, processTracked3Bodys, "process tracked 3 body decays", true);
+};
+
+WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
+{
+  return WorkflowSpec{
+    adaptAnalysisTask<StrangenessTrackingTask>(cfgc)};
+}