#include <RAT/Log.hh>
#include <RAT/DB.hh>
#include <RAT/DU/PMTInfo.hh>
using namespace RAT;
using namespace RAT::DU;

#include <vector>
using namespace std;

ClassImp( RAT::DU::PMTInfo )

void
PMTInfo::BeginOfRun()
{
  DBLinkPtr pmtDB = DB::Get()->GetLink( "PMTINFO" );
  fPanels = pmtDB->GetIArray("panelnumber");
  vector<int> types = pmtDB->GetIArray("pmt_type");
  // make this optional
  vector<int> petalstatus;
  try{
    petalstatus = pmtDB->GetIArray("petal_status");
  }catch( DBNotFoundError& e ){
    warn << "DU::PMTInfo::BeginOfRun() petal_status not available \n";
    for(size_t id=0; id<fPanels.size();++id){
      petalstatus.push_back(UNAVAILABLE);
    }
  }
  vector<double> pmtx = pmtDB->GetDArray("x");
  vector<double> pmty = pmtDB->GetDArray("y");
  vector<double> pmtz = pmtDB->GetDArray("z");
  vector<double> dirx = pmtDB->GetDArray("u");
  vector<double> diry = pmtDB->GetDArray("v");
  vector<double> dirz = pmtDB->GetDArray("w");
  unsigned int size = types.size();
  Log::Assert( size == fPanels.size() && size == pmtx.size(), "PMTInfo::BeginOfRun: Mis-sized PMT database arrays." );
  fTypes.resize( size );
  fPetalStatus.resize( size );
  fPositions.resize( size );
  fDirections.resize( size );
  for( size_t id = 0; id < types.size(); id++ )
    {
      fTypes[id] = static_cast<EPMTType>( types[id] );
      fPetalStatus[id] = static_cast<EPetalStatus>( petalstatus[id] );
      fPositions[id] = TVector3( pmtx[id], pmty[id], pmtz[id] );
      fDirections[id] = TVector3( dirx[id], diry[id], dirz[id] ).Unit();
    }
  debug << "PMTInfo::BeginOfRun: Loaded " << fTypes.size() << " PMTs.\n";
}