Skip to content
Merged
Changes from all commits
Commits
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
25 changes: 24 additions & 1 deletion ALICE3/Tasks/alice3-dilepton.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
/// \author s.scheid@cern.ch, daiki.sekihata@cern.ch
///

#include "ALICE3/DataModel/OTFCollision.h"
#include "ALICE3/DataModel/OTFRICH.h"
#include "ALICE3/DataModel/OTFTOF.h"
#include "ALICE3/DataModel/tracksAlice3.h"
Expand All @@ -24,6 +25,7 @@
#include <Framework/AnalysisDataModel.h>
#include <Framework/AnalysisTask.h>
#include <Framework/HistogramRegistry.h>
#include <Framework/O2DatabasePDGPlugin.h>
#include <Framework/runDataProcessing.h>

#include <Math/Vector4D.h>
Expand Down Expand Up @@ -54,6 +56,8 @@ struct Alice3Dilepton {
SliceCache cache_mc;
SliceCache cache_rec;

Service<o2::framework::O2DatabasePDG> inspdg;

Configurable<int> pdg{"pdg", 11, "pdg code for analysis. dielectron:11, dimuon:13"};
Configurable<bool> requireHFEid{"requireHFEid", true, "Require HFE identification for both leptons"};
Configurable<float> ptMin{"pt-min", 0.f, "Lower limit in pT"};
Expand All @@ -68,6 +72,7 @@ struct Alice3Dilepton {
Configurable<float> nSigmaPionCutInnerTOF{"nSigmaPionCutInnerTOF", 3., "Pion exclusion in inner TOF"};
Configurable<float> nSigmaElectronRich{"nSigmaElectronRich", 3., "Electron inclusion RICH"};
Configurable<float> nSigmaPionRich{"nSigmaPionRich", 3., "Pion exclusion RICH"};
Configurable<int> otfConfig{"otfConfig", 0, "OTF configuration flag"};

HistogramRegistry registry{"Histos", {}, OutputObjHandlingPolicy::AnalysisObject};

Expand Down Expand Up @@ -100,6 +105,7 @@ struct Alice3Dilepton {
registry.add("Generated/Particle/prodVy", "Particle Prod. Vertex Y", kTH1F, {axisPrody});
registry.add("Generated/Particle/prodVz", "Particle Prod. Vertex Z", kTH1F, {axisProdz});
registry.add("Generated/Particle/ParticlesPerEvent", "Particles per event", kTH1F, {{100, 0, 100}});
registry.add("Generated/Particle/ParticlesFit", "Charged Particles in Fit acceptance per event", kTH1F, {{15000, 0, 15000}});

registry.add("Generated/Pair/ULS/Tried", "Pair tries", kTH1F, {{10, -0.5, 9.5}});
registry.add("Generated/Pair/ULS/Mass", "Pair Mass", kTH1F, {axisM});
Expand Down Expand Up @@ -601,7 +607,19 @@ struct Alice3Dilepton {

auto mcParticles_per_coll = mcParticles.sliceBy(perMCCollision, mccollision.globalIndex());
int nParticlesInEvent = 0;
int nParticlesFIT = 0;
for (const auto& mcParticle : mcParticles_per_coll) {
if (mcParticle.isPhysicalPrimary()) {
if ((2.2 < mcParticle.eta() && mcParticle.eta() < 5.0) || (-3.4 < mcParticle.eta() && mcParticle.eta() < -2.3)) {
auto pdgParticle = inspdg->GetParticle(mcParticle.pdgCode());
if (pdgParticle) {
float charge = pdgParticle->Charge() / 3.f; // Charge in units of |e|
if (std::abs(charge) >= 1.) {
nParticlesFIT++;
}
}
}
}
if (std::abs(mcParticle.pdgCode()) != pdg) {
continue;
}
Expand All @@ -624,6 +642,7 @@ struct Alice3Dilepton {

} // end of mc particle loop
registry.fill(HIST("Generated/Particle/ParticlesPerEvent"), nParticlesInEvent);
registry.fill(HIST("Generated/Particle/ParticlesFIT"), nParticlesFIT);

auto neg_mcParticles_coll = neg_mcParticles->sliceByCached(o2::aod::mcparticle::mcCollisionId, mccollision.globalIndex(), cache_mc);
auto pos_mcParticles_coll = pos_mcParticles->sliceByCached(o2::aod::mcparticle::mcCollisionId, mccollision.globalIndex(), cache_mc);
Expand All @@ -636,18 +655,22 @@ struct Alice3Dilepton {
} // end of processGen

using MyTracksMC = soa::Join<aod::Tracks, aod::TracksCov, aod::TracksDCA, aod::McTrackLabels, aod::UpgradeTofs, aod::UpgradeTofMCs, aod::UpgradeRichs, aod::TracksAlice3>;
using Alice3Collision = soa::Join<aod::Collisions, aod::OTFLUTConfigId>;

// Filter trackFilter = etaMin < o2::aod::track::eta &&
// o2::aod::track::eta < etaMax &&
// ptMin < o2::aod::track::pt &&
// o2::aod::track::pt < ptMax &&
// o2::aod::track_alice3::isReconstructed == selectReconstructed;
Filter trackFilter = o2::aod::track_alice3::isReconstructed == selectReconstructed;
using MyFilteredTracksMC = soa::Filtered<MyTracksMC>;
Filter configFilter = (aod::upgrade_collision::lutConfigId == otfConfig);
using MyFilteredAlice3Collision = soa::Filtered<Alice3Collision>;
Preslice<MyFilteredTracksMC> perCollision = aod::track::collisionId;
Partition<MyFilteredTracksMC> posTracks = o2::aod::track::signed1Pt > 0.f;
Partition<MyFilteredTracksMC> negTracks = o2::aod::track::signed1Pt < 0.f;

void processRec(const o2::aod::Collisions& collisions,
void processRec(MyFilteredAlice3Collision const& collisions,
MyFilteredTracksMC const& tracks,
const o2::aod::McCollisions&,
const aod::McParticles& mcParticles)
Expand Down
Loading