Skip to content

Commit ae476fb

Browse files
dsekihatdsekihat
committed
PWGEM/Dilepton: support Q vector of FV0A for flow
1 parent 50f0282 commit ae476fb

File tree

1 file changed

+18
-12
lines changed

1 file changed

+18
-12
lines changed

‎PWGEM/Dilepton/Core/Dilepton.h‎

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -973,21 +973,23 @@ struct Dilepton {
973973
std::array<float, 2> q2ft0m = {collision.q2xft0m(), collision.q2yft0m()};
974974
std::array<float, 2> q2ft0a = {collision.q2xft0a(), collision.q2yft0a()};
975975
std::array<float, 2> q2ft0c = {collision.q2xft0c(), collision.q2yft0c()};
976+
std::array<float, 2> q2fv0a = {collision.q2xfv0a(), collision.q2yfv0a()};
976977
std::array<float, 2> q2btot = {collision.q2xbtot(), collision.q2ybtot()};
977978
std::array<float, 2> q2bpos = {collision.q2xbpos(), collision.q2ybpos()};
978979
std::array<float, 2> q2bneg = {collision.q2xbneg(), collision.q2ybneg()};
979980
std::array<float, 2> q3ft0m = {collision.q3xft0m(), collision.q3yft0m()};
980981
std::array<float, 2> q3ft0a = {collision.q3xft0a(), collision.q3yft0a()};
981982
std::array<float, 2> q3ft0c = {collision.q3xft0c(), collision.q3yft0c()};
983+
std::array<float, 2> q3fv0a = {collision.q3xfv0a(), collision.q3yfv0a()};
982984
std::array<float, 2> q3btot = {collision.q3xbtot(), collision.q3ybtot()};
983985
std::array<float, 2> q3bpos = {collision.q3xbpos(), collision.q3ybpos()};
984986
std::array<float, 2> q3bneg = {collision.q3xbneg(), collision.q3ybneg()};
985987

986988
std::vector<std::vector<std::array<float, 2>>> qvectors = {
987-
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 0th harmonics
988-
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 1st harmonics
989-
{q2ft0m, q2ft0a, q2ft0c, q2btot, q2bpos, q2bneg}, // 2nd harmonics
990-
{q3ft0m, q3ft0a, q3ft0c, q3btot, q3bpos, q3bneg}, // 3rd harmonics
989+
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 0th harmonics
990+
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 1st harmonics
991+
{q2ft0m, q2ft0a, q2ft0c, q2fv0a, q2btot, q2bpos, q2bneg}, // 2nd harmonics
992+
{q3ft0m, q3ft0a, q3ft0c, q3fv0a, q3btot, q3bpos, q3bneg}, // 3rd harmonics
991993
};
992994

993995
if constexpr (ev_id == 0) {
@@ -1167,23 +1169,25 @@ struct Dilepton {
11671169
std::array<float, 2> q2ft0m = {collision.q2xft0m(), collision.q2yft0m()};
11681170
std::array<float, 2> q2ft0a = {collision.q2xft0a(), collision.q2yft0a()};
11691171
std::array<float, 2> q2ft0c = {collision.q2xft0c(), collision.q2yft0c()};
1172+
std::array<float, 2> q2fv0a = {collision.q2xfv0a(), collision.q2yfv0a()};
11701173
std::array<float, 2> q2btot = {collision.q2xbtot(), collision.q2ybtot()};
11711174
std::array<float, 2> q2bpos = {collision.q2xbpos(), collision.q2ybpos()};
11721175
std::array<float, 2> q2bneg = {collision.q2xbneg(), collision.q2ybneg()};
11731176
std::array<float, 2> q3ft0m = {collision.q3xft0m(), collision.q3yft0m()};
11741177
std::array<float, 2> q3ft0a = {collision.q3xft0a(), collision.q3yft0a()};
11751178
std::array<float, 2> q3ft0c = {collision.q3xft0c(), collision.q3yft0c()};
1179+
std::array<float, 2> q3fv0a = {collision.q3xfv0a(), collision.q3yfv0a()};
11761180
std::array<float, 2> q3btot = {collision.q3xbtot(), collision.q3ybtot()};
11771181
std::array<float, 2> q3bpos = {collision.q3xbpos(), collision.q3ybpos()};
11781182
std::array<float, 2> q3bneg = {collision.q3xbneg(), collision.q3ybneg()};
11791183
const float eventplanes_2_for_mix[6] = {collision.ep2ft0m(), collision.ep2ft0a(), collision.ep2ft0c(), collision.ep2btot(), collision.ep2bpos(), collision.ep2bneg()};
11801184
float ep2 = eventplanes_2_for_mix[cfgEP2Estimator_for_Mix];
11811185

11821186
std::vector<std::vector<std::array<float, 2>>> qvectors = {
1183-
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 0th harmonics
1184-
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 1st harmonics
1185-
{q2ft0m, q2ft0a, q2ft0c, q2btot, q2bpos, q2bneg}, // 2nd harmonics
1186-
{q3ft0m, q3ft0a, q3ft0c, q3btot, q3bpos, q3bneg}, // 3rd harmonics
1187+
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 0th harmonics
1188+
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 1st harmonics
1189+
{q2ft0m, q2ft0a, q2ft0c, q2fv0a, q2btot, q2bpos, q2bneg}, // 2nd harmonics
1190+
{q3ft0m, q3ft0a, q3ft0c, q3fv0a, q3btot, q3bpos, q3bneg}, // 3rd harmonics
11871191
};
11881192

11891193
if (nmod == 2) {
@@ -1423,21 +1427,23 @@ struct Dilepton {
14231427
std::array<float, 2> q2ft0m = {collision.q2xft0m(), collision.q2yft0m()};
14241428
std::array<float, 2> q2ft0a = {collision.q2xft0a(), collision.q2yft0a()};
14251429
std::array<float, 2> q2ft0c = {collision.q2xft0c(), collision.q2yft0c()};
1430+
std::array<float, 2> q2fv0a = {collision.q2xfv0a(), collision.q2yfv0a()};
14261431
std::array<float, 2> q2btot = {collision.q2xbtot(), collision.q2ybtot()};
14271432
std::array<float, 2> q2bpos = {collision.q2xbpos(), collision.q2ybpos()};
14281433
std::array<float, 2> q2bneg = {collision.q2xbneg(), collision.q2ybneg()};
14291434
std::array<float, 2> q3ft0m = {collision.q3xft0m(), collision.q3yft0m()};
14301435
std::array<float, 2> q3ft0a = {collision.q3xft0a(), collision.q3yft0a()};
14311436
std::array<float, 2> q3ft0c = {collision.q3xft0c(), collision.q3yft0c()};
1437+
std::array<float, 2> q3fv0a = {collision.q3xfv0a(), collision.q3yfv0a()};
14321438
std::array<float, 2> q3btot = {collision.q3xbtot(), collision.q3ybtot()};
14331439
std::array<float, 2> q3bpos = {collision.q3xbpos(), collision.q3ybpos()};
14341440
std::array<float, 2> q3bneg = {collision.q3xbneg(), collision.q3ybneg()};
14351441

14361442
std::vector<std::vector<std::array<float, 2>>> qvectors = {
1437-
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 0th harmonics
1438-
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 1st harmonics
1439-
{q2ft0m, q2ft0a, q2ft0c, q2btot, q2bpos, q2bneg}, // 2nd harmonics
1440-
{q3ft0m, q3ft0a, q3ft0c, q3btot, q3bpos, q3bneg}, // 3rd harmonics
1443+
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 0th harmonics
1444+
{{999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}, {999.f, 999.f}}, // 1st harmonics
1445+
{q2ft0m, q2ft0a, q2ft0c, q2fv0a, q2btot, q2bpos, q2bneg}, // 2nd harmonics
1446+
{q3ft0m, q3ft0a, q3ft0c, q3fv0a, q3btot, q3bpos, q3bneg}, // 3rd harmonics
14411447
};
14421448

14431449
if (!fEMEventCut.IsSelected(collision)) {

0 commit comments

Comments
 (0)