Skip to content

Commit

Permalink
Merge pull request #19 from alibuild/alibot-cleanup-5159
Browse files Browse the repository at this point in the history 
Please consider the following formatting changes to AliceO2Group#5159
  • Loading branch information
@abylinkin
abylinkin authored Mar 14, 2024
2 parents 5687d6c + c693570 commit eb85efd
Showing 1 changed file with 74 additions and 75 deletions.
149 changes: 74 additions & 75 deletions PWGUD/Tasks/SG_D0_Analyzer.cxx
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,10 @@ using namespace o2;
using namespace o2::aod;
using namespace o2::framework;
using namespace o2::framework::expressions;
#define mpion 0.1396
#define mpion 0.1396
#define mkaon 0.4937
#define mproton 0.9383
struct SG_D0_Analyzer{
struct SG_D0_Analyzer {
HistogramRegistry registry{
"registry",
{
Expand Down Expand Up @@ -42,125 +42,124 @@ struct SG_D0_Analyzer{
{"ss_KPi_pT_2", "#K#Pi pT (GeV/c); Entries", {HistType::kTH1F, {{5000, 0, 10}}}},
{"ss_KPi_eTa_2", "#K#Pi eTa (GeV/c); Entries", {HistType::kTH1F, {{100, -1., 1.}}}},
{"ss_KPi_invm_2", "#K#Pi Mass (GeV/c^2); Entries", {HistType::kTH1F, {{5000, 0, 10}}}},
}
};
using udtracks = soa::Join<aod::UDTracks,aod::UDTracksExtra,aod::UDTracksPID>;
using udtracksfull = soa::Join<aod::UDTracks, aod::UDTracksPID, aod::UDTracksExtra, aod::UDTracksFlags, aod::UDTracksDCA>;
using UDCollisionsFull = soa::Join<aod::UDCollisions, aod::SGCollisions, aod::UDCollisionsSels, aod::UDZdcsReduced>; //
using UDCollisionFull = UDCollisionsFull::iterator;
}};
using udtracks = soa::Join<aod::UDTracks, aod::UDTracksExtra, aod::UDTracksPID>;
using udtracksfull = soa::Join<aod::UDTracks, aod::UDTracksPID, aod::UDTracksExtra, aod::UDTracksFlags, aod::UDTracksDCA>;
using UDCollisionsFull = soa::Join<aod::UDCollisions, aod::SGCollisions, aod::UDCollisionsSels, aod::UDZdcsReduced>; //
using UDCollisionFull = UDCollisionsFull::iterator;

void process(UDCollisionFull const& collision, udtracksfull const& tracks){
void process(UDCollisionFull const& collision, udtracksfull const& tracks)
{
TLorentzVector a;
int gapSide = collision.gapSide();
if(gapSide < 0 || gapSide > 2) return;
// Single gap either side
TLorentzVector v0;
TLorentzVector v1;
TLorentzVector v01;
// Look for D0 and D0bar
for (auto& [t0, t1] : combinations(tracks, tracks)) {
// PID cut - t0=K, t1=pi
if (std::abs(t0.tpcNSigmaKa()) < 3 && std::abs(t1.tpcNSigmaPi()) < 3
&& std::abs(t0.tofNSigmaKa()) < 3 && std::abs(t1.tofNSigmaPi()) < 3) {
// Apply pion hypothesis and create pairs
v0.SetXYZM(t0.px(), t0.py(), t0.pz(), o2::constants::physics::MassKaonCharged);
v1.SetXYZM(t1.px(), t1.py(), t1.pz(), o2::constants::physics::MassPionCharged);
v01 = v0 + v1;
// Opposite sign pairs
if (t0.sign() != t1.sign()) {
registry.fill(HIST("os_KPi_pT"), v01.Pt());
registry.fill(HIST("os_KPi_eTa"), v01.Eta());
registry.fill(HIST("os_KPi_invm"), v01.M());
if (gapSide==0){
if (gapSide < 0 || gapSide > 2)
return;
// Single gap either side
TLorentzVector v0;
TLorentzVector v1;
TLorentzVector v01;
// Look for D0 and D0bar
for (auto& [t0, t1] : combinations(tracks, tracks)) {
// PID cut - t0=K, t1=pi
if (std::abs(t0.tpcNSigmaKa()) < 3 && std::abs(t1.tpcNSigmaPi()) < 3 && std::abs(t0.tofNSigmaKa()) < 3 && std::abs(t1.tofNSigmaPi()) < 3) {
// Apply pion hypothesis and create pairs
v0.SetXYZM(t0.px(), t0.py(), t0.pz(), o2::constants::physics::MassKaonCharged);
v1.SetXYZM(t1.px(), t1.py(), t1.pz(), o2::constants::physics::MassPionCharged);
v01 = v0 + v1;
// Opposite sign pairs
if (t0.sign() != t1.sign()) {
registry.fill(HIST("os_KPi_pT"), v01.Pt());
registry.fill(HIST("os_KPi_eTa"), v01.Eta());
registry.fill(HIST("os_KPi_invm"), v01.M());
if (gapSide == 0) {
registry.fill(HIST("os_KPi_pT_0"), v01.Pt());
registry.fill(HIST("os_KPi_eTa_0"), v01.Eta());
registry.fill(HIST("os_KPi_invm_0"), v01.M());
}
if (gapSide==1){
}
if (gapSide == 1) {
registry.fill(HIST("os_KPi_pT_1"), v01.Pt());
registry.fill(HIST("os_KPi_eTa_1"), v01.Eta());
registry.fill(HIST("os_KPi_invm_1"), v01.M());
}
if (gapSide==2){
}
if (gapSide == 2) {
registry.fill(HIST("os_KPi_pT_2"), v01.Pt());
registry.fill(HIST("os_KPi_eTa_2"), v01.Eta());
registry.fill(HIST("os_KPi_invm_2"), v01.M());
}
}
// Same sign pairs
else if (t0.sign() == t1.sign()) {
registry.fill(HIST("ss_KPi_pT"), v01.Pt());
registry.fill(HIST("ss_KPi_eTa"), v01.Eta());
registry.fill(HIST("ss_KPi_invm"), v01.M());
if (gapSide==0){
}
// Same sign pairs
else if (t0.sign() == t1.sign()) {
registry.fill(HIST("ss_KPi_pT"), v01.Pt());
registry.fill(HIST("ss_KPi_eTa"), v01.Eta());
registry.fill(HIST("ss_KPi_invm"), v01.M());
if (gapSide == 0) {
registry.fill(HIST("ss_KPi_pT_0"), v01.Pt());
registry.fill(HIST("ss_KPi_eTa_0"), v01.Eta());
registry.fill(HIST("ss_KPi_invm_0"), v01.M());
}
if (gapSide==1){
}
if (gapSide == 1) {
registry.fill(HIST("ss_KPi_pT_1"), v01.Pt());
registry.fill(HIST("ss_KPi_eTa_1"), v01.Eta());
registry.fill(HIST("ss_KPi_invm_1"), v01.M());
}
if (gapSide==2){
}
if (gapSide == 2) {
registry.fill(HIST("ss_KPi_pT_2"), v01.Pt());
registry.fill(HIST("ss_KPi_eTa_2"), v01.Eta());
registry.fill(HIST("ss_KPi_invm_2"), v01.M());
}
}
}
// PID cut - t0=pi, t1=K
else if (std::abs(t1.tpcNSigmaKa()) < 3 && std::abs(t0.tpcNSigmaPi()) < 3
&& std::abs(t1.tofNSigmaKa()) < 3 && std::abs(t0.tofNSigmaPi()) < 3) {
// Apply pion hypothesis and create pairs
v0.SetXYZM(t0.px(), t0.py(), t0.pz(), o2::constants::physics::MassPionCharged);
v1.SetXYZM(t1.px(), t1.py(), t1.pz(), o2::constants::physics::MassKaonCharged);
v01 = v0 + v1;
// Opposite sign pairs
if (t0.sign() != t1.sign()) {
registry.fill(HIST("os_KPi_pT"), v01.Pt());
registry.fill(HIST("os_KPi_eTa"), v01.Eta());
registry.fill(HIST("os_KPi_invm"), v01.M());
if (gapSide==0){
}
// PID cut - t0=pi, t1=K
else if (std::abs(t1.tpcNSigmaKa()) < 3 && std::abs(t0.tpcNSigmaPi()) < 3 && std::abs(t1.tofNSigmaKa()) < 3 && std::abs(t0.tofNSigmaPi()) < 3) {
// Apply pion hypothesis and create pairs
v0.SetXYZM(t0.px(), t0.py(), t0.pz(), o2::constants::physics::MassPionCharged);
v1.SetXYZM(t1.px(), t1.py(), t1.pz(), o2::constants::physics::MassKaonCharged);
v01 = v0 + v1;
// Opposite sign pairs
if (t0.sign() != t1.sign()) {
registry.fill(HIST("os_KPi_pT"), v01.Pt());
registry.fill(HIST("os_KPi_eTa"), v01.Eta());
registry.fill(HIST("os_KPi_invm"), v01.M());
if (gapSide == 0) {
registry.fill(HIST("os_KPi_pT_0"), v01.Pt());
registry.fill(HIST("os_KPi_eTa_0"), v01.Eta());
registry.fill(HIST("os_KPi_invm_0"), v01.M());
}
if (gapSide==1){
}
if (gapSide == 1) {
registry.fill(HIST("os_KPi_pT_1"), v01.Pt());
registry.fill(HIST("os_KPi_eTa_1"), v01.Eta());
registry.fill(HIST("os_KPi_invm_1"), v01.M());
}
if (gapSide==2){
}
if (gapSide == 2) {
registry.fill(HIST("os_KPi_pT_2"), v01.Pt());
registry.fill(HIST("os_KPi_eTa_2"), v01.Eta());
registry.fill(HIST("os_KPi_invm_2"), v01.M());
}
}
// Same sign pairs
else if (t0.sign() == t1.sign()) {
registry.fill(HIST("ss_KPi_pT"), v01.Pt());
registry.fill(HIST("ss_KPi_eTa"), v01.Eta());
registry.fill(HIST("ss_KPi_invm"), v01.M());
if (gapSide==0){
}
// Same sign pairs
else if (t0.sign() == t1.sign()) {
registry.fill(HIST("ss_KPi_pT"), v01.Pt());
registry.fill(HIST("ss_KPi_eTa"), v01.Eta());
registry.fill(HIST("ss_KPi_invm"), v01.M());
if (gapSide == 0) {
registry.fill(HIST("ss_KPi_pT_0"), v01.Pt());
registry.fill(HIST("ss_KPi_eTa_0"), v01.Eta());
registry.fill(HIST("ss_KPi_invm_0"), v01.M());
}
if (gapSide==1){
}
if (gapSide == 1) {
registry.fill(HIST("ss_KPi_pT_1"), v01.Pt());
registry.fill(HIST("ss_KPi_eTa_1"), v01.Eta());
registry.fill(HIST("ss_KPi_invm_1"), v01.M());
}
if (gapSide==2){
}
if (gapSide == 2) {
registry.fill(HIST("ss_KPi_pT_2"), v01.Pt());
registry.fill(HIST("ss_KPi_eTa_2"), v01.Eta());
registry.fill(HIST("ss_KPi_invm_2"), v01.M());
}
}
}
}
}
}
}
};

WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
Expand Down

0 comments on commit eb85efd

Please sign in to comment.