Replace *pygraphviz* with *pydot* to simplify installation.

References colour-science/colour#1298.

Signed-off-by: Thomas Mansencal <[email protected]>

Author: KelSolaar

README.rst

@@ -87,10 +87,9 @@ as follows::
     poetry install --with optional
 
 The *aces-dev* *CTL* reference graph can be plotted but it requires `Graphviz <https://graphviz.org>`__
-to be installed on the system and having installed the optional `pygraphviz <https://pypi.org/project/pygraphviz>`__:
-python package::
+to be installed on the system::
 
-    poetry install --with graphviz,optional
+    poetry install --with optional
 
 Docker
 ~~~~~~
@@ -134,7 +133,7 @@ Optional Dependencies
 -   `graphviz <https://www.graphviz.org>`__
 -   `jsonpickle <https://jsonpickle.github.io>`__
 -   `networkx <https://pypi.org/project/networkx>`__
--   `pygraphviz <https://pypi.org/project/pygraphviz>`__
+-   `pydot <https://pypi.org/project/pydot>`__
 
 Development Dependencies
 ************************

docs/installation.rst

@@ -35,10 +35,9 @@ as follows::
     poetry install --with optional
 
 The *aces-dev* *CTL* reference graph can be plotted but it requires `Graphviz <https://graphviz.org>`__
-to be installed on the system and having installed the optional `pygraphviz <https://pypi.org/project/pygraphviz>`__:
-python package::
+to be installed on the system::
 
-    poetry install --with graphviz,optional
+    poetry install --with optional
 
 Docker
 ------
@@ -82,7 +81,7 @@ Optional Dependencies
 -   `graphviz <https://www.graphviz.org>`__
 -   `jsonpickle <https://jsonpickle.github.io>`__
 -   `networkx <https://pypi.org/project/networkx>`__
--   `pygraphviz <https://pypi.org/project/pygraphviz>`__
+-   `pydot <https://pypi.org/project/pydot>`__
 
 Development Dependencies
 ************************

opencolorio_config_aces/config/reference/discover/graph.py

@@ -17,6 +17,7 @@
 
 import codecs
 import logging
+import os
 import pickle
 
 from opencolorio_config_aces.config.reference.discover.classify import (
@@ -299,11 +300,11 @@ def conversion_path(graph, source, target):
 
 
 @required("NetworkX")
-def plot_aces_conversion_graph(graph, filename, prog="dot", args=""):
+def plot_aces_conversion_graph(graph, filename, prog="dot"):
     """
     Plot given *aces-dev* conversion graph using
     `Graphviz <https://www.graphviz.org/>`__ and
-    `pyraphviz <https://pygraphviz.github.io>`__.
+    `Pydot <https://pypi.org/project/pydot/>`__.
 
     Parameters
     ----------
@@ -314,72 +315,96 @@ def plot_aces_conversion_graph(graph, filename, prog="dot", args=""):
     prog : unicode, optional
         {'neato', 'dot', 'twopi', 'circo', 'fdp', 'nop'},
         *Graphviz* layout method.
-    args : unicode, optional
-         Additional arguments for *Graphviz*.
 
     Returns
     -------
-    AGraph
-        *PyGraphviz* graph.
+    Dot
+        *Pydot* graph.
     """
 
     import networkx as nx
+    from pydot import Subgraph
 
-    agraph = nx.nx_agraph.to_agraph(graph)
-
-    agraph.node_attr.update(
-        style="filled", shape="circle", fontname="Helvetica", fontsize=20
-    )
+    dot = nx.drawing.nx_pydot.to_pydot(graph)
 
     ctl_transforms_csc = []
     ctl_transforms_idt = []
     ctl_transforms_odt = []
     ctl_transforms_output_transform = []
     ctl_transforms_lmt = []
 
-    for node in agraph.nodes():
+    for node in dot.get_nodes():
+        node.set_style("filled")
+        node.set_shape("circle")
+        node.set_fontname("Helvetica")
+        node.set_fontsize(20)
+
         unserialized = pickle.loads(  # noqa: S301
-            codecs.decode(node.attr["serialized"].encode(), "base64")
+            codecs.decode(node.get_attributes()["serialized"].encode(), "base64")
         )
 
         ctl_transform_type = unserialized.type
         if node in ("ACES2065-1", "OCES"):
-            node.attr.update(
-                shape="doublecircle",
-                color="#673AB7FF",
-                fillcolor="#673AB770",
-                fontsize=30,
-            )
+            node.set_shape("doublecircle")
+            node.set_color("#673AB7FF")
+            node.set_fillcolor("#673AB7FF")
+            node.set_fontsize(30)
         elif ctl_transform_type == "ACEScsc":
-            node.attr.update(color="#00BCD4FF", fillcolor="#00BCD470")
+            node.set_color("#00BCD4FF")
+            node.set_fillcolor("#00BCD470")
             ctl_transforms_csc.append(node)
         elif ctl_transform_type == "IDT":
-            node.attr.update(color="#B3BC6D", fillcolor="#E6EE9C")
+            node.set_color("#B3BC6D")
+            node.set_fillcolor("#E6EE9C")
             ctl_transforms_idt.append(node)
         elif ctl_transform_type in ("ODT", "InvODT"):
-            node.attr.update(color="#CA9B52", fillcolor="#FFCC80")
+            node.set_color("#CA9B52")
+            node.set_fillcolor("#FFCC80")
             ctl_transforms_odt.append(node)
         elif ctl_transform_type in ("RRTODT", "InvRRTODT"):
-            node.attr.update(color="#C88719", fillcolor="#FFB74D")
+            node.set_color("#C88719")
+            node.set_fillcolor("#FFB74D")
             ctl_transforms_output_transform.append(node)
         elif ctl_transform_type == "LMT":
-            node.attr.update(color="#4BA3C7", fillcolor="#81D4FA")
+            node.set_color("#4BA3C7")
+            node.set_fillcolor("#81D4FA")
             ctl_transforms_lmt.append(node)
 
-    agraph.add_subgraph(ctl_transforms_csc, name="cluster_ACEScsc", color="#00BCD4FF")
-    agraph.add_subgraph(ctl_transforms_idt, name="cluster_IDT", color="#B3BC6D")
-    agraph.add_subgraph(ctl_transforms_odt, name="cluster_ODT", color="#CA9B52")
-    agraph.add_subgraph(
-        ctl_transforms_output_transform,
-        name="cluster_OutputTransform",
-        color="#C88719",
+    ctl_transforms_csc_subgraph = Subgraph(name="cluster_ACEScsc", color="#00BCD4FF")
+    for node in ctl_transforms_csc:
+        ctl_transforms_csc_subgraph.add_node(node)
+    dot.add_subgraph(ctl_transforms_csc_subgraph)
+
+    ctl_transforms_idt_subgraph = Subgraph(name="cluster_IDT", color="#B3BC6D")
+    for node in ctl_transforms_idt:
+        ctl_transforms_idt_subgraph.add_node(node)
+    dot.add_subgraph(ctl_transforms_idt_subgraph)
+
+    ctl_transforms_odt_subgraph = Subgraph(name="cluster_ODT", color="#CA9B52")
+    for node in ctl_transforms_odt:
+        ctl_transforms_odt_subgraph.add_node(node)
+    dot.add_subgraph(ctl_transforms_odt_subgraph)
+
+    ctl_transforms_output_transform_subgraph = Subgraph(
+        name="cluster_OutputTransform", color="#C88719"
     )
-    agraph.add_subgraph(ctl_transforms_lmt, name="cluster_LMT", color="#4BA3C7")
+    for node in ctl_transforms_output_transform:
+        ctl_transforms_output_transform_subgraph.add_node(node)
+    dot.add_subgraph(ctl_transforms_output_transform_subgraph)
+
+    ctl_transforms_lmt_subgraph = Subgraph(name="cluster_LMT", color="#4BA3C7")
+    for node in ctl_transforms_lmt:
+        ctl_transforms_lmt_subgraph.add_node(node)
+    dot.add_subgraph(ctl_transforms_lmt_subgraph)
+
+    for edge in dot.get_edges():
+        edge.set_color("#26323870")
 
-    agraph.edge_attr.update(color="#26323870")
-    agraph.draw(filename, prog=prog, args=args)
+    file_format = os.path.splitext(filename)[-1][1:]
+    write_method = getattr(dot, f"write_{file_format}")
+    write_method(filename, prog=prog, f=file_format)
 
-    return agraph
+    return dot
 
 
 if __name__ == "__main__":

pyproject.toml

@@ -32,9 +32,7 @@ semver = ">= 3, < 4"
 colour-science = ">= 0.4.3"
 jsonpickle = ">= 2, < 3"
 networkx = "*"
-
-[tool.poetry.group.graphviz.dependencies]
-pygraphviz = "*"
+pydot = ">= 3, 4"
 
 [tool.poetry.group.dev.dependencies]
 coverage = "*"