ShowerTrackTrajToSpacePoint_tool.cc

Go to the documentation of this file.

//############################################################################
//### Name:        ShowerTrackTrajToSpacePoint                             ###
//### Author:      Dominic Barker                                          ###
//### Date:        01.10.19                                                ###
//### Description: Tool to associate the initial track trajectory points   ###
//###              to the spacepoints                                      ###
//############################################################################

//Framework Includes
#include "art/Utilities/ToolMacros.h"

//LArSoft Includes
#include "lardataobj/RecoBase/Hit.h"
#include "lardataobj/RecoBase/SpacePoint.h"
#include "lardataobj/RecoBase/Track.h"
#include "larpandora/LArPandoraEventBuilding/LArPandoraShower/Tools/IShowerTool.h"

namespace ShowerRecoTools {

  class ShowerTrackTrajToSpacePoint : public IShowerTool {

  public:
    ShowerTrackTrajToSpacePoint(const fhicl::ParameterSet& pset);

    //Match trajectory points to the spacepoints
    int CalculateElement(const art::Ptr<recob::PFParticle>& pfparticle,
                         art::Event& Event,
                         reco::shower::ShowerElementHolder& ShowerEleHolder) override;

  private:
    float fMaxDist; //Max distance that a spacepoint can be from a trajectory
    //point to be matched
    art::InputTag fPFParticleLabel;
    int fVerbose;

    std::string fInitialTrackSpacePointsOutputLabel;
    std::string fInitialTrackHitsOutputLabel;
    std::string fInitialTrackInputTag;
    std::string fShowerStartPositionInputTag;
    std::string fInitialTrackSpacePointsInputTag;
  };

  ShowerTrackTrajToSpacePoint::ShowerTrackTrajToSpacePoint(const fhicl::ParameterSet& pset)
    : IShowerTool(pset.get<fhicl::ParameterSet>("BaseTools"))
    , fMaxDist(pset.get<float>("MaxDist"))
    , fPFParticleLabel(pset.get<art::InputTag>("PFParticleLabel"))
    , fVerbose(pset.get<int>("Verbose"))
    , fInitialTrackSpacePointsOutputLabel(
        pset.get<std::string>("InitialTrackSpacePointsOutputLabel"))
    , fInitialTrackHitsOutputLabel(pset.get<std::string>("InitialTrackHitsOutputLabel"))
    , fInitialTrackInputTag(pset.get<std::string>("InitialTrackInputTag"))
    , fShowerStartPositionInputTag(pset.get<std::string>("ShowerStartPositionInputTag"))
    , fInitialTrackSpacePointsInputTag(pset.get<std::string>("InitialTrackSpacePointsInputTag"))
  {}

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

    //Check the Track has been defined
    if (!ShowerEleHolder.CheckElement(fInitialTrackInputTag)) {
      if (fVerbose)
        mf::LogError("ShowerTrackTrajToSpacePoint") << "Initial track not set" << std::endl;
      return 0;
    }

    //Check the start position is set.
    if (!ShowerEleHolder.CheckElement(fShowerStartPositionInputTag)) {
      if (fVerbose)
        mf::LogError("ShowerTrackTrajToSpacePoint")
          << "Start position not set, returning " << std::endl;
      return 0;
    }

    //Check the Track Hits has been defined
    if (!ShowerEleHolder.CheckElement(fInitialTrackSpacePointsInputTag)) {
      if (fVerbose)
        mf::LogError("ShowerTrackTrajToSpacePoint")
          << "Initial track spacepoints not set" << std::endl;
      return 0;
    }

    //Get the start poistion
    geo::Point_t ShowerStartPosition = {-999, -999, -999};
    ShowerEleHolder.GetElement(fShowerStartPositionInputTag, ShowerStartPosition);

    //Get the initial track hits.
    std::vector<art::Ptr<recob::SpacePoint>> intitaltrack_sp;
    ShowerEleHolder.GetElement(fInitialTrackSpacePointsInputTag, intitaltrack_sp);

    //Get the track
    recob::Track InitialTrack;
    ShowerEleHolder.GetElement(fInitialTrackInputTag, InitialTrack);

    std::vector<art::Ptr<recob::SpacePoint>> new_intitaltrack_sp;
    //Loop over the trajectory points
    for (unsigned int traj = 0; traj < InitialTrack.NumberTrajectoryPoints(); ++traj) {

      //ignore bogus info.
      auto flags = InitialTrack.FlagsAtPoint(traj);
      if (flags.isSet(recob::TrajectoryPointFlagTraits::NoPoint)) { continue; }

      geo::Point_t TrajPosition = InitialTrack.LocationAtPoint(traj);
      geo::Point_t TrajPositionStart = InitialTrack.LocationAtPoint(0);

      //Ignore values with 0 mag from the start position
      if ((TrajPosition - TrajPositionStart).R() == 0) { continue; }
      if ((TrajPosition - ShowerStartPosition).R() == 0) { continue; }

      float MinDist = 9999;
      unsigned int index = 999;
      for (unsigned int sp = 0; sp < intitaltrack_sp.size(); ++sp) {
        //Find the spacepoint closest to the trajectory point.
        art::Ptr<recob::SpacePoint> spacepoint = intitaltrack_sp[sp];
        auto const dist = (spacepoint->position() - TrajPosition).R();
        if (dist < MinDist && dist < fMaxDist) {
          MinDist = dist;
          index = sp;
        }
      }

      if (index == 999) { continue; }
      //Add the spacepoint to the track spacepoints.
      new_intitaltrack_sp.push_back(intitaltrack_sp[index]);

      //Delete the spacepoint so it can not be used again.
      intitaltrack_sp.erase(intitaltrack_sp.begin() + index);
    }

    // Get the spacepoints
    auto const spHandle = Event.getValidHandle<std::vector<recob::SpacePoint>>(fPFParticleLabel);

    // Get the hits associated with the space points
    const art::FindOneP<recob::Hit>& fohsp =
      ShowerEleHolder.GetFindOneP<recob::Hit>(spHandle, Event, fPFParticleLabel);

    //Save the corresponding hits
    std::vector<art::Ptr<recob::Hit>> trackHits;
    for (auto const& spacePoint : new_intitaltrack_sp) {
      //Get the hits
      const art::Ptr<recob::Hit> hit = fohsp.at(spacePoint.key());
      trackHits.push_back(hit);
    }

    //Save the spacepoints.
    ShowerEleHolder.SetElement(new_intitaltrack_sp, fInitialTrackSpacePointsOutputLabel);
    ShowerEleHolder.SetElement(trackHits, fInitialTrackHitsOutputLabel);

    return 0;
  }

}

DEFINE_ART_CLASS_TOOL(ShowerRecoTools::ShowerTrackTrajToSpacePoint)