Program Listing for File ShowerRecoKeypoint.h

↰ Return to documentation for file (larflow/Reco/ShowerRecoKeypoint.h)

#ifndef __LARFLOW_SHOWER_RECO_KEYPOINT_H__
#define __LARFLOW_SHOWER_RECO_KEYPOINT_H__

#include <vector>
#include <set>

#include "larcv/core/DataFormat/IOManager.h"
#include "larcv/core/DataFormat/Image2D.h"
#include "DataFormat/storage_manager.h"
#include "DataFormat/larflowcluster.h"

#include "cluster_functions.h"

namespace larflow {
namespace reco {

  class ShowerRecoKeypoint : public larcv::larcv_base {

  public:

    ShowerRecoKeypoint()
      : larcv::larcv_base("ShowerRecoKeypoint"),
      _ssnet_lfhit_tree_name("showerhit"),
      _larmatch_score_threshold(0.5)
      {};
    virtual ~ShowerRecoKeypoint() {};

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

  protected:

    struct ShowerTrunk_t {
      int idx_keypoint;
      const larlite::larflow3dhit* keypoint;
      std::vector< float > pcaxis_v;
      std::vector< float > center_v;
      std::vector< float > start_v;
      float pca_eigenval_ratio;
      int npts;
      float gapdist;
      float impact_par;
      ShowerTrunk_t() {
        idx_keypoint = -1;
        keypoint = nullptr;
        pcaxis_v.resize(3,0);
        center_v.resize(3,0);
        start_v.resize(3,0);
        pca_eigenval_ratio = -1;
        npts = -1;
        gapdist = -1;
        impact_par = -1;
      };
    };

    struct ShowerCandidate_t {
      int cluster_idx;
      const cluster_t* cluster;
      std::vector< ShowerTrunk_t > trunk_candidates_v;
    };

    struct Shower_t  {
      ShowerTrunk_t trunk;
      std::set<int> cluster_idx;
      std::vector< std::vector<float> > points_v;
      std::vector< int > hitidx_v;
    };

    std::vector< ShowerCandidate_t > _shower_cand_v;
    std::vector< Shower_t >          _recod_shower_v;

    void _reconstructClusterTrunks( const std::vector<const cluster_t*>&    showercluster_v,
                                    const std::vector<const larlite::larflow3dhit*>& keypoint_v );
    void _buildShowers( const std::vector<const cluster_t*>&  showerhit_cluster_v );
    Shower_t _buildShowerCandidate( const ShowerCandidate_t& shower_cand,
                                    const std::vector<const cluster_t*>& showerhit_cluster_v );
    std::set<int> _buildoutShowerTrunkCandidate( const ShowerTrunk_t& trunk_cand,
                                                 const std::vector<const cluster_t*>& showerhit_cluster_v );

    Shower_t _fillShowerObject( const ShowerCandidate_t& shower_cand,
                                const std::set<int>& cluster_idx_set,
                                const int trunk_idx,
                                const std::vector< const cluster_t* >& showerhit_cluster_v );

    void _fillShowerObject( Shower_t& shower,
                            const std::vector< const cluster_t* >& showerhit_cluster_v );

    int _chooseBestTrunk( const ShowerCandidate_t& shower_cand,
                          const std::set<int>& cluster_idx_v,
                          const std::vector< const cluster_t* >& showerhit_cluster_v );

    int _chooseBestShowerForCluster( const cluster_t& cluster,
                                     const std::set<int>& shower_idx_v,
                                     const std::vector< const cluster_t* >& showerhit_cluster_v );



  protected:

    // PARAMETERS
    std::string _ssnet_lfhit_tree_name;
    float _larmatch_score_threshold;

  public:

    void set_ssnet_lfhit_tree_name( std::string name ) { _ssnet_lfhit_tree_name=name; };

    void set_larmatch_score_threshold( float thresh ) { _larmatch_score_threshold = thresh; };


  };


}
}

#endif