First attempt at saving the node positions to the dot file.

Author: CoderAllan

src/commands/showHierarchyBase.ts

@@ -112,8 +112,9 @@ export class ShowHierarchyBase extends CommandBase {
   }
 
   protected saveAsGraphViz(graphVizFilename: string, messageText: string, graphType: string, popMessageText: string) {
+    const message = JSON.parse(messageText);
     const graphVizManager = new GraphVizManager();
-    const fileContent = graphVizManager.createGraphVizDiagram(graphType, this.nodes, this.edges);
+    const fileContent = graphVizManager.createGraphVizDiagram(graphType, this.nodes, message.nodes, this.edges);
 
     // Write the prettified xml string to the ReadMe-ProjectStructure.dgml file.
     var directoryPath: string = this.fsUtils.getWorkspaceFolder();

src/graphvizManager.ts

@@ -1,8 +1,26 @@
-import { Edge, Node } from "@model";
+import { Edge, NetworkNode, Node, Position } from "@model";
 
 export class GraphVizManager {
 
-  public createGraphVizDiagram(graphType: string, nodes: Node[], edges: Edge[]): string {
+  public createGraphVizDiagram(graphType: string, nodes: Node[], nodeInfoDictionary: { [id: string]: NetworkNode }, edges: Edge[]): string {
+    const maxX = Math.max(...Object.keys(nodeInfoDictionary).map(id => nodeInfoDictionary[id].position.x));
+    const minX = Math.min(...Object.keys(nodeInfoDictionary).map(id => nodeInfoDictionary[id].position.x));
+    const diffX = Math.max(...[Math.abs(maxX),Math.abs(minX)]);
+    const maxY = Math.max(...Object.keys(nodeInfoDictionary).map(id => nodeInfoDictionary[id].position.y));
+    const minY = Math.min(...Object.keys(nodeInfoDictionary).map(id => nodeInfoDictionary[id].position.y));
+    const diffY = Math.max(...[Math.abs(maxY),Math.abs(minY)]);
+
+    nodes.forEach(node => {
+      if(node.id in nodeInfoDictionary){
+        const networkNode = nodeInfoDictionary[node.id];
+        if (networkNode.label) {
+          node.name = networkNode.label;
+        }
+        if (node.id in nodeInfoDictionary) {
+          node.position = {x: Math.round(((networkNode.position.x + diffX) / maxX) * nodes.length), y: Math.round(((networkNode.position.y + diffY) / maxY) * nodes.length)};
+        }
+      }
+    });
     const digraphNodes = this.generateDigraphNodes(nodes);
     const digraphEdges = this.generateDigraphEdges(edges);
     const graphVizDigraph = this.addRootNode(graphType, digraphNodes, digraphEdges);

src/model/Node.ts

@@ -56,12 +56,18 @@ export class Node {
     const regex = /\W/g;
     const id = this.id.replace(regex, '_');
     const attributes: string[] = [];
-    attributes.push(`label="${this.name}"`);
+    const label = this.config.maximumNodeLabelLength !== -1 && this.name.length > this.config.maximumNodeLabelLength ? this.name.substr(0, this.config.maximumNodeLabelLength) + '...' : this.name;
+    attributes.push(`label="${label}"`);
     const color = this.getNodeTypeColor(this.nodeType);
     if (color) {
       attributes.push(`color="${color}"`);
       attributes.push(`style="filled"`);
     }
+    if (this.position !== undefined) {
+      const x = this.position.x;
+      const y = this.position.y;
+      attributes.push(`pos="${x},${y}!"`);
+    }
     let attributesStr: string = '';
     if (attributes.length > 0) {
       attributesStr = ` [${attributes.join(', ')}]`;