factories/pid/IrtCherenkovParticleID_factory.h

0001 // Copyright 2023, Christopher Dilks
0002 // Subject to the terms in the LICENSE file found in the top-level directory.
0003
0004 #pragma once
0005
0006 #include <IRT/CherenkovDetectorCollection.h>
0007 #include <JANA/JEvent.h>
0008 #include <edm4eic/CherenkovParticleIDCollection.h>
0009 #include <algorithm>
0010 #include <memory>
0011 #include <string>
0012 #include <utility>
0013 #include <vector>
0014
0015 #include "algorithms/pid/IrtCherenkovParticleID.h"
0016 #include "algorithms/pid/IrtCherenkovParticleIDConfig.h"
0017 #include "extensions/jana/JOmniFactory.h"
0018 #include "services/algorithms_init/AlgorithmsInit_service.h"
0019 #include "services/geometry/richgeo/RichGeo_service.h"
0020
0021 namespace eicrecon {
0022
0023 class IrtCherenkovParticleID_factory
0024     : public JOmniFactory<IrtCherenkovParticleID_factory, IrtCherenkovParticleIDConfig> {
0025
0026 private:
0027   using AlgoT = eicrecon::IrtCherenkovParticleID;
0028   std::unique_ptr<AlgoT> m_algo;
0029
0030   PodioInput<edm4eic::TrackSegment> m_aerogel_tracks_input{this};
0031   PodioInput<edm4eic::TrackSegment> m_gas_tracks_input{this};
0032   PodioInput<edm4eic::TrackSegment> m_merged_tracks_input{this};
0033   PodioInput<edm4eic::RawTrackerHit> m_raw_hits_input{this};
0034   PodioInput<edm4eic::MCRecoTrackerHitAssociation> m_raw_hit_assoc_input{this};
0035   PodioOutput<edm4eic::CherenkovParticleID> m_aerogel_particleIDs_output{this};
0036   PodioOutput<edm4eic::CherenkovParticleID> m_gas_particleIDs_output{this};
0037
0038   ParameterRef<unsigned int> m_numRIndexBins{this, "numRIndexBins", config().numRIndexBins, ""};
0039   ParameterRef<std::vector<int>> m_pdgList{this, "pdgList", config().pdgList, ""};
0040
0041   ParameterRef<double> m_aerogel_referenceRIndex{this, "aerogel:referenceRIndex",
0042                                                  config().radiators["Aerogel"].referenceRIndex, ""};
0043   ParameterRef<double> m_aerogel_attenuation{this, "aerogel:attenuation",
0044                                              config().radiators["Aerogel"].attenuation, ""};
0045   ParameterRef<std::string> m_aerogel_smearingMode{this, "aerogel:smearingMode",
0046                                                    config().radiators["Aerogel"].smearingMode, ""};
0047   ParameterRef<double> m_aerogel_smearing{this, "aerogel:smearing",
0048                                           config().radiators["Aerogel"].smearing, ""};
0049
0050   ParameterRef<double> m_gas_referenceRIndex{this, "gas:referenceRIndex",
0051                                              config().radiators["Gas"].referenceRIndex, ""};
0052   ParameterRef<double> m_gas_attenuation{this, "gas:attenuation",
0053                                          config().radiators["Gas"].attenuation, ""};
0054   ParameterRef<std::string> m_gas_smearingMode{this, "gas:smearingMode",
0055                                                config().radiators["Gas"].smearingMode, ""};
0056   ParameterRef<double> m_gas_smearing{this, "gas:smearing", config().radiators["Gas"].smearing, ""};
0057
0058   ParameterRef<bool> m_cheatPhotonVertex{this, "cheatPhotonVertex", config().cheatPhotonVertex, ""};
0059   ParameterRef<bool> m_cheatTrueRadiator{this, "cheatTrueRadiator", config().cheatTrueRadiator, ""};
0060
0061   Service<AlgorithmsInit_service> m_algorithmsInit{this};
0062   Service<RichGeo_service> m_RichGeoSvc{this};
0063
0064 public:
0065   void Configure() {
0066     m_algo = std::make_unique<AlgoT>(GetPrefix());
0067     m_algo->level(static_cast<algorithms::LogLevel>(logger()->level()));
0068     m_algo->applyConfig(config());
0069     m_algo->init(m_RichGeoSvc().GetIrtGeo("DRICH")->GetIrtDetectorCollection());
0070   }
0071
0072   void ChangeRun(int32_t /* run_number */) {}
0073
0074   void Process(int32_t /* run_number */, uint64_t /* event_number */) {
0075     m_algo->process({m_aerogel_tracks_input(), m_gas_tracks_input(), m_merged_tracks_input(),
0076                      m_raw_hits_input(), m_raw_hit_assoc_input()},
0077                     {m_aerogel_particleIDs_output().get(), m_gas_particleIDs_output().get()});
0078   }
0079 };
0080
0081 } // namespace eicrecon