g3dZddlmZddlmZddlZddlmZddl m Z m Z ddl m Z ddlmZd d lmZGd d e ZdS) z=Piecewise polynomial Chebyshev approximation to a given f(x).) annotations)CallableN) Chebyshev)QuantumRegisterAncillaRegister)BlueprintCircuit) CircuitError)!PiecewisePolynomialPauliRotationscleZdZdZ d&d'fd Zd(d)dZed*dZejd+dZed,dZ e jd-dZ ed.dZ e jd/dZ ed0dZ e jd1d!Z ed,d"Z e jd2d#Z d2d$Z fd%ZxZS)3PiecewiseChebyshevaPiecewise Chebyshev approximation to an input function. For a given function :math:`f(x)` and degree :math:`d`, this class implements a piecewise polynomial Chebyshev approximation on :math:`n` qubits to :math:`f(x)` on the given intervals. All the polynomials in the approximation are of degree :math:`d`. The values of the parameters are calculated according to [1] and see [2] for a more detailed explanation of the circuit construction and how it acts on the qubits. Examples: .. plot:: :alt: Circuit diagram output by the previous code. :include-source: import numpy as np from qiskit import QuantumCircuit from qiskit.circuit.library.arithmetic.piecewise_chebyshev import PiecewiseChebyshev f_x, degree, breakpoints, num_state_qubits = lambda x: np.arcsin(1 / x), 2, [2, 4], 2 pw_approximation = PiecewiseChebyshev(f_x, degree, breakpoints, num_state_qubits) pw_approximation._build() qc = QuantumCircuit(pw_approximation.num_qubits) qc.h(list(range(num_state_qubits))) qc.append(pw_approximation.to_instruction(), qc.qubits) qc.draw(output='mpl') References: [1]: Haener, T., Roetteler, M., & Svore, K. M. (2018). Optimizing Quantum Circuits for Arithmetic. `arXiv:1805.12445 `_ [2]: Carrera Vazquez, A., Hiptmair, H., & Woerner, S. (2022). Enhancing the Quantum Linear Systems Algorithm Using Richardson Extrapolation. `ACM Transactions on Quantum Computing 3, 1, Article 2 `_ Npw_chebf_xfloat | Callable[[int], float]degree int | None breakpointslist[int] | Nonenum_state_qubitsnamestrreturnNonect|d|_||_||nd|_||ndg|_d|_||_dS)a Args: f_x: the function to be approximated. Constant functions should be specified as f_x = constant. degree: the degree of the polynomials. Defaults to ``1``. breakpoints: the breakpoints to define the piecewise-linear function. Defaults to the full interval. num_state_qubits: number of qubits representing the state. name: The name of the circuit object. rNr r)super__init___num_state_qubits_f_x_degree _breakpoints _polynomialsr)selfrrrrr __class__s k/granary/edev/quantum/lib/python3.11/site-packages/qiskit/circuit/library/arithmetic/piecewise_chebyshev.pyrzPiecewiseChebyshev.__init__@sl& d###"& !'!3vv +6+BKK6: 0Traise_on_failureboolc>d}|jd}|rtd|jd}|rtd|jd}|rtd|jd}|rtd|j|jdzkrd}|rt d |jdzd |S) z,Check if the current configuration is valid.TNFz1The function to be approximated has not been set.z/The degree of the polynomials has not been set.z"The breakpoints have not been set.z&The number of qubits has not been set.r z0Not enough qubits in the circuit, need at least .)rAttributeErrorr r!r num_qubitsr )r#r'valids r%_check_configurationz'PiecewiseChebyshev._check_configurationas 9 E Z$%XYYY < E X$%VWWW   $E K$%IJJJ  (E O$%MNNN ?T2Q6 6 6E "4,q0444  r&c|jS)zhThe function to be approximated. Returns: The function to be approximated. )rr#s r%rzPiecewiseChebyshev.f_xs yr&%float | Callable[[int], float] | Nonec|j ||jkr7|||_||jdSdS)zSet the function to be approximated. Note that this may change the underlying quantum register, if the number of state qubits changes. Args: f_x: The new function to be approximated. N)r _invalidate_reset_registersr)r#rs r%rzPiecewiseChebyshev.f_xsV 9 ty 0 0      DI  ! !$"7 8 8 8 8 8 !1 0r&intc|jS)zdThe degree of the polynomials. Returns: The degree of the polynomials. )r r0s r%rzPiecewiseChebyshev.degrees |r&c|j ||jkr7|||_||jdSdS)zSet the error tolerance. Note that this may change the underlying quantum register, if the number of state qubits changes. Args: degree: The new degree. N)r r3r4r)r#rs r%rzPiecewiseChebyshev.degreesV < 6T\#9#9      !DL  ! !$"7 8 8 8 8 8 $:#9r& list[int]c~|j}|j.d|jz}|d|kr||gz}|ddkrdg|z}|S)zThe breakpoints for the piecewise approximation. Returns: The breakpoints for the piecewise approximation. Nr)r!r)r#r num_statess r%rzPiecewiseChebyshev.breakpointssd'   ,D11J2++)ZL8 1~!! cK/ r&c|j ||jkr<|||ndg|_||jdSdS)a Set the breakpoints for the piecewise approximation. Note that this may change the underlying quantum register, if the number of state qubits changes. Args: breakpoints: The new breakpoints for the piecewise approximation. Nr)r!r3r4r)r#rs r%rzPiecewiseChebyshev.breakpointsse   $ t7H(H(H      /:/F QCD   ! !$"7 8 8 8 8 8 )I(Hr&list[list[float]]c V|jggS|j}|dgkr dd|jzg}t|}g}td|dz D]} t |jt tfr||jgntj |j|j ||||dzg}d| tjjjz}||#t$$r3}t'd|jjjdzdzd z|d}~wwxYw|d d|jzkr|dtjdzggz}|ddkrdtjdzgg|z}|S) zThe polynomials for the piecewise approximation. Returns: The polynomials for the piecewise approximation. Raises: TypeError: If the input function is not in the correct format. Nrr:r )domain)kindz5 () missing 1 required positional argument: 'z'.z< Constant functions should be specified as 'f_x = constant'.r;)rr!lenrange isinstancerfloatr5appendr interpolaterconvertnp polynomial Polynomialcoeftolist ValueError TypeError__code__ co_varnamesarcsin)r#r num_intervals polynomialsipolyerrs r%rTzPiecewiseChebyshev.polynomialss  (4K ' 1#  a!667KK((  q-!+,,  A dh 55 6&&z2222$0$+{1~{STWXSXGY6ZDt||1I|JJOOD&&t{{}}555   Kh'3A67UU    r?Q 55 5 5%!bill*:);(<A   ! ,-.`Nr) _is_builtr_buildr rrrTrrFto_gatequbits)r#poly_rr$s r%rgzPiecewiseChebyshev._buildQsx >  F 2  !4#3T5EDI    FNN$$dk22222r&)NNNr) rrrrrrrrrrrr)T)r'r(rr()rr)rr1rr)rr5)rrrr)rr8)rrrr)rr>)rTrXrr)rrrr)__name__ __module__ __qualname____doc__rr.propertyrsetterrrrTrr4rg __classcell__)r$s@r%r r s""N"(,'+ 1111111B     DX Z 9 9 9Z 9X ] 9 9 9] 9X, 9 9 9 9777X7r 9 9 9 9&&&X&4444& . . . .333333333r&r )rn __future__rtypingrnumpyrInumpy.polynomial.chebyshevrqiskit.circuitrr'qiskit.circuit.library.blueprintcircuitrqiskit.circuit.exceptionsr $piecewise_polynomial_pauli_rotationsr r r&r%r{sDC""""""000000;;;;;;;;DDDDDD222222SSSSSSG3G3G3G3G3)G3G3G3G3G3r&