B q\@sldZddlmZddlmZyddlmZdZWnek rHdZYnXGddde Z Gd d d Z d S) zc The SCEngine class uses the ``sortedcontainers`` package to implement an Index engine for Tables. ) OrderedDict)starmap) SortedListTFc@sTeZdZdZddZddZddZdd Zd d Zd d Z ddZ dZ ddZ dS)Node)keyvaluecCs||_||_dS)N)rr)selfrrr 1lib/python3.7/site-packages/astropy/table/soco.py__init__sz Node.__init__cCs,|jtkr"|j|jf|j|jfkS|j|kS)N) __class__rrr)rotherr r r __lt__s z Node.__lt__cCs,|jtkr"|j|jf|j|jfkS|j|kS)N)r rrr)rr r r r __le__s z Node.__le__cCs,|jtkr"|j|jf|j|jfkS|j|kS)N)r rrr)rr r r r __eq__#s z Node.__eq__cCs,|jtkr"|j|jf|j|jfkS|j|kS)N)r rrr)rr r r r __ne__(s z Node.__ne__cCs,|jtkr"|j|jf|j|jfkS|j|kS)N)r rrr)rr r r r __gt__-s z Node.__gt__cCs,|jtkr"|j|jf|j|jfkS|j|kS)N)r rrr)rr r r r __ge__2s z Node.__ge__NcCsd|j|jS)NzNode({0!r}, {1!r}))formatrr)rr r r __repr__9sz Node.__repr__) __name__ __module__ __qualname__ __slots__r rrrrrr__hash__rr r r r rsrc@sveZdZdZdddZddZddZdd d Zd d ZddZ ddZ ddZ ddZ dddZ ddZddZd S) SCEngineal Fast tree-based implementation for indexing, using the ``sortedcontainers`` package. Parameters ---------- data : Table Sorted columns of the original table row_index : Column object Row numbers corresponding to data columns unique : bool (defaults to False) Whether the values of the index must be unique FcCs*tt|}tttt|||_||_dS)N)maptuplerrrzip_nodes_unique)rdataZ row_indexuniqueZ node_keysr r r r Ks zSCEngine.__init__cCs8|jr"||jkr"d|}t||jt||dS)z( Add a key, value pair. z duplicate {0:!r} in unique indexN)r rr ValueErroraddr)rrrmessager r r r$Ps z SCEngine.addcCsdd|j||DS)z; Find rows corresponding to the given key. cSsg|] }|jqSr )r).0noder r r ]sz!SCEngine.find..)rirange)rrr r r findYsz SCEngine.findNcCsp|dk r.)r)rr r r sorted_dataszSCEngine.sorted_dataTTcCs|j|||}dd|DS)z7 Return row values in the given range. cSsg|] }|jqSr )r)r&r'r r r r(sz"SCEngine.range..)rr))rlowerupperZboundsiteratorr r r rangeszSCEngine.rangecsHfdd|jD}x|D]}|j|_qW|j|j|dS)z: Replace rows with the values in row_map. csg|]}|jkr|qSr )r)r&r')row_mapr r r(sz)SCEngine.replace_rows..N)rrclearupdate)rr?Znodesr'r )r?r replace_rowss   zSCEngine.replace_rowscCsdt|jS)Nz{0!r})rr,r)rr r r rszSCEngine.__repr__)F)N)r:)rrr__doc__r r$r*r+r2r3r/r8r9r>rBrr r r r r=s       rN) rC collectionsr itertoolsrZsortedcontainersrZHAS_SOCO ImportErrorobjectrrr r r r s    +