.. _program_listing_file_larflow_Voxelizer_VoxelizeTriplets.h:

Program Listing for File VoxelizeTriplets.h
===========================================

|exhale_lsh| :ref:`Return to documentation for file <file_larflow_Voxelizer_VoxelizeTriplets.h>` (``larflow/Voxelizer/VoxelizeTriplets.h``)

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

.. code-block:: cpp

   #ifndef __VOXELIZE_TRIPLETS_H__
   #define __VOXELIZE_TRIPLETS_H__
   
   #include <Python.h>
   #include "bytesobject.h"
   
   #define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
   #include <numpy/ndarrayobject.h>
   
   #include <string>
   #include <vector>
   
   #include "larcv/core/DataFormat/IOManager.h"
   #include "DataFormat/storage_manager.h"
   
   #include "larflow/PrepFlowMatchData/PrepMatchTriplets.h"
   
   namespace larflow {
   namespace voxelizer {
   
     class VoxelizeTriplets {
   
     public:
   
       VoxelizeTriplets();
   
       VoxelizeTriplets( std::vector<float> origin,
                         std::vector<float> dim_len,
                         float voxel_size );
       ~VoxelizeTriplets() {};
   
       void process_fullchain( larcv::IOManager& iolcv,
                               std::string adc_producer,
                               std::string chstatus_producer,
                               bool has_mc=false );
   
     protected:
   
       int _ndims; 
       std::vector<float> _origin;  
       std::vector<float> _len;     
       float _voxel_size;           
       std::vector<int>   _nvoxels; 
       void _define_voxels();
   
       std::set< std::array<int,3> >      _voxel_set;  
       std::map< std::array<int,3>, int > _voxel_list; 
       std::vector< std::vector<int> >    _voxelidx_to_tripidxlist; 
       std::vector<int>                   _trip2voxelidx; 
       
   
       larflow::prep::PrepMatchTriplets _triplet_maker; 
   
     public:
   
       int get_axis_voxel( int axis, float coord ) const;
       
       
       void make_voxeldata( const larflow::prep::PrepMatchTriplets& triplet_data );
       PyObject* make_voxeldata_dict( const larflow::prep::PrepMatchTriplets& triplet_data );
       PyObject* make_voxeldata_dict();    
   
       const std::vector<int>& get_nvoxels() const  { return _nvoxels; };
   
       const std::vector<float>& get_origin() const { return _origin; };
   
       const std::vector<float>& get_dim_len() const { return _len; };
   
       float get_voxel_size() const { return _voxel_size; };
   
       
     protected:
       
       
     private:
   
       static bool _setup_numpy;
       
     };
     
   }
   }
   
   #endif