PCASeedFinderAlg.h

Go to the documentation of this file.

#ifndef PCASeedFinderAlg_h
#define PCASeedFinderAlg_h

// Framework Includes
namespace fhicl {
  class ParameterSet;
}

// LArSoft includes
#include "larreco/RecoAlg/Cluster3DAlgs/Cluster3D.h"
#include "larreco/RecoAlg/Cluster3DAlgs/PrincipalComponentsAlg.h"
#include "larreco/RecoAlg/Cluster3DAlgs/SeedFinderAlgBase.h"
namespace geo {
  class Geometry;
}

// ROOT includes
class TVector3;

//------------------------------------------------------------------------------------------------------------------------------------------

namespace lar_cluster3d {

  class PCASeedFinderAlg final : public SeedFinderAlgBase {
  public:
    PCASeedFinderAlg(fhicl::ParameterSet const& pset);

    bool findTrackSeeds(reco::HitPairListPtr& hitPairListPtr,
                        reco::PrincipalComponents& inputPCA,
                        SeedHitPairListPairVec& seedHitMap) const override;

  private:
    bool getHitsAtEnd(reco::HitPairListPtr& hit3DList, reco::PrincipalComponents& seedPca) const;

    void LineFit2DHits(const reco::HitPairListPtr& hitList,
                       double XOrigin,
                       TVector3& Pos,
                       TVector3& Dir,
                       double& ChiDOF) const;

    geo::Geometry const* m_geometry; // pointer to the Geometry service

    double m_gapDistance;      
    size_t m_numSeed2DHits;    
    double m_minAllowedCosAng; 

    PrincipalComponentsAlg m_pcaAlg; // For running Principal Components Analysis
  };

} // namespace lar_cluster3d
#endif