g'rdZddlmZddlZddlmZddlmZmZ ddl m Z ddl m Z mZmZddlmZddlZdd lmZdd lmZmZmZd d lmZd d lmZmZddgZe eeeee eefeffZ e e efZ! GddeZ"e d ddZ#e d ddZ$dS)z5 ND-Array container class for Estimator observables. ) annotationsN) defaultdict)IterableMapping) lru_cache)Unionroverload)Complex) ArrayLike)Pauli PauliList SparsePauliOp) object_array) ShapedMixin shape_tupleObservableLikeObservablesArrayLikeceZdZUdZdZdZded< d&d'fd Zd Zd(dZ d)dZ e d*dZ e d+dZ dZ d,dZ d-dZed.d Zed/d!Zd"Zed0d%ZxZS)1ObservablesArrayzJAn ND-array of Hermitian observables for an :class:`.Estimator` primitive.)_array_shapez IXYZ01+-lrstr ALLOWED_BASIST observablesrcopyboolvalidatectt|tr|j}t ||t f|_|jj|_|rd}tj |jD]i\}}| |}ttt|}||}n||krtd||j|<hdSdS)aInitialize an observables array. Args: observables: An array-like of basis observable compatible objects. copy: Specify the ``copy`` kwarg of the :func:`.object_array` function when initializing observables. validate: If true, coerce entries into the internal format and validate them. If false, the input should already be an array-like. Raises: ValueError: If ``validate=True`` and the input observables is not valid. )r list_typesNSThe number of qubits must be the same for all observables in the observables array.)super__init__ isinstancerrrr shapernp ndenumeratecoerce_observablelennextiter ValueError) selfrrr num_qubitsndiobs basis_obsbasis_num_qubits __class__s d/granary/edev/quantum/lib/python3.11/site-packages/qiskit/primitives/containers/observables_array.pyr#zObservablesArray.__init__9s $  k#3 4 4 -%,K";TylSSS k'  -JN4;77 - -S 22377 #&tDOO'<'<#=#= %!1JJ%33$-$- C   - - - -ct|jd}d|jd}tj|j||d}||z|zS)N(z, shape=)2)prefixsuffix threshold)type__name__r%r& array2stringr)r-r:r;arrays r4__repr__zObservablesArray.__repr__^sUJJ'***)DJ))) F6UWXXX~&&r5returnlistc4|jS)zConvert to a nested list)rtolistr-s r4rEzObservablesArray.tolistds{!!###r5Nc~| |tkr"|r|jn|jStd)zConvert to an Numpy.ndarrayNzType must be 'None' or 'object')objectrrr,)r-dtypers r4 __array__zObservablesArray.__array__hs> =EVOO)->4;##%%%4; >:;;;r5argsint | tuple[int, ...]Mapping[str, float]cdSNr-rKs r4 __getitem__zObservablesArray.__getitem__nsORsr5slicecdSrOrPrQs r4rRzObservablesArray.__getitem__qsz Non-Hermitian input observable: z& term has a complex value coefficient.zDNumber of qubits must be the same for all observable basis elements.zInvalid observable type: )r$rsimplifyr& real_if_closecoeffs iscomplexobjr,paulis to_labelsdictzipr to_labelphaser_validate_basis_Mappingr)r*r+rfloatitemsr absimag TypeErrorrealr=) clsr_rfrhlabelrmr.uniquebasiscoeffs r4r(z"ObservablesArray.coerce_observables j- 0 0 -#,,!,44J%j&788Fv&&  ; &0022FFF++,, , j% ( ( =%aaa=1133Z5E5Eqy  ]"'!E1::% < j# & & #    + + +? " j( + + T$z"2"23344J ''F * 0 0 2 2 ' ' ueU++'#(8#4#4#6#6 5Eqy(ezz!&eW--'5:--',u,,,"JE##E***u::++$^u & << FD4D4DFFGGGr5cFt|tr|S||S)zCoerce ObservablesArrayLike into ObservableArray. Args: observables: an object to be observables array. Returns: A coerced observables array. )r$r)rvrs r4coercezObservablesArray.coerces- k#3 4 4  s;r5cd}|jdD]E}ttt |}||}0||krt dFdS)z.Validate the consistency in observables array.Nrcr!)rr[r)r*r+r,)r-r.r0r2s r4rzObservablesArray.validates~ ;&&r**  C"4S ??33 !- !Z// )0   r5ryNonec 4t|j}||snt|j}t t ||}td|d|dt |jddS)zValidate a basis string. Args: basis: a basis string to validate. Raises: ValueError: If basis string contains invalid characters zObservable basis string 'z' contains invalid characters z, allowed characters are .N) _regex_matchrmatch_regex_invalidrCsetfindallr,)rvryallowed_patterninvalid_pattern invalid_charss r4rnz ObservablesArray._validate_basiss's'899$$U++ ,S->??O _%<% __module__ __qualname____doc__ __slots__r__annotations__r#rArErJr rRr[r^ classmethodr(r|rrn __classcell__)r3s@r4rr2sTT$I%M%%%%2  #-#-#-#-#-#-#-J''' $$$$<<<< RRRXR ???X?BBB X X X X ' ' ' 'GHGHGH[GHR    [     [r5r allowed_charsrrB re.PatterncVtjdtj|dS)zMReturn pattern for matching if a string contains only the allowed characters.z^[z]*$recompileescapers r4rrs* :829]33888 9 99r5cVtjdtj|dS)z,Return pattern for selecting invalid stringsz[^]rrs r4rrs* :629]33666 7 77r5)rrrBr)%r __future__rr collectionsrcollections.abcrrro functoolsrtypingrr numbersr numpyr& numpy.typingr qiskit.quantum_infor r rrr%rr__all__rrprrrrrrPr5r4rs#""""" ######99999999++++++++++""""""??????????&&&&&&++++++++ 3 4  E#u* u $%' U^Y67ZYYYYY{YYYx 1::::  1888888r5