////////////////////////////////////////////////////////////////////
/// \file inroot.cc
///
/// \brief Extracts some standard data from a root file.
///
/// \author P G Jones <p.g.jones@qmul.ac.uk>
///
/// REVISION HISTORY:\n
///     2014-05-29 : P G Jones - Added header information.\n
///
/// \details The standard data is used to check the inroot producer
/// works.
///
////////////////////////////////////////////////////////////////////
#include <RAT/DU/DSReader.hh>
#include <RAT/DS/Entry.hh>

#include <TH1D.h>
#include <TFile.h>

void make_plots( RAT::DU::DSReader* dsReader, TFile *out_file )
{
  TH1D* hMC = new TH1D( "hMC", "Number of mc hits per mc-event", 200, 0.0, 2000.0 );
  hMC->SetXTitle( "Number of mc hits per mc-event." );
  hMC->SetYTitle( "Count per 10 hit bin." );
  TH1D* hUnCal = new TH1D( "hUnCal", "Number of UnCal hits per (triggered) ev-event", 200, 0.0, 2000.0 );
  hUnCal->SetXTitle( "Number of mc hits per ev-event." );
  hUnCal->SetYTitle( "Count per 10 hit bin." );
  TH1D* hCal = new TH1D( "hCal", "Number of Cal hits per (triggered) ev-event", 200, 0.0, 2000.0 );
  hCal->SetXTitle( "Number of mc hits per ev-event." );
  hCal->SetYTitle( "Count per 10 hit bin." );

  for( size_t iEntry = 0; iEntry < dsReader->GetEntryCount(); iEntry++ )
    {
      const RAT::DS::Entry& rDS = dsReader->GetEntry( iEntry );
      const RAT::DS::MC& rMC = rDS.GetMC();
      hMC->Fill( rMC.GetMCPMTCount() );
      for( size_t iEV = 0; iEV < rDS.GetEVCount(); iEV++ )
        {
          const RAT::DS::EV& rEV = rDS.GetEV( iEV );
          hUnCal->Fill( rEV.GetUncalPMTs().GetCount() );
          hCal->Fill( rEV.GetCalPMTs().GetCount() );
        }
    }

  out_file->cd();
  hMC->Write();
  hUnCal->Write();
  hCal->Write();
}