Skip to content

[PWGJE] Please consider the following formatting changes to #9999 #35

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

Merged
smaff92 merged 1 commit into from
Feb 18, 2025
Merged

[PWGJE] Please consider the following formatting changes to #9999 #35

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
185 changes: 91 additions & 94 deletions PWGJE/Tasks/statPromptPhoton.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
/// \author Adrian Fereydon Nassirpour <adrian.fereydon.nassirpour@cern.ch>
#include <CCDB/BasicCCDBManager.h>

#include <iostream>

Check warning on line 19 in PWGJE/Tasks/statPromptPhoton.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[include-iostream] 

Including iostream is discouraged. Use O2 logging instead.
#include <vector>
#include <string>

Expand Down Expand Up @@ -260,14 +260,13 @@
using filteredMCCollisions = soa::Filtered<selectedMCCollisions>;
*/


Filter PosZFilter_JE = nabs(aod::jcollision::posZ) < cfgVtxCut;
Filter clusterDefinitionSelection_JE = (o2::aod::jcluster::definition == cfgClusterDefinition) && (o2::aod::jcluster::time >= cfgMinTime) && (o2::aod::jcluster::time <= cfgMaxTime) && (o2::aod::jcluster::energy > cfgMinClusterEnergy) && (o2::aod::jcluster::nCells >= cfgMinNCells) && (o2::aod::jcluster::nlm <= cfgMaxNLM) && (o2::aod::jcluster::isExotic == cfgExoticContribution);

using selectedMCCollisions = aod::JMcCollisions;
using filteredMCCollisions = soa::Filtered<selectedMCCollisions>;
using filteredMCCollisions = soa::Filtered<selectedMCCollisions>;
using TrackCandidates = soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::TrackSelection>;
using BcCandidates = soa::Join<aod::JBCs, aod::JBCPIs>;
using BcCandidates = soa::Join<aod::JBCs, aod::JBCPIs>;

// using jTrackCandidates = soa::Join<aod::JTracks, aod::JTrackPIs, aod::McTrackLabels>;
using jTrackCandidates = soa::Join<aod::JTracks, aod::JTrackPIs, aod::JMcTrackLbs>;
Expand Down Expand Up @@ -433,13 +432,13 @@
void processMCGen(filteredMCCollisions::iterator const& collision, soa::SmallGroups<soa::Join<aod::JMcCollisionLbs, jfilteredCollisions>> const& recocolls, aod::JMcParticles const& mcParticles, jfilteredMCClusters const&)
{
nEventsGenMC++;
if(cfgDebug){
if (cfgDebug) {
if ((nEventsGenMC + 1) % 10000 == 0) {
std::cout << "Processed Gen MC Events: " << nEventsGenMC << std::endl;
std::cout << "Processed Gen MC Events: " << nEventsGenMC << std::endl;
}
}
histos.fill(HIST("GEN_nEvents"), 0.5);
if (fabs(collision.posZ()) > cfgVtxCut)

Check warning on line 441 in PWGJE/Tasks/statPromptPhoton.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
return;

if (recocolls.size() <= 0) // not reconstructed
Expand All @@ -447,7 +446,7 @@
for (auto& recocoll : recocolls) { // poorly reconstructed
if (!recocoll.sel8())
return;
if (fabs(recocoll.posZ()) > cfgVtxCut)

Check warning on line 449 in PWGJE/Tasks/statPromptPhoton.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.

return;
histos.fill(HIST("GEN_nEvents"), 1.5);
Expand All @@ -465,7 +464,7 @@
continue;
if (std::abs(mcPhoton.eta()) > cfgtrkMaxEta)
continue;
double pdgcode = fabs(mcPhoton.pdgCode());

Check warning on line 467 in PWGJE/Tasks/statPromptPhoton.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
if (mcPhoton.isPhysicalPrimary()) {
if (pdgcode == 211 || pdgcode == 321 || pdgcode == 2212 || pdgcode == 11) {
histos.fill(HIST("GEN_Particle_Pt"), mcPhoton.pt());
Expand All @@ -481,7 +480,7 @@
bool sterntrigger = false;
double sternPt = 0.0;
if (mcPhoton.pt() > cfgMinTrig && mcPhoton.pt() < cfgMaxTrig) {
if (fabs(mcPhoton.eta()) <= cfgtrkMaxEta) {

Check warning on line 483 in PWGJE/Tasks/statPromptPhoton.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
sterntrigger = true;
sternPt = mcPhoton.pt();
}
Expand Down Expand Up @@ -509,7 +508,7 @@
if (mcPhoton.pdgCode() == 22) {
histos.fill(HIST("GEN_True_Photon_Energy"), mcPhoton.e());
if (mcPhoton.pt() > cfgMinTrig && mcPhoton.pt() < cfgMaxTrig) {
if (fabs(mcPhoton.eta()) <= cfgtrkMaxEta) {

Check warning on line 511 in PWGJE/Tasks/statPromptPhoton.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
photontrigger = true;
}
} // check for photon trigger
Expand Down Expand Up @@ -568,25 +567,25 @@
} // 3rd photon loop
} // 2nd photon check

if(cfgDebug){
std::cout << "We have a GEN prompt photon" << std::endl;
std::cout << "Photon gen status code chain: " << std::endl;
std::cout << "Photon stat: " << mcPhoton.getGenStatusCode() << std::endl;
std::cout << "Photon index: " << mcPhoton.globalIndex() << std::endl;
std::cout << "Photon mompdg 1: " << mompdg1 << std::endl;
std::cout << "Photon momstatus 1: " << momstatus1 << std::endl;
std::cout << "Photon momindex 1: " << momindex1 << std::endl;
std::cout << "Photon mompdg 2: " << mompdg2 << std::endl;
std::cout << "Photon momstatus 2: " << momstatus2 << std::endl;
std::cout << "Photon momindex 2: " << momindex2 << std::endl;
std::cout << "Photon mompdg 3: " << mompdg3 << std::endl;
std::cout << "Photon momstatus 3: " << momstatus3 << std::endl;
std::cout << "Photon momindex 3: " << momindex3 << std::endl;
}
if (cfgDebug) {
std::cout << "We have a GEN prompt photon" << std::endl;
std::cout << "Photon gen status code chain: " << std::endl;
std::cout << "Photon stat: " << mcPhoton.getGenStatusCode() << std::endl;
std::cout << "Photon index: " << mcPhoton.globalIndex() << std::endl;
std::cout << "Photon mompdg 1: " << mompdg1 << std::endl;
std::cout << "Photon momstatus 1: " << momstatus1 << std::endl;
std::cout << "Photon momindex 1: " << momindex1 << std::endl;
std::cout << "Photon mompdg 2: " << mompdg2 << std::endl;
std::cout << "Photon momstatus 2: " << momstatus2 << std::endl;
std::cout << "Photon momindex 2: " << momindex2 << std::endl;
std::cout << "Photon mompdg 3: " << mompdg3 << std::endl;
std::cout << "Photon momstatus 3: " << momstatus3 << std::endl;
std::cout << "Photon momindex 3: " << momindex3 << std::endl;
}
} // check for mother of OG
else {
continue;
}
else {
continue;
}

if (std::abs(mcPhoton.getGenStatusCode()) > 19 && std::abs(mcPhoton.getGenStatusCode()) < 90) {
if (mcPhoton.isPhysicalPrimary()) {
Expand All @@ -613,15 +612,15 @@
{

nEventsRecMC_JE++;
if(cfgDebug){
if (cfgDebug) {
if ((nEventsRecMC_JE + 1) % 10000 == 0) {
std::cout << "Processed JE Rec MC Events: " << nEventsRecMC_JE << std::endl;
std::cout << "Processed JE Rec MC Events: " << nEventsRecMC_JE << std::endl;
}
}
histos.fill(HIST("REC_nEvents"), 0.5);

// required cuts
if (fabs(collision.posZ()) > cfgVtxCut)

Check warning on line 623 in PWGJE/Tasks/statPromptPhoton.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
return;
if (!collision.sel8())
return;
Expand Down Expand Up @@ -662,7 +661,7 @@
continue;
if (mccluster.energy() > cfgHighClusterE)
continue;
if (fabs(mccluster.eta()) > cfgtrkMaxEta)

Check warning on line 664 in PWGJE/Tasks/statPromptPhoton.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
continue;

histos.fill(HIST("REC_Cluster_QA"), 0.5);
Expand Down Expand Up @@ -715,35 +714,34 @@
continue;
}

// bool nodoublecount = false;
// bool nodoublecount = false;
double etaT = ogtrack.trackEtaEmcal();
double etaC = mccluster.eta();
double phiT = ogtrack.trackPhiEmcal();
double phiC = mccluster.phi();
double ptT = ctrack.pt();
double ptT = ctrack.pt();
bool etatrigger = false;
bool phitrigger = false;
bool phitrigger = false;
double phidiff = TVector2::Phi_mpi_pi(mccluster.phi() - ogtrack.phi());
double etadiff = mccluster.eta() - ogtrack.eta();

if(cfgPtClusterCut){
if(fabs(etaT - etaC) < (0.010 + pow(ptT+4.07,-2.5)) ) {
if (cfgPtClusterCut) {
if (fabs(etaT - etaC) < (0.010 + pow(ptT + 4.07, -2.5))) {

Check warning on line 729 in PWGJE/Tasks/statPromptPhoton.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
etatrigger = true;
}
}

if(fabs(TVector2::Phi_mpi_pi(phiT - phiC)) < (0.015 + pow(ptT+3.65,-2.0)) ) {
if (fabs(TVector2::Phi_mpi_pi(phiT - phiC)) < (0.015 + pow(ptT + 3.65, -2.0))) {

Check warning on line 733 in PWGJE/Tasks/statPromptPhoton.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[std-prefix] 

Use std:: prefix for names from the std namespace.
phitrigger = true;
}
}
else {
if (fabs(etadiff) < 0.05) {
etatrigger = true;
}

if (fabs(phidiff) < 0.05) {
phitrigger = true;
}
}
}
} else {
if (fabs(etadiff) < 0.05) {
etatrigger = true;
}

if (fabs(phidiff) < 0.05) {
phitrigger = true;
}
}

if (etatrigger && phitrigger) {
chargetrigger = true;
Expand Down Expand Up @@ -792,29 +790,29 @@
if (!ogtrack.isGlobalTrack()) {
continue;
}
bool etatrigger = false;
bool phitrigger = false;
bool etatrigger = false;
bool phitrigger = false;
// double ptT = ctrack.pt();
double phidiff = TVector2::Phi_mpi_pi(mccluster.phi() - ogtrack.phi());
double etadiff = mccluster.eta() - ogtrack.eta();
if (fabs(etadiff) < 0.05) {
etatrigger = true;
}
double phidiff = TVector2::Phi_mpi_pi(mccluster.phi() - ogtrack.phi());
double etadiff = mccluster.eta() - ogtrack.eta();
if (fabs(etadiff) < 0.05) {
etatrigger = true;
}

if (fabs(phidiff) < 0.05) {
phitrigger = true;
}
if (fabs(phidiff) < 0.05) {
phitrigger = true;
}

if (chargetrigger) {
histos.fill(HIST("REC_Track_v_Cluster_Phi_C"), phidiff);
histos.fill(HIST("REC_Track_v_Cluster_Eta_C"), etadiff);
histos.fill(HIST("REC_Track_v_Cluster_Phi_Eta_C"), phidiff, etadiff);
} else {
if (etatrigger && chargetrigger) {
if(cfgDebug){
std::cout << "????????????????????" << std::endl;
}
}
if (etatrigger && chargetrigger) {
if (cfgDebug) {
std::cout << "????????????????????" << std::endl;
}
}
histos.fill(HIST("REC_Track_v_Cluster_Phi_AC"), phidiff);
histos.fill(HIST("REC_Track_v_Cluster_Eta_AC"), etadiff);
histos.fill(HIST("REC_Track_v_Cluster_Phi_Eta_AC"), phidiff, etadiff);
Expand All @@ -833,7 +831,7 @@
histos.fill(HIST("REC_PtHadSum_Photon"), pthadsum);
histos.fill(HIST("REC_Trigger_Energy"), mccluster.energy());
}

auto ClusterParticles = mccluster.mcParticles_as<aod::JMcParticles>();

// now we check the realness of our prompt photons
Expand Down Expand Up @@ -952,12 +950,12 @@
if (std::fabs(mom1) > 40 && std::fabs(mom1) > 0)
continue;

if(cfgDebug){
std::cout<<"We have a REC prompt photon"<<std::endl;
std::cout<<"Photon gen status code: "<<clusterparticle.getGenStatusCode()<<std::endl;
std::cout<<"Photon mom 1: "<<mom1<<std::endl;
std::cout<<"Photon mom 2: "<<mom2<<std::endl;
}
if (cfgDebug) {
std::cout << "We have a REC prompt photon" << std::endl;
std::cout << "Photon gen status code: " << clusterparticle.getGenStatusCode() << std::endl;
std::cout << "Photon mom 1: " << mom1 << std::endl;
std::cout << "Photon mom 2: " << mom2 << std::endl;
}
if (std::abs(clusterparticle.getGenStatusCode()) > 19 && std::abs(clusterparticle.getGenStatusCode()) < 90) {
histos.fill(HIST("REC_True_Prompt_Trigger_Energy"), clusterparticle.e());
TLorentzVector lRealPhoton;
Expand All @@ -969,12 +967,12 @@
} // photon check
} // clusterparticle loop
/*
if(chPe>0){
if(photontrigger){
if(chPe/mccluster.energy() < 0.50){
goodgentrigger=true;
}
}
if(chPe>0){
if(photontrigger){
if(chPe/mccluster.energy() < 0.50){
goodgentrigger=true;
}
}
*/
if (goodgentrigger && photontrigger) {
histos.fill(HIST("REC_Trigger_Purity"), 0.5);
Expand Down Expand Up @@ -1094,12 +1092,12 @@
{

nEventsData++;
if(cfgDebug){
if (nEventsData==1){
std::cout << "Starting Data Processing: " << nEventsData << std::endl;
if (cfgDebug) {
if (nEventsData == 1) {
std::cout << "Starting Data Processing: " << nEventsData << std::endl;
}
if ((nEventsData + 1) % 10000 == 0) {
std::cout << "Processed Data Events: " << nEventsData << std::endl;
std::cout << "Processed Data Events: " << nEventsData << std::endl;
}
}
histos.fill(HIST("DATA_nEvents"), 0.5);
Expand Down Expand Up @@ -1208,25 +1206,24 @@
bool phitrigger = false;
double phidiff = TVector2::Phi_mpi_pi(cluster.phi() - ogtrack.phi());
double etadiff = cluster.eta() - ogtrack.eta();
if(cfgPtClusterCut){
if (fabs(etaT - etaC) < (0.010 + pow(ptT + 4.07, -2.5))) {
etatrigger = true;
}

if (fabs(TVector2::Phi_mpi_pi(phiT - phiC)) < (0.015 + pow(ptT + 3.65, -2.0))) {
phitrigger = true;
}
}
else {
if (fabs(etadiff) < 0.05) {
etatrigger = true;
}

if (fabs(phidiff) < 0.05) {
phitrigger = true;
}
}

if (cfgPtClusterCut) {
if (fabs(etaT - etaC) < (0.010 + pow(ptT + 4.07, -2.5))) {
etatrigger = true;
}

if (fabs(TVector2::Phi_mpi_pi(phiT - phiC)) < (0.015 + pow(ptT + 3.65, -2.0))) {
phitrigger = true;
}
} else {
if (fabs(etadiff) < 0.05) {
etatrigger = true;
}

if (fabs(phidiff) < 0.05) {
phitrigger = true;
}
}

if (etatrigger && phitrigger) {
chargetrigger = true;
sumptT += ptT;
Expand Down
Loading