g2^dZddlmZmZddlZddlmZmZm Z ddl m Z Gdde Z dS) z)Compute the weighted sum of qubit states.)ListOptionalN)QuantumRegisterAncillaRegisterQuantumCircuit)BlueprintCircuitc peZdZdZ ddeedeeededdffd Ze defd Z e deefd Z e j deeddfd Z e defd Z e j deddfd Z dZe defdZe defdZddZfdZxZS) WeightedAdderu8A circuit to compute the weighted sum of qubit registers. Given :math:`n` qubit basis states :math:`q_0, \ldots, q_{n-1} \in \{0, 1\}` and non-negative integer weights :math:`\lambda_0, \ldots, \lambda_{n-1}`, this circuit performs the operation .. math:: |q_0 \ldots q_{n-1}\rangle |0\rangle_s \mapsto |q_0 \ldots q_{n-1}\rangle |\sum_{j=0}^{n-1} \lambda_j q_j\rangle_s where :math:`s` is the number of sum qubits required. This can be computed as .. math:: s = 1 + \left\lfloor \log_2\left( \sum_{j=0}^{n-1} \lambda_j \right) \right\rfloor or :math:`s = 1` if the sum of the weights is 0 (then the expression in the logarithm is invalid). For qubits in a circuit diagram, the first weight applies to the upper-most qubit. For an example where the state of 4 qubits is added into a sum register, the circuit can be schematically drawn as .. code-block:: text ┌────────┐ state_0: ┤0 ├ | state_0 * weights[0] │ │ | state_1: ┤1 ├ | + state_1 * weights[1] │ │ | state_2: ┤2 ├ | + state_2 * weights[2] │ │ | state_3: ┤3 ├ | + state_3 * weights[3] │ │ sum_0: ┤4 ├ | │ Adder │ | sum_1: ┤5 ├ | = sum_0 * 2^0 + sum_1 * 2^1 + sum_2 * 2^2 │ │ | sum_2: ┤6 ├ | │ │ carry_0: ┤7 ├ │ │ carry_1: ┤8 ├ │ │ control_0: ┤9 ├ └────────┘ Naddernum_state_qubitsweightsnamereturnct|d|_d|_||_||_dS)a:Computes the weighted sum controlled by state qubits. Args: num_state_qubits: The number of state qubits. weights: List of weights, one for each state qubit. If none are provided they default to 1 for every qubit. name: The name of the circuit. rN)super__init___weights_num_state_qubitsrr )selfr rr __class__s f/granary/edev/quantum/lib/python3.11/site-packages/qiskit/circuit/library/arithmetic/weighted_adder.pyrzWeightedAdder.__init__IsD d### !% 0c t|jdkrHttjtjt|jdzSdS)zThe number of sum qubits in the circuit. Returns: The number of qubits needed to represent the weighted sum of the qubits. r)sumrintnpfloorlog2rs rnum_sum_qubitszWeightedAdder.num_sum_qubits_sO t|  q rxDL(9(9 : :;;a?@@ @qrcF|jr|jS|jr dg|jzSdS)ziThe weights for the qubit states. Returns: The weight for the qubit states. rN)rr r"s rrzWeightedAdder.weightsjs6 = !=   /3.. .trc,|rbt|D]R\}}tj|tj|st dtj|||<S|||_|dS)zSet the weights for summing the qubit states. Args: weights: The new weights. Raises: ValueError: If not all weights are close to an integer. z&Non-integer weights are not supported!N) enumerateriscloseround ValueError _invalidater_reset_registers)rriweights rrzWeightedAdder.weightsws  .&w// . . 6z&"(6*:*:;;O$%MNNNXf--    rc|jS)zeThe number of qubits to be summed. Returns: The number of state qubits. )rr"s rr zWeightedAdder.num_state_qubitss %%rc|j ||jkr1|||_|dSdS)zuSet the number of state qubits. Args: num_state_qubits: The new number of state qubits. N)rr*r+)rr s rr zWeightedAdder.num_state_qubitssV  ! )-=AW-W-W      %5D "  ! ! # # # # #.X-Wrclg|_|jrt|jd}t|jd}||g|_|jdkr+t |jd}|||jdkr/t |jd}||dSdSdS)zReset the registers.staterrrcarrycontrolN)qregsr rr#num_carry_qubitsr add_registernum_control_qubits)rqr_stateqr_sumqr_carry qr_controls rr+zWeightedAdder._reset_registerss  .&t'<7KKKH$T%8uEEEF"F+DJ$q((*4+@wOOO!!(+++&**,T-D9UUU !!*----- . .+*rc|jdz S)a#The number of carry qubits required to compute the sum. Note that this is not necessarily equal to the number of ancilla qubits, these can be queried using ``num_ancilla_qubits``. Returns: The number of carry qubits required to compute the sum. r)r#r"s rr5zWeightedAdder.num_carry_qubitss"Q&&rc2t|jdkS)aThe number of additional control qubits required. Note that the total number of ancilla qubits can be obtained by calling the method ``num_ancilla_qubits``. Returns: The number of additional control qubits required (0 or 1). r)rr#r"s rr7z WeightedAdder.num_control_qubitss4&*+++rTcd}|jd}|rtd|jt|jkrd}|rt d|S)z,Check if the current configuration is valid.TNFz,The number of state qubits has not been set.z/Mismatching number of state qubits and weights.)rAttributeErrorlenrr))rraise_on_failurevalids r_check_configurationz"WeightedAdder._check_configurationsg  ! )E U$%STTT  !S%6%6 6 6E T !RSSS rc  |jrdSt|j|jz}t |j}|jd|j}|j|j|}|j|||jz}|j||jzd}t|j D]\}}tj |dr||} t|d|jdddd} t| D]q\} } | dkr|jdkr|| || 5| dkr@|| || || || || {| |jdz krC|| || || || dz || ||| ||| dz || || || dz g|| |d || || ||| ||| dz || || || || dz || |jdkr| dkr| |jdz kr(|| || dz || || || || dz g|| |d || || dz || st't)t+| D]} | | } | dkr|jdkr| dkrZ||| || || || ||| ~| |jdz kr||| dz || || || dz g|| |d || || ||| dz |jdkr%| dkr-| |jdz kr=||| || || || dz g|| |d ||| |||jdS) z(If not already built, build the circuit.Nrb01rzv-chain)mode) _is_builtr_buildr r#rr4qubitsr5r&rrr'rrjustcxccxxmcxreversedranger@appendto_gate)rnum_result_qubitscircuitr8r9r:r;r,r-q_state weight_binaryjbitrs rrKzWeightedAdder._builds >  F  1D4GG $*->"9D$9"9: 58I IJ>"36G$J_6_"_`^$58M$M$O$OP #4<00g -g -IAvz&!$$ qkG #6{{..44T5H#NNttQStTM$M222 I2 I3#::*a// 7F1I6666a GVAY DDD 7F1I6666d1A555 7F1I666 GXa!e_fQiHHHH &),,, (1q5/222 $fQi!a%A$QK&!* $  7HQK888 &),,, (1q5/222 7F1I666 GXa!e_fQiHHHH*a//ad1A555 GXa!e_fQiHHHH $fQi!a%A$QK&!* $  GXa!e_fQiHHHHeC $6$67788& -& -#A&#::*a//a &),,, GVAY DDD &),,,,d1A555 (1q5/222 $fQi!a%A$QK&!* $  7HQK888 (1q5/2222*a//ad1A555 &),,, $fQi!a%A$QK&!* $  &),,,,M& -P GOO%%t{33333r)NNr )T)__name__ __module__ __qualname____doc__rrrstrrpropertyr#rsetterr r+r5r7rCrK __classcell__)rs@rr r s//f+/'+ 11"3-1$s)$1 1  111111,X c   X  ^ tCy T   ^ &&#&&&X& $ $ $ $ $ $..." '# ' ' 'X ' ,C , , ,X ,    y4y4y4y4y4y4y4y4y4rr ) r_typingrrnumpyrqiskit.circuitrrrblueprintcircuitr r rrris0/!!!!!!!!KKKKKKKKKK//////z4z4z4z4z4$z4z4z4z4z4r