LArSoft  v10_04_05
Liquid Argon Software toolkit - https://larsoft.org/
ShowerReco3D_module.cc
Go to the documentation of this file.
1 // Class: ShowerReco3D
3 // Module Type: producer
4 // File: ShowerReco3D_module.cc
5 //
6 // Generated at Tue Jan 20 07:26:48 2015 by Kazuhiro Terao using artmod
7 // from cetpkgsupport v1_08_02.
9 
16 #include "fhiclcpp/ParameterSet.h"
17 
36 
39 
40 #include <memory>
41 #include <string>
42 
43 class ShowerReco3D : public art::EDProducer {
44 public:
45  explicit ShowerReco3D(fhicl::ParameterSet const& p);
46  // The destructor generated by the compiler is fine for classes
47  // without bare pointers or other resource use.
48 
49  // Plugins should not be copied or assigned.
50  ShowerReco3D(ShowerReco3D const&) = delete;
51  ShowerReco3D(ShowerReco3D&&) = delete;
52  ShowerReco3D& operator=(ShowerReco3D const&) = delete;
53  ShowerReco3D& operator=(ShowerReco3D&&) = delete;
54 
55 private:
56  // Required functions.
57  void produce(art::Event& e) override;
58 
59  // Declare member data here.
61  std::string fInputProducer;
69 };
70 
73 {
74  fInputProducer = p.get<std::string>("InputProducer");
75  fUsePFParticle = p.get<bool>("UsePFParticle");
76  produces<std::vector<recob::Shower>>();
77  produces<art::Assns<recob::Shower, recob::Cluster>>();
78  produces<art::Assns<recob::Shower, recob::Hit>>();
79  if (fUsePFParticle) produces<art::Assns<recob::PFParticle, recob::Shower>>();
80 
81  // Instantiate algorithms
82  fCaloAlgo = new ::calo::CalorimetryAlg(p.get<fhicl::ParameterSet>("CalorimetryAlg"));
83  fShowerAlgo = new ::showerreco::ShowerRecoAlg;
84  fCPAlgoArray = new ::cmtool::CPAlgoArray;
85  fCPAlgoNHits = new ::cmtool::CPAlgoNHits;
86  fCPAlgoIgnoreTracks = new ::cmtool::CPAlgoIgnoreTracks;
87  fCFAlgoTimeOverlap = new ::cmtool::CFAlgoTimeOverlap;
88 
89  // Configure
90  fCPAlgoNHits->SetMinHits(p.get<int>("MinHits"));
91 
94 
97 
98  fShowerAlgo->Verbose(p.get<bool>("Verbosity"));
99  fShowerAlgo->SetUseArea(p.get<bool>("UseArea"));
100  fShowerAlgo->setEcorrection(p.get<bool>("ApplyMCEnergyCorrection"));
101  fShowerAlgo->CaloAlgo(fCaloAlgo);
102 
104 }
105 
107 {
108  auto const& geom = *lar::providerFrom<geo::Geometry>();
109  auto const& wireReadoutGeom = art::ServiceHandle<geo::WireReadout const>()->Get();
110  auto const clockData = art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(e);
111  auto const detProp =
113  util::GeometryUtilities const gser{geom, wireReadoutGeom, clockData, detProp};
114 
115  // Create output data product containers
116  std::unique_ptr<std::vector<recob::Shower>> out_shower_v(new std::vector<recob::Shower>);
117  std::unique_ptr<art::Assns<recob::Shower, recob::Cluster>> sc_assn(
119  std::unique_ptr<art::Assns<recob::Shower, recob::Hit>> sh_assn(
121  std::unique_ptr<art::Assns<recob::PFParticle, recob::Shower>> sp_assn(
123 
124  // Preparation
125 
126  // Reset ShowerRecoManager
127  fManager.Reset();
128 
129  // Retrieve input clusters
131  e.getByLabel(fInputProducer, cHandle);
132 
133  if (!cHandle.isValid())
134  throw cet::exception(__FUNCTION__) << "Invalid input cluster label!" << std::endl;
135 
136  // Cluster type conversion: recob::Hit => util::PxHit
137  std::vector<std::vector<::util::PxHit>> local_clusters;
138  art::FindManyP<recob::Hit> hit_m(cHandle, e, fInputProducer);
139  ::util::PxHitConverter conv{gser};
140  for (size_t i = 0; i < cHandle->size(); ++i) {
141  local_clusters.push_back(std::vector<::util::PxHit>());
142 
143  const std::vector<art::Ptr<recob::Hit>>& hits = hit_m.at(i);
144 
145  conv.GeneratePxHit(hits, local_clusters.back());
146  }
147 
148  //
149  // Run shower reconstruction
150  //
151  // shower pfparticle index (later used to make an association)
152  std::vector<size_t> shower_pfpart_index;
153  // cluster combination in terms of cluster index (for ShowerRecoManager::Reconstruct)
154  std::vector<std::vector<unsigned int>> matched_pairs;
155  // shower vector container to receive from ShowerRecoManager::Reconstruct
156  std::vector<recob::Shower> shower_v;
157 
158  if (!fUsePFParticle) {
159  matched_pairs =
160  fManager.Reconstruct(geom, wireReadoutGeom, clockData, detProp, local_clusters, shower_v);
161  }
162  else {
163 
164  // Retrieve PFParticle
166  e.getByLabel(fInputProducer, pfHandle);
167  if (!pfHandle.isValid())
168  throw cet::exception(__FUNCTION__) << "Invalid input PFParticle label!" << std::endl;
169 
170  // Make a cluster ptr => index map to fill matched_pairs
171  std::map<art::Ptr<recob::Cluster>, size_t> cmap;
172  for (size_t i = 0; i < cHandle->size(); ++i) {
173 
174  const art::Ptr<recob::Cluster> cptr(cHandle, i);
175 
176  cmap[cptr] = i;
177  }
178  // Now to fill matched_pairs retrieve association of pfpart => cluster(s)
179  art::FindManyP<recob::Cluster> cluster_m(pfHandle, e, fInputProducer);
180 
181  for (size_t i = 0; i < pfHandle->size(); ++i) {
182 
183  const art::Ptr<recob::PFParticle> pf(pfHandle, i);
184 
185  if (pf->PdgCode() != 11) continue;
186 
187  const std::vector<art::Ptr<recob::Cluster>>& clusters = cluster_m.at(i);
188 
189  std::vector<unsigned int> one_pair;
190  one_pair.reserve(clusters.size());
191 
192  for (auto const& cptr : clusters) {
193 
194  auto iter = cmap.find(cptr);
195  if (iter == cmap.end())
196  throw cet::exception(__FUNCTION__)
197  << "PFParticle=>Cluster association not valid!" << std::endl;
198 
199  one_pair.push_back((*iter).second);
200  }
201  matched_pairs.push_back(one_pair);
202  shower_pfpart_index.push_back(i);
203  }
204  // Run reconstruction
206  geom, wireReadoutGeom, clockData, detProp, local_clusters, matched_pairs, shower_v);
207  }
208 
209  // Make sure output shower vector size is same as expected length
210  if (shower_v.size() != matched_pairs.size())
211  throw cet::exception(__FUNCTION__)
212  << "Logic error: # of matched pairs != # of reco-ed showers!" << std::endl;
213 
214  // Fill output shower vector data container
215  out_shower_v->reserve(shower_v.size());
216 
217  for (size_t i = 0; i < shower_v.size(); ++i) {
218 
219  // Set output shower ID
220  shower_v[i].set_id(i);
221 
222  out_shower_v->push_back(shower_v[i]);
223 
224  // Create shower=>cluster association
225  std::vector<art::Ptr<recob::Cluster>> ass_clusters;
226  // Create shower=>hit association
227  std::vector<art::Ptr<recob::Hit>> ass_hits;
228  for (auto const& cindex : matched_pairs[i]) {
229 
230  ass_clusters.push_back(art::Ptr<recob::Cluster>(cHandle, cindex));
231 
232  const std::vector<art::Ptr<recob::Hit>>& hits = hit_m.at(cindex);
233 
234  for (auto const& ptr : hits)
235  ass_hits.push_back(ptr);
236  }
237  // Shower=>Cluster
238  util::CreateAssn(e, *(out_shower_v.get()), ass_clusters, *(sc_assn.get()));
239  // Shower=>Hits
240  util::CreateAssn(e, *(out_shower_v.get()), ass_hits, *(sh_assn.get()));
241 
242  if (fUsePFParticle) {
243 
245  e.getByLabel(fInputProducer, pfHandle);
246 
247  art::Ptr<recob::PFParticle> pf_ptr(pfHandle, shower_pfpart_index[i]);
248 
249  util::CreateAssn(e, *(out_shower_v.get()), pf_ptr, *(sp_assn.get()));
250  }
251  }
252 
253  // Store in an event record
254  e.put(std::move(out_shower_v));
255  e.put(std::move(sh_assn));
256  e.put(std::move(sc_assn));
257  if (fUsePFParticle) e.put(std::move(sp_assn));
258 }
259 
void Algo(ShowerRecoAlgBase *alg)
::calo::CalorimetryAlg * fCaloAlgo
Utilities related to art service access.
::cmtool::CPAlgoArray * fCPAlgoArray
Declaration of signal hit object.
::showerreco::ShowerRecoManager fManager
EDProducer(fhicl::ParameterSet const &pset)
Definition: EDProducer.cc:6
void AddAlgo(CPriorityAlgoBase *algo)
Setter to add a new algorithm.
Definition: CPAlgoArray.h:62
void SetMinHits(size_t n)
Setter for minimum # hits.
Definition: CPAlgoNHits.h:41
ClusterAss_t Reconstruct(geo::GeometryCore const &geom, geo::WireReadoutGeom const &wireReadoutGeom, detinfo::DetectorClocksData const &clockData, detinfo::DetectorPropertiesData const &detProp, const std::vector< std::vector< util::PxHit >> &clusters, std::vector<::recob::Shower > &showers)
cout<< "Opened file "<< fin<< " ixs= "<< ixs<< endl;if(ixs==0) hhh=(TH1F *) fff-> Get("h1")
Definition: AddMC.C:8
int PdgCode() const
Return the type of particle as a PDG ID.
Definition: PFParticle.h:79
::cmtool::CFAlgoTimeOverlap * fCFAlgoTimeOverlap
void AddMatchAlgo(CFloatAlgoBase *algo)
A simple method to add an algorithm for merging.
Definition: CMatchManager.h:46
void produce(art::Event &e) override
void CaloAlgo(calo::CalorimetryAlg *alg)
Definition: ShowerRecoAlg.h:24
bool isValid() const noexcept
Definition: Handle.h:203
PutHandle< PROD > put(std::unique_ptr< PROD > &&edp, std::string const &instance={})
Definition: Event.h:77
void hits()
Definition: readHits.C:15
std::string fInputProducer
#define DEFINE_ART_MODULE(klass)
Definition: ModuleMacros.h:65
void setEcorrection(bool on)
Function to set whether to use E correction.
Definition: ShowerRecoAlg.h:27
Class def header for a class CPAlgoArray.
Declaration of cluster object.
void SetUseArea(bool on)
Function to decide if to use Area or Pulse Amplitude for calculations.
Definition: ShowerRecoAlg.h:20
Class def header for a class CPAlgoNHits.
::cmtool::CMatchManager & MatchManager()
void Verbose(bool verbose)
Definition: ShowerRecoAlg.h:22
bool CreateAssn(art::Event &evt, std::vector< T > const &a, art::Ptr< U > const &b, art::Assns< U, T > &assn, std::string a_instance, size_t index=UINT_MAX)
Creates a single one-to-one association.
ShowerReco3D(fhicl::ParameterSet const &p)
::cmtool::CPAlgoIgnoreTracks * fCPAlgoIgnoreTracks
bool getByLabel(std::string const &label, std::string const &instance, Handle< PROD > &result) const
Utility object to perform functions of association.
::showerreco::ShowerRecoAlg * fShowerAlgo
Class def header for a class CMatchManager.
Class def header for a class CPAlgoIgnoreTracks.
ShowerReco3D & operator=(ShowerReco3D const &)=delete
Class def header for a class CFAlgoTimeOverlap.
::cmtool::CPAlgoNHits * fCPAlgoNHits
void AddPriorityAlgo(CPriorityAlgoBase *algo)
Setter to add an algorithm for priority determination.
Definition: CMManagerBase.h:71
Float_t e
Definition: plot.C:35
art framework interface to geometry description
cet::coded_exception< error, detail::translate > exception
Definition: exception.h:33
Class def header for a class ShowerRecoManager.