O(1) UUIDs lookup

ekuiter · Nov 20, 2018 · a8001b1 · a8001b1
1 parent 165ea2f
commit a8001b1
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/client/src/modeling/GraphicalFeatureModel.ts b/client/src/modeling/GraphicalFeatureModel.ts
@@ -204,6 +204,7 @@ class GraphicalFeatureModel {
     _actualNodes: GraphicalFeatureNode[];
     _visibleNodes: GraphicalFeatureNode[];
     _constraints: GraphicalConstraint[];
+    _UUIDsToFeatures: {[x: string]: GraphicalFeatureNode} = {};
 
     // feature model as supplied by feature model messages from the server
     static fromJSON(serializedFeatureModel: SerializedFeatureModel): GraphicalFeatureModel {
@@ -254,6 +255,8 @@ class GraphicalFeatureModel {
 
                 if (isVisible(node))
                     this._visibleNodes.push(node);
+
+                this._UUIDsToFeatures[getUUID(node)] = node;
             });
         }
     }
@@ -279,8 +282,8 @@ class GraphicalFeatureModel {
     }
 
     getNode(featureUUID: string): GraphicalFeatureNode | undefined {
-        // TODO: use a map instead?
-        return this.actualNodes.find(node => getUUID(node) === featureUUID);
+        this.prepare();
+        return this._UUIDsToFeatures[featureUUID];
     }
 
     getFeature(featureUUID: string): GraphicalFeature | undefined {