ShowerDirectionTopologyDecision_tool.cc

Go to the documentation of this file.

//############################################################################
//### Name:        ShowerDirectionTopologyDecisionTool                     ###
//### Author:      Dom Barker                                              ###
//### Date:        13.05.19                                                ###
//### Description: Decision tool to pick direction based on angular        ###
//###              seperation of two candidate direction vectors           ###
//############################################################################

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

//LArSoft Includes
#include "larpandora/LArPandoraEventBuilding/LArPandoraShower/Tools/IShowerTool.h"

namespace ShowerRecoTools {

  class ShowerDirectionTopologyDecisionTool : public IShowerTool {

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

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

  private:
    int fVerbose;
    float fAngleCut;
    std::string fFirstDirectionInputLabel;
    std::string fSecondDirectionInputLabel;
    std::string fShowerDirectionOutputLabel;
  };

  ShowerDirectionTopologyDecisionTool::ShowerDirectionTopologyDecisionTool(
    const fhicl::ParameterSet& pset)
    : IShowerTool(pset.get<fhicl::ParameterSet>("BaseTools"))
    , fVerbose(pset.get<int>("Verbose"))
    , fAngleCut(pset.get<float>("AngleCut"))
    , fFirstDirectionInputLabel(pset.get<std::string>("FirstDirectionInputLabel"))
    , fSecondDirectionInputLabel(pset.get<std::string>("SecondDirectionInputLabel"))
    , fShowerDirectionOutputLabel(pset.get<std::string>("ShowerDirectionOutputLabel"))
  {}

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

    //Check the relevent products
    if (!ShowerEleHolder.CheckElement(fFirstDirectionInputLabel)) {
      if (fVerbose)
        mf::LogError("ShowerDirectionTopologyDecision")
          << "fFirstDirectionInputLabel is is not set. Stopping.";
      return 1;
    }
    if (!ShowerEleHolder.CheckElement(fSecondDirectionInputLabel)) {
      if (fVerbose)
        mf::LogError("ShowerDirectionTopologyDecision")
          << "fSecondDirectionInputLabel is is not set. Stopping.";
      return 1;
    }

    //Get the relevent products
    TVector3 FirstShowerDirection;
    TVector3 FirstShowerDirectionError;
    ShowerEleHolder.GetElementAndError(
      fFirstDirectionInputLabel, FirstShowerDirection, FirstShowerDirectionError);

    TVector3 SecondShowerDirection;
    TVector3 SecondShowerDirectionError;
    ShowerEleHolder.GetElementAndError(
      fSecondDirectionInputLabel, SecondShowerDirection, SecondShowerDirectionError);

    //Use the first tool if directions agree within the chosen angle
    if (FirstShowerDirection.Angle(SecondShowerDirection) < fAngleCut) {
      ShowerEleHolder.SetElement(
        FirstShowerDirection, FirstShowerDirectionError, fShowerDirectionOutputLabel);
    }
    else {
      ShowerEleHolder.SetElement(
        SecondShowerDirection, SecondShowerDirectionError, fShowerDirectionOutputLabel);
    }
    return 0;
  }
}

DEFINE_ART_CLASS_TOOL(ShowerRecoTools::ShowerDirectionTopologyDecisionTool)