Class SplitHitsBySSNet

Inheritance Relationships

Base Type

  • public larcv_base

Class Documentation

class SplitHitsBySSNet : public larcv_base

Separate a set of larflow3dhit using the SSNet score.

Public Functions

SplitHitsBySSNet()
virtual ~SplitHitsBySSNet()
void split(const std::vector<larcv::Image2D> &ssnet_score_v, const larlite::event_larflow3dhit &lfhit_v, const float ssnet_score_threshold, const float larmatch_score_threshold, std::vector<larlite::larflow3dhit> &accept_v, std::vector<larlite::larflow3dhit> &reject_v)

split-up container of larflow3dhit using ssnet output images

Parameters
  • ssnet_score_v: SSNet shower score images for each plane
  • larmatch_hit_v: LArMatch hits
  • ssnet_score_threshold: Threshold shower score
  • larmatch_score_threshold: Threshold larmatch score
  • accept_v: Hits above threshold
  • reject_v: Hits below threshold

void process(larcv::IOManager &iolcv, larlite::storage_manager &ioll)

Process event data in the larcv and larlite IO managers.

Parameters
  • iolcv: LArCV IO manager
  • ioll: larlite IO manager

void set_ssnet_threshold(float thresh)

set the ssnet shower score threshold

void set_larmatch_threshold(float thresh)

set the larmatch treshold score

void set_ssnet_tree_stem_name(std::string stem)

set stem name of tree holding ssnet images

void set_larmatch_tree_name(std::string hitname)

set name of tree holding larmatch hits

void set_adc_tree_name(std::string name)

set name of tree holding wire plane images

void set_output_tree_stem_name(std::string stem)

set name of tree to put output hits

std::vector<larlite::larflow3dhit> &get_shower_hits()

get mutable shower larflow3dhit container

std::vector<larlite::larflow3dhit> &get_track_hits()

get mutable track larflow3dhit container

const std::vector<larlite::larflow3dhit> &get_shower_hits() const

get const shower larflow3dhit container

const std::vector<larlite::larflow3dhit> &get_track_hits() const

get const track larflow3dhit container

Protected Attributes

float _score_threshold

ssnet shower score threshold to be labeled shower

float _larmatch_threshold

larmatch score threshold to be considered

std::string _ssnet_stem_name

stem name of tree holding ssnet images

std::string _adc_name

name of tree holding wire plane images

std::string _input_larmatch_hit_tree_name

name of tree holding larflow3dhit made by larmatch network

std::string _output_larmatch_hit_stem_name

stem name of tree holding output hits

std::vector<larlite::larflow3dhit> _shower_hit_v

container holding shower-labeled hits

std::vector<larlite::larflow3dhit> _track_hit_v

container holding track-labeled hits