cqlib-core 0.1.0-beta.7

// This code is part of Cqlib.
//
// (C) Copyright China Telecom Quantum Group 2026
//
// This code is licensed under the Apache License, Version 2.0. You may
// obtain a copy of this license in the LICENSE.txt file in the root directory
// of this source tree or at http://www.apache.org/licenses/LICENSE-2.0.
//
// Any modifications or derivative works of this code must retain this
// copyright notice, and modified files need to carry a notice indicating
// that they have been altered from the originals.

//! Unified Quantum Instruction Type
//!
//! This module defines the [`Instruction`] enum, which serves as the universal wrapper for all
//! operations that can be placed in a quantum circuit. It unifies:
//! - **Standard Gates**: Highly optimized, commonly used gates (e.g., H, CX).
//! - **Extended Gates**: Multi-controlled operations, custom unitaries, and circuit-backed gates.
//! - **Directives**: Non-unitary operations such as measurement, barrier, and reset.
//! - **Classical Control**: Structured dynamic-circuit control flow driven by typed classical
//!   expressions.
//!
//! By using `Instruction`, the circuit can store a heterogeneous list of operations in a single vector.

use crate::circuit::gate::circuit_gate::{CircuitGate, FrozenCircuit};
use crate::circuit::gate::directive::Directive;
use crate::circuit::gate::standard_gate::StandardGate;
use crate::circuit::gate::{ClassicalDataOp, MCGate, UnitaryGate, gate_matrix};
use crate::circuit::{ClassicalControlOp, ClassicalValue, Parameter};
use alloc::borrow::Cow;
use ndarray::Array2;
use num_complex::Complex64;
use smallvec::SmallVec;
use std::fmt;
use std::sync::Arc;

/// A unified representation of any operation in a quantum circuit.
///
/// This enum acts as a sum type for all possible instructions.
/// - Prefer [`Instruction::Standard`] for common gates to leverage simulator optimizations.
/// - Use [`Instruction::McGate`] for generalized controls over standard gates.
/// - Use [`Instruction::UnitaryGate`] or [`Instruction::CircuitGate`] for custom unitary behavior.
/// - Use [`Instruction::Directive`] for non-reversible actions such as measurement and reset.
/// - Use [`Instruction::ClassicalControl`] for expression-based dynamic control flow.
#[derive(Debug, Clone, PartialEq)]
pub enum Instruction {
    /// A standard, natively supported quantum gate (e.g., `H`, `CX`).
    Standard(StandardGate),
    /// An extended gate, such as a multi-controlled gate or a user-defined unitary.
    McGate(Box<MCGate>),
    /// User-defined unitary gate backed by a matrix or circuit definition.
    UnitaryGate(Box<UnitaryGate>),
    /// Reusable gate defined by a frozen circuit body.
    CircuitGate(Box<CircuitGate>),
    /// A non-unitary operation, such as `Measure`, `Barrier`, or `Reset`.
    Directive(Directive),

    /// Runtime classical data operation such as assignment or measurement
    /// into typed classical storage.
    ClassicalData(ClassicalDataOp),

    /// Structured classical control flow.
    ///
    /// This variant carries expression-based IR such as `if`, `while`, `for`,
    /// `switch`, `break`, and `continue`. It is a runtime control instruction,
    /// not a unitary gate: it has no fixed arity, cannot be represented as a
    /// matrix, has no static inverse, and cannot be promoted to a
    /// quantum-controlled instruction.
    ClassicalControl(ClassicalControlOp),
    /// I gate in QCIS, represented here as Delay, unit is 0.5ns
    Delay,
}

impl Instruction {
    /// Returns the stable concrete operation name or user-defined gate label.
    ///
    /// Unlike [`Self::instruction_type`], this distinguishes individual gates.
    /// It is independent of `Debug` and of presentation-only formatting.
    pub fn name(&self) -> String {
        match self {
            Self::Standard(gate) => gate.name().to_string(),
            Self::McGate(gate) => gate.name(),
            Self::UnitaryGate(gate) => gate.label().to_string(),
            Self::CircuitGate(gate) => gate.name().to_string(),
            Self::Directive(directive) => directive.name().to_string(),
            Self::ClassicalData(operation) => operation.name().to_string(),
            Self::ClassicalControl(operation) => operation.name().to_string(),
            Self::Delay => "delay".to_string(),
        }
    }

    /// Returns a stable category name for this instruction.
    pub const fn instruction_type(&self) -> &'static str {
        match self {
            Self::Standard(_) => "standard",
            Self::McGate(_) => "mcgate",
            Self::UnitaryGate(_) => "unitary",
            Self::CircuitGate(_) => "circuit",
            Self::Directive(_) => "directive",
            Self::ClassicalData(_) => "classical_data",
            Self::ClassicalControl(_) => "classical_control",
            Self::Delay => "delay",
        }
    }

    pub const fn is_standard(&self) -> bool {
        matches!(self, Self::Standard(_))
    }

    pub const fn is_mcgate(&self) -> bool {
        matches!(self, Self::McGate(_))
    }

    pub const fn is_unitary(&self) -> bool {
        matches!(self, Self::UnitaryGate(_))
    }

    pub const fn is_circuit_gate(&self) -> bool {
        matches!(self, Self::CircuitGate(_))
    }

    pub const fn is_directive(&self) -> bool {
        matches!(self, Self::Directive(_))
    }

    pub const fn is_classical_data(&self) -> bool {
        matches!(self, Self::ClassicalData(_))
    }

    pub const fn is_classical_control(&self) -> bool {
        matches!(self, Self::ClassicalControl(_))
    }

    pub const fn is_delay(&self) -> bool {
        matches!(self, Self::Delay)
    }

    /// Returns whether this instruction is a unitary quantum gate.
    pub const fn is_quantum_gate(&self) -> bool {
        match self {
            Self::Standard(_) | Self::McGate(_) | Self::UnitaryGate(_) | Self::CircuitGate(_) => {
                true
            }
            Self::Directive(_)
            | Self::ClassicalData(_)
            | Self::ClassicalControl(_)
            | Self::Delay => false,
        }
    }

    /// Returns the standard gate represented directly by this instruction.
    pub fn standard_gate(&self) -> Option<StandardGate> {
        match self {
            Self::Standard(gate) => Some(*gate),
            _ => None,
        }
    }

    /// Returns the multi-controlled gate when this is an `McGate` instruction.
    pub fn mc_gate(&self) -> Option<&MCGate> {
        match self {
            Self::McGate(gate) => Some(gate),
            _ => None,
        }
    }

    /// Returns the user-defined unitary gate when this is a `UnitaryGate` instruction.
    pub fn unitary_gate(&self) -> Option<&UnitaryGate> {
        match self {
            Self::UnitaryGate(gate) => Some(gate),
            _ => None,
        }
    }

    /// Returns the circuit-backed gate when this is a `CircuitGate` instruction.
    pub fn circuit_gate(&self) -> Option<&CircuitGate> {
        match self {
            Self::CircuitGate(gate) => Some(gate),
            _ => None,
        }
    }

    /// Returns the directive when this is a `Directive` instruction.
    pub fn directive(&self) -> Option<Directive> {
        match self {
            Self::Directive(directive) => Some(*directive),
            _ => None,
        }
    }

    /// Returns the classical data operation when this is a `ClassicalData` instruction.
    pub fn classical_data(&self) -> Option<&ClassicalDataOp> {
        match self {
            Self::ClassicalData(operation) => Some(operation),
            _ => None,
        }
    }

    /// Returns the classical control operation when this is a `ClassicalControl` instruction.
    pub fn classical_control(&self) -> Option<&ClassicalControlOp> {
        match self {
            Self::ClassicalControl(operation) => Some(operation),
            _ => None,
        }
    }

    /// Returns true when this instruction directly or recursively contains a measurement.
    pub fn has_measurement(&self) -> bool {
        match self {
            Self::Directive(Directive::Measure)
            | Self::ClassicalData(ClassicalDataOp::MeasureBit { .. })
            | Self::ClassicalData(ClassicalDataOp::MeasureBits { .. }) => true,
            Self::ClassicalControl(control) => control.has_measurement(),
            _ => false,
        }
    }

    /// Returns true when this instruction directly or recursively reads `value`.
    pub fn reads_value(&self, value: ClassicalValue) -> bool {
        match self {
            Self::ClassicalData(ClassicalDataOp::Store {
                value: expression, ..
            }) => expression.values().contains(&value),
            Self::ClassicalControl(control) => control.reads_value(value),
            _ => false,
        }
    }

    /// Breaks a standard gate into its canonical base gate plus explicit
    /// control count.
    ///
    /// This helper centralizes the standard-gate promotion table so
    /// instruction-form canonicalization and `control()` share exactly the
    /// same notion of which multi-controlled instructions deserve a dedicated
    /// `StandardGate` representation.
    fn decompose_standard_control(g: StandardGate) -> (StandardGate, usize) {
        match g {
            StandardGate::CX => (StandardGate::X, 1),
            StandardGate::CCX => (StandardGate::X, 2),
            StandardGate::CY => (StandardGate::Y, 1),
            StandardGate::CZ => (StandardGate::Z, 1),
            StandardGate::CRX => (StandardGate::RX, 1),
            StandardGate::CRY => (StandardGate::RY, 1),
            StandardGate::CRZ => (StandardGate::RZ, 1),
            _ => (g, 0),
        }
    }

    /// Attempts to recompose a base gate and explicit control count into the
    /// most specific `StandardGate` form supported by the current IR.
    fn compose_standard_control(base: StandardGate, total_controls: usize) -> Option<StandardGate> {
        match (base, total_controls) {
            (StandardGate::X, 1) => Some(StandardGate::CX),
            (StandardGate::X, 2) => Some(StandardGate::CCX),
            (StandardGate::Y, 1) => Some(StandardGate::CY),
            (StandardGate::Z, 1) => Some(StandardGate::CZ),
            (StandardGate::RX, 1) => Some(StandardGate::CRX),
            (StandardGate::RY, 1) => Some(StandardGate::CRY),
            (StandardGate::RZ, 1) => Some(StandardGate::CRZ),
            (gate, 0) => Some(gate),
            _ => None,
        }
    }

    /// Returns the canonical instruction-form representation for this
    /// instruction without changing its semantics.
    ///
    /// This only collapses `McGate` values back to dedicated `StandardGate`
    /// variants when such a representation already exists in the IR. It uses
    /// the same control-promotion table as [`Instruction::control`], preserving
    /// the total control count, target convention, and parameter arity of the
    /// base gate. Operation qubit order is not changed here; callers that
    /// rewrite an operation keep the original qubit list attached to the new
    /// instruction.
    ///
    /// It intentionally does not perform decomposition, target-aware lowering,
    /// or matrix-based equivalence detection.
    pub(crate) fn canonicalize_form(&self) -> Instruction {
        match self {
            Instruction::McGate(mc) => {
                let total_ctrls = mc.num_ctrl_qubits();
                let (base, _) = Self::decompose_standard_control(mc.base_gate().to_owned());

                if let Some(std) = Self::compose_standard_control(base, total_ctrls) {
                    Instruction::Standard(std)
                } else {
                    self.clone()
                }
            }
            _ => self.clone(),
        }
    }

    /// Returns the fixed `(qubit_count, parameter_count)` for gate-like instructions.
    ///
    /// This method describes arity that is intrinsic to the instruction itself.
    /// It returns `None` for instructions whose qubit count is variable or
    /// context-dependent, such as barriers and classical control operations.
    /// Callers that need the qubits used by a classical control body should use
    /// [`ClassicalControlOp::used_qubits`] instead.
    pub fn gate_arity(&self) -> Option<(usize, usize)> {
        match self {
            Instruction::Standard(gate) => Some((gate.num_qubits(), gate.num_params())),
            Instruction::McGate(gate) => Some((gate.num_qubits(), gate.num_params())),
            Instruction::UnitaryGate(gate) => {
                Some((gate.num_qubits() as usize, gate.num_params() as usize))
            }
            Instruction::CircuitGate(gate) => Some((gate.num_qubits(), gate.num_params())),
            Instruction::Directive(Directive::Measure | Directive::Reset) => Some((1, 0)),
            Instruction::ClassicalData(ClassicalDataOp::Store { .. }) => Some((0, 0)),
            Instruction::ClassicalData(ClassicalDataOp::MeasureBit { .. }) => Some((1, 0)),
            Instruction::Delay => Some((1, 1)),
            Instruction::Directive(Directive::Barrier)
            | Instruction::ClassicalData(ClassicalDataOp::MeasureBits { .. })
            | Instruction::ClassicalControl(_) => None,
        }
    }

    /// Computes the unitary matrix representation of the instruction.
    ///
    /// The first qubit argument is the most-significant bit of the local
    /// matrix for every supported gate, including circuit-backed UnitaryGate
    /// definitions. Full circuit matrices instead use an LSB-first basis.
    ///
    /// # Arguments
    ///
    /// * `params` - A slice of floating-point parameters (e.g., rotation angles).
    ///   CircuitGate values bind positionally to its declared signature.
    ///
    /// # Returns
    ///
    /// - `Some(Cow<Array2>)`: A numeric matrix for a supported gate instruction.
    /// - `None`: If parameters are invalid or unresolved, or the instruction has
    ///   no matrix (e.g., `Measure`, `Barrier`, `Reset`, or classical control).
    pub fn matrix(&self, params: &[f64]) -> Option<Cow<'_, Array2<Complex64>>> {
        match self {
            Instruction::Standard(g) => g.matrix(params).ok(),
            Instruction::McGate(g) => g.matrix(params).ok(),
            Instruction::UnitaryGate(g) => g.matrix_for_params_in_gate_order(params).ok(),
            Instruction::CircuitGate(g) => g.matrix_for_params(params).ok().map(Cow::Owned),
            Instruction::Directive(_) => None,
            Instruction::ClassicalData(_) => None,
            Instruction::Delay => None,
            Instruction::ClassicalControl(_) => None,
        }
    }

    /// Computes the inverse (Hermitian conjugate) of the instruction.
    ///
    /// # Arguments
    ///
    /// * `params` - The parameters of the instruction instance.
    ///
    /// # Returns
    ///
    /// - `Some(...)`: The inverse instruction and its transformed parameters.
    /// - `None`: If the instruction is non-invertible (e.g., `Measure` or classical control)
    ///   or if the inverse cannot be symbolically determined (e.g., some custom unitaries).
    pub fn inverse(&self, params: &[Parameter]) -> Option<(Instruction, SmallVec<[Parameter; 3]>)> {
        match self {
            Instruction::Standard(g) => {
                if let Some((gate, ps)) = g.inverse(params) {
                    Some((Self::Standard(gate), ps))
                } else {
                    None
                }
            }
            Instruction::McGate(g) => {
                if let Some((gate, ps)) = g.inverse(params) {
                    Some((Self::McGate(Box::new(gate)), ps))
                } else {
                    None
                }
            }
            Instruction::UnitaryGate(g) => {
                // Try to invert via circuit representation first
                if let Some(c) = g.circuit().as_ref() {
                    // Invert the internal circuit
                    if let Ok(c_inv) = c.circuit.inverse() {
                        // Create frozen circuit from inverted circuit
                        let frozen_inv = FrozenCircuit::new(c_inv);
                        // Create new UnitaryGate with inverted circuit
                        let u_inv = UnitaryGate::new(
                            format!("{}_dg", g.label()).as_str(),
                            g.num_qubits(),
                            g.num_params(),
                        )
                        .with_circuit(Arc::new(frozen_inv))
                        .ok()?;
                        return Some((Self::UnitaryGate(Box::new(u_inv)), SmallVec::new()));
                    }
                }
                None
            }
            Instruction::CircuitGate(circuit_gate) => {
                if let Ok(inv_gate) = circuit_gate.inverse() {
                    Some((
                        Instruction::CircuitGate(Box::new(inv_gate)),
                        params.iter().cloned().collect(),
                    ))
                } else {
                    None
                }
            }
            Instruction::Directive(d) => match d {
                Directive::Barrier => Some((Self::Directive(Directive::Barrier), SmallVec::new())),
                _ => None,
            },
            Instruction::ClassicalData(_) => None,
            Instruction::Delay => Some((Self::Delay, SmallVec::new())),
            Instruction::ClassicalControl(_) => None,
        }
    }

    /// Returns a new instruction that applies the current operation conditioned on extra control qubits.
    ///
    /// This method employs a **canonicalization strategy** to return the most optimized gate representation:
    ///
    /// 1. **Promotion**: If adding controls to a `StandardGate` results in another `StandardGate`
    ///    (e.g., $X \xrightarrow{+1} CX \xrightarrow{+1} CCX$), it returns the upgraded standard gate.
    /// 2. **Fallback**: If no standard equivalent exists (e.g., $C^3-X$), it returns
    ///    an [`Instruction::McGate`].
    /// 3. **Aggregation**: If the instruction is already an `MCGate`, the new controls are merged into it.
    ///
    /// # Arguments
    ///
    /// * `num_new_ctrls` - The number of additional control qubits to add.
    ///
    /// # Returns
    ///
    /// - `Some(Instruction)`: The new controlled instruction.
    /// - `None`: If the instruction cannot be controlled (e.g., `Barrier`, `Measure`,
    ///   or classical control).
    pub fn control(&self, num_new_ctrls: usize) -> Option<Instruction> {
        if num_new_ctrls == 0 {
            return Some(self.clone());
        }

        match self {
            Instruction::Standard(g) => {
                let (base, curr_ctrls) = Self::decompose_standard_control(*g);
                let total_ctrls = curr_ctrls + num_new_ctrls;

                if let Some(std) = Self::compose_standard_control(base, total_ctrls) {
                    Some(Instruction::Standard(std))
                } else {
                    Some(Instruction::McGate(Box::from(MCGate::new(
                        total_ctrls as u8,
                        base,
                    ))))
                }
            }
            Instruction::McGate(mc) => {
                let total_ctrls = mc.num_ctrl_qubits() + num_new_ctrls;
                let (base, _) = Self::decompose_standard_control(mc.base_gate().to_owned());

                if let Some(std) = Self::compose_standard_control(base, total_ctrls) {
                    Some(Instruction::Standard(std))
                } else {
                    Some(Instruction::McGate(Box::from(MCGate::new(
                        total_ctrls as u8,
                        base,
                    ))))
                }
            }
            Instruction::UnitaryGate(uni) => {
                let mut g = UnitaryGate::new(
                    format!("ctl_{}_{}", num_new_ctrls, uni.label()).as_str(),
                    uni.num_qubits() + num_new_ctrls as u16,
                    uni.num_params(),
                );
                if let Some(m) = uni.matrix() {
                    let controlled = gate_matrix::control_matrix(m, num_new_ctrls);
                    // Handle possible matrix dimension error, return None on failure
                    g = match g.with_matrix(controlled) {
                        Ok(gate) => gate,
                        Err(_) => return None,
                    };
                } else if let Some(m) = uni.symbolic_matrix() {
                    let controlled =
                        crate::circuit::symbolic_matrix::gate::control_matrix(m, num_new_ctrls);
                    let params = uni.matrix_params().unwrap_or(&[]);
                    g = match g.with_symbolic_matrix(params.iter().cloned(), controlled) {
                        Ok(gate) => gate,
                        Err(_) => return None,
                    };
                }
                // Copy circuit field if present
                if let Some(c) = uni.circuit() {
                    g = match g.with_circuit(c.clone()) {
                        Ok(gate) => gate,
                        Err(_) => return None,
                    };
                }

                Some(Instruction::UnitaryGate(Box::from(g)))
            }
            Instruction::CircuitGate(_) => None, // CircuitGate does not support control yet
            Instruction::Directive(_) => None,
            Instruction::ClassicalData(_) => None,
            Instruction::Delay => None,
            Instruction::ClassicalControl(_) => None,
        }
    }
}

impl fmt::Display for Instruction {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.write_str(&self.name())
    }
}

impl From<StandardGate> for Instruction {
    fn from(g: StandardGate) -> Self {
        Self::Standard(g)
    }
}

impl From<Directive> for Instruction {
    fn from(d: Directive) -> Self {
        Self::Directive(d)
    }
}

impl From<ClassicalControlOp> for Instruction {
    fn from(cf: ClassicalControlOp) -> Self {
        Self::ClassicalControl(cf)
    }
}

impl From<ClassicalDataOp> for Instruction {
    fn from(op: ClassicalDataOp) -> Self {
        Self::ClassicalData(op)
    }
}

#[cfg(test)]
#[path = "instruction_test.rs"]
mod instruction_test;