struggling with gtfs

Author: chw-03

Diff for: src/main.rs

@@ -30,14 +30,14 @@ struct Args {
     /// Number of times to greet
     #[arg(long, default_value_t = true)]
     makequerygraph: bool,
+    #[arg(long, default_value_t = true)]
     debugmode: bool,
 }
 
 #[tokio::main]
 async fn main() {
     let mut args = Args::parse();
-
-    args.debugmode = true;
+    let preset_distance = 250.0;
 
     if !args.debugmode {
         args.makequerygraph = false;
@@ -64,8 +64,6 @@ async fn main() {
         //see following link, anything but first option (which includes walking between stations, hasnt been implemented yet)
         //https://www.google.com/maps/dir/Bloomfield,+Connecticut+06002/77+Forest+St,+Hartford,+CT+06105/@41.823207,-72.7745391,34082m/data=!3m1!1e3!4m20!4m19!1m5!1m1!1s0x89e7001af40714d7:0xc4be608b22d7e4a8!2m2!1d-72.7197095!2d41.8683576!1m5!1m1!1s0x89e653502e880197:0xc1f0096f7d179457!2m2!1d-72.7005256!2d41.7671825!2m4!4e3!6e0!7e2!8j1727241000!3e3!5i1
 
-        let preset_distance = 250.0;
-
         if (args.makequerygraph) {
             //pepperidge farm to harriet beecher stowe center
             let (source, target) = make_points_from_coords(
@@ -133,6 +131,7 @@ async fn main() {
     }
 
     else {
+        println!("debug mode");
         let gtfs = read_from_gtfs_zip("test2.zip");
 
         //overhead for cloning these strings is very low, it's just for displaying anyway
@@ -142,5 +141,44 @@ async fn main() {
 
         let (transit_graph, connections) = TimeExpandedGraph::new(gtfs, "Wednesday".to_string(), 0);
         let (mut router, mut paths) = TransitDijkstra::new(transit_graph);
+
+        let (source, target) = make_points_from_coords(
+            0.0, //always 0
+            0.0,
+            0.0, //always 0
+            5.0,
+        );
+
+        let now = Instant::now();
+        let graph = query_graph_construction(
+            &mut router,
+            &mut paths,
+            source,
+            target,
+            21600, //5:10 AM
+            86400, //24 hour searchspace
+            preset_distance,
+        );
+
+        let output = File::create("test.json").unwrap();
+        println!("query graph constructed in {:?}", now.elapsed());
+        serde_json::to_writer(output, &graph).unwrap();
+
+        let run_query = query_graph_search(connections, graph, &mut paths);
+
+        if let Some(stuff) = run_query {
+            let path = stuff.1.get_tp(stuff.0, &paths, &trips);
+            for (node, route) in path.0 {
+                println!("{:?}", node);
+                if let Some(route) = route {
+                    println!("via {:?}", routes.get(&route).unwrap().short_name);
+                } else {
+                    println!("start {}", stops.get(&node.station_id.to_string()).unwrap());
+                }
+            }
+        } else {
+            println!("no path");
+        }
+
     }
 }

Diff for: src/transit_dijkstras.rs

@@ -181,7 +181,6 @@ impl TransitDijkstra {
             //stop conditions
             //cost or # of settled nodes goes over limit
             if pathed_node.cost_from_start > self.cost_upper_bound {
-                println!("cost over");
                 return visited_nodes;
             }
 

Diff for: test.json

@@ -0,0 +1,87 @@
+{
+    "source": {
+        "x": 0.0,
+        "y": 0.0
+    },
+    "target": {
+        "x": 5.0,
+        "y": 0.0
+    },
+    "edges": {},
+    "source_stations": [
+        {
+            "id": 5,
+            "lat": 0,
+            "lon": 0
+        }
+    ],
+    "target_stations": [
+        {
+            "id": 4,
+            "lat": 0,
+            "lon": 500000000000000
+        }
+    ],
+    "hubs": [
+        3
+    ],
+    "source_nodes": [
+        "Departure,5,21900,5",
+        "Arrival,5,21600,9",
+        "Departure,5,21900,8",
+        "Arrival,5,21600,5",
+        "Transfer,5,21600,6",
+        "Transfer,5,21600,8",
+        "Transfer,5,21600,5",
+        "Arrival,5,21600,6",
+        "Transfer,5,21600,9",
+        "Departure,5,21900,12",
+        "Arrival,5,21600,8",
+        "Transfer,5,21600,12",
+        "Departure,5,21900,6",
+        "Departure,5,21900,9",
+        "Arrival,5,21600,12"
+    ],
+    "target_nodes": [
+        "Transfer,4,21600,10",
+        "Departure,4,21900,10",
+        "Arrival,4,21600,10"
+    ],
+    "station_map": {
+        "C": {
+            "id": 2,
+            "lat": 0,
+            "lon": 200000000000000
+        },
+        "B": {
+            "id": 0,
+            "lat": 0,
+            "lon": 100000000000000
+        },
+        "G": {
+            "id": 1,
+            "lat": 0,
+            "lon": 600000000000000
+        },
+        "F": {
+            "id": 4,
+            "lat": 0,
+            "lon": 500000000000000
+        },
+        "D": {
+            "id": 3,
+            "lat": 0,
+            "lon": 300000000000000
+        },
+        "A": {
+            "id": 5,
+            "lat": 0,
+            "lon": 0
+        },
+        "E": {
+            "id": 6,
+            "lat": 0,
+            "lon": 400000000000000
+        }
+    }
+}

Diff for: test/agency.txt

@@ -0,0 +1,2 @@
+agency_url,agency_name,agency_timezone,agency_id
+http://www.thebus.org,TheBus,Pacific/Honolulu,1

Diff for: test/calendar.txt

@@ -0,0 +1,3 @@
+service_id,start_date,end_date,monday,tuesday,wednesday,thursday,friday,saturday,sunday
+1,20250101,20251231,1,1,1,1,1,1,1
+

Diff for: test/routes.txt

@@ -0,0 +1,8 @@
+route_long_name,route_id,route_type,route_text_color,agency_id,route_color,route_url,route_desc,route_short_name
+AE,AE,0,,1,,,,
+ABE,ABE,0,,1,,,,
+ABC,ABC,0,,1,,,,
+ABDE,ABDE,0,,1,,,,
+ABCDE,ABCDE,0,,1,,,,
+CFG,CFG,0,,1,,,,
+

Diff for: test/stop_times.txt

@@ -0,0 +1,54 @@
+trip_id,arrival_time,departure_time,stop_id,stop_sequence,stop_headsign,pickup_type,drop_off_type,shape_dist_traveled
+a1,06:00:00,06:05:00,A,1,,,,
+a1,06:10:00,06:15:00,E,2,,,,
+
+b1,07:00:00,07:05:00,E,1,,,,
+b1,07:10:00,07:15:00,A,2,,,,
+
+d1,06:00:00,06:05:00,A,1,,,,
+d1,06:10:00,06:15:00,B,2,,,,
+d1,06:20:00,06:25:00,E,3,,,,
+
+e1,07:00:00,07:05:00,E,1,,,,
+e1,07:10:00,07:15:00,B,2,,,,
+e1,07:20:00,07:25:00,A,3,,,,
+
+
+g1,06:00:00,06:05:00,A,1,,,,
+g1,06:10:00,06:15:00,B,2,,,,
+g1,06:20:00,06:25:00,C,3,,,,
+
+h1,07:00:00,07:05:00,C,1,,,,
+h1,07:10:00,07:15:00,B,2,,,,
+h1,07:20:00,07:25:00,A,3,,,,
+
+j1,06:00:00,06:05:00,A,1,,,,
+j1,06:10:00,06:15:00,B,2,,,,
+j1,06:20:00,06:25:00,D,3,,,,
+j1,06:30:00,06:35:00,E,4,,,,
+
+k1,07:00:00,07:05:00,E,1,,,,
+k1,07:10:00,07:15:00,D,2,,,,
+k1,07:20:00,07:25:00,B,3,,,,
+k1,07:30:00,07:35:00,A,4,,,,
+
+m1,06:00:00,06:05:00,A,1,,,,
+m1,06:10:00,06:15:00,B,2,,,,
+m1,06:20:00,06:25:00,C,3,,,,
+m1,06:30:00,06:35:00,D,4,,,,
+m1,06:40:00,06:45:00,E,5,,,,
+
+n1,07:00:00,07:05:00,E,1,,,,
+n1,07:10:00,07:15:00,D,2,,,,
+n1,07:20:00,07:25:00,C,3,,,,
+n1,07:30:00,07:35:00,B,4,,,,
+n1,07:40:00,07:45:00,A,5,,,,
+
+p1,06:20:00,06:25:00,C,1,,,,
+p1,07:20:00,07:25:00,F,2,,,,
+p1,08:20:00,08:25:00,G,3,,,,
+
+q1,05:00:00,05:05:00,G,1,,,,
+q1,06:00:00,06:05:00,F,2,,,,
+q1,07:00:00,07:05:00,C,3,,,,
+

Diff for: test/stops.txt

@@ -0,0 +1,8 @@
+stop_id,stop_code,stop_name,stop_lat,stop_lon,zone_id,stop_url,location_type,parent_station,stop_timezone,wheelchair_boarding
+A,,,33.0,-117.00001,,,,,,0
+B,,,33.0,-117.00002,,,,,,0
+C,,,33.0,-117.00003,,,,,,0
+D,,,33.0,-117.00004,,,,,,0
+E,,,33.0,-117.00005,,,,,,0
+F,,,33.0,-117.00006,,,,,,0
+G,,,33.0,-117.00007,,,,,,0

Diff for: test/trips.txt

@@ -0,0 +1,13 @@
+block_id,route_id,direction_id,trip_headsign,shape_id,service_id,trip_id
+0,AE,0,xxxx,,1,a1
+0,AE,1,xxxx,,1,b1
+0,ABE,0,xxxx,,1,d1
+0,ABE,1,xxxx,,1,e1
+0,ABC,0,xxxx,,1,g1
+0,ABC,1,xxxx,,1,h1
+0,ABDE,0,xxxx,,1,j1
+0,ABDE,1,xxxx,,1,k1
+0,ABCDE,0,xxxx,,1,m1
+0,ABCDE,1,xxxx,,1,n1
+0,CFG,0,xxxx,,1,p1
+0,CFG,1,xxxx,,1,q1