"""This module implements the StairLayerGenerator class."""
from __future__ import annotations
import logging
from bqskit.compiler.passdata import PassData
from bqskit.ir.circuit import Circuit
from bqskit.ir.gate import Gate
from bqskit.passes.search.generator import LayerGenerator
from bqskit.qis.state.state import StateVector
from bqskit.qis.state.system import StateSystem
from bqskit.qis.unitary.unitarymatrix import UnitaryMatrix
_logger = logging.getLogger(__name__)
[docs]
class StairLayerGenerator(LayerGenerator):
"""Layer Generator for search that builds circuits from a single gate."""
[docs]
def __init__(self, gate: Gate) -> None:
"""
Construct a StairLayerGenerator.
Args:
gate (Gate): The gate to build from.
"""
if not isinstance(gate, Gate):
raise TypeError(f'Expected Gate for gate, got {type(gate)}')
self.gate = gate
[docs]
def gen_initial_layer(
self,
target: UnitaryMatrix | StateVector | StateSystem,
data: PassData,
) -> Circuit:
"""
Generate the initial layer, see LayerGenerator for more.
Raises:
ValueError: If `target` has a size or radix mismatch with
`self.seed`.
"""
if not isinstance(target, (UnitaryMatrix, StateVector, StateSystem)):
raise TypeError(
'Expected unitary or state, got %s.' % type(target),
)
init_circuit = Circuit(target.num_qudits, target.radixes)
# Place RXX Gates on consective pairs of qudits
nq = self.gate.num_qudits
for i in range(init_circuit.num_qudits - (nq - 1)):
init_circuit.append_gate(self.gate, [i + j for j in range(nq)])
return init_circuit
[docs]
def gen_successors(self, circuit: Circuit, data: PassData) -> list[Circuit]:
"""
Generate the successors of a circuit node.
Raises:
ValueError: If circuit is a single-qudit circuit.
"""
if not isinstance(circuit, Circuit):
raise TypeError('Expected circuit, got %s.' % type(circuit))
nq = self.gate.num_qudits
successor = circuit.copy()
for i in range(circuit.num_qudits - (nq - 1)):
successor.append_gate(self.gate, [i + j for j in range(nq)])
return [successor]