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;