VertexDecoder Class Reference

Inheritance diagram for VertexDecoder:

Public Member Functions
	VertexDecoder (const fhicl::ParameterSet &pset)
	Constructor. More...
virtual	~VertexDecoder () noexcept=default
	Virtual Destructor. More...
void	declareProducts (art::ProducesCollector &collector) override
	declareProducts function More...
void	writeEmptyToEvent (art::Event &e, const vector< vector< size_t >> &idsmap) override
	writeEmptyToEvent function More...
void	writeToEvent (art::Event &e, const vector< vector< size_t >> &idsmap, const vector< NuGraphOutput > &infer_output) override
	Decoder function. More...
void	printVector (const std::vector< float > &vec)
template<typename T , size_t N>
void	softmax (std::array< T, N > &arr)
void	setDebugAndPlanes (bool d, vector< std::string > &p)

Protected Attributes
bool	debug
vector< std::string >	planes
std::string	instancename
std::string	outputname

Private Attributes
string	outputDictElem

Detailed Description

Definition at line 6 of file VertexDecoder_tool.cc.

Constructor & Destructor Documentation

VertexDecoder::VertexDecoder

(

const fhicl::ParameterSet &

pset

)

Constructor.

Parameters

pset

Definition at line 51 of file VertexDecoder_tool.cc.

References fhicl::ParameterSet::get().

  : DecoderToolBase(p), outputDictElem{p.get<string>("outputDictElem")}
{}

virtual VertexDecoder::~VertexDecoder ( )

virtualdefaultnoexcept

Virtual Destructor.

Member Function Documentation

void VertexDecoder::declareProducts ( art::ProducesCollector &  collector )

inlineoverridevirtual

declareProducts function

Parameters

art::ProducesCollector

Implements DecoderToolBase.

Definition at line 26 of file VertexDecoder_tool.cc.

References e, DecoderToolBase::instancename, art::ProducesCollector::produces(), lar::dump::vector(), writeEmptyToEvent(), and writeToEvent().

  {
    collector.produces<vector<recob::Vertex>>(instancename);
  }

void DecoderToolBase::printVector ( const std::vector< float > &  vec )

inlineinherited

Definition at line 69 of file DecoderToolBase.h.

Referenced by FilterDecoder::writeToEvent(), and SemanticDecoder::writeToEvent().

  {
    for (size_t i = 0; i < vec.size(); ++i) {
      std::cout << vec[i];
      // Print space unless it's the last element
      if (i != vec.size() - 1) { std::cout << " "; }
    }
    std::cout << std::endl;
    std::cout << std::endl;
  }

void DecoderToolBase::setDebugAndPlanes ( bool  d, vector< std::string > &  p  )

inlineinherited

Definition at line 98 of file DecoderToolBase.h.

References d, and tca::debug.

  {
    debug = d;
    planes = p;
  }

template<typename T , size_t N>

void DecoderToolBase::softmax ( std::array< T, N > &  arr )

inlineinherited

Definition at line 81 of file DecoderToolBase.h.

References sum.

Referenced by SemanticDecoder::writeToEvent().

  {
    T m = -std::numeric_limits<T>::max();
    for (size_t i = 0; i < arr.size(); i++) {
      if (arr[i] > m) { m = arr[i]; }
    }
    T sum = 0.0;
    for (size_t i = 0; i < arr.size(); i++) {
      sum += expf(arr[i] - m);
    }
    T offset = m + logf(sum);
    for (size_t i = 0; i < arr.size(); i++) {
      arr[i] = expf(arr[i] - offset);
    }
    return;
  }

void VertexDecoder::writeEmptyToEvent ( art::Event &  e, const vector< vector< size_t >> &  idsmap  )

overridevirtual

writeEmptyToEvent function

Parameters

art::Event

event record

Implements DecoderToolBase.

Definition at line 55 of file VertexDecoder_tool.cc.

References DecoderToolBase::instancename, and art::Event::put().

Referenced by declareProducts().

{
  //
  auto vertcol = std::make_unique<vector<recob::Vertex>>();
  e.put(std::move(vertcol), instancename);
  //
}

void VertexDecoder::writeToEvent ( art::Event &  e, const vector< vector< size_t >> &  idsmap, const vector< NuGraphOutput > &  infer_output  )

overridevirtual

Decoder function.

Parameters

art::Event

event record for decoder

Implements DecoderToolBase.

Definition at line 63 of file VertexDecoder_tool.cc.

References DecoderToolBase::debug, DEFINE_ART_CLASS_TOOL, DecoderToolBase::instancename, outputDictElem, and art::Event::put().

Referenced by declareProducts().

{
  //
  auto vertcol = std::make_unique<vector<recob::Vertex>>();

  const std::vector<float>* x_vertex_data = nullptr;
  for (auto& io : infer_output) {
    if (io.output_name == outputDictElem) x_vertex_data = &io.output_vec;
  }
  if (x_vertex_data->size() == 3) {
    double vpos[3] = {(*x_vertex_data)[0], (*x_vertex_data)[1], (*x_vertex_data)[2]};
    vertcol->push_back(recob::Vertex(vpos));
    if (debug)
      std::cout << "NuGraph vertex pos=" << vpos[0] << ", " << vpos[1] << ", " << vpos[2]
                << std::endl;
  }
  else {
    std::cout << "ERROR -- Wrong size returned by NuGraph vertex decoder" << std::endl;
  }
  e.put(std::move(vertcol), instancename);
  //
}

Member Data Documentation

bool DecoderToolBase::debug

protectedinherited

Definition at line 105 of file DecoderToolBase.h.

Referenced by writeToEvent(), FilterDecoder::writeToEvent(), and SemanticDecoder::writeToEvent().

std::string DecoderToolBase::instancename

protectedinherited

Definition at line 107 of file DecoderToolBase.h.

Referenced by declareProducts(), FilterDecoder::declareProducts(), SemanticDecoder::declareProducts(), writeEmptyToEvent(), FilterDecoder::writeEmptyToEvent(), SemanticDecoder::writeEmptyToEvent(), writeToEvent(), FilterDecoder::writeToEvent(), and SemanticDecoder::writeToEvent().

string VertexDecoder::outputDictElem

private

Definition at line 48 of file VertexDecoder_tool.cc.

Referenced by writeToEvent().

std::string DecoderToolBase::outputname

protectedinherited

Definition at line 108 of file DecoderToolBase.h.

Referenced by FilterDecoder::writeToEvent(), and SemanticDecoder::writeToEvent().

vector<std::string> DecoderToolBase::planes

protectedinherited

Definition at line 106 of file DecoderToolBase.h.

Referenced by FilterDecoder::writeToEvent(), and SemanticDecoder::writeToEvent().

---

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

• larrecodnn/v10_01_05/source/larrecodnn/NuGraph/Tools/VertexDecoder_tool.cc