Skip to content

Commit 69758e7

Browse files
committed
Feat: add universal getters for pt
1 parent 26c07a1 commit 69758e7

1 file changed

Lines changed: 24 additions & 5 deletions

File tree

PWGCF/Femto/DataModel/FemtoTables.h

Lines changed: 24 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -234,10 +234,6 @@ DECLARE_SOA_DYNAMIC_COLUMN(SignedPt, signedPt,
234234
[](uint16_t signedBinnedPt) -> float {
235235
return unBinSignedPt(signedBinnedPt);
236236
});
237-
DECLARE_SOA_DYNAMIC_COLUMN(UnsignedPt, unsignedPt,
238-
[](uint16_t unsignedBinnedPt) -> float {
239-
return unBinUnsignedPt(unsignedBinnedPt);
240-
});
241237
DECLARE_SOA_DYNAMIC_COLUMN(Eta, eta,
242238
[](uint16_t binnedEta) -> float {
243239
return unBinEta(binnedEta);
@@ -246,6 +242,23 @@ DECLARE_SOA_DYNAMIC_COLUMN(Phi, phi,
246242
[](uint16_t binnedPhi) -> float {
247243
return unBinPhi(binnedPhi);
248244
});
245+
246+
namespace signedpt
247+
{
248+
DECLARE_SOA_DYNAMIC_COLUMN(Pt, pt,
249+
[](uint16_t signedBinnedPt) -> float {
250+
return std::fabs(unBinSignedPt(signedBinnedPt));
251+
});
252+
} // namespace signedpt
253+
254+
namespace unsignedpt
255+
{
256+
DECLARE_SOA_DYNAMIC_COLUMN(Pt, pt,
257+
[](uint16_t unsignedBinnedPt) -> float {
258+
return unBinUnsignedPt(unsignedBinnedPt);
259+
});
260+
} // namespace unsignedpt
261+
249262
} // namespace lite
250263
} // namespace femtobase
251264

@@ -355,6 +368,7 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FLiteTracks_001, "FLITETRACK", 1, //! femto t
355368
femtobase::lite::BinnedPhi,
356369
femtobase::lite::Sign<femtobase::lite::SignedBinnedPt>,
357370
femtobase::lite::SignedPt<femtobase::lite::SignedBinnedPt>,
371+
femtobase::lite::signedpt::Pt<femtobase::lite::SignedBinnedPt>,
358372
femtobase::lite::Eta<femtobase::lite::BinnedEta>,
359373
femtobase::lite::Phi<femtobase::lite::BinnedPhi>);
360374
using FLiteTracks = FLiteTracks_001;
@@ -628,6 +642,7 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FLiteLambdas_001, "FLITELAMBDA", 1,
628642
femtov0s::lite::NegDauId,
629643
femtobase::lite::Sign<femtobase::lite::SignedBinnedPt>,
630644
femtobase::lite::SignedPt<femtobase::lite::SignedBinnedPt>,
645+
femtobase::lite::signedpt::Pt<femtobase::lite::SignedBinnedPt>,
631646
femtobase::lite::Eta<femtobase::lite::BinnedEta>,
632647
femtobase::lite::Phi<femtobase::lite::BinnedPhi>,
633648
femtov0s::lite::LambdaMass<femtov0s::lite::BinnedLambdaMass>);
@@ -681,7 +696,7 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FLiteK0shorts_001, "FLITEK0SHORT", 1,
681696
femtov0s::lite::BinnedK0shortMass,
682697
femtov0s::lite::PosDauId,
683698
femtov0s::lite::NegDauId,
684-
femtobase::lite::UnsignedPt<femtobase::lite::UnsignedBinnedPt>,
699+
femtobase::lite::unsignedpt::Pt<femtobase::lite::UnsignedBinnedPt>,
685700
femtobase::lite::Eta<femtobase::lite::BinnedEta>,
686701
femtobase::lite::Phi<femtobase::lite::BinnedPhi>,
687702
femtov0s::lite::K0shortMass<femtov0s::lite::BinnedK0shortMass>);
@@ -795,6 +810,7 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FLiteSigmas_001, "FLITESIGMA", 1,
795810
femtokinks::lite::ChaDauId,
796811
femtobase::lite::Sign<femtobase::lite::SignedBinnedPt>,
797812
femtobase::lite::SignedPt<femtobase::lite::SignedBinnedPt>,
813+
femtobase::lite::signedpt::Pt<femtobase::lite::SignedBinnedPt>,
798814
femtobase::lite::Eta<femtobase::lite::BinnedEta>,
799815
femtobase::lite::Phi<femtobase::lite::BinnedPhi>,
800816
femtokinks::lite::SigmaMass<femtokinks::lite::BinnedSigmaMass>);
@@ -850,6 +866,7 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FLiteSigmaPlus_001, "FLITESIGMAPLUS", 1,
850866
femtokinks::lite::ChaDauId,
851867
femtobase::lite::Sign<femtobase::lite::SignedBinnedPt>,
852868
femtobase::lite::SignedPt<femtobase::lite::SignedBinnedPt>,
869+
femtobase::lite::signedpt::Pt<femtobase::lite::SignedBinnedPt>,
853870
femtobase::lite::Eta<femtobase::lite::BinnedEta>,
854871
femtobase::lite::Phi<femtobase::lite::BinnedPhi>,
855872
femtokinks::lite::SigmaMass<femtokinks::lite::BinnedSigmaMass>);
@@ -963,6 +980,7 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FLiteXis_001, "FLITEXI", 1,
963980
femtov0s::lite::NegDauId,
964981
femtobase::lite::Sign<femtobase::lite::SignedBinnedPt>,
965982
femtobase::lite::SignedPt<femtobase::lite::SignedBinnedPt>,
983+
femtobase::lite::signedpt::Pt<femtobase::lite::SignedBinnedPt>,
966984
femtobase::lite::Eta<femtobase::lite::BinnedEta>,
967985
femtobase::lite::Phi<femtobase::lite::BinnedPhi>,
968986
femtocascades::lite::XiMass<femtocascades::lite::BinnedXiMass>);
@@ -1021,6 +1039,7 @@ DECLARE_SOA_TABLE_STAGED_VERSIONED(FLiteOmegas_001, "FLITEOMEGA", 1,
10211039
femtov0s::lite::NegDauId,
10221040
femtobase::lite::Sign<femtobase::lite::SignedBinnedPt>,
10231041
femtobase::lite::SignedPt<femtobase::lite::SignedBinnedPt>,
1042+
femtobase::lite::signedpt::Pt<femtobase::lite::SignedBinnedPt>,
10241043
femtobase::lite::Eta<femtobase::lite::BinnedEta>,
10251044
femtobase::lite::Phi<femtobase::lite::BinnedPhi>,
10261045
femtocascades::lite::OmegaMass<femtocascades::lite::BinnedOmegaMass>);

0 commit comments

Comments
 (0)