afonso/core-extra
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

add wireless network variable to CanvasGraph that maps a wireless/EMANE node to all MDRs connected to it

Browse source
This commit is contained in:
Huy Pham 2020-02-18 15:58:18 -08:00
parent 08e652633f
commit d8f586bd2b
3 changed files with 25 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
daemon/core/gui/graph/edges.py
View file

@ -177,6 +177,17 @@ class CanvasEdge:
dst_node_type = dst_node.core_node.type
is_src_wireless = NodeUtils.is_wireless_node(src_node_type)
is_dst_wireless = NodeUtils.is_wireless_node(dst_node_type)
# update the wlan/EMANE network
wlan_network = self.canvas.wireless_network
if is_src_wireless and not is_dst_wireless:
if self.src not in wlan_network:
wlan_network[self.src] = set()
wlan_network[self.src].add(self.dst)
elif not is_src_wireless and is_dst_wireless:
if self.dst not in wlan_network:
wlan_network[self.dst] = set()
wlan_network[self.dst].add(self.src)
return is_src_wireless or is_dst_wireless
def check_wireless(self):