allow to filter by element id

model-command/src/main/java/lcsb/mapviewer/commands/ColorModelCommand.java

@@ -209,6 +209,11 @@ public class ColorModelCommand extends ModelCommand {
           return false;
         }
       }
+      if (schema.getElementId() != null && !schema.getElementId().isEmpty()) {
+        if (!element.getElementId().equalsIgnoreCase(schema.getElementId())) {
+          return false;
+        }
+      }
       if (schema.getTypes().size() > 0) {
         boolean found = false;
         for (Class<?> clazz : schema.getTypes()) {

model-command/src/test/java/lcsb/mapviewer/commands/ColorModelCommandTest.java

@@ -452,4 +452,46 @@ public class ColorModelCommandTest extends CommandTestFunctions {
     }
   }
 
+
+  public void testAliasColoringWithUnknownElementSourceId() throws Exception {
+    try {
+      Model model = getModelForFile("testFiles/sample.xml", false);
+
+      ColorSchema schema = new GenericColorSchema();
+      schema.setElementId("xxx");
+
+      Collection<ColorSchema> schemas = new ArrayList<>();
+      schemas.add(schema);
+
+      ColorModelCommand factory = new ColorModelCommand(model, schemas, colorExtractor);
+      Map<Object, ColorSchema> map = factory.getModifiedElements();
+      assertEquals(0, map.values().size());
+
+    } catch (Exception e) {
+      e.printStackTrace();
+      throw e;
+    }
+  }
+
+  @Test
+  public void testAliasColoringWithElementIdMatch() throws Exception {
+    try {
+      Model model = getModelForFile("testFiles/sample.xml", false);
+
+      ColorSchema schema = new GenericColorSchema();
+      schema.setElementId(model.getElements().iterator().next().getElementId());
+
+      Collection<ColorSchema> schemas = new ArrayList<>();
+      schemas.add(schema);
+
+      ColorModelCommand factory = new ColorModelCommand(model, schemas, colorExtractor);
+      Map<Object, ColorSchema> map = factory.getModifiedElements();
+      assertEquals(1, map.values().size());
+
+    } catch (Exception e) {
+      e.printStackTrace();
+      throw e;
+    }
+  }
+
 }

model/src/main/java/lcsb/mapviewer/model/map/layout/ColorSchema.java

@@ -42,6 +42,8 @@ public abstract class ColorSchema implements Serializable {
    * Identifier of the reaction to change the color.
    */
   private String reactionIdentifier = null;
+  
+  private String elementId = null;
 
   /**
    * Should the direction of highlighted reaction be reversed.
@@ -467,4 +469,12 @@ public abstract class ColorSchema implements Serializable {
     this.modelName = modelName;
   }
 
+  public String getElementId() {
+    return elementId;
+  }
+
+  public void setElementId(String elementId) {
+    this.elementId = elementId;
+  }
+
 }