# Copyright 2025 qBraid
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""
Module containing PyTKET conversion extras.
"""
from __future__ import annotations
from typing import TYPE_CHECKING
from qbraid_core._import import LazyLoader
from qbraid.transpiler.annotations import requires_extras
from qbraid.transpiler.exceptions import ProgramConversionError
from ._utils import _validate_resolved_parameters
pytket_braket = LazyLoader("pytket_braket", globals(), "pytket.extensions.braket")
pytket_qiskit = LazyLoader("pytket_qiskit", globals(), "pytket.extensions.qiskit")
pytket_cirq = LazyLoader("pytket_cirq", globals(), "pytket.extensions.cirq")
pytket_qir = LazyLoader("pytket_qir", globals(), "pytket.qir")
pyqir = LazyLoader("pyqir", globals(), "pyqir")
if TYPE_CHECKING:
import braket.circuits
import cirq
import pytket.circuit
import qiskit
from pyqir import Module
[docs]
@requires_extras("pytket.extensions.braket")
def pytket_to_braket(circuit: pytket.circuit.Circuit) -> braket.circuits.Circuit:
"""Returns an Amazon Braket circuit equivalent to the input pytket circuit.
``tk_to_braket`` drops measurement instructions, so they are re-appended here
from the source circuit in classical-bit order.
Args:
circuit (pytket.circuit.Circuit): PyTKET circuit to convert to Braket circuit.
Returns:
braket.circuits.Circuit: Braket circuit equivalent to input pytket circuit.
Raises:
ProgramConversionError: If the circuit cannot be represented by an Amazon Braket
circuit.
"""
from pytket.circuit import OpType # pylint: disable=import-outside-toplevel
_validate_resolved_parameters(circuit, "Amazon Braket")
braket_circuit, _, measure_map = pytket_braket.braket_convert.tk_to_braket(circuit)
measured_qubits = set()
measured_bits = set()
for command in circuit.get_commands():
if command.op.type == OpType.Barrier:
continue
if measured_qubits.intersection(command.qubits):
# Covers a gate on a measured qubit and a qubit measured into a second bit;
# silently converting the latter would drop one of its classical bits.
raise ProgramConversionError(
"Braket circuits do not support mid-circuit measurement: a qubit is "
"acted on after it has been measured."
)
if command.op.type == OpType.Measure:
if measured_bits.intersection(command.bits):
# pytket's later measurement overwrites the bit; Braket gives every
# measurement its own result bit, so converting silently would widen
# the register and report both outcomes.
raise ProgramConversionError(
"Braket circuits cannot represent measuring two qubits into the "
"same classical bit."
)
measured_qubits.update(command.qubits)
measured_bits.update(command.bits)
# measure_map maps braket qubit id -> pytket bit index; re-add in classical-bit order.
for braket_qubit, _ in sorted(measure_map.items(), key=lambda item: item[1]):
braket_circuit.measure(braket_qubit)
return braket_circuit
[docs]
@requires_extras("pytket.extensions.qiskit")
def pytket_to_qiskit(circuit: pytket.circuit.Circuit) -> qiskit.QuantumCircuit:
"""Returns a Qiskit QuantumCircuit equivalent to the input pytket circuit.
Args:
circuit (pytket.circuit.Circuit): PyTKET circuit to convert to a Qiskit circuit.
Returns:
qiskit.QuantumCircuit: Qiskit circuit equivalent to input pytket circuit.
"""
return pytket_qiskit.tk_to_qiskit(circuit)
[docs]
@requires_extras("pytket.extensions.cirq")
def pytket_to_cirq(circuit: pytket.circuit.Circuit) -> cirq.Circuit:
"""Returns a Cirq circuit equivalent to the input pytket circuit.
Args:
circuit (pytket.circuit.Circuit): PyTKET circuit to convert to a Cirq circuit.
Returns:
cirq.Circuit: Cirq circuit equivalent to input pytket circuit.
"""
return pytket_cirq.tk_to_cirq(circuit)
[docs]
@requires_extras("pytket.qir")
def pytket_to_pyqir(circuit: pytket.circuit.Circuit) -> Module:
"""Returns a PyQIR module equivalent to the input pytket circuit.
Args:
circuit (pytket.circuit.Circuit): PyTKET circuit to convert to a PyQIR module.
Returns:
pyqir.Module: PyQIR module equivalent to input pytket circuit.
Raises:
ProgramConversionError: If the circuit contains unresolved parameters.
"""
_validate_resolved_parameters(circuit, "PyQIR")
llvm_ir = pytket_qir.pytket_to_qir(
circuit,
qir_format=pytket_qir.QIRFormat.STRING,
profile=pytket_qir.QIRProfile.BASE,
)
context = pyqir.Context()
return pyqir.Module.from_ir(context, llvm_ir, "Circuit")