Inheritance diagram for ShowerRecoTools::ShowerTrackColinearTrajPointDirection:

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

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

Private Attributes
int	fVerbose

bool	fUsePandoraVertex

bool	fAllowDynamicSliding

bool	fUsePositionInfo

bool	fUseStartPos

float	fAngleCut

std::string	fInitialTrackInputLabel

std::string	fShowerStartPositionInputLabel

std::string	fShowerDirectionOutputLabel

Additional Inherited Members
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)

Detailed Description

Definition at line 19 of file ShowerTrackColinearTrajPointDirection_tool.cc.

Constructor & Destructor Documentation

ShowerRecoTools::ShowerTrackColinearTrajPointDirection::ShowerTrackColinearTrajPointDirection ( const fhicl::ParameterSet & pset )

Definition at line 50 of file ShowerTrackColinearTrajPointDirection_tool.cc.

     : IShowerTool(pset.get<fhicl::ParameterSet>("BaseTools"))
     , fVerbose(pset.get<int>("Verbose"))
     , fUsePandoraVertex(pset.get<bool>("UsePandoraVertex"))
     , fAllowDynamicSliding(pset.get<bool>("AllowDynamicSliding"))
     , fUsePositionInfo(pset.get<bool>("UsePositionInfo"))
     , fUseStartPos(pset.get<bool>("UseStartPos"))
     , fAngleCut(pset.get<float>("AngleCut"))
     , fInitialTrackInputLabel(pset.get<std::string>("InitialTrackInputLabel"))
     , fShowerStartPositionInputLabel(pset.get<std::string>("ShowerStartPositionInputLabel"))
     , fShowerDirectionOutputLabel(pset.get<std::string>("ShowerDirectionOutputLabel"))
 
   {}

Member Function Documentation

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

overridevirtual

Implements ShowerRecoTools::IShowerTool.

Definition at line 66 of file ShowerTrackColinearTrajPointDirection_tool.cc.

   {
 
     //Check the Track has been defined
     if (!ShowerEleHolder.CheckElement(fInitialTrackInputLabel)) {
       if (fVerbose)
         mf::LogError("ShowerTrackColinearTrajPointDirection")
           << "Initial track not set" << std::endl;
       return 1;
     }
     recob::Track InitialTrack;
     ShowerEleHolder.GetElement(fInitialTrackInputLabel, InitialTrack);
 
     //Smartly choose the which trajectory point to look at by ignoring the smush of hits at the vertex.
     if (InitialTrack.NumberTrajectoryPoints() == 1) {
       if (fVerbose)
         mf::LogError("ShowerTrackColinearTrajPointDirection")
           << "Not Enough trajectory points." << std::endl;
       return 1;
     }
 
     //Trajectory point which the direction is calcualted for.
     int trajpoint = 0;
     geo::Vector_t Direction_vec;
 
     if (fUsePositionInfo) {
       //Get the start position.
       geo::Point_t StartPosition;
 
       if (fUsePandoraVertex) {
         //Check the Track has been defined
         if (!ShowerEleHolder.CheckElement(fShowerStartPositionInputLabel)) {
           if (fVerbose)
             mf::LogError("ShowerTrackColinearTrajPointDirection")
               << "Shower start position not set" << std::endl;
           return 1;
         }
         TVector3 StartPosition_vec = {-999, -999, -999};
         ShowerEleHolder.GetElement(fShowerStartPositionInputLabel, StartPosition_vec);
         StartPosition.SetCoordinates(
           StartPosition_vec.X(), StartPosition_vec.Y(), StartPosition_vec.Z());
       }
       else {
         StartPosition = InitialTrack.Start();
       }
 
       //Loop over the trajectory points and find two corresponding trajectory points where the angle between themselves (or themsleves and the start position) is less the fMinAngle.
       for (unsigned int traj = 0; traj < InitialTrack.NumberTrajectoryPoints() - 2; ++traj) {
         ++trajpoint;
 
         //ignore bogus info.
         auto trajflags = InitialTrack.FlagsAtPoint(trajpoint);
         if (trajflags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) { continue; }
 
         bool bail = false;
 
         //ignore bogus info.
         auto flags = InitialTrack.FlagsAtPoint(traj);
         if (flags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) { continue; }
 
         //find the next non bogus traj point.
         int nexttraj = traj + 1;
         auto nextflags = InitialTrack.FlagsAtPoint(nexttraj);
         while (nextflags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) {
           if (nexttraj == (int)InitialTrack.NumberTrajectoryPoints() - 2) {
             bail = true;
             break;
           }
           ++nexttraj;
           nextflags = InitialTrack.FlagsAtPoint(nexttraj);
         }
 
         //find the next next non bogus traj point.
         int nextnexttraj = nexttraj + 1;
         auto nextnextflags = InitialTrack.FlagsAtPoint(nextnexttraj);
         while (nextnextflags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) {
           if (nexttraj == (int)InitialTrack.NumberTrajectoryPoints() - 1) {
             bail = true;
             break;
           }
           ++nextnexttraj;
           nextnextflags = InitialTrack.FlagsAtPoint(nextnexttraj);
         }
 
         if (bail) {
           if (fVerbose)
             mf::LogError("ShowerTrackColinearTrajPointDirection")
               << "Trajectory point not set as rest of the traj points are bogus." << std::endl;
           break;
         }
 
         //Get the directions.
         geo::Vector_t TrajPosition_vec = InitialTrack.LocationAtPoint(traj) - StartPosition;
         geo::Vector_t NextTrajPosition_vec;
         geo::Vector_t NextNextTrajPosition_vec;
         if (fUseStartPos) {
           NextTrajPosition_vec = InitialTrack.LocationAtPoint(nexttraj) - StartPosition;
           NextNextTrajPosition_vec = InitialTrack.LocationAtPoint(nextnexttraj) - StartPosition;
         }
         else {
           NextTrajPosition_vec =
             InitialTrack.LocationAtPoint(nexttraj) - InitialTrack.LocationAtPoint(traj);
           NextNextTrajPosition_vec =
             InitialTrack.LocationAtPoint(nextnexttraj) - InitialTrack.LocationAtPoint(traj + 1);
         }
 
         //Get the directions.
         TVector3 TrajPosition = {TrajPosition_vec.X(), TrajPosition_vec.Y(), TrajPosition_vec.Z()};
         TVector3 NextTrajPosition = {
           NextTrajPosition_vec.X(), NextTrajPosition_vec.Y(), NextTrajPosition_vec.Z()};
         TVector3 NextNextTrajPosition = {
           NextNextTrajPosition_vec.X(), NextNextTrajPosition_vec.Y(), NextNextTrajPosition_vec.Z()};
 
         //Might still be bogus and we can't use the start point
         if (TrajPosition.Mag() == 0) { continue; }
         if (NextTrajPosition.Mag() == 0) { continue; }
         if (NextNextTrajPosition.Mag() == 0) { continue; }
 
         //Check to see if the angle between the directions is small enough.
         if (TrajPosition.Angle(NextTrajPosition) < fAngleCut &&
             TrajPosition.Angle(NextNextTrajPosition) < fAngleCut) {
           break;
         }
 
         //Move the start position onwords.
         if (fAllowDynamicSliding) {
           StartPosition = {InitialTrack.LocationAtPoint(traj).X(),
                            InitialTrack.LocationAtPoint(traj).Y(),
                            InitialTrack.LocationAtPoint(traj).Z()};
         }
       }
 
       geo::Point_t TrajPosition = InitialTrack.LocationAtPoint(trajpoint);
       Direction_vec = (TrajPosition - StartPosition).Unit();
     }
     else {
       //Loop over the trajectory points and find two corresponding trajectory points where the angle between themselves (or themsleves and the start position) is less the fMinAngle.
       for (unsigned int traj = 0; traj < InitialTrack.NumberTrajectoryPoints() - 2; ++traj) {
         ++trajpoint;
 
         //ignore bogus info.
         auto trajflags = InitialTrack.FlagsAtPoint(trajpoint);
         if (trajflags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) { continue; }
 
         //ignore bogus info.
         auto flags = InitialTrack.FlagsAtPoint(traj);
         if (flags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) { continue; }
 
         bool bail = false;
 
         geo::Vector_t TrajDirection_vec;
 
         //Get the next non bogus trajectory points
         if (fUseStartPos) {
           int prevtraj = 0;
           auto prevflags = InitialTrack.FlagsAtPoint(prevtraj);
           while (prevflags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) {
             if (prevtraj == (int)InitialTrack.NumberTrajectoryPoints() - 2) {
               bail = true;
               break;
             }
             ++prevtraj;
             prevflags = InitialTrack.FlagsAtPoint(prevtraj);
           }
           TrajDirection_vec = InitialTrack.DirectionAtPoint(prevtraj);
         }
         else if (fAllowDynamicSliding && traj != 0) {
           int prevtraj = traj - 1;
           auto prevflags = InitialTrack.FlagsAtPoint(prevtraj);
           while (prevflags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) {
             if (prevtraj == 0) {
               bail = true;
               break;
             }
             --prevtraj;
             prevflags = InitialTrack.FlagsAtPoint(prevtraj);
           }
           TrajDirection_vec = InitialTrack.DirectionAtPoint(prevtraj);
         }
         else {
           TrajDirection_vec = InitialTrack.DirectionAtPoint(traj);
         }
 
         //find the next non bogus traj point.
         int nexttraj = traj + 1;
         auto nextflags = InitialTrack.FlagsAtPoint(nexttraj);
         while (nextflags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) {
           if (nexttraj == (int)InitialTrack.NumberTrajectoryPoints() - 2) {
             bail = true;
             break;
           }
           ++nexttraj;
           nextflags = InitialTrack.FlagsAtPoint(nexttraj);
         }
 
         //find the next next non bogus traj point.
         int nextnexttraj = nexttraj + 1;
         auto nextnextflags = InitialTrack.FlagsAtPoint(nextnexttraj);
         while (nextnextflags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) {
           if (nexttraj == (int)InitialTrack.NumberTrajectoryPoints() - 1) {
             bail = true;
             break;
           }
           ++nextnexttraj;
           nextnextflags = InitialTrack.FlagsAtPoint(nextnexttraj);
         }
 
         if (bail) {
           if (fVerbose)
             mf::LogError("ShowerTrackColinearTrajPointDirection")
               << "Trajectory point not set as rest of the traj points are bogus." << std::endl;
           break;
         }
 
         //Get the directions.
         geo::Vector_t NextTrajDirection_vec = InitialTrack.DirectionAtPoint(nexttraj);
         geo::Vector_t NextNextTrajDirection_vec = InitialTrack.DirectionAtPoint(nextnexttraj);
 
         TVector3 TrajDirection = {
           TrajDirection_vec.X(), TrajDirection_vec.Y(), TrajDirection_vec.Z()};
         TVector3 NextTrajDirection = {
           NextTrajDirection_vec.X(), NextTrajDirection_vec.Y(), NextTrajDirection_vec.Z()};
         TVector3 NextNextTrajDirection = {NextNextTrajDirection_vec.X(),
                                           NextNextTrajDirection_vec.Y(),
                                           NextNextTrajDirection_vec.Z()};
 
         //Might still be bogus and we can't use the start point
         if (TrajDirection.Mag() == 0) { continue; }
         if (NextTrajDirection.Mag() == 0) { continue; }
         if (NextNextTrajDirection.Mag() == 0) { continue; }
 
         //See if the angle is small enough.
         if (TrajDirection.Angle(NextTrajDirection) < fAngleCut &&
             TrajDirection.Angle(NextNextTrajDirection) < fAngleCut) {
           break;
         }
       }
       Direction_vec = InitialTrack.DirectionAtPoint(trajpoint).Unit();
     }
 
     if (trajpoint == (int)InitialTrack.NumberTrajectoryPoints() - 3) {
       if (fVerbose)
         mf::LogError("ShowerSmartTrackTrajectoryPointDirectio")
           << "Trajectory point not set." << std::endl;
       return 1;
     }
 
     //Set the direction.
     TVector3 Direction = {Direction_vec.X(), Direction_vec.Y(), Direction_vec.Z()};
     TVector3 DirectionErr = {-999, -999, -999};
     ShowerEleHolder.SetElement(Direction, DirectionErr, fShowerDirectionOutputLabel);
     return 0;
   }