visits_between.cc

#include "timetable.h"


MsgPack do_visits_between(std::string stop_code, DateTime start, DateTime end, int count) {
  std::vector<Visit> results;
  auto &index = tt->st_indices["station.departure"];
  auto stop   = tt->stops[stop_code];

  auto lower_bound    = index.lower_bound(stop.id);
  auto initial_bound  = index.lower_bound(stop.id + start.time());

  for(auto today = start.without_time(); today <= end; today++) {
    auto bound = today.date() == start.date() ? initial_bound : lower_bound;

    for(auto it = bound; ; ++it) {
      auto &stop_time = *it->second;
      if(stop_time.stop_id != stop.id) break;
      if(!tt->is_active(stop_time, today)) continue;

      auto departure_dt = DateTime(today.date(), stop_time.departure_time);
      departure_dt.resolve();
      if(departure_dt < start)  continue;
      if(departure_dt > end)    goto finish;

      results.push_back({stop_time, departure_dt, departure_dt, tt});

      if((int) results.size() >= count) goto finish;
    }
  }

  finish:
  return make_payload(results);
}


MsgPack visits_between(std::string stop, std::string start, std::string end, int count) {
  std::cout << "Received call to `visits_between`: \n"
            << "\tstop:   " << stop   << "\n"
            << "\tstart:  " << start  << "\n"
            << "\tend:    " << end    << "\n"
            << "\tcount:  " << count  << "\n";

  auto response_start  = std::chrono::system_clock::now();
  MsgPack result = do_visits_between(stop, start, end, count);
  auto response_end    = std::chrono::system_clock::now();
  std::chrono::duration<double, std::milli> elapsed_time = response_end - response_start;

  std::cout << "Responded in " << elapsed_time.count() << "ms with:\n"
            << result << "\n\n";

  return result;
}