From 6b6842c72b28c1346ef501bc656ec3016d00e1c7 Mon Sep 17 00:00:00 2001 From: Jacek Kowalski <Jacek@jacekk.info> Date: Mon, 10 Jun 2019 20:49:08 +0000 Subject: [PATCH] Add procedure to match TTSS to GTFS via bus positions --- lib/mapper.php | 100 ++++++++++++++++++++++++++++++++++++++++--------- 1 files changed, 81 insertions(+), 19 deletions(-) diff --git a/lib/mapper.php b/lib/mapper.php index a993f4c..933305f 100644 --- a/lib/mapper.php +++ b/lib/mapper.php @@ -5,8 +5,12 @@ use transit_realtime\FeedMessage; class Mapper { - private $jsonTrips = []; + private $ttssDate = NULL; + private $ttssTrips = []; + private $gtfsrtDate = NULL; + private $gtfsrtTrips = []; private $gtfsTrips = []; + private $gtfsRoutes = []; private $logger = NULL; private $specialNames = [ @@ -27,8 +31,9 @@ } public function loadTTSS($file) { - $json = json_decode(file_get_contents($file)); - foreach($json->vehicles as $vehicle) { + $ttss = json_decode(file_get_contents($file)); + $this->ttssDate = $ttss->lastUpdate; + foreach($ttss->vehicles as $vehicle) { if(isset($vehicle->isDeleted) && $vehicle->isDeleted) continue; if(!isset($vehicle->tripId) || !$vehicle->tripId) continue; if(!isset($vehicle->name) || !$vehicle->name) continue; @@ -39,26 +44,57 @@ continue; } } - $this->jsonTrips[(int)$vehicle->tripId] = [ + $name = explode(' ', $vehicle->name, 2); + $this->ttssTrips[(int)$vehicle->tripId] = [ 'id' => $vehicle->id, + 'line' => $name[0], + 'direction' => $name[1], 'latitude' => (float)$vehicle->latitude / 3600000.0, 'longitude' => (float)$vehicle->longitude / 3600000.0, ]; } - ksort($this->jsonTrips); + ksort($this->ttssTrips); + } + + public function getTTSSDate() { + return $this->ttssDate / 1000.0; } public function loadGTFS($file) { + $buffer_size = 512; + + $routes = fopen('phar://'.$file.'/routes.txt', 'r'); + $route_header = fgetcsv($routes, $buffer_size); + while(($route = fgetcsv($routes, $buffer_size)) !== FALSE) { + $this->gtfsRoutes[$route[0]] = $route[2]; + } + fclose($routes); + + $trips = fopen('phar://'.$file.'/trips.txt', 'r'); + $trip_header = fgetcsv($trips, $buffer_size); + var_dump($trip_header); + while(($trip = fgetcsv($trips, $buffer_size)) !== FALSE) { + $this->gtfsTrips[$trip[0]] = [ + 'line' => $this->gtfsRoutes[$trip[1]] ?? NULL, + 'route' => $trip[1], + 'direction' => $trip[3], + ]; + } + fclose($trips); + } + + public function loadGTFSRT($file) { $data = file_get_contents($file); $feed = new FeedMessage(); $feed->parse($data); + $this->gtfsrtDate = $feed->header->timestamp; foreach ($feed->getEntityList() as $entity) { $vehiclePosition = $entity->getVehicle(); $position = $vehiclePosition->getPosition(); $vehicle = $vehiclePosition->getVehicle(); $trip = $vehiclePosition->getTrip(); $tripId = $trip->getTripId(); - $this->gtfsTrips[self::convertTripId($tripId)] = [ + $this->gtfsrtTrips[self::convertTripId($tripId)] = [ 'id' => $entity->getId(), 'num' => $vehicle->getLicensePlate(), 'tripId' => $tripId, @@ -66,21 +102,47 @@ 'longitude' => $position->getLongitude(), ]; } - ksort($this->gtfsTrips); + ksort($this->gtfsrtTrips); + } + + public function getGTFSRTDate() { + return $this->gtfsrtDate; + } + + public function mapUsingCoords() { + $level = 0.002; + foreach($this->gtfsrtTrips as $gkey => $gtrip) { + $trip = $this->gtfsTrips[$gtrip['tripId']] ?? NULL; + + if($trip === NULL) continue; + + foreach($this->ttssTrips as $jkey => $jtrip) { + echo $gtrip['latitude'].' '.$jtrip['latitude']."\n"; + echo $gtrip['longitude'].' '.$jtrip['longitude']."\n"; + echo $jtrip['line'].' '.$trip['line']."\n"; + echo "\n"; + + if($jtrip['line'] != $trip['line']) continue; + if(abs($gtrip['latitude'] - $jtrip['latitude']) > $level) continue; + if(abs($gtrip['longitude'] - $jtrip['longitude']) < $level) continue; + + echo 'MATCH '.$gtrip['num'].' '.$gkey.' '.$jkey.' ('.($jkey-$gkey).')'."\n"; + } + } } public function findOffset() { - if(count($this->jsonTrips) == 0 || count($this->gtfsTrips) == 0) { + if(count($this->ttssTrips) == 0 || count($this->gtfsrtTrips) == 0) { return NULL; } - $jsonTripIds = array_keys($this->jsonTrips); - $gtfsTripIds = array_keys($this->gtfsTrips); + $ttssTripIds = array_keys($this->ttssTrips); + $gtfsTripIds = array_keys($this->gtfsrtTrips); $possibleOffsets = []; - for($i = 0; $i < count($this->jsonTrips); $i++) { - for($j = 0; $j < count($this->gtfsTrips); $j++) { - $possibleOffsets[$jsonTripIds[$i] - $gtfsTripIds[$j]] = TRUE; + for($i = 0; $i < count($this->ttssTrips); $i++) { + for($j = 0; $j < count($this->gtfsrtTrips); $j++) { + $possibleOffsets[$ttssTripIds[$i] - $gtfsTripIds[$j]] = TRUE; } } $possibleOffsets = array_keys($possibleOffsets); @@ -94,7 +156,7 @@ foreach($gtfsTripIds as $tripId) { $tripId += $offset; - if(isset($this->jsonTrips[$tripId])) { + if(isset($this->ttssTrips[$tripId])) { $matched++; } } @@ -114,11 +176,11 @@ return $bestOffset; } - public function getMapping($offset) { + public function mapUsingOffset($offset) { $result = []; - foreach($this->gtfsTrips as $gtfsTripId => $gtfsTrip) { - $jsonTripId = $gtfsTripId + $offset; - if(isset($this->jsonTrips[$jsonTripId])) { + foreach($this->gtfsrtTrips as $gtfsTripId => $gtfsTrip) { + $ttssTripId = $gtfsTripId + $offset; + if(isset($this->ttssTrips[$ttssTripId])) { $data = numToTypeB($gtfsTrip['id']); $num = $gtfsTrip['num']; if(!is_array($data) || !isset($data['num'])) { @@ -130,7 +192,7 @@ // Ignore due to incorrect depot markings in the data //$this->logger->warn('Got '.$num.', database has '.$data['num']); } - $result[$this->jsonTrips[$jsonTripId]['id']] = $data; + $result[$this->ttssTrips[$ttssTripId]['id']] = $data; } } return $result; -- Gitblit v1.9.1