From 3e7c1003377dfdf6dcf99e8685336da5ce0958cd Mon Sep 17 00:00:00 2001
From: Adam Rupe <adam.rupe@pnnl.gov>
Date: Tue, 17 Sep 2024 12:51:28 -0500
Subject: [PATCH] added convert_to_HCGM

---
 src/y0/hierarchical.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/y0/hierarchical.py b/src/y0/hierarchical.py
index 02ac891a..3cbad20f 100644
--- a/src/y0/hierarchical.py
+++ b/src/y0/hierarchical.py
@@ -125,6 +125,21 @@ def create_Qvar(HCM: pgv.AGraph, subunit_node: pgv.Node) -> Variable:
         Q_str = "Q_{" + subunit_node.lower() + "|" + parent_str + "}"
     return Variable(Q_str)
 
+def convert_to_HCGM(HCM: pgv.AGraph) -> pgv.AGraph:
+    """Convert an HCM to an HCGM with promoted Q variables"""
+    HCGM = copy_HCM(HCM)
+    subunits = get_subunits(HCM)
+    for s in subunits:
+        Q = create_Qvar(HCGM, s)
+        HCGM.add_node(Q, color="blue")
+        parent_set = set(parents(HCGM, s))
+        for unit_parent in (parent_set & get_units(HCGM)):
+            HCGM.delete_edge(unit_parent, s)
+            HCGM.add_edge(unit_parent, Q)
+        HCGM.add_edge(Q, s)
+    HCGM.subgraphs()[0].graph_attr['style'] = 'solid'
+    return HCGM
+
 
 def direct_unit_descendents(HCM: pgv.AGraph, subunit_node: pgv.Node) -> set[pgv.Node]:
     """Return the set of direct unit descendents of the given subunit variable in the HCM."""