Skip to content

Commit f675dbb

Browse files
scattaruscattaru
authored
Update taskCorrelationDsHadrons.cxx
1 parent 7f60023 commit f675dbb

File tree

1 file changed

+13
-19
lines changed

1 file changed

+13
-19
lines changed

‎PWGHF/HFC/Tasks/taskCorrelationDsHadrons.cxx‎

Lines changed: 13 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,6 @@ struct HfTaskCorrelationDsHadrons {
523523
int const poolBin = pairEntry.poolBin();
524524
int const ptBinD = o2::analysis::findBin(binsPtD, std::abs(ptD));
525525
const bool haveSameSign = ptD * ptHadron > 0.;
526-
const bool haveOppositeSign = ptD * ptHadron < 0.;
527526

528527
if (!isSelectedCandidate(ptBinD, bdtScorePrompt, bdtScoreBkg)) {
529528
continue;
@@ -544,7 +543,7 @@ struct HfTaskCorrelationDsHadrons {
544543
if (massD > signalRegionInner->at(ptBinD) && massD < signalRegionOuter->at(ptBinD)) {
545544
if (doLSpair && haveSameSign) { // like-sign pairs
546545
registry.fill(HIST("hCorrel2DVsPtSignalRegionLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
547-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
546+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
548547
registry.fill(HIST("hCorrel2DVsPtSignalRegionULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
549548
} else if (fillHistoData) { // default case
550549
registry.fill(HIST("hCorrel2DVsPtSignalRegion"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
@@ -556,7 +555,7 @@ struct HfTaskCorrelationDsHadrons {
556555
if (massD > sidebandLeftOuter->at(ptBinD) && massD < sidebandLeftInner->at(ptBinD)) {
557556
if (doLSpair && haveSameSign) { // like-sign pairs
558557
registry.fill(HIST("hCorrel2DVsPtSidebandLeftLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
559-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
558+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
560559
registry.fill(HIST("hCorrel2DVsPtSidebandLeftULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
561560
} else if (fillHistoData) { // default case
562561
registry.fill(HIST("hCorrel2DVsPtSidebandLeft"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
@@ -568,7 +567,7 @@ struct HfTaskCorrelationDsHadrons {
568567
if (massD > sidebandRightInner->at(ptBinD) && massD < sidebandRightOuter->at(ptBinD)) {
569568
if (doLSpair && haveSameSign) { // like-sign pairs
570569
registry.fill(HIST("hCorrel2DVsPtSidebandRightLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
571-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
570+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
572571
registry.fill(HIST("hCorrel2DVsPtSidebandRightULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
573572
} else if (fillHistoData) { // default case
574573
registry.fill(HIST("hCorrel2DVsPtSidebandRight"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
@@ -634,7 +633,6 @@ struct HfTaskCorrelationDsHadrons {
634633
int const ptBinD = o2::analysis::findBin(binsPtD, std::abs(ptD));
635634
bool const isPhysicalPrimary = pairEntry.isPhysicalPrimary();
636635
const bool haveSameSign = ptD * ptHadron > 0.;
637-
const bool haveOppositeSign = ptD * ptHadron < 0.;
638636

639637
if (!isSelectedCandidate(ptBinD, bdtScorePrompt, bdtScoreBkg)) {
640638
continue;
@@ -661,7 +659,7 @@ struct HfTaskCorrelationDsHadrons {
661659
if (isPhysicalPrimary) {
662660
if (doLSpair && haveSameSign) { // like-sign pairs
663661
registry.fill(HIST("hCorrel2DVsPtPhysicalPrimaryMcRecLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), statusDsPrompt, poolBin, efficiencyWeight);
664-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
662+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
665663
registry.fill(HIST("hCorrel2DVsPtPhysicalPrimaryMcRecULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), statusDsPrompt, poolBin, efficiencyWeight);
666664
} else { // default case
667665
registry.fill(HIST("hCorrel2DVsPtPhysicalPrimaryMcRec"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), statusDsPrompt, poolBin, efficiencyWeight);
@@ -703,14 +701,13 @@ struct HfTaskCorrelationDsHadrons {
703701
int const statusPromptHadron = pairEntry.trackOrigin();
704702
bool const isDsPrompt = pairEntry.isPrompt();
705703
const bool haveSameSign = ptD * ptHadron > 0.;
706-
const bool haveOppositeSign = ptD * ptHadron < 0.;
707704

708705
registry.fill(HIST("hCorrel2DVsPtMcGen"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin);
709706
registry.fill(HIST("hDeltaEtaPtIntMcGen"), deltaEta);
710707
registry.fill(HIST("hDeltaPhiPtIntMcGen"), deltaPhi);
711708
if (isDsPrompt) {
712709
registry.fill(HIST("hCorrel2DVsPtMcGenPrompt"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin);
713-
if (doULSpair && haveOppositeSign) {
710+
if (doULSpair && !haveSameSign) {
714711
registry.fill(HIST("hCorrel2DVsPtMcGenPromptULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin);
715712
}
716713
if (doLSpair && haveSameSign) {
@@ -721,7 +718,7 @@ struct HfTaskCorrelationDsHadrons {
721718
}
722719
} else {
723720
registry.fill(HIST("hCorrel2DVsPtMcGenNonPrompt"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin);
724-
if (doULSpair && haveOppositeSign) {
721+
if (doULSpair && !haveSameSign) {
725722
registry.fill(HIST("hCorrel2DVsPtMcGenNonPromptULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin);
726723
}
727724
if (doLSpair && haveSameSign) {
@@ -753,7 +750,6 @@ struct HfTaskCorrelationDsHadrons {
753750
int const poolBin = pairEntry.poolBin();
754751
int const ptBinD = o2::analysis::findBin(binsPtD, std::abs(ptD));
755752
const bool haveSameSign = ptD * ptHadron > 0.;
756-
const bool haveOppositeSign = ptD * ptHadron < 0.;
757753

758754
if (!isSelectedCandidate(ptBinD, bdtScorePrompt, bdtScoreBkg)) {
759755
continue;
@@ -774,7 +770,7 @@ struct HfTaskCorrelationDsHadrons {
774770
if (massD > signalRegionInner->at(ptBinD) && massD < signalRegionOuter->at(ptBinD)) {
775771
if (doLSpair && haveSameSign) { // like-sign pairs
776772
registry.fill(HIST("hCorrel2DVsPtSignalRegionLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
777-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
773+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
778774
registry.fill(HIST("hCorrel2DVsPtSignalRegionULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
779775
} else if (fillHistoData) { // default case
780776
registry.fill(HIST("hCorrel2DVsPtSignalRegion"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
@@ -786,7 +782,7 @@ struct HfTaskCorrelationDsHadrons {
786782
if (massD > sidebandLeftOuter->at(ptBinD) && massD < sidebandLeftInner->at(ptBinD)) {
787783
if (doLSpair && haveSameSign) { // like-sign pairs
788784
registry.fill(HIST("hCorrel2DVsPtSidebandLeftLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
789-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
785+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
790786
registry.fill(HIST("hCorrel2DVsPtSidebandLeftULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
791787
} else if (fillHistoData) { // default case
792788
registry.fill(HIST("hCorrel2DVsPtSidebandLeft"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
@@ -798,7 +794,7 @@ struct HfTaskCorrelationDsHadrons {
798794
if (massD > sidebandRightInner->at(ptBinD) && massD < sidebandRightOuter->at(ptBinD)) {
799795
if (doLSpair && haveSameSign) { // like-sign pairs
800796
registry.fill(HIST("hCorrel2DVsPtSidebandRightLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
801-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
797+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
802798
registry.fill(HIST("hCorrel2DVsPtSidebandRightULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
803799
} else if (fillHistoData) { // default case
804800
registry.fill(HIST("hCorrel2DVsPtSidebandRight"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
@@ -823,7 +819,6 @@ struct HfTaskCorrelationDsHadrons {
823819
int const poolBin = pairEntry.poolBin();
824820
int const ptBinD = o2::analysis::findBin(binsPtD, std::abs(ptD));
825821
const bool haveSameSign = ptD * ptHadron > 0.;
826-
const bool haveOppositeSign = ptD * ptHadron < 0.;
827822

828823
double efficiencyWeight = 1.;
829824
if (useHighDimHistoForEff) {
@@ -836,7 +831,7 @@ struct HfTaskCorrelationDsHadrons {
836831
if (massD > signalRegionInner->at(ptBinD) && massD < signalRegionOuter->at(ptBinD)) {
837832
if (doLSpair && haveSameSign) { // like-sign pairs
838833
registry.fill(HIST("hCorrel2DVsPtSignalRegionLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
839-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
834+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
840835
registry.fill(HIST("hCorrel2DVsPtSignalRegionULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
841836
} else if (fillHistoData) { // default case
842837
registry.fill(HIST("hCorrel2DVsPtSignalRegion"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
@@ -848,7 +843,7 @@ struct HfTaskCorrelationDsHadrons {
848843
if (massD > sidebandLeftOuter->at(ptBinD) && massD < sidebandLeftInner->at(ptBinD)) {
849844
if (doLSpair && haveSameSign) { // like-sign pairs
850845
registry.fill(HIST("hCorrel2DVsPtSidebandLeftLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
851-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
846+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
852847
registry.fill(HIST("hCorrel2DVsPtSidebandLeftULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
853848
} else if (fillHistoData) { // default case
854849
registry.fill(HIST("hCorrel2DVsPtSidebandLeft"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
@@ -860,7 +855,7 @@ struct HfTaskCorrelationDsHadrons {
860855
if (massD > sidebandRightInner->at(ptBinD) && massD < sidebandRightOuter->at(ptBinD)) {
861856
if (doLSpair && haveSameSign) { // like-sign pairs
862857
registry.fill(HIST("hCorrel2DVsPtSidebandRightLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
863-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
858+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
864859
registry.fill(HIST("hCorrel2DVsPtSidebandRightULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
865860
} else if (fillHistoData) { // default case
866861
registry.fill(HIST("hCorrel2DVsPtSidebandRight"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), poolBin, efficiencyWeight);
@@ -894,7 +889,6 @@ struct HfTaskCorrelationDsHadrons {
894889
int const ptBinD = o2::analysis::findBin(binsPtD, std::abs(ptD));
895890
bool const isPhysicalPrimary = pairEntry.isPhysicalPrimary();
896891
const bool haveSameSign = ptD * ptHadron > 0.;
897-
const bool haveOppositeSign = ptD * ptHadron < 0.;
898892

899893
if (!isSelectedCandidate(ptBinD, bdtScorePrompt, bdtScoreBkg)) {
900894
continue;
@@ -921,7 +915,7 @@ struct HfTaskCorrelationDsHadrons {
921915
if (isPhysicalPrimary) {
922916
if (doLSpair && haveSameSign) { // like-sign pairs
923917
registry.fill(HIST("hCorrel2DVsPtPhysicalPrimaryMcRecLS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), statusDsPrompt, poolBin, efficiencyWeight);
924-
} else if (doULSpair && haveOppositeSign) { // unlike-sign pairs
918+
} else if (doULSpair && !haveSameSign) { // unlike-sign pairs
925919
registry.fill(HIST("hCorrel2DVsPtPhysicalPrimaryMcRecULS"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), statusDsPrompt, poolBin, efficiencyWeight);
926920
} else { // default case
927921
registry.fill(HIST("hCorrel2DVsPtPhysicalPrimaryMcRec"), deltaPhi, deltaEta, std::abs(ptD), std::abs(ptHadron), statusDsPrompt, poolBin, efficiencyWeight);

0 commit comments

Comments
 (0)