g(FdZddlmZddlZddlmZGddeZdS)zQuasidistribution class)sqrtN)ProbDistributionczeZdZdZejdZdZd fd Zd dZ ddZ d Z e d Z d ZxZS)QuasiDistributionz7A dict-like class for representing quasi-probabilities.z^[01]+$Nc||_||_d|_|rtt |}t |trEttt|dz |_nt |tr| ds| drbd| D}ttt|dz |_nu|j|r=td|D|_d| D}nt!dt#d t%|d S) afBuilds a quasiprobability distribution object. .. note:: The quasiprobability values might include floating-point errors. ``QuasiDistribution.__repr__`` rounds using :meth:`numpy.round` and the parameter ``ndigits`` can be manipulated with the class attribute ``__ndigits__``. The default is ``15``. Parameters: data (dict): Input quasiprobability data. Where the keys represent a measured classical value and the value is a float for the quasiprobability of that result. The keys can be one of several formats: * A hexadecimal string of the form ``"0x4a"`` * A bit string e.g. ``'0b1011'`` or ``"01011"`` * An integer shots (int): Number of shots the distribution was derived from. stddev_upper_bound (float): An upper bound for the standard deviation Raises: TypeError: If the input keys are not a string or int ValueError: If the string format of the keys is incorrect r0x0bc6i|]\}}t|d|S)rint.0keyvalues W/granary/edev/quantum/lib/python3.11/site-packages/qiskit/result/distributions/quasi.py z.QuasiDistribution.__init__..F&NNN:3CQKKNNNc34K|]}t|VdSN)len)rrs r z-QuasiDistribution.__init__..Ls((B(BcS(B(B(B(B(B(Brc6i|]\}}t|d|S)r rrs rrz.QuasiDistribution.__init__..MrrzThe input keys are not a valid string format, must either be a hex string prefixed by '0x' or a binary string optionally prefixed with 0bz9Input data's keys are of invalid type, must be str or intN)shots_stddev_upper_bound _num_bitsnextiterkeys isinstancerrbinmaxstr startswithitems_bitstring_regexsearch ValueError TypeErrorsuper__init__)selfdatarstddev_upper_bound first_key __class__s rr.zQuasiDistribution.__init__ s6 #5   ]T$))++..//I)S)) ]"%STYY[[)9)9%:%:!;!;a!?Is++ ]''--1E1Ed1K1KNNNNND&)S-=-=)>)>%?%?!%CDNN*11)<< %((B(BT(B(B(B%B%BDNNNNNNDD$6  [\\\ rFctt|d}t|}i}d}d}|D]E\}}|||z z} | dkr||z }|dz}|||zz }&|||z ||z zz }||||z z||<F|r$t ||jt |fSt ||jS)aTakes a quasiprobability distribution and maps it to the closest probability distribution as defined by the L2-norm. Parameters: return_distance (bool): Return the L2 distance between distributions. Returns: ProbDistribution: Nearest probability distribution. float: Euclidean (L2) distance of distributions. Notes: Method from Smolin et al., Phys. Rev. Lett. 108, 070502 (2012). c|dS)Nr)items rzDQuasiDistribution.nearest_probability_distribution..gs $q'r)rrr)dictsortedr(rrrr) r/return_distance sorted_probs num_elems new_probsbetadiffrvaltemps r nearest_probability_distributionz2QuasiDistribution.nearest_probability_distributionXs F4::<<5I5IJJJKK  %%  $**,, F FHC ))Daxx Q c ! )dY.>??!-c!2TI5E!E #  G#Itz::DJJF F 4:666rcZ||jn|fd|DS)aBuild a quasi-probabilities dictionary with binary string keys Parameters: num_bits (int): number of bits in the binary bitstrings (leading zeros will be padded). If None, a default value will be used. If keys are given as integers or strings with binary or hex prefix, the default value will be derived from the largest key present. If keys are given as bitstrings without prefix, the default value will be derived from the largest key length. Returns: dict: A dictionary where the keys are binary strings in the format ``"0110"`` Nc^i|])\}}t|d|*S)b)formatzfill)rrrns rrz:QuasiDistribution.binary_probabilities..s7PPPZS%sC  &&q))5PPPr)rr()r/num_bitsrIs @rbinary_probabilitiesz&QuasiDistribution.binary_probabilitiesys7'.DNNHPPPP4::<<PPPPrc>d|DS)zBuild a quasi-probabilities dictionary with hexadecimal string keys Returns: dict: A dictionary where the keys are hexadecimal strings in the format ``"0x1a"`` c4i|]\}}t||Sr6)hexrs rrz7QuasiDistribution.hex_probabilities..s$???JCC%???r)r(r/s rhex_probabilitiesz#QuasiDistribution.hex_probabilitiess!@?$**,,????rc|jS)z@Return an upper bound on standard deviation of expval estimator.)rrOs rr1z$QuasiDistribution.stddev_upper_bounds ''rc^tfdDS)NcDi|]\}}|t|jS))ndigits)round __ndigits__)rrrr/s rrz.QuasiDistribution.__repr__..s/___JCCud.>???___r)r&r(rOs`r__repr__zQuasiDistribution.__repr__s3____RVR\R\R^R^___```r)NN)Fr)__name__ __module__ __qualname____doc__recompiler)rVr.rCrKrPpropertyr1rW __classcell__)r3s@rrrsAA!rz*--K666666p7777BQQQQ$@@@((X(aaaaaaarr)r[mathrr\ probabilityrr9rr6rrrbs ))))))@a@a@a@a@a@a@a@a@a@ar