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)
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.

References art::errors::Configuration, fNfithits, fNfitpass, and fToler.

    : 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

virtual int ShowerRecoTools::IShowerTool::AddAssociations ( const art::Ptr< recob::PFParticle > &  pfpPtr, art::Event &  Event, reco::shower::ShowerElementHolder &  ShowerEleHolder  )

inlinevirtualinherited

Reimplemented in ShowerRecoTools::ShowerPCADirection, ShowerRecoTools::ShowerPandoraSlidingFitTrackFinder, ShowerRecoTools::ShowerExampleTool, and ShowerRecoTools::ShowerSkeletonTool.

Definition at line 74 of file IShowerTool.h.

    {
      return 0;
    }

template<class T , class A , class B >

void ShowerRecoTools::IShowerTool::AddSingle ( A &  a, B &  b, std::string  Name  )

inlineprotectedinherited

Definition at line 152 of file IShowerTool.h.

References reco::shower::ShowerProducedPtrsHolder::AddSingle().

    {
      UniquePtrs->AddSingle<T>(a, b, Name);
    }

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

overridevirtual

Implements ShowerRecoTools::IShowerTool.

Definition at line 86 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

References geo::PlaneID::asPlaneID(), reco::shower::ShowerElementHolder::CheckElement(), fHitLabel, FindInitialTrackHits(), fInitialTrackHitsOutputLabel, fInitialTrackSpacePointsOutputLabel, fPFParticleLabel, fShowerDirectionInputLabel, fShowerStartPositionInputLabel, fVerbose, reco::shower::ShowerElementHolder::GetElement(), reco::shower::ShowerElementHolder::GetFindManyP(), ShowerRecoTools::IShowerTool::GetLArPandoraShowerAlg(), art::ProductRetriever::getValidHandle(), hits(), art::Ptr< T >::key(), shower::LArPandoraShowerAlg::OrderShowerHits(), and reco::shower::ShowerElementHolder::SetElement().

  {

    //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;
    }

    geo::Point_t ShowerStartPosition{-999, -999, -999};
    ShowerEleHolder.GetElement(fShowerStartPositionInputLabel, ShowerStartPosition);

    geo::Vector_t 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 196 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

References util::abs(), geo::vect::coord(), fApplyChargeWeight, fNfithits, fNfitpass, fToler, ShowerRecoTools::IShowerTool::GetLArPandoraShowerAlg(), shower::LArPandoraShowerAlg::HitCoordinates(), hits(), and WeightedFit().

Referenced by CalculateElement().

  {

    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;
  }

const shower::LArPandoraShowerAlg& ShowerRecoTools::IShowerTool::GetLArPandoraShowerAlg ( ) const

inlineprotectedinherited

Definition at line 82 of file IShowerTool.h.

Referenced by ShowerRecoTools::ShowerLengthPercentile::CalculateElement(), ShowerRecoTools::ShowerUnidirectiondEdx::CalculateElement(), ShowerRecoTools::Shower3DCylinderTrackHitFinder::CalculateElement(), ShowerRecoTools::ShowerPCAPropergationStartPosition::CalculateElement(), CalculateElement(), ShowerRecoTools::ShowerPFPVertexStartPosition::CalculateElement(), ShowerRecoTools::ShowerDirectionCheater::CalculateElement(), ShowerRecoTools::ShowerPCADirection::CalculateElement(), ShowerRecoTools::ShowerIncrementalTrackHitFinder::CalculateElement(), ShowerRecoTools::ShowerTrajPointdEdx::CalculateElement(), ShowerRecoTools::ShowerPCADirection::CalculateShowerPCA(), FindInitialTrackHits(), ShowerRecoTools::Shower3DCylinderTrackHitFinder::FindTrackSpacePoints(), ShowerRecoTools::ShowerIncrementalTrackHitFinder::FitSegmentAndCalculateResidual(), ShowerRecoTools::ShowerIncrementalTrackHitFinder::PruneFrontOfSPSPool(), ShowerRecoTools::ShowerIncrementalTrackHitFinder::PruneTrack(), ShowerRecoTools::ShowerIncrementalTrackHitFinder::RunTestOfIncrementalSpacePointFinder(), ShowerRecoTools::ShowerTrackPCADirection::ShowerPCAVector(), and ShowerRecoTools::ShowerIncrementalTrackHitFinder::ShowerPCAVector().

    {
      return fLArPandoraShowerAlg;
    };

template<class T >

art::Ptr<T> ShowerRecoTools::IShowerTool::GetProducedElementPtr ( std::string  Name, reco::shower::ShowerElementHolder &  ShowerEleHolder, int  iter = -1  )

inlineprotectedinherited

Definition at line 102 of file IShowerTool.h.

References reco::shower::ShowerElementHolder::CheckElement(), reco::shower::ShowerProducedPtrsHolder::CheckUniqueProduerPtr(), reco::shower::ShowerProducedPtrsHolder::GetArtPtr(), and reco::shower::ShowerElementHolder::GetShowerNumber().

    {

      //Check the element has been set
      bool check_element = ShowerEleHolder.CheckElement(Name);
      if (!check_element) {
        throw cet::exception("IShowerTool") << "tried to get a element that does not exist. Failed "
                                               "at making the art ptr for Element: "
                                            << Name << std::endl;
      }

      //Check the unique ptr has been set.
      bool check_ptr = UniquePtrs->CheckUniqueProduerPtr(Name);
      if (!check_ptr) {
        throw cet::exception("IShowerTool")
          << "tried to get a ptr that does not exist. Failed at making the art ptr for Element"
          << Name;
      }

      //Check if the user has defined an index if not just use the current shower index/
      int index;
      if (iter != -1) { index = iter; }
      else {
        index = ShowerEleHolder.GetShowerNumber();
      }

      //Make the ptr
      return UniquePtrs->GetArtPtr<T>(Name, index);
    }

int ShowerRecoTools::IShowerTool::GetVectorPtrSize ( std::string  Name )

inlineprotectedinherited

Definition at line 158 of file IShowerTool.h.

References reco::shower::ShowerProducedPtrsHolder::GetVectorPtrSize().

Referenced by ShowerRecoTools::ShowerExampleTool::AddAssociations(), ShowerRecoTools::ShowerPandoraSlidingFitTrackFinder::AddAssociations(), and ShowerRecoTools::ShowerPCADirection::AddAssociations().

{ return UniquePtrs->GetVectorPtrSize(Name); }

void ShowerRecoTools::IShowerTool::InitaliseProducerPtr ( reco::shower::ShowerProducedPtrsHolder &  uniqueproducerPtrs )

inlineinherited

Definition at line 68 of file IShowerTool.h.

    {
      UniquePtrs = &uniqueproducerPtrs;
    }

virtual void ShowerRecoTools::IShowerTool::InitialiseProducers ( )

inlinevirtualinherited

Reimplemented in ShowerRecoTools::ShowerPCADirection, ShowerRecoTools::ShowerPandoraSlidingFitTrackFinder, and ShowerRecoTools::ShowerExampleTool.

Definition at line 62 of file IShowerTool.h.

{}

template<class T >

void ShowerRecoTools::IShowerTool::InitialiseProduct ( std::string  Name, std::string  InstanceName = ""  )

inlineprotectedinherited

Definition at line 137 of file IShowerTool.h.

References art::ProducesCollector::produces(), and reco::shower::ShowerProducedPtrsHolder::SetShowerUniqueProduerPtr().

    {

      if (collectorPtr == nullptr) {
        mf::LogWarning("IShowerTool") << "The art::ProducesCollector ptr has not been set";
        return;
      }

      collectorPtr->produces<T>(InstanceName);
      UniquePtrs->SetShowerUniqueProduerPtr(type<T>(), Name, InstanceName);
    }

void ShowerRecoTools::IShowerTool::PrintPtr ( std::string  Name )

inlineprotectedinherited

Definition at line 162 of file IShowerTool.h.

References reco::shower::ShowerProducedPtrsHolder::PrintPtr().

Referenced by ShowerRecoTools::ShowerExampleTool::CalculateElement().

{ UniquePtrs->PrintPtr(Name); }

void ShowerRecoTools::IShowerTool::PrintPtrs ( )

inlineprotectedinherited

Definition at line 160 of file IShowerTool.h.

References reco::shower::ShowerProducedPtrsHolder::PrintPtrs().

Referenced by ShowerRecoTools::ShowerExampleTool::CalculateElement().

{ UniquePtrs->PrintPtrs(); }

int ShowerRecoTools::IShowerTool::RunShowerTool ( const art::Ptr< recob::PFParticle > &  pfparticle, art::Event &  Event, reco::shower::ShowerElementHolder &  ShowerEleHolder, std::string  evd_display_name_append = ""  )

inlineinherited

Definition at line 46 of file IShowerTool.h.

    {

      int calculation_status = CalculateElement(pfparticle, Event, ShowerEleHolder);
      if (calculation_status != 0) return calculation_status;
      if (fRunEventDisplay) {
        IShowerTool::GetLArPandoraShowerAlg().DebugEVD(
          pfparticle, Event, ShowerEleHolder, evd_display_name_append);
      }
      return calculation_status;
    }

void ShowerRecoTools::IShowerTool::SetPtr ( art::ProducesCollector *  collector )

inlineinherited

Definition at line 65 of file IShowerTool.h.

{ collectorPtr = collector; }

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 248 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

References DEFINE_ART_CLASS_TOOL, and n.

Referenced by FindInitialTrackHits().

  {

    Double_t sumx = 0.;
    Double_t sumx2 = 0.;
    Double_t sumy = 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];
      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.

Referenced by FindInitialTrackHits().

art::InputTag ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fHitLabel

private

Definition at line 56 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Referenced by CalculateElement().

std::string ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fInitialTrackHitsOutputLabel

private

Definition at line 59 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Referenced by CalculateElement().

std::string ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fInitialTrackSpacePointsOutputLabel

private

Definition at line 60 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Referenced by CalculateElement().

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

private

Definition at line 49 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Referenced by FindInitialTrackHits(), and Shower2DLinearRegressionTrackHitFinder().

unsigned int ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fNfitpass

private

Definition at line 47 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Referenced by FindInitialTrackHits(), and Shower2DLinearRegressionTrackHitFinder().

art::InputTag ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fPFParticleLabel

private

Definition at line 54 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Referenced by CalculateElement().

std::string ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fShowerDirectionInputLabel

private

Definition at line 58 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Referenced by CalculateElement().

std::string ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fShowerStartPositionInputLabel

private

Definition at line 57 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Referenced by CalculateElement().

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

private

Definition at line 50 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Referenced by FindInitialTrackHits(), and Shower2DLinearRegressionTrackHitFinder().

int ShowerRecoTools::Shower2DLinearRegressionTrackHitFinder::fVerbose

private

Definition at line 55 of file Shower2DLinearRegressionTrackHitFinder_tool.cc.

Referenced by CalculateElement().

---

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

• larpandora/v09_22_05/source/larpandora/LArPandoraEventBuilding/LArPandoraShower/Tools/Shower2DLinearRegressionTrackHitFinder_tool.cc