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);
 }