[PWGJE] Add photon-hadron correlation analysis task (#11433)

Author: jkinner2

PWGJE/DataModel/PhotonChargedTriggerCorrelation.h

@@ -0,0 +1,132 @@
+// 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.
+
+///
+/// \brief table definitions for photon-hadron correlation analyses
+///
+/// \author Julius Kinner
+/// \file PhotonChargedTriggerCorrelation.h
+
+#ifndef PWGJE_DATAMODEL_PHOTONCHARGEDTRIGGERCORRELATION_H_
+#define PWGJE_DATAMODEL_PHOTONCHARGEDTRIGGERCORRELATION_H_
+
+#include "Framework/AnalysisDataModel.h"
+#include "PWGJE/Core/JetDerivedDataUtilities.h"
+#include "PWGJE/DataModel/Jet.h"
+#include "PWGEM/PhotonMeson/DataModel/gammaTables.h"
+#include "PWGEM/PhotonMeson/Utils/PCMUtilities.h"
+
+namespace o2::aod
+{
+
+// basic correlation particle columns
+namespace corr_particle
+{
+DECLARE_SOA_INDEX_COLUMN_FULL(JetCollision, jetCollision, int, JCollisions, "");
+DECLARE_SOA_INDEX_COLUMN_FULL(JetMcCollision, jetMcCollision, int, JMcCollisions, "");
+DECLARE_SOA_COLUMN(Pt, pt, float);
+DECLARE_SOA_COLUMN(Phi, phi, float);
+DECLARE_SOA_COLUMN(Eta, eta, float);
+} // namespace corr_particle
+
+// reco
+
+// collision extension
+namespace collision_extra_corr
+{
+DECLARE_SOA_COLUMN(SelEv, selEv, bool);
+DECLARE_SOA_COLUMN(TrigEv, trigEv, bool);
+} // namespace collision_extra_corr
+DECLARE_SOA_TABLE(CollisionsExtraCorr, "AOD", "COLLISIONSEXTRACORR",
+                  collision_extra_corr::SelEv, collision_extra_corr::TrigEv);
+
+// trigger
+namespace trigger
+{
+DECLARE_SOA_INDEX_COLUMN_FULL(JetTrack, jetTrack, int, JetTracks, "");
+} // namespace trigger
+DECLARE_SOA_TABLE(Triggers, "AOD", "TRIGGERS",
+                  o2::soa::Index<>, corr_particle::JetCollisionId, trigger::JetTrackId,
+                  corr_particle::Pt, corr_particle::Phi, corr_particle::Eta);
+using Trigger = Triggers::iterator;
+
+// hadrons (global tracks)
+namespace hadron
+{
+DECLARE_SOA_INDEX_COLUMN_FULL(JetTrack, jetTrack, int, JetTracks, "");
+} // namespace hadron
+DECLARE_SOA_TABLE(Hadrons, "AOD", "HADRONS",
+                  o2::soa::Index<>, corr_particle::JetCollisionId, hadron::JetTrackId,
+                  corr_particle::Pt, corr_particle::Phi, corr_particle::Eta);
+using Hadron = Hadrons::iterator;
+
+// pipm
+namespace pipm
+{
+DECLARE_SOA_INDEX_COLUMN_FULL(JetTrack, jetTrack, int, JetTracks, "");
+} // namespace pipm
+DECLARE_SOA_TABLE(Pipms, "AOD", "PIPMS",
+                  o2::soa::Index<>, corr_particle::JetCollisionId, pipm::JetTrackId,
+                  corr_particle::Pt, corr_particle::Phi, corr_particle::Eta);
+using Pipm = Pipms::iterator;
+
+// photonPCM
+namespace photon_pcm
+{
+DECLARE_SOA_INDEX_COLUMN_FULL(V0PhotonKF, v0PhotonKF, int, V0PhotonsKF, "");
+DECLARE_SOA_COLUMN(PosTrackId, posTrackId, int);
+DECLARE_SOA_COLUMN(NegTrackId, negTrackId, int);
+} // namespace photon_pcm
+DECLARE_SOA_TABLE(PhotonPCMs, "AOD", "PHOTONPCMS",
+                  o2::soa::Index<>, corr_particle::JetCollisionId, photon_pcm::V0PhotonKFId,
+                  photon_pcm::PosTrackId, photon_pcm::NegTrackId,
+                  corr_particle::Pt, corr_particle::Phi, corr_particle::Eta);
+using PhotonPCM = PhotonPCMs::iterator;
+
+// photonPCM pairs (pi0)
+namespace photon_pcm_pair
+{
+DECLARE_SOA_INDEX_COLUMN_FULL(V0PhotonKF1, v0PhotonKF1, int, V0PhotonsKF, "_1");
+DECLARE_SOA_INDEX_COLUMN_FULL(V0PhotonKF2, v0PhotonKF2, int, V0PhotonsKF, "_2");
+DECLARE_SOA_COLUMN(PosTrack1Id, posTrack1Id, int);
+DECLARE_SOA_COLUMN(NegTrack1Id, negTrack1Id, int);
+DECLARE_SOA_COLUMN(PosTrack2Id, posTrack2Id, int);
+DECLARE_SOA_COLUMN(NegTrack2Id, negTrack2Id, int);
+DECLARE_SOA_COLUMN(Mgg, mgg, float);
+} // namespace photon_pcm_pair
+DECLARE_SOA_TABLE(PhotonPCMPairs, "AOD", "PHOTONPCMPAIRS",
+                  o2::soa::Index<>, corr_particle::JetCollisionId, photon_pcm_pair::V0PhotonKF1Id, photon_pcm_pair::V0PhotonKF2Id,
+                  photon_pcm_pair::PosTrack1Id, photon_pcm_pair::NegTrack1Id, photon_pcm_pair::PosTrack2Id, photon_pcm_pair::NegTrack2Id,
+                  corr_particle::Pt, corr_particle::Phi, corr_particle::Eta, photon_pcm_pair::Mgg);
+using PhotonPCMPair = PhotonPCMPairs::iterator;
+
+// mc
+
+// mcCollision extension
+namespace mc_collision_extra_corr
+{
+DECLARE_SOA_COLUMN(TrigEv, trigEv, bool);
+} // namespace mc_collision_extra_corr
+DECLARE_SOA_TABLE(McCollisionsExtraCorr, "AOD", "MCCOLLISIONSEXTRACORR",
+                  mc_collision_extra_corr::TrigEv);
+
+// trigger
+namespace trigger_particle
+{
+DECLARE_SOA_INDEX_COLUMN_FULL(JetMcParticle, jetMcParticle, int, JetParticles, "");
+} // namespace trigger_particle
+DECLARE_SOA_TABLE(TriggerParticles, "AOD", "TRIGGERPARTICLES",
+                  o2::soa::Index<>, corr_particle::JetMcCollisionId, trigger_particle::JetMcParticleId,
+                  corr_particle::Pt, corr_particle::Phi, corr_particle::Eta);
+using TriggerParticle = TriggerParticles::iterator;
+} // namespace o2::aod
+
+#endif // PWGJE_DATAMODEL_PHOTONCHARGEDTRIGGERCORRELATION_H_

PWGJE/Tasks/CMakeLists.txt

@@ -50,6 +50,10 @@ o2physics_add_dpl_workflow(photon-isolation-qa
                     SOURCES photonIsolationQA.cxx
                     PUBLIC_LINK_LIBRARIES O2::Framework O2::EMCALBase O2::EMCALCalib O2Physics::AnalysisCore
                     COMPONENT_NAME Analysis)
+o2physics_add_dpl_workflow(photon-charged-trigger-correlation
+                    SOURCES photonChargedTriggerCorrelation.cxx
+                    PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore
+                    COMPONENT_NAME Analysis)
 
 if(FastJet_FOUND)
     o2physics_add_dpl_workflow(jet-background-analysis