Add table for ECAL

Author: njacazio

ALICE3/DataModel/ECAL.h

@@ -0,0 +1,48 @@
+// 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.
+
+///
+/// \file   ECAL.h
+/// \author Nicolo' Jacazio
+/// \since  14/09/2021
+/// \brief  Set of tables for the ALICE3 ECAL information
+///
+
+#ifndef O2_ANALYSIS_ALICE3_ECAL_H_
+#define O2_ANALYSIS_ALICE3_ECAL_H_
+
+// O2 includes
+#include "Framework/AnalysisDataModel.h"
+
+namespace o2::aod
+{
+namespace alice3ecal
+{
+DECLARE_SOA_INDEX_COLUMN(Track, track);    //! Index to travel from track to ECAL
+DECLARE_SOA_COLUMN(Energy, energy, float); //! Signal in ECAL
+DECLARE_SOA_COLUMN(PosX, posX, float);     //! Error on the ECAL signal
+DECLARE_SOA_COLUMN(PosY, posY, float);     //! signal - exp. signal for electrons
+DECLARE_SOA_COLUMN(PosZ, posZ, float);     //! signal - exp. signal for muons
+} // namespace alice3ecal
+
+DECLARE_SOA_TABLE(ECALs, "AOD", "A3ECAL", //! Table for the ALICE3 ECAL detector
+                  o2::soa::Index<>,
+                  alice3ecal::TrackId,
+                  alice3ecal::Energy,
+                  alice3ecal::PosX,
+                  alice3ecal::PosY,
+                  alice3ecal::PosZ);
+
+using ECAL = ECALs::iterator;
+
+} // namespace o2::aod
+
+#endif // O2_ANALYSIS_ALICE3_ECAL_H_

ALICE3/Tasks/CMakeLists.txt

@@ -29,6 +29,11 @@ o2physics_add_dpl_workflow(alice3-pid-rich-qa
                     PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
                     COMPONENT_NAME Analysis)
 
+o2physics_add_dpl_workflow(alice3-ecal-qa
+                    SOURCES ECALqa.cxx
+                    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
+                    COMPONENT_NAME Analysis)
+
 o2physics_add_dpl_workflow(alice3-pid-ftof-qa
                     SOURCES pidFTOFqa.cxx
                     PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore

ALICE3/Tasks/ECALqa.cxx

@@ -0,0 +1,80 @@
+// 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.
+
+///
+/// \file   ECALqa.cxx
+/// \author Nicolo' Jacazio
+/// \since  14/09/2021
+/// \brief  Task to use the ALICE3 ECAL table
+///
+
+// O2 includes
+#include "Framework/AnalysisTask.h"
+#include "ALICE3/DataModel/ECAL.h"
+#include "Common/Core/MC.h"
+#include "Common/Core/PID/PIDResponse.h"
+#include "ReconstructionDataFormats/PID.h"
+#include "Framework/HistogramRegistry.h"
+#include "Framework/runDataProcessing.h"
+
+using namespace o2;
+using namespace o2::track;
+using namespace o2::framework;
+using namespace o2::framework::expressions;
+
+namespace o2::aod
+{
+
+namespace indices
+{
+DECLARE_SOA_INDEX_COLUMN(Track, track);
+DECLARE_SOA_INDEX_COLUMN(ECAL, ecal);
+} // namespace indices
+
+DECLARE_SOA_INDEX_TABLE_USER(ECALTracksIndex, Tracks, "ECALTRK", indices::TrackId, indices::ECALId);
+} // namespace o2::aod
+
+struct ecalIndexBuilder { // Builder of the ECAL-track index linkage
+  Builds<o2::aod::ECALTracksIndex> ind;
+  void init(o2::framework::InitContext&)
+  {
+  }
+};
+
+struct ecalQaMc {
+  HistogramRegistry histos{"Histos", {}, OutputObjHandlingPolicy::QAObject};
+
+  void init(o2::framework::InitContext&)
+  {
+    histos.add("energy", ";Energy;Entries", HistType::kTH1F, {{1000, 0, 10000}});
+  }
+
+  using Trks = soa::Join<aod::Tracks, aod::ECALTracksIndex, aod::TracksExtra>;
+  void process(const aod::McParticles& mcParticles,
+               const Trks& tracks,
+               const aod::McTrackLabels& labels,
+               const aod::ECALs&,
+               const aod::Collisions& colls)
+  {
+    for (auto& track : tracks) {
+      if (!track.has_ecal())
+        continue;
+      histos.fill(HIST("energy"), track.ecal().energy());
+    }
+  }
+};
+
+WorkflowSpec defineDataProcessing(ConfigContext const& cfg)
+{
+  auto workflow = WorkflowSpec{adaptAnalysisTask<ecalIndexBuilder>(cfg)};
+  workflow.push_back(adaptAnalysisTask<ecalQaMc>(cfg));
+  return workflow;
+}