RdStationRejector.h

Go to the documentation of this file.

#ifndef _RdStationRejector_RdStationRejector_h_
#define _RdStationRejector_RdStationRejector_h_

#include <fwk/VModule.h>

#include <utl/TimeStamp.h>
#include <revt/Station.h>

#include <map>


namespace RdStationRejector {

  class RdStationRejector : public fwk::VModule {

  public:
    fwk::VModule::ResultFlag Init() override;
    fwk::VModule::ResultFlag Run(evt::Event& event) override;
    fwk::VModule::ResultFlag Finish() override;

  private:
    bool fExcludeStations = true;
    bool fUseStationsInStationList = true;
    std::vector<int> fStationList;

    bool fExcludeNlStations = true;

    bool fUseBadPeriodDB = false;
    unsigned long long int fBadPeriodReasons = 0;

    typedef std::multimap<utl::TimeStamp, std::pair<utl::TimeStamp, size_t>> ParentMap;

    class TimeRangeMap : public ParentMap {
    public:

      typedef std::vector<size_t> IdCollection;

      IdCollection
      GetBadStations(const utl::TimeStamp& time) const
      {
        ParentMap::const_iterator it = upper_bound(time);  // first key larger than time
        if (it == begin())
          return IdCollection();

        IdCollection ids;
        do {
          --it;
          if (it->first <= time && time <= it->second.first)
            ids.push_back(it->second.second);
        } while (it != begin());

        return ids;
      }
    };
    TimeRangeMap fRejectMap;

    void ExcludeOrRejectStation(revt::Station& station, unsigned long long int reason);

    REGISTER_MODULE("RdStationRejector", RdStationRejector);
  };

}

#endif