cnn::RegCNNVtxHandler Class Reference

RegCNNVtxHandler, basic output of CNN neural net. More...

#include <RegCNNVtxHandler.h>

Public Member Functions
	RegCNNVtxHandler (const fhicl::ParameterSet &pset)
std::vector< float >	GetVertex (detinfo::DetectorClocksData const &clockData, detinfo::DetectorPropertiesData const &detProp, art::Event &evt, const RegPixelMap &pixelmap)

Private Member Functions
void	FindGlobalVertices (const RegPixelMap &pm, std::vector< float > &outputs)
void	getCM (const RegPixelMap &pm, float *cm_list)

Private Attributes
std::string	fHitsModuleLabel
cnn::TFRegNetHandler	fTFHandler1st
cnn::TFRegNetHandler	fTFHandler2nd
unsigned int	fTdcWidth
unsigned int	fWireLength
unsigned int	fTimeResolution
unsigned int	fWireResolution
unsigned int	fGlobalWireMethod
RegPixelMapProducer	fProducer

Detailed Description

RegCNNVtxHandler, basic output of CNN neural net.

Definition at line 23 of file RegCNNVtxHandler.h.

Constructor & Destructor Documentation

cnn::RegCNNVtxHandler::RegCNNVtxHandler

(

const fhicl::ParameterSet &

pset

)

Definition at line 21 of file RegCNNVtxHandler.cxx.

                                                                 :
    fHitsModuleLabel  (pset.get<std::string>    ("HitsModuleLabel")),
    fTFHandler1st     (pset.get<fhicl::ParameterSet> ("TFNetHandler1st")),
    fTFHandler2nd     (pset.get<fhicl::ParameterSet> ("TFNetHandler2nd")),
    fTdcWidth         (pset.get<unsigned short>     ("TdcWidth")),
    fWireLength       (pset.get<unsigned short>     ("WireLength")),
    fTimeResolution   (pset.get<unsigned short>     ("TimeResolution")),
    fWireResolution   (pset.get<unsigned short>     ("WireResolution")),
    fGlobalWireMethod (pset.get<int>                ("GlobalWireMethod")),
//    fProngOnly        (pset.get<bool>               ("ProngOnly")),
    fProducer      (fWireLength, fWireResolution, fTdcWidth, fTimeResolution, fGlobalWireMethod, 0, 1)
  {
  }

Member Function Documentation

void cnn::RegCNNVtxHandler::FindGlobalVertices ( const RegPixelMap &  pm, std::vector< float > &  outputs  )

private

Definition at line 69 of file RegCNNVtxHandler.cxx.

  {
    for (int ii = 0; ii < 3; ii++){
        float vtx_wire = outputs[ii*2+1];
        float vtx_tick = outputs[ii*2];
        float dwire = pm.fNWire/2-vtx_wire;
        float dtick = (pm.fNTdc/2-vtx_tick)*pm.fNTRes;
        float cm_wire = pm.fBound.fFirstWire[ii]+pm.fNWire/2-dwire;
        float cm_tdc  = pm.fBound.fFirstTDC[ii]+pm.fNTdc*pm.fNTRes/2-dtick;
        outputs[ii*2+1] = cm_wire;
        outputs[ii*2] = cm_tdc;
    }
  }

void cnn::RegCNNVtxHandler::getCM ( const RegPixelMap &  pm, float *  cm_list  )

private

Definition at line 82 of file RegCNNVtxHandler.cxx.

  {
    for (int ii = 0; ii < 3; ii++){
        float mean_wire = pm.fBound.fFirstWire[ii]+pm.fNWire/2;
        float mean_tdc  = pm.fBound.fFirstTDC[ii]+pm.fNTdc*pm.fNTRes/2;
        cm_list[2*ii] = mean_tdc;
        cm_list[2*ii+1] = mean_wire;
    }
  }

std::vector< float > cnn::RegCNNVtxHandler::GetVertex	(	detinfo::DetectorClocksData const &	clockData,
		detinfo::DetectorPropertiesData const &	detProp,
		art::Event &	evt,
		const RegPixelMap &	pixelmap
	)

Load in the pixel map

Definition at line 34 of file RegCNNVtxHandler.cxx.

                                                                                            {

      std::vector< art::Ptr< recob::Hit > > hitlist;
      auto hitListHandle = evt.getHandle< std::vector< recob::Hit > >(fHitsModuleLabel);
      if (hitListHandle)
        art::fill_ptr_vector(hitlist, hitListHandle);
      art::FindManyP<recob::Wire> fmwire(hitListHandle, evt, fHitsModuleLabel);

      /// Load in the pixel map
      // get vertex
      std::vector <float> Result(3, -99999);
      if (pixelmap.fInPM){
              std::vector<float> networkOutput(6, -99999);
              networkOutput = fTFHandler1st.Predict(pixelmap);
              FindGlobalVertices(pixelmap, networkOutput);
              std::vector <float> center_of_mass(7,0);
              for (int ii = 0; ii < 6; ii++){
                      center_of_mass[ii] = networkOutput[ii];
              }
              networkOutput[1] += 7; networkOutput[5] += 7;
              networkOutput[3] -= 9;

              RegPixelMap pm;
              pm = fProducer.CreateMap(clockData, detProp, hitlist, fmwire, networkOutput);
              if (pm.fInPM){
                      center_of_mass[6] = (float)(pm.fTPC%4); // add TPC info
                      Result = fTFHandler2nd.Predict(pm, center_of_mass);
              }
      } // end of pixelmap
      return Result;
  } // end og GetVertex

Member Data Documentation

unsigned int cnn::RegCNNVtxHandler::fGlobalWireMethod

private

Definition at line 43 of file RegCNNVtxHandler.h.

std::string cnn::RegCNNVtxHandler::fHitsModuleLabel

private

Definition at line 36 of file RegCNNVtxHandler.h.

RegPixelMapProducer cnn::RegCNNVtxHandler::fProducer

private

Definition at line 45 of file RegCNNVtxHandler.h.

unsigned int cnn::RegCNNVtxHandler::fTdcWidth

private

Definition at line 39 of file RegCNNVtxHandler.h.

cnn::TFRegNetHandler cnn::RegCNNVtxHandler::fTFHandler1st

private

Definition at line 37 of file RegCNNVtxHandler.h.

cnn::TFRegNetHandler cnn::RegCNNVtxHandler::fTFHandler2nd

private

Definition at line 38 of file RegCNNVtxHandler.h.

unsigned int cnn::RegCNNVtxHandler::fTimeResolution

private

Definition at line 41 of file RegCNNVtxHandler.h.

unsigned int cnn::RegCNNVtxHandler::fWireLength

private

Definition at line 40 of file RegCNNVtxHandler.h.

unsigned int cnn::RegCNNVtxHandler::fWireResolution

private

Definition at line 42 of file RegCNNVtxHandler.h.

---

The documentation for this class was generated from the following files:

• dunereco/dunereco/RegCNN/art/RegCNNVtxHandler.h
• dunereco/dunereco/RegCNN/art/RegCNNVtxHandler.cxx