# Repository: https://gitlab.com/quantify-os/quantify-scheduler
# Licensed according to the LICENCE file on the main branch
"""Plotting functions used in the visualization backend of the sequencer."""
from __future__ import annotations
from copy import deepcopy
from typing import TYPE_CHECKING, Dict, List, Optional, Tuple, Union
from matplotlib.axes import Axes
from matplotlib.figure import Figure
from quantify_scheduler.operations.operation import Operation
import quantify_scheduler.schedules._visualization.pulse_scheme as ps
from quantify_scheduler.helpers.importers import import_python_object_from_string
from quantify_scheduler.schedules._visualization import constants
if TYPE_CHECKING:
from quantify_scheduler import Schedule
[docs]
def gate_box(ax: Axes, time: float, qubit_idxs: List[int], text: str, **kw):
"""
A box for a single gate containing a label.
Parameters
----------
ax :
The matplotlib Axes.
time :
The time of the gate.
qubit_idxs :
The qubit indices.
text :
The gate name.
kw :
Additional keyword arguments to be passed to drawing the gate box.
"""
for qubit_idx in qubit_idxs:
ps.box_text(
ax,
x0=time,
y0=qubit_idx,
text=text,
fillcolor=constants.COLOR_LAZURE,
width=0.8,
height=0.5,
**kw,
)
[docs]
def pulse_baseband(ax: Axes, time: float, qubit_idxs: List[int], text: str, **kw):
"""
Adds a visual indicator for a Baseband pulse to the `matplotlib.axes.Axis`
instance.
Parameters
----------
ax :
The matplotlib Axes.
time :
The time of the pulse.
qubit_idxs :
The qubit indices.
text :
The pulse name.
kw :
Additional keyword arguments to be passed to drawing the pulse.
"""
cartoon_width = 0.6
for qubit_idx in qubit_idxs:
ps.fluxPulse(
ax,
pos=time - cartoon_width / 2,
y_offs=qubit_idx,
width=cartoon_width,
s=0.0025,
amp=0.33,
**kw,
)
ax.text(time, qubit_idx + 0.45, text, ha="center", va="center", zorder=6)
[docs]
def pulse_modulated(ax: Axes, time: float, qubit_idxs: List[int], text: str, **kw):
"""
Adds a visual indicator for a Modulated pulse to the `matplotlib.axes.Axis`
instance.
Parameters
----------
ax :
The matplotlib Axes.
time :
The time of the pulse.
qubit_idxs :
The qubit indices.
text :
The pulse name.
kw :
Additional keyword arguments to be passed to drawing the pulse.
"""
cartoon_width = 0.6
for qubit_idx in qubit_idxs:
ps.mwPulse(
ax,
pos=time - cartoon_width / 2,
y_offs=qubit_idx,
width=cartoon_width,
amp=0.33,
**kw,
)
ax.text(time, qubit_idx + 0.45, text, ha="center", va="center", zorder=6)
[docs]
def meter(ax: Axes, time: float, qubit_idxs: List[int], text: str, **kw):
"""
A simple meter to depict a measurement.
Parameters
----------
ax :
The matplotlib Axes.
time :
The time of the measurement.
qubit_idxs :
The qubit indices.
text :
The measurement name.
kw :
Additional keyword arguments to be passed to drawing the meter.
"""
for qubit_idx in qubit_idxs:
ps.meter(
ax,
x0=time,
y0=qubit_idx,
fillcolor=constants.COLOR_GREY,
y_offs=0,
width=0.8,
height=0.5,
**kw,
)
[docs]
def acq_meter(ax: Axes, time: float, qubit_idxs: List[int], text: str, **kw):
"""
Variation of the meter to depict a acquisition.
Parameters
----------
ax :
The matplotlib Axes.
time :
The time of the measurement.
qubit_idxs :
The qubit indices.
text :
The measurement name.
kw :
Additional keyword arguments to be passed to drawing the acq meter.
"""
for qubit_idx in qubit_idxs:
ps.meter(
ax,
x0=time,
y0=qubit_idx,
fillcolor="white",
y_offs=0.0,
width=0.8,
height=0.5,
framewidth=constants.ACQ_METER_LINEWIDTH,
**kw,
)
[docs]
def acq_meter_text(ax: Axes, time: float, qubit_idxs: List[int], text: str, **kw):
"""
Same as acq_meter, but also displays text.
Parameters
----------
ax :
The matplotlib Axes.
time :
The time of the measurement.
qubit_idxs :
The qubit indices.
text :
The measurement name.
kw :
Additional keyword arguments to be passed to drawing the acq meter.
"""
acq_meter(ax, time, qubit_idxs, text, **kw)
ax.text(time, max(qubit_idxs) + 0.45, text, ha="center", va="center", zorder=6)
[docs]
def cnot(ax: Axes, time: float, qubit_idxs: List[int], text: str, **kw):
"""
Markers to denote a CNOT gate between two qubits.
Parameters
----------
ax :
The matplotlib Axes.
time :
The time of the CNOT.
qubit_idxs :
The qubit indices.
text :
The CNOT name.
kw :
Additional keyword arguments to be passed to drawing the CNOT.
"""
ax.plot(
[time, time], qubit_idxs, marker="o", markersize=15, color=constants.COLOR_BLUE
)
ax.plot([time], qubit_idxs[1], marker="+", markersize=12, color="white")
[docs]
def cz(ax: Axes, time: float, qubit_idxs: List[int], text: str, **kw):
"""
Markers to denote a CZ gate between two qubits.
Parameters
----------
ax :
The matplotlib Axes.
time :
The time of the CZ.
qubit_idxs :
The qubit indices.
text :
The CZ name.
kw :
Additional keyword arguments to be passed to drawing the CZ.
"""
ax.plot(
[time, time], qubit_idxs, marker="o", markersize=15, color=constants.COLOR_BLUE
)
[docs]
def reset(ax: Axes, time: float, qubit_idxs: List[int], text: str, **kw):
"""
A broken line to denote qubit initialization.
Parameters
----------
ax
matplotlib axis object.
time
x position to draw the reset on
qubit_idxs
indices of the qubits that the reset is performed on.
text :
The reset name.
kw :
Additional keyword arguments to be passed to drawing the reset.
"""
for qubit_idx in qubit_idxs:
ps.box_text(
ax,
x0=time,
y0=qubit_idx,
text=text,
color="white",
fillcolor="white",
width=0.4,
height=0.5,
**kw,
)
[docs]
def _locate_qubit_in_address(qubit_map, address):
"""Returns the name of a qubit in a pulse address."""
if address is None:
raise ValueError(f"Could not resolve address '{address}'")
for sub_addr in address.split(":"):
if sub_addr in qubit_map:
return sub_addr
raise ValueError(f"Could not resolve address '{address}'")
[docs]
def circuit_diagram_matplotlib(
schedule: Schedule,
figsize: Tuple[int, int] = None,
ax: Optional[Axes] = None,
) -> Tuple[Figure, Union[Axes, List[Axes]]]:
"""
Creates a circuit diagram visualization of a schedule using matplotlib.
Parameters
----------
schedule
the schedule to render.
figsize
matplotlib figsize.
ax
Axis handle to use for plotting.
Returns
-------
fig
matplotlib figure object.
ax
matplotlib axis object.
"""
# to prevent the original input schedule from being modified.
schedule = deepcopy(schedule)
# importing inside function scope to prevent circular import
from quantify_scheduler.compilation import _determine_absolute_timing
schedule = _determine_absolute_timing(schedule, "ideal")
qubit_map: Dict[str, int] = {}
qubits: List[str] = set()
for _, operation in schedule.operations.items():
if operation.valid_gate:
qubits.update(operation.data["gate_info"]["qubits"])
for index, qubit in enumerate(sorted(qubits)):
qubit_map[qubit] = index
# Validate pulses
# Note: needs to be done before creating figure and axhline
# in order to avoid unnecessary redraws.
for schedulable in schedule.schedulables.values():
operation = schedule.operations[schedulable["operation_id"]]
if not isinstance(operation, Operation):
# FIXME #461 We do not support schedule operations in pulse diagrams yet.
continue
if operation.valid_pulse:
try:
for pulse_info in operation["pulse_info"]:
_locate_qubit_in_address(qubit_map, pulse_info["port"])
except ValueError:
for key in qubit_map:
qubit_map[key] += 1
qubit_map["other"] = 0
break
if operation.valid_acquisition:
try:
for acq_info in operation["acquisition_info"]:
_locate_qubit_in_address(qubit_map, acq_info["port"])
except ValueError:
for key in qubit_map:
qubit_map[key] += 1
qubit_map["other"] = 0
break
if figsize is None:
figsize = (10, len(qubit_map))
fig, ax = ps.new_pulse_fig(figsize=figsize, ax=ax)
ax.set_title(schedule.data["name"])
ax.set_aspect("equal")
ax.set_ylim(-0.5, len(qubit_map) - 0.5)
ax.axhline(0, color="0.1", linewidth=0.9)
for qubit in qubits:
ax.axhline(qubit_map[qubit], color="0.1", linewidth=0.9)
# plot the qubit names on the y-axis
ax.set_yticks(list(qubit_map.values()))
ax.set_yticklabels(qubit_map.keys())
t0, tf, time = 0, 0, 0
for schedulable in sorted(
schedule.schedulables.values(), key=lambda sch: sch["abs_time"]
):
operation = schedule.operations[schedulable["operation_id"]]
if not isinstance(operation, Operation):
# FIXME #461 We do not support schedule operations in pulse diagrams yet.
continue
tf = schedulable["abs_time"]
time += 1 if tf != t0 else 0
t0 = tf
if operation.valid_gate:
plot_func = import_python_object_from_string(
operation["gate_info"]["plot_func"]
)
idxs = [qubit_map[qubit] for qubit in operation["gate_info"]["qubits"]]
plot_func(
ax, time=time, qubit_idxs=idxs, text=operation["gate_info"]["tex"]
)
elif operation.valid_pulse:
idxs: List[int]
try:
idxs = [
qubit_map[_locate_qubit_in_address(qubit_map, pulse_info["port"])]
for pulse_info in operation["pulse_info"]
]
except ValueError:
# The pulse port was not found in the qubit_map
# move this pulse to the 'other' timeline
idxs = [0]
for pulse_info in operation["pulse_info"]:
clock_id: str = pulse_info["clock"]
clock_resource: dict = schedule.data["resource_dict"][clock_id]
if clock_resource["freq"] == 0:
pulse_baseband(ax, time=time, qubit_idxs=idxs, text=operation.name)
else:
pulse_modulated(ax, time=time, qubit_idxs=idxs, text=operation.name)
elif operation.valid_acquisition:
idxs: List[int]
try:
idxs = [
qubit_map[_locate_qubit_in_address(qubit_map, acq_info["port"])]
for acq_info in operation["acquisition_info"]
]
except ValueError:
# The pulse port was not found in the qubit_map
# move this pulse to the 'other' timeline
idxs = [0]
for _ in operation["acquisition_info"]:
acq_meter(ax, time=time, qubit_idxs=idxs, text=operation.name)
else:
raise ValueError("Unknown operation")
ax.set_xlim(-1, time + 1)
return fig, ax
