Skip to content

Commit 9455ee5

Browse files
authored
Refactor correlation calculation using utility functions
1 parent 690c0e4 commit 9455ee5

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

PWGCF/TwoParticleCorrelations/Tasks/lambdaR2Correlation.cxx

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1586,13 +1586,9 @@ struct LambdaSpinCorrelation {
15861586
boost(pr1, v1);
15871587
boost(pr2, v2);
15881588

1589-
float cdot = pr1[0] * pr2[0] + pr1[1] * pr2[1] + pr1[2] * pr2[2];
1590-
float mag1 = std::sqrt(pr1[0] * pr1[0] + pr1[1] * pr1[1] + pr1[2] * pr1[2]);
1591-
float mag2 = std::sqrt(pr2[0] * pr2[0] + pr2[1] * pr2[1] + pr2[2] * pr2[2]);
1592-
float cphi = cdot / (mag1 * mag2);
1593-
15941589
std::array<float, 3> pr1tv = {pr1[0], pr1[1], pr1[2]};
15951590
std::array<float, 3> pr2tv = {pr2[0], pr2[1], pr2[2]};
1591+
float cphi = RecoDecay::dotProd(pr1tv, pr2tv) / (RecoDecay::sqrtSumOfSquares(pr1tv[0], pr1tv[1], pr1tv[2]) * RecoDecay::sqrtSumOfSquares(pr2tv[0], pr2tv[1], pr2tv[2]));
15961592
float prdphi = RecoDecay::constrainAngle(RecoDecay::phi(pr1) - RecoDecay::phi(pr2), -PI);
15971593
float prdrap = RecoDecay::eta(pr1tv) - RecoDecay::eta(pr2tv);
15981594
float dr = std::sqrt(prdrap * prdrap + prdphi * prdphi);

0 commit comments

Comments
 (0)