|
9 | 9 | // granted to it by virtue of its status as an Intergovernmental Organization |
10 | 10 | // or submit itself to any jurisdiction. |
11 | 11 |
|
12 | | -#include <vector> |
13 | | -#include <TMath.h> |
| 12 | +#include "PWGEM/Dilepton/DataModel/dileptonTables.h" |
14 | 13 |
|
15 | 14 | #include "Common/Core/RecoDecay.h" |
16 | 15 | #include "Common/DataModel/CaloClusters.h" |
| 16 | +#include "Common/DataModel/PIDResponseTOF.h" |
| 17 | +#include "Common/DataModel/PIDResponseTPC.h" |
| 18 | +#include "Common/DataModel/TrackSelectionTables.h" |
17 | 19 |
|
18 | | -#include "PWGEM/Dilepton/DataModel/dileptonTables.h" |
| 20 | +#include <Framework/ASoA.h> |
| 21 | +#include <Framework/AnalysisDataModel.h> |
19 | 22 |
|
20 | | -#include "PWGJE/DataModel/EMCALClusters.h" |
| 23 | +#include <array> |
| 24 | +#include <cmath> |
| 25 | +#include <cstdint> |
| 26 | +#include <vector> |
21 | 27 |
|
22 | 28 | #ifndef PWGEM_PHOTONMESON_DATAMODEL_GAMMATABLES_H_ |
23 | 29 | #define PWGEM_PHOTONMESON_DATAMODEL_GAMMATABLES_H_ |
@@ -462,16 +468,16 @@ DECLARE_SOA_COLUMN(Time, time, float); |
462 | 468 | DECLARE_SOA_COLUMN(IsExotic, isExotic, bool); //! flag to mark cluster as exotic |
463 | 469 | DECLARE_SOA_COLUMN(Definition, definition, int); //! cluster definition, see EMCALClusterDefinition.h |
464 | 470 | DECLARE_SOA_ARRAY_INDEX_COLUMN(Track, track); //! TrackIds |
465 | | -DECLARE_SOA_COLUMN(TrackEta, tracketa, std::vector<float>); //! eta values of the matched tracks |
466 | | -DECLARE_SOA_COLUMN(TrackPhi, trackphi, std::vector<float>); //! phi values of the matched tracks |
| 471 | +DECLARE_SOA_COLUMN(DeltaPhi, deltaPhi, std::vector<float>); //! phi values of the matched tracks |
| 472 | +DECLARE_SOA_COLUMN(DeltaEta, deltaEta, std::vector<float>); //! eta values of the matched tracks |
467 | 473 | DECLARE_SOA_COLUMN(TrackP, trackp, std::vector<float>); //! momentum values of the matched tracks |
468 | 474 | DECLARE_SOA_COLUMN(TrackPt, trackpt, std::vector<float>); //! pt values of the matched tracks |
469 | 475 | DECLARE_SOA_DYNAMIC_COLUMN(Pt, pt, [](float e, float eta, float m = 0) -> float { return sqrt(e * e - m * m) / cosh(eta); }); //! cluster pt, mass to be given as argument when getter is called! |
470 | 476 | } // namespace emccluster |
471 | | -DECLARE_SOA_TABLE(SkimEMCClusters, "AOD", "SKIMEMCCLUSTERS", //! table of skimmed EMCal clusters |
| 477 | +DECLARE_SOA_TABLE(SkimEMCClusters, "AOD", "SKIMEMCCLUSTER", //! table of skimmed EMCal clusters |
472 | 478 | o2::soa::Index<>, skimmedcluster::CollisionId, emccluster::Definition, skimmedcluster::E, skimmedcluster::Eta, skimmedcluster::Phi, |
473 | | - skimmedcluster::M02, skimmedcluster::NCells, skimmedcluster::Time, emccluster::IsExotic, emccluster::TrackEta, |
474 | | - emccluster::TrackPhi, emccluster::TrackP, emccluster::TrackPt, emccluster::Pt<skimmedcluster::E, skimmedcluster::Eta>); |
| 479 | + skimmedcluster::M02, skimmedcluster::NCells, skimmedcluster::Time, emccluster::IsExotic, emccluster::DeltaPhi, |
| 480 | + emccluster::DeltaEta, emccluster::TrackP, emccluster::TrackPt, emccluster::Pt<skimmedcluster::E, skimmedcluster::Eta>); |
475 | 481 | using SkimEMCCluster = SkimEMCClusters::iterator; |
476 | 482 |
|
477 | 483 | DECLARE_SOA_TABLE(EMCEMEventIds, "AOD", "EMCEMEVENTID", emccluster::EMEventId); // To be joined with SkimEMCClusters table at analysis level. |
@@ -523,42 +529,11 @@ namespace caloextra |
523 | 529 | { |
524 | 530 | DECLARE_SOA_INDEX_COLUMN_FULL(Cluster, cluster, int, SkimEMCClusters, ""); //! reference to the gamma in the skimmed EMCal table |
525 | 531 | DECLARE_SOA_INDEX_COLUMN_FULL(Cell, cell, int, Calos, ""); //! reference to the gamma in the skimmed EMCal table |
526 | | -// DECLARE_SOA_INDEX_COLUMN(Track, track); //! TrackID |
527 | | -DECLARE_SOA_COLUMN(TrackEta, tracketa, float); //! eta of the matched track |
528 | | -DECLARE_SOA_COLUMN(TrackPhi, trackphi, float); //! phi of the matched track |
529 | | -DECLARE_SOA_COLUMN(TrackP, trackp, float); //! momentum of the matched track |
530 | | -DECLARE_SOA_COLUMN(TrackPt, trackpt, float); //! pt of the matched track |
531 | 532 | } // namespace caloextra |
532 | 533 |
|
533 | 534 | DECLARE_SOA_TABLE(SkimEMCCells, "AOD", "SKIMEMCCELLS", //! table of link between skimmed EMCal clusters and their cells |
534 | 535 | o2::soa::Index<>, caloextra::ClusterId, caloextra::CellId); //! |
535 | 536 | using SkimEMCCell = SkimEMCCells::iterator; |
536 | | - |
537 | | -DECLARE_SOA_TABLE(SkimEMCMTs, "AOD", "SKIMEMCMTS", //! table of link between skimmed EMCal clusters and their matched tracks |
538 | | - o2::soa::Index<>, caloextra::ClusterId, caloextra::TrackEta, |
539 | | - caloextra::TrackPhi, caloextra::TrackP, caloextra::TrackPt); |
540 | | -using SkimEMCMT = SkimEMCMTs::iterator; |
541 | | - |
542 | | -namespace gammareco |
543 | | -{ |
544 | | -DECLARE_SOA_COLUMN(Method, method, int); //! cut bit for PCM photon candidates |
545 | | -DECLARE_SOA_INDEX_COLUMN_FULL(SkimmedPCM, skimmedPCM, int, V0PhotonsKF, ""); //! reference to the gamma in the skimmed PCM table |
546 | | -DECLARE_SOA_INDEX_COLUMN_FULL(SkimmedPHOS, skimmedPHOS, int, PHOSClusters, ""); //! reference to the gamma in the skimmed PHOS table |
547 | | -DECLARE_SOA_INDEX_COLUMN_FULL(SkimmedEMC, skimmedEMC, int, SkimEMCClusters, ""); //! reference to the gamma in the skimmed EMCal table |
548 | | -DECLARE_SOA_COLUMN(PCMCutBit, pcmcutbit, uint64_t); //! cut bit for PCM photon candidates |
549 | | -DECLARE_SOA_COLUMN(PHOSCutBit, phoscutbit, uint64_t); //! cut bit for PHOS photon candidates |
550 | | -DECLARE_SOA_COLUMN(EMCCutBit, emccutbit, uint64_t); //! cut bit for EMCal photon candidates |
551 | | -} // namespace gammareco |
552 | | -DECLARE_SOA_TABLE(SkimGammas, "AOD", "SKIMGAMMAS", //! table of all gamma candidates (PCM, EMCal and PHOS) after cuts |
553 | | - o2::soa::Index<>, skimmedcluster::CollisionId, gammareco::Method, |
554 | | - skimmedcluster::E, skimmedcluster::Eta, skimmedcluster::Phi, |
555 | | - gammareco::SkimmedEMCId, gammareco::SkimmedPHOSId); |
556 | | -DECLARE_SOA_TABLE(SkimPCMCuts, "AOD", "SKIMPCMCUTS", //! table of link between skimmed PCM photon candidates and their cuts |
557 | | - o2::soa::Index<>, gammareco::SkimmedPCMId, gammareco::PCMCutBit); //! |
558 | | -DECLARE_SOA_TABLE(SkimPHOSCuts, "AOD", "SKIMPHOSCUTS", //! table of link between skimmed PHOS photon candidates and their cuts |
559 | | - o2::soa::Index<>, gammareco::SkimmedPHOSId, gammareco::PHOSCutBit); //! |
560 | | -DECLARE_SOA_TABLE(SkimEMCCuts, "AOD", "SKIMEMCCUTS", //! table of link between skimmed EMCal photon candidates and their cuts |
561 | | - o2::soa::Index<>, gammareco::SkimmedEMCId, gammareco::EMCCutBit); //! |
562 | 537 | } // namespace o2::aod |
563 | 538 |
|
564 | 539 | #endif // PWGEM_PHOTONMESON_DATAMODEL_GAMMATABLES_H_ |
0 commit comments