GridCal 5.1.19__tar.gz → 5.1.21__tar.gz

{GridCal-5.1.19 → GridCal-5.1.21}/GridCal/Gui/Diagrams/MapWidget/Branches/map_line_container.py

@@ -15,6 +15,8 @@
 # along with this program; if not, write to the Free Software Foundation,
 # Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 from __future__ import annotations
+
+# import uuid
 from typing import TYPE_CHECKING, List, Union
 
 import logging
@@ -28,6 +30,7 @@ from GridCalEngine.Devices.types import BRANCH_TYPES, FluidPath
 from GridCalEngine.Devices.Branches.line import Line
 from GridCalEngine.enumerations import DeviceType
 from GridCal.Gui.Diagrams.generic_graphics import GenericDiagramWidget
+from GridCal.Gui.messages import error_msg
 
 if TYPE_CHECKING:
     from GridCal.Gui.Diagrams.MapWidget.Substation.node_graphic_item import NodeGraphicItem
@@ -78,7 +81,7 @@ class MapLineContainer(GenericDiagramWidget):
         Remove all segments from the scene
         """
         for segment in self.segments_list:
-            self.editor.remove_from_scene(segment)
+            self.editor.remove_only_graphic_element(segment)
 
         self.segments_list = list()
 
@@ -158,39 +161,15 @@ class MapLineContainer(GenericDiagramWidget):
         """
         self.clean()
 
-        diagram_locations: PointsGroup = self.editor.diagram.data.get(DeviceType.LineLocation.value, None)
-
         # draw line locations
         for elm in self.api_object.locations.data:
+            graphic_obj = self.editor.create_node(line_container=self,
+                                                  api_object=elm,
+                                                  lat=elm.lat,  # 42.0 ...
+                                                  lon=elm.long,
+                                                  index=self.number_of_nodes())  # 2.7 ...
 
-            if diagram_locations is None:
-                # no locations found, use the data from the api object
-                # lat = elm.lat
-                # lon = elm.long
-                pass
-            else:
-
-                # try to get location from the diagram
-                # We will not take the location of the element in the database because we want to keep...
-                # ... the diagram separated from database
-                # diagram_location = diagram_locations.locations.get(elm.idtag, None)
-
-                # if diagram_location is None:
-                #     # no particular location found, use the data from the api object
-                #     # lat = elm.lat
-                #     # lon = elm.long
-                #     pass
-                # else:
-                #     # Draw only what's on the diagram
-                #     # diagram data found, use it
-
-                graphic_obj = self.editor.create_node(line_container=self,
-                                                      api_object=elm,
-                                                      lat=elm.lat,  # 42.0 ...
-                                                      lon=elm.long,
-                                                      index=self.number_of_nodes())  # 2.7 ...
-
-                self.register_new_node(node=graphic_obj)
+            self.register_new_node(node=graphic_obj)
 
         # second pass: create the segments
         self.redraw_segments()
@@ -206,6 +185,7 @@ class MapLineContainer(GenericDiagramWidget):
                 self.editor.map.diagram_scene.removeItem(seg)
 
         self.nodes_list.remove(node)
+        self.api_object.locations.remove(node.api_object)
 
         for nod in self.nodes_list:
             if nod.index > node.index:
@@ -257,48 +237,16 @@ class MapLineContainer(GenericDiagramWidget):
             # draw the segment in the scene
             self.editor.add_to_scene(graphic_object=segment_graphic_object)
 
-        # diagram_locations: PointsGroup = self.editor.diagram.data.get(DeviceType.LineLocation.value, None)
-        #
-        # for idx, elm in enumerate(self.api_object.locations.data):
-        #
-        #     if diagram_locations is None:
-        #         # no locations found, use the data from the api object
-        #         # lat = elm.lat
-        #         # lon = elm.long
-        #         pass
-        #     else:
-        #
-        #         # try to get location from the diagram
-        #         diagram_location = diagram_locations.locations.get(elm.idtag, None)
-        #
-        #         if diagram_location is None:
-        #             # no particular location found, use the data from the api object
-        #             # lat = elm.lat
-        #             # lon = elm.long
-        #             pass
-        #         else:
-        #             # Draw only what's on the diagram
-        #             # diagram data found, use it
-        #
-        #             if idx > 0:
-        #                 i1 = idx
-        #                 i2 = idx - 1
-        #                 # Assuming Connector takes (scene, node1, node2) as arguments
-        #                 segment_graphic_object = Segment(first=self.nodes_list[i1],
-        #                                                  second=self.nodes_list[i2])
-        #
-        #                 self.nodes_list[i1].needsUpdateFirst = True
-        #                 self.nodes_list[i2].needsUpdateSecond = True
-        #                 segment_graphic_object.needsUpdate = True
-        #
-        #                 # register the segment in the line
-        #                 self.add_segment(segment=segment_graphic_object)
-        #
-        #                 # draw the segment in the scene
-        #                 self.editor.add_to_scene(graphic_object=segment_graphic_object)
-
         self.update_connectors()
 
+        self.editor.Update_widths()
+
+    def substation_to(self):
+        return self.editor.graphics_manager.query(elm=self.api_object.get_substation_to())
+
+    def substation_from(self):
+        return self.editor.graphics_manager.query(elm=self.api_object.get_substation_from())
+
     def insert_new_node_at_position(self, index: int):
         """
         Creates a new node in the list at the given position
@@ -307,7 +255,7 @@ class MapLineContainer(GenericDiagramWidget):
         """
 
         # Check if the index is valid
-        if 1 <= index < len(self.api_object.locations.data):
+        if 1 <= index < len(self.api_object.locations.data) and len(self.api_object.locations.data) > 1:
 
             nd1 = self.nodes_list[index]
             nd2 = self.nodes_list[index - 1]
@@ -323,7 +271,7 @@ class MapLineContainer(GenericDiagramWidget):
                                           z=new_api_node_data.alt,
                                           seq=new_api_node_data.seq,
                                           name=new_api_node_data.name,
-                                          idtag=new_api_node_data.idtag,
+                                          idtag=None,  # generates new UUID
                                           code=new_api_node_data.code)
 
             self.api_object.locations.data.insert(index, new_api_object)
@@ -356,7 +304,7 @@ class MapLineContainer(GenericDiagramWidget):
             # Return the newly created node
             return graphic_obj
 
-        elif index == 0 and len(self.api_object.locations.data) == 0:
+        elif len(self.api_object.locations.data) == 0:
 
             substation_from_graphics = self.editor.graphics_manager.query(elm=self.api_object.get_substation_from())
             substation_to_graphics = self.editor.graphics_manager.query(elm=self.api_object.get_substation_to())
@@ -375,7 +323,7 @@ class MapLineContainer(GenericDiagramWidget):
                                           idtag="",
                                           code="")
 
-            self.api_object.locations.data.insert(index, new_api_object)
+            self.api_object.locations.data.insert(0, new_api_object)
 
             # Create a new graphical node item
 
@@ -383,19 +331,10 @@ class MapLineContainer(GenericDiagramWidget):
                                                   api_object=new_api_object,
                                                   lat=new_api_object.lat,
                                                   lon=new_api_object.long,
-                                                  index=index)
-
-            idx = 0
-
-            for nod in self.nodes_list:
-
-                if idx >= index:
-                    nod.index = nod.index + 1
-
-                idx = idx + 1
+                                                  index=0)
 
             # Add the node to the nodes list
-            self.nodes_list.insert(index, graphic_obj)
+            self.nodes_list.insert(0, graphic_obj)
 
             graphic_obj.updatePosition()
 
@@ -405,7 +344,7 @@ class MapLineContainer(GenericDiagramWidget):
             # Return the newly created node
             return graphic_obj
 
-        elif 0 == index or index == len(self.api_object.locations.data):
+        elif (0 == index or index >= len(self.api_object.locations.data) - 1):
 
             substation_from_graphics = self.editor.graphics_manager.query(elm=self.api_object.get_substation_from())
             substation_to_graphics = self.editor.graphics_manager.query(elm=self.api_object.get_substation_to())
@@ -417,7 +356,7 @@ class MapLineContainer(GenericDiagramWidget):
                 nd2 = self.nodes_list[0]
 
             if index >= len(self.nodes_list):
-                nd2 = self.nodes_list[len(self.nodes_list) - 1]
+                nd1 = self.nodes_list[len(self.nodes_list) - 1]
 
             new_lat = ((nd2.lat + nd1.lat) / 2)
             new_long = ((nd2.lon + nd1.lon) / 2)
@@ -475,11 +414,11 @@ class MapLineContainer(GenericDiagramWidget):
         if 0 < index < len(self.api_object.locations.data) and len(self.api_object.locations.data) > 3:
 
             ln1 = Line()
-            ln1.copyData(self.api_object)
+            ln1.set_data_from(self.api_object)
             # ln1 = self.api_object.copy()
 
             ln2 = Line()
-            ln2.copyData(self.api_object)
+            ln2.set_data_from(self.api_object)
 
             first_list = self.api_object.locations.data[:index]
             second_list = self.api_object.locations.data[index:]
@@ -498,6 +437,9 @@ class MapLineContainer(GenericDiagramWidget):
                 api_obj.long = self.nodes_list[idx].lon
                 idx = idx + 1
 
+            ln1.bus_from = self.api_object.bus_from
+            ln2.bus_to = self.api_object.bus_to
+
             l1 = self.editor.add_api_line(ln1, original=False)
             l2 = self.editor.add_api_line(ln2, original=False)
 
@@ -506,10 +448,7 @@ class MapLineContainer(GenericDiagramWidget):
             return first_list, second_list
         else:
             # Handle invalid index
-            raise ValueError("Index out of range or invalid")
-
-    def merge_line(self):
-        return 0
+            error_msg("Index out of range or invalid", "split line")
 
     def disable_line(self):
         """

{GridCal-5.1.19 → GridCal-5.1.21}/GridCal/Gui/Diagrams/MapWidget/Branches/map_line_segment.py

@@ -324,8 +324,21 @@ class MapLineSegment(QGraphicsLineItem):
 
         if len(self.container.nodes_list) == 0:
             self.container.insert_new_node_at_position(0)
+        elif len(self.container.nodes_list) == 1:
+            if self.second == self.container.substation_to() or self.first == self.container.substation_to():
+                self.container.insert_new_node_at_position(1)
+            else:
+                self.container.insert_new_node_at_position(0)
         else:
-            self.container.insert_new_node_at_position(self.second.index)
+            if self.second == self.container.substation_to() or self.first == self.container.substation_to():
+                self.container.insert_new_node_at_position(len(self.container.nodes_list))
+            elif self.second == self.container.substation_from() or self.first == self.container.substation_from():
+                self.container.insert_new_node_at_position(0)
+            else:
+                if self.first.index > self.second.index:
+                    self.container.insert_new_node_at_position(self.second.index)
+                elif self.first.index < self.second.index:
+                    self.container.insert_new_node_at_position(self.second.index)
 
         # TODO implement
         # self.editor.split_line(line_graphics=self)

{GridCal-5.1.19 → GridCal-5.1.21}/GridCal/Gui/Diagrams/MapWidget/Substation/node_graphic_item.py

@@ -79,7 +79,7 @@ class NodeGraphicItem(QtWidgets.QGraphicsRectItem, NodeTemplate):
         # self.api_object: LineLocation = api_object
         self.index = index
 
-        self.resize(r)
+        self.resize(r * 10)
         self.setAcceptHoverEvents(True)  # Enable hover events for the item
         self.setFlag(QtWidgets.QGraphicsItem.GraphicsItemFlag.ItemIsMovable)  # Allow moving the node
         self.setFlag(QtWidgets.QGraphicsItem.GraphicsItemFlag.ItemIsSelectable)  # Allow selecting the node
@@ -88,7 +88,6 @@ class NodeGraphicItem(QtWidgets.QGraphicsRectItem, NodeTemplate):
 
         self.hovered = False
         self.enabled = True
-        self.itemSelected = False
 
         # Create a pen with reduced line width
         self.change_pen_width(0.001)
@@ -173,13 +172,13 @@ class NodeGraphicItem(QtWidgets.QGraphicsRectItem, NodeTemplate):
         Event handler for mouse press events.
         """
         super().mousePressEvent(event)
-
+        selected_items = self.editor.map.view._scene.selectedItems()
+        if len(selected_items) < 2:
+            self.setSelected(True)
         if self.enabled:
             self.editor.map.view.disableMove = True
             if event.button() == Qt.RightButton:
                 pass
-            elif event.button() == Qt.LeftButton:
-                self.selectItem()
 
     def mouseReleaseEvent(self, event):
         """
@@ -190,11 +189,15 @@ class NodeGraphicItem(QtWidgets.QGraphicsRectItem, NodeTemplate):
         self.updateDiagram()
         self.editor.map.view._scene.update()
 
+        selected_items = self.editor.map.view._scene.selectedItems()
+        if len(selected_items) < 2:
+            self.setSelected(True)
+
     def hoverEnterEvent(self, event: QtWidgets.QGraphicsSceneHoverEvent) -> None:
         """
         Event handler for when the mouse enters the item.
         """
-        self.editor.inItem = True
+        self.editor.map.view.inItem = True
         self.hovered = True
         self.setNodeColor(QColor(Qt.red), QColor(Qt.red))
         QApplication.instance().setOverrideCursor(Qt.PointingHandCursor)
@@ -203,7 +206,7 @@ class NodeGraphicItem(QtWidgets.QGraphicsRectItem, NodeTemplate):
         """
         Event handler for when the mouse leaves the item.
         """
-        self.editor.inItem = False
+        self.editor.map.view.inItem = False
         self.hovered = False
         self.setDefaultColor()
         QApplication.instance().restoreOverrideCursor()
@@ -217,45 +220,22 @@ class NodeGraphicItem(QtWidgets.QGraphicsRectItem, NodeTemplate):
         menu = QMenu()
 
         add_menu_entry(menu=menu,
-                       text="Add",
+                       text="Merge",
                        icon_path="",
-                       function_ptr=self.AddFunction)
+                       function_ptr=self.MergeFunction)
 
         add_menu_entry(menu=menu,
                        text="Split",
-                       icon_path="",
+                       icon_path=":/Icons/icons/divide.svg",
                        function_ptr=self.SplitFunction)
 
         add_menu_entry(menu=menu,
-                       text="Merge",
-                       icon_path="",
-                       function_ptr=self.MergeFunction)
-
-        add_menu_entry(menu=menu,
-                       text="Remove",
-                       icon_path="",
+                       text="Delete",
+                       icon_path=":/Icons/icons/delete.svg",
                        function_ptr=self.RemoveFunction)
 
         menu.exec_(event.screenPos())
 
-    def selectItem(self):
-        """
-
-        :return:
-        """
-        if not self.itemSelected:
-            self.editor.map.view.selectedItems.append(self)
-            self.setNodeColor(QColor(Qt.yellow), QColor(Qt.yellow))
-        self.itemSelected = True
-
-    def deSelectItem(self):
-        """
-
-        :return:
-        """
-        self.itemSelected = False
-        self.setDefaultColor()
-
     def AddFunction(self):
         """
 
@@ -312,10 +292,7 @@ class NodeGraphicItem(QtWidgets.QGraphicsRectItem, NodeTemplate):
         :return:
         """
         # Example: color assignment
-        if self.itemSelected:
-            self.setNodeColor(QColor(Qt.yellow), QColor(Qt.yellow))
-        else:
-            self.setNodeColor(self.colorInner, self.colorBorder)
+        self.setNodeColor(self.colorInner, self.colorBorder)
 
     def getPos(self) -> QPointF:
         """

{GridCal-5.1.19 → GridCal-5.1.21}/GridCal/Gui/Diagrams/MapWidget/Substation/substation_graphic_item.py

@@ -25,6 +25,8 @@ from GridCal.Gui.Diagrams.MapWidget.Substation.node_template import NodeTemplate
 from GridCal.Gui.GuiFunctions import add_menu_entry
 from GridCal.Gui.messages import yes_no_question
 from GridCal.Gui.GeneralDialogues import InputNumberDialogue
+
+from GridCalEngine.Devices.Substation.bus import Bus
 from GridCalEngine.Devices import VoltageLevel
 from GridCalEngine.Devices.Substation.substation import Substation
 from GridCalEngine.enumerations import DeviceType
@@ -69,7 +71,7 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
                               lon=lon)
 
         self.line_container = None
-        self.editor: GridMapWidget = editor  # re assign for the types to be clear
+        self.editor: GridMapWidget = editor  # reassign for the types to be clear
         self.api_object: Substation = api_object
 
         self.lat = lat
@@ -101,16 +103,21 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
         # self.setZValue(1)
         self.voltage_level_graphics: List[VoltageLevelGraphicItem] = list()
 
-    def move_to(self, lat: float, lon: float):
+    def move_to(self, lat: float, lon: float) -> Tuple[float, float]:
         """
 
         :param lat:
         :param lon:
-        :return:
+        :return: x, y
         """
         x, y = self.editor.to_x_y(lat=lat, lon=lon)
         self.setRect(x, y, self.rect().width(), self.rect().height())
 
+        for vl in self.voltage_level_graphics:
+            vl.center_on_substation()
+
+        return x, y
+
     def register_voltage_level(self, vl: VoltageLevelGraphicItem):
         """
 
@@ -179,7 +186,6 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
             for vl_graphics in self.voltage_level_graphics:
                 vl_graphics.center_on_substation()
 
-            self.editor.update_connectors()
             self.updateDiagram()  # always update
 
     def mousePressEvent(self, event: QGraphicsSceneMouseEvent):
@@ -187,10 +193,13 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
         Event handler for mouse press events.
         """
         super().mousePressEvent(event)
+        selected_items = self.editor.map.view._scene.selectedItems()
+        if len(selected_items) < 2:
+            self.setSelected(True)
+
         event.setAccepted(True)
         self.editor.map.view.disableMove = True
 
-        print("SE mouse press")
         if self.api_object is not None:
             self.editor.set_editor_model(api_object=self.api_object,
                                          dictionary_of_lists={
@@ -202,6 +211,7 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
                                              DeviceType.ZoneDevice: self.editor.circuit.get_zones(),
                                          })
 
+
     def mouseReleaseEvent(self, event: QGraphicsSceneMouseEvent):
         """
         Event handler for mouse release events.
@@ -215,6 +225,7 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
         """
         Event handler for when the mouse enters the item.
         """
+        self.editor.map.view.inItem = True
         self.setNodeColor(QColor(Qt.red), QColor(Qt.red))
         self.hovered = True
         QApplication.instance().setOverrideCursor(Qt.PointingHandCursor)
@@ -223,6 +234,7 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
         """
         Event handler for when the mouse leaves the item.
         """
+        self.editor.map.view.inItem = False
         self.hovered = False
         self.setDefaultColor()
         QApplication.instance().restoreOverrideCursor()
@@ -239,6 +251,11 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
                        icon_path="",
                        function_ptr=self.add_voltage_level)
 
+        add_menu_entry(menu=menu,
+                       text="Create line",
+                       icon_path="",
+                       function_ptr=self.create_new_line)
+
         add_menu_entry(menu=menu,
                        text="Move to API coordinates",
                        icon_path="",
@@ -251,7 +268,7 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
 
         add_menu_entry(menu=menu,
                        text="ADD node",
-                       icon_path="",
+                       icon_path=":/Icons/icons/divide.svg",
                        function_ptr=self.add_function)
 
         add_menu_entry(menu=menu,
@@ -259,8 +276,16 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
                        icon_path="",
                        function_ptr=self.new_substation_diagram)
 
+
+
         menu.exec_(event.screenPos())
 
+    def create_new_line(self):
+        """
+        Create a new line in the map wizard
+        """
+        self.editor.createNewLineWizard()
+
     def add_function(self):
         """
 
@@ -277,9 +302,11 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
 
             for idtag, graphic_object in dev_dict.items():
                 substation_from_graphics = self.editor.graphics_manager.query(
-                    elm=graphic_object.api_object.get_substation_from())
+                    elm=graphic_object.api_object.get_substation_from()
+                )
                 substation_to_graphics = self.editor.graphics_manager.query(
-                    elm=graphic_object.api_object.get_substation_to())
+                    elm=graphic_object.api_object.get_substation_to()
+                )
                 lines_info.append((idtag, graphic_object, substation_from_graphics, substation_to_graphics))
 
             # Now, iterate over the collected information
@@ -311,8 +338,8 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
                              "Move substation graphics")
 
         if ok:
-            self.move_to(lat=self.api_object.latitude, lon=self.api_object.longitude)
-            self.editor.update_connectors()
+            x, y = self.move_to(lat=self.api_object.latitude, lon=self.api_object.longitude)  # this moves the vl too
+            self.set_callabacks(x, y)
 
     def new_substation_diagram(self):
         """
@@ -329,7 +356,7 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
         inpt = InputNumberDialogue(
             min_value=1.0,
             max_value=100000.0,
-            default_value=1.0,
+            default_value=self.editor.diagram.default_bus_voltage,
             title="Add voltage level",
             text="Voltage (kV)",
         )
@@ -342,7 +369,14 @@ class SubstationGraphicItem(QGraphicsRectItem, NodeTemplate):
                               Vnom=kv,
                               substation=self.api_object)
 
+            bus = Bus(name=f"Bus {self.api_object.name}",
+                      Vnom=kv,
+                      substation=self.api_object,
+                      voltage_level=vl)
+
             self.editor.circuit.add_voltage_level(vl)
+            self.editor.circuit.add_bus(obj=bus)
+
             self.editor.add_api_voltage_level(substation_graphics=self,
                                               api_object=vl)
 

{GridCal-5.1.19 → GridCal-5.1.21}/GridCal/Gui/Diagrams/MapWidget/Substation/voltage_level_graphic_item.py

@@ -19,10 +19,13 @@ from typing import TYPE_CHECKING
 from PySide6 import QtWidgets
 from PySide6.QtCore import Qt, QPointF
 from PySide6.QtGui import QBrush, QColor
-from PySide6.QtWidgets import QGraphicsSceneMouseEvent, QGraphicsEllipseItem
+from PySide6.QtWidgets import (QMenu, QGraphicsSceneContextMenuEvent, QGraphicsSceneMouseEvent, QGraphicsEllipseItem)
+
 from GridCal.Gui.Diagrams.generic_graphics import GenericDiagramWidget
+from GridCal.Gui.GuiFunctions import add_menu_entry
 
 from GridCalEngine.Devices.Substation.voltage_level import VoltageLevel
+from GridCalEngine.Devices.Substation.bus import Bus
 from GridCalEngine.enumerations import DeviceType
 
 if TYPE_CHECKING:  # Only imports the below statements during type checking
@@ -61,19 +64,21 @@ class VoltageLevelGraphicItem(GenericDiagramWidget, QGraphicsEllipseItem):
                                       api_object=api_object,
                                       editor=editor,
                                       draw_labels=draw_labels)
-        QGraphicsEllipseItem.__init__(self, parent_center.x(), parent_center.y(), r, r, parent)
+        QGraphicsEllipseItem.__init__(self, parent_center.x(), parent_center.y(), r * api_object.Vnom * 0.01,
+                                      r * api_object.Vnom * 0.01, parent)
 
         parent.register_voltage_level(vl=self)
 
         self.editor: GridMapWidget = editor  # to reinforce the type
         self.api_object: VoltageLevel = api_object  # to reinforce the type
 
-        self.radius = r
-        print(f"VL created at x:{parent_center.x()}, y:{parent_center.y()}")
+        self.radius = r * api_object.Vnom * 0.01
+        # print(f"VL created at x:{parent_center.x()}, y:{parent_center.y()}")
 
         self.setAcceptHoverEvents(True)  # Enable hover events for the item
         # self.setFlag(QtWidgets.QGraphicsItem.GraphicsItemFlag.ItemIsMovable)  # Allow moving the node
-        self.setFlag(QtWidgets.QGraphicsItem.GraphicsItemFlag.ItemIsSelectable)  # Allow selecting the node
+        self.setFlag(
+            self.GraphicsItemFlag.ItemIsSelectable | self.GraphicsItemFlag.ItemIsMovable)  # Allow selecting the node
 
         # Create a pen with reduced line width
         self.change_pen_width(0.5)
@@ -90,7 +95,7 @@ class VoltageLevelGraphicItem(GenericDiagramWidget, QGraphicsEllipseItem):
         self.needsUpdate = False
         self.setZValue(0)
 
-    def center_on_substation(self):
+    def center_on_substation(self) -> None:
         """
         Centers the graphic item on the substation
         """
@@ -99,6 +104,16 @@ class VoltageLevelGraphicItem(GenericDiagramWidget, QGraphicsEllipseItem):
         yc = parent_center.y() - self.rect().height() / 2
         self.setRect(xc, yc, self.rect().width(), self.rect().height())
 
+    def move_to_xy(self, x: float, y: float):
+        """
+
+        :param x:
+        :param y:
+        :return:
+        """
+        self.setRect(x, y, self.rect().width(), self.rect().height())
+        return x, y
+
     def updateDiagram(self) -> None:
         """
 
@@ -120,10 +135,9 @@ class VoltageLevelGraphicItem(GenericDiagramWidget, QGraphicsEllipseItem):
         """
         Event handler for mouse move events.
         """
-        super().mouseMoveEvent(event)
+        # super().mouseMoveEvent(event)
         if self.hovered:
-            # self.updatePosition()
-            self.editor.update_connectors()
+            self.parent.mouseMoveEvent(event)
 
     def mousePressEvent(self, event: QGraphicsSceneMouseEvent):
         """
@@ -160,6 +174,18 @@ class VoltageLevelGraphicItem(GenericDiagramWidget, QGraphicsEllipseItem):
         self.hovered = False
         self.setNodeColor(self.colorInner, self.colorBorder)
 
+    def contextMenuEvent(self, event: QGraphicsSceneContextMenuEvent):
+        """
+
+        :param event:
+        """
+        menu = QMenu()
+
+        add_menu_entry(menu=menu,
+                       text="Add bus",
+                       icon_path="",
+                       function_ptr=self.add_bus)
+
     def setNodeColor(self, inner_color: QColor = None, border_color: QColor = None) -> None:
         """
 
@@ -197,3 +223,14 @@ class VoltageLevelGraphicItem(GenericDiagramWidget, QGraphicsEllipseItem):
         pen = self.pen()
         pen.setWidth(width)
         self.setPen(pen)
+
+    def add_bus(self):
+        """
+        Add bus
+        """
+        bus = Bus(name=f"Bus {self.api_object.name}",
+                  Vnom=self.api_object.Vnom,
+                  substation=self.parent.api_object,
+                  voltage_level=self.api_object)
+
+        self.editor.circuit.add_bus(obj=bus)