RdAirplaneArrivalDirectionSetter.cc

Go to the documentation of this file.

#include "RdAirplaneArrivalDirectionSetter.h"
#include "Ads_bDataBase.h"
#include "Airplane.h"
#include "Flight.h"

using namespace evt;
using namespace revt;
using namespace std;
using namespace utl;
using namespace fwk;
using namespace RdAirplane;
using namespace boost;
using namespace det;
namespace fs = boost::filesystem;

RdAirplaneArrivalDirectionSetter::RdAirplaneArrivalDirectionSetter() :
  _ads_bDataBase_(NULL) {}

RdAirplaneArrivalDirectionSetter::~RdAirplaneArrivalDirectionSetter() {}

VModule::ResultFlag RdAirplaneArrivalDirectionSetter::Init() {
  Detector& detector = Detector::GetInstance();
  _globalCoordinateSystem_ = detector.GetReferenceCoordinateSystem();
  string eventAirplaneAssociationFile;
  CentralConfig* centralConfig = CentralConfig::GetInstance();
  Branch directionSetterTopBranch = centralConfig->GetTopBranch("RdAirplaneArrivalDirectionSetter");
  directionSetterTopBranch.GetChild("AirplaneEventAssociationFile").GetData(eventAirplaneAssociationFile);
  INFO("Loading event ids and associated airplane ids from file " + eventAirplaneAssociationFile);
  if(!_readInEventAirplaneAssociations(eventAirplaneAssociationFile)) {
    ERROR("Could not load event-airplane-associations.");
    return eFailure;
  }

  _ads_bDataBase_ = Ads_bDataBase::getSharedDataBase();

  return eSuccess;
}


VModule::ResultFlag RdAirplaneArrivalDirectionSetter::Run(Event& event) {
  evt::Header& header = event.GetHeader();
  EventIdToEventInfoMap::iterator iter = _eventAirplaneAssociation_.find(header.GetId());
  if(iter==_eventAirplaneAssociation_.end()) {
    INFO("Event is not from an airplane");
    return eContinueLoop;
  }
 const EventInfo& eventInfo = iter->second;
  Airplane* airplane=_ads_bDataBase_->getAirplaneById(eventInfo.airplaneId);
  if(airplane==NULL) {
    INFO("No airplane with the right id found.");
    return eContinueLoop;
  }
  const TimeStamp& time = header.GetTime();
  Flight* flight = airplane->getFlight(time);
  if(flight==NULL) {
    INFO("Airplane was not over the AERA field at that time.");
    return eContinueLoop;
   }
  Point* airplanePosition = flight->getPosition(time);
  if(airplanePosition==NULL) {
    INFO("No position data for airplane at that time.");
    return eContinueLoop;
  }
  REvent& radioEvent = event.GetREvent();
  const det::Detector& detector = det::Detector::GetInstance();
  const rdet::RDetector& radioDetector = detector.GetRDetector();

  for (REvent::StationIterator stationIter = radioEvent.StationsBegin(); stationIter != radioEvent.StationsEnd(); ++stationIter) {
    Station& station = *stationIter;
    const rdet::Station& radioStation = radioDetector.GetStation(station);
    const CoordinateSystemPtr stationCS = radioStation.GetLocalCoordinateSystem();
    StationRecData& recData = station.GetRecData();
    recData.SetParameter(eSignalArrivalDirectionX, airplanePosition->GetX(stationCS));
    recData.SetParameter(eSignalArrivalDirectionY, airplanePosition->GetY(stationCS));
    recData.SetParameter(eSignalArrivalDirectionZ, airplanePosition->GetZ(stationCS));
  }

  const Point& coreCoordinates = eventInfo.coreCoordinates;
  utl::CoordinateSystemPtr coreCoordinateSystem = LocalCoordinateSystem::Create(coreCoordinates);
  double coreX = coreCoordinates.GetX(_globalCoordinateSystem_);
  double coreY = coreCoordinates.GetY(_globalCoordinateSystem_);
  double coreZ = coreCoordinates.GetZ(_globalCoordinateSystem_);
  ShowerRRecData& showerRRecData = event.GetRecShower().GetRRecShower();
  showerRRecData.SetParameter(eCoreX, coreX);
  showerRRecData.SetParameter(eCoreY, coreY);
  showerRRecData.SetParameter(eCoreZ, coreZ);
  showerRRecData.SetParameter(eShowerAxisX, airplanePosition->GetX(coreCoordinateSystem));
  showerRRecData.SetParameter(eShowerAxisY, airplanePosition->GetY(coreCoordinateSystem));
  showerRRecData.SetParameter(eShowerAxisZ, airplanePosition->GetZ(coreCoordinateSystem));

  showerRRecData.SetParameter(eRecStage, ShowerRRecData::kSphericalWaveFitVarC);



  return eSuccess;
}


VModule::ResultFlag RdAirplaneArrivalDirectionSetter::Finish() {
  return eSuccess;
}

bool RdAirplaneArrivalDirectionSetter::_readInEventAirplaneAssociations(const string& pFile) {
  fs::path pathToFile = fs::path(pFile);
  if(!fs::exists(pathToFile)) {
    return false;
  }
  ifstream file(pFile.c_str());
  try {
    if(!file.is_open())
      return false;
    string line;
    while(getline(file,line)) {
      if(line.empty() || line==" ")
        continue;
      string eventId;
      string airplaneId;
      double coreX = 0.0;
      double coreY = 0.0;
      double coreZ = 0.0;
      stringstream lineAsStream(line);
      lineAsStream >> eventId;
      lineAsStream >> airplaneId;
      lineAsStream >> coreX;
      lineAsStream >> coreY;
      lineAsStream >> coreZ;
      _eventAirplaneAssociation_.insert(make_pair(eventId,EventInfo(eventId,airplaneId,Point(coreX,coreY,coreZ,_globalCoordinateSystem_))));
      }
  } catch(...) {
    file.close();
    throw;
  }
  file.close();
  return true;
}