Remove stop-ref guessing and add route number to stop/segs

This commit is contained in:
Mattéo Delabre 2020-07-22 23:10:43 +02:00
parent 4eca29c0c7
commit b67543cc0c
Signed by: matteo
GPG Key ID: AE3FBD02DC583ABB
2 changed files with 3067 additions and 1021 deletions

View File

@ -17,111 +17,6 @@ const util = require('../util');
const osm = require('./sources/osm'); const osm = require('./sources/osm');
const tam = require('./sources/tam'); const tam = require('./sources/tam');
/**
* Use theoretical passings data to guess which lines use which stops in which
* direction.
*
* This is used for suggesting possible stop IDs for stops that dont have
* one in OSM.
*
* @return Map containing for each stop its abbreviated name, the lines that
* use it and in which directions it is used.
*/
const fetchStopsRefAssociations = () => new Promise((res, rej) =>
{
const stops = {};
tam.fetchTheoretical((err, row) =>
{
if (err)
{
rej(err);
return;
}
if (!row)
{
res(stops);
return;
}
let line = row.routeShortName;
if (line === '4')
{
line += row.directionId === '0' ? 'A' : 'B';
}
if (!(row.stopId in stops))
{
stops[row.stopId] = {
name: row.stopName,
lines: new Set([line]),
directions: new Set([row.tripHeadsign]),
};
}
else
{
const stop = stops[row.stopId];
if (stop.name !== row.stopName)
{
console.warn(`Stop ${row.stopId} has multiple names: \
${row.stopName} and ${stop.name}. Only the first one will be considered.`);
}
stop.lines.add(line);
stop.directions.add(row.tripHeadsign);
}
});
});
// Mapping for abbreviations used in stop names
const stopAbbreviations = {
st: 'saint',
};
/**
* Convert a stop name to a canonical representation suitable for
* comparing two names.
*
* @param stopName Original stop name.
* @return List of normalized tokens in the name.
*/
const canonicalizeStopName = stopName => stopName
.toLowerCase()
// Remove diacritics
.normalize('NFD').replace(/[\u0300-\u036f]/g, '')
// Only keep alpha-numeric characters
.replace(/[^a-z0-9]/g, ' ')
// Split in tokens longer than two characters
.split(/\s+/g).filter(part => part.length >= 2)
// Resolve well-known abbreviations
.map(part => part in stopAbbreviations ? stopAbbreviations[part] : part);
/**
* Compute a matching score between two stop names.
*
* @param fullName Stop name in full.
* @param abbrName Abbreviated stop name.
* @return Matching score (number of common tokens).
*/
const matchStopNames = (fullName, abbrName) =>
{
const canonicalFullName = canonicalizeStopName(fullName);
const canonicalAbbrName = canonicalizeStopName(abbrName);
return canonicalFullName.filter(part =>
canonicalAbbrName.findIndex(abbrPart =>
part.startsWith(abbrPart)
) !== -1
).length;
};
/** /**
* Fetch stops and lines of the network. * Fetch stops and lines of the network.
* *
@ -142,9 +37,6 @@ relation[network="TaM"][type="route_master"][ref~"^(${lineRefs.join('|')})$"];
out body qt; out body qt;
`); `);
// Retrieve stop associations from TaM
const associations = await fetchStopsRefAssociations();
// List of retrieved objects // List of retrieved objects
const elementsList = rawData.elements; const elementsList = rawData.elements;
@ -173,11 +65,15 @@ out body qt;
const color = routeMaster.tags.colour || '#000000'; const color = routeMaster.tags.colour || '#000000';
// Extract all routes for the given line // Extract all routes for the given line
const rawRoutes = routeMaster.members.map(({ref}) => elements[ref]); const routes = [];
// Add missing stops to the result object for (let [routeRef, {ref: routeId}] of routeMaster.members.entries())
for (let route of rawRoutes)
{ {
const route = elements[routeId];
const {from, to, name} = route.tags;
const state = route.tags.state || 'normal';
// Add missing stops to the global stops object
for (let {ref, role} of route.members) for (let {ref, role} of route.members)
{ {
if (role === 'stop') if (role === 'stop')
@ -186,69 +82,9 @@ out body qt;
if (!('ref' in stop.tags)) if (!('ref' in stop.tags))
{ {
console.warn(`Stop ${stop.id} is missing a “ref” tag throw new Error(`Stop ${stop.id}
Name: ${stop.tags.name} (${osm.viewNode(stop.id)}) on line ${route.tags.name} is missing
Part of line: ${route.tags.name} a ref tag`);
URI: ${osm.viewNode(stop.id)}
`);
// Try to identify stops matching this stop in the
// TaM-provided data, using the stop name, line number
// and trip direction
const candidates = Object.entries(associations).filter(
([, {lines}]) => lines.has(route.tags.ref)
).map(([stopRef, {name, lines, directions}]) => ({
stopRef,
lines,
name,
nameScore: matchStopNames(stop.tags.name, name),
directions,
directionScore: Math.max(
...Array.from(directions).map(direction =>
matchStopNames(route.tags.to, direction)
)
),
}))
// Only keep non-zero scores for both criteria
.filter(({nameScore, directionScore}) =>
nameScore && directionScore
)
// Sort by best name score then best direction
.sort(({
nameScore: nameScore1,
directionScore: directionScore1,
}, {
nameScore: nameScore2,
directionScore: directionScore2,
}) =>
(nameScore2 - nameScore1)
|| (directionScore2 - directionScore1)
)
.slice(0, 4);
if (candidates.length === 0)
{
console.warn('No candidate found in TaM data.');
}
else
{
console.warn('Candidates:');
for (let candidate of candidates)
{
console.warn(`\
Stop ${candidate.stopRef} with name ${candidate.name} used by \
${util.choosePlural(candidate.lines.length, 'line', '.s')} \
${util.joinSentence(Array.from(candidate.lines), ', ', ' and ')} going to \
${util.joinSentence(Array.from(candidate.directions), ', ', ' or ')}
Apply in JOSM: ${osm.addTagsToNode(stop.id, ['ref=' + candidate.stopRef])}
`);
}
}
console.warn('');
} }
if (!(stop.tags.ref in stops)) if (!(stop.tags.ref in stops))
@ -257,24 +93,15 @@ ${util.joinSentence(Array.from(candidate.directions), ', ', ' or ')}
lat: stop.lat, lat: stop.lat,
lon: stop.lon, lon: stop.lon,
name: stop.tags.name, name: stop.tags.name,
lines: new Set([lineRef]), routes: [[lineRef, routeRef]],
}; };
} }
else else
{ {
stops[stop.tags.ref].lines.add(lineRef); stops[stop.tags.ref].routes.push([lineRef, routeRef]);
} }
} }
} }
}
// Reconstruct the lines route from stop to stop
const routes = [];
for (let route of rawRoutes)
{
const {from, to, name} = route.tags;
const state = route.tags.state || 'normal';
// Check that the route consists of a block of stops and platforms // Check that the route consists of a block of stops and platforms
// followed by a block of routes as dictated by PTv2 // followed by a block of routes as dictated by PTv2
@ -301,7 +128,7 @@ of ${name}`);
// List of stops in the route, expected to be in the timetable // List of stops in the route, expected to be in the timetable
// order as per PTv2 and to be traversed in order by the sequence // order as per PTv2 and to be traversed in order by the sequence
// of ways extracted below // of ways extracted below
const stops = route.members.slice(0, relationPivot) const lineStops = route.members.slice(0, relationPivot)
.filter(({role}) => role === 'stop') .filter(({role}) => role === 'stop')
.map(({ref}) => ref); .map(({ref}) => ref);
@ -312,7 +139,7 @@ of ${name}`);
// Merge all used ways in a single path // Merge all used ways in a single path
let path = []; let path = [];
let currentNode = stops[0]; let currentNode = lineStops[0];
for (let wayIndex = 0; wayIndex < ways.length; wayIndex += 1) for (let wayIndex = 0; wayIndex < ways.length; wayIndex += 1)
{ {
@ -373,15 +200,15 @@ ${name} is one-way and cannot be used in reverse.`);
} }
// Split the path into segments between stops // Split the path into segments between stops
for (let stopIndex = 0; stopIndex + 1 < stops.length; ++stopIndex) for (let stopIdx = 0; stopIdx + 1 < lineStops.length; ++stopIdx)
{ {
const begin = elements[stops[stopIndex]].tags.ref; const begin = elements[lineStops[stopIdx]].tags.ref;
const end = elements[stops[stopIndex + 1]].tags.ref; const end = elements[lineStops[stopIdx + 1]].tags.ref;
const id = `${begin}-${end}`; const id = `${begin}-${end}`;
const nodes = path.slice( const nodes = path.slice(
path.indexOf(stops[stopIndex]), path.indexOf(lineStops[stopIdx]),
path.indexOf(stops[stopIndex + 1]) + 1, path.indexOf(lineStops[stopIdx + 1]) + 1,
); );
if (id in segments) if (id in segments)
@ -392,7 +219,7 @@ ${name} is one-way and cannot be used in reverse.`);
different sequence of nodes in two or more lines.`); different sequence of nodes in two or more lines.`);
} }
segments[id].lines.add(lineRef); segments[id].routes.push([lineRef, routeRef]);
} }
else else
{ {
@ -419,8 +246,9 @@ different sequence of nodes in two or more lines.`);
// Keep track of the original sequence of nodes to // Keep track of the original sequence of nodes to
// compare with duplicates // compare with duplicates
nodes, nodes,
points, points,
lines: new Set([lineRef]), routes: [[lineRef, routeRef]],
}; };
} }
} }
@ -428,7 +256,7 @@ different sequence of nodes in two or more lines.`);
routes.push({ routes.push({
from, to, from, to,
name, state, name, state,
stops: stops.map(id => elements[id].tags.ref), stops: lineStops.map(id => elements[id].tags.ref),
}); });
} }

File diff suppressed because it is too large Load Diff