|
33 | 33 | #include "TableHelper.h" |
34 | 34 | #include "pidTPCBase.h" |
35 | 35 |
|
| 36 | +#include "Common/CCDB/ctpRateFetcher.h" |
36 | 37 | #include "Common/Core/PID/TPCPIDResponse.h" |
37 | 38 | #include "Common/DataModel/EventSelection.h" |
38 | 39 | #include "Common/DataModel/Multiplicity.h" |
39 | 40 | #include "Common/DataModel/PIDResponseTPC.h" |
40 | | -#include "Common/CCDB/ctpRateFetcher.h" |
41 | 41 | #include "Tools/ML/model.h" |
42 | 42 |
|
43 | 43 | #include "CCDB/BasicCCDBManager.h" |
@@ -154,7 +154,7 @@ struct tpcPid { |
154 | 154 | Configurable<float> networkInputBatchedMode{"networkInputBatchedMode", -1, {"-1: Takes all tracks, >0: Takes networkInputBatchedMode number of tracks at once"}}; |
155 | 155 | Configurable<std::string> irSource{"irSource", "ZNC hadronic", "Estimator of the interaction rate (Recommended: pp --> T0VTX, Pb-Pb --> ZNC hadronic)"}; |
156 | 156 | ctpRateFetcher mRateFetcher; |
157 | | - |
| 157 | + |
158 | 158 | // Parametrization configuration |
159 | 159 | bool useCCDBParam = false; |
160 | 160 | std::vector<float> track_properties; |
@@ -404,12 +404,12 @@ struct tpcPid { |
404 | 404 | track_properties[counter_track_props + 6] = trk.has_collision() ? collisions.iteratorAt(trk.collisionId()).ft0cOccupancyInTimeRange() / 60000. : 1.; |
405 | 405 | } |
406 | 406 | if (input_dimensions == 8 && networkVersion == "3") { |
407 | | - if(trk.has_collision()){ |
| 407 | + if (trk.has_collision()) { |
408 | 408 | auto trk_bc = (collisions.iteratorAt(trk.collisionId())).template bc_as<B>(); |
409 | 409 | float hadronicRate = mRateFetcher.fetch(ccdb.service, trk_bc.timestamp(), trk_bc.runNumber(), irSource) * 1.e-3; |
410 | | - track_properties[counter_track_props + 7] = hadronicRate/50.; |
411 | | - }else{ |
412 | | - track_properties[counter_track_props + 7] =1; |
| 410 | + track_properties[counter_track_props + 7] = hadronicRate / 50.; |
| 411 | + } else { |
| 412 | + track_properties[counter_track_props + 7] = 1; |
413 | 413 | } |
414 | 414 | } |
415 | 415 |
|
|
0 commit comments