cvn::GCNGraphMaker2D Class Reference

Inheritance diagram for cvn::GCNGraphMaker2D:

Public Member Functions
	GCNGraphMaker2D (fhicl::ParameterSet const &pset)
	~GCNGraphMaker2D ()
void	produce (art::Event &evt)
void	beginJob ()
void	endJob ()
Public Member Functions inherited from art::EDProducer
	EDProducer (fhicl::ParameterSet const &pset)
template<typename Config >
	EDProducer (Table< Config > const &config)
std::string	workerType () const
Public Member Functions inherited from art::detail::Producer
virtual	~Producer () noexcept
	Producer (fhicl::ParameterSet const &)
	Producer (Producer const &)=delete
	Producer (Producer &&)=delete
Producer &	operator= (Producer const &)=delete
Producer &	operator= (Producer &&)=delete
void	doBeginJob (SharedResources const &resources)
void	doEndJob ()
void	doRespondToOpenInputFile (FileBlock const &fb)
void	doRespondToCloseInputFile (FileBlock const &fb)
void	doRespondToOpenOutputFiles (FileBlock const &fb)
void	doRespondToCloseOutputFiles (FileBlock const &fb)
bool	doBeginRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doEndRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doBeginSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEndSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEvent (EventPrincipal &ep, ModuleContext const &mc, std::atomic< std::size_t > &counts_run, std::atomic< std::size_t > &counts_passed, std::atomic< std::size_t > &counts_failed)
Public Member Functions inherited from art::Modifier
	~Modifier () noexcept
	Modifier ()
	Modifier (Modifier const &)=delete
	Modifier (Modifier &&)=delete
Modifier &	operator= (Modifier const &)=delete
Modifier &	operator= (Modifier &&)=delete
Public Member Functions inherited from art::ModuleBase
virtual	~ModuleBase () noexcept
	ModuleBase ()
ModuleDescription const &	moduleDescription () const
void	setModuleDescription (ModuleDescription const &)
std::array< std::vector< ProductInfo >, NumBranchTypes > const &	getConsumables () const
void	sortConsumables (std::string const &current_process_name)
template<typename T , BranchType BT>
ViewToken< T >	consumesView (InputTag const &tag)
template<typename T , BranchType BT>
ViewToken< T >	mayConsumeView (InputTag const &tag)

Private Attributes
std::string	fPixelMapLabel
	Module label for input space points. More...
float	fChargeThreshold
	Minimum charge to created a graph node from a pixel. More...
float	fNeighbourPixels
	Radius for calculating number of neighbours. More...

Additional Inherited Members
Public Types inherited from art::EDProducer
using	ModuleType = EDProducer
using	WorkerType = WorkerT< EDProducer >
Public Types inherited from art::detail::Producer
template<typename UserConfig , typename KeysToIgnore = void>
using	Table = Modifier::Table< UserConfig, KeysToIgnore >
Public Types inherited from art::Modifier
template<typename UserConfig , typename UserKeysToIgnore = void>
using	Table = ProducerTable< UserConfig, detail::ModuleConfig, UserKeysToIgnore >
Static Public Member Functions inherited from art::EDProducer
static void	commitEvent (EventPrincipal &ep, Event &e)
Protected Member Functions inherited from art::ModuleBase
ConsumesCollector &	consumesCollector ()
template<typename T , BranchType = InEvent>
ProductToken< T >	consumes (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	consumesView (InputTag const &)
template<typename T , BranchType = InEvent>
void	consumesMany ()
template<typename T , BranchType = InEvent>
ProductToken< T >	mayConsume (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	mayConsumeView (InputTag const &)
template<typename T , BranchType = InEvent>
void	mayConsumeMany ()

Detailed Description

Definition at line 32 of file GCNGraphMaker2D_module.cc.

Constructor & Destructor Documentation

cvn::GCNGraphMaker2D::GCNGraphMaker2D ( fhicl::ParameterSet const &  pset )

explicit

Definition at line 58 of file GCNGraphMaker2D_module.cc.

                                                               : art::EDProducer(pset),
  fPixelMapLabel (pset.get<std::string>    ("PixelMapLabel")),
  fChargeThreshold  (pset.get<float> ("ChargeThreshold")),
  fNeighbourPixels (pset.get<float>("NeighbourPixels"))
  {
    produces< std::vector<cvn::GCNGraph>   >();
  }

cvn::GCNGraphMaker2D::~GCNGraphMaker2D

(

)

Definition at line 67 of file GCNGraphMaker2D_module.cc.

  {
    //======================================================================
    // Clean up any memory allocated by your module
    //======================================================================
  }

Member Function Documentation

void cvn::GCNGraphMaker2D::beginJob ( )

virtual

Reimplemented from art::EDProducer.

Definition at line 75 of file GCNGraphMaker2D_module.cc.

  {  }

void cvn::GCNGraphMaker2D::endJob ( )

virtual

Reimplemented from art::EDProducer.

Definition at line 79 of file GCNGraphMaker2D_module.cc.

  {
  }

void cvn::GCNGraphMaker2D::produce ( art::Event &  evt )

virtual

Implements art::EDProducer.

Definition at line 84 of file GCNGraphMaker2D_module.cc.

  {
    auto pixelMaps = evt.getValidHandle<std::vector<cvn::PixelMap>>(fPixelMapLabel);

    // Create the Graph vector and fill it if we have enough hits
    std::unique_ptr<std::vector<cvn::GCNGraph>> graphs(new std::vector<cvn::GCNGraph>);

    if(pixelMaps->size() > 0){

      cvn::GCNFeatureUtils graphUtil;

      // Get the three graphs from the pixel map
      // These graphs contain N nodes with (wire,time) position and (charge) features
      std::vector<cvn::GCNGraph> graphs2D = graphUtil.ExtractGraphsFromPixelMap(pixelMaps->at(0),fChargeThreshold);

      // For each of the graphs we want to add a number of neighbours feature
      for(cvn::GCNGraph g : graphs2D){
        std::map<unsigned int, unsigned int> neighbourMap = graphUtil.Get2DGraphNeighbourMap(g,fNeighbourPixels);
        std::cout << "Built graph with " << g.GetNumberOfNodes() << " nodes" << std::endl;
        for(unsigned int n = 0; n < g.GetNumberOfNodes(); ++n){
          cvn::GCNGraphNode &node = g.GetNodeEditable(n);
          node.AddFeature(neighbourMap.at(n));
        }
        // Add the graph to the output vector
        graphs->push_back(g);
      } 
      
    }
    // Write our graph to the event
    evt.put(std::move(graphs));
  }

Member Data Documentation

float cvn::GCNGraphMaker2D::fChargeThreshold

private

Minimum charge to created a graph node from a pixel.

Definition at line 48 of file GCNGraphMaker2D_module.cc.

float cvn::GCNGraphMaker2D::fNeighbourPixels

private

Radius for calculating number of neighbours.

Definition at line 51 of file GCNGraphMaker2D_module.cc.

std::string cvn::GCNGraphMaker2D::fPixelMapLabel

private

Module label for input space points.

Definition at line 45 of file GCNGraphMaker2D_module.cc.

---

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

• dunereco/dunereco/CVN/art/GCNGraphMaker2D_module.cc