RdStationDebugWriter.cc

Go to the documentation of this file.

#include "RdStationDebugWriter.h"

#include <fwk/CentralConfig.h>

#include <utl/ErrorLogger.h>
#include <utl/Reader.h>
#include <utl/config.h>

#include <evt/Event.h>
#include <revt/REvent.h>
#include <revt/Header.h>
#include <revt/Station.h>


using namespace fwk;
using namespace utl;
using namespace revt;
using namespace evt;
using std::ostringstream;
using std::ofstream;
using std::endl;
using std::ios;


namespace RdStationDebugWriter {

  VModule::ResultFlag
  RdStationDebugWriter::Init()
  {
    INFO("RdStationDebugWriter::Init()");
    // Read in of the xml-configuration
    Branch topBranch =
      CentralConfig::GetInstance()->GetTopBranch("RdStationDebugWriter");
    topBranch.GetChild("AsciiSpectrumNameBase").GetData(fAsciiSpectrumNameBase);
    topBranch.GetChild("AsciiTraceNameBase").GetData(fAsciiTraceNameBase);

    return eSuccess;
  }


  VModule::ResultFlag
  RdStationDebugWriter::Run(evt::Event& event)
  {
    // Check if radio event data exists
    if (!event.HasREvent()){
      ERROR ("Event has no radio event. Use RdStationAssociator i.e.!");
      return eContinueLoop;
    }
    REvent& rEvent = event.GetREvent();

    // Get the event id and and check if it has changed since the last time
    if ( fLastEventID != rEvent.GetHeader().GetId() ) {
      ++fEventCounter;
      fStepCounter = 0;
      fLastEventID = rEvent.GetHeader().GetId();
    } else {
      ++fStepCounter;
    }

    ostringstream info;

    // Loop over all stations in the radio event
    for (const auto& station : rEvent.StationsRange()) {

      // Get the frequency spectrum
      const auto& spectrum = station.GetStationFrequencySpectrum();

      info.str("");
      info << "Spectrum reaches from " << station.GetFrequencyOfBin(0) / megahertz
           << " MHz to " << station.GetFrequencyOfBin(spectrum.GetSize() - 1) / megahertz
           << " MHz with a binning of " << spectrum.GetBinning() / megahertz << " MHz.";
      INFO(info);

      // Open ascii file for specrum data
      ostringstream filenamespec;
      filenamespec << fAsciiSpectrumNameBase << "_" << fLastEventID
                   << "_s" << station.GetId()
                   << "_Step-" << fStepCounter << ".dat";

      ofstream outfilespec;
      outfilespec.open(filenamespec.str().c_str(), ios::out);

      // Write the frequency, the amplitude, and the phase of the spectrum binwise to the file
      for (auto i = 0LU; i < spectrum.GetSize(); ++i){
        outfilespec << station.GetFrequencyOfBin(i) / megahertz
                    << " " << abs(spectrum[i][0]) / ((micro*volt)/(meter*megahertz))
                    << " " << abs(spectrum[i][1]) / ((micro*volt)/(meter*megahertz))
                    << " " << abs(spectrum[i][2]) / ((micro*volt)/(meter*megahertz))
                    << endl;
      }
      outfilespec.close();

      // Get the time series
      const auto& timeSeries = station.GetStationTimeSeries();

      info.str("");
      info << "Trace consists of " << timeSeries.GetSize() << " bins with a binning of "
           << timeSeries.GetBinning()/nanosecond << " ns.";
      INFO(info);

      // Open ascii file for trace data
      ostringstream filenametrc;
      filenametrc << fAsciiTraceNameBase << "_" << fLastEventID
                  << "_s" << station.GetId()
                  << "_Step-" << fStepCounter << ".dat";

      ofstream outfiletrc;
      outfiletrc.open(filenametrc.str().c_str(), ios::out);

      // Write the time and the amplitudes of the trace binwise to the file
      for (auto i = 0LU; i < timeSeries.GetSize(); ++i){
        outfiletrc << timeSeries.GetBinning() * static_cast<double>(i) / nanosecond
                   << " " << timeSeries[i][0] / (micro*volt/meter)
                   << " " << timeSeries[i][1] / (micro*volt/meter)
                   << " " << timeSeries[i][2] / (micro*volt/meter)
                   << endl;
      }
      outfiletrc.close();
    }

    return eSuccess;
  }


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

}