g0 dZddlmZmZmZmZmZddlmZddl m Z ddl Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZ d#deeefdeeedededeedeeeefeff dZ dZ! d$deeee j"fdeeedededededeeee j"ffdZ# d%dedeeededeeeffdZ$d&d Z%d!Z&d"Z'dS)'z+Utility functions for working with Results.)SequenceUnionOptionalDictList)Counter)deepcopyN) QiskitError)Result)Counts)ProbDistribution)QuasiDistribution) _bin_to_hex)resultsFTresultindicesinplaceformat_marginalmarginalize_memoryreturnct|trL|st|}t|jD]#\}}||}t ||}i} |D]\} } | | t| <| |j _ |Kt||j _ t|j dd} | t| ||j _t|j dd[|Y|durt#|j d|t%|d} t'j|j j| d|j _%|St ||}|r|t-|||}|S)aMarginalize counts from an experiment over some indices of interest. Args: result: result to be marginalized (a Result object or a dict(str, int) of counts). indices: The bit positions of interest to marginalize over. If ``None`` (default), do not marginalize at all. inplace: Default: False. Operates on the original Result argument if True, leading to loss of original Job Result. It has no effect if ``result`` is a dict. format_marginal: Default: False. If True, takes the output of marginalize and formats it with placeholders between cregs and for non-indices. marginalize_memory: If True, then also marginalize the memory field (if present). If False, remove the memory field from the result. If None, leave the memory field as is. Returns: Result or dict(str, int): A Result object or a dictionary with the observed counts, marginalized to only account for frequency of observations of bits of interest. Raises: QiskitError: in case of invalid indices to marginalize over. N creg_sizesmemoryFTreverse) return_hex) isinstancer r enumerater get_counts _marginalizeitemsrdatacountslenheader memory_slotsgetattr_adjust_creg_sizesrdelattrsorted results_rsmarginal_memoryr_format_marginal)rrrrriexperiment_resultr# new_countsnew_counts_hexkvcsizesorted_indices marg_countss I/granary/edev/quantum/lib/python3.11/site-packages/qiskit/result/utils.pymarginal_countsr8!s@&&!!" &f%%F$-fn$=$=   A &&q))F%fg66JN"((** 3 3112{1~~..,:  " )"8;G !(5 1 8,MM$:LUT[:\:\%,7(-x>>JwOb%..-2H===='/%+&&&N5?4N).5~RV555%*1 "6733  Iw2*6;HHKcpd|D}t|}d|DfdtdtdzDdd}d}|D]H}t|t|D](}|||kr|}||dxxdz cc<n)Id|D}|S)z,Helper to reduce creg_sizes to match indicesc"g|] }|ddg S)r.0cregs r7 z&_adjust_creg_sizes..js :::ttAwl:::r9cg|]\}}|Sr<r<)r>_xs r7r@z&_adjust_creg_sizes..os***tq!***r9cFg|]}td|ddz S)r)sum)r>rC creg_numss r7r@z&_adjust_creg_sizes..ps1SSS3y1Q'((1,SSSr9rrENc*g|]}|ddk|S)rErr<r=s r7r@z&_adjust_creg_sizes..|s!FFFtaAdr9)r*ranger$) rrnew_creg_sizes indices_sort creg_limitscreg_idxindidxrGs @r7r(r(fs;:z:::N'??L+*z***ISSSSeAs9~~PQ?Q6R6RSSSTUTVTVWKH3{#3#344  Ck#&&&s#A&&&!+&&&' GF~FFFN r9r int_return hex_returnavg_dataparallel_thresholdc |r|rtdt|tjr|rtd|rtd||S|jdkrt j||S|jdkr,|rt j||St j ||S|jdkrt j ||Stdt j ||||| S) aMarginalize shot memory This function is multithreaded and will launch a thread pool with threads equal to the number of CPUs by default. You can tune the number of threads with the ``RAYON_NUM_THREADS`` environment variable. For example, setting ``RAYON_NUM_THREADS=4`` would limit the thread pool to 4 threads. Args: memory: The input memory list, this is either a list of hexadecimal strings to be marginalized representing measure level 2 memory or a numpy array representing level 0 measurement memory (single or avg) or level 1 measurement memory (single or avg). indices: The bit positions of interest to marginalize over. If ``None`` (default), do not marginalize at all. int_return: If set to ``True`` the output will be a list of integers. By default the return type is a bit string. This and ``hex_return`` are mutually exclusive and can not be specified at the same time. This option only has an effect with memory level 2. hex_return: If set to ``True`` the output will be a list of hexadecimal strings. By default the return type is a bit string. This and ``int_return`` are mutually exclusive and can not be specified at the same time. This option only has an effect with memory level 2. avg_data: If a 2 dimensional numpy array is passed in for ``memory`` this can be set to ``True`` to indicate it's a avg level 0 data instead of level 1 single data. parallel_threshold: The number of elements in ``memory`` to start running in multiple threads. If ``len(memory)`` is >= this value, the function will run in multiple threads. By default this is set to 1000. Returns: marginal_memory: The list of marginalized memory Raises: ValueError: if both ``int_return`` and ``hex_return`` are set to ``True`` z=Either int_return or hex_return can be specified but not bothz3int_return option only works with memory list inputz3hex_return option only works with memory list inputNrEzInvalid input memory array) return_intrrT) ValueErrorrnpndarraycopyndimr+marginal_measure_level_1_avgmarginal_measure_level_0_avgmarginal_measure_level_1marginal_measure_level_0r,)rrrQrRrSrTs r7r,r,s0TZjZXYYY&"*%%7  TRSS S  TRSS S ?;;== ;!  :67KK K ;!   L!>vwOOO!:67KKK ;!  6vwGG G5666  %-    r9r#cxtt|dd}|Xt|dks/t |t |std|dz dt|trtj ||}nt|ttfrtj||}ntt!|}t|t$t&jfrtj ||}nFt|t*t&jfrtj||}ntd|r|t/|||S|S) alMarginalize counts from an experiment over some indices of interest. Unlike :func:`~.marginal_counts` this function respects the order of the input ``indices``. If the input ``indices`` list is specified then the order the bit indices are specified will be the output order of the bitstrings in the marginalized output. Args: counts: result to be marginalized indices: The bit positions of interest to marginalize over. If ``None`` (default), do not marginalize at all. format_marginal: Default: False. If True, takes the output of marginalize and formats it with placeholders between cregs and for non-indices. Returns: dict(str, int): A marginalized dictionary Raises: QiskitError: If any value in ``indices`` is invalid or the ``counts`` dict is invalid.  Nrindices must be in range [0, rE].z(Values of counts must be an int or float)r$maxkeysreplacesetissubsetrIr rr r+r8r rmarginal_distributionnextitervaluesintrZintegerfloatfloatingr-)r#rr num_clbitsres first_values r7rlrls2S''//R8899JG  1 1W9N9NuU_O`O`9a9a 1L*q.LLLMMM&&!! J(99 F-/@A B B J.vw??4 0011 kC#4 5 5 J,VW==CC  eR[%9 : : J267CCCCHII I67.W555 Jr9cttt|dd}|-t t |t |kr2i}|D]\}t||<|S|rrOkeys r7r@z _marginalize.. s-VVVs3C88#BVVVr9)r$rmrnrirjrIr!ryrkr r*rjoindict)r#rrtretvalr0new_keyrzs @r7r r srT$v,,''//R8899J Cj 1 122c'llBB   HC*3//CCHH N#g,,//E*4E4E0F0FGGNL*q.LLLMMMWd+++GJLLNN##S''VVVVgVVVWW7s"   r9c i}tt|}t|dd}t |d}|D]}t t || d fdt|Dddd}t|D]#\} } | dkr|d| dz|| dz}$||||<|S)z[Take the output of marginalize and add placeholders for multiple cregs and non-indices.rcrdTrc,g|]}|vr|ndS)rBr<)r>index index_dicts r7r@z$_format_marginal..s- ^ ^ ^5%:"5"5Z  3 ^ ^ ^r9N) rmrnr$rir*r|zipr{rIr) r#r6r format_countscounts_template counts_len indices_revcount count_bitsrbitrs @r7r-r-sM4<<((O_,,S"5566J$///K77#k51122 WW ^ ^ ^ ^ER\L]L] ^ ^ ^  $B$ $O44 K KJE3czz'/#5 5668JJ $/$6 j!! r9cV|ddddS)z1Removes all spaces and underscores from bitstringrcrdrB)ri) bitstrings r7ryry%s(   S" % % - -c2 6 66r9)NFFT)NFFFrP)NF)N)(__doc__typingrrrrr collectionsrr\r numpyrZqiskit.exceptionsr qiskit.result.resultr qiskit.result.countsr 'qiskit.result.distributions.probabilityr !qiskit.result.distributions.quasirqiskit.result.postprocessrqiskit._acceleraterr+r|rpboolstrr8r(r[r,rlr r-ryr<r9r7rs2188888888888888))))))''''''''''''DDDDDD??????111111444444 $(!)- BB $, B d3i BB B ! B  4S>6 !" BBBBJ8$(" DD $s)RZ' (D d3i DD D  D  D 49bj !DDDDR(,!,, , hsm $,, #s(^ ,,,,^6(77777r9