g"TdZddlmZddlZddlZddlZddlmZGddZ dS)z! Base register reference object. ) annotationsN) CircuitErrorceZdZdZgdZejZdZdZ ddd Z d Z e d Z e d Zd ZdZdZdZdZdZdZdZdZdZdS)RegisterzImplement a generic register. .. note:: This class should not be instantiated directly. This is just a superclass for :class:`~.ClassicalRegister` and :class:`~.QuantumRegister`. )_name_size_bits _bit_indices_hash_reprregNsize int | Nonename str | Nonec||fdks||td|d|d|t|} |t|k}n#ttf$rd}YnwxYw|s(tdt |jd|d t|}|d kr(td t |jd|d |jtj }n3 t|}n"#t$r}td |d}~wwxYw|_ |_ tt j j f_jjd jdjd_|j tt+|krtd|t-fd|Drtd|t/|_dt3jD_dSfdt7|D_d_dS)aCreate a new generic register. Either the ``size`` or the ``bits`` argument must be provided. If ``size`` is not None, the register will be pre-populated with bits of the correct type. Args: size (int): Optional. The number of bits to include in the register. name (str): Optional. The name of the register. If not provided, a unique name will be auto-generated from the register type. bits (list[Bit]): Optional. A list of Bit() instances to be used to populate the register. Raises: CircuitError: if both the ``size`` and ``bits`` arguments are provided, or if neither are. CircuitError: if ``size`` is not valid. CircuitError: if ``name`` is not a valid name according to the OpenQASM spec. CircuitError: if ``bits`` contained duplicated bits. CircuitError: if ``bits`` contained bits of an incorrect type. )NNNzIExactly one of the size or bits arguments can be provided. Provided size=z bits=.Fz#Register size must be an integer. (z 'z' was provided)rz$Register size must be non-negative (zRThe circuit name should be castable to a string (or None for autogenerate a name).(z, 'z')z+Register bits must not be duplicated. bits=c3DK|]}t|j VdSN) isinstancebit_type).0bitselfs M/granary/edev/quantum/lib/python3.11/site-packages/qiskit/circuit/register.py z$Register.__init__..us2FF#z#t}555FFFFFFz4Provided bits did not all match register type. bits=ci|]\}}|| Sr ridxrs r z%Register.__init__..x P P Phc3c P P Prc<g|]}|Sr )rrr"rs r z%Register.__init__..zs'JJJs$--c22JJJr)rlenint ValueError TypeErrortype__name__prefixnextinstances_counterstr Exceptionrrhashr __class__ __qualname__rrr setanylistr enumerater range)rrrbits valid_sizeexs` r__init__zRegister.__init__-s0 $<< ' 'D,<AQ?+/??7;???   t99D T*JJI&   JJJ  bd4jj6IbbTbbb 4yy !88ctDzz7Jccdccc  <kA4(>#?#?AADD 4yy   "9    4::tz4:>?? 3QQdiQQDIQQQ  zST^^++"#WQU#W#WXXXFFFFFFFFF b"#`Z^#`#`aaadDJ P P)DJ:O:O P P PD   JJJJeDkkJJJDJ !%D   s)A A A +C;; DDDc|jdddtdddkrtjdt d|S)NrrzNsubclassing 'Register' is not supported, and may not be possible in Qiskit 2.0)category stacklevel) __module__splitr-warningswarnDeprecationWarning)clss r__init_subclass__zRegister.__init_subclass__sb >  Q ' ' *hnnS!.D.DQ.G G G M`+      rc|jS)zGet the register name.)rrs rrz Register.name zrc|jS)zGet the register size.rrKs rrz Register.sizerLrc|jS)z5Return the official string representing the register.)r rKs r__repr__zRegister.__repr__ zrc|jS)zReturn register size.rNrKs r__len__zRegister.__len__rQrc~t|ttjtt fst dt|tr j|St|t r=t|tkrfd|DSt dj|S)a Arg: bit_type (Qubit or Clbit): a constructor type return element/s. key (int or slice or list): index of the bit to be retrieved. Returns: Qubit or Clbit or list(Qubit) or list(Clbit): a Qubit or Clbit instance if key is int. If key is a slice, returns a list of these instances. Raises: CircuitError: if the `key` is not an integer or not in the range `(0, self.size)`. z-expected integer or slice index into registerc*g|]}j|Sr )r r&s rr'z(Register.__getitem__..s777C 3777rzregister index out of range) rr)npintegerslicer8rr maxr()rkeys` r __getitem__zRegister.__getitem__s#RZ=>> PNOO O c5 ! ! #:c? " T " " #3xx#d))##777737777"#@AAA:c? "rc#VKt|jD]}|j|VdSr)r:rr )rr"s r__iter__zRegister.__iter__s>$$ " "C*S/ ! ! ! ! " "rch|j#dt|jD|_||jvS)Nci|]\}}|| Sr r r!s rr#z)Register.__contains__..r$r)r r9r )rrs r __contains__zRegister.__contains__s9   $ P P)DJ:O:O P P PD d'''rc|j#dt|jD|_ |j|S#t$r}t d|d|d|d}~wwxYw)z8Find the index of the provided bit within this register.Nci|]\}}|| Sr r r!s rr#z"Register.index..r$rzBit z not found in Register r)r r9r KeyErrorr*)rrerrs rindexzRegister.indexs   $ P P)DJ:O:O P P PD  R$S) ) R R RGCGGGGGHHc Q Rs 9 AAAc||urdSd}t|t|ur9|j|jkr)tdt||Drd}|S)aTwo Registers are the same if they are of the same type (i.e. quantum/classical), and have the same name and size. Additionally, if either Register contains new-style bits, the bits in both registers will be checked for pairwise equality. If two registers are equal, they will have behave identically when specified as circuit args. Args: other (Register): other Register Returns: bool: `self` and `other` are equal. TFc3`K|])\}}d|j|j|j|jfv!||kV*dSr) _register_index)rsbitobits rrz"Register.__eq__..sTD$DNDKUUU UUUU r)r,r allzip)rotherress r__eq__zRegister.__eq__s 5==4 JJ$u++ % % ek))#&dE"2"2*C rc|jS)z9Make object hashable, based on the name and size to hash.)r rKs r__hash__zRegister.__hash__rQrcB|j|j|j|j|jfSr)rrr r r rKs r __getstate__zRegister.__getstate__sz4:tz4:tzIIrcV|\|_|_|_|_|_d|_dSr)rrr r r r )rstates r __setstate__zRegister.__setstate__s*EJB DJ DJ  r)NNN)rrrr)r-rCr5__doc__ __slots__ itertoolscountr0r.rr>rIpropertyrrrPrSr[r]r`rerprrrtrwr rrrrsHNMMI( )) FHS%S%S%S%S%j   XX###2"""((( RRR<JJJ !!!!!rr) rx __future__rrzrEnumpyrVqiskit.circuit.exceptionsrrr rrrs#"""""222222Z!Z!Z!Z!Z!Z!Z!Z!Z!Z!r