Skip to content

[Common] TOF/FT0: merge event time tasks #12736

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

Closed
njacazio wants to merge 1 commit into from
Closed

[Common] TOF/FT0: merge event time tasks #12736

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
92 changes: 84 additions & 8 deletions Common/TableProducer/PID/pidTOFMerge.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
// Copyright 2019-2020 CERN and copyright holders of ALICE O2.

Check failure on line 1 in Common/TableProducer/PID/pidTOFMerge.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/workflow-file] 

Name of a workflow file must match the name of the main struct in it (without the PWG prefix). (Class implementation files should be in "Core" directories.)
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
Expand All @@ -25,6 +25,7 @@
#include "Common/DataModel/PIDResponseTOF.h"

#include <CCDB/BasicCCDBManager.h>
#include <DataFormatsFIT/Triggers.h>
#include <DataFormatsParameters/GRPLHCIFData.h>
#include <DataFormatsTOF/ParameterContainers.h>
#include <Framework/ASoA.h>
Expand Down Expand Up @@ -55,6 +56,7 @@
#include <map>
#include <memory>
#include <string>
#include <utility>
#include <vector>

using namespace o2;
Expand All @@ -72,7 +74,12 @@
using Run3TrksWtofWevTime = soa::Join<Run3TrksWtof, aod::TOFEvTime, aod::pidEvTimeFlags>;

using EvTimeCollisions = soa::Join<Run3Cols, aod::EvSels>;
// #define MERGEWITHFT0
#ifndef MERGEWITHFT0
using EvTimeCollisionsFT0 = soa::Join<EvTimeCollisions, aod::FT0sCorrected>;
#else
using EvTimeCollisionsFT0 = EvTimeCollisions;
#endif

using Run2Trks = o2::soa::Join<aod::Tracks, aod::TracksExtra>;
using Run2TrksWtofWevTime = soa::Join<Run2Trks, aod::TOFSignal, aod::TOFEvTime, aod::pidEvTimeFlags>;
Expand Down Expand Up @@ -387,15 +394,15 @@
o2::pid::tof::TOFResoParamsV3 mRespParamsV3;
Service<o2::ccdb::BasicCCDBManager> ccdb;
struct : ConfigurableGroup {
Configurable<std::string> cfgUrl{"ccdb-url", "http://alice-ccdb.cern.ch", "url of the ccdb repository"};

Check failure on line 397 in Common/TableProducer/PID/pidTOFMerge.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/configurable] 

Use lowerCamelCase for names of configurables and use the same name for the struct member as for the JSON string. (Declare the type and names on the same line.)
Configurable<std::string> cfgPathGrpLhcIf{"ccdb-path-grplhcif", "GLO/Config/GRPLHCIF", "Path on the CCDB for the GRPLHCIF object"};

Check failure on line 398 in Common/TableProducer/PID/pidTOFMerge.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/configurable] 

Use lowerCamelCase for names of configurables and use the same name for the struct member as for the JSON string. (Declare the type and names on the same line.)
Configurable<int64_t> cfgTimestamp{"ccdb-timestamp", -1, "timestamp of the object"};

Check failure on line 399 in Common/TableProducer/PID/pidTOFMerge.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/configurable] 

Use lowerCamelCase for names of configurables and use the same name for the struct member as for the JSON string. (Declare the type and names on the same line.)
Configurable<std::string> cfgTimeShiftCCDBPathPos{"timeShiftCCDBPathPos", "", "Path of the TOF time shift vs eta for pos. tracks. If empty none is taken"};

Check failure on line 400 in Common/TableProducer/PID/pidTOFMerge.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/configurable] 

Use lowerCamelCase for names of configurables and use the same name for the struct member as for the JSON string. (Declare the type and names on the same line.)
Configurable<std::string> cfgTimeShiftCCDBPathNeg{"timeShiftCCDBPathNeg", "", "Path of the TOF time shift vs eta for neg. tracks. If empty none is taken"};

Check failure on line 401 in Common/TableProducer/PID/pidTOFMerge.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/configurable] 

Use lowerCamelCase for names of configurables and use the same name for the struct member as for the JSON string. (Declare the type and names on the same line.)
Configurable<std::string> cfgTimeShiftCCDBPathPosMC{"timeShiftCCDBPathPosMC", "", "Path of the TOF time shift for MC vs eta for pos. tracks. If empty none is taken"};

Check failure on line 402 in Common/TableProducer/PID/pidTOFMerge.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/configurable] 

Use lowerCamelCase for names of configurables and use the same name for the struct member as for the JSON string. (Declare the type and names on the same line.)
Configurable<std::string> cfgTimeShiftCCDBPathNegMC{"timeShiftCCDBPathNegMC", "", "Path of the TOF time shift for MC vs eta for neg. tracks. If empty none is taken"};

Check failure on line 403 in Common/TableProducer/PID/pidTOFMerge.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/configurable] 

Use lowerCamelCase for names of configurables and use the same name for the struct member as for the JSON string. (Declare the type and names on the same line.)
Configurable<std::string> cfgParamFileName{"paramFileName", "", "Path to the parametrization object. If empty the parametrization is not taken from file"};

Check failure on line 404 in Common/TableProducer/PID/pidTOFMerge.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/configurable] 

Use lowerCamelCase for names of configurables and use the same name for the struct member as for the JSON string. (Declare the type and names on the same line.)
Configurable<std::string> cfgParametrizationPath{"parametrizationPath", "TOF/Calib/Params", "Path of the TOF parametrization on the CCDB or in the file, if the paramFileName is not empty"};

Check failure on line 405 in Common/TableProducer/PID/pidTOFMerge.cxx

View workflow job for this annotation

GitHub Actions / O2 linter

[name/configurable] 

Use lowerCamelCase for names of configurables and use the same name for the struct member as for the JSON string. (Declare the type and names on the same line.)
Configurable<std::string> cfgReconstructionPass{"reconstructionPass", "", {"Apass to use when fetching the calibration tables. Empty (default) does not check for any pass. Use `metadata` to fetch it from the AO2D metadata. Otherwise it will override the metadata."}};
Configurable<std::string> cfgReconstructionPassDefault{"reconstructionPassDefault", "unanchored", {"Default pass to get if the standard one is not found"}};
Configurable<bool> cfgFatalOnPassNotAvailable{"fatalOnPassNotAvailable", true, "Flag to throw a fatal if the pass is not available in the retrieved CCDB object"};
Expand Down Expand Up @@ -536,6 +543,9 @@
Produces<o2::aod::TOFEvTime> tableEvTime;
Produces<o2::aod::EvTimeTOFOnly> tableEvTimeTOFOnly;
Produces<o2::aod::pidEvTimeFlags> tableFlags;
#ifdef MERGEWITHFT0
Produces<o2::aod::FT0sCorrected> tableFt0;
#endif
static constexpr bool kRemoveTOFEvTimeBias = true; // Flag to subtract the Ev. Time bias for low multiplicity events with TOF
static constexpr float kDiamond = 6.0; // Collision diamond used in the estimation of the TOF event time
static constexpr float kErrDiamond = kDiamond * 33.356409f;
Expand Down Expand Up @@ -658,14 +668,48 @@
using ResponseImplementationEvTime = o2::pid::tof::ExpTimes<Run3TrksWtof::iterator, pid>;
void processRun3(Run3TrksWtof const& tracks,
aod::FT0s const&,
#ifdef MERGEWITHFT0
EvTimeCollisions const& collisions,
#else
EvTimeCollisionsFT0 const&,
#endif
aod::BCsWithTimestamps const& bcs)
{
if (!enableTableTOFEvTime) {
return;
}
LOG(debug) << "Processing Run3 data for TOF event time";

#ifdef MERGEWITHFT0

std::map<uint64_t, std::pair<float, float>> collisionFt0Map;
tableFt0.reserve(collisions.size());
for (const auto& collision : collisions) {
static constexpr float DefaultValue = 1e10f;
float t0A = DefaultValue;
float t0C = DefaultValue;
const float vertexPV = collision.posZ();
static constexpr float kInvLightSpeedCm2NS = 1.f / o2::constants::physics::LightSpeedCm2NS;
const float vertexCorr = vertexPV * kInvLightSpeedCm2NS;
constexpr float DummyTime = 30.f; // Due to HW limitations time can be only within range (-25,25) ns, dummy time is around 32 ns
if (collision.has_foundFT0()) {
const auto& ft0 = collision.foundFT0();
const std::bitset<8>& triggers = ft0.triggerMask();
const bool ora = triggers[o2::fit::Triggers::bitA];
const bool orc = triggers[o2::fit::Triggers::bitC];
LOGF(debug, "triggers OrA %i OrC %i ", ora, orc);
LOGF(debug, " T0A = %f, T0C %f, vertexCorr %f", ft0.timeA(), ft0.timeC(), vertexCorr);
if (ora && ft0.timeA() < DummyTime) {
t0A = ft0.timeA() + vertexCorr;
}
if (orc && ft0.timeC() < DummyTime) {
t0C = ft0.timeC() - vertexCorr;
}
}
LOGF(debug, " T0 collision time T0A = %f, T0C = %f", t0A, t0C);
tableFt0(t0A, t0C);
collisionFt0Map[collision.globalIndex()] = std::make_pair(t0A, t0C); // Store the T0A and T0C for the collision
}
#endif
tableEvTime.reserve(tracks.size());
tableFlags.reserve(tracks.size());
if (enableTableEvTimeTOFOnly) {
Expand All @@ -692,7 +736,8 @@
}
LOG(debug) << "Running on " << CollisionSystemType::getCollisionSystemName(mTOFCalibConfig.collisionSystem()) << " mComputeEvTimeWithTOF " << mComputeEvTimeWithTOF.value << " mComputeEvTimeWithFT0 " << mComputeEvTimeWithFT0.value;

if (mComputeEvTimeWithTOF == 1 && mComputeEvTimeWithFT0 == 1) {
if (mComputeEvTimeWithTOF == 1 && mComputeEvTimeWithFT0 == 1) { // Use both FT0 and TOF event times

int lastCollisionId = -1; // Last collision ID analysed
for (auto const& t : tracks) { // Loop on collisions
if (!t.has_collision() || ((sel8TOFEvTime.value == true) && !t.collision_as<EvTimeCollisionsFT0>().sel8())) { // Track was not assigned, cannot compute event time or event did not pass the event selection
Expand All @@ -710,8 +755,9 @@
lastCollisionId = t.collisionId(); /// Cache last collision ID

const auto& tracksInCollision = tracks.sliceBy(perCollision, lastCollisionId);
#ifndef MERGEWITHFT0
const auto& collision = t.collision_as<EvTimeCollisionsFT0>();

#endif
// Compute the TOF event time
const auto evTimeMakerTOF = evTimeMakerForTracks<Run3TrksWtof::iterator, filterForTOFEventTime, o2::pid::tof::ExpTimes>(tracksInCollision, mRespParamsV3, kDiamond);

Expand Down Expand Up @@ -743,14 +789,25 @@
sumOfWeights += weight;
}

#ifdef MERGEWITHFT0
float t0A = collisionFt0Map[t.collisionId()].first; // T0A for the collision
float t0C = collisionFt0Map[t.collisionId()].second; // T0C for the collision

static constexpr float ThresholdValue = 1e9f; // I the value of the FT0s are defined, the values are lower than 10e9
if ((t0A < ThresholdValue) && (t0C < ThresholdValue)) {
t0AC[0] = 0.5 * (t0A + t0C) * 1000.f;
t0AC[1] = 0.5 * std::abs(t0A - t0C) * 1000.f;
flags |= o2::aod::pidflags::enums::PIDFlags::EvTimeT0AC;

#else
if (collision.has_foundFT0()) { // T0 measurement is available
// const auto& ft0 = collision.foundFT0();
if (collision.t0ACValid()) {
t0AC[0] = collision.t0AC() * 1000.f;
t0AC[1] = collision.t0resolution() * 1000.f;
flags |= o2::aod::pidflags::enums::PIDFlags::EvTimeT0AC;
}

#endif
weight = 1.f / (t0AC[1] * t0AC[1]);
eventTime += t0AC[0] * weight;
sumOfWeights += weight;
Expand All @@ -769,7 +826,8 @@
}
}
}
} else if (mComputeEvTimeWithTOF == 1 && mComputeEvTimeWithFT0 == 0) {
} else if (mComputeEvTimeWithTOF == 1 && mComputeEvTimeWithFT0 == 0) { // Use TOF event time only

int lastCollisionId = -1; // Last collision ID analysed
for (auto const& t : tracks) { // Loop on collisions
if (!t.has_collision() || ((sel8TOFEvTime.value == true) && !t.collision_as<EvTimeCollisions>().sel8())) { // Track was not assigned, cannot compute event time or event did not pass the event selection
Expand Down Expand Up @@ -812,7 +870,8 @@
}
}
}
} else if (mComputeEvTimeWithTOF == 0 && mComputeEvTimeWithFT0 == 1) {
} else if (mComputeEvTimeWithTOF == 0 && mComputeEvTimeWithFT0 == 1) { // Use FT0 event time only

for (auto const& t : tracks) { // Loop on collisions
if (enableTableEvTimeTOFOnly) {
tableEvTimeTOFOnly((uint8_t)0, 0.f, 0.f, -1);
Expand All @@ -822,6 +881,22 @@
tableEvTime(0.f, 999.f);
continue;
}
#ifdef MERGEWITHFT0

float t0A = collisionFt0Map[t.collisionId()].first; // T0A for the collision
float t0C = collisionFt0Map[t.collisionId()].second; // T0C for the collision

static constexpr float ThresholdValue = 1e9f; // I the value of the FT0s are defined, the values are lower than 10e9
float t0AC[2] = {.0f, 999.f};
uint8_t flags = 0;
if ((t0A < ThresholdValue) && (t0C < ThresholdValue)) {
t0AC[0] = 0.5 * (t0A + t0C) * 1000.f;
t0AC[1] = 0.5 * std::abs(t0A - t0C) * 1000.f;
flags = o2::aod::pidflags::enums::PIDFlags::EvTimeT0AC;
}
tableFlags(flags);
tableEvTime(t0AC[0], t0AC[1]);
#else
const auto& collision = t.collision_as<EvTimeCollisionsFT0>();

if (collision.has_foundFT0()) { // T0 measurement is available
Expand All @@ -834,8 +909,9 @@
}
tableFlags(0);
tableEvTime(0.f, 999.f);
#endif
}
} else {
} else { // Error
LOG(fatal) << "Invalid configuration for TOF event time computation";
}
}
Expand Down Expand Up @@ -942,7 +1018,7 @@
doprocessRun2.value = false;
} else {
if (mTOFCalibConfig.autoSetProcessFunctions()) {
LOG(info) << "Autodetecting process functions";
LOG(info) << "Autodetecting process functions for mass and beta";
if (metadataInfo.isFullyDefined()) {
if (metadataInfo.isRun3()) {
doprocessRun3.value = true;
Expand Down
Loading