Add read-only vector GTFS (General Transit Feed Specification) driver

OSGeo · Jan 26, 2023 · bcd4ed2 · bcd4ed2
1 parent 986b841
commit bcd4ed2
Show file tree

Hide file tree

Showing 10 changed files with 924 additions and 0 deletions.
diff --git a/autotest/ogr/data/gtfs/gtfs_extract.zip b/autotest/ogr/data/gtfs/gtfs_extract.zip
diff --git a/autotest/ogr/ogr_gtfs.py b/autotest/ogr/ogr_gtfs.py
diff --git a/doc/source/drivers/vector/gtfs.rst b/doc/source/drivers/vector/gtfs.rst
@@ -0,0 +1,41 @@
+.. _vector.gtfs:
+
+GTFS - General Transit Feed Specification
+=========================================
+
+.. versionadded:: 3.7
+
+.. shortname:: GTFS
+
+.. built_in_by_default::
+
+This driver can read a .zip archive containing text files following the GTFS
+specification.
+
+The driver presents layers with their original structure, and:
+
+- enhances the ``stops`` layer with Point geometries
+- enhances the ``trips`` layer with LineString geometries reconstructed by
+  joining the ``trips`` layer with ``stops`` and ``stop_times`` layers
+- expose a ``shapes_geom`` layer, if the optional ``shapes`` is present by
+  constructing LineString geometries.
+
+Open syntax
+-----------
+
+The connection name can be:
+
+- a .zip filename containing GTFS .txt files
+- a directory name prefixed with ``GTFS:`` (potentially a
+  ``/vsizip/path/to/the.zip`` filename prefixed with ``GTFS:``)
+
+Driver capabilities
+-------------------
+
+.. supports_virtualio::
+
+
+Links
+-----
+
+-  `GTFS Wikipedia page <https://en.wikipedia.org/wiki/GTFS>`__
diff --git a/doc/source/drivers/vector/index.rst b/doc/source/drivers/vector/index.rst
@@ -55,6 +55,7 @@ Vector drivers
    gpsbabel
    gpx
    grass
+   gtfs
    hana
    idb
    idrisi

diff --git a/frmts/drivers.ini b/frmts/drivers.ini
@@ -255,6 +255,7 @@ MapML
 HANA
 Parquet
 Arrow
+GTFS
 
 # Put TIGER and AVCBIN at end since they need poOpenInfo->GetSiblingFiles()
 Tiger

diff --git a/ogr/ogrsf_frmts/CMakeLists.txt b/ogr/ogrsf_frmts/CMakeLists.txt
@@ -96,6 +96,8 @@ ogr_dependent_driver(cad "OpenCAD" "GDAL_USE_OPENCAD OR GDAL_USE_OPENCAD_INTERNA
 ogr_dependent_driver(parquet "Parquet" "GDAL_USE_PARQUET")
 ogr_dependent_driver(arrow "Arrow" "GDAL_USE_ARROW")
 
+ogr_dependent_driver(gtfs "GTFS" "OGR_ENABLE_DRIVER_CSV")
+
 # ######################################################################################################################
 # proprietary libraries
 ogr_dependent_driver(oci "Oracle OCI" "GDAL_USE_ORACLE")

diff --git a/ogr/ogrsf_frmts/generic/ogrregisterall.cpp b/ogr/ogrsf_frmts/generic/ogrregisterall.cpp
@@ -254,6 +254,9 @@ void OGRRegisterAllInternal()
 #ifdef ARROW_ENABLED
     RegisterOGRArrow();
 #endif
+#ifdef GTFS_ENABLED
+    RegisterOGRGTFS();
+#endif
 
     // NOTE: you need to generally insert your own driver before that line.
 

diff --git a/ogr/ogrsf_frmts/gtfs/CMakeLists.txt b/ogr/ogrsf_frmts/gtfs/CMakeLists.txt
@@ -0,0 +1,6 @@
+add_gdal_driver(
+  TARGET ogr_GTFS
+  SOURCES ogrgtfsdriver.cpp
+  PLUGIN_CAPABLE NO_DEPS)
+gdal_standard_includes(ogr_GTFS)
+