gdZddlmZddlmZddlmZmZmZddl Z ddl m Z m Z eedfZ eGdd eZGd d eZddZddZddZdS)z+ Array shape related classes and functions ) annotations)Iterable)ProtocolUnionruntime_checkableN) ArrayLikeNDArrayzIterable[ShapeInput]cZeZdZdZed dZed dZed dZdS) Shapeda%Protocol that defines what it means to be a shaped object. Note that static type checkers will classify ``numpy.ndarray`` as being :class:`Shaped`. Moreover, since this protocol is runtime-checkable, we will even have ``isinstance(, Shaped) == True``. returntuple[int, ...]c td)zThe array shape of this object.z7A `Shaped` protocol must implement the `shape` propertyNotImplementedErrorselfs X/granary/edev/quantum/lib/python3.11/site-packages/qiskit/primitives/containers/shape.pyshapez Shaped.shape%s""[\\\intc td)z.The number of array dimensions of this object.z6A `Shaped` protocol must implement the `ndim` propertyrrs rndimz Shaped.ndim*""Z[[[rc td)zVThe total dimension of this object, i.e. the product of the entries of :attr:`~shape`.z6A `Shaped` protocol must implement the `size` propertyrrs rsizez Shaped.size/rrN)r r )r r)__name__ __module__ __qualname____doc__propertyrrrrrr r s]]]X]\\\X\\\\X\\\rr cfeZdZUdZded<dZedZedZedZ dS) ShapedMixinzXMixin class to create :class:`~Shaped` types by only providing :attr:`_shape` attribute.r _shapec@t|jd|jdS)Nz())typerrrs r__repr__zShapedMixin.__repr__:s$t**%777777rc|jSN)r$rs rrzShapedMixin.shape=s {rc*t|jSr))lenr$rs rrzShapedMixin.ndimAs4;rc\ttj|jtS)N)dtype)rnpprodr$rs rrzShapedMixin.sizeEs!274;c222333rN) rrrr__annotations__r'r rrrr!rrr#r#5sbb888X  X 44X444rr#arrArrayLike | Shapedr NDArray | ShapedcXt|tr|Stj|S)aCoerce the input into an object with a shape attribute. Copies are avoided. Args: arr: The object to coerce. Returns: Something that is :class:`~Shaped`, and always ``numpy.ndarray`` if the input is not already :class:`~Shaped`. ) isinstancer r.asarray)r1s r array_coercer7Js)#v :c??rarg ShapeInput Iterable[int]c#6K|D]} t|trt|Ed{Vn8t||krt|Vnt d|dg#t t f$r}t d|d|d}~wwxYwdS)a Yield one integer at a time. Args: arg: Integers or iterables of integers, possibly nested, to be yielded. Yields: The provided integers in depth-first recursive order. Raises: ValueError: If an input is not an iterable or an integer. Nz Expected z to be iterable or an integer.)r5r_flatten_to_intsr ValueError TypeErrorRecursionError)r8itemexs rr<r<[s W W W$)) S+D1111111111Td""$ii !QT!Q!Q!QRRR>* W W WMMMMNNTV V W W WsA#A,,B=BBshapesr c:tt|S)a% Flatten the input into a single tuple of integers, preserving order. Args: shapes: Integers or iterables of integers, possibly nested. Returns: A tuple of integers. Raises: ValueError: If some member of ``shapes`` is not an integer or iterable. )tupler<)rBs r shape_tuplerEts !&)) * **r)r1r2r r3)r8r9r r:)rBr9r r )r __future__rcollections.abcrtypingrrrnumpyr. numpy.typingrr rr9r r#r7r<rEr!rrrKs=#"""""$$$$$$5555555555++++++++ 3.. / 4\\\\\X\\\044444&444*"WWWW2 + + + + + +r