QasmModule Circuit Drawer

docs/conf.py

@@ -15,7 +15,7 @@
 # -- Project information -----------------------------------------------------
 
 project = "qBraid"
-copyright = "2024, qBraid Development Team"
+copyright = "2025, qBraid Development Team"
 author = "qBraid Development Team"
 
 # The full version, including alpha/beta/rc tags
@@ -41,7 +41,7 @@
 # set_type_checking_flag = True
 autodoc_member_order = "bysource"
 autoclass_content = "both"
-autodoc_mock_imports = ["openqasm3"]
+autodoc_mock_imports = ["openqasm3", "matplotlib"]
 napoleon_numpy_docstring = False
 todo_include_todos = True
 mathjax_path = "https://cdn.jsdelivr.net/npm/mathjax@2/MathJax.js?config=TeX-AMS-MML_HTMLorMML"

pyproject.toml

@@ -32,6 +32,7 @@ classifiers = [
     "Operating System :: Unix",
     "Operating System :: MacOS",
 ]
+
 dependencies = ["numpy", "openqasm3[parser]>=1.0.0,<2.0.0"]
 
 [project.urls]
@@ -40,9 +41,10 @@ dependencies = ["numpy", "openqasm3[parser]>=1.0.0,<2.0.0"]
 
 [project.optional-dependencies]
 cli = ["typer>=0.12.1", "rich>=10.11.0", "typing-extensions"]
-test = ["pytest", "pytest-cov"]
+test = ["pytest", "pytest-cov", "pytest-mpl", "matplotlib"]
 lint = ["black", "isort>=6.0.0", "pylint", "mypy", "qbraid-cli>=0.10.2"]
 docs = ["sphinx>=7.3.7,<8.2.0", "sphinx-autodoc-typehints>=1.24,<3.1", "sphinx-rtd-theme>=2.0.0,<4.0.0", "docutils<0.22", "sphinx-copybutton"]
+visualization = ["matplotlib"]
 
 [tool.setuptools.package-data]
 pyqasm = ["py.typed", "*.pyx"]

src/pyqasm/__init__.py

@@ -23,6 +23,7 @@
    loads
    dump
    dumps
+   draw
 
 Classes
 ---------
@@ -58,6 +59,7 @@
 from .entrypoint import dump, dumps, load, loads
 from .exceptions import PyQasmError, QasmParsingError, ValidationError
 from .modules import Qasm2Module, Qasm3Module, QasmModule
+from .printer import draw
 
 __all__ = [
     "PyQasmError",
@@ -67,6 +69,7 @@
     "loads",
     "dump",
     "dumps",
+    "draw",
     "QasmModule",
     "Qasm2Module",
     "Qasm3Module",

src/pyqasm/maps/gates.py

@@ -8,6 +8,7 @@
 #
 # THERE IS NO WARRANTY for PyQASM, as per Section 15 of the GPL v3.
 
+
 # pylint: disable=too-many-lines
 
 """
@@ -1193,6 +1194,20 @@ def map_qasm_inv_op_to_callable(op_name: str):
     raise ValidationError(f"Unsupported / undeclared QASM operation: {op_name}")
 
 
+REV_CTRL_GATE_MAP = {
+    "cx": "x",
+    "cy": "y",
+    "cz": "z",
+    "crx": "rx",
+    "cry": "ry",
+    "crz": "rz",
+    "cp": "p",
+    "ch": "h",
+    "cu": "u",
+    "cswap": "swap",
+    "ccx": "cx",
+}
+
 CTRL_GATE_MAP = {
     "x": "cx",
     "y": "cy",