X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Forg%2Fonap%2Fclamp%2Fclds%2Futil%2Fdrawing%2FDocumentBuilder.java;h=7c624be0fbec58ec08e6b377909b1d763678d593;hb=061ecec6a6099ad7fb480b0fe67f6e5154f9db23;hp=f34eaf2ee603c72912f196949bd804a07d1c18c2;hpb=515251fb5ce2d8f2b4ccb5bd5d3ebd46deb4d29a;p=clamp.git

diff --git a/src/main/java/org/onap/clamp/clds/util/drawing/DocumentBuilder.java b/src/main/java/org/onap/clamp/clds/util/drawing/DocumentBuilder.java
index f34eaf2e..7c624be0 100644
--- a/src/main/java/org/onap/clamp/clds/util/drawing/DocumentBuilder.java
+++ b/src/main/java/org/onap/clamp/clds/util/drawing/DocumentBuilder.java
@@ -33,6 +33,8 @@ public class DocumentBuilder {
     private final Document documentFactory;
 
     static final String DATA_ELEMENT_ID_ATTRIBUTE = "data-element-id";
+    static final String DATA_ELEMENT_GROUPING_ATTRIBUTE = "data-grouping-id";
+    static final String DATA_FOR_UI_ATTRIBUTE = "data-for-ui";
 
     DocumentBuilder(Document groupingDocument, Document documentFactory) {
         this.groupingDocument = groupingDocument;
@@ -40,10 +42,20 @@ public class DocumentBuilder {
     }
 
     void pushChangestoDocument(SVGGraphics2D g2d, String dataElementId) {
+        pushChangestoDocument(g2d, dataElementId,null,null);
+    }
+
+    void pushChangestoDocument(SVGGraphics2D g2d, String dataElementId, String dataGroupingId, String dataForUI) {
         Element element =
             this.documentFactory.createElementNS(SVGGraphics2D.SVG_NAMESPACE_URI,
                 SVGGraphics2D.SVG_G_TAG);
         element.setAttribute(DATA_ELEMENT_ID_ATTRIBUTE, dataElementId);
+        if (dataGroupingId != null) {
+            element.setAttribute(DATA_ELEMENT_GROUPING_ATTRIBUTE, dataGroupingId);
+        }
+        if (dataForUI != null) {
+            element.setAttribute(DATA_FOR_UI_ATTRIBUTE, dataForUI);
+        }
         g2d.getRoot(element);
         Node node = this.groupingDocument.importNode(element, true);
         this.groupingDocument.getDocumentElement().appendChild(node);