RdChannelTimeSeriesWindowCutter.cc

Go to the documentation of this file.

#include "RdChannelTimeSeriesWindowCutter.h"

#include <fwk/CentralConfig.h>

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

#include <evt/Event.h>

#include <revt/REvent.h>
#include <revt/Station.h>
#include <revt/Channel.h>
#include <revt/StationRecData.h>


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


namespace RdChannelTimeSeriesWindowCutter {

  VModule::ResultFlag
  RdChannelTimeSeriesWindowCutter::Init()
  {
    Branch topBranch = CentralConfig::GetInstance()->GetTopBranch("RdChannelTimeSeriesWindowCutter");
    topBranch.GetChild("InfoLevel").GetData(fInfoLevel);
    topBranch.GetChild("WindowSize").GetData(fWindowSize);
    topBranch.GetChild("WindowStart").GetData(fWindowStart);

    return eSuccess;
  }


  VModule::ResultFlag
  RdChannelTimeSeriesWindowCutter::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();

    ostringstream info;

    // Loop through stations and channels
    for (auto& station : rEvent.StationsRange()) {
      for (auto& channel : station.ChannelsRange()) {

        if (!channel.IsActive()) // skip inactive channels
          continue;

        // Work on the time trace of this channel
        ChannelTimeSeries& timeSeries = channel.GetChannelTimeSeries();
        const double binning = timeSeries.GetBinning();

        // Check if xml settings are sensible
        if (fWindowStart > timeSeries.GetSize() * binning) {
          info.str("");
          info << "Station "<< station.GetId() << " Channel " << channel.GetId()
               << " StartTime > Length of time series, not a good idea! Skip to next channel";
          WARNING(info);

          continue;
        }

        if (fWindowStart + fWindowSize > timeSeries.GetSize() * binning) {
          info.str("");
          info << "Station "<< station.GetId() << " Channel " << channel.GetId()
               << " StopTime > Length of time series. Stop time will be set at end of the time series";
          WARNING(info);

          fWindowSize = (timeSeries.GetSize() - 1) * binning - fWindowStart;
        }

        ChannelTimeSeries tempTimeSeries;
        tempTimeSeries.SetBinning(binning);

        // Clipping
        const auto iBegin = ChannelTimeSeries::SizeType(fWindowStart / binning);
        const auto iEnd = ChannelTimeSeries::SizeType((fWindowStart + fWindowSize) / binning);
        for (auto i = iBegin; i < iEnd; ++i)
          tempTimeSeries.PushBack(timeSeries[i]);
        timeSeries = tempTimeSeries;
      }

      // Redefining the start time (on station level)
      const double traceStartTime = station.GetRecData().GetParameter(eTraceStartTime);

      // keep parameter open because it might be overwritten by other modules
      station.GetRecData().SetParameter(eTraceStartTime, traceStartTime + fWindowStart, false);
    }

    return eSuccess;
  }


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

}