Skip to content

Commit 03e85bc

Browse files
njacazionjacazio
committed
Update
1 parent 3ea6b72 commit 03e85bc

File tree

6 files changed

+505
-445
lines changed

6 files changed

+505
-445
lines changed

‎Common/Core/CMakeLists.txt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ o2physics_add_library(AnalysisCore
1414
OrbitRange.cxx
1515
PID/ParamBase.cxx
1616
PID/PIDTOF.cxx
17+
PID/PIDTOFService.cxx
1718
CollisionAssociation.cxx
1819
TrackSelectionDefaults.cxx
1920
EventPlaneHelper.cxx
@@ -34,6 +35,7 @@ o2physics_target_root_dictionary(AnalysisCore
3435
PID/DetectorResponse.h
3536
PID/PIDTOF.h
3637
PID/TPCPIDResponse.h
38+
PID/PIDTOFService.h
3739
CollisionTypeHelper.h
3840
FFitWeights.h
3941
LINKDEF AnalysisCoreLinkDef.h)

‎Common/Core/PID/PIDTOF.cxx‎

Lines changed: 0 additions & 273 deletions
Original file line numberDiff line numberDiff line change
@@ -111,276 +111,3 @@ float TOFResoParamsV3::getTimeShift(float eta, int16_t sign) const
111111
}
112112

113113
} // namespace o2::pid::tof
114-
115-
#include "Framework/CommonServices.h"
116-
#include "Framework/Plugins.h"
117-
#include "Framework/ServiceHandle.h"
118-
#include "Framework/ServiceSpec.h"
119-
120-
#include "TDatabasePDG.h"
121-
122-
using namespace o2::framework;
123-
124-
o2::pid::tof::TOFResoParamsV3 o2::pid::tof::TOFResponseImpl::parameters;
125-
o2::common::core::MetadataHelper o2::pid::tof::TOFResponseImpl::metadataInfo;
126-
bool o2::pid::tof::TOFResponseImpl::mIsInit = false;
127-
int o2::pid::tof::TOFResponseImpl::mLastRunNumber = -1;
128-
129-
void o2::pid::tof::TOFResponseImpl::inheritFromBaseTask(o2::framework::InitContext& initContext, const std::string task)
130-
{
131-
if (mIsInit) {
132-
LOG(fatal) << "TOFResponseImpl already initialized, cannot re-initialize";
133-
}
134-
getCfg(initContext, "ccdb-url", mUrl, task);
135-
getCfg(initContext, "ccdb-path-grplhcif", mPathGrpLhcIf, task);
136-
getCfg(initContext, "ccdb-timestamp", mTimestamp, task);
137-
getCfg(initContext, "timeShiftCCDBPathPos", mTimeShiftCCDBPathPos, task);
138-
getCfg(initContext, "timeShiftCCDBPathNeg", mTimeShiftCCDBPathNeg, task);
139-
getCfg(initContext, "timeShiftCCDBPathPosMC", mTimeShiftCCDBPathPosMC, task);
140-
getCfg(initContext, "timeShiftCCDBPathNegMC", mTimeShiftCCDBPathNegMC, task);
141-
getCfg(initContext, "paramFileName", mParamFileName, task);
142-
getCfg(initContext, "parametrizationPath", mParametrizationPath, task);
143-
getCfg(initContext, "reconstructionPass", mReconstructionPass, task);
144-
getCfg(initContext, "reconstructionPassDefault", mReconstructionPassDefault, task);
145-
getCfg(initContext, "fatalOnPassNotAvailable", mFatalOnPassNotAvailable, task);
146-
getCfg(initContext, "enableTimeDependentResponse", mEnableTimeDependentResponse, task);
147-
getCfg(initContext, "collisionSystem", mCollisionSystem, task);
148-
getCfg(initContext, "autoSetProcessFunctions", mAutoSetProcessFunctions, task);
149-
}
150-
151-
void o2::pid::tof::TOFResponseImpl::initSetup(o2::ccdb::BasicCCDBManager* ccdb,
152-
o2::framework::InitContext& initContext)
153-
{
154-
if (mIsInit) {
155-
LOG(fatal) << "TOFResponseImpl already initialized, cannot re-initialize";
156-
}
157-
158-
if (!ccdb) {
159-
LOG(fatal) << "CCDB manager is not set, cannot initialize TOFResponseImpl";
160-
}
161-
inheritFromBaseTask(initContext); // Gets the configuration parameters from the base task (tof-signal)
162-
mCcdb = ccdb; // Set the CCDB manager
163-
mCcdb->setURL(mUrl);
164-
mCcdb->setTimestamp(mTimestamp);
165-
mCcdb->setCaching(true);
166-
mCcdb->setLocalObjectValidityChecking();
167-
// Not later than now objects
168-
mCcdb->setCreatedNotAfter(std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch()).count());
169-
170-
mIsInit = true; // Set the initialization flag
171-
172-
// Then the information about the metadata
173-
if (mReconstructionPass == "metadata") {
174-
LOG(info) << "Getting pass from metadata";
175-
if (metadataInfo.isMC()) {
176-
mReconstructionPass = metadataInfo.get("AnchorPassName");
177-
} else {
178-
mReconstructionPass = metadataInfo.get("RecoPassName");
179-
}
180-
LOG(info) << "Passed autodetect mode for pass. Taking '" << mReconstructionPass << "'";
181-
}
182-
LOG(info) << "Using parameter collection, starting from pass '" << mReconstructionPass << "'";
183-
184-
if (!mParamFileName.empty()) { // Loading the parametrization from file
185-
LOG(info) << "Loading exp. sigma parametrization from file " << mParamFileName << ", using param: " << mParametrizationPath << " and pass " << mReconstructionPass;
186-
o2::tof::ParameterCollection paramCollection;
187-
paramCollection.loadParamFromFile(mParamFileName, mParametrizationPath);
188-
LOG(info) << "+++ Loaded parameter collection from file +++";
189-
if (!paramCollection.retrieveParameters(parameters, mReconstructionPass)) {
190-
if (mFatalOnPassNotAvailable) {
191-
LOG(fatal) << "Pass '" << mReconstructionPass << "' not available in the retrieved object from file";
192-
} else {
193-
LOG(warning) << "Pass '" << mReconstructionPass << "' not available in the retrieved object from file, fetching '" << mReconstructionPassDefault << "'";
194-
if (!paramCollection.retrieveParameters(parameters, mReconstructionPassDefault)) {
195-
paramCollection.print();
196-
LOG(fatal) << "Cannot get default pass for calibration " << mReconstructionPassDefault;
197-
} else {
198-
if (metadataInfo.isRun3()) {
199-
parameters.setResolutionParametrization(paramCollection.getPars(mReconstructionPassDefault));
200-
} else {
201-
parameters.setResolutionParametrizationRun2(paramCollection.getPars(mReconstructionPassDefault));
202-
}
203-
parameters.setMomentumChargeShiftParameters(paramCollection.getPars(mReconstructionPassDefault));
204-
}
205-
}
206-
} else { // Pass is available, load non standard parameters
207-
if (metadataInfo.isRun3()) {
208-
parameters.setResolutionParametrization(paramCollection.getPars(mReconstructionPass));
209-
} else {
210-
parameters.setResolutionParametrizationRun2(paramCollection.getPars(mReconstructionPass));
211-
}
212-
parameters.setMomentumChargeShiftParameters(paramCollection.getPars(mReconstructionPass));
213-
}
214-
} else if (!mEnableTimeDependentResponse) { // Loading it from CCDB
215-
LOG(info) << "Loading initial exp. sigma parametrization from CCDB, using path: " << mParametrizationPath << " for timestamp " << mTimestamp;
216-
o2::tof::ParameterCollection* paramCollection = mCcdb->getSpecific<o2::tof::ParameterCollection>(mParametrizationPath, mTimestamp);
217-
if (!paramCollection->retrieveParameters(parameters, mReconstructionPass)) { // Attempt at loading the parameters with the pass defined
218-
if (mFatalOnPassNotAvailable) {
219-
LOG(fatal) << "Pass '" << mReconstructionPass << "' not available in the retrieved CCDB object";
220-
} else {
221-
LOG(warning) << "Pass '" << mReconstructionPass << "' not available in the retrieved CCDB object, fetching '" << mReconstructionPassDefault << "'";
222-
if (!paramCollection->retrieveParameters(parameters, mReconstructionPassDefault)) {
223-
paramCollection->print();
224-
LOG(fatal) << "Cannot get default pass for calibration " << mReconstructionPassDefault;
225-
} else {
226-
if (metadataInfo.isRun3()) {
227-
parameters.setResolutionParametrization(paramCollection->getPars(mReconstructionPassDefault));
228-
} else {
229-
parameters.setResolutionParametrizationRun2(paramCollection->getPars(mReconstructionPassDefault));
230-
}
231-
parameters.setMomentumChargeShiftParameters(paramCollection->getPars(mReconstructionPassDefault));
232-
}
233-
}
234-
} else { // Pass is available, load non standard parameters
235-
if (metadataInfo.isRun3()) {
236-
parameters.setResolutionParametrization(paramCollection->getPars(mReconstructionPass));
237-
} else {
238-
parameters.setResolutionParametrizationRun2(paramCollection->getPars(mReconstructionPass));
239-
}
240-
parameters.setMomentumChargeShiftParameters(paramCollection->getPars(mReconstructionPass));
241-
}
242-
}
243-
244-
// Loading additional calibration objects
245-
std::map<std::string, std::string> metadata;
246-
if (!mReconstructionPass.empty()) {
247-
metadata["RecoPassName"] = mReconstructionPass;
248-
}
249-
250-
auto updateTimeShift = [&](const std::string& nameShift, bool isPositive) {
251-
if (nameShift.empty()) {
252-
return;
253-
}
254-
const bool isFromFile = nameShift.find(".root") != std::string::npos;
255-
if (isFromFile) {
256-
LOG(info) << "Initializing the time shift for " << (isPositive ? "positive" : "negative") << " from file '" << nameShift << "'";
257-
parameters.setTimeShiftParameters(nameShift, "ccdb_object", isPositive);
258-
} else if (!mEnableTimeDependentResponse) { // If the response is fixed fetch it at the init time
259-
LOG(info) << "Initializing the time shift for " << (isPositive ? "positive" : "negative")
260-
<< " from ccdb '" << nameShift << "' and timestamp " << mTimestamp
261-
<< " and pass '" << mReconstructionPass << "'";
262-
mCcdb->setFatalWhenNull(false);
263-
parameters.setTimeShiftParameters(mCcdb->getSpecific<TGraph>(nameShift, mTimestamp, metadata), isPositive);
264-
mCcdb->setFatalWhenNull(true);
265-
}
266-
LOG(info) << " test getTimeShift at 0 " << (isPositive ? "pos" : "neg") << ": "
267-
<< parameters.getTimeShift(0, isPositive);
268-
};
269-
270-
const std::string nameShiftPos = metadataInfo.isMC() ? mTimeShiftCCDBPathPosMC : mTimeShiftCCDBPathPos;
271-
updateTimeShift(nameShiftPos, true);
272-
const std::string nameShiftNeg = metadataInfo.isMC() ? mTimeShiftCCDBPathNegMC : mTimeShiftCCDBPathNeg;
273-
updateTimeShift(nameShiftNeg, false);
274-
275-
// Calibration object is defined
276-
LOG(info) << "Parametrization at init time:";
277-
parameters.printFullConfig();
278-
}
279-
280-
void o2::pid::tof::TOFResponseImpl::processSetup(const int runNumber, const int64_t timeStamp)
281-
{
282-
LOG(debug) << "Processing setup for run number " << runNumber << " from run " << mLastRunNumber;
283-
// First we check if this run number was already processed
284-
if (mLastRunNumber == runNumber) {
285-
return;
286-
}
287-
LOG(info) << "Updating the parametrization from last run " << mLastRunNumber << " to " << runNumber << " and timestamp from " << mTimestamp << " " << timeStamp;
288-
mLastRunNumber = runNumber;
289-
mTimestamp = timeStamp;
290-
291-
// Check the beam type
292-
if (mCollisionSystem == o2::common::core::CollisionSystemType::kCollSysUndef) {
293-
o2::parameters::GRPLHCIFData* grpo = mCcdb->getSpecific<o2::parameters::GRPLHCIFData>(mPathGrpLhcIf,
294-
mTimestamp);
295-
mCollisionSystem = CollisionSystemType::getCollisionTypeFromGrp(grpo);
296-
} else {
297-
LOG(debug) << "Not setting collisions system as already set to " << mCollisionSystem << " " << CollisionSystemType::getCollisionSystemName(mCollisionSystem);
298-
}
299-
300-
if (!mEnableTimeDependentResponse) {
301-
return;
302-
}
303-
LOG(info) << "Updating parametrization from path '" << mParametrizationPath << "' and timestamp " << mTimestamp << " and reconstruction pass '" << mReconstructionPass << "' for run number " << runNumber;
304-
if (mParamFileName.empty()) { // Not loading if parametrization was taken from file
305-
LOG(info) << "Updating parametrization from ccdb";
306-
const o2::tof::ParameterCollection* paramCollection = mCcdb->getSpecific<o2::tof::ParameterCollection>(mParametrizationPath, mTimestamp);
307-
if (!paramCollection->retrieveParameters(parameters, mReconstructionPass)) {
308-
if (mFatalOnPassNotAvailable) {
309-
LOGF(fatal, "Pass '%s' not available in the retrieved CCDB object", mReconstructionPass.data());
310-
} else {
311-
LOGF(warning, "Pass '%s' not available in the retrieved CCDB object, fetching '%s'", mReconstructionPass.data(), mReconstructionPassDefault.data());
312-
if (!paramCollection->retrieveParameters(parameters, mReconstructionPassDefault)) {
313-
paramCollection->print();
314-
LOG(fatal) << "Cannot get default pass for calibration " << mReconstructionPassDefault;
315-
} else { // Found the default case
316-
if (metadataInfo.isRun3()) {
317-
parameters.setResolutionParametrization(paramCollection->getPars(mReconstructionPassDefault));
318-
} else {
319-
parameters.setResolutionParametrizationRun2(paramCollection->getPars(mReconstructionPassDefault));
320-
}
321-
parameters.setMomentumChargeShiftParameters(paramCollection->getPars(mReconstructionPassDefault));
322-
}
323-
}
324-
} else { // Found the non default case
325-
if (metadataInfo.isRun3()) {
326-
parameters.setResolutionParametrization(paramCollection->getPars(mReconstructionPass));
327-
} else {
328-
parameters.setResolutionParametrizationRun2(paramCollection->getPars(mReconstructionPass));
329-
}
330-
parameters.setMomentumChargeShiftParameters(paramCollection->getPars(mReconstructionPass));
331-
}
332-
}
333-
334-
// Loading additional calibration objects
335-
std::map<std::string, std::string> metadata;
336-
if (!mReconstructionPass.empty()) {
337-
metadata["RecoPassName"] = mReconstructionPass;
338-
}
339-
340-
auto updateTimeShift = [&](const std::string& nameShift, bool isPositive) {
341-
if (nameShift.empty()) {
342-
return;
343-
}
344-
const bool isFromFile = nameShift.find(".root") != std::string::npos;
345-
if (isFromFile) {
346-
return;
347-
}
348-
LOG(info) << "Updating the time shift for " << (isPositive ? "positive" : "negative")
349-
<< " from ccdb '" << nameShift << "' and timestamp " << mTimestamp
350-
<< " and pass '" << mReconstructionPass << "'";
351-
mCcdb->setFatalWhenNull(false);
352-
parameters.setTimeShiftParameters(mCcdb->getSpecific<TGraph>(nameShift, mTimestamp, metadata), isPositive);
353-
mCcdb->setFatalWhenNull(true);
354-
LOG(info) << " test getTimeShift at 0 " << (isPositive ? "pos" : "neg") << ": "
355-
<< parameters.getTimeShift(0, isPositive);
356-
};
357-
358-
updateTimeShift(metadataInfo.isMC() ? mTimeShiftCCDBPathPosMC : mTimeShiftCCDBPathPos, true);
359-
updateTimeShift(metadataInfo.isMC() ? mTimeShiftCCDBPathNegMC : mTimeShiftCCDBPathNeg, false);
360-
361-
LOG(info) << "Parametrization at setup time:";
362-
parameters.printFullConfig();
363-
}
364-
365-
struct TOFSupport : o2::framework::ServicePlugin {
366-
o2::framework::ServiceSpec* create() final
367-
{
368-
return new ServiceSpec{
369-
.name = "tof-response",
370-
.init = [](ServiceRegistryRef, DeviceState&, fair::mq::ProgOptions&) -> ServiceHandle {
371-
auto* wrapper = new o2::pid::tof::TOFResponse();
372-
auto* ptr = new o2::pid::tof::TOFResponseImpl();
373-
wrapper->setInstance(ptr);
374-
return ServiceHandle{TypeIdHelpers::uniqueId<o2::pid::tof::TOFResponse>(), wrapper, ServiceKind::Serial, "database-pdg"};
375-
},
376-
.configure = CommonServices::noConfiguration(),
377-
.exit = [](ServiceRegistryRef, void* service) {
378-
auto* resp = reinterpret_cast<o2::pid::tof::TOFResponse*>(service);
379-
delete resp; },
380-
.kind = ServiceKind::Serial};
381-
}
382-
};
383-
384-
DEFINE_DPL_PLUGINS_BEGIN
385-
DEFINE_DPL_PLUGIN_INSTANCE(TOFSupport, CustomService);
386-
DEFINE_DPL_PLUGINS_END

0 commit comments

Comments
 (0)