RegCNNVtxHandler.h
Go to the documentation of this file.
1 ////////////////////////////////////////////////////////////////////////
2 /// \file RegCNNVtxNetHandler.h
3 /// \brief Reg CNN Vertex Handler
4 /// \author Ilsoo Seong - iseong@uci.edu
5 ////////////////////////////////////////////////////////////////////////
6 
7 #ifndef REGCNN_VTXHANDLER_H
8 #define REGCNN_VTXHANDLER_H
9 
10 #include <vector>
11 #include <memory>
12 
13 #include "fhiclcpp/ParameterSet.h"
19 
20 namespace cnn
21 {
22  /// RegCNNVtxHandler, basic output of CNN neural net
24  {
25  public:
27 
28  std::vector<float> GetVertex(detinfo::DetectorClocksData const& clockData,
29  detinfo::DetectorPropertiesData const& detProp,
30  art::Event& evt, const RegPixelMap &pixelmap);
31 
32  private:
33  void FindGlobalVertices(const RegPixelMap &pm, std::vector<float> &outputs);
34  void getCM(const RegPixelMap& pm, float* cm_list);
35 
39  unsigned int fTdcWidth;
40  unsigned int fWireLength;
41  unsigned int fTimeResolution;
42  unsigned int fWireResolution;
43  unsigned int fGlobalWireMethod;
44 // bool fProngOnly;
46 
47  };
48 
49 }
50 
51 #endif // REGCNN_VTXHANDLER_H
RegPixelMapProducer fProducer
unsigned int fWireResolution
std::string string
Definition: nybbler.cc:12
RegPixelMap, basic input to CNN neural net.
Definition: RegPixelMap.h:22
void FindGlobalVertices(const RegPixelMap &pm, std::vector< float > &outputs)
Producer algorithm for RegPixelMap, input to CVN neural net.
unsigned int fGlobalWireMethod
RegCNNVtxHandler, basic output of CNN neural net.
cnn::TFRegNetHandler fTFHandler1st
cnn::TFRegNetHandler fTFHandler2nd
void getCM(const RegPixelMap &pm, float *cm_list)
RegPixelMapProducer for RegCNN modified from PixelMapProducer.h.
Wrapper for caffe::Net which handles construction and prediction.
std::string fHitsModuleLabel
Defines an enumeration for cellhit classification.
Contains all timing reference information for the detector.
TFRegNetHandler for RegCNN modified from TFNetHandler.h.
TCEvent evt
Definition: DataStructs.cxx:7
unsigned int fTimeResolution
RegPixelMap for RegCNN modified from PixelMap.h.
std::vector< float > GetVertex(detinfo::DetectorClocksData const &clockData, detinfo::DetectorPropertiesData const &detProp, art::Event &evt, const RegPixelMap &pixelmap)
RegCNNVtxHandler(const fhicl::ParameterSet &pset)