Skip to content

[PWGLF] added MC weight in kstar flow task #9829

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
skundu692 merged 1 commit into from
Feb 7, 2025
Merged

[PWGLF] added MC weight in kstar flow task #9829

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
177 changes: 93 additions & 84 deletions PWGLF/Tasks/Resonances/kstarpbpb.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,7 @@
#include "DataFormatsParameters/GRPObject.h"
#include "DataFormatsParameters/GRPMagField.h"
#include "CCDB/BasicCCDBManager.h"
#include "CCDB/CcdbApi.h"

using namespace o2;
using namespace o2::framework;
Expand All @@ -63,15 +64,23 @@ struct kstarpbpb {
int mRunNumber;
int multEstimator;
float d_bz;

struct : ConfigurableGroup {
Configurable<std::string> cfgURL{"cfgURL", "http://alice-ccdb.cern.ch", "Address of the CCDB to browse"};
Configurable<int64_t> nolaterthan{"ccdb-no-later-than", std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch()).count(), "Latest acceptable timestamp of creation for the object"};
} cfgCcdbParam;

// Enable access to the CCDB for the offset and correction constants and save them in dedicated variables.
Service<o2::ccdb::BasicCCDBManager> ccdb;
o2::ccdb::CcdbApi ccdbApi;
Service<o2::framework::O2DatabasePDG> pdg;

// CCDB options
Configurable<std::string> ccdburl{"ccdb-url", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};
Configurable<std::string> grpPath{"grpPath", "GLO/GRP/GRP", "Path of the grp file"};
Configurable<std::string> grpmagPath{"grpmagPath", "GLO/Config/GRPMagField", "CCDB path of the GRPMagField object"};
Configurable<std::string> lutPath{"lutPath", "GLO/Param/MatLUT", "Path of the Lut parametrization"};
Configurable<std::string> geoPath{"geoPath", "GLO/Config/GeometryAligned", "Path of the geometry file"};
// Configurable<std::string> ccdburl{"ccdb-url", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};
// Configurable<std::string> grpPath{"grpPath", "GLO/GRP/GRP", "Path of the grp file"};
// Configurable<std::string> grpmagPath{"grpmagPath", "GLO/Config/GRPMagField", "CCDB path of the GRPMagField object"};
// Configurable<std::string> lutPath{"lutPath", "GLO/Param/MatLUT", "Path of the Lut parametrization"};
// Configurable<std::string> geoPath{"geoPath", "GLO/Config/GeometryAligned", "Path of the geometry file"};

// events
Configurable<float> cfgCutVertex{"cfgCutVertex", 10.0f, "Accepted z-vertex range"};
Expand All @@ -98,7 +107,7 @@ struct kstarpbpb {
Configurable<bool> removefaketrak{"removefaketrack", true, "Remove fake track from momentum difference"};
Configurable<float> ConfFakeKaonCut{"ConfFakeKaonCut", 0.1, "Cut based on track from momentum difference"};
ConfigurableAxis configThnAxisPhiminusPsi{"configThnAxisPhiminusPsi", {6, 0.0, TMath::Pi()}, "#phi - #psi"};
ConfigurableAxis configThnAxisV2{"configThnAxisV2", {200, -1, 1}, "V2"};
ConfigurableAxis configThnAxisV2{"configThnAxisV2", {400, -16, 16}, "V2"};
Configurable<bool> additionalEvsel{"additionalEvsel", false, "Additional event selcection"};
Configurable<bool> timFrameEvsel{"timFrameEvsel", false, "TPC Time frame boundary cut"};
Configurable<bool> additionalEvselITS{"additionalEvselITS", true, "Additional event selcection for ITS"};
Expand All @@ -117,7 +126,8 @@ struct kstarpbpb {
Configurable<bool> mix{"mix", false, "mix"};
Configurable<bool> fillOccupancy{"fillOccupancy", false, "fill Occupancy"};
Configurable<int> cfgOccupancyCut{"cfgOccupancyCut", 500, "Occupancy cut"};
Configurable<bool> useSP{"useSP", true, "useSP"};
Configurable<bool> useWeight{"useWeight", false, "use EP dep effi weight"};
Configurable<std::string> ConfWeightPath{"ConfWeightPath", "Users/s/skundu/My/Object/PbPb2024/MCWeight2", "Path to gain calibration"};

Filter collisionFilter = nabs(aod::collision::posZ) < cfgCutVertex;
Filter centralityFilter = nabs(aod::cent::centFT0C) < cfgCutCentrality;
Expand Down Expand Up @@ -152,9 +162,8 @@ struct kstarpbpb {
AxisSpec centAxis = {8, 0, 80, "V0M (%)"};
AxisSpec occupancyAxis = {occupancyBinning, "Occupancy"};

if (same) {
histos.add("hSparseV2SASameEvent_V2", "hSparseV2SASameEvent_V2", HistType::kTHnSparseF, {thnAxisInvMass, thnAxisPt, thnAxisV2, thnAxisCentrality});
}
histos.add("hSparseV2SASameEvent_V2", "hSparseV2SASameEvent_V2", HistType::kTHnSparseF, {thnAxisInvMass, thnAxisPt, thnAxisV2, thnAxisCentrality});

if (like) {
histos.add("hSparseV2SAlikeEventNN_V2", "hSparseV2SAlikeEventNN_V2", HistType::kTHnSparseF, {thnAxisInvMass, thnAxisPt, thnAxisV2, thnAxisCentrality});
histos.add("hSparseV2SAlikeEventPP_V2", "hSparseV2SAlikeEventPP_V2", HistType::kTHnSparseF, {thnAxisInvMass, thnAxisPt, thnAxisV2, thnAxisCentrality});
Expand Down Expand Up @@ -225,6 +234,11 @@ struct kstarpbpb {
fMultMultPVCut = new TF1("fMultMultPVCut", "[0]+[1]*x+[2]*x*x", 0, 5000);
fMultMultPVCut->SetParameters(-0.1, 0.785, -4.7e-05);
}
ccdb->setURL(cfgCcdbParam.cfgURL);
ccdbApi.init("http://alice-ccdb.cern.ch");
ccdb->setCaching(true);
ccdb->setLocalObjectValidityChecking();
ccdb->setCreatedNotAfter(std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch()).count());
}

double massKa = o2::constants::physics::MassKPlus;
Expand Down Expand Up @@ -418,7 +432,10 @@ struct kstarpbpb {
using BinningTypeVertexContributor = ColumnBinningPolicy<aod::collision::PosZ, aod::cent::CentFT0C, aod::epcalibrationtable::PsiFT0C>;
ROOT::Math::PxPyPzMVector KstarMother, daughter1, daughter2, kaonrot, kstarrot;

void processSE(EventCandidates::iterator const& collision, TrackCandidates const& tracks, aod::BCs const&)
int currentRunNumber = -999;
int lastRunNumber = -999;
TH3D* hweight;
void processSE(EventCandidates::iterator const& collision, TrackCandidates const& tracks, aod::BCsWithTimestamps const&)
{
if (!collision.sel8() || !collision.triggereventep() || !collision.selection_bit(aod::evsel::kNoTimeFrameBorder) || !collision.selection_bit(aod::evsel::kNoITSROFrameBorder) || !collision.selection_bit(aod::evsel::kNoSameBunchPileup) || !collision.selection_bit(aod::evsel::kIsGoodZvtxFT0vsPV) || !collision.selection_bit(o2::aod::evsel::kNoCollInTimeRangeStandard)) {
return;
Expand Down Expand Up @@ -456,6 +473,14 @@ struct kstarpbpb {
histos.fill(HIST("hOccupancy"), occupancy);
histos.fill(HIST("hVtxZ"), collision.posZ());
}
auto bc = collision.bc_as<aod::BCsWithTimestamps>();
currentRunNumber = collision.bc_as<aod::BCsWithTimestamps>().runNumber();
if (useWeight && (currentRunNumber != lastRunNumber)) {
hweight = ccdb->getForTimeStamp<TH3D>(ConfWeightPath.value, bc.timestamp());
}
lastRunNumber = currentRunNumber;
float weight1 = 1.0;
float weight2 = 1.0;
for (auto track1 : tracks) {
if (!selectionTrack(track1)) {
continue;
Expand All @@ -466,6 +491,13 @@ struct kstarpbpb {
continue;
}
track1kaon = true;
if (useWeight) {
if (track1.pt() < 10.0 && track1.pt() > 0.15) {
weight1 = hweight->GetBinContent(hweight->FindBin(centrality, GetPhiInRange(track1.phi() - psiFT0C), track1.pt() + 0.000005));
} else {
weight1 = 1;
}
}
for (auto track2 : tracks) {
if (!selectionTrack(track2)) {
continue;
Expand All @@ -482,66 +514,59 @@ struct kstarpbpb {
if (!track1kaon || !track2pion) {
continue;
}
if (useWeight) {
if (track2.pt() < 10.0 && track2.pt() > 0.15) {
weight2 = hweight->GetBinContent(hweight->FindBin(centrality, GetPhiInRange(track2.phi() - psiFT0C), track2.pt() + 0.000005));
} else {
weight2 = 1;
}
}
daughter1 = ROOT::Math::PxPyPzMVector(track1.px(), track1.py(), track1.pz(), massKa);
daughter2 = ROOT::Math::PxPyPzMVector(track2.px(), track2.py(), track2.pz(), massPi);
KstarMother = daughter1 + daughter2;
if (TMath::Abs(KstarMother.Rapidity()) > confRapidity) {
continue;
}
auto phiminuspsi = GetPhiInRange(KstarMother.Phi() - psiFT0C);
if (useSP) {
v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
}
if (!useSP) {
v2 = TMath::Cos(2.0 * phiminuspsi);

v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
auto totalweight = weight1 * weight2;
if (totalweight <= 0.0005) {
totalweight = 1.0;
}
// unlike sign
if (track1.sign() * track2.sign() < 0) {
if (same) {
histos.fill(HIST("hSparseV2SASameEvent_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
}
if (fillRotation) {
for (int nrotbkg = 0; nrotbkg < nBkgRotations; nrotbkg++) {
auto anglestart = confMinRot;
auto angleend = confMaxRot;
auto anglestep = (angleend - anglestart) / (1.0 * (nBkgRotations - 1));
auto rotangle = anglestart + nrotbkg * anglestep;
histos.fill(HIST("hRotation"), rotangle);
auto rotkaonPx = track1.px() * std::cos(rotangle) - track1.py() * std::sin(rotangle);
auto rotkaonPy = track1.px() * std::sin(rotangle) + track1.py() * std::cos(rotangle);
kaonrot = ROOT::Math::PxPyPzMVector(rotkaonPx, rotkaonPy, track1.pz(), massKa);
kstarrot = kaonrot + daughter2;
if (TMath::Abs(kstarrot.Rapidity()) > confRapidity) {
continue;
}
auto phiminuspsiRot = GetPhiInRange(kstarrot.Phi() - psiFT0C);
if (useSP) {
v2Rot = TMath::Cos(2.0 * phiminuspsiRot) * QFT0C;
}
if (!useSP) {
v2Rot = TMath::Cos(2.0 * phiminuspsiRot);
}
histos.fill(HIST("hSparseV2SASameEventRotational_V2"), kstarrot.M(), kstarrot.Pt(), v2Rot, centrality);
}
}

if (useWeight) {
histos.fill(HIST("hSparseV2SASameEvent_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality, 1 / totalweight);
} else {
histos.fill(HIST("hSparseV2SASameEvent_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
}
// like sign
if (track1.sign() * track2.sign() > 0) {
if (like) {
if (track1.sign() > 0 && track2.sign() > 0) {
histos.fill(HIST("hSparseV2SAlikeEventPP_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
}
if (track1.sign() < 0 && track2.sign() < 0) {
histos.fill(HIST("hSparseV2SAlikeEventNN_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
if (fillRotation) {
for (int nrotbkg = 0; nrotbkg < nBkgRotations; nrotbkg++) {
auto anglestart = confMinRot;
auto angleend = confMaxRot;
auto anglestep = (angleend - anglestart) / (1.0 * (nBkgRotations - 1));
auto rotangle = anglestart + nrotbkg * anglestep;
histos.fill(HIST("hRotation"), rotangle);
auto rotkaonPx = track1.px() * std::cos(rotangle) - track1.py() * std::sin(rotangle);
auto rotkaonPy = track1.px() * std::sin(rotangle) + track1.py() * std::cos(rotangle);
kaonrot = ROOT::Math::PxPyPzMVector(rotkaonPx, rotkaonPy, track1.pz(), massKa);
kstarrot = kaonrot + daughter2;
if (TMath::Abs(kstarrot.Rapidity()) > confRapidity) {
continue;
}
auto phiminuspsiRot = GetPhiInRange(kstarrot.Phi() - psiFT0C);

v2Rot = TMath::Cos(2.0 * phiminuspsiRot) * QFT0C;

histos.fill(HIST("hSparseV2SASameEventRotational_V2"), kstarrot.M(), kstarrot.Pt(), v2Rot, centrality);
}
}
}
}
}
PROCESS_SWITCH(kstarpbpb, processSE, "Process Same event latest", true);

void processSameEvent(EventCandidates::iterator const& collision, TrackCandidates const& /*tracks*/, aod::BCs const&)
void processSameEvent(EventCandidates::iterator const& collision, TrackCandidates const& /*tracks, aod::BCs const&*/, aod::BCsWithTimestamps const&)
{
if (!collision.sel8()) {
return;
Expand Down Expand Up @@ -697,12 +722,9 @@ struct kstarpbpb {
continue;
}
auto phiminuspsi = GetPhiInRange(KstarMother.Phi() - psiFT0C);
if (useSP) {
v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
}
if (!useSP) {
v2 = TMath::Cos(2.0 * phiminuspsi);
}

v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;

histos.fill(HIST("hSparseV2SASameEvent_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
}
if (fillRotation) {
Expand Down Expand Up @@ -730,12 +752,9 @@ struct kstarpbpb {
continue;
}
auto phiminuspsiRot = GetPhiInRange(kstarrot.Phi() - psiFT0C);
if (useSP) {
v2Rot = TMath::Cos(2.0 * phiminuspsiRot) * QFT0C;
}
if (!useSP) {
v2Rot = TMath::Cos(2.0 * phiminuspsiRot);
}

v2Rot = TMath::Cos(2.0 * phiminuspsiRot) * QFT0C;

histos.fill(HIST("hSparseV2SASameEventRotational_V2"), kstarrot.M(), kstarrot.Pt(), v2Rot, centrality);
}
}
Expand Down Expand Up @@ -846,12 +865,8 @@ struct kstarpbpb {
continue;
}
auto phiminuspsi = GetPhiInRange(KstarMother.Phi() - psiFT0C);
if (useSP) {
v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
}
if (!useSP) {
v2 = TMath::Cos(2.0 * phiminuspsi);
}

v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
histos.fill(HIST("hSparseV2SAlikeEventNN_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
}
} else if (track1pion && track2kaon) {
Expand All @@ -863,12 +878,9 @@ struct kstarpbpb {
continue;
}
auto phiminuspsi = GetPhiInRange(KstarMother.Phi() - psiFT0C);
if (useSP) {
v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
}
if (!useSP) {
v2 = TMath::Cos(2.0 * phiminuspsi);
}

v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;

histos.fill(HIST("hSparseV2SAlikeEventPP_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
}
}
Expand Down Expand Up @@ -1000,12 +1012,9 @@ struct kstarpbpb {
continue;
}
auto phiminuspsi = GetPhiInRange(KstarMother.Phi() - psiFT0C);
if (useSP) {
v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;
}
if (!useSP) {
v2 = TMath::Cos(2.0 * phiminuspsi);
}

v2 = TMath::Cos(2.0 * phiminuspsi) * QFT0C;

if (mix) {
histos.fill(HIST("hSparseV2SAMixedEvent_V2"), KstarMother.M(), KstarMother.Pt(), v2, centrality);
}
Expand Down