[PWGMM] New MM task computing MFT tracks histograms (#363)

Author: sarahherrmann

PWGMM/Tasks/CMakeLists.txt

@@ -19,6 +19,11 @@ o2physics_add_dpl_workflow(dndeta
                     PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore
                     COMPONENT_NAME Analysis)
 
+o2physics_add_dpl_workflow(analyse-mft-tracks
+                    SOURCES analyse-mft-tracks.cxx
+                    PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore
+                    COMPONENT_NAME Analysis)
+
 o2physics_add_dpl_workflow(ue-charged
                     SOURCES uecharged.cxx
                     PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore

PWGMM/Tasks/analyse-mft-tracks.cxx

@@ -0,0 +1,101 @@
+// 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/runDataProcessing.h"
+#include "Framework/AnalysisTask.h"
+
+#include "TDatabasePDG.h"
+#include "Common/Core/MC.h"
+
+using namespace o2;
+using namespace o2::framework;
+
+using Particles = aod::McParticles;
+
+//the task analyseMFTTracks loops over MFT tracks and generated particles and fills basic histograms
+
+struct analyseMFTTracks {
+  int icoll = 0;
+  Service<TDatabasePDG> pdg;
+  HistogramRegistry registry{
+    "registry",
+    {
+      {"TracksPhiEta_in_coll", "; #varphi; #eta; tracks", {HistType::kTH2F, {{600, -M_PI, M_PI}, {35, -4.5, -1.}}}}, //
+      {"TracksEtaZvtx", "; #eta; Z_{vtx}; tracks", {HistType::kTH2F, {{35, -4.5, -1.}, {201, -20.1, 20.1}}}},        //
+      {"NtrkZvtx", "; N_{trk}; Z_{vtx}; events", {HistType::kTH2F, {{301, -0.5, 300.5}, {201, -20.1, 20.1}}}},       //
+      {"NtrkEta", "#eta; N_{trk}; events", {HistType::kTH1F, {{35, -4.5, -1.}}}},                                    //
+      {"TracksPhiEtaGen", "; #varphi; #eta; tracks", {HistType::kTH2F, {{600, 0, 2 * M_PI}, {35, -4.5, -1.}}}},      //
+      {"TracksEtaZvtxGen", "; #eta; Z_{vtx}; tracks", {HistType::kTH2F, {{35, -4.5, -1.}, {201, -20.1, 20.1}}}},     //
+      //{"NtrkZvtxGen", "; N_{trk}; Z_{vtx}; events", {HistType::kTH2F, {{301, -0.5, 300.5}, {201, -20.1, 20.1}}}},            //
+      {"NtrkEtaGen", "#eta; N_{trk}; events", {HistType::kTH1F, {{35, -4.5, -1.}}}}, //
+    }                                                                                //
+  };
+
+  void processRec(o2::aod::Collision const& collision, o2::aod::MFTTracks const& tracks)
+  {
+
+    auto z = collision.posZ();
+    registry.fill(HIST("NtrkZvtx"), tracks.size(), z);
+
+    for (auto& track : tracks) {
+      registry.fill(HIST("TracksPhiEta_in_coll"), track.phi(), track.eta());
+      registry.fill(HIST("TracksEtaZvtx"), track.eta(), z);
+      registry.fill(HIST("NtrkEta"), track.eta());
+    }
+    icoll++;
+  }
+  //end of processRec
+  PROCESS_SWITCH(analyseMFTTracks, processRec, "Process rec level", true);
+
+  void processGen(aod::McCollisions::iterator const& mcCollision, Particles const& particles)
+  {
+
+    int nChargedPrimaryParticles = 0;
+    auto z = mcCollision.posZ();
+
+    for (auto& particle : particles) {
+      auto p = pdg->GetParticle(particle.pdgCode());
+      int charge = 0;
+      if (p == nullptr) {
+        // unknown particles will be skipped
+        if (particle.pdgCode() > 1000000000) {
+          //          auto x = (std::trunc(particle.pdgCode() / 10000) - 100000);
+          //          charge = x - std::trunc(x / 1000) * 1000;
+          LOGF(debug, "[{}] Nucleus with PDG code {}", particle.globalIndex(), particle.pdgCode() /*, charge*/); // (charge %d)
+        } else {
+          LOGF(debug, "[{}] Unknown particle with PDG code {}", particle.globalIndex(), particle.pdgCode());
+        }
+      } else {
+        charge = p->Charge();
+      }
+      if (charge != 0 && particle.isPhysicalPrimary()) {
+        registry.fill(HIST("TracksEtaZvtxGen"), particle.eta(), z);
+        registry.fill(HIST("TracksPhiEtaGen"), particle.phi(), particle.eta());
+
+        registry.fill(HIST("TracksPhiEtaGen"), particle.phi(), particle.eta());
+        registry.fill(HIST("NtrkEtaGen"), particle.eta());
+        nChargedPrimaryParticles++;
+      }
+    }
+
+    registry.fill(HIST("NtrkZvtxGen"), nChargedPrimaryParticles, mcCollision.posZ());
+  }
+
+  PROCESS_SWITCH(analyseMFTTracks, processGen, "Process gen level", true);
+};
+//end of the task analyseMFTTracks
+
+WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
+{
+  return WorkflowSpec{
+    adaptAnalysisTask<analyseMFTTracks>(cfgc),
+  };
+}