updates to hardware layout hierarchy stuff

thegridelectric · Nov 26, 2024 · 991785b · 991785b
1 parent b09d856
commit 991785b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/gwproto/data_classes/hardware_layout.py b/src/gwproto/data_classes/hardware_layout.py
@@ -653,6 +653,9 @@ def boss_node(self, node: ShNode) -> Optional[ShNode]:
             raise DcError(f"{node} is missing boss {boss_handle}")
         return boss
 
+    def direct_reports(self, node: ShNode) -> List[ShNode]:
+        return [n for n in self.nodes.values() if self.boss_node(n) == node]
+
     def node_from_handle(self, handle: str) -> Optional[ShNode]:
         return next((n for n in self.nodes.values() if n.handle == handle), None)