RdStationTimeSeriesTaperer.cc

Go to the documentation of this file.

#include "RdStationTimeSeriesTaperer.h"

#include <fwk/CentralConfig.h>

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

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


using namespace revt;
using namespace fwk;
using namespace utl;
using namespace std;


namespace RdStationTimeSeriesTaperer {


  VModule::ResultFlag
  RdStationTimeSeriesTaperer::Init()
  {
    Branch topBranch =
      CentralConfig::GetInstance()->GetTopBranch("RdStationTimeSeriesTaperer");
    topBranch.GetChild("InfoLevel").GetData(fInfoLevel);

    topBranch.GetChild("RelativeWindowWidthOnEachSide").GetData(fWindowSizeOnEachSide);
    topBranch.GetChild("Renormalize").GetData(fRenormalize);

    string tmpstring = topBranch.GetChild("WindowType").Get<string>();
    if (tmpstring == "Hann") {
      fWindowType = Hann;
      fWindow = new HannWindow(fWindowSizeOnEachSide);
    }
/*  else if (tmpstring== "Hamming") {
      fWindowType = Hamming;
      fWindow = new HammingWindow(fWindowSizeOnEachSide);
    } */

    ostringstream info;
    info << "Station TimeSeries Windower is set up as " << tmpstring
         << " window with a relative width of " << fWindowSizeOnEachSide
         << " on each side.";
    INFO(info);

    return eSuccess;
  }


  VModule::ResultFlag
  RdStationTimeSeriesTaperer::Run(evt::Event& event)
  {
    // Check if there are events at all
    if(!event.HasREvent()) {
      WARNING("No radio event found!");
      return eContinueLoop;
    }

    REvent& rEvent = event.GetREvent();
    for (auto& station : rEvent.StationsRange()) {
      // Work on the time series this station
      StationTimeSeries& timeSeries = station.GetStationTimeSeries();

      // set the trace length of the window for the specific station
      fWindow->SetTraceLength(timeSeries.GetSize());

      // apply the window
      for (StationTimeSeries::SizeType i = 0; i < timeSeries.GetSize(); ++i) {
        timeSeries[i] *= fWindow->GetWeightAtBin(i);
        if (fRenormalize)
          timeSeries[i] *= fWindow->GetRenormalizationFactor();
      }
    }

    return eSuccess;
  }


  VModule::ResultFlag
  RdStationTimeSeriesTaperer::Finish()
  {
    delete fWindow;
    return eSuccess;
  }

}