gdZddlZddlZddlmZmZmZmZddlZ ddl m Z ddl m Z ddlmZddlmZejeZe d d d d efdZe d d d d de jde jdedeeeffdZe d d d d dZe d d d d dZe d d d d dededee jeffdZe d d d d  d de jdedeeede jfdZe d d d d  d!dedeeefdeeedeeede jf dZe d d d d  d!dedeeefdeeedeeedee jeff dZdS)"z( Readout mitigation data handling utils N)OptionalListTupleDict) QiskitError)deprecate_func)marginal_counts)Countsz1.3Qiskitz in Qiskit 2.0zzThe `qiskit.result.mitigation` module is deprecated in favor of the https://github.com/Qiskit/qiskit-addon-mthree package.)since package_nameremoval_timelineadditional_msgctj||}t|D]/}t|ddd||<0dtj|dzS)z8Return the diagonal for the operator :math:`Z^\otimes n`dtyper N1)npzerosrangebincountmod)dimrparityis T/granary/edev/quantum/lib/python3.11/site-packages/qiskit/result/mitigation/utils.py z_diagonalr skXc ' ' 'F 3ZZ**FF122J$$S))q 26&!$$ $$coeffsprobsshotsreturnc||}|dz|}||dzz |z }|dkr0tj|dstd||dkrt j|nd}||gS)aACompute expectation value and standard deviation. Args: coeffs: array of diagonal operator coefficients. probs: array of measurement probabilities. shots: total number of shots to obtain probabilities. Returns: tuple: (expval, stddev) expectation value and standard deviation. r rzqEncountered a negative variance in expectation value calculation.(%f). Setting standard deviation of result to 0.g)dotriscloseloggerwarningmathsqrt)r"r#r$expval sq_expvalvariance calc_stddevs rexpval_with_stddevr1,s"ZZ  F&&IFAI%.H!||BJx33| ?    *2A$)H%%%3K K  r!ci}|D]'\}}tj|d|z z|z }|||<(|S)zCalculate stddev dict)itemsr+r,)r#r$retkeyprobstd_errs rstddevr9NsQ C[[]] T)DAH-566C Jr!ctjddgttjddgttjddgttjddgtd}tjdgt}t|D]3}||vrt d|tj|||}4|S)z1Transform diagonal from a string to a numpy arrayr3rrr)IZ0rz"Invalid diagonal string character )rarrayfloatreversedrkron)stringcharsr5rs rstr2diagrD^sXq!fE * * * Xq"gU + + + Xq!fE * * * Xq!fE * * *   E (A3e $ $ $C f  %% E>>F1FFGG GgeAh$$ Jr!counts num_qubitsctjd|zt}d}|D]\}}||z }||t |d<||z}||fS)z)Transforms Counts to a probability vectorr rr)rrr?r4int)rErFvecr$r6vals rcounts_to_vectorrKusm (1j= . . .C ELLNNS  CQKK5LC :r!rIqubitsc2|t|krtdfdttj|D}tj|dgz||j}|S)zRemapping the qubitsNz(Num qubits does not match vector length.c g|] }dz |z  S)r3).0rrFs r z remap_qubits..s"IIIq Q"IIIr!r )lenrr@rargsortreshape transposeshape)rIrFrLaxess ` r remap_qubitsrXs v;;* $ $HII IIIIIHRZ5G5G,H,HIIIjjA3.//99$??GG RR Jr! qubit_indexclbitsc|R|t|nd}|t|nd}|d|fvrtd|d|dt||}| |fd|D}t||}|S)z[Marginalization of the Counts. Verify that number of clbits equals to the number of qubits.Nrz Num qubits (z#) does not match number of clbits (z).c g|] }| SrOrO)rPqubitrYs rrQz&marganalize_counts..s999+e$999r!)rRrr )rErYrLrZ qubits_len clbits_lens ` rmarganalize_countsr`s(.S[[[A (.S[[[A a_ , ,\z\\j\\\ !00 ~&,9999&999 00 Mr!ct||||}|t|}n!t|}t||\}}t |||}||fS)aCompute a probability vector for all count outcomes. Args: counts: counts object qubit_index: For each qubit, its index in the mitigator qubits list qubits: qubits the count bitstrings correspond to. clbits: Optional, marginalize counts to just these bits. Raises: QiskitError: if qubits and clbits kwargs are not valid. Returns: np.ndarray: a probability vector for all count outcomes. int: Number of shots in the counts )r`rRkeysrKrX)rErYrLrZrFrIr$s rcounts_probability_vectorrcsq8  VV D DF [[ ))++,, !&*55JC sJ / /C :r!)N)NN) __doc__loggingr+typingrrrrnumpyrqiskit.exceptionsrqiskit.utils.deprecationrutilsr rEr getLogger__name__r)r?r ndarrayrHr1r9rDrKrXr`rcrOr!rrns ............))))))333333######  8 $ $ $A  %%%% $A !rz!"*!S!USXZ_S_M`!!!!6 $A  $A      $A Vrz39O $A EI    !$ .6tCy.A Z     $A #'"&  c3h T#Y  T#Y   Z * $A #'"&  c3h T#Y  T#Y    2:s? r!