raptor-journey-planner/dist/src/transfer-patterns.js

Implementation of the Round bAsed Public Transit Optimized Router (Raptor) journey planning algorithm.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const cp = require("node:child_process");
const ProgressBar = require("progress");
const gtfs_stream_1 = require("gtfs-stream");
const fs = require("node:fs");
const os = require("node:os");
const numCPUs = os.cpus().length;
async function run(filename, dateString) {
    const date = new Date(dateString);
    const stops = await getStops(filename);
    const bar = new ProgressBar("  [:current of :total] [:bar] :percent eta :eta  ", { total: stops.length });
    for (let i = 0; i < Math.min(numCPUs - 2, stops.length); i++) {
        const worker = cp.fork(`${__dirname}/transfer-pattern-worker`, [filename, date.toISOString()]);
        worker.on("message", () => {
            if (stops.length > 0) {
                bar.tick();
                worker.send(stops.pop());
            }
            else {
                worker.kill("SIGUSR2");
            }
        });
    }
}
async function getStops(filename) {
    return new Promise((resolve, reject) => {
        const stops = [];
        fs.createReadStream(filename)
            .pipe((0, gtfs_stream_1.plain)({ raw: true }))
            .on("data", entity => entity.type === "stop"
            && entity.data.stop_timezone === "Europe/London"
            && stops.push(entity.data.stop_id))
            .on("error", e => reject(e))
            .on("end", () => resolve(stops));
    });
}
if (process.argv[2] && process.argv[3]) {
    run(process.argv[2], process.argv[3]).catch(e => console.error(e));
}
else {
    console.log("Please specify a GTFS file and date.");
}