1 files changed, 245 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/datatypes/heattotosca/unifiedmodel/consolidation/EntityConsolidationData.java b/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/datatypes/heattotosca/unifiedmodel/consolidation/EntityConsolidationData.java
new file mode 100644
index 0000000000..4545d5cdbd
--- /dev/null
+++ b/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/datatypes/heattotosca/unifiedmodel/consolidation/EntityConsolidationData.java
@@ -0,0 +1,245 @@
+package org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.consolidation;
+
+import org.openecomp.sdc.tosca.datatypes.model.RequirementAssignment;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * The type Entity consolidation data.
+ */
+public class EntityConsolidationData {
+
+  private String nodeTemplateId;
+
+  //groups that point to this entity node template
+  private List<String> groupIds;
+
+  // key - node template id which has connection to this entity
+  // value - List of Requirement assignment data which connect to this entity
+  private Map<String, List<RequirementAssignmentData>> nodesConnectedIn;
+
+  // key - node template id which connected from this entity
+  // List of Requirement assignment data which connect to the key node template id
+  private Map<String, List<RequirementAssignmentData>> nodesConnectedOut;
+
+  //key - node template id which include get attribute function from this entity
+  //key - node template id which include get attribute function from this entity
+  //value - List of getAttr data
+  private Map<String, List<GetAttrFuncData>> nodesGetAttrIn;
+
+  //key - node template id which is pointed by this entity using get attribute function
+  //value - List of getAttr data
+  private Map<String, List<GetAttrFuncData>> nodesGetAttrOut;
+
+  //List of getAttr data
+  private List<GetAttrFuncData> outputParametersGetAttrIn;
+
+  /**
+   * Gets node template id.
+   *
+   * @return the node template id
+   */
+  public String getNodeTemplateId() {
+    return nodeTemplateId;
+  }
+
+  /**
+   * Sets node template id.
+   *
+   * @param nodeTemplateId the node template id
+   */
+  public void setNodeTemplateId(String nodeTemplateId) {
+    this.nodeTemplateId = nodeTemplateId;
+  }
+
+  /**
+   * Gets group ids point to me.
+   *
+   * @return the group ids point to me
+   */
+  public List<String> getGroupIds() {
+    return groupIds;
+  }
+
+  /**
+   * Sets group ids point to me.
+   *
+   * @param groupIds the group ids point to me
+   */
+  public void setGroupIds(List<String> groupIds) {
+    this.groupIds = groupIds;
+  }
+
+
+  /**
+   * Sets node connected to me.
+   *
+   * @param nodesConnectedIn the node connected to me
+   */
+  public void setNodesConnectedIn(Map<String, List<RequirementAssignmentData>> nodesConnectedIn) {
+    this.nodesConnectedIn = nodesConnectedIn;
+  }
+
+  /**
+   * Add nodeConnectedIn.
+   *
+   * @param nodeTemplateId        the node template id which has connection to me
+   * @param requirementId         the requirement id
+   * @param requirementAssignment the requirement assignment
+   */
+  public void addNodesConnectedIn(String nodeTemplateId, String requirementId,
+                                  RequirementAssignment requirementAssignment) {
+    if (this.nodesConnectedIn == null) {
+      this.nodesConnectedIn = new HashMap<>();
+    }
+    this.nodesConnectedIn.computeIfAbsent(nodeTemplateId, k -> new ArrayList<>());
+    this.nodesConnectedIn.get(nodeTemplateId).add(new RequirementAssignmentData(requirementId,
+        requirementAssignment));
+  }
+
+  /**
+   * Gets node connected to me.
+   *
+   * @return the node connected to me
+   */
+  public Map<String, List<RequirementAssignmentData>> getNodesConnectedIn() {
+    return nodesConnectedIn;
+  }
+
+
+  /**
+   * Gets node connected from me.
+   *
+   * @return the node connected from me
+   */
+  public Map<String, List<RequirementAssignmentData>> getNodesConnectedOut() {
+    return nodesConnectedOut;
+  }
+
+  /**
+   * Sets node connected from me.
+   *
+   * @param nodesConnectedOut the node connected from me
+   */
+  public void setNodesConnectedOut(
+      Map<String, List<RequirementAssignmentData>> nodesConnectedOut) {
+    this.nodesConnectedOut = nodesConnectedOut;
+  }
+
+  /**
+   * Add nodeConnectedOut.
+   *
+   * @param nodeTemplateId        the node template id which is connected from me
+   * @param requirementId         the requirement id
+   * @param requirementAssignment the requirement assignment
+   */
+  public void addNodesConnectedOut(String nodeTemplateId, String requirementId,
+                                   RequirementAssignment
+                                       requirementAssignment) {
+    if (this.nodesConnectedOut == null) {
+      this.nodesConnectedOut = new HashMap<>();
+    }
+    this.nodesConnectedOut.computeIfAbsent(nodeTemplateId, k -> new ArrayList<>());
+    this.nodesConnectedOut.get(nodeTemplateId).add(new RequirementAssignmentData(requirementId,
+        requirementAssignment));
+  }
+
+  /**
+   * Gets nodes get attr in.
+   *
+   * @return the get attr in
+   */
+  public Map<String, List<GetAttrFuncData>> getNodesGetAttrIn() {
+    return nodesGetAttrIn;
+  }
+
+  /**
+   * Sets nodes get attr in.
+   *
+   * @param nodesGetAttrIn the get attr in
+   */
+  public void setNodesGetAttrIn(
+      Map<String, List<GetAttrFuncData>> nodesGetAttrIn) {
+    this.nodesGetAttrIn = nodesGetAttrIn;
+  }
+
+  /**
+   * Add nodes get attr in data.
+   *
+   * @param nodeTemplateId  the node template id
+   * @param getAttrFuncData get attr data
+   */
+  public void addNodesGetAttrIn(String nodeTemplateId, GetAttrFuncData getAttrFuncData) {
+    if (nodesGetAttrIn == null) {
+      nodesGetAttrIn = new HashMap<>();
+    }
+    this.nodesGetAttrIn.putIfAbsent(nodeTemplateId, new ArrayList<>());
+    this.nodesGetAttrIn.get(nodeTemplateId).add(getAttrFuncData);
+  }
+
+  /**
+   * Gets output parameters get attr from me.
+   *
+   * @return the get attr from me
+   */
+  public List<GetAttrFuncData> getOutputParametersGetAttrIn() {
+    return outputParametersGetAttrIn;
+  }
+
+  /**
+   * Sets output parameters get attr from me.
+   *
+   * @param outputParametersGetAttrIn the output parameters get attr from me
+   */
+  public void setOutputParametersGetAttrIn(List<GetAttrFuncData> outputParametersGetAttrIn) {
+    this.outputParametersGetAttrIn = outputParametersGetAttrIn;
+  }
+
+  /**
+   * Add output parameters get attr data.
+   *
+   * @param getAttrFuncData get attr data
+   */
+  public void addOutputParamGetAttrIn(GetAttrFuncData getAttrFuncData) {
+    if (outputParametersGetAttrIn == null) {
+      outputParametersGetAttrIn = new ArrayList<>();
+    }
+    this.outputParametersGetAttrIn.add(getAttrFuncData);
+  }
+
+  /**
+   * Gets nodes get attr out.
+   *
+   * @return the get attr out
+   */
+  public Map<String, List<GetAttrFuncData>> getNodesGetAttrOut() {
+    return nodesGetAttrOut;
+  }
+
+  /**
+   * Sets nodes get attr out.
+   *
+   * @param nodesGetAttrOut the get attr out
+   */
+  public void setNodesGetAttrOut(
+      Map<String, List<GetAttrFuncData>> nodesGetAttrOut) {
+    this.nodesGetAttrOut = nodesGetAttrOut;
+  }
+
+  /**
+   * Add nodes get attr out data.
+   *
+   * @param nodeTemplateId  the node template id
+   * @param getAttrFuncData get attr data
+   */
+  public void addNodesGetAttrOut(String nodeTemplateId, GetAttrFuncData getAttrFuncData) {
+    if (nodesGetAttrOut == null) {
+      nodesGetAttrOut = new HashMap<>();
+    }
+    this.nodesGetAttrOut.putIfAbsent(nodeTemplateId, new ArrayList<>());
+    this.nodesGetAttrOut.get(nodeTemplateId).add(getAttrFuncData);
+  }
+}