U e{j:@sXdZddlmZddlmZddlmZer8ddlmZGdddZ Gdd d Z d S) zc The SCEngine class uses the ``sortedcontainers`` package to implement an Index engine for Tables. ) OrderedDict)starmap)HAS_SORTEDCONTAINERS) SortedListc@sTeZdZdZddZddZddZdd Zd d Zd d Z ddZ dZ ddZ dS)NodekeyvaluecCs||_||_dSNr)selfrr r Q/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/table/soco.py__init__sz Node.__init__cCs,|jtkr"|j|jf|j|jfkS|j|kSr  __class__rrr r otherr r r __lt__s z Node.__lt__cCs,|jtkr"|j|jf|j|jfkS|j|kSr rrr r r __le__s z Node.__le__cCs,|jtkr"|j|jf|j|jfkS|j|kSr rrr r r __eq__"s z Node.__eq__cCs,|jtkr"|j|jf|j|jfkS|j|kSr rrr r r __ne__'s z Node.__ne__cCs,|jtkr"|j|jf|j|jfkS|j|kSr rrr r r __gt__,s z Node.__gt__cCs,|jtkr"|j|jf|j|jfkS|j|kSr rrr r r __ge__1s z Node.__ge__NcCsd|jd|jdS)NzNode(, )rr r r r __repr__8sz Node.__repr__) __name__ __module__ __qualname__ __slots__rrrrrrr__hash__rr r r r rsrc@sveZdZdZdddZddZddZdd d Zd d ZddZ ddZ ddZ ddZ dddZ ddZddZd S) SCEngineat 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 Whether the values of the index must be unique. Defaults to False. FcCs6ts tdtt|}tttt|||_||_ dS)Nz-sortedcontainers is needed for using SCEngine) r ImportErrormaptuplerrrzip_nodes_unique)r dataZ row_indexuniqueZ node_keysr r r rLs  zSCEngine.__init__cCs:|jr$||jkr$d|d}t||jt||dS)z( Add a key, value pair. z duplicate z in unique indexN)r(r' ValueErroraddr)r rr messager r r r,Ts z SCEngine.addcCsdd|j||DS)z; Find rows corresponding to the given key. cSsg|] }|jqSr r .0noder r r asz!SCEngine.find..r'irange)r rr r r find]sz SCEngine.findNcCsn|dk r>t||}z|j|Wntk r8YdSXdSt|j||}|D]}|j|qTt|S)z1 Remove data from the given key. NFT)rr'remover+listr4bool)r rr)itemitemsr r r r6cs zSCEngine.removecCs(|jD]}|j|kr|jd8_qdS)z; Decrement rows larger than the given row. Nr'r r rowr1r r r shift_leftss  zSCEngine.shift_leftcCs(|jD]}|j|kr|jd7_qdS)zH Increment rows greater than or equal to the given row. r;Nr<r=r r r shift_right{s  zSCEngine.shift_rightcCsFt}|jD]0}|j|kr.||j|jq |jg||j<q |S)z4 Return a list of key, data tuples. )rr'rappendr r:)r resultr1r r r r:s   zSCEngine.itemscCst|jD]\}}||_q dS)z6 Make row order align with key order. N) enumerater'r )r indexr1r r r sortsz SCEngine.sortcCsdd|jDS)z? Return a list of rows in order sorted by key. cSsg|] }|jqSr r.r/r r r r2sz(SCEngine.sorted_data..)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 r2sz"SCEngine.range..r3)r lowerupperZboundsiteratorr r r rangeszSCEngine.rangecsDfdd|jD}|D]}|j|_q|j|j|dS)z: Replace rows with the values in row_map. csg|]}|jkr|qSr r.r/row_mapr r r2s z)SCEngine.replace_rows..N)r'r clearupdate)r rMnodesr1r rLr replace_rowss  zSCEngine.replace_rowscCsjt|jdkr:t|jdddgt|jdd}n|j}ddd|D}d|jjd |d S) Nz...rcss|]}t|VqdSr )strr/r r r sz$SCEngine.__repr__..)lenr'r7joinrr)r rPZ nodes_strr r r rs ,zSCEngine.__repr__)F)N)rG)rrr__doc__rr,r5r6r?r@r:rErFrKrQrr r r r r"<s      r"N) r[ collectionsr itertoolsrZ"astropy.utils.compat.optional_depsrZsortedcontainersrrr"r r r r s    +