Skip to content

Commit 5a70c01

Browse files
subhadee-alisubhadee-ali
authored
Refactor variable names and apply static_cast
1 parent a53d9cf commit 5a70c01

File tree

1 file changed

+70
-70
lines changed

1 file changed

+70
-70
lines changed

‎PWGCF/TwoParticleCorrelations/Tasks/lambdaSpinPolarization.cxx‎

Lines changed: 70 additions & 70 deletions
Original file line numberDiff line numberDiff line change
@@ -1064,7 +1064,7 @@ struct LambdaTableProducer {
10641064
if (!selTrackBasic(track))
10651065
return false;
10661066
if (cApplyFakeDcaCuts) {
1067-
float minDca = isProton ? (float)cMinDcaProtonToPV : (float)cMinDcaPionToPV;
1067+
float minDca = isProton ? static_cast<float>(cMinDcaProtonToPV) : static_cast<float>(cMinDcaPionToPV);
10681068
if (std::abs(track.dcaXY()) < minDca)
10691069
return false;
10701070
}
@@ -1822,14 +1822,14 @@ struct LambdaSpinPolarization {
18221822
static constexpr std::string_view SubDirHistUS[] = {"LaPLaM", "LaMLaP", "LaPLaP", "LaMLaM"};
18231823
static constexpr std::string_view SubDirHistBkg[] = {"LaPFkLaM", "LaMLFkLaP", "LaPFkLaP", "LaMLFkLaM"};
18241824

1825-
constexpr bool isBkg = (part_pair == kLambdaFakeAntiLambda || part_pair == kAntiLambdaFakeLambda || part_pair == kLambdaFakeLambda || part_pair == kAntiLambdaFakeAntiLambda);
1825+
constexpr bool IsBkg = (part_pair == kLambdaFakeAntiLambda || part_pair == kAntiLambdaFakeLambda || part_pair == kLambdaFakeLambda || part_pair == kAntiLambdaFakeAntiLambda);
18261826

18271827
// Fill pair invariant mass histogram
1828-
if constexpr (!isBkg) {
1828+
if constexpr (!IsBkg) {
18291829
histos.fill(HIST("Reco/h2f_n2_mass_") + HIST(SubDirHistUS[part_pair]), p1.mass(), p2.mass(), p1.pt(), p2.pt());
18301830
} else {
1831-
constexpr int bkgIdx = (int)part_pair - 4;
1832-
histos.fill(HIST("RecoBkg/h2f_n2_mass_") + HIST(SubDirHistBkg[bkgIdx]), p1.mass(), p2.mass(), p1.pt(), p2.pt());
1831+
constexpr int BkgIdx = static_cast<int>(part_pair) - 4;
1832+
histos.fill(HIST("RecoBkg/h2f_n2_mass_") + HIST(SubDirHistBkg[BkgIdx]), p1.mass(), p2.mass(), p1.pt(), p2.pt());
18331833
}
18341834

18351835
float drap = p1.rap() - p2.rap();
@@ -1842,71 +1842,71 @@ struct LambdaSpinPolarization {
18421842
std::array<float, 4> pr2 = {p2.prPx(), p2.prPy(), p2.prPz(), MassProton};
18431843

18441844
if (cDoAtlasMethod) {
1845-
std::array<float, 4> l1_atlas = l1;
1846-
std::array<float, 4> l2_atlas = l2;
1847-
std::array<float, 4> pr1_atlas = pr1;
1848-
std::array<float, 4> pr2_atlas = pr2;
1845+
std::array<float, 4> l1Atlas = l1;
1846+
std::array<float, 4> l2Atlas = l2;
1847+
std::array<float, 4> pr1Atlas = pr1;
1848+
std::array<float, 4> pr2Atlas = pr2;
18491849

18501850
std::array<float, 4> llpair = {
1851-
l1_atlas[0] + l2_atlas[0],
1852-
l1_atlas[1] + l2_atlas[1],
1853-
l1_atlas[2] + l2_atlas[2],
1854-
l1_atlas[3] + l2_atlas[3]};
1851+
l1Atlas[0] + l2Atlas[0],
1852+
l1Atlas[1] + l2Atlas[1],
1853+
l1Atlas[2] + l2Atlas[2],
1854+
l1Atlas[3] + l2Atlas[3]};
18551855

18561856
std::array<float, 3> vPair;
18571857
getBoostVector(llpair, vPair, cInvBoostFlag);
1858-
boost(l1_atlas, vPair);
1859-
boost(l2_atlas, vPair);
1860-
boost(pr1_atlas, vPair);
1861-
boost(pr2_atlas, vPair);
1862-
1863-
std::array<float, 3> v1_pair, v2_pair;
1864-
getBoostVector(l1_atlas, v1_pair, cInvBoostFlag);
1865-
getBoostVector(l2_atlas, v2_pair, cInvBoostFlag);
1866-
boost(pr1_atlas, v1_pair);
1867-
boost(pr2_atlas, v2_pair);
1868-
1869-
std::array<float, 3> pr1tv_atlas = {pr1_atlas[0], pr1_atlas[1], pr1_atlas[2]};
1870-
std::array<float, 3> pr2tv_atlas = {pr2_atlas[0], pr2_atlas[1], pr2_atlas[2]};
1858+
boost(l1Atlas, vPair);
1859+
boost(l2Atlas, vPair);
1860+
boost(pr1Atlas, vPair);
1861+
boost(pr2Atlas, vPair);
1862+
1863+
std::array<float, 3> v1Pair, v2Pair;
1864+
getBoostVector(l1Atlas, v1Pair, cInvBoostFlag);
1865+
getBoostVector(l2Atlas, v2Pair, cInvBoostFlag);
1866+
boost(pr1Atlas, v1Pair);
1867+
boost(pr2Atlas, v2Pair);
1868+
1869+
std::array<float, 3> pr1tvAtlas = {pr1Atlas[0], pr1Atlas[1], pr1Atlas[2]};
1870+
std::array<float, 3> pr2tvAtlas = {pr2Atlas[0], pr2Atlas[1], pr2Atlas[2]};
18711871
float ctheta =
1872-
RecoDecay::dotProd(pr1tv_atlas, pr2tv_atlas) /
1873-
(RecoDecay::sqrtSumOfSquares(pr1tv_atlas[0], pr1tv_atlas[1], pr1tv_atlas[2]) *
1874-
RecoDecay::sqrtSumOfSquares(pr2tv_atlas[0], pr2tv_atlas[1], pr2tv_atlas[2]));
1872+
RecoDecay::dotProd(pr1tvAtlas, pr2tvAtlas) /
1873+
(RecoDecay::sqrtSumOfSquares(pr1tvAtlas[0], pr1tvAtlas[1], pr1tvAtlas[2]) *
1874+
RecoDecay::sqrtSumOfSquares(pr2tvAtlas[0], pr2tvAtlas[1], pr2tvAtlas[2]));
18751875

1876-
if constexpr (!isBkg) {
1876+
if constexpr (!IsBkg) {
18771877
histos.fill(HIST("RecoCorr/h2f_n2_ctheta_") + HIST(SubDirHistUS[part_pair]), cent, drap, dphi, ctheta);
18781878
histos.fill(HIST("RecoCorr/h2f_n2_dltaR_") + HIST(SubDirHistUS[part_pair]), cent, dR, ctheta);
18791879
} else {
1880-
constexpr int bkgIdx = (int)part_pair - 4;
1881-
histos.fill(HIST("RecoCorrBkg/h2f_n2_ctheta_") + HIST(SubDirHistBkg[bkgIdx]), cent, drap, dphi, ctheta);
1882-
histos.fill(HIST("RecoCorrBkg/h2f_n2_dltaR_") + HIST(SubDirHistBkg[bkgIdx]), cent, dR, ctheta);
1880+
constexpr int BkgIdx = static_cast<int>(part_pair) - 4;
1881+
histos.fill(HIST("RecoCorrBkg/h2f_n2_ctheta_") + HIST(SubDirHistBkg[BkgIdx]), cent, drap, dphi, ctheta);
1882+
histos.fill(HIST("RecoCorrBkg/h2f_n2_dltaR_") + HIST(SubDirHistBkg[BkgIdx]), cent, dR, ctheta);
18831883
}
18841884
}
18851885

18861886
if (cDoStarMethod) {
1887-
std::array<float, 4> pr1_star = pr1;
1888-
std::array<float, 4> pr2_star = pr2;
1887+
std::array<float, 4> pr1Star = pr1;
1888+
std::array<float, 4> pr2Star = pr2;
18891889

1890-
std::array<float, 3> v1_lab, v2_lab;
1891-
getBoostVector(l1, v1_lab, cInvBoostFlag);
1892-
getBoostVector(l2, v2_lab, cInvBoostFlag);
1893-
boost(pr1_star, v1_lab);
1894-
boost(pr2_star, v2_lab);
1890+
std::array<float, 3> v1Lab, v2Lab;
1891+
getBoostVector(l1, v1Lab, cInvBoostFlag);
1892+
getBoostVector(l2, v2Lab, cInvBoostFlag);
1893+
boost(pr1Star, v1Lab);
1894+
boost(pr2Star, v2Lab);
18951895

1896-
std::array<float, 3> pr1tv_star = {pr1_star[0], pr1_star[1], pr1_star[2]};
1897-
std::array<float, 3> pr2tv_star = {pr2_star[0], pr2_star[1], pr2_star[2]};
1896+
std::array<float, 3> pr1tvStar = {pr1Star[0], pr1Star[1], pr1Star[2]};
1897+
std::array<float, 3> pr2tvStar = {pr2Star[0], pr2Star[1], pr2Star[2]};
18981898
float ctheta =
1899-
RecoDecay::dotProd(pr1tv_star, pr2tv_star) /
1900-
(RecoDecay::sqrtSumOfSquares(pr1tv_star[0], pr1tv_star[1], pr1tv_star[2]) *
1901-
RecoDecay::sqrtSumOfSquares(pr2tv_star[0], pr2tv_star[1], pr2tv_star[2]));
1899+
RecoDecay::dotProd(pr1tvStar, pr2tvStar) /
1900+
(RecoDecay::sqrtSumOfSquares(pr1tvStar[0], pr1tvStar[1], pr1tvStar[2]) *
1901+
RecoDecay::sqrtSumOfSquares(pr2tvStar[0], pr2tvStar[1], pr2tvStar[2]));
19021902

1903-
if constexpr (!isBkg) {
1903+
if constexpr (!IsBkg) {
19041904
histos.fill(HIST("RecoCorr/h2f_n2_ctheta_") + HIST(SubDirHistUS[part_pair]), cent, drap, dphi, ctheta);
19051905
histos.fill(HIST("RecoCorr/h2f_n2_dltaR_") + HIST(SubDirHistUS[part_pair]), cent, dR, ctheta);
19061906
} else {
1907-
constexpr int bkgIdx = (int)part_pair - 4;
1908-
histos.fill(HIST("RecoCorrBkg/h2f_n2_ctheta_") + HIST(SubDirHistBkg[bkgIdx]), cent, drap, dphi, ctheta);
1909-
histos.fill(HIST("RecoCorrBkg/h2f_n2_dltaR_") + HIST(SubDirHistBkg[bkgIdx]), cent, dR, ctheta);
1907+
constexpr int BkgIdx = static_cast<int>(part_pair) - 4;
1908+
histos.fill(HIST("RecoCorrBkg/h2f_n2_ctheta_") + HIST(SubDirHistBkg[BkgIdx]), cent, drap, dphi, ctheta);
1909+
histos.fill(HIST("RecoCorrBkg/h2f_n2_dltaR_") + HIST(SubDirHistBkg[BkgIdx]), cent, dR, ctheta);
19101910
}
19111911
}
19121912
}
@@ -1949,7 +1949,7 @@ struct LambdaSpinPolarization {
19491949

19501950
decltype(&*trks_2.begin()) currentPartner = nullptr;
19511951

1952-
for (auto const* trk_2 : candidates) {
1952+
for (auto const& trk_2 : candidates) {
19531953

19541954
if constexpr (samelambda) {
19551955
if (trk_1.index() == trk_2->index())
@@ -2071,36 +2071,36 @@ struct LambdaSpinPolarization {
20712071
histos.fill(HIST("QA/ME/hPoolCentVz"), col1.cent(), col1.posZ());
20722072

20732073
// Lambda slices
2074-
auto lambdaTracks_col1 = partLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col1.globalIndex(), cache);
2075-
auto lambdaTracks_col2 = partLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col2.globalIndex(), cache);
2074+
auto lambdaTracksCol1 = partLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col1.globalIndex(), cache);
2075+
auto lambdaTracksCol2 = partLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col2.globalIndex(), cache);
20762076

20772077
// Anti-lambda slices
2078-
auto antiLambdaTracks_col1 = partAntiLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col1.globalIndex(), cache);
2079-
auto antiLambdaTracks_col2 = partAntiLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col2.globalIndex(), cache);
2078+
auto antiLambdaTracksCol1 = partAntiLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col1.globalIndex(), cache);
2079+
auto antiLambdaTracksCol2 = partAntiLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col2.globalIndex(), cache);
20802080

20812081
// Fake lambda slices (LS background)
2082-
auto fakeLambdaTracks_col2 = partFakeLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col2.globalIndex(), cache);
2083-
auto fakeAntiLambdaTracks_col2 = partFakeAntiLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col2.globalIndex(), cache);
2082+
auto fakeLambdaTracksCol2 = partFakeLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col2.globalIndex(), cache);
2083+
auto fakeAntiLambdaTracksCol2 = partFakeAntiLambdaTracks->sliceByCached(aod::lambdatrack::lambdaCollisionId, col2.globalIndex(), cache);
20842084

20852085
// QA: multiplicities
2086-
histos.fill(HIST("QA/ME/hLambdaMultVsCent"), col1.cent(), lambdaTracks_col1.size());
2087-
histos.fill(HIST("QA/ME/hAntiLambdaMultVsCent"), col1.cent(), antiLambdaTracks_col1.size());
2086+
histos.fill(HIST("QA/ME/hLambdaMultVsCent"), col1.cent(), lambdaTracksCol1.size());
2087+
histos.fill(HIST("QA/ME/hAntiLambdaMultVsCent"), col1.cent(), antiLambdaTracksCol1.size());
20882088

2089-
analyzePairsME<kLambdaAntiLambda, false>(lambdaTracks_col1, antiLambdaTracks_col2);
2090-
analyzePairsME<kAntiLambdaLambda, false>(antiLambdaTracks_col1, lambdaTracks_col2);
2091-
analyzePairsME<kLambdaLambda, true>(lambdaTracks_col1, lambdaTracks_col2);
2092-
analyzePairsME<kAntiLambdaAntiLambda, true>(antiLambdaTracks_col1, antiLambdaTracks_col2);
2089+
analyzePairsME<kLambdaAntiLambda, false>(lambdaTracksCol1, antiLambdaTracksCol2);
2090+
analyzePairsME<kAntiLambdaLambda, false>(antiLambdaTracksCol1, lambdaTracksCol2);
2091+
analyzePairsME<kLambdaLambda, true>(lambdaTracksCol1, lambdaTracksCol2);
2092+
analyzePairsME<kAntiLambdaAntiLambda, true>(antiLambdaTracksCol1, antiLambdaTracksCol2);
20932093

2094-
analyzePairsME<kLambdaFakeAntiLambda, false>(lambdaTracks_col1, fakeAntiLambdaTracks_col2);
2095-
analyzePairsME<kLambdaFakeAntiLambda, false>(fakeLambdaTracks_col2, antiLambdaTracks_col1);
2094+
analyzePairsME<kLambdaFakeAntiLambda, false>(lambdaTracksCol1, fakeAntiLambdaTracksCol2);
2095+
analyzePairsME<kLambdaFakeAntiLambda, false>(fakeLambdaTracksCol2, antiLambdaTracksCol1);
20962096

2097-
analyzePairsME<kAntiLambdaFakeLambda, false>(antiLambdaTracks_col1, fakeLambdaTracks_col2);
2098-
analyzePairsME<kAntiLambdaFakeLambda, false>(fakeAntiLambdaTracks_col2, lambdaTracks_col1);
2097+
analyzePairsME<kAntiLambdaFakeLambda, false>(antiLambdaTracksCol1, fakeLambdaTracksCol2);
2098+
analyzePairsME<kAntiLambdaFakeLambda, false>(fakeAntiLambdaTracksCol2, lambdaTracksCol1);
20992099

2100-
analyzePairsME<kLambdaFakeLambda, false>(lambdaTracks_col1, fakeLambdaTracks_col2);
2101-
analyzePairsME<kLambdaFakeLambda, false>(fakeLambdaTracks_col2, lambdaTracks_col1);
2102-
analyzePairsME<kAntiLambdaFakeAntiLambda, false>(antiLambdaTracks_col1, fakeAntiLambdaTracks_col2);
2103-
analyzePairsME<kAntiLambdaFakeAntiLambda, false>(fakeAntiLambdaTracks_col2, antiLambdaTracks_col1);
2100+
analyzePairsME<kLambdaFakeLambda, false>(lambdaTracksCol1, fakeLambdaTracksCol2);
2101+
analyzePairsME<kLambdaFakeLambda, false>(fakeLambdaTracksCol2, lambdaTracksCol1);
2102+
analyzePairsME<kAntiLambdaFakeAntiLambda, false>(antiLambdaTracksCol1, fakeAntiLambdaTracksCol2);
2103+
analyzePairsME<kAntiLambdaFakeAntiLambda, false>(fakeAntiLambdaTracksCol2, antiLambdaTracksCol1);
21042104
}
21052105
}
21062106

0 commit comments

Comments
 (0)