Program Listing for File PyFlowMatchData.hh

↰ Return to documentation for file (larflow/PrepFlowMatchData/arxiv/PyFlowMatchData.hh)

#ifndef __LARFLOW_PYFLOWMATCHDATA_H__
#define __LARFLOW_PYFLOWMATCHDATA_H__

struct _object;
typedef _object PyObject;

#include <Python.h>
#include "bytesobject.h"
#include <vector>
#include "PrepFlowMatchData.hh"
#include "core/DataFormat/larflow3dhit.h"
#include "larcv/core/DataFormat/ImageMeta.h"
#include "larcv/core/DataFormat/Image2D.h"
#include "larcv/core/DataFormat/EventChStatus.h"

namespace larflow {

  // get a randomly sampled set of flow pixel pairs, returned in a numpy array
  // used for training
  PyObject* sample_pair_array( const int& nsamples, const FlowMatchMap& matchdata, int& nfilled, bool with_truth=false );

  // get a set of flow pixel pairs for a squential set of source pixels
  // used for deploy
  PyObject* get_chunk_pair_array( const int& start_source_pixel_index,
                                  const int& max_num_pairs,
                                  const FlowMatchMap& matchdata,
                                  int& last_source_pixel_index,
                                  int& num_pairs_filled,
                                  bool with_truth=false );

  // internal function that makes array, filling it with source pixels specified by indices
  PyObject* _make_pair_array( const FlowMatchMap& matchdata,
                              const std::vector<size_t>& idx_v,
                              const int start_idx,
                              const int max_num_pairs,
                              int& nsource_pixels_covered,
                              int& num_pairs_filled,
                              bool withtruth );

  // convert to larflow3dhits
  void make_larflow_hits( PyObject* pair_probs,
                          PyObject* source_sparseimg, PyObject* target_sparseimg,
                          PyObject* matchpairs,
                          const int source_plane, const int target_plane,
                          const larcv::ImageMeta& source_meta,
                          const std::vector<larcv::Image2D>& img_v,
                          larlite::event_larflow3dhit& hit_v,
                          const larcv::EventChStatus* ev_chstatus=0 );

  // convert to larflow3dhits
  void make_larflow_hits_with_deadchs( PyObject* pair_probs,
                                       PyObject* source_sparseimg, PyObject* target_sparseimg,
                                       PyObject* matchpairs,
                                       const int source_plane, const int target_plane,
                                       const larcv::ImageMeta& source_meta,
                                       const std::vector<larcv::Image2D>& img_v,
                                       const larcv::EventChStatus& ev_status,
                                       larlite::event_larflow3dhit& hit_v );



}

#endif