DisambigAlg.h

Go to the documentation of this file.

//  \fileDisambigAlg.h
//  tylerdalion@gmail.com
#ifndef DisambigAlg_H
#define DisambigAlg_H

#include <map>
#include <utility> // std::pair<>
#include <vector>

#include "art/Framework/Principal/Handle.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"
#include "canvas/Persistency/Common/Ptr.h"
#include "fhiclcpp/fwd.h"

#include "larcore/Geometry/Geometry.h"
#include "larcoreobj/SimpleTypesAndConstants/RawTypes.h" // raw::ChannelID_t
#include "larcoreobj/SimpleTypesAndConstants/geo_types.h"
#include "lardataobj/RecoBase/Hit.h"
#include "larreco/RecoAlg/APAGeometryAlg.h"
#include "larsim/MCCheater/BackTrackerService.h"
namespace detinfo {
  class DetectorClocksData;
  class DetectorPropertiesData;
}

namespace apa {

  //---------------------------------------------------------------
  class DisambigAlg {
  public:
    explicit DisambigAlg(fhicl::ParameterSet const& pset);

    void RunDisambig(detinfo::DetectorClocksData const& clockData,
                     detinfo::DetectorPropertiesData const& detProp,
                     art::Handle<std::vector<recob::Hit>> GausHits);

    void TrivialDisambig(detinfo::DetectorClocksData const& clockData,
                         detinfo::DetectorPropertiesData const& detProp,
                         unsigned int apa); 
    void Crawl(unsigned int apa);           
    unsigned int FindChanTimeEndPts(detinfo::DetectorPropertiesData const& detProp,
                                    unsigned int apa); 
    void UseEndPts(detinfo::DetectorPropertiesData const& detProp,
                   unsigned int apa); 
    unsigned int CompareViews(
      detinfo::DetectorPropertiesData const& detProp,
      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 const> geom;
    // **temporarily** here to look at performance without noise hits
    art::ServiceHandle<cheat::BackTrackerService const> bt_serv;
    geo::WireReadoutGeom const* fWireReadoutGeom;

    // 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> const& 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(detinfo::DetectorPropertiesData const& detProp,
                           recob::Hit const& hitA,
                           recob::Hit const& 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