Skip to content

[PWGCF] Please consider the following formatting changes to #16632 #18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
alibuild wants to merge 1 commit into from
Closed

[PWGCF] Please consider the following formatting changes to #16632 #18

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
165 changes: 88 additions & 77 deletions PWGCF/EbyEFluctuations/Tasks/netchargeFluctuations.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ struct NetchargeFluctuations {
Configurable<bool> cfgLoadEff{"cfgLoadEff", true, "Load efficiency"};
Configurable<bool> cfgEffNue{"cfgEffNue", false, "efficiency correction to nu_dyn"};

Configurable<int> cfgCentEstimator{ "cfgCentEstimator", 0, "0=FT0C, 1=FT0A, 2=FT0M"};
Configurable<int> cfgCentEstimator{"cfgCentEstimator", 0, "0=FT0C, 1=FT0A, 2=FT0M"};
Configurable<int> cut05{"cut05", 155, "0-5% boundary"};
Configurable<int> cut10{"cut10", 130, "5-10% boundary"};
Configurable<int> cut20{"cut20", 97, "10-20% boundary"};
Expand Down Expand Up @@ -262,11 +262,10 @@ struct NetchargeFluctuations {
histogramRegistry.add("QA/hCentrality", "", kTH1F, {centAxis});
histogramRegistry.add("QA/hMultiplicity", "", kTH1F, {multAxis});

histogramRegistry.add("cent/multFT0C", "", kTH1F, {nchAxis});
histogramRegistry.add("cent/multFT0C", "", kTH1F, {nchAxis});
histogramRegistry.add("cent/multFT0A", "", kTH1F, {nchAxis});
histogramRegistry.add("cent/multFT0M", "", kTH1F, {nchAxis});


histogramRegistry.add("gen/hVtxZ_before", "", kTH1F, {vtxzAxis});
histogramRegistry.add("gen/hVtxZ_after", "", kTH1F, {vtxzAxis});
histogramRegistry.add("gen/hPt", "", kTH1F, {ptAxis});
Expand Down Expand Up @@ -597,12 +596,12 @@ struct NetchargeFluctuations {
int binx = hEff->GetXaxis()->FindBin(pt);
int biny = hEff->GetYaxis()->FindBin(eta);

if (binx < 1 || binx > hEff->GetNbinsX() ||
biny < 1 || biny > hEff->GetNbinsY()) {
return -1;
}
if (binx < 1 || binx > hEff->GetNbinsX() ||
biny < 1 || biny > hEff->GetNbinsY()) {
return -1;
}

double eff = hEff->GetBinContent(binx, biny);
double eff = hEff->GetBinContent(binx, biny);
return eff;
}

Expand Down Expand Up @@ -779,7 +778,7 @@ struct NetchargeFluctuations {
nch += 1;
fillAfterQA(track);

if (track.sign() == 1) {
if (track.sign() == 1) {
histogramRegistry.fill(HIST("eff/hPt_np"), track.pt());
histogramRegistry.fill(HIST("eff/hPt_hEta_np"), track.pt(), track.eta());
} else if (track.sign() == -1) {
Expand Down Expand Up @@ -860,7 +859,7 @@ struct NetchargeFluctuations {
if ((mcpart.pt() <= ptMinCut) || (mcpart.pt() >= ptMaxCut)) {
continue;
}
if (sign == 1) {
if (sign == 1) {
histogramRegistry.fill(HIST("eff/hPt_np_gen"), mcpart.pt());
histogramRegistry.fill(HIST("eff/hPt_hEta_np_gen"), mcpart.pt(), mcpart.eta());
} else if (sign == -1) {
Expand Down Expand Up @@ -1055,7 +1054,7 @@ struct NetchargeFluctuations {
continue;

histogramRegistry.fill(HIST("data/delta_eta_eta"), eta);
double eff = getEfficiency(track.pt(), track.eta(), track.sign());
double eff = getEfficiency(track.pt(), track.eta(), track.sign());

if (eff < threshold)
continue;
Expand Down Expand Up @@ -1178,75 +1177,86 @@ struct NetchargeFluctuations {

} // void


int getCentrality(aod::McParticles const& particles)
{
int multFT0A = 0;
int multFT0C = 0;
{
int multFT0A = 0;
int multFT0C = 0;

for (auto const& part : particles) {
for (auto const& part : particles) {

if (!part.isPhysicalPrimary()) {
continue;
}
if (!part.isPhysicalPrimary()) {
continue;
}

// FT0C
if (part.eta() > ft0Cmin && part.eta() < ft0Cmax) {
multFT0C++;
}
// FT0C
if (part.eta() > ft0Cmin && part.eta() < ft0Cmax) {
multFT0C++;
}

// FT0A
if (part.eta() > ft0Amin && part.eta() < ft0Amax) {
multFT0A++;
// FT0A
if (part.eta() > ft0Amin && part.eta() < ft0Amax) {
multFT0A++;
}
}
}


int multFT0M = multFT0A + multFT0C;
//LOGF(info, "multFT0C = %d", multFT0C);
histogramRegistry.fill(HIST("cent/multFT0C"), multFT0C);
histogramRegistry.fill(HIST("cent/multFT0A"), multFT0A);
histogramRegistry.fill(HIST("cent/multFT0M"), multFT0M);

int multEstimator = 0;

if (cfgCentEstimator == 0) { multEstimator = multFT0C; }
else if (cfgCentEstimator == 1) { multEstimator = multFT0A; }
else { multEstimator = multFT0M; }

// centrality cuts
if (multEstimator >= cut05) return 2.5;
if (multEstimator >= cut10) return 7.5;
if (multEstimator >= cut20) return 15;
if (multEstimator >= cut30) return 25;
if (multEstimator >= cut40) return 35;
if (multEstimator >= cut50) return 45;
if (multEstimator >= cut60) return 55;
if (multEstimator >= cut70) return 65;
if (multEstimator >= cut80) return 75;
if (multEstimator >= cut90) return 85;



return -1;
}
int multFT0M = multFT0A + multFT0C;
// LOGF(info, "multFT0C = %d", multFT0C);
histogramRegistry.fill(HIST("cent/multFT0C"), multFT0C);
histogramRegistry.fill(HIST("cent/multFT0A"), multFT0A);
histogramRegistry.fill(HIST("cent/multFT0M"), multFT0M);

int multEstimator = 0;

void mcPredictionsCent(aod::McCollision const& collision, aod::McParticles const& particles)
{
(void)collision;
int posGen = 0, negGen = 0, posNegGen = 0, termNGen = 0, termPGen = 0, nchGen = 0;
int cent = getCentrality(particles);
if (cfgCentEstimator == 0) {
multEstimator = multFT0C;
} else if (cfgCentEstimator == 1) {
multEstimator = multFT0A;
} else {
multEstimator = multFT0M;
}

if (cent < 0) {
return;
// centrality cuts
if (multEstimator >= cut05)
return 2.5;
if (multEstimator >= cut10)
return 7.5;
if (multEstimator >= cut20)
return 15;
if (multEstimator >= cut30)
return 25;
if (multEstimator >= cut40)
return 35;
if (multEstimator >= cut50)
return 45;
if (multEstimator >= cut60)
return 55;
if (multEstimator >= cut70)
return 65;
if (multEstimator >= cut80)
return 75;
if (multEstimator >= cut90)
return 85;

return -1;
}

for (auto const& mcpart : particles) {
void mcPredictionsCent(aod::McCollision const& collision, aod::McParticles const& particles)
{
(void)collision;
int posGen = 0, negGen = 0, posNegGen = 0, termNGen = 0, termPGen = 0, nchGen = 0;
int cent = getCentrality(particles);

if (cent < 0) {
return;
}

if (!mcpart.isPhysicalPrimary()) { continue; }
int pid = mcpart.pdgCode();
auto sign = 0;
for (auto const& mcpart : particles) {

if (!mcpart.isPhysicalPrimary()) {
continue;
}
int pid = mcpart.pdgCode();
auto sign = 0;
auto* pd = pdgService->GetParticle(pid);
if (pd != nullptr) {
sign = pd->Charge() / 3.;
Expand All @@ -1260,9 +1270,13 @@ int cent = getCentrality(particles);
std::abs(pid) != kProton)
continue;

if (std::fabs(mcpart.eta()) >= etaCut) { continue; }
if (std::fabs(mcpart.eta()) >= etaCut) {
continue;
}

if (mcpart.pt() <= ptMinCut || mcpart.pt() >= ptMaxCut) { continue; }
if (mcpart.pt() <= ptMinCut || mcpart.pt() >= ptMaxCut) {
continue;
}

histogramRegistry.fill(HIST("gen/hPt"), mcpart.pt());
histogramRegistry.fill(HIST("gen/cent_hPt"), cent, mcpart.pt());
Expand All @@ -1278,8 +1292,8 @@ int cent = getCentrality(particles);
if (sign == -1) {
negGen += 1;
}
}
//LOGF(info, "cent = %d nch = %d", cent, nchGen);
}
// LOGF(info, "cent = %d nch = %d", cent, nchGen);
termPGen = posGen * (posGen - 1);
termNGen = negGen * (negGen - 1);
posNegGen = posGen * negGen;
Expand All @@ -1303,13 +1317,13 @@ int cent = getCentrality(particles);
histogramRegistry.fill(HIST("subsample/gen/pos_sq"), cent, sampleIndex, posGen * posGen);
histogramRegistry.fill(HIST("subsample/gen/neg_sq"), cent, sampleIndex, negGen * negGen);
histogramRegistry.fill(HIST("subsample/gen/posneg"), cent, sampleIndex, posNegGen);
}//void
} // void

void mcPredictionsDeltaEta(aod::McCollision const& collision, aod::McParticles const& particles, float deta1, float deta2)
{
(void)collision;

float deltaEtaWidth = deta2 - deta1+ 1e-5f;
float deltaEtaWidth = deta2 - deta1 + 1e-5f;

int posGen = 0, negGen = 0, posNegGen = 0, termNGen = 0, termPGen = 0, nchGen = 0;
for (const auto& mcpart : particles) {
Expand Down Expand Up @@ -1377,7 +1391,6 @@ int cent = getCentrality(particles);

} // void


SliceCache cache;
Preslice<aod::McParticles> mcTrack = aod::mcparticle::mcCollisionId;

Expand Down Expand Up @@ -1438,19 +1451,17 @@ int cent = getCentrality(particles);

PROCESS_SWITCH(NetchargeFluctuations, processMcRun2, "Process reconstructed", false);

void processMcPrediction(aod::McCollision const& collision, aod::McParticles const& particles)
void processMcPrediction(aod::McCollision const& collision, aod::McParticles const& particles)
{
mcPredictionsCent(collision, particles);
for (int ii = 0; ii < deltaEta; ii++) {
float etaMin = -0.1f * (ii + 1);
float etaMax = 0.1f * (ii + 1);
mcPredictionsDeltaEta(collision, particles, etaMin, etaMax);
}

}

PROCESS_SWITCH(NetchargeFluctuations, processMcPrediction, "Process Prediction", false);

};

// struct
Expand Down
Loading