Refactor to use PostGIS database instead of gtfs tools
[busui.git] / include / db / route-dao.inc.php
blob:a/include/db/route-dao.inc.php -> blob:b/include/db/route-dao.inc.php
<?php <?php
   
  /*
  * Copyright 2010,2011 Alexander Sadleir
   
  Licensed under the Apache License, Version 2.0 (the "License");
  you may not use this file except in compliance with the License.
  You may obtain a copy of the License at
   
  http://www.apache.org/licenses/LICENSE-2.0
   
  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.
  */
   
function getRoute($routeID) { function getRoute($routeID) {
/* global $conn;
def handle_json_GET_routerow(self, params): $query = "Select * from routes where route_id = :routeID LIMIT 1";
schedule = self.server.schedule debug($query, "database");
route = schedule.GetRoute(params.get('route', None)) $query = $conn->prepare($query);
return [transitfeed.Route._FIELD_NAMES, route.GetFieldValuesTuple()] $query->bindParam(":routeID", $routeID);
*/ $query->execute();
} if (!$query) {
  databaseError($conn->errorInfo());
  return Array();
  }
  return $query->fetch(PDO :: FETCH_ASSOC);
  }
   
  function getRouteByFullName($routeFullName) {
  global $conn;
  $query = "Select * from routes where route_short_name||route_long_name = :routeFullName LIMIT 1";
  debug($query, "database");
  $query = $conn->prepare($query);
  $query->bindParam(":routeFullName", $routeFullName);
  $query->execute();
  if (!$query) {
  databaseError($conn->errorInfo());
  return Array();
  }
  return $query->fetch(PDO :: FETCH_ASSOC);
  }
   
function getRoutes() { function getRoutes() {
/* def handle_json_GET_routes(self, params): global $conn;
"""Return a list of all routes.""" $query = "Select * from routes order by route_short_name;";
schedule = self.server.schedule debug($query, "database");
result = [] $query = $conn->prepare($query);
for r in schedule.GetRouteList(): $query->execute();
servicep = None if (!$query) {
for t in schedule.GetTripList(): databaseError($conn->errorInfo());
if t.route_id == r.route_id: return Array();
servicep = t.service_period }
break return $query->fetchAll();
result.append( (r.route_id, r.route_short_name, r.route_long_name, servicep.service_id) ) }
result.sort(key = lambda x: x[1:3])  
return result function getRoutesByNumber($routeNumber = "") {
*/ global $conn;
} if ($routeNumber != "") {
  $query = "Select distinct routes.route_id,routes.route_short_name,routes.route_long_name,service_id from routes join trips on trips.route_id =
function findRouteByNumber($routeNumber) { routes.route_id join stop_times on stop_times.trip_id = trips.trip_id
/* where route_short_name = :routeNumber OR route_short_name LIKE :routeNumber2 order by route_short_name;";
def handle_json_GET_routesearch(self, params): } else {
"""Return a list of routes with matching short name.""" $query = "SELECT DISTINCT route_short_name from routes order by route_short_name";
schedule = self.server.schedule }
routeshortname = params.get('routeshortname', None) debug($query, "database");
result = [] $query = $conn->prepare($query);
for r in schedule.GetRouteList(): if ($routeNumber != "") {
if r.route_short_name == routeshortname: $query->bindParam(":routeNumber", $routeNumber);
servicep = None $routeNumber2 = "% " . $routeNumber;
for t in schedule.GetTripList(): $query->bindParam(":routeNumber2", $routeNumber2);
if t.route_id == r.route_id: }
servicep = t.service_period $query->execute();
break if (!$query) {
result.append( (r.route_id, r.route_short_name, r.route_long_name, servicep.service_id) ) databaseError($conn->errorInfo());
result.sort(key = lambda x: x[1:3]) return Array();
return result }
*/ return $query->fetchAll();
  }
   
  function getRoutesByNumberSeries($routeNumberSeries = "") {
  global $conn;
  if (strlen($routeNumberSeries) == 1) {
  return getRoutesByNumber($routeNumberSeries);
  }
  $seriesMin = substr($routeNumberSeries, 0, -1) . "0";
  $seriesMax = substr($routeNumberSeries, 0, -1) . "9";
  $query = "Select distinct routes.route_id,routes.route_short_name,routes.route_long_name,service_id from routes join trips on trips.route_id =
  routes.route_id join stop_times on stop_times.trip_id = trips.trip_id where to_number(route_short_name, 'FM999') between :seriesMin and :seriesMax OR route_short_name LIKE :routeNumberSeries order by route_short_name;";
  debug($query, "database");
  $query = $conn->prepare($query);
  $query->bindParam(":seriesMin", $seriesMin);
  $query->bindParam(":seriesMax", $seriesMax);
  $routeNumberSeries = "% " . substr($routeNumberSeries, 0, -1) . "%";
  $query->bindParam(":routeNumberSeries", $routeNumberSeries);
  $query->execute();
  if (!$query) {
  databaseError($conn->errorInfo());
  return Array();
  }
  return $query->fetchAll();
} }
   
function getRouteNextTrip($routeID) { function getRouteNextTrip($routeID) {
/* global $conn;
def handle_json_GET_routetrips(self, params): $query = "select * from routes join trips on trips.route_id = routes.route_id
""" Get a trip for a route_id (preferablly the next one) """ join stop_times on stop_times.trip_id = trips.trip_id where
schedule = self.server.schedule arrival_time > :currentTime and routes.route_id = :routeID order by
query = params.get('route_id', None).lower() arrival_time limit 1";
result = [] debug($query, "database");
for t in schedule.GetTripList(): $query = $conn->prepare($query);
if t.route_id == query: $query->bindParam(":currentTime", current_time());
try: $query->bindParam(":routeID", $routeID);
starttime = t.GetStartTime() $query->execute();
except: if (!$query) {
print "Error for GetStartTime of trip #" + t.trip_id + sys.exc_info()[0] databaseError($conn->errorInfo());
else: return Array();
cursor = t._schedule._connection.cursor() }
cursor.execute( $r = $query->fetch(PDO :: FETCH_ASSOC);
'SELECT arrival_secs,departure_secs FROM stop_times WHERE '  
'trip_id=? ORDER BY stop_sequence DESC LIMIT 1', (t.trip_id,)) // past last trip of the day special case
(arrival_secs, departure_secs) = cursor.fetchone() if (sizeof($r) < 16) {
if arrival_secs != None: $query = "select * from routes join trips on trips.route_id = routes.route_id
endtime = arrival_secs join stop_times on stop_times.trip_id = trips.trip_id where routes.route_id = :routeID order by
elif departure_secs != None: arrival_time DESC limit 1";
endtime = departure_secs debug($query, "database");
else: $query = $conn->prepare($query);
endtime =0 $query->bindParam(":routeID", $routeID);
result.append ( (starttime, t.trip_id, endtime) ) $query->execute();
return sorted(result, key=lambda trip: trip[2]) if (!$query) {
*/ databaseError($conn->errorInfo());
  return Array();
  }
   
  $r = $query->fetch(PDO :: FETCH_ASSOC);
  }
  return $r;
  }
   
  function getRouteAtStop($routeID, $stop_id) {
  $nextTrip = getRouteNextTrip($routeID);
  if ($nextTrip['trip_id']) {
  foreach (getTripStopTimes($nextTrip['trip_id']) as $tripStop) {
  if ($tripStop['stop_id'] == $stop_id)
  return $tripStop;
  }
  }
  return Array();
  }
   
  function getRouteTrips($routeID) {
  global $conn;
  $query = "select routes.route_id,trips.trip_id,service_id,arrival_time, stop_id, stop_sequence from routes join trips on trips.route_id = routes.route_id
  join stop_times on stop_times.trip_id = trips.trip_id where routes.route_id = :routeID and stop_sequence = '1' order by
  arrival_time ";
  debug($query, "database");
  $query = $conn->prepare($query);
  $query->bindParam(":routeID", $routeID);
  $query->execute();
  if (!$query) {
  databaseError($conn->errorInfo());
  return Array();
  }
  return $query->fetchAll();
  }
   
  function getRoutesByDestination($destination = "", $service_period = "") {
  global $conn;
  if ($service_period == "")
  $service_period = service_period();
  if ($destination != "") {
  $query = "SELECT DISTINCT trips.route_id,route_short_name,route_long_name, service_id
  FROM stop_times join trips on trips.trip_id =
  stop_times.trip_id join routes on trips.route_id = routes.route_id
  WHERE route_long_name = :destination AND service_id=:service_period order by route_short_name";
  } else {
  $query = "SELECT DISTINCT route_long_name
  FROM stop_times join trips on trips.trip_id =
  stop_times.trip_id join routes on trips.route_id = routes.route_id
  WHERE service_id= :service_period order by route_long_name";
  }
  debug($query, "database");
  $query = $conn->prepare($query);
  $query->bindParam(":service_period", $service_period);
  if ($destination != "")
  $query->bindParam(":destination", $destination);
  $query->execute();
  if (!$query) {
  databaseError($conn->errorInfo());
  return Array();
  }
  return $query->fetchAll();
  }
   
  function getRoutesBySuburb($suburb, $service_period = "") {
  if ($service_period == "")
  $service_period = service_period();
  global $conn;
  $query = "SELECT DISTINCT service_id,trips.route_id,route_short_name,route_long_name
  FROM stop_times join trips on trips.trip_id = stop_times.trip_id
  join routes on trips.route_id = routes.route_id
  join stops on stops.stop_id = stop_times.stop_id
  WHERE zone_id LIKE ':suburb AND service_id=:service_period ORDER BY route_short_name";
  debug($query, "database");
  $query = $conn->prepare($query);
  $query->bindParam(":service_period", $service_period);
  $suburb = "%" . $suburb . ";%";
  $query->bindParam(":suburb", $suburb);
  $query->execute();
  if (!$query) {
  databaseError($conn->errorInfo());
  return Array();
  }
  return $query->fetchAll();
  }
   
  function getRoutesNearby($lat, $lng, $limit = "", $distance = 500) {
  if ($service_period == "")
  $service_period = service_period();
  $service_ids = service_ids($service_period);
  $sidA = $service_ids[0];
  $sidB = $service_ids[1];
  if ($limit != "")
  $limitSQL = " LIMIT :limit ";
  global $conn;
  $query = "SELECT service_id,trips.route_id,route_short_name,route_long_name,min(stops.stop_id) as stop_id,
  min(ST_Distance(position, ST_GeographyFromText('SRID=4326;POINT($lng $lat)'), FALSE)) as distance
  FROM stop_times
  join trips on trips.trip_id = stop_times.trip_id
  join routes on trips.route_id = routes.route_id
  join stops on stops.stop_id = stop_times.stop_id
  WHERE (service_id=:service_periodA OR service_id=:service_periodB)
  AND ST_DWithin(position, ST_GeographyFromText('SRID=4326;POINT($lng $lat)'), :distance, FALSE)
  group by service_id,trips.route_id,route_short_name,route_long_name
  order by distance $limitSQL";
  debug($query, "database");
  $query = $conn->prepare($query);
  $query->bindParam(":service_periodA", $sidA);
  $query->bindParam(":service_periodB", $sidB);
  $query->bindParam(":distance", $distance);
  if ($limit != "")
  $query->bindParam(":limit", $limit);
  $query->execute();
  if (!$query) {
  databaseError($conn->errorInfo());
  return Array();
  }
  return $query->fetchAll();
} }
   
?> ?>