WeatherStore.h

Go to the documentation of this file.

#ifndef _anal_WeatherStore_h_
#define _anal_WeatherStore_h_

#include <LineStringTo.h>
#include <set>
#include <iostream>
#include <fstream>
#include <iterator>
#include <exception>


namespace anal {

  template<class Data, class RawRow = Data>
  struct WeatherStore {
    template<class Iterator>
    WeatherStore(const Iterator& begin, const Iterator& end) :
      fSet(begin, end)
    {
      Check();
    }

    WeatherStore(const std::string& filename)
    {
      std::ifstream file(filename);
      if (!file.is_open()) {
        std::cerr << "file " << filename << " missing!" << std::endl;
        return;
      }
      typedef std::istream_iterator<LineStringTo<RawRow>> Iterator;
      fSet.insert(Iterator(file), Iterator());
      Check();
    }

    const Data&
    GetData(const unsigned long int gps)
    {
      fFound.fGPSSecond = gps;
      const auto it = fSet.lower_bound(fFound);
      if (it == fSet.begin()) {
        std::cerr << "Warning: GPS second " << gps << " is before the earliest available weather data!" << std::endl;
        fFound = *it;
      } else if (it == fSet.end()) {
        std::cerr << "Warning: GPS second " << gps << " is after the last available weather data!" << std::endl;
        fFound = *fSet.rbegin();
      } else {
        if (it->fGPSSecond == gps)
          fFound = *it;
        else {
          auto pit = it;
          --pit;
          fFound.Interpolate(*pit, *it, gps);
        }
      }
      return fFound;
    }

    std::size_t GetNEntries() const { return fSet.size(); }

    friend
    std::ostream&
    operator<<(std::ostream& os, const WeatherStore<Data>& s)
    {
      for (const auto& w : s.fSet)
        os << w << '\n';
      return os;
    }

  private:
    void
    Check()
    {
      if (fSet.size() < 3)
        throw std::logic_error("not enough weather data!");
    }

    Data fFound;
    std::set<Data> fSet;
  };

}


#endif