New tables for purity studies

Author: scannito

PWGLF/DataModel/LFPhiStrangeCorrelationTables.h

@@ -22,6 +22,48 @@
 
 namespace o2::aod
 {
+namespace lf_selection_event
+{
+DECLARE_SOA_COLUMN(DefaultSel, defaultSel, bool);
+DECLARE_SOA_COLUMN(PhimesonSel, phimesonSel, bool);
+} // namespace lf_selection_event
+
+/*DECLARE_SOA_TABLE(PhiStrangeDefEvtSelDataLike, "AOD", "DEFEVTSELDATA",
+                  lf_selection_event::DefaultSel);
+
+DECLARE_SOA_TABLE(PhiStrangeDefEvtSelMcGen, "AOD", "DEFEVTSELMCGEN",
+                  lf_selection_event::DefaultSel);*/
+
+DECLARE_SOA_TABLE(PhiStrangeEvtSelDataLike, "AOD", "EVTSELDATA",
+                  lf_selection_event::DefaultSel,
+                  lf_selection_event::PhimesonSel);
+
+DECLARE_SOA_TABLE(PhiStrangeEvtSelMcGen, "AOD", "EVTSELMCGEN",
+                  lf_selection_event::DefaultSel,
+                  lf_selection_event::PhimesonSel);
+
+/*namespace lf_selection_default_collision
+{
+DECLARE_SOA_COLUMN(DefaultSel, defaultSel, bool);
+} // namespace lf_selection_default_collision
+
+DECLARE_SOA_TABLE(DefaultSelectionData, "AOD", "DEFSELDATA",
+                  lf_selection_default_collision::DefaultSel);
+
+DECLARE_SOA_TABLE(DefaultSelectionMcGen, "AOD", "DEFSELMCGEN",
+                  lf_selection_default_collision::DefaultSel);
+
+namespace lf_selection_phi_collision
+{
+DECLARE_SOA_COLUMN(PhimesonSel, phimesonSel, bool);
+} // namespace lf_selection_phi_collision
+
+DECLARE_SOA_TABLE(PhimesonSelectionData, "AOD", "PHIINCOLLDATA",
+                  lf_selection_phi_collision::PhimesonSel);
+
+DECLARE_SOA_TABLE(PhimesonSelectionMcGen, "AOD", "PHIINCOLLMCGEN",
+                  lf_selection_phi_collision::PhimesonSel);*/
+
 namespace lf_selection_phi_candidate
 {
 DECLARE_SOA_INDEX_COLUMN(Collision, collision);
@@ -64,19 +106,67 @@ DECLARE_SOA_TABLE(PhimesonCandidatesMcGen, "AOD", "PHICANDMCGEN",
                   lf_selection_phi_candidate::Phi,
                   lf_selection_phi_candidate::InMassRegion<lf_selection_phi_candidate::M>);
 
-namespace lf_selection_phi_collision
+namespace lf_selection_k0s_reduced
 {
-DECLARE_SOA_COLUMN(PhimesonSel, phimesonSel, bool);
-} // namespace lf_selection_phi_collision
+DECLARE_SOA_INDEX_COLUMN(Collision, collision);
 
-DECLARE_SOA_TABLE(PhimesonSelectionData, "AOD", "PHIINCOLLDATA",
-                  lf_selection_phi_collision::PhimesonSel);
+DECLARE_SOA_COLUMN(M, m, float);
+DECLARE_SOA_COLUMN(Pt, pt, float);
+DECLARE_SOA_COLUMN(Y, y, float);
+DECLARE_SOA_COLUMN(Phi, phi, float);
 
-/*DECLARE_SOA_TABLE(PhimesonSelectionMcReco, "AOD", "PHIINCOLLMCRECO",
-                  lf_selection_phi_collision::PhimesonSel);*/
+DECLARE_SOA_DYNAMIC_COLUMN(InMassRegion, inMassRegion,
+                           [](float m, float minM, float maxM) -> bool {
+                             return (m >= minM && m <= maxM);
+                           });
+} // namespace lf_selection_k0s_reduced
 
-DECLARE_SOA_TABLE(PhimesonSelectionMcGen, "AOD", "PHIINCOLLMCGEN",
-                  lf_selection_phi_collision::PhimesonSel);
+DECLARE_SOA_TABLE(K0sReducedCandidatesData, "AOD", "K0SCANDDATA",
+                  soa::Index<>,
+                  lf_selection_k0s_reduced::CollisionId,
+                  lf_selection_k0s_reduced::M,
+                  lf_selection_k0s_reduced::Pt,
+                  lf_selection_k0s_reduced::Y,
+                  lf_selection_k0s_reduced::Phi,
+                  lf_selection_k0s_reduced::InMassRegion<lf_selection_k0s_reduced::M>);
+
+DECLARE_SOA_TABLE(K0sReducedCandidatesMcReco, "AOD", "K0SCANDMCRECO",
+                  soa::Index<>,
+                  lf_selection_k0s_reduced::CollisionId,
+                  lf_selection_k0s_reduced::M,
+                  lf_selection_k0s_reduced::Pt,
+                  lf_selection_k0s_reduced::Y,
+                  lf_selection_k0s_reduced::Phi,
+                  lf_selection_k0s_reduced::InMassRegion<lf_selection_k0s_reduced::M>);
+
+namespace lf_selection_pion_track
+{
+DECLARE_SOA_INDEX_COLUMN(Collision, collision);
+
+DECLARE_SOA_COLUMN(NSigmaTPC, nSigmaTPC, float);
+DECLARE_SOA_COLUMN(NSigmaTOF, nSigmaTOF, float);
+DECLARE_SOA_COLUMN(Pt, pt, float);
+DECLARE_SOA_COLUMN(Y, y, float);
+DECLARE_SOA_COLUMN(Phi, phi, float);
+} // namespace lf_selection_pion_track
+
+DECLARE_SOA_TABLE(PionTracksData, "AOD", "PITRACKSDATA",
+                  soa::Index<>,
+                  lf_selection_pion_track::CollisionId,
+                  lf_selection_pion_track::NSigmaTPC,
+                  lf_selection_pion_track::NSigmaTOF,
+                  lf_selection_pion_track::Pt,
+                  lf_selection_pion_track::Y,
+                  lf_selection_pion_track::Phi);
+
+DECLARE_SOA_TABLE(PionTracksMcReco, "AOD", "PITRACKSMCRECO",
+                  soa::Index<>,
+                  lf_selection_pion_track::CollisionId,
+                  lf_selection_pion_track::NSigmaTPC,
+                  lf_selection_pion_track::NSigmaTOF,
+                  lf_selection_pion_track::Pt,
+                  lf_selection_pion_track::Y,
+                  lf_selection_pion_track::Phi);
 } // namespace o2::aod
 
 #endif // PWGLF_DATAMODEL_LFPHISTRANGECORRELATIONTABLES_H_