o Nrf8@sddlmZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl Z zddl mZWn ey?eZYnwddlZddlZddlmZmZdd lmZd d lmZmZd d lmZmZgd Zd]ddZ d^ddZ!d_ddZ"GdddZ#Gddde#Z$ d`dad'd(Z%dbd+d,Z&dcd.d/Z'ddded3d4Z(dfd5d6Z)dgd8d9Z*dgd:d;Z+dhdPsz)TreeNode.get_children..)r.r)r6r4r@)r9childrenr#rCr$ get_childrenKs    zTreeNode.get_childrencCsB|jjddp d}t|jdr|d|jjd|jj7}|S)N/shape )r)namesplitr.rIZdtype)r9rKr#r#r$get_textUs zTreeNode.get_text)rN)r)rr4r5r6r7rr)rr?)r< __module__ __qualname__r:r>rFrMr#r#r#r$r3Bs     r3c@seZdZdddZdS)AttrNoderrcCs|jjddp dS)NrGrH)r)rKrLr=r#r#r$rM]szAttrNode.get_textNrN)r<rOrPrMr#r#r#r$rQ\srQgroup h5py.Groupfunc/Callable[[str, h5py.Group | h5py.Dataset], Any]r6r7dict[str, Any]cs&dfdd t||d}||iS)zALike :py:method:`h5py.Group.visititems`, but much faster somehow.Ncs>|}|r|D]}||jj|j||jj<|||q|Sr8)rFr)rK)noderTresultrEchild _visititemsr#r$r[hs zvisititems.._visititemsr6r8)r3)rRrTr6rootr#rZr$ visititemsas  r^grpboolcsLjdd}|tkr$d}tfdd|Ds"tdjddSdS) Nformat)chromsbinsZpixelsZindexesc3|] }|vVqdSr8keysrBrKr_r#r$ xz_is_cooler..z Cooler path z appears to be corruptTF)attrsgetr allwarningswarnrK)r_fmtrfr#rhr$ _is_coolertsrquricCsTt|\}}t|s dSt|}t||WdS1s#wYdS)z Determine if a URI string references a cooler data collection. Returns False if the file or group path doesn't exist. FN)rh5pyis_hdf5Filerq)rrfilepath grouppathfr#r#r$r~s    $rrv min_versionr5cCst|sdSt|q}|jdd}d|vrDt|ddkrDtt|d}|dkrCt |d|rC WddSn!d|vr]t |drm|d kru WddSWddSWddSWddSWddS1swYdS) zg Determine if a file is a multi-res cooler file. Returns False if the file doesn't exist. FraN resolutionsrz HDF5::MCOOLT0) rsrtrurkrlrflennextiterrq)rvryrxrprKr#r#r$rs6             rcst|s td|dt|rjdd}|tkrld}tfdd|Ds9t d  WddSd vrtt d  d kr|d  D]}t d |sbWddSqO Wdd SWddSWddSWddS1swYdS) zi Determine if a file is a single-cell cooler file. Returns False if the file doesn't exist. '' is not an HDF5 file.FraN)rbrccellsc3rdr8rergrxr#r$rirjz is_scool_file..z Scool file appears to be corruptrrT) rsrtOSErrorrurkrlrrmrnrorfr}rq)rvrprfrr#rr$ is_scool_files:            r list[str]csxt|s td|dgfdd}t|d}|d|t||WdtS1s3wYtS)z List group paths to all cooler data collections in a file. Parameters ---------- filepath : str Returns ------- list Cooler group paths in the file. rrc,t|r|dsd|n|dSdSNrGrqappend startswithpthr_Zlistingr#r$ _check_cooler z#list_coolers.._check_coolerrrGN)rsrtrrur^rrvrrxr#rr$rs     rcsfdd}t|r9gt|d}|d|t||Wdn1s'wYdvr5dtStd|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. crrrrrr#r$rrz'list_scool_cells.._check_coolerrrGNrz' is not a scool file.)rrsrur^removerrrr#rr$list_scool_cellss    rcst|\}}t|std|dgfdd}t|d}|||t|||WdS1s9wYS)z Get all groups and datasets in an HDF5 file. Parameters ---------- uri : str Returns ------- list Group and dataset paths. rrcs&|dsd|dS|dSr)rrrrr#r$ _check_all s&zls.._check_allrN)rrsrtrrur^)rrrvrwrrxr#rr$lss     rsrc_uridst_uri overwritelinkrename soft_linkNonec Cst|\}}t|\}} t|||gdkrtdtj|r!|r$d} nd} ||kr-d} nd} t|| } t|| o} ||krb|sE|rQ| || | <|rP| |=nS|r[t|| | <nI| || nB|rht d|rst ||| | <n1| dkr| | D] }| |d|| |q}| | j | |j n | || | dkr| ndWdn1swYWddSWddS1swYdS)Nr z:Must provide at most one of: "link", "rename", "soft_link"wzr+rz,Can't hard link between two different files.rG)rsumr1ospathisfilersruZSoftLinkcopyrZ ExternalLinkrfrkupdate)rrrrrrZsrc_pathZ src_groupZdst_pathZ dst_groupZdst_write_modeZsrc_write_modesrcdstZsubgrpr#r#r$_copysD    PrFcCst|||dddddS)zICopy a group or dataset from one file to another or within the same file.FrrrNrrrrr#r#r$rErcCst|||dddddS)z/Rename a group or dataset within the same file.FTrNrrr#r#r$rJrrsoftcCst|||| d|ddS)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)rrrrr#r#r$rOsrexpandrc szddlWnty}zt|ddd}~wwddd ddfdd }t||d}||d|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']cSs&|dvrdS|dvr dStd|)N>ZDatasetZArraytable>GrouprufolderzUnknown type: )r1)rr#r#r$_tree_get_iconfs z#tree_widget.._tree_get_iconFrWr3r]r`r ipytree.Nodecsr}||_|sdusttr|jkrd|_nd|_||_fdd| D|_ d|_ |S)NTFcsg|]}|dqS)rr#)rBc)_tree_widget_sublistrr#r$rD~s z=tree_widget.._tree_widget_sublist..) Noder>iconr/r5r4ZopenedrMrKrFnodesdisabled)rWr]rrXrripytreerr$rps    z)tree_widget.._tree_widget_sublistr\T)r]r)rrrrFF)rWr3r]r`rr`rr)r ImportErrorZTreer3add_node)rRrr6errorrXr]r#rr$ tree_widget[s"    rc@s*eZdZdddZddd Zdd d Zd S) TreeTraversalrWr3rr?cC|Sr8)rFr9rWr#r#r$rFzTreeTraversal.get_childrentreecCs|Sr8r#r9rr#r#r$get_rootszTreeTraversal.get_rootrcCrr8)rMrr#r#r$rMrzTreeTraversal.get_textN)rWr3rr?)rr3rr3)rWr3rr)r<rOrPrFrrMr#r#r#r$rs  rc@sJeZdZdZddefdd d ZdddZdddZdddZdddZ 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 FNrRrSrr`r6r7node_clsrcCsJ||_||_||_dddd|_ddddd|_dd d d d|_||_dS) Nr|r )Z horiz_lenZ label_spacer+-|)Z UP_AND_RIGHTZ HORIZONTALZVERTICALZVERTICAL_AND_RIGHTu└u─u│u├)rRrr6 text_kwargs bytes_kwargsunicode_kwargsr)r9rRrr6rr#r#r$r:s zTreeViewer.__init__rbytescCsFtttdd|ji|jd}|j|j|jd}||}|}|SNZgfx)traverseZdrawr\r#) r rr rrrrRr6encode)r9drawerr]rXr#r#r$ __bytes__szTreeViewer.__bytes__rcCs:tttdd|ji|jd}|j|j|jd}||Sr)r rr rrrrRr6)r9rr]r#r#r$ __unicode__s zTreeViewer.__unicode__cCs|Sr8)rr=r#r#r$__repr__rzTreeViewer.__repr__cCs t|j|j|jd}||S)N)rr6)rrRrr6_repr_mimebundle_rr#r#r$rszTreeViewer._repr_mimebundle_)rRrSrr`r6r7rr)rrrN)rr) r<rOrP__doc__r3r:rrrrr#r#r#r$rs   rcs"dfdd t||dddS)NFcsH|jj}ddd|Di}|}|r"|D] }|||<q|S)Nz@attrscSsi|] \}}|t|qSr#)r2)rBkvr#r#r$ sz4read_attr_tree.._getdict..)r)rkitemsrFrM)rWr]rkrXrErY_getdictr#r$rsz read_attr_tree.._getdictr\T)r]F)rQ)rRr6r#rr$read_attr_trees rc Csxddlm}ddl}t|\}}t|d}||}|}|t||||WdS1s5wYdS)Nr)StringIOr) ioryamlrrsrudumprgetvalue) rrr6rrrrRrxr_r&r#r#r$pprint_attr_trees  $rcCsTt|\}}t|d}||}tt||dWdS1s#wYdS)Nrr\)rrsrureprr)rrr6rrRrxr_r#r#r$pprint_data_trees $r)rrrr)r&rrr)r)r*rr+r8)rRrSrTrUr6r7rrV)r_rSrr`)rrrrr`)r )rvrryr5rr`)rvrrr`)rvrrr)rrrrr)rrrrrr`rr`rr`rr`rrr)rrrrrr`rrr) rrrrrr`rr`rr)rRrSrr`r6r7rr)rRrSr6r7rrV)rrrr6r7rr)7 __future__rrrncollections.abcrrnumbersrtypingrrZ simplejsonr!rrr1rsnumpynpZ asciitreer r Zasciitree.traversalr creater rutilrr__all__r%r(r2r3rQr^rqrrrrrrrrrrrrrrrrr#r#r#r$sZ                1  1 H