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