Update platform parser

1 files changed, 13 insertions, 1 deletions

diff --git a/esds/helpers/platform.py b/esds/helpers/platform.py
index cc08709..658b05d 100644
--- a/esds/helpers/platform.py
+++ b/esds/helpers/platform.py
@@ -1,5 +1,6 @@
 
 import yaml, os
+import numpy as np
 
 class YAMLPlatformFile:
 
@@ -84,6 +85,7 @@ class YAMLPlatformFile:
 
     def parse_interfaces(self):
         interfaces=self.platform["interfaces"]
+        node_count=self.default["node_count"]
         for i in interfaces:
             is_wired=interfaces[i]["wireless"]
             links=list()
@@ -94,7 +96,17 @@ class YAMLPlatformFile:
                 links.append(self.parse_link(interfaces[i]["links"]))
             ##### Create network matrix
             for link in links:
-                print(link)
+                BW=np.full((node_count,node_count),0)
+                LAT=np.full((node_count,node_count),0)
+                for n1 in link[0]:
+                    for n2 in link[3]:
+                        BW[n1][n2]=link[1]
+                        LAT[n1][n2]=link[2]
+            self.default["interfaces"][i]={
+                "is_wired": is_wired,
+                "bandwidth": BW,
+                "latency": LAT
+            }
 
     def parse_nodes(self):
         nodes=self.platform["nodes"]