wifi_intrastructure_link.py

Go to the documentation of this file.

import math
 
try:
    from ns import ns
except ModuleNotFoundError:
    raise SystemExit(
        "Error: ns3 Python module not found;"
        " Python bindings may not be enabled"
        " or your PYTHONPATH might not be properly configured"
    )
from gi.repository import GooCanvas
 
try:
    from ns3.visualizer.base import Link, transform_distance_canvas_to_simulation
except ModuleNotFoundError:
    from visualizer.base import Link, transform_distance_canvas_to_simulation
 
 
 
class WifiLink(Link):
    
    def __init__(self, parent_canvas_item, sta, dev):
        """! Initialize function.
        @param self The object pointer.
        @param parent_canvas_item: parent canvas
        @param sta The STA node
        @param dev The dev
        """
        super(WifiLink, self).__init__()
        self.node1node1 = sta
        self.devdev = dev
        self.node2node2 = None  # ap
        self.canvas_itemcanvas_item = GooCanvas.CanvasGroup(parent=parent_canvas_item)
        self.invisible_lineinvisible_line = GooCanvas.CanvasPolyline(
            parent=self.canvas_itemcanvas_item,
            line_width=25.0,
            visibility=GooCanvas.CanvasItemVisibility.HIDDEN,
        )
        self.visible_linevisible_line = GooCanvas.CanvasPolyline(
            parent=self.canvas_itemcanvas_item,
            line_width=1.0,
            stroke_color_rgba=0xC00000FF,
            line_dash=GooCanvas.CanvasLineDash.newv([2.0, 2.0]),
        )
        # self.invisible_line.set_property("pointer-events", (GooCanvas.CanvasPointerEvents.STROKE_MASK
        #                                             |GooCanvas.CanvasPointerEvents.FILL_MASK
        #                                             |GooCanvas.CanvasPointerEvents.PAINTED_MASK))
        self.canvas_itemcanvas_item.pyviz_object = self
        self.canvas_itemcanvas_item.lower(None)
        self.set_apset_ap(None)
 
    def set_ap(self, ap):
        """! Set AP.
        @param self The object pointer.
        @param ap The AP node
        @return none
        """
        if ap is self.node2node2:
            return
        if self.node2node2 is not None:
            self.node2node2.remove_link(self)
        self.node2node2 = ap
        if self.node2node2 is None:
            self.canvas_itemcanvas_item.set_property("visibility", GooCanvas.CanvasItemVisibility.HIDDEN)
        else:
            self.node2node2.add_link(self)
            self.canvas_itemcanvas_item.set_property("visibility", GooCanvas.CanvasItemVisibility.VISIBLE)
        self.update_pointsupdate_points()
 
    def update_points(self):
        """! Update points function.
        @param self The object pointer.
        @return none
        """
        if self.node2node2 is None:
            return
        pos1_x, pos1_y = self.node1node1.get_position()
        pos2_x, pos2_y = self.node2node2.get_position()
        points = GooCanvas.CanvasPoints.new(2)
        points.set_point(0, pos1_x, pos1_y)
        points.set_point(1, pos2_x, pos2_y)
        self.visible_linevisible_line.set_property("points", points)
        self.invisible_lineinvisible_line.set_property("points", points)
 
    def destroy(self):
        """! Destroy function.
        @param self The object pointer.
        @return none
        """
        self.canvas_itemcanvas_item.destroy()
        self.node1node1 = None
        self.node2node2 = None
 
    def tooltip_query(self, tooltip):
        """! Destroy function.
        @param self The object pointer.
        @param tooltip The tooltip.
        @return tooltip
        """
        pos1_x, pos1_y = self.node1node1.get_position()
        pos2_x, pos2_y = self.node2node2.get_position()
        dx = pos2_x - pos1_x
        dy = pos2_y - pos1_y
        d = transform_distance_canvas_to_simulation(math.sqrt(dx * dx + dy * dy))
        mac = self.devdev.GetMac()
        tooltip.set_text(
            (
                "WiFi link between STA Node %i and AP Node %i; distance=%.2f m.\n"
                "SSID: %s\n"
                "BSSID: %s"
            )
            % (self.node1node1.node_index, self.node2node2.node_index, d, mac.GetSsid(), mac.GetBssid())
        )
 
 
 
class WifiLinkMonitor(object):
    
    def __init__(self, dummy_viz):
        """! Initialize function.
        @param self The object pointer.
        @param dummy_viz A dummy visualizer
        """
        self.access_pointsaccess_points = {}  # bssid -> node
        self.stationsstations = []  # list of (sta_netdevice, viz_node, wifi_link)
 
    def scan_nodes(self, viz):
        """! Scan nodes function.
        @param self The object pointer.
        @param viz The visualizer object
        @return none
        """
        for sta_netdevice, viz_node, wifi_link in self.stationsstations:
            wifi_link.destroy()
 
        self.access_pointsaccess_points = {}
        self.stationsstations = []
 
        for node in viz.nodes.values():
            ns3_node = ns.network.NodeList.GetNode(node.node_index)
            for devI in range(ns3_node.GetNDevices()):
                dev = ns3_node.GetDevice(devI)
                if not isinstance(dev, ns.wifi.WifiNetDevice):
                    continue
                wifi_mac = dev.GetMac()
                if isinstance(wifi_mac, ns.wifi.StaWifiMac):
                    wifi_link = WifiLink(viz.links_group, node, dev)
                    self.stationsstations.append((dev, node, wifi_link))
                elif isinstance(wifi_mac, ns.wifi.ApWifiMac):
                    bssid = ns.network.Mac48Address.ConvertFrom(dev.GetAddress())
                    self.access_pointsaccess_points[str(bssid)] = node
        # print "APs: ", self.access_points
        # print "STAs: ", self.stations
 
    def simulation_periodic_update(self, viz):
        """! Simulation Periodic Update function.
        @param self The object pointer.
        @param viz The visualizer object
        @return none
        """
        for sta_netdevice, viz_node, wifi_link in self.stationsstations:
            if not sta_netdevice.IsLinkUp():
                wifi_link.set_ap(None)
                continue
            bssid = str(sta_netdevice.GetMac().GetBssid())
            if bssid == "00:00:00:00:00:00":
                wifi_link.set_ap(None)
                continue
            ap = self.access_pointsaccess_points[bssid]
            wifi_link.set_ap(ap)
 
    def update_view(self, viz):
        """! Update View function.
        @param self The object pointer.
        @param viz The visualizer object
        @return none
        """
        for dummy_sta_netdevice, dummy_viz_node, wifi_link in self.stationsstations:
            if wifi_link is not None:
                wifi_link.update_points()
 
 
def register(viz):
    link_monitor = WifiLinkMonitor(viz)
    viz.connect("simulation-periodic-update", link_monitor.simulation_periodic_update)
    viz.connect("update-view", link_monitor.update_view)
    viz.connect("topology-scanned", link_monitor.scan_nodes)