"""Lowering a Labelled Graph State
==================================

This example starts from a labelled graph state and an explicit feedforward map,
then lowers them to an executable MBQC pattern.
"""

# %%
import sys
from pathlib import Path

from graphqomb.pattern import print_pattern
from graphqomb.ptn_format import dump
from graphqomb.qompiler import qompile
from graphqomb.random_objects import generate_random_flow_graph

# %%
# Generate a labelled graph state and its X-correction map.
graphstate, xflow = generate_random_flow_graph(width=3, depth=5)

print("graph nodes:", len(graphstate.nodes))
print("graph edges:", len(graphstate.edges))

# Lower the graph/feedforward IRs to an executable pattern.
pattern = qompile(graphstate, xflow)
print("pattern depth:", pattern.depth)
print("pattern max space:", pattern.max_space)
print_pattern(pattern)

# Dump the pattern in GraphQOMB's text format.
script_path = Path(globals().get("__file__", sys.argv[0])).resolve()
example_dir = script_path.parent
output_dir = example_dir / "generated"
output_dir.mkdir(exist_ok=True)
ptn_path = output_dir / "pattern_generation.ptn"
dump(pattern, ptn_path)
print("wrote pattern:", ptn_path)