Skip to content

Commit

Permalink
Adding Neutron Proton correlation task in PWGCF/TwoParticleCorrelatio…
Browse files Browse the repository at this point in the history 
…ns/Task
  • Loading branch information
@omassen
omassen committed Dec 4, 2024
1 parent 29c7274 commit 9d089d0
Show file tree
Hide file tree
Showing 2 changed files with 256 additions and 0 deletions.
5 changes: 5 additions & 0 deletions PWGCF/TwoParticleCorrelations/Tasks/CMakeLists.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,11 @@ o2physics_add_dpl_workflow(corrsparses
PUBLIC_LINK_LIBRARIES O2::Framework O2Physics::AnalysisCore O2Physics::PWGCFCore
COMPONENT_NAME Analysis)

o2physics_add_dpl_workflow(neutronszdc
SOURCES neutronSzdc.cxx
PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
COMPONENT_NAME Analysis)




251 changes: 251 additions & 0 deletions PWGCF/TwoParticleCorrelations/Tasks/neutronSzdc.cxx
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,251 @@
// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
///
/// \file dptDptEfficiencyAndQc.cxx
/// \brief Correlations between protons and neutrons in the ZDC
/// \author Olaf Massen <[email protected]>

//| o2-analysis-pid-tpc ${OPTION} | o2-analysis-pid-tpc-base ${OPTION} | o2-analysis-pid-tof ${OPTION} | o2-analysis-pid-tof-base ${OPTION}

#include "Framework/runDataProcessing.h"
#include "Framework/AnalysisTask.h"
#include "Framework/ASoAHelpers.h"

#include "Common/DataModel/TrackSelectionTables.h"
#include "Common/DataModel/EventSelection.h"
#include "Common/DataModel/PIDResponse.h"
#include "Common/DataModel/Centrality.h"
#include "Common/DataModel/Multiplicity.h"

using namespace o2;
using namespace o2::framework;
using namespace o2::framework::expressions;

struct neutronSzdc {
// Histogram registry: an object to hold your histograms
HistogramRegistry histos{"histos", {}, OutputObjHandlingPolicy::AnalysisObject};

Configurable<int> cfgNBinsZN{"cfgNBinsZN", 100, "N bins for ZNA and ZNC"};
Configurable<int> cfgNBinsZP{"cfgNBinsZP", 100, "N bins for ZPA and ZPC"};
Configurable<double> cfgZNmin{"cfgZNmin", -10, "Minimum value for ZN signal"};
Configurable<double> cfgZNmax{"cfgZNmax", 350, "Maximum value for ZN signal"};
Configurable<double> cfgZPmin{"cfgZPmin", -10, "Minimum value for ZP signal"};
Configurable<double> cfgZPmax{"cfgZPmax", 200, "Maximum value for ZP signal"};
Configurable<double> cfgDiffZNmin{"cfgdiffZmin", -20, "Minimum value for the diffZN signal"};
Configurable<double> cfgDiffZNmax{"cfgdiffZmax", 40, "Maximum value for the diffZN signal"};
Configurable<int> cfgNBinsAlpha{"cfgNBinsAlpha", 100, "Number of bins for ZDC asymmetry"};
Configurable<double> cfgAlphaZmin{"cfgAlphaZmin", -1, "Minimum value for ZDC asymmetry"};
Configurable<double> cfgAlphaZmax{"cfgAlphaZmax", 1, "Maximum value for ZDC asymmetry"};
Configurable<bool> cfgProcessRun2{"cfgProcessRun2", false, "Analyse Run 2 converted data"};
Configurable<int> cfgCentralityEstimator{"cfgCentralityEstimator", 0, "Choice of centrality estimator"};//0 for FTOC, 1 for FTOA, 2 for FTOM, 3 for FVOA.

ConfigurableAxis cfgAxisCent{"cfgAxisCent", {VARIABLE_WIDTH, 0.0f, 0.1f, 0.2f, 0.3f, 0.4f, 0.5f, 0.6f, 0.7f, 0.8f, 0.9f, 1.0f, 1.1f, 1.2f, 1.3f, 1.4f, 1.5f, 1.6f, 1.7f, 1.8f, 1.9f, 2.0f, 2.2f, 2.4f, 2.6f, 2.8f, 3.0f, 3.2f, 3.4f, 3.6f, 3.8f, 4.0f, 4.2f, 4.4f, 4.6f, 4.8f, 5.0f, 5.5f, 6.0f, 6.5f, 7.0f, 7.5f, 8.0f, 8.5f, 9.0f, 9.5f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f, 17.0f, 18.0f, 19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f , 25.0f, 26.0f, 27.0f, 28.0f, 29.0f, 30.0f, 31.0f, 32.0f, 33.0f, 34.0f, 35.0f, 36.0f, 37.0f, 38.0f, 39.0f, 40.0f, 41.0f, 42.0f, 43.0f, 44.0f, 45.0f, 46.0f, 47.0f, 48.0f, 49.0f, 50.0f, 51.0f, 52.0f, 53.0f, 54.0f, 55.0f, 56.0f, 57.0f, 58.0f, 59.0f, 60.0f, 61.0f, 62.0f, 63.0f, 64.0f, 65.0f, 66.0f, 67.0f, 68.0f, 69.0f, 70.0f, 71.0f, 72.0f, 73.0f, 74.0f, 75.0f, 76.0f, 77.0f, 78.0f, 79.0f, 80.0f, 81.0f, 82.0f, 83.0f, 84.0f, 85.0f, 86.0f, 87.0f, 88.0f, 89.0f, 90.0f, 91.0f, 92.0f, 93.0f, 94.0f, 95.0f, 96.0f, 97.0f, 98.0f, 99.0f, 100.0f}, "Centrality [%]"};

Filter collisionVtxZ = nabs(aod::collision::posZ) < 10.f;

using CentralitiesRun3 = aod::CentFT0Cs;
using CentralitiesRun2 = aod::CentRun2V0Ms;
using BCsRun3 = soa::Join<aod::BCs, aod::Timestamps, aod::BcSels, aod::Run3MatchedToBCSparse>;

void init(InitContext const&)
{
// define axes you want to use
const AxisSpec axisCounter{4, 0, +2, ""};
const AxisSpec axisZNSectorSignal{cfgNBinsZN,cfgZNmin,cfgZNmax/3.};
const AxisSpec axisZPSectorSignal{cfgNBinsZP,cfgZPmin,cfgZPmax/3.};
const AxisSpec axisZNASignal{cfgNBinsZN,cfgZNmin,cfgZNmax, "ZNA (a.u.)"};
const AxisSpec axisZNCSignal{cfgNBinsZN,cfgZNmin,cfgZNmax, "ZNC (a.u.)"};
const AxisSpec axisZPASignal{cfgNBinsZP,cfgZPmin,cfgZPmax, "ZPA (a.u.)"};
const AxisSpec axisZPCSignal{cfgNBinsZP,cfgZPmin,cfgZPmax, "ZPC (a.u.)"};
const AxisSpec axisZNSignal{2*cfgNBinsZN, cfgZNmin, 1.5*cfgZNmax,"ZN (a.u.)"};
const AxisSpec axisZPSignal{2*cfgNBinsZP, cfgZPmin, 1.5*cfgZPmax,"ZP (a.u.)"};
const AxisSpec axisAlphaZ{cfgNBinsAlpha,cfgAlphaZmin,cfgAlphaZmax, "#alpha_{spec}"};
const AxisSpec axisZDiffSignal{cfgNBinsZN,cfgDiffZNmin,cfgDiffZNmax,"#Delta E"};


// create histograms
histos.add("eventCounter", "eventCounter", kTH1F, {axisCounter});

histos.add("ZNASector0Signal","ZNASector0Signal",kTH1F,{axisZNSectorSignal});
histos.add("ZNASector1Signal","ZNASector1Signal",kTH1F,{axisZNSectorSignal});
histos.add("ZNASector2Signal","ZNASector2Signal",kTH1F,{axisZNSectorSignal});
histos.add("ZNASector3Signal","ZNASector3Signal",kTH1F,{axisZNSectorSignal});

histos.add("ZNCSector0Signal","ZNCSector0Signal",kTH1F,{axisZNSectorSignal});
histos.add("ZNCSector1Signal","ZNCSector1Signal",kTH1F,{axisZNSectorSignal});
histos.add("ZNCSector2Signal","ZNCSector2Signal",kTH1F,{axisZNSectorSignal});
histos.add("ZNCSector3Signal","ZNCSector3Signal",kTH1F,{axisZNSectorSignal});

histos.add("ZPASector0Signal","ZPASector0Signal",kTH1F,{axisZPSectorSignal});
histos.add("ZPASector1Signal","ZPASector1Signal",kTH1F,{axisZPSectorSignal});
histos.add("ZPASector2Signal","ZPASector2Signal",kTH1F,{axisZPSectorSignal});
histos.add("ZPASector3Signal","ZPASector3Signal",kTH1F,{axisZPSectorSignal});

histos.add("ZPCSector0Signal","ZPCSector0Signal",kTH1F,{axisZPSectorSignal});
histos.add("ZPCSector1Signal","ZPCSector1Signal",kTH1F,{axisZPSectorSignal});
histos.add("ZPCSector2Signal","ZPCSector2Signal",kTH1F,{axisZPSectorSignal});
histos.add("ZPCSector3Signal","ZPCSector3Signal",kTH1F,{axisZPSectorSignal});

histos.add("ZNASignal","ZNASignal",kTH1F,{axisZNASignal});
histos.add("ZNCSignal","ZNCSignal",kTH1F,{axisZNCSignal});
histos.add("ZNSignal", "ZNSignal", kTH1F,{axisZNSignal});
histos.add("ZPASignal","ZPASignal",kTH1F,{axisZPASignal});
histos.add("ZPCSignal","ZPCSignal",kTH1F,{axisZPCSignal});
histos.add("ZPSignal","ZPSignal", kTH1F, {axisZPSignal});

histos.add("alphaZN", "alphaZN", kTH1F, {axisAlphaZ});
histos.add("alphaZP", "alphaZP", kTH1F, {axisAlphaZ});

histos.add("diffZNASignal","diffZNASignal",kTH1F,{axisZDiffSignal});
histos.add("diffZNCSignal","diffZNCSignal",kTH1F,{axisZDiffSignal});
histos.add("diffZPASignal","diffZPASignal",kTH1F,{axisZDiffSignal});
histos.add("diffZPCSignal","diffZPCSignal",kTH1F,{axisZDiffSignal});
histos.add("diffZNSignal","diffZNSignal",kTH1F,{axisZDiffSignal});
histos.add("diffZPSignal","diffZPSignal",kTH1F,{axisZDiffSignal});

histos.add("CentralityPercentile","CentralityPercentile",kTH1F,{cfgAxisCent});

histos.add("CentvsZNASignal","FT0CvsZNASignal",kTH2F,{cfgAxisCent,axisZNASignal});
histos.add("CentvsZNCSignal","FT0CvsZNCSignal",kTH2F,{cfgAxisCent,axisZNCSignal});
histos.add("CentvsZPASignal","FT0CvsZPASignal",kTH2F,{cfgAxisCent,axisZPASignal});
histos.add("CentvsZPCSignal","FT0CvsZPCSignal",kTH2F,{cfgAxisCent,axisZPCSignal});
histos.add("CentvsZNSignal","FT0CvsZNSignal",kTH2F,{cfgAxisCent,axisZNSignal});
histos.add("CentvsZPSignal","FT0CvsZPSignal",kTH2F,{cfgAxisCent,axisZPSignal});

}

void processRun3(soa::Filtered<soa::Join< aod::Collisions, aod::EvSels, CentralitiesRun3>>::iterator const& collision , BCsRun3 const&, aod::Zdcs const&)
{
if(!collision.sel8()){
histos.fill(HIST("eventCounter"), 0.25);
return;
}
const auto& foundBC = collision.foundBC_as<BCsRun3>();
if (foundBC.has_zdc()){
const auto& zdcread = foundBC.zdc();
const auto cent = collision.centFT0C();

histos.fill(HIST("eventCounter"), 1.25);
histos.fill(HIST("ZNASignal"), zdcread.energyCommonZNA());
histos.fill(HIST("ZNCSignal"), zdcread.energyCommonZNC());
histos.fill(HIST("ZPASignal"), zdcread.energyCommonZPA());
histos.fill(HIST("ZPCSignal"), zdcread.energyCommonZPC());
histos.fill(HIST("ZNASector0Signal"),(zdcread.energySectorZNA())[0]);
histos.fill(HIST("ZNASector1Signal"),(zdcread.energySectorZNA())[1]);
histos.fill(HIST("ZNASector2Signal"),(zdcread.energySectorZNA())[2]);
histos.fill(HIST("ZNASector3Signal"),(zdcread.energySectorZNA())[3]);
histos.fill(HIST("ZNCSector0Signal"),(zdcread.energySectorZNC())[0]);
histos.fill(HIST("ZNCSector1Signal"),(zdcread.energySectorZNC())[1]);
histos.fill(HIST("ZNCSector2Signal"),(zdcread.energySectorZNC())[2]);
histos.fill(HIST("ZNCSector3Signal"),(zdcread.energySectorZNC())[3]);
histos.fill(HIST("ZPASector0Signal"),(zdcread.energySectorZPA())[0]);
histos.fill(HIST("ZPASector1Signal"),(zdcread.energySectorZPA())[1]);
histos.fill(HIST("ZPASector2Signal"),(zdcread.energySectorZPA())[2]);
histos.fill(HIST("ZPASector3Signal"),(zdcread.energySectorZPA())[3]);
histos.fill(HIST("ZPCSector0Signal"),(zdcread.energySectorZPC())[0]);
histos.fill(HIST("ZPCSector1Signal"),(zdcread.energySectorZPC())[1]);
histos.fill(HIST("ZPCSector2Signal"),(zdcread.energySectorZPC())[2]);
histos.fill(HIST("ZPCSector3Signal"),(zdcread.energySectorZPC())[3]);

float sumZNC = (zdcread.energySectorZNC())[0] + (zdcread.energySectorZNC())[1] + (zdcread.energySectorZNC())[2] + (zdcread.energySectorZNC())[3];
float sumZNA = (zdcread.energySectorZNA())[0] + (zdcread.energySectorZNA())[1] + (zdcread.energySectorZNA())[2] + (zdcread.energySectorZNA())[3];
float sumZPC = (zdcread.energySectorZPC())[0] + (zdcread.energySectorZPC())[1] + (zdcread.energySectorZPC())[2] + (zdcread.energySectorZPC())[3];
float sumZPA = (zdcread.energySectorZPA())[0] + (zdcread.energySectorZPA())[1] + (zdcread.energySectorZPA())[2] + (zdcread.energySectorZPA())[3];
float alphaZN = (sumZNA - sumZNC) / (sumZNA + sumZNC);
float alphaZP = (sumZPA - sumZPC) / (sumZPA + sumZPC);

histos.fill(HIST("alphaZN"), alphaZN);
histos.fill(HIST("alphaZP"), alphaZP);

histos.fill(HIST("diffZNASignal"),sumZNA - zdcread.energyCommonZNA());
histos.fill(HIST("diffZNCSignal"),sumZNC - zdcread.energyCommonZNC());
histos.fill(HIST("diffZPASignal"),sumZPA - zdcread.energyCommonZPA());
histos.fill(HIST("diffZPCSignal"),sumZPC - zdcread.energyCommonZPC());
histos.fill(HIST("ZNSignal"), sumZNA + sumZNC);
histos.fill(HIST("ZPSignal"), sumZPA + sumZPC);
histos.fill(HIST("diffZNSignal"), (sumZNA + sumZNC) - (zdcread.energyCommonZNA()+zdcread.energyCommonZNC()));
histos.fill(HIST("diffZPSignal"), (sumZPA + sumZPC) - (zdcread.energyCommonZPA()+zdcread.energyCommonZPC()));
histos.fill(HIST("CentralityPercentile"), cent);
histos.fill(HIST("CentvsZNASignal"), cent, sumZNA);
histos.fill(HIST("CentvsZNCSignal"), cent, sumZNC);
histos.fill(HIST("CentvsZPASignal"), cent, sumZPA);
histos.fill(HIST("CentvsZPCSignal"), cent, sumZPC);
histos.fill(HIST("CentvsZNSignal"), cent, sumZNA + sumZNC);
histos.fill(HIST("CentvsZPSignal"), cent, sumZPA + sumZPC);
}
}
PROCESS_SWITCH(neutronSzdc, processRun3, "Process analysis for Run 3 data", true);

void processRun2(soa::Filtered<soa::Join<aod::Collisions, aod::EvSels, aod::Run2MatchedSparse, CentralitiesRun2>>::iterator const& collision ,aod::BCsWithTimestamps const&, aod::Zdcs const&)
{
if(!collision.alias_bit(kINT7)){
histos.fill(HIST("eventCounter"), 0.25);
return;
}

if (collision.has_zdc()){
const auto& zdcread = collision.zdc();
const auto cent = collision.centRun2V0M();
histos.fill(HIST("eventCounter"), 1.25);
histos.fill(HIST("ZNASignal"), zdcread.energyCommonZNA());
histos.fill(HIST("ZNCSignal"), zdcread.energyCommonZNC());
histos.fill(HIST("ZPASignal"), zdcread.energyCommonZPA());
histos.fill(HIST("ZPCSignal"), zdcread.energyCommonZPC());
histos.fill(HIST("ZNASector0Signal"),(zdcread.energySectorZNA())[0]);
histos.fill(HIST("ZNASector1Signal"),(zdcread.energySectorZNA())[1]);
histos.fill(HIST("ZNASector2Signal"),(zdcread.energySectorZNA())[2]);
histos.fill(HIST("ZNASector3Signal"),(zdcread.energySectorZNA())[3]);
histos.fill(HIST("ZNCSector0Signal"),(zdcread.energySectorZNC())[0]);
histos.fill(HIST("ZNCSector1Signal"),(zdcread.energySectorZNC())[1]);
histos.fill(HIST("ZNCSector2Signal"),(zdcread.energySectorZNC())[2]);
histos.fill(HIST("ZNCSector3Signal"),(zdcread.energySectorZNC())[3]);
histos.fill(HIST("ZPASector0Signal"),(zdcread.energySectorZPA())[0]);
histos.fill(HIST("ZPASector1Signal"),(zdcread.energySectorZPA())[1]);
histos.fill(HIST("ZPASector2Signal"),(zdcread.energySectorZPA())[2]);
histos.fill(HIST("ZPASector3Signal"),(zdcread.energySectorZPA())[3]);
histos.fill(HIST("ZPCSector0Signal"),(zdcread.energySectorZPC())[0]);
histos.fill(HIST("ZPCSector1Signal"),(zdcread.energySectorZPC())[1]);
histos.fill(HIST("ZPCSector2Signal"),(zdcread.energySectorZPC())[2]);
histos.fill(HIST("ZPCSector3Signal"),(zdcread.energySectorZPC())[3]);
float sumZNC = (zdcread.energySectorZNC())[0] + (zdcread.energySectorZNC())[1] + (zdcread.energySectorZNC())[2] + (zdcread.energySectorZNC())[3];
float sumZNA = (zdcread.energySectorZNA())[0] + (zdcread.energySectorZNA())[1] + (zdcread.energySectorZNA())[2] + (zdcread.energySectorZNA())[3];
float sumZPC = (zdcread.energySectorZPC())[0] + (zdcread.energySectorZPC())[1] + (zdcread.energySectorZPC())[2] + (zdcread.energySectorZPC())[3];
float sumZPA = (zdcread.energySectorZPA())[0] + (zdcread.energySectorZPA())[1] + (zdcread.energySectorZPA())[2] + (zdcread.energySectorZPA())[3];
float alphaZN = (sumZNA - sumZNC) / (sumZNA + sumZNC);
float alphaZP = (sumZPA - sumZPC) / (sumZPA + sumZPC);
histos.fill(HIST("alphaZN"), alphaZN);
histos.fill(HIST("alphaZP"), alphaZP);

histos.fill(HIST("diffZNASignal"),sumZNA - zdcread.energyCommonZNA());
histos.fill(HIST("diffZNCSignal"),sumZNC - zdcread.energyCommonZNC());
histos.fill(HIST("diffZPASignal"),sumZPA - zdcread.energyCommonZPA());
histos.fill(HIST("diffZPCSignal"),sumZPC - zdcread.energyCommonZPC());
histos.fill(HIST("ZNSignal"), sumZNA + sumZNC);
histos.fill(HIST("ZPSignal"), sumZPA + sumZPC);
histos.fill(HIST("diffZNSignal"), (sumZNA + sumZNC) - (zdcread.energyCommonZNA()+zdcread.energyCommonZNC()));
histos.fill(HIST("diffZPSignal"), (sumZPA + sumZPC) - (zdcread.energyCommonZPA()+zdcread.energyCommonZPC()));
histos.fill(HIST("CentralityPercentile"), cent);
histos.fill(HIST("CentvsZNASignal"), cent, sumZNA);
histos.fill(HIST("CentvsZNCSignal"), cent, sumZNC);
histos.fill(HIST("CentvsZPASignal"), cent, sumZPA);
histos.fill(HIST("CentvsZPCSignal"), cent, sumZPC);
histos.fill(HIST("CentvsZNSignal"), cent, sumZNA + sumZNC);
histos.fill(HIST("CentvsZPSignal"), cent, sumZPA + sumZPC);
}
}
PROCESS_SWITCH(neutronSzdc, processRun2, "Process analysis for Run 2 converted data", false);
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
{
return WorkflowSpec{
adaptAnalysisTask<neutronSzdc>(cfgc)};
}

0 comments on commit 9d089d0

Please sign in to comment.