DUf8ddlmZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl Z ddl mZn #e$reZYnwxYwddlZddlZddlmZmZdd lmZd d lmZmZd d lmZmZgd ZdJdZ dKdZ!dLdZ"GddZ#Gdde#Z$ dMdNd$Z%dOd'Z&dPd)Z'dQdRd-Z(dSd.Z)dTd0Z*dTd1Z+dUd2Z,dVd:Z-dWdXd<Z.dWdXd=Z/ dYdZd?Z0d[dBZ1GdCdDeZ2GdEdFZ3d\dGZ4d]dHZ5d]dIZ6dS)^) annotationsN)Callable)datetime)Number)AnyLiteral)JSONDecodeError)BoxStyle LeftAligned) Traversal)MAGIC MAGIC_SCOOL) natsortedparse_cooler_uri) is_cooleris_multires_file list_coolerscpmvlnoobjectreturnstrc4tj|ddddS)z/Write JSON in a consistent, human-readable way.T),z: )indent sort_keys ensure_ascii separators)jsondumps)rs K/var/www/html/software/conda/lib/python3.11/site-packages/cooler/fileops.py json_dumpsr&s& : !t$;   sc*tj|S)zRead JSON in a consistent way.)r#loads)r(s r% json_loadsr+%s :a==r'obj,str | np.generic | np.ndarray | h5py.Dataset%str | Number | datetime | list | dictcft|dr|}nt|dr|}nft|trO t j|d}n:#t$r+ tj |}n#t$r|}YnwxYwYnwxYw|}|S)zf Decode a HDF5 attribute value (numpy object or string) into something JSON serializable. itemtolistz%Y-%m-%dT%H:%M:%S.%f) hasattrr0r1 isinstancerrstrptime ValueErrorr#r*r )r,rs r%decode_attr_valuer6*ssF  HHJJ h    JJLL C    !#'=>>AA    JsOO"       Hs6!A77 B,BB, B&#B,%B&&B,+B,c0eZdZddd Zdd ZddZddZdS)TreeNoderNr,rdepthintlevel int | Nonec0||_||_||_dSN)r,r9r;)selfr,r9r;s r%__init__zTreeNode.__init__Cs  r'rrc4t|jjSr>)typer,__name__r?s r%get_typezTreeNode.get_typeHsDH~~&&r'list[TreeNode]ctjdrIjjjkr2jdzj}fd|DSgS)Nvaluesr cJg|]}|j S))r9r;) __class__r;).0rr9r?s r% z)TreeNode.get_children..Ps:IJDNN1ENDDr')r2r,r;r9rH)r?childrenr9s` @r% get_childrenzTreeNode.get_childrenKs 48X & & z!TZ$*%<%< Q8??,,NV r'c|jjddpd}t|jdr|d|jjd|jjz }|S)N/shape )r,namesplitr2rRdtype)r?rTs r%get_textzTreeNode.get_textUs_x}""3''+2s 48W % % : 99999 9D r')rN)r,rr9r:r;r<rr)rrF)rC __module__ __qualname__r@rErNrWr'r%r8r8Bsi ''''r'r8ceZdZddZdS)AttrNoderrcP|jjddpdS)NrPrQ)r,rTrUrDs r%rWzAttrNode.get_text]s$x}""3''+2s2r'NrX)rCrYrZrWr[r'r%r]r]\s(333333r'r]group h5py.Groupfunc/Callable[[str, h5py.Group | h5py.Dataset], Any]r;r<dict[str, Any]cLdfd t||}||iS)zALike :py:method:`h5py.Group.visititems`, but much faster somehow.Nc|}|r:|D]7}||jj|j||jj<|||8|Sr>)rNr,rT)noderaresultrMchild _visititemss r%rizvisititems.._visititemshsg$$&&  1! 1 1)-einei)H)Huy~& E40000 r'r;r>)r8)r_rar;rootris @r% visititemsrlasI E ' ' 'D ;tT2 & &&r'grpboolcjdd}|tkrkeys)rKrTrms r% z_is_cooler..xs077$4388::%777777r'z Cooler path z appears to be corruptTF)attrsgetrallwarningswarnrT)rmfmtrws` r% _is_coolerrtst )--$ ' 'C e||67777$77777 K MIIII J J Jt 5r'urict|\}}tj|sdStj|5}t ||cdddS#1swxYwYdS)z Determine if a URI string references a cooler data collection. Returns False if the file or group path doesn't exist. FN)rh5pyis_hdf5Filer)rfilepath grouppathfs r%rr~s +3//Hi < ! !u 8  (!I,''((((((((((((((((((sAA#&A#r min_versionr:ctj|sdStj|5}|jdd}d|vrt |ddkrdtt|d}|dkr)t|d|r ddddSn?d|vr)t|dr|d kr ddddSdddn #1swxYwYdS) zg Determine if a file is a multi-res cooler file. Returns False if the file doesn't exist. FrpN resolutionsrz HDF5::MCOOLT0) rrrryrzrwlennextiterr)rrrr~rTs r%rrs < ! !u 8  gkk(D)) AFFHH $ $Q}-=-B-B-D-D)E)E)I)IQ}-22445566Dm## 1]3CD3I(J(J#  AFFHH__AcF!3!3_ a 5sB3D7+3D77D;>D;ctj|std|dtj|5jdd}|t krd}tfd|Ds"tj d ddddSd  vrtd  d krTd  D]+}td |sddddS, dddd Sdddn #1swxYwYdS) zi Determine if a file is a single-cell cooler file. Returns False if the file doesn't exist. '' is not an HDF5 file.FrpN)rqrrcellsc3DK|]}|vVdSr>rv)rKrTrs r%rxz is_scool_file..s099Dtqvvxx'999999r'z Scool file appears to be corruptrrT) rrOSErrorrryrzrr{r|r}rwrr)rr~rwrrs @r% is_scool_filers < ! !:(:::;;; 8   gkk(D)) +  .D9999D99999  @AAA        !&&((""s1W:??+<+<'='='A'AwZ__..%%E%aj&788%$        %                        5sAE#A9E)EE E list[str]ctj|std|dgfd}tj|d5}|d|t ||dddn #1swxYwYt S)z List group paths to all cooler data collections in a file. Parameters ---------- filepath : str Returns ------- list Cooler group paths in the file. rrct|r1|dsd|zn|dSdSNrPrappend startswithpthrmlistings r% _check_coolerz#list_coolers.._check_coolerO c?? J NNCNN3,?,?H399S I I I I I J Jr'rrPN)rrrrrlrrrrrs @r%rrs < ! !<:(:::;;;GJJJJJ 8S ! !%Q c11m$$$%%%%%%%%%%%%%%% W  sA..A25A2c8fd}t|rsgtj|d5}|d|t||dddn #1swxYwYdvrdt St d|d)z List the paths to all single-cell cool matrices in a file scool file. Parameters ---------- filepath : str Returns ------- list Cooler group paths of all cells in the file. ct|r1|dsd|zn|dSdSrrrs r%rz'list_scool_cells.._check_coolerrr'rrPNrz' is not a scool file.)rrrrlremoverrrs @r%list_scool_cellsrsJJJJJX < Yx % % ) M#q ! ! ! q- ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) '>> NN3   !!!:(:::;;;sAAAc&t|\}}tj|std|dgfd}tj|d5}|||t |||dddn #1swxYwYS)z Get all groups and datasets in an HDF5 file. Parameters ---------- uri : str Returns ------- list Group and dataset paths. rrcf|dsd|zn|dSr)rrrs r% _check_allzls.._check_all s4s(;(;DsSyyEEEEEr'rN)rrrrrrl)rrrrrrs @r%lsrs+3//Hi < ! !<:(:::;;;GFFFFF 8S ! !-Q 9a   1Y<,,,--------------- Ns#BB  B src_uridst_uri overwritelinkrename soft_linkNonect|\}}t|\}} t|||gdkrtdtj|r|rd} nd} ||krd} nd} t j|| 5} t j|| 5} ||krF|s|r| || | <|r| |=n|rt j|| | <n| || n|rtd|rt j ||| | <n| dkrf| | D]}| |dz|z| | | | j | |j n| || | dkr| nddddn #1swxYwYddddS#1swxYwYdS)Nr z:Must provide at most one of: "link", "rename", "soft_link"wzr+rz,Can't hard link between two different files.rP)rsumr5ospathisfilerrSoftLinkcopyr ExternalLinkrwryupdate)rrrrrrsrc_path src_groupdst_path dst_groupdst_write_modesrc_write_modesrcdstsubgrps r%_copyrs+733Hi*733Hi D&) $%%))UVVV 7>>( # #y8 8^ , ,V 8^ , ,V03 x   /v /!$YI'I /!%y!9!9II.... VLMMM V!%!28Y!G!GI##"%i."5"5"7"7HHS6!93GGGG N(//I0DEEEEHHY)s:J:JYYPTUUU-VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVs7G&DF6* G6F: :G=F: >GGGFc0t|||ddddS)zICopy a group or dataset from one file to another or within the same file.FrrrNrrrrs r%rrEs# '7IE%5QQQQQQr'c0t|||ddddS)z/Rename a group or dataset within the same file.FTrNrrs r%rrJs# '7IE$%PPPPPPr'softc2t|||| d|dS)zCreate a hard link to a group or dataset in the same file. Also supports soft links (in the same file) or external links (different files). FrNr)rrrrs r%rrOs'  '7IHUdSSSSSSr'expandrc  ddln%#t$r}t|ddd}~wwxYwdd ddfd }t||}||d||S)Nrz: Run `pip install ipytree` or `conda install ipytree`to get the required ipytree dependency for displaying the tree widget. If using jupyterlab, you also need to run `jupyter labextension install ipytree`stype,Literal['Dataset', 'Array', 'Group', 'File']rLiteral['table', 'folder']c>|dvrdS|dvrdStd|)N>ArrayDatasettable>rGroupfolderzUnknown type: )r5)rs r%_tree_get_iconz#tree_widget.._tree_get_iconfs= ( ( (7 ' ' '85e5566 6r'Frfr8rkrnr ipytree.Nodecl}||_|s$dus ttr|jkrd|_nd|_||_fd| D|_ d|_ |S)NTFc*g|]}|S))rr[)rKc_tree_widget_sublistrs r%rLz=tree_widget.._tree_widget_sublist..~s7   78 6 2 2 2   r') NoderEiconr3r:r9openedrWrTrNnodesdisabled)rfrkrrgrripytrees ` r%rz)tree_widget.._tree_widget_sublistps$nT]]__55  "t^^  $ $$)-f)<)< FMM!FMmmoo      <@)rNr?rfs r%rNzTreeTraversal.get_childrens  """r'treec|Sr>r[r?rs r%get_rootzTreeTraversal.get_roots r'rc*|Sr>)rWrs r%rWzTreeTraversal.get_texts}}r'N)rfr8rrF)rr8rr8)rfr8rr)rCrYrZrNrrWr[r'r%rrsP####r'rcBeZdZdZddefdd ZddZddZddZddZ dS) TreeViewera  Generates ascii- or html-based reprs for "Groupy" objects. Borrowed with minor modifications from the zarr project (Zarr Developers, MIT-licensed). See: zarr.util.TreeViewer, zarr.util.tree_html FNr_r`rrnr;r<node_clsrc||_||_||_dddd|_ddddd|_dd d d d|_||_dS) Nrr ) horiz_len label_spacer+-|) UP_AND_RIGHT HORIZONTALVERTICALVERTICAL_AND_RIGHTu└u─u│u├)r_rr; text_kwargs bytes_kwargsunicode_kwargsr)r?r_rr;rs r%r@zTreeViewer.__init__sr   )*1JJ "%   %" "*   ! r'rbytesctttdd|ji|j}||j|j}||}|}|SNgfx)traversedrawrjr[) r rr rrrr_r;encode)r?drawerrkrgs r% __bytes__zTreeViewer.__bytes__su"__DDd/D43CDD   }}TZtz}:: r'rctttdd|ji|j}||j|j}||Sr)r rr rrrr_r;)r?r rks r% __unicode__zTreeViewer.__unicode__sb"__FFd1FT5EFF   }}TZtz}::vd||r'c*|Sr>)rrDs r%__repr__zTreeViewer.__repr__s!!!r'cpt|j|j|j}||S)N)rr;)rr_rr;_repr_mimebundle_rs r%rzTreeViewer._repr_mimebundle_s34:dkLLL     r')r_r`rrnr;r<rr)rrrX)rr) rCrYrZ__doc__r8r@r rrrr[r'r%rrs    !!!!!:    """"r'rcHdfd t||dS)NFc|jj}dd|Di}|}|r%|D]"}|||<#|S)Nz@attrsc4i|]\}}|t|Sr[)r6)rKkvs r% z4read_attr_tree.._getdict..s'OOOAQ 1! 4 4OOOr')r,ryitemsrNrW)rfrkryrgrMrh_getdicts r%rz read_attr_tree.._getdictsyOOOOOP$$&&  ;! ; ;+38E??u~~''(( r'rjT)rkF)r])r_r;rs @r%read_attr_treersE 8HU%000t < < <)r_r`rarbr;r<rrc)rmr`rrn)rrrrn)r )rrrr:rrn)rrrrn)rrrr)rrrr)rrrrrrnrrnrrnrrnrrr)rrrrrrnrrr) rrrrrrnrrnrr)r_r`rrnr;r<rr)r_r`r;r<rrc)rrr;r<rr)7 __future__rrr|collections.abcrrnumbersrtypingrr simplejsonr#r rr5rnumpynp asciitreer r asciitree.traversalr createrrutilrr__all__r&r+r6r8r]rlrrrrrrrrrrrrrrrr%r(r[r'r%r5s"""""" $$$$$$!*******!!! OOO! ++++++++))))))&&&&&&&&-------- M M M     0433333x333'''''& ( ( ( (*2<<<<<><.V.V.V.VbRRRRR QQQQQ GLTTTTT....bIEEEEEEEEP = = = =    222222s 5??