Inheritance diagram for ShowerRecoTools::ShowerTrackHitDirection:

Public Member Functions
	ShowerTrackHitDirection (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

art::InputTag	fHitModuleLabel

art::InputTag	fPFParticleLabel

std::string	fInitialTrackHitsInputLabel

std::string	fShowerStartPositionInputLabel

std::string	fInitialTrackInputLabel

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 17 of file ShowerTrackHitDirection_tool.cc.

Constructor & Destructor Documentation

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

Definition at line 41 of file ShowerTrackHitDirection_tool.cc.

     : IShowerTool(pset.get<fhicl::ParameterSet>("BaseTools"))
     , fVerbose(pset.get<int>("Verbose"))
     , fUsePandoraVertex(pset.get<bool>("UsePandoraVertex"))
     , fHitModuleLabel(pset.get<art::InputTag>("HitModuleLabel"))
     , fPFParticleLabel(pset.get<art::InputTag>("PFParticleLabel"))
     , fInitialTrackHitsInputLabel(pset.get<std::string>("InitialTrackHitsInputLabel"))
     , fShowerStartPositionInputLabel(pset.get<std::string>("ShowerStartPositionInputLabel"))
     , fInitialTrackInputLabel(pset.get<std::string>("InitialTrackInputLabel"))
     , fShowerDirectionOutputLabel(pset.get<std::string>("ShowerDirectionOutputLabel"))
   {}

Member Function Documentation

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

overridevirtual

Implements ShowerRecoTools::IShowerTool.

Definition at line 54 of file ShowerTrackHitDirection_tool.cc.

   {
 
     //Check the Track Hits has been defined
     if (!ShowerEleHolder.CheckElement(fInitialTrackHitsInputLabel)) {
       if (fVerbose)
         mf::LogError("ShowerTrackHitDirection") << "Initial track hits not set" << std::endl;
       return 0;
     }
 
     //Check the start position is set.
     if (fUsePandoraVertex && !ShowerEleHolder.CheckElement(fShowerStartPositionInputLabel)) {
       if (fVerbose)
         mf::LogError("ShowerTrackHitDirection")
           << "Start position not set, returning " << std::endl;
       return 0;
     }
 
     //Get the start poistion
     TVector3 StartPosition = {-999, -999, -99};
     if (fUsePandoraVertex) {
       ShowerEleHolder.GetElement(fShowerStartPositionInputLabel, StartPosition);
     }
     else {
       //Check the Tracks has been defined
       if (!ShowerEleHolder.CheckElement(fInitialTrackInputLabel)) {
         if (fVerbose)
           mf::LogError("ShowerTrackHitDirection") << "Initial track not set" << std::endl;
         return 0;
       }
       recob::Track InitialTrack;
       ShowerEleHolder.GetElement(fInitialTrackInputLabel, InitialTrack);
       geo::Point_t Start_point = InitialTrack.Start();
       StartPosition = {Start_point.X(), Start_point.Y(), Start_point.Z()};
     }
 
     //Get the spacepoints associated to hits
     auto const hitHandle = Event.getValidHandle<std::vector<recob::Hit>>(fHitModuleLabel);
 
     //Get the spacepoint handle. We need to do this in 3D.
     const art::FindManyP<recob::SpacePoint>& fmsp =
       ShowerEleHolder.GetFindManyP<recob::SpacePoint>(hitHandle, Event, fPFParticleLabel);
 
     //Get the initial track hits.
     std::vector<art::Ptr<recob::Hit>> InitialTrackHits;
     ShowerEleHolder.GetElement(fInitialTrackHitsInputLabel, InitialTrackHits);
 
     //Calculate the mean direction and the the standard deviation
     float sumX = 0, sumX2 = 0;
     float sumY = 0, sumY2 = 0;
     float sumZ = 0, sumZ2 = 0;
 
     //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);
 
         //Get the direction relative to the start positon
         TVector3 pos = IShowerTool::GetLArPandoraShowerAlg().SpacePointPosition(sp) - StartPosition;
         if (pos.Mag() == 0) { continue; }
 
         sumX = pos.X();
         sumX2 += pos.X() * pos.X();
         sumY = pos.Y();
         sumY2 += pos.Y() * pos.Y();
         sumZ = pos.Z();
         sumZ2 += pos.Z() * pos.Z();
       }
     }
 
     float NumSps = intitaltrack_sp.size();
     TVector3 Mean = {sumX / NumSps, sumY / NumSps, sumZ / NumSps};
     Mean = Mean.Unit();
 
     float RMSX = 999;
     float RMSY = 999;
     float RMSZ = 999;
     if (sumX2 / NumSps - ((sumX / NumSps) * ((sumX / NumSps))) > 0) {
       RMSX = std::sqrt(sumX2 / NumSps - ((sumX / NumSps) * ((sumX / NumSps))));
     }
     if (sumY2 / NumSps - ((sumY / NumSps) * ((sumY / NumSps))) > 0) {
       RMSY = std::sqrt(sumY2 / NumSps - ((sumY / NumSps) * ((sumY / NumSps))));
     }
     if (sumZ2 / NumSps - ((sumZ / NumSps) * ((sumZ / NumSps))) > 0) {
       RMSZ = std::sqrt(sumZ2 / NumSps - ((sumZ / NumSps) * ((sumZ / NumSps))));
     }
 
     //Loop over the spacepoints and remove ones the relative direction is not within one sigma.
     TVector3 Direction_Mean = {0, 0, 0};
     int N = 0;
     for (auto const sp : intitaltrack_sp) {
       TVector3 Direction =
         IShowerTool::GetLArPandoraShowerAlg().SpacePointPosition(sp) - StartPosition;
       if ((std::abs((Direction - Mean).X()) < 1 * RMSX) &&
           (std::abs((Direction - Mean).Y()) < 1 * RMSY) &&
           (std::abs((Direction - Mean).Z()) < 1 * RMSZ)) {
         if (Direction.Mag() == 0) { continue; }
         ++N;
         Direction_Mean += Direction;
       }
     }
 
     if (N > 0) {
       //Take the mean value
       TVector3 Direction = Direction_Mean.Unit();
       ShowerEleHolder.SetElement(Direction, fShowerDirectionOutputLabel);
     }
     else {
       if (fVerbose)
         mf::LogError("ShowerTrackHitDirection")
           << "None of the points are within 1 sigma" << std::endl;
       return 1;
     }
     return 0;
   }