ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder Class Reference

Inheritance diagram for ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder:

Public Member Functions
	Shower2DLinearRegressionTrackHitFinder (const fhicl::ParameterSet &pset)
int	CalculateElement (const art::Ptr< recob::PFParticle > &pfparticle, art::Event &Event, reco::shower::ShowerElementHolder &ShowerEleHolder) override

Private Member Functions
std::vector< art::Ptr < recob::Hit > >	FindInitialTrackHits (const detinfo::DetectorPropertiesData &detProp, std::vector< art::Ptr< recob::Hit >> &hits)
Int_t	WeightedFit (const Int_t n, const Double_t *x, const Double_t *y, const Double_t *w, Double_t *parm)
Private Member Functions inherited from ShowerRecoTools::IShowerTool
	IShowerTool (const fhicl::ParameterSet &pset)
virtual	~IShowerTool () noexcept=default
int	RunShowerTool (const art::Ptr< recob::PFParticle > &pfparticle, art::Event &Event, reco::shower::ShowerElementHolder &ShowerEleHolder, std::string evd_display_name_append="")
virtual void	InitialiseProducers ()
void	SetPtr (art::ProducesCollector *collector)
void	InitaliseProducerPtr (reco::shower::ShowerProducedPtrsHolder &uniqueproducerPtrs)
virtual int	AddAssociations (const art::Ptr< recob::PFParticle > &pfpPtr, art::Event &Event, reco::shower::ShowerElementHolder &ShowerEleHolder)
const shower::LArPandoraShowerAlg &	GetLArPandoraShowerAlg () const
template<class T >
art::Ptr< T >	GetProducedElementPtr (std::string Name, reco::shower::ShowerElementHolder &ShowerEleHolder, int iter=-1)
template<class T >
void	InitialiseProduct (std::string Name, std::string InstanceName="")
template<class T , class A , class B >
void	AddSingle (A &a, B &b, std::string Name)
int	GetVectorPtrSize (std::string Name)
void	PrintPtrs ()
void	PrintPtr (std::string Name)

Private Attributes
unsigned int	fNfitpass
std::vector< unsigned int >	fNfithits
std::vector< double >	fToler
bool	fApplyChargeWeight
art::InputTag	fPFParticleLabel
int	fVerbose
art::InputTag	fHitLabel
std::string	fShowerStartPositionInputLabel
std::string	fShowerDirectionInputLabel
std::string	fInitialTrackHitsOutputLabel
std::string	fInitialTrackSpacePointsOutputLabel

Detailed Description

Definition at line 24 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Constructor & Destructor Documentation

ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::Shower2DLinearRegressionTrackHitFinder

(

const fhicl::ParameterSet &

pset

)

Definition at line 63 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

    : IShowerTool(pset.get<fhicl::ParameterSet>("BaseTools"))
    , fNfitpass(pset.get<unsigned int>("Nfitpass"))
    , fNfithits(pset.get<std::vector<unsigned int>>("Nfithits"))
    , fToler(pset.get<std::vector<double>>("Toler"))
    , fApplyChargeWeight(pset.get<bool>("ApplyChargeWeight"))
    , fPFParticleLabel(pset.get<art::InputTag>("PFParticleLabel"))
    , fVerbose(pset.get<int>("Verbose"))
    , fHitLabel(pset.get<art::InputTag>("HitsModuleLabel"))
    , fShowerStartPositionInputLabel(pset.get<std::string>("ShowerStartPositionInputLabel"))
    , fShowerDirectionInputLabel(pset.get<std::string>("ShowerDirectionInputLabel"))
    , fInitialTrackHitsOutputLabel(pset.get<std::string>("InitialTrackHitsOutputLabel"))
    , fInitialTrackSpacePointsOutputLabel(
        pset.get<std::string>("InitialTrackSpacePointsOutputLabel"))
  {
    if (fNfitpass != fNfithits.size() || fNfitpass != fToler.size()) {
      throw art::Exception(art::errors::Configuration)
        << "Shower2DLinearRegressionTrackHitFinderEMShower: fNfithits and fToler need to have size "
           "fNfitpass";
    }
  }

Member Function Documentation

int ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::CalculateElement ( const art::Ptr< recob::PFParticle > &  pfparticle, art::Event &  Event, reco::shower::ShowerElementHolder &  ShowerEleHolder  )

overridevirtual

Implements ShowerRecoTools::IShowerTool.

Definition at line 87 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

  {

    //This is all based on the shower vertex being known. If it is not lets not do the track
    if (!ShowerEleHolder.CheckElement(fShowerStartPositionInputLabel)) {
      if (fVerbose)
        mf::LogError("Shower2DLinearRegressionTrackHitFinder")
          << "Start position not set, returning " << std::endl;
      return 1;
    }
    if (!ShowerEleHolder.CheckElement(fShowerDirectionInputLabel)) {
      if (fVerbose)
        mf::LogError("Shower2DLinearRegressionTrackHitFinder")
          << "Direction not set, returning " << std::endl;
      return 1;
    }

    TVector3 ShowerStartPosition = {-999, -999, -999};
    ShowerEleHolder.GetElement(fShowerStartPositionInputLabel, ShowerStartPosition);

    TVector3 ShowerDirection = {-999, -999, -999};
    ShowerEleHolder.GetElement(fShowerDirectionInputLabel, ShowerDirection);

    // Get the assocated pfParicle vertex PFParticles
    auto const pfpHandle = Event.getValidHandle<std::vector<recob::PFParticle>>(fPFParticleLabel);

    //Get the clusters
    auto const clusHandle = Event.getValidHandle<std::vector<recob::Cluster>>(fPFParticleLabel);

    const art::FindManyP<recob::Cluster>& fmc =
      ShowerEleHolder.GetFindManyP<recob::Cluster>(pfpHandle, Event, fPFParticleLabel);
    std::vector<art::Ptr<recob::Cluster>> clusters = fmc.at(pfparticle.key());

    if (clusters.size() < 2) {
      if (fVerbose)
        mf::LogError("Shower2DLinearRegressionTrackHitFinder")
          << "Not enough clusters: " << clusters.size() << std::endl;
      return 1;
    }

    //Get the hit association
    const art::FindManyP<recob::Hit>& fmhc =
      ShowerEleHolder.GetFindManyP<recob::Hit>(clusHandle, Event, fPFParticleLabel);
    std::map<geo::PlaneID, std::vector<art::Ptr<recob::Hit>>> plane_clusters;
    //Loop over the clusters in the plane and get the hits
    for (auto const& cluster : clusters) {

      //Get the hits
      std::vector<art::Ptr<recob::Hit>> hits = fmhc.at(cluster.key());

      for (auto hit : hits) {
        geo::WireID wire = hit->WireID();
        geo::PlaneID plane = wire.asPlaneID();
        plane_clusters[plane].push_back(hit);
      }

      // Was having issues with clusters having hits in multiple planes breaking PMA
      // So switched to the method above. May want to switch back when using PandoraTrack
      //plane_clusters[plane].insert(plane_clusters[plane].end(),hits.begin(),hits.end());
    }

    auto const clockData =
      art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(Event);
    auto const detProp =
      art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(Event, clockData);

    std::vector<art::Ptr<recob::Hit>> InitialTrackHits;
    //Loop over the clusters and order the hits and get the initial track hits in that plane
    for (auto const& cluster : plane_clusters) {

      //Get the hits
      std::vector<art::Ptr<recob::Hit>> hits = cluster.second;

      //Order the hits
      IShowerTool::GetLArPandoraShowerAlg().OrderShowerHits(
        detProp, hits, ShowerStartPosition, ShowerDirection);

      //Find the initial track hits
      std::vector<art::Ptr<recob::Hit>> trackhits = FindInitialTrackHits(detProp, hits);

      InitialTrackHits.insert(InitialTrackHits.end(), trackhits.begin(), trackhits.end());
    }

    //Holders for the initial track values.
    ShowerEleHolder.SetElement(InitialTrackHits, fInitialTrackHitsOutputLabel);

    //Get the associated spacepoints
    //Get the hits
    auto const hitHandle = Event.getValidHandle<std::vector<recob::Hit>>(fHitLabel);

    //get the sp<->hit association
    const art::FindManyP<recob::SpacePoint>& fmsp =
      ShowerEleHolder.GetFindManyP<recob::SpacePoint>(hitHandle, Event, fPFParticleLabel);

    //Get the spacepoints associated to the track hit
    std::vector<art::Ptr<recob::SpacePoint>> intitaltrack_sp;
    for (auto const& hit : InitialTrackHits) {
      std::vector<art::Ptr<recob::SpacePoint>> sps = fmsp.at(hit.key());
      for (auto const sp : sps) {
        intitaltrack_sp.push_back(sp);
      }
    }
    ShowerEleHolder.SetElement(intitaltrack_sp, fInitialTrackSpacePointsOutputLabel);
    return 0;
  }

std::vector< art::Ptr< recob::Hit > > ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::FindInitialTrackHits ( const detinfo::DetectorPropertiesData &  detProp, std::vector< art::Ptr< recob::Hit >> &  hits  )

private

Definition at line 198 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

  {

    std::vector<art::Ptr<recob::Hit>> trackHits;

    double parm[2];
    int fitok = 0;
    std::vector<double> wfit;
    std::vector<double> tfit;
    std::vector<double> cfit;

    for (size_t i = 0; i < fNfitpass; ++i) {

      // Fit a straight line through hits
      unsigned int nhits = 0;
      for (auto& hit : hits) {

        //Not sure I am a fan of doing things in wire tick space. What if id doesn't not iterate properly or the
        //two planes in each TPC are not symmetric.
        TVector2 coord = IShowerTool::GetLArPandoraShowerAlg().HitCoordinates(detProp, hit);

        if (i == 0 ||
            (std::abs((coord.Y() - (parm[0] + coord.X() * parm[1])) *
                      std::cos(std::atan(parm[1]))) < fToler[i - 1]) ||
            fitok == 1) {
          ++nhits;
          if (nhits == fNfithits[i] + 1) break;
          wfit.push_back(coord.X());
          tfit.push_back(coord.Y());

          if (fApplyChargeWeight) { cfit.push_back(hit->Integral()); }
          else {
            cfit.push_back(1.);
          };
          if (i == fNfitpass - 1) { trackHits.push_back(hit); }
        }
      }

      if (i < fNfitpass - 1 && wfit.size()) {
        fitok = WeightedFit(wfit.size(), &wfit[0], &tfit[0], &cfit[0], &parm[0]);
      }

      wfit.clear();
      tfit.clear();
      cfit.clear();
    }
    return trackHits;
  }

Int_t ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::WeightedFit ( const Int_t  n, const Double_t *  x, const Double_t *  y, const Double_t *  w, Double_t *  parm  )

private

Definition at line 251 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

  {

    Double_t sumx = 0.;
    Double_t sumx2 = 0.;
    Double_t sumy = 0.;
    Double_t sumy2 = 0.;
    Double_t sumxy = 0.;
    Double_t sumw = 0.;
    Double_t eparm[2];

    parm[0] = 0.;
    parm[1] = 0.;
    eparm[0] = 0.;
    eparm[1] = 0.;

    for (Int_t i = 0; i < n; i++) {
      sumx += x[i] * w[i];
      sumx2 += x[i] * x[i] * w[i];
      sumy += y[i] * w[i];
      sumy2 += y[i] * y[i] * w[i];
      sumxy += x[i] * y[i] * w[i];
      sumw += w[i];
    }

    if (sumx2 * sumw - sumx * sumx == 0.) return 1;
    if (sumx2 - sumx * sumx / sumw == 0.) return 1;

    parm[0] = (sumy * sumx2 - sumx * sumxy) / (sumx2 * sumw - sumx * sumx);
    parm[1] = (sumxy - sumx * sumy / sumw) / (sumx2 - sumx * sumx / sumw);

    eparm[0] = sumx2 * (sumx2 * sumw - sumx * sumx);
    eparm[1] = (sumx2 - sumx * sumx / sumw);

    if (eparm[0] < 0. || eparm[1] < 0.) return 1;

    eparm[0] = sqrt(eparm[0]) / (sumx2 * sumw - sumx * sumx);
    eparm[1] = sqrt(eparm[1]) / (sumx2 - sumx * sumx / sumw);

    return 0;
  }

Member Data Documentation

bool ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fApplyChargeWeight

private

Definition at line 53 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

art::InputTag ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fHitLabel

private

Definition at line 56 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

std::string ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fInitialTrackHitsOutputLabel

private

Definition at line 59 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

std::string ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fInitialTrackSpacePointsOutputLabel

private

Definition at line 60 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

std::vector<unsigned int> ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fNfithits

private

Definition at line 49 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

unsigned int ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fNfitpass

private

Definition at line 47 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

art::InputTag ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fPFParticleLabel

private

Definition at line 54 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

std::string ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fShowerDirectionInputLabel

private

Definition at line 58 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

std::string ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fShowerStartPositionInputLabel

private

Definition at line 57 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

std::vector<double> ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fToler

private

Definition at line 50 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

int ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fVerbose

private

Definition at line 55 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

---

The documentation for this class was generated from the following file:

• Shower2DLinearRegressionTrackHitFinder_tool.cc