DisambigAlg.h

Go to the documentation of this file.

//  \fileDisambigAlg.h
//  tylerdalion@gmail.com
#ifndef DisambigAlg_H
#define DisambigAlg_H
#include <vector>
#include <map>
#include <cmath>
#include <iostream>
#include <stdint.h>

#include "fhiclcpp/ParameterSet.h"
#include "canvas/Persistency/Common/Ptr.h"
#include "canvas/Persistency/Common/PtrVector.h"
#include "art/Framework/Principal/Event.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"

#include "larcore/Geometry/Geometry.h"
#include "lardata/DetectorInfoServices/DetectorPropertiesService.h"
#include "larcoreobj/SimpleTypesAndConstants/geo_types.h"
#include "larcoreobj/SimpleTypesAndConstants/RawTypes.h" // raw::ChannelID_t
#include "lardataobj/RecoBase/Wire.h"
#include "lardataobj/RecoBase/Hit.h"
#include "lardataobj/RecoBase/Cluster.h"
#include "larreco/RecoAlg/APAGeometryAlg.h"

#include "larsim/MCCheater/BackTrackerService.h"

#include "TMatrixD.h"
#include "TVectorD.h"



namespace apa{



  //--------------------------------------------------------------- 
  class DisambigAlg {
  public:
    
    
    DisambigAlg(fhicl::ParameterSet const& pset);
    virtual ~DisambigAlg();
    
    void               reconfigure(fhicl::ParameterSet const& p);

    void               RunDisambig( art::Handle< std::vector<recob::Hit> > GausHits );

    void               TrivialDisambig    ( unsigned int apa );   
    void               Crawl              ( unsigned int apa );   
    unsigned int       FindChanTimeEndPts ( unsigned int apa );   
    void               UseEndPts          ( unsigned int apa );   
    unsigned int       CompareViews       ( unsigned int apa );   
    void               AssessDisambigSoFar( unsigned int apa );   


    std::map<unsigned int, double>         fUeffSoFar;
    std::map<unsigned int, double>         fVeffSoFar;
    std::map<unsigned int, unsigned int>   fnUSoFar;
    std::map<unsigned int, unsigned int>   fnVSoFar;
    std::map<unsigned int, unsigned int>   fnDUSoFar;
    std::map<unsigned int, unsigned int>   fnDVSoFar;

    std::vector< std::pair<art::Ptr<recob::Hit>, geo::WireID> > fDisambigHits;

    private:

    // other classes we will use
    apa::APAGeometryAlg                           fAPAGeo;
    art::ServiceHandle<geo::Geometry>             geom;
    const detinfo::DetectorProperties*           detprop;
    art::ServiceHandle<cheat::BackTrackerService> bt_serv;                     

    // Hits organization
    std::map< raw::ChannelID_t, std::vector< art::Ptr< recob::Hit > > > fChannelToHits; 
    std::map< unsigned int, std::vector< art::Ptr< recob::Hit > > >    fAPAToUVHits, fAPAToZHits; 
    std::map< unsigned int, std::vector< art::Ptr< recob::Hit > > >    fAPAToHits; 
    std::map< unsigned int, std::vector< art::Ptr< recob::Hit > > >    fAPAToEndPHits;
    std::map< unsigned int, std::vector< std::pair<art::Ptr<recob::Hit>, geo::WireID> > >  fAPAToDHits;



    // data/function to keep track of disambiguation along the way
    std::map<std::pair<double,double>, geo::WireID>                      fChanTimeToWid;
    std::map< unsigned int, std::map<std::pair<double,double>, bool> >   fHasBeenDisambiged;
    void          MakeDisambigHit( art::Ptr<recob::Hit> hit, 
                                   geo::WireID, 
                                    unsigned int apa);



    // Functions that support disambiguation methods
    unsigned int  MakeCloseHits        (int ext, geo::WireID wid, double Dmin, double Dmax);
    bool          HitsOverlapInTime    ( art::Ptr<recob::Hit> hitA, art::Ptr<recob::Hit> hitB );
    bool          HitsReasonablyMatch  ( art::Ptr<recob::Hit> hitA, art::Ptr<recob::Hit> hitB );



    // Configure the disambiguation
    bool         fCrawl;
    bool         fUseEndP;
    bool         fCompareViews;
    unsigned int fNChanJumps;       
    double       fCloseHitsRadius;  
    double       fMaxEndPDegRange;  

  }; // class DisambigAlg

} // namespace apa

#endif // ifndef DisambigAlg_H