.. _program_listing_file_larflow_PrepFlowMatchData_arxiv_PyFlowMatchData.hh:

Program Listing for File PyFlowMatchData.hh
===========================================

|exhale_lsh| :ref:`Return to documentation for file <file_larflow_PrepFlowMatchData_arxiv_PyFlowMatchData.hh>` (``larflow/PrepFlowMatchData/arxiv/PyFlowMatchData.hh``)

.. |exhale_lsh| unicode:: U+021B0 .. UPWARDS ARROW WITH TIP LEFTWARDS

.. code-block:: cpp

   #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