#include "TrackShowerSeparationAlg.h"

Public Member Functions
	TrackShowerSeparationAlg (fhicl::ParameterSet const &pset)

void	reconfigure (fhicl::ParameterSet const &pset)
	Read in configurable parameters from provided parameter set. More...

std::vector< art::Ptr< recob::Hit > >	SelectShowerHits (int event, const std::vector< art::Ptr< recob::Hit > > &hits, const std::vector< art::Ptr< recob::Track > > &tracks, const std::vector< art::Ptr< recob::SpacePoint > > &spacePoints, const art::FindManyP< recob::Hit > &fmht, const art::FindManyP< recob::Track > &fmth, const art::FindManyP< recob::SpacePoint > &fmspt, const art::FindManyP< recob::Track > &fmtsp)

Private Member Functions
std::vector< int >	InitialTrackLikeSegment (std::map< int, std::unique_ptr< ReconTrack > > &reconTracks)

TVector3	Gradient (const std::vector< TVector3 > &points, const std::unique_ptr< TVector3 > &dir)

TVector3	Gradient (const art::Ptr< recob::Track > &track)

TVector3	Gradient (const std::vector< art::Ptr< recob::SpacePoint > > &spacePoints)

TVector3	ProjPoint (const TVector3 &point, const TVector3 &direction, const TVector3 &origin=TVector3(0, 0, 0))

TVector3	SpacePointPos (const art::Ptr< recob::SpacePoint > &spacePoint)
	Return 3D point of this space point. More...

double	SpacePointsRMS (const std::vector< art::Ptr< recob::SpacePoint > > &spacePoints)

Private Attributes
int	fDebug

double	fConeAngle

double	fCylinderRadius

double	fTrackVertexCut

double	fCylinderCut

double	fShowerConeCut

Detailed Description

Definition at line 169 of file TrackShowerSeparationAlg.h.

Constructor & Destructor Documentation

shower::TrackShowerSeparationAlg::TrackShowerSeparationAlg ( fhicl::ParameterSet const & pset )

Definition at line 14 of file TrackShowerSeparationAlg.cxx.

References reconfigure().

                                                                                       {
   this->reconfigure(pset);
 }

Member Function Documentation

TVector3 shower::TrackShowerSeparationAlg::Gradient	(	const std::vector< TVector3 > &	points,
		const std::unique_ptr< TVector3 > &	dir
	)

private

Definition at line 346 of file TrackShowerSeparationAlg.cxx.

Referenced by Gradient(), SelectShowerHits(), and SpacePointsRMS().

                                                                                                                        {
 
   int nhits = 0;
   double sumx=0., sumy=0., sumz=0., sumx2=0., sumy2=0., sumxy=0., sumxz=0., sumyz=0.;
   for (std::vector<TVector3>::const_iterator pointIt = points.begin(); pointIt != points.end(); ++pointIt) {
     ++nhits;
     sumx += pointIt->X();
     sumy += pointIt->Y();
     sumz += pointIt->Z();
     sumx2 += pointIt->X() * pointIt->X();
     sumy2 += pointIt->Y() * pointIt->Y();
     sumxy += pointIt->X() * pointIt->Y();
     sumxz += pointIt->X() * pointIt->Z();
     sumyz += pointIt->Y() * pointIt->Z();
   }
 
   double dydx = (nhits * sumxy - sumx * sumy) / (nhits * sumx2 - sumx * sumx);
   double yint = (sumy * sumx2 - sumx * sumxy) / (nhits * sumx2 - sumx * sumx);
   double dzdx = (nhits * sumxz - sumx * sumz) / (nhits * sumx2 - sumx * sumx);
   double zint = (sumz * sumx2 - sumx * sumxz) / (nhits * sumx2 - sumx * sumx);
   TVector2 directionXY = TVector2(1,dydx).Unit(), directionXZ = TVector2(1,dzdx).Unit();
   TVector3 direction = TVector3(1,dydx,dzdx).Unit();
   TVector3 intercept = TVector3(0,yint,zint);
 
   // Make sure the best fit direction is pointing correctly
   if (dir and TMath::Abs(direction.Angle(*dir)) > TMath::Pi() / 2.)
     direction *= -1;
 
   return direction;
 
 }

TVector3 shower::TrackShowerSeparationAlg::Gradient ( const art::Ptr< recob::Track > & track )

private

Definition at line 378 of file TrackShowerSeparationAlg.cxx.

References dir, Gradient(), recob::Track::LocationAtPoint(), recob::Track::NumberTrajectoryPoints(), and recob::Track::VertexDirection().

                                                                                  {
 
   std::vector<TVector3> points;
   std::unique_ptr<TVector3> dir;
 
   for (unsigned int traj = 0; traj < track->NumberTrajectoryPoints(); ++traj)
     points.push_back(track->LocationAtPoint(traj));
   dir = std::make_unique<TVector3>(track->VertexDirection());
 
   return Gradient(points, dir);
 
 }

TVector3 shower::TrackShowerSeparationAlg::Gradient ( const std::vector< art::Ptr< recob::SpacePoint > > & spacePoints )

private

Definition at line 391 of file TrackShowerSeparationAlg.cxx.

References dir, Gradient(), SpacePointPos(), and lar::dump::vector().

                                                                                                         {
 
   std::vector<TVector3> points;
   std::unique_ptr<TVector3> dir;
 
   for (std::vector<art::Ptr<recob::SpacePoint> >::const_iterator spacePointIt = spacePoints.begin(); spacePointIt != spacePoints.end(); ++spacePointIt)
     points.push_back(SpacePointPos(*spacePointIt));
 
   return Gradient(points, dir);
 
 }

std::vector< int > shower::TrackShowerSeparationAlg::InitialTrackLikeSegment ( std::map< int, std::unique_ptr< ReconTrack > > & reconTracks )

private

Definition at line 280 of file TrackShowerSeparationAlg.cxx.

References fConeAngle.

                                                                                                                           {
 
   // Consider the cones for each track
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt) {
 
     for (std::map<int,std::unique_ptr<ReconTrack> >::iterator otherTrackIt = reconTracks.begin(); otherTrackIt != reconTracks.end(); ++otherTrackIt) {
 
       if (trackIt->first == otherTrackIt->first)
         continue;
       if ((otherTrackIt->second->Vertex() - trackIt->second->Vertex()).Angle(trackIt->second->VertexDirection()) < fConeAngle * TMath::Pi() / 180 or
           (otherTrackIt->second->End() - trackIt->second->Vertex()).Angle(trackIt->second->VertexDirection()) < fConeAngle * TMath::Pi() / 180) {
         trackIt->second->AddForwardTrack(otherTrackIt->first);
         otherTrackIt->second->AddShowerTrack(trackIt->first);
       }
       if ((otherTrackIt->second->Vertex() - trackIt->second->Vertex()).Angle(-1*trackIt->second->VertexDirection()) < fConeAngle * TMath::Pi() / 180 or
           (otherTrackIt->second->End() - trackIt->second->Vertex()).Angle(-1*trackIt->second->VertexDirection()) < fConeAngle * TMath::Pi() / 180)
         trackIt->second->AddBackwardTrack(otherTrackIt->first);
 
     }
 
   }
 
   // Determine if any of these tracks are actually shower tracks
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt) {
 
     if (!trackIt->second->ShowerTrackCandidate())
       continue;
 
     std::cout << "Track " << trackIt->first << " is a candidate, with shower cone tracks:" << std::endl;
     const std::vector<int>& showerConeTracks = trackIt->second->ForwardConeTracks();
     for (std::vector<int>::const_iterator showerConeTrackIt = showerConeTracks.begin(); showerConeTrackIt != showerConeTracks.end(); ++showerConeTrackIt)
       std::cout << "  " << *showerConeTrackIt << std::endl;
 
     bool isBestCandidate = true;
     const std::vector<int>& showerTracks = trackIt->second->ShowerTracks();
     for (std::vector<int>::const_iterator showerTrackIt = showerTracks.begin(); showerTrackIt != showerTracks.end(); ++showerTrackIt) {
       if (!reconTracks[*showerTrackIt]->ShowerTrackCandidate())
         continue;
       if (std::find(showerConeTracks.begin(), showerConeTracks.end(), *showerTrackIt) == showerConeTracks.end())
         continue;
       if (reconTracks[*showerTrackIt]->IsShowerTrack())
         continue;
       if (trackIt->second->TrackConeSize() < reconTracks[*showerTrackIt]->TrackConeSize())
         isBestCandidate = false;
     }
 
     if (isBestCandidate)
       trackIt->second->MakeShowerTrack();
 
   }
 
   // Determine which tracks are shower cones
   std::vector<int> showerTracks;
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt) {
     if (trackIt->second->IsShowerTrack()) {
       showerTracks.push_back(trackIt->first);
       const std::vector<int>& coneTracks = trackIt->second->ForwardConeTracks();
       for (std::vector<int>::const_iterator coneTrackIt = coneTracks.begin(); coneTrackIt != coneTracks.end(); ++coneTrackIt)
         reconTracks[*coneTrackIt]->MakeShowerCone();
     }
   }
 
   return showerTracks;
 
 }

TVector3 shower::TrackShowerSeparationAlg::ProjPoint	(	const TVector3 &	point,
		const TVector3 &	direction,
		const TVector3 &	origin = `TVector3(0,0,0)`
	)

private

Projects the 3D point given along the line defined by the specified direction Coordinate system is assumed to be centred at the origin unless a difference point is specified

Definition at line 403 of file TrackShowerSeparationAlg.cxx.

References geo::origin().

Referenced by SelectShowerHits(), and SpacePointsRMS().

                                                                                                                            {
   return (point-origin).Dot(direction) * direction + origin;
 }

void shower::TrackShowerSeparationAlg::reconfigure ( fhicl::ParameterSet const & pset )

Read in configurable parameters from provided parameter set.

Definition at line 18 of file TrackShowerSeparationAlg.cxx.

References fConeAngle, fCylinderCut, fCylinderRadius, fDebug, fShowerConeCut, fTrackVertexCut, and fhicl::ParameterSet::get().

Referenced by cluster::BlurredClustering::reconfigure(), and TrackShowerSeparationAlg().

                                                                               {
   fConeAngle      = pset.get<double>("ConeAngle");
   fCylinderRadius = pset.get<double>("CylinderRadius");
   fTrackVertexCut = pset.get<double>("TrackVertexCut");
   fCylinderCut    = pset.get<double>("CylinderCut");
   fShowerConeCut  = pset.get<double>("ShowerConeCut");
 
   fDebug = pset.get<int>("Debug",0);
 }

std::vector< art::Ptr< recob::Hit > > shower::TrackShowerSeparationAlg::SelectShowerHits	(	int	event,
		const std::vector< art::Ptr< recob::Hit > > &	hits,
		const std::vector< art::Ptr< recob::Track > > &	tracks,
		const std::vector< art::Ptr< recob::SpacePoint > > &	spacePoints,
		const art::FindManyP< recob::Hit > &	fmht,
		const art::FindManyP< recob::Track > &	fmth,
		const art::FindManyP< recob::SpacePoint > &	fmspt,
		const art::FindManyP< recob::Track > &	fmtsp
	)

Definition at line 28 of file TrackShowerSeparationAlg.cxx.

References fConeAngle, fCylinderCut, fCylinderRadius, fDebug, for(), fTrackVertexCut, Gradient(), hits(), proj, ProjPoint(), SpacePointPos(), track, and lar::dump::vector().

Referenced by cluster::BlurredClustering::produce().

                                                                                                                              {
 
   // Ok, here we are again
   // Playing the game in which no one wins, but everyone loses
   // Trying to separate tracks from showers
   //    Ode to track/shower separation
   //    M Wallbank, Oct 2016
 
   // Showers are comprised of two topologically different parts:
   //  -- the 'shower track' (the initial part of the shower)
   //  -- the 'shower cone' (the part of the shower after the initial track when showering happens)
 
   //                                            -
   //                                          --
   //                 -             --       --
   //               --         ---         ---
   //             --      ----       --
   //  -----------   -----       -  --        ---
   //                    ---             ---
   //                       --               ---
   //  {=========}{==============================}
   // shower track          shower cone
 
   std::map<int,std::unique_ptr<ReconTrack> > reconTracks;
   for (std::vector<art::Ptr<recob::Track> >::const_iterator trackIt = tracks.begin(); trackIt != tracks.end(); ++trackIt) {
     std::unique_ptr<ReconTrack> track = std::make_unique<ReconTrack>(trackIt->key());
     track->SetVertex((*trackIt)->Vertex());
     track->SetEnd((*trackIt)->End());
     track->SetVertexDir((*trackIt)->VertexDirection());
     track->SetLength((*trackIt)->Length());
     track->SetDirection(Gradient(*trackIt));
     track->SetHits(fmht.at(trackIt->key()));
     track->SetSpacePoints(fmspt.at(trackIt->key()));
     const std::vector<art::Ptr<recob::SpacePoint> > spsss = fmspt.at(trackIt->key());
     // std::cout << "Track " << trackIt->key() << " has " << spsss.size() << " space points and " << (*trackIt)->NumberTrajectoryPoints() << " traj points" << std::endl;
     // if (trackIt->key() == 5)
     //   for (unsigned int i = 0; i < (*trackIt)->NumberTrajectoryPoints(); ++i)
     //  std::cout << "Traj point " << i << " has position (" << (*trackIt)->LocationAtPoint(i).X() << ", " << (*trackIt)->LocationAtPoint(i).Y() << ", " << (*trackIt)->LocationAtPoint(i).Z() << ")" << std::endl;
     reconTracks[trackIt->key()] = std::move(track);
   }
 
   // std::vector<int> showerLikeTracks, trackLikeTracks;
   // std::vector<int> showerTracks = InitialTrackLikeSegment(reconTracks);
 
   // Consider the space point cylinder situation
   double avCylinderSpacePoints = 0;
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt) {
     // Get the 3D properties of the track
     TVector3 point = trackIt->second->Vertex();
     TVector3 direction = trackIt->second->Direction();
     // if (trackIt->second->Vertex().X() > 250 and trackIt->second->Vertex().X() < 252 and
     //  trackIt->second->Vertex().Y() > -440 and trackIt->second->Vertex().Y() < -430 and
     //  trackIt->second->Vertex().Z() > 1080 and trackIt->second->Vertex().Z() < 1090)
     //   std::cout << "Track " << trackIt->first << " begins at the most upstream vertex" << std::endl;
     // if (trackIt->second->Vertex().X() > 254 and trackIt->second->Vertex().X() < 258 and
     //  trackIt->second->Vertex().Y() > -440 and trackIt->second->Vertex().Y() < -420 and
     //  trackIt->second->Vertex().Z() > 1115 and trackIt->second->Vertex().Z() < 1130)
     //   std::cout << "Track " << trackIt->first << " begins at the supposed vertex" << std::endl;
     // if (trackIt->second->End().X() > 254 and trackIt->second->End().X() < 258 and
     //  trackIt->second->End().Y() > -440 and trackIt->second->End().Y() < -420 and
     //  trackIt->second->End().Z() > 1115 and trackIt->second->End().Z() < 1130)
     //   std::cout << "Track " << trackIt->first << " ends at the supposed vertex" << std::endl;
     // std::cout << "Track " << trackIt->first << " has vertex (" << trackIt->second->Vertex().X() << ", " << trackIt->second->Vertex().Y() << ", " << trackIt->second->Vertex().Z() << ") and end (" << trackIt->second->End().X() << ", " << trackIt->second->End().Y() << ", " << trackIt->second->End().Z() << "), with vertex direction (" << trackIt->second->VertexDirection().X() << ", " << trackIt->second->VertexDirection().Y() << ", " << trackIt->second->VertexDirection().Z() << ")" << std::endl;
     // Count space points in the volume around the track
     for (std::vector<art::Ptr<recob::SpacePoint> >::const_iterator spacePointIt = spacePoints.begin(); spacePointIt != spacePoints.end(); ++spacePointIt) {
       const std::vector<art::Ptr<recob::Track> > spTracks = fmtsp.at(spacePointIt->key());
       if (find_if(spTracks.begin(), spTracks.end(), [&trackIt](const art::Ptr<recob::Track>& t){ return (int)t.key() == trackIt->first; }) != spTracks.end())
         continue;
       // Get the properties of this space point
       TVector3 pos = SpacePointPos(*spacePointIt);
       TVector3 proj = ProjPoint(pos, direction, point);
       if ((pos-proj).Mag() < fCylinderRadius)
         trackIt->second->AddCylinderSpacePoint(spacePointIt->key());
       // if ((pos-proj).Mag() < fCylinderRadius and
       //          (pos-point)*direction > 0 and
       //          (pos-point)*direction < trackIt->second->Length())
       //        std::cout << "Space point " << spacePointIt->key() << " (" << pos.X() << ", " << pos.Y() << ", " << pos.Z() << ") in cylinder around track " << trackIt->first << " (assocatied with track " << spTracks.at(0).key() << "); point is (" << point.X() << ", " << point.Y() << ", " << point.Z() << "), proj end (" << ((trackIt->second->Length()*trackIt->second->VertexDirection())+point).X() << ", " << ((trackIt->second->Length()*trackIt->second->VertexDirection())+point).Y() << ", " << ((trackIt->second->Length()*trackIt->second->VertexDirection())+point).Z() << ")" << std::endl;
     }
     avCylinderSpacePoints += trackIt->second->CylinderSpacePointRatio();
   }
   avCylinderSpacePoints /= (double)reconTracks.size();
 
   if (fDebug > 1) {
     std::cout << std::endl << "Cylinder space point ratios:" << std::endl;
     for (std::map<int,std::unique_ptr<ReconTrack> >::const_iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt)
       std::cout << "  Track " << trackIt->first << " has cylinder space point ratio " << trackIt->second->CylinderSpacePointRatio() << " (event average " << avCylinderSpacePoints << ")" << std::endl;
   }
 
   // Identify tracks
   if (fDebug > 0)
     std::cout << std::endl << "Identifying tracks:" << std::endl;
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt)
     if (trackIt->second->CylinderSpacePointRatio() / avCylinderSpacePoints < fCylinderCut) {
       if (fDebug > 0)
         std::cout << "  Making track " << trackIt->first << " a track (Type I)" << std::endl;
       trackIt->second->MakeTrack();
     }
 
   // for (std::map<int,std::unique_ptr<ReconTrack> >::const_iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt) {
   //   if (!trackIt->second->IsTrack())
   //     continue;
   //   std::cout << "Track " << trackIt->first << " (tagged as track) is this close to other tracks:" << std::endl;
   //   for (std::map<int,std::unique_ptr<ReconTrack> >::const_iterator otherTrackIt = reconTracks.begin(); otherTrackIt != reconTracks.end(); ++otherTrackIt) {
   //     if (trackIt->first == otherTrackIt->first)
   //    continue;
   //     std::cout << "  Other track " << otherTrackIt->first << " from track " << trackIt->first << " vertex: " << (otherTrackIt->second->Vertex()-trackIt->second->Vertex()).Mag() << " (vertex) and " << (otherTrackIt->second->End()-trackIt->second->Vertex()).Mag() << " (end); end: " << (otherTrackIt->second->Vertex()-trackIt->second->End()).Mag() << " (vertex) and " << (otherTrackIt->second->End()-trackIt->second->End()).Mag() << " (end)" << std::endl;
   //   }
   // }
 
   // Identify further tracks
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt) {
     if (trackIt->second->IsTrack())
       continue;
     for (std::map<int,std::unique_ptr<ReconTrack> >::const_iterator otherTrackIt = reconTracks.begin(); otherTrackIt != reconTracks.end(); ++otherTrackIt) {
       if (trackIt->first == otherTrackIt->first or !otherTrackIt->second->IsTrack())
         continue;
       if ((trackIt->second->Vertex() - otherTrackIt->second->Vertex()).Mag() < fTrackVertexCut or
           (trackIt->second->Vertex() - otherTrackIt->second->End()).Mag() < fTrackVertexCut or
           (trackIt->second->End() - otherTrackIt->second->Vertex()).Mag() < fTrackVertexCut or
           (trackIt->second->End() - otherTrackIt->second->End()).Mag() < fTrackVertexCut) {
         if (fDebug > 0)
           std::cout << "  Making track " << trackIt->first << " a track (Type II)" << std::endl;
         trackIt->second->MakeTrack();
       }
     }
   }
 
   // Consider removing false tracks by looking at their closest approach to any other track
 
   // Consider the space point cone situation
   std::vector<art::Ptr<recob::SpacePoint> > showerSpacePoints;
   for (std::vector<art::Ptr<recob::SpacePoint> >::const_iterator spacePointIt = spacePoints.begin(); spacePointIt != spacePoints.end(); ++spacePointIt) {
     bool showerSpacePoint = true;
     const std::vector<art::Ptr<recob::Track> > spacePointTracks = fmtsp.at(spacePointIt->key());
     for (std::vector<art::Ptr<recob::Track> >::const_iterator trackIt = spacePointTracks.begin(); trackIt != spacePointTracks.end(); ++trackIt)
       if (reconTracks[trackIt->key()]->IsTrack())
         showerSpacePoint = false;
     if (showerSpacePoint)
       showerSpacePoints.push_back(*spacePointIt);
   }
 
   // Identify tracks which slipped through and shower tracks
   // For the moment, until the track tagging gets better at least, don't try to identify tracks from this
   double avConeSize = 0;
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt) {
     for (std::vector<art::Ptr<recob::SpacePoint> >::const_iterator spacePointIt = showerSpacePoints.begin(); spacePointIt != showerSpacePoints.end(); ++spacePointIt) {
       bool associatedSpacePoint = false;
       const std::vector<art::Ptr<recob::Track> > spTracks = fmtsp.at(spacePointIt->key());
       for (std::vector<art::Ptr<recob::Track> >::const_iterator spTrackIt = spTracks.begin(); spTrackIt != spTracks.end(); ++spTrackIt)
         if ((int)spTrackIt->key() == trackIt->first)
           associatedSpacePoint = true;
       if (associatedSpacePoint)
         continue;
       if ((SpacePointPos(*spacePointIt) - trackIt->second->Vertex()).Angle(trackIt->second->Direction()) < fConeAngle * TMath::Pi() / 180) {
         trackIt->second->AddForwardSpacePoint(spacePointIt->key());
         trackIt->second->AddForwardTrack(spTracks.at(0).key());
       }
       if ((SpacePointPos(*spacePointIt) - trackIt->second->Vertex()).Angle(-1*trackIt->second->Direction()) < fConeAngle * TMath::Pi() / 180) {
         trackIt->second->AddBackwardSpacePoint(spacePointIt->key());
         trackIt->second->AddBackwardTrack(spTracks.at(0).key());
       }
     }
     avConeSize += trackIt->second->ConeSize();
   }
   avConeSize /= (double)reconTracks.size();
   if (fDebug > 0)
     std::cout << std::endl << "Identifying showers:" << std::endl;
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt) {
     // if (trackIt->second->ForwardSpacePoints() == 0)
     //   trackIt->second->MakeTrack();
     // double distanceFromAverage = (trackIt->second->ConeSize() - avConeSize) / TMath::Abs(avConeSize);
     // if (fDebug > 1)
     //   std::cout << "    Track " << trackIt->first << " has cone size " << trackIt->second->ConeSize() << " (forward " << trackIt->second->ForwardSpacePoints() << ") and tracks " << trackIt->second->TrackConeSize() << " (average " << avConeSize << ", distance from average " << distanceFromAverage << ")" << std::endl;
     // if (distanceFromAverage > fShowerConeCut) {
     //   trackIt->second->MakeShower();
     //   if (fDebug > 0)
     //  std::cout << "  Making track " << trackIt->first << " a shower (Type I)" << std::endl;
     // }
     if (fDebug > 1)
       std::cout << "    Track " << trackIt->first << " has space point cone size " << trackIt->second->ConeSize() << " and track cone size " << trackIt->second->TrackConeSize() << std::endl;
     if (TMath::Abs(trackIt->second->ConeSize()) > 30 and TMath::Abs(trackIt->second->TrackConeSize()) > 3) {
       trackIt->second->MakeShower();
       if (fDebug > 0)
         std::cout << "  Making track " << trackIt->first << " a shower (Type I)" << std::endl;
       if (trackIt->second->ConeSize() < 0)
         trackIt->second->FlipTrack();
     }
   }
 
   // Look for shower cones
   std::cout << std::endl << "  Shower cones:" << std::endl;
   for (std::map<int,std::unique_ptr<ReconTrack> >::const_iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt) {
     if (trackIt->second->IsShower()) {
       if (fDebug > 1)
         std::cout << "    Track " << trackIt->first << std::endl;
       for (std::map<int,std::unique_ptr<ReconTrack> >::iterator otherTrackIt = reconTracks.begin(); otherTrackIt != reconTracks.end(); ++otherTrackIt) {
         if (trackIt->first == otherTrackIt->first or !otherTrackIt->second->IsUndetermined())
           continue;
         if ((otherTrackIt->second->Vertex()-trackIt->second->Vertex()).Angle(trackIt->second->Direction()) < fConeAngle * TMath::Pi() / 180 or
             (otherTrackIt->second->End()-trackIt->second->Vertex()).Angle(trackIt->second->Direction()) < fConeAngle * TMath::Pi() / 180) {
           std::cout << "      " << otherTrackIt->first << std::endl;
           otherTrackIt->second->MakeShower();
           if (fDebug > 0)
             std::cout << "      Making track " << otherTrackIt->first << " a shower (Type II)" << std::endl;
         }
       }
     }
   }
 
   // Look at remaining undetermined tracks
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt) {
     if (trackIt->second->IsUndetermined()) {
       // std::cout << "Remaining undetermined track " << trackIt->first << std::endl;
       // std::cout << "Length is " << trackIt->second->Length() << " and rms of space points is " << SpacePointsRMS(trackIt->second->SpacePoints()) << std::endl;
       trackIt->second->MakeShower();
     }
   }
 
   std::cout << std::endl << "Event " << event << " track shower separation:" << std::endl;
   std::cout << "Shower initial tracks are:" << std::endl;
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt)
     if (trackIt->second->IsShowerTrack())
       std::cout << "  " << trackIt->first << std::endl;
 
   std::cout << "Track tracks are:" << std::endl;
   for (std::map<int,std::unique_ptr<ReconTrack> >::iterator trackIt = reconTracks.begin(); trackIt != reconTracks.end(); ++trackIt)
     if (trackIt->second->IsTrack())
       std::cout << "  " << trackIt->first << std::endl;
 
   // Shower hits -- select all hits which aren't associated with a determined track
   std::vector<art::Ptr<recob::Hit> > showerHits;
   for (std::vector<art::Ptr<recob::Hit> >::const_iterator hitIt = hits.begin(); hitIt != hits.end(); ++hitIt) {
     bool showerHit = true;
     const std::vector<art::Ptr<recob::Track> > hitTracks = fmth.at(hitIt->key());
     for (std::vector<art::Ptr<recob::Track> >::const_iterator hitTrackIt = hitTracks.begin(); hitTrackIt != hitTracks.end(); ++hitTrackIt)
       if (reconTracks[hitTrackIt->key()]->IsTrack())
         showerHit = false;
     if (showerHit)
       showerHits.push_back(*hitIt);
   }
 
   return showerHits;
 
 }

TVector3 shower::TrackShowerSeparationAlg::SpacePointPos ( const art::Ptr< recob::SpacePoint > & spacePoint )

private

Return 3D point of this space point.

Definition at line 407 of file TrackShowerSeparationAlg.cxx.

References recob::SpacePoint::XYZ().

Referenced by Gradient(), SelectShowerHits(), and SpacePointsRMS().

                                                                                                 {
   const double* xyz = spacePoint->XYZ();
   return TVector3(xyz[0], xyz[1], xyz[2]);
 }

double shower::TrackShowerSeparationAlg::SpacePointsRMS ( const std::vector< art::Ptr< recob::SpacePoint > > & spacePoints )

private

Definition at line 412 of file TrackShowerSeparationAlg.cxx.

References Gradient(), proj, ProjPoint(), SpacePointPos(), and lar::dump::vector().

                                                                                                             {
 
   TVector3 point = SpacePointPos(spacePoints.at(0));
   TVector3 direction = Gradient(spacePoints);
 
   std::vector<double> distances;
   for (std::vector<art::Ptr<recob::SpacePoint> >::const_iterator spacePointIt = spacePoints.begin(); spacePointIt != spacePoints.end(); ++spacePointIt) {
     TVector3 pos = SpacePointPos(*spacePointIt);
     TVector3 proj = ProjPoint(pos, direction, point);
     distances.push_back((pos-proj).Mag());
   }
 
   double rms = TMath::RMS(distances.begin(), distances.end());
 
   return rms;
 
 }