////////////////////////////////////////////////////////////////////////
/// \class RAT::PDFs::DirectionPDF
///
/// \brief  A simple direction PDF using the angle of PMT hits relative
///         to direction of incoming electron
///
/// \author name Ian Coulter <icoulter@hep.upenn.edu> -- contact person
///
/// REVISION HISTORY:\n
///     26/04/2012 : I Coulter - New file \n
///     2019/06/16 : Logan Lebanowski - Handle outlier positions
///
////////////////////////////////////////////////////////////////////////

#ifndef __RAT_PDF_DirectionPDF_
#define __RAT_PDF_DirectionPDF_

#include <RAT/PDF.hh>

#include <string>

namespace RAT
{

class FitterPMT;

namespace DS
{
  class FitVertex;
}

namespace PDFs
{

class DirectionPDF : public PDF
{
public:
  double GetProbability( const FitterPMT& pmt, const DS::FitVertex& vertex );

  std::string GetName() const { return Name(); }
  static std::string Name() { return std::string( "directionPDF" ); }

  /// Initialise the scintTiming PDF
  void Initialise( const std::string& param );

  void BeginOfRun( DS::Run& run );

  void EndOfRun( DS::Run& run );
protected:
  G4PhysicsOrderedFreeVector* fProbability; /// <PDF Probability
  double fConcentratorRadius; /// <Radius of PMT concentrators
  std::string fIndex; ///< Optional database index
};

} //::PDFs

} //::RAT

#endif