Skip to content

Commit ab16f49

Browse files
committed
code check fomate
1 parent 46e6df9 commit ab16f49

File tree

3 files changed

+12
-9
lines changed

3 files changed

+12
-9
lines changed

RecoHGCal/TICL/plugins/TracksterInferenceByPFN.cc

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ namespace ticl {
7474
input_Data_[1][index_tr + 3] = static_cast<float>(trackster.barycenter().y());
7575
input_Data_[1][index_tr + 4] = static_cast<float>(std::abs(trackster.barycenter().z()));
7676
input_Data_[1][index_tr + 5] = static_cast<float>(std::abs(trackster.barycenter().eta()));
77-
input_Data_[1][index_tr + 6] = static_cast<float>(trackster.barycenter().phi());
77+
input_Data_[1][index_tr + 6] = static_cast<float>(trackster.barycenter().phi());
7878

7979
// Prepare indices and sort clusters based on energy
8080
std::vector<int> clusterIndices(trackster.vertices().size());
@@ -85,7 +85,7 @@ namespace ticl {
8585
std::sort(clusterIndices.begin(), clusterIndices.end(), [&layerClusters, &trackster](const int& a, const int& b) {
8686
return layerClusters[trackster.vertices(a)].energy() > layerClusters[trackster.vertices(b)].energy();
8787
});
88-
88+
8989
std::vector<int> seenClusters(eidNLayers_, 0);
9090

9191
// Fill input data with cluster information
@@ -94,7 +94,7 @@ namespace ticl {
9494
int j = rhtools_.getLayerWithOffset(cluster.hitsAndFractions()[0].first) - 1;
9595
if (j < eidNLayers_ && seenClusters[j] < eidNClusters_) {
9696
auto index_lc = (i * eidNLayers_ + j) * eidNFeatures_ * eidNClusters_ + seenClusters[j] * eidNFeatures_;
97-
// Adding more features regarding LC, such as E, eta, phi, x, y, z, and nhits.
97+
// Adding more features regarding LC, such as E, eta, phi, x, y, z, and nhits.
9898
input_Data_[0][index_lc] =
9999
static_cast<float>(cluster.energy() / static_cast<float>(trackster.vertex_multiplicity(k)));
100100
input_Data_[0][index_lc + 1] = static_cast<float>(std::abs(cluster.eta()));
@@ -142,15 +142,16 @@ namespace ticl {
142142
void TracksterInferenceByPFN::fillPSetDescription(edm::ParameterSetDescription& iDesc) {
143143
iDesc.add<int>("algo_verbosity", 0);
144144
iDesc
145-
.add<edm::FileInPath>("onnxPIDModelPath",
146-
edm::FileInPath("RecoHGCal/TICL/data/ticlv5/onnx_models/PFN/patternrecognition/id_v0.onnx"))
145+
.add<edm::FileInPath>(
146+
"onnxPIDModelPath",
147+
edm::FileInPath("RecoHGCal/TICL/data/ticlv5/onnx_models/PFN/patternrecognition/id_v0.onnx"))
147148
->setComment("Path to ONNX PID model CLU3D");
148149
iDesc
149150
.add<edm::FileInPath>(
150151
"onnxEnergyModelPath",
151152
edm::FileInPath("RecoHGCal/TICL/data/ticlv5/onnx_models/PFN/patternrecognition/energy_v0.onnx"))
152153
->setComment("Path to ONNX Energy model CLU3D");
153-
iDesc.add<std::vector<std::string>>("inputNames", {"input","input_tr_features"});
154+
iDesc.add<std::vector<std::string>>("inputNames", {"input", "input_tr_features"});
154155
iDesc.add<std::vector<std::string>>("output_en", {"enreg_output"});
155156
iDesc.add<std::vector<std::string>>("output_id", {"pid_output"});
156157
iDesc.add<double>("eid_min_cluster_energy", 1.0);

RecoHGCal/TICL/plugins/TracksterLinksProducer.cc

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ void TracksterLinksProducer::produce(edm::Event &evt, const edm::EventSetup &es)
241241
// Run inference algorithm
242242
inferenceAlgo_->inputData(layerClusters, *resultTracksters);
243243
inferenceAlgo_->runInference(
244-
*resultTracksters); //option to use "Linking" instead of "CLU3D"/"energyAndPid" instead of "PID"
244+
*resultTracksters); //option to use "Linking" instead of "CLU3D"/"energyAndPid" instead of "PID"
245245
}
246246

247247
evt.put(std::move(linkedResultTracksters));
@@ -285,7 +285,8 @@ void TracksterLinksProducer::fillDescriptions(edm::ConfigurationDescriptions &de
285285
desc.add<edm::ParameterSetDescription>("pluginInferenceAlgoTracksterInferenceByDNN", inferenceDesc);
286286

287287
edm::ParameterSetDescription inferenceDescPFN;
288-
inferenceDescPFN.addNode(edm::PluginDescription<TracksterInferenceAlgoFactory>("type", "TracksterInferenceByPFN", true));
288+
inferenceDescPFN.addNode(
289+
edm::PluginDescription<TracksterInferenceAlgoFactory>("type", "TracksterInferenceByPFN", true));
289290
desc.add<edm::ParameterSetDescription>("pluginInferenceAlgoTracksterInferenceByPFN", inferenceDescPFN);
290291

291292
edm::ParameterSetDescription inferenceDescCNNv4;

RecoHGCal/TICL/plugins/TrackstersProducer.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,8 @@ void TrackstersProducer::fillDescriptions(edm::ConfigurationDescriptions& descri
147147
desc.add<edm::ParameterSetDescription>("pluginInferenceAlgoTracksterInferenceByDNN", inferenceDesc);
148148

149149
edm::ParameterSetDescription inferenceDescPFN;
150-
inferenceDescPFN.addNode(edm::PluginDescription<TracksterInferenceAlgoFactory>("type", "TracksterInferenceByPFN", true));
150+
inferenceDescPFN.addNode(
151+
edm::PluginDescription<TracksterInferenceAlgoFactory>("type", "TracksterInferenceByPFN", true));
151152
desc.add<edm::ParameterSetDescription>("pluginInferenceAlgoTracksterInferenceByPFN", inferenceDescPFN);
152153

153154
edm::ParameterSetDescription inferenceDescANN;

0 commit comments

Comments
 (0)