Skip to content

[PWGHF] Update of post processing macros for hf correlation studies #12815

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
apalasciano merged 2 commits into from
Sep 2, 2025
Merged

[PWGHF] Update of post processing macros for hf correlation studies #12815

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
e311b01
Update of post processing macros for hf correlation studies
scattaru Sep 1, 2025
83b2c3b
Fix MegaLinter
scattaru Sep 1, 2025
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
1,066 changes: 1,025 additions & 41 deletions PWGHF/HFC/Macros/DhCorrelationExtraction.cxx
View file
Open in desktop

Large diffs are not rendered by default.

168 changes: 132 additions & 36 deletions PWGHF/HFC/Macros/DhCorrelationExtraction.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,6 @@
#ifndef PWGHF_HFC_MACROS_DHCORRELATIONEXTRACTION_H_
#define PWGHF_HFC_MACROS_DHCORRELATIONEXTRACTION_H_

#include <Rtypes.h>
#include <RtypesCore.h>
#include <TAttMarker.h>
#include <TDirectoryFile.h>
#include <TFile.h>
Expand All @@ -28,6 +26,9 @@
#include <TObject.h>
#include <TString.h>

#include <Rtypes.h>
#include <RtypesCore.h>

class DhCorrelationExtraction : public TObject
{

Expand All @@ -40,6 +41,10 @@ class DhCorrelationExtraction : public TObject
kME };
enum selectInvMassRegion { kSign,
kSideb };
enum selectDmesonOrigin { kPrompt,
kFD };
enum selectParticleType { kPrimaryPart,
kAllPart };

DhCorrelationExtraction(); // default constructor
DhCorrelationExtraction(const DhCorrelationExtraction& source);
Expand All @@ -51,15 +56,36 @@ class DhCorrelationExtraction : public TObject
void SetInputFilenameMass(TString filenameMass) { fFileNameMass = filenameMass; }
void SetInputFilenameSE(TString filenameSE) { fFileNameSE = filenameSE; }
void SetInputFilenameME(TString filenameME) { fFileNameME = filenameME; }
void SetInputFilenameSecPart(TString filenameSecPart) { fFileSecPartName = filenameSecPart; }
void SetInputFilenameBiasBtoD(TString filenamePromptMcRec, TString filenameNonPromptMcRec)
{
fFilePromptMcRecName = filenamePromptMcRec;
fFileNonPromptMcRecName = filenameNonPromptMcRec;
}
void SetDirNameSE(TString dirNameSE) { fDirNameSE = dirNameSE; }
void SetDirNameME(TString dirNameME) { fDirNameME = dirNameME; }
void SetDirNameSecPart(TString dirNameSecPart) { fDirSecPartName = dirNameSecPart; }
void SetMassHistoNameSgn(TString massHistoNameSgn) { fMassHistoNameSgn = massHistoNameSgn; }
void SetMassHistoNameBkg(TString massHistoNameBkg) { fMassHistoNameBkg = massHistoNameBkg; }
void SetMassHistoNameSBs(TString massHistoNameSBs) { fMassHistoNameSBs = massHistoNameSBs; }
void SetSECorrelHistoSignalName(TString correlNameSigSE) { fSECorrelSignalRegionName = correlNameSigSE; }
void SetSECorrelHistoSidebandName(TString correlNameSbSE) { fSECorrelSidebandsName = correlNameSbSE; }
void SetSECorrelHistoSidebandLeftName(TString correlNameSbSE) { fSECorrelSidebandLeftName = correlNameSbSE; }
void SetSECorrelHistoSidebandRightName(TString correlNameSbSE) { fSECorrelSidebandRightName = correlNameSbSE; }
void SetMECorrelHistoSignalName(TString correlNameSigME) { fMECorrelSignalRegionName = correlNameSigME; }
void SetMECorrelHistoSidebandName(TString correlNameSbME) { fMECorrelSidebandsName = correlNameSbME; }
void SetMECorrelHistoSidebandLeftName(TString correlNameSbME) { fMECorrelSidebandLeftName = correlNameSbME; }
void SetMECorrelHistoSidebandRightName(TString correlNameSbME) { fMECorrelSidebandRightName = correlNameSbME; }
void SetHistoSecPartName(TString histoPrimaryPartName, TString histoAllPartName)
{
fHistoPrimaryPartName = histoPrimaryPartName;
fHistoAllPartName = histoAllPartName;
}
void SetInputFilenameFDTemplate(TString filenameFDTemplate) { fFileFDTemplateName = filenameFDTemplate; }
void SetInputFilenameFDPromptFrac(TString filenameFDPromptFrac) { fFileFDPromptFracName = filenameFDPromptFrac; }
void SetInputHistoNameFDTemplatePrompt(TString hNameFDTemplatePrompt) { fHistoFDTemplatePromptName = hNameFDTemplatePrompt; }
void SetInputHistoNameFDTemplateNonPrompt(TString hNameFDTemplateNonPrompt) { fHistoFDTemplateNonPromptName = hNameFDTemplateNonPrompt; }
void SetInputHistoNameFDPromptFrac(TString hNameFDPromptFrac) { fHistoFDPromptFracName = hNameFDPromptFrac; }

// Input conditions: PtCand, PtHad, PoolBins
void SetNpools(Int_t npools) { fNpools = npools; }
Expand All @@ -72,66 +98,136 @@ class DhCorrelationExtraction : public TObject
void SetSubtractSoftPiInMEdistr(Bool_t subtractSoftPiME) { fSubtractSoftPiME = subtractSoftPiME; }
void SetBkgScaleFactor(Double_t scaleFactor) { fBkgScaleFactor = scaleFactor; }
void SetSignalYieldforNorm(Double_t sgnYield) { fSgnYieldNorm = sgnYield; }
void SetBkgYield(Double_t bkgYield) { fBkgYield = bkgYield; }
void SetSBYield(Double_t SBYield) { fSBYield = SBYield; }
void SetRebin2DcorrelHisto(Int_t rebinDeltaEta, Int_t rebinDeltaPhi)
{
fRebin2Dhisto = kTRUE;
fRebinAxisDeltaEta = rebinDeltaEta;
fRebinAxisDeltaPhi = rebinDeltaPhi;
}
void SetRebinOptions(Bool_t rebinAngCorr, Bool_t rebinFDCorr, Bool_t rebinSecPart)
{
fRebinAngCorr = rebinAngCorr;
fRebinFDCorr = rebinFDCorr;
fRebinSecPart = rebinSecPart;
}
void GetSignalAndBackgroundForNorm(Double_t PtCandMin, Double_t PtCandMax);
void NormalizeMEplot(TH2D*& histoME, TH2D*& histoMEsoftPi);
void SetDebugLevel(Int_t debug) { fDebug = debug; }
void SetDividedSidebands(Bool_t dividedSideb, Bool_t useSidebLeft, Bool_t useSidebRight)
{
fSidebandDivided = dividedSideb;
fUseSidebLeft = useSidebLeft;
fUseSidebRight = useSidebRight;
}
void SetFDSubtraction(Bool_t subtractFD) { fFDsubtraction = subtractFD; }
void SetSecPartContamination(Bool_t secPartContamination) { fSecPartContamination = secPartContamination; }
void SetCorrBiasBtoD(Bool_t corrbiasBtoD) { fCorrBiasBtoD = corrbiasBtoD; }
void SetBinCandAndHad(Int_t binCand, Int_t binHad)
{
fBinPtCand = binCand;
fBinPtHad = binHad;
}

/// Analysis methods
TH2D* GetCorrelHisto(Int_t SEorME, Int_t SorSB, Int_t pool, Double_t PtCandMin, Double_t PtCandMax, Double_t PtHadMin, Double_t PtHadMax);
TH2D* GetFDTemplateHisto(Int_t PromptOrFD, Double_t PtCandMin, Double_t PtCandMax, Double_t PtHadMin, Double_t PtHadMax);
TH1D* GetCorrelHistoSecondaryPart(Int_t PrimaryPart, Double_t PtCandMin, Double_t PtCandMax, Double_t PtHadMin, Double_t PtHadMax);
TH1D* ReflectCorrHistogram(TH1D*& histo);
TH1D* ReflectHistoRun2(TH1D* h, Double_t scale);
TH1D* EvaluateMCClosModulations(Double_t PtCandMin, Double_t PtCandMax, Double_t PtHadMin, Double_t PtHadMax);
Double_t GetFDPromptFrac(Double_t PtCandMin, Double_t PtCandMax, Double_t PtHadMin, Double_t PtHadMax);
Double_t CalculateBaseline(TH1D*& histo, Bool_t totalRange = kTRUE, Bool_t reflected = kFALSE);
Double_t CalculateBaselineError(TH1D*& histo, Bool_t totalRange = kTRUE, Bool_t reflected = kFALSE);
Bool_t ReadInputSEandME();
Bool_t ReadInputInvMass();
Bool_t ReadInputFDSubtr();
Bool_t ReadInputSecondaryPartContamination();
Bool_t ExtractCorrelations(Double_t PtCandMin, Double_t PtCandMax, Double_t PtHadMin, Double_t PtHadMax, TString codeName);
TH1D* GetCorrectedCorrHisto() { return fCorrectedCorrHisto; }
TH1D* GetCorrectedCorrHisto_BaselineSubtr() { return fCorrectedCorrHisto_BaselineSubtr; }
TH1D* GetCorrectedCorrHisto_Reflected() { return fCorrectedCorrHisto_Reflected; }
TH1D* GetCorrectedCorrHisto_Reflected_BaselineSubtr() { return fCorrectedCorrHisto_Reflected_BaselineSubtr; }

/// Histogram style
void SetTH1HistoStyle(TH1D*& histo, TString hTitle, TString hXaxisTitle, TString hYaxisTitle, Style_t markerStyle = kFullCircle, Color_t markerColor = kRed + 1, Double_t markerSize = 1.4, Color_t lineColor = kRed + 1, Int_t lineWidth = 3, Float_t hTitleXaxisOffset = 1.0, Float_t hTitleYaxisOffset = 1.0, Float_t hTitleXaxisSize = 0.060, Float_t hTitleYaxisSize = 0.060, Float_t hLabelXaxisSize = 0.060, Float_t hLabelYaxisSize = 0.060, Bool_t centerXaxisTitle = false, Bool_t centerYaxisTitle = false);
void SetTH2HistoStyle(TH2D*& histo, TString hTitle, TString hXaxisTitle, TString hYaxisTitle, TString hZaxisTitle, Float_t hTitleXaxisOffset = 1.8, Float_t hTitleYaxisOffset = 1.8, Float_t hTitleZaxisOffset = 1.2, Float_t hTitleXaxisSize = 0.060, Float_t hTitleYaxisSize = 0.060, Float_t hTitleZaxisSize = 0.060, Float_t hLabelXaxisSize = 0.060, Float_t hLabelYaxisSize = 0.060, Float_t hLabelZaxisSize = 0.060, Bool_t centerXaxisTitle = true, Bool_t centerYaxisTitle = true);

private:
TFile* fFileMass; // file containing the mass histograms
TFile* fFileSE; // file containing the Same Event (SE) output
TFile* fFileME; // file containing the Mixed Event (ME) output

TDirectoryFile* fDirMass; // TDirectory for mass histos
TDirectoryFile* fDirSE; // TDirectory for SE info
TDirectoryFile* fDirME; // TDirectory for ME info

TH1D* fCorrectedCorrHisto; // Corrected correlation histogram

DmesonSpecie fDmesonSpecies; // D meson specie
TString fDmesonLabel; // D meson label
TString fFileNameMass; // File cntaining inv. mass histograms
TString fFileNameSE; // File contaning Same Event (SE) output
TString fFileNameME; // File contaning Mixed Event (ME) output
TString fDirNameSE; // Directory in the file containing SE output
TString fDirNameME; // Directory in the file containing ME output
TString fMassHistoNameSgn; // Inv. mass histo name signal yield
TString fMassHistoNameBkg; // Inv. mass histo name background yield
TString fMassHistoNameSBs; // Inv. mass histo name sideband yield
TString fSECorrelSignalRegionName; // THnSparse name containing SE output for signal region
TString fSECorrelSidebandsName; // THnSparse name containing SE output for sideband region
TString fMECorrelSignalRegionName; // THnSparse name containing ME output for signal region
TString fMECorrelSidebandsName; // THnSparse name containing ME output for sideband region

Int_t fNpools; // number of pools used for the ME correction
Int_t fRebinAxisDeltaEta; // rebin deltaEta axis
Int_t fRebinAxisDeltaPhi; // rebin deltaPhi axis
Int_t fDebug; // debug level

Double_t fDeltaEtaMin; // deltaEta min value
Double_t fDeltaEtaMax; // deltaEta max value
TFile* fFileMass; // File containing the mass histograms
TFile* fFileSE; // File containing the Same Event (SE) output
TFile* fFileME; // File containing the Mixed Event (ME) output
TFile* fFileFDTemplate; // File containing FD angular correlation templates
TFile* fFileFDPromptFrac; // File containing prompt fraction (used fo FD subtraction)
TFile* fFileSecPart; // File containing secondary particle contaminaion teplates
TFile* fFilePromptMc; // File containing prompt ratio taken from MC Closure test study (to use for B to D bias correction)
TFile* fFileNonPromptMc; // File containing non-prompt ratio taken from MC Closure test study (to use for B to D bias correction)

TDirectoryFile* fDirMass; // TDirectory for mass histos
TDirectoryFile* fDirSE; // TDirectory for SE info
TDirectoryFile* fDirME; // TDirectory for ME info
TDirectoryFile* fDirSecPart; // TDirectory for seondary particle correction

TH1D* fCorrectedCorrHisto; // Corrected correlation histogram
TH1D* fCorrectedCorrHisto_BaselineSubtr; // Corrected correlation histogram with baseline subtracion
TH1D* fCorrectedCorrHisto_Reflected; // Corrected correlation histogram relected in azimuth
TH1D* fCorrectedCorrHisto_Reflected_BaselineSubtr; // Corrected correlation histogram reflected in azimuth with baseline subtraction

DmesonSpecie fDmesonSpecies; // D meson specie
TString fDmesonLabel; // D meson label
TString fFileNameMass; // File name containing inv. mass histograms
TString fFileNameSE; // File name contaning Same Event (SE) output
TString fFileNameME; // File name contaning Mixed Event (ME) output
TString fFileSecPartName; // File name contaning secondary particle correction output
TString fFileFDTemplateName; // File name contaning FD angular correlation templates
TString fFileFDPromptFracName; // File name contaning prompt fraction (used for FD subtraction)
TString fFilePromptMcRecName; // File name contaning prompt angular correlation (used for B to d bias correction)
TString fFileNonPromptMcRecName; // File name contaning non-prompt angular correlation (used for B to d bias correction)
TString fDirNameSE; // Directory in the file containing SE output
TString fDirNameME; // Directory in the file containing ME output
TString fDirSecPartName; // Directory in the file containing secondary particle correction output
TString fMassHistoNameSgn; // Inv. mass histo name signal yield
TString fMassHistoNameBkg; // Inv. mass histo name background yield
TString fMassHistoNameSBs; // Inv. mass histo name sideband yield
TString fSECorrelSignalRegionName; // THnSparse name containing SE output for signal region
TString fSECorrelSidebandsName; // THnSparse name containing SE output for sideband region
TString fSECorrelSidebandLeftName; // THnSparse name containing SE output for sideband left region
TString fSECorrelSidebandRightName; // THnSparse name containing SE output for sideband right region
TString fMECorrelSignalRegionName; // THnSparse name containing ME output for signal region
TString fMECorrelSidebandsName; // THnSparse name containing ME output for sideband regions
TString fMECorrelSidebandLeftName; // THnSparse name containing ME output for sideband left region
TString fMECorrelSidebandRightName; // THnSparse name containing ME output for sideband right region
TString fHistoFDTemplatePromptName; // Prompt angular correlation histogram name
TString fHistoFDTemplateNonPromptName; // FD angular correlation histogram name
TString fHistoFDPromptFracName; // Prompt fraction histogram name
TString fHistoPrimaryPartName; // Primary particle histogram (to be used for secondary particle contamination correction)
TString fHistoAllPartName; // All particle histogram (to be used for secondary particle contamination correction)

Int_t fNpools; // Number of pools used for the ME correction
Int_t fRebinAxisDeltaEta; // Rebin deltaEta axis value
Int_t fRebinAxisDeltaPhi; // Rebin deltaPhi axis value
Int_t fDebug; // Debug level
Int_t fBinPtCand; // Pt bin of the candidate
Int_t fBinPtHad; // Pt bin of the hadron

Double_t fDeltaEtaMin; // DeltaEta min value
Double_t fDeltaEtaMax; // DeltaEta max value
Double_t fBkgScaleFactor; // Bkg/SB factor to scale correlation plots obtained in the sideband region
Double_t fSgnYieldNorm; // Signal yield (used for normalize correlation plots after bkg subtraction)
Double_t fBkgYield; // Bkg yield under signal peak region
Double_t fSBYield; // Sideband yield

Bool_t fCorrectPoolsSeparately; // Possibility to do the ME correction pool-by-pool (kTRUE) or merging all pools (kFALSE)
Bool_t fSubtractSoftPiME; // Soft pion subtraction (for D0 case)
Bool_t fRebin2Dhisto; // Flag to rebin the 2D correlation plots
Bool_t fRebinAngCorr; // Rebin angular correlaion distributons (SE and ME)
Bool_t fRebinFDCorr; // Rebin angular correlaion distributon templates used for FD correction (theory driven)
Bool_t fRebinSecPart; // Rebin angular correlaion distributon templates used for secodary particle contamination correction
Bool_t fSidebandDivided; // To be set to TRUE if two sideband corrlaion histograms are passed inteh config file
Bool_t fUseSidebLeft; // To be set to TRUE if only sideband left is used for the bkg correction
Bool_t fUseSidebRight; // To be set to TRUE if only sideband right is used for the bkg correction
Bool_t fFDsubtraction; // Enable feed-down (FD) correction
Bool_t fSecPartContamination; // Enable seconday particle contamination correction
Bool_t fCorrBiasBtoD; // Enable bias B to D correction (perfrmed with angular correlaion templates taken from MC simulaions)
};

#endif // PWGHF_HFC_MACROS_DHCORRELATIONEXTRACTION_H_
Loading
Loading