Merge pull request github#17 from warerebel/github#16---Bugfix

github#16 - Ensure map gets built with end nodes
warerebel · Aug 26, 2022 · 08e1f23 · 08e1f23
2 parents 83060b7 + 9c86a4d
commit 08e1f23
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
  "name": "openlr_decoder",
- "version": "0.4.0",
+ "version": "0.4.1",
  "description": "A general purpose road network OpenLR decoding solution",
  "keywords": [
  "openlr",

diff --git a/src/graph.ts b/src/graph.ts
@@ -4,10 +4,17 @@ import type {node, linkLookup, nodeChildLink, graphInput} from "./nodes";
 export function buildLinkLookups(nodeCollection: Array<node>) {
  const linkLookup = {};
  const graphInput = {};
- nodeCollection.map(node => node.startLinks.map(link => addLinkToGraph(link, linkLookup, graphInput)));
+ nodeCollection.map(node => checkForEndpoints(node, linkLookup, graphInput));
  return {links: linkLookup as linkLookup, graphInput: graphInput as graphInput};
 }
 
+function checkForEndpoints(node: node, linkLookup: linkLookup, graphInput: graphInput){
+ if(node.startLinks)
+ node.startLinks.map(link => addLinkToGraph(link, linkLookup, graphInput));
+ else
+ node.endLinks.map(link => addLinkToGraph(link, linkLookup, graphInput));
+}
+
 export function getGraph(input: graphInput){
  return new Graph(input);
 }