From a6114be086cfd4bfda8e445e89da5eac06f11fe3 Mon Sep 17 00:00:00 2001
From: jsimomaa <jani.simomaa@gmail.com>
Date: Fri, 18 Oct 2019 10:05:39 +0300
Subject: [PATCH] Add configurationBySolverName map to MappingBase

gitlab #402

Cherry picked manually from I3a315e48a79efbcbca73408de8f0f172cf6d8344

Change-Id: I0e7ec025e0eb2e0e215661eb92eee8376b83408a
---
 .../synchronization/base/MappingBase.java     | 35 +++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/bundles/org.simantics.structural.synchronization.client/src/org/simantics/structural/synchronization/base/MappingBase.java b/bundles/org.simantics.structural.synchronization.client/src/org/simantics/structural/synchronization/base/MappingBase.java
index 78678b505..aa1afd2f0 100644
--- a/bundles/org.simantics.structural.synchronization.client/src/org/simantics/structural/synchronization/base/MappingBase.java
+++ b/bundles/org.simantics.structural.synchronization.client/src/org/simantics/structural/synchronization/base/MappingBase.java
@@ -6,6 +6,7 @@ import gnu.trove.procedure.TObjectProcedure;
 import gnu.trove.set.hash.THashSet;
 
 import java.io.PrintWriter;
+import java.util.Map;
 
 /**
  * The entry point to the mapping structure between Simantics database and a
@@ -22,6 +23,11 @@ abstract public class MappingBase<T extends ComponentBase<T>> {
      * Set of all components indexed by their UID.
      */
     transient protected THashMap<String, T> configurationByUid;
+
+    /**
+     * Set of all components indexed by their solver name.
+     */
+    transient protected Map<String, T> configurationBySolverName;
     
     /** 
      * Set of components whose removal is delayed because they might
@@ -89,6 +95,29 @@ abstract public class MappingBase<T extends ComponentBase<T>> {
         }
     }
 
+    public Map<String, T> getConfigurationBySolverName() {
+        Map<String, T> result = configurationBySolverName;
+        if (result == null)
+            result = configurationBySolverName = createConfigurationBySolverName(getConfiguration());
+        return result;
+    }
+
+    protected Map<String, T> createConfigurationBySolverName(T configuration) {
+        THashMap<String, T> configurationBySolverName = new THashMap<>();
+        browseConfigurationBySolverName(configurationBySolverName, configuration);
+        return configurationBySolverName;
+    }
+
+    private void browseConfigurationBySolverName(
+            THashMap<String, T> configurationBySolverName,
+            T configuration) {
+        configurationBySolverName.put(configuration.solverComponentName, configuration);
+        for(T child : configuration.getChildren()) {
+            browseConfigurationBySolverName(configurationBySolverName, child);
+            child.parent = configuration;
+        }
+    }
+
     public T detachOrCreateComponent(String uid) {
         T result = configurationByUid.get(uid);
         if(result == null) {
@@ -140,6 +169,8 @@ abstract public class MappingBase<T extends ComponentBase<T>> {
     public void remove(final Solver solver, T component) {
         if(configurationByUid != null)
             configurationByUid.remove(component.uid);
+        if (configurationBySolverName != null)
+            configurationBySolverName.remove(component.solverComponentName);
         if(component.getChildMap() != null)
             component.getChildMap().forEachValue(new TObjectProcedure<T>() {
                 @Override
@@ -212,6 +243,10 @@ abstract public class MappingBase<T extends ComponentBase<T>> {
     public void dispose() {
         if (configurationByUid != null)
             configurationByUid.clear();
+        if (configurationBySolverName != null) {
+            configurationBySolverName.clear();
+            configurationBySolverName = null;
+        }
         pendingRemoval.clear();
     }
     
-- 
2.43.2