#include <RAT/PDFFactory.hh>
#include <RAT/SimpleTiming.hh>
#include <RAT/ET1D.hh>
#include <RAT/GV1D.hh>
#include <RAT/PartialET1D.hh>
#include <RAT/CherenkovSourcePDF.hh>
#include <RAT/RadialPDF.hh>
#include <RAT/DirectionPDF.hh>
#include <RAT/PositionDirectionPDF.hh>
#include <RAT/Processor.hh>

using namespace RAT;
using namespace RAT::PDFs;

#include <RAT/string_utilities.hpp>
#include <string>
using namespace std;

PDFFactory::PDFFactory()
{
  fFactory.Register( SimpleTiming::Name(), new Alloc<PDF, SimpleTiming>() );
  fFactory.Register( ET1D::Name(), new Alloc<PDF, ET1D>() );
  fFactory.Register( GV1D::Name(), new Alloc<PDF, GV1D>() );
  fFactory.Register( PartialET1D::Name(), new Alloc<PDF, PartialET1D>() );
  fFactory.Register( RadialPDF::Name(), new Alloc<PDF, RadialPDF>() );
  fFactory.Register( DirectionPDF::Name(), new Alloc<PDF, DirectionPDF>() );
  fFactory.Register( PositionDirectionPDF::Name(), new Alloc<PDF, PositionDirectionPDF>() );
  fFactory.Register( CherenkovSourcePDF::Name(), new Alloc<PDF, CherenkovSourcePDF>() );
}

void
PDFFactory::Register( std::string name, AllocBase<PDF>* allocator )
{
  fFactory.Register( name, allocator );
}


RAT::PDFs::PDF*
PDFFactory::GetPDF( const std::string& name )
{
  string pdfName, initialisation("");
  const size_t hyphenPos = name.find( "-" );
  if( hyphenPos != string::npos )
    {
      pdfName = name.substr( 0, hyphenPos );
      initialisation = name.substr( hyphenPos + 1 );
    }
  else
    pdfName = name;

  PDF* pdf = fFactory.New( pdfName );
  pdf->Initialise( initialisation );
  return pdf;
}