g.dZddlZGddeZdS)z$Class for probability distributions.NcReZdZdZejdZdfd ZddZdZ xZ S)ProbDistributionz8A generic dict-like class for probability distributions.z^[01]+$Nc||_d|_|rtt|}t |t rEttt|dz |_nt |tr| ds| drbd| D}ttt|dz |_nu|j |r=td|D|_d| D}ntdt!d t#|d S) aBuilds a probability distribution object. Args: data (dict): Input probability data. Where the keys represent a measured classical value and the value is a float for the probability 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. 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 ]/granary/edev/quantum/lib/python3.11/site-packages/qiskit/result/distributions/probability.py z-ProbDistribution.__init__..9&NNN:3CQKKNNNc34K|]}t|VdSN)len)r rs r z,ProbDistribution.__init__..?s((B(BcS(B(B(B(B(B(Brc6i|]\}}t|d|S)rr r s rrz-ProbDistribution.__init__..@rrzThe 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 _num_bitsnextiterkeys isinstancer rbinmaxstr startswithitems_bitstring_regexsearch ValueError TypeErrorsuper__init__)selfdatar first_key __class__s rr)zProbDistribution.__init__s&   ]T$))++..//I)S)) ]"%STYY[[)9)9%:%:!;!;a!?Is++ ]''--1E1Ed1K1KNNNNND&)S-=-=)>)>%?%?!%CDNN*11)<< %((B(BT(B(B(B%B%BDNNNNNNDD$6  [\\\ rcZ||jn|fd|DS)aBuild a 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)r rrns rrz9ProbDistribution.binary_probabilities..[s7PPPZS%sC  &&q))5PPPr)rr#)r*num_bitsr3s @rbinary_probabilitiesz%ProbDistribution.binary_probabilitiesKs7'.DNNHPPPP4::<<PPPPrc>d|DS)zBuild a probabilities dictionary with hexadecimal string keys Returns: dict: A dictionary where the keys are hexadecimal strings in the format ``"0x1a"`` c4i|]\}}t||S)hexr s rrz6ProbDistribution.hex_probabilities..ds$???JCC%???r)r#)r*s rhex_probabilitiesz"ProbDistribution.hex_probabilities]s!@?$**,,????rr) __name__ __module__ __qualname____doc__recompiler$r)r5r: __classcell__)r-s@rrrsBB!rz*--------^QQQQ$@@@@@@@rr)r>r?dictrr8rrrCs^+* M@M@M@M@M@tM@M@M@M@M@r