Merge pull request #1264 from transitland/rsp-headways

route_stop_patterns/headways

Author: irees

app/controllers/api/v1/route_stop_patterns_controller.rb

@@ -3,8 +3,40 @@ def self.model
     RouteStopPattern
   end
 
+  def headways
+    set_model
+    render :json => rsp_headways([@model]).map { |k,v| [k.join(':'), v] }.to_h
+  end
+
   private
 
+  def rsp_headways(rsps)
+    # headway_* query parameters
+    dates = (params[:headway_dates] || "").split(",")
+    between = (params[:headway_departure_between] || "").split(",")
+    departure_span = params[:headway_departure_span].presence
+    h = params[:headway_percentile].presence    
+    headway_percentile = h ? h.to_f : 0.5
+    headways = {}
+    begin
+      headways = ScheduleStopPair.headways({
+        dates: dates, 
+        key: [:route_stop_pattern_id, :origin_id, :destination_id],
+        q: {route_stop_pattern_id: rsps.map(&:id)}, 
+        departure_start: between[0], 
+        departure_end: between[1], 
+        departure_span: departure_span, 
+        headway_percentile: headway_percentile
+      })
+    rescue StandardError => e
+      return {}
+    end
+    sids = Stop.select([:id, :onestop_id]).where(id: headways.keys.map { |k| k[1..-1] }.flatten.sort.uniq ).map { |s| [s.id, s.onestop_id] }.to_h
+    rids = rsps.map { |i| [i.id, i.onestop_id] }.to_h
+    headways.map { |k,v| [[rids[k[0]],sids[k[1]],sids[k[2]]], v]}.to_h
+  end
+
+
   def index_query
     super
     if params[:traversed_by].present?

app/models/schedule_stop_pair.rb

@@ -287,7 +287,7 @@ def self.headways(dates: [], q: {}, departure_start: nil, departure_end: nil, de
       ScheduleStopPair
         .where(q)
         .where_service_on_date(date)
-        .select([:id, :route_id, :origin_id, :destination_id, :origin_arrival_time, :origin_departure_time, :destination_arrival_time, :destination_departure_time, :frequency_start_time, :frequency_end_time, :frequency_headway_seconds])
+        .select([:id, :route_id, :route_stop_pattern_id, :origin_id, :destination_id, :origin_arrival_time, :origin_departure_time, :destination_arrival_time, :destination_departure_time, :frequency_start_time, :frequency_end_time, :frequency_headway_seconds])
         .find_each do |ssp|
           ssp.expand_frequency.each do |ssp|
             t = GTFS::WideTime.parse(ssp.origin_arrival_time).to_seconds

config/routes.rb

@@ -49,7 +49,11 @@
           get 'aggregate'
         end
       end
-      resources :route_stop_patterns, only: [:index, :show]
+      resources :route_stop_patterns, only: [:index, :show] do
+        member do
+          get 'headways'
+        end
+      end
       resources :schedule_stop_pairs, only: [:index]
       resources :feeds, only: [:index, :show] do
         member do