Inheritance diagram for FilterDecoder:

Public Member Functions
	FilterDecoder (const fhicl::ParameterSet &pset)
	Constructor. More...

virtual	~FilterDecoder () 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

Detailed Description

Definition at line 9 of file FilterDecoder_tool.cc.

Constructor & Destructor Documentation

FilterDecoder::FilterDecoder ( const fhicl::ParameterSet & pset )

Constructor.

Parameters

pset

Definition at line 51 of file FilterDecoder_tool.cc.

51 : DecoderToolBase{p} {}

DecoderToolBase

Definition: DecoderToolBase.h:25

virtual FilterDecoder::~FilterDecoder ( )

virtualdefaultnoexcept

Virtual Destructor.

Member Function Documentation

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

inlineoverridevirtual

declareProducts function

Parameters

art::ProducesCollector

Implements DecoderToolBase.

Definition at line 29 of file FilterDecoder_tool.cc.

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

   {
     collector.produces<vector<FeatureVector<1>>>(instancename);
   }

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

inlineinherited

Definition at line 69 of file DecoderToolBase.h.

Referenced by 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 FilterDecoder::writeEmptyToEvent	(	art::Event &	e,
		const vector< vector< size_t >> &	idsmap
	)

overridevirtual

writeEmptyToEvent function

Parameters

art::Event event record

Implements DecoderToolBase.

Definition at line 53 of file FilterDecoder_tool.cc.

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

Referenced by declareProducts().

 {
   //
   size_t size = 0;
   for (auto& v : idsmap)
     size += v.size();
   auto filtcol =
     std::make_unique<vector<FeatureVector<1>>>(size, FeatureVector<1>(std::array<float, 1>({-1.})));
   e.put(std::move(filtcol), instancename);
   //
 }

void FilterDecoder::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 65 of file FilterDecoder_tool.cc.

References DecoderToolBase::debug, DEFINE_ART_CLASS_TOOL, f, DecoderToolBase::instancename, DecoderToolBase::outputname, DecoderToolBase::planes, DecoderToolBase::printVector(), art::Event::put(), and util::size().

Referenced by declareProducts().

 {
   //
   size_t size = 0;
   for (auto& v : idsmap)
     size += v.size();
   auto filtcol =
     std::make_unique<vector<FeatureVector<1>>>(size, FeatureVector<1>(std::array<float, 1>({-1.})));
   //
   for (size_t p = 0; p < planes.size(); p++) {
     //
     const std::vector<float>* x_filter_data = 0;
     for (auto& io : infer_output) {
       if (io.output_name == outputname + planes[p]) x_filter_data = &io.output_vec;
     }
     if (debug) {
       std::cout << outputname + planes[p] << std::endl;
       printVector(*x_filter_data);
     }
     //
     torch::TensorOptions options = torch::TensorOptions().dtype(torch::kFloat32);
     int64_t num_elements = x_filter_data->size();
     const torch::Tensor f =
       torch::from_blob(const_cast<float*>(x_filter_data->data()), {num_elements}, options);
     //
     for (int i = 0; i < f.numel(); ++i) {
       size_t idx = idsmap[p][i];
       std::array<float, 1> input({f[i].item<float>()});
       (*filtcol)[idx] = FeatureVector<1>(input);
     }
   }
   e.put(std::move(filtcol), instancename);
 }