tf $ddlZddlmcmZddlZddlZddlZ ddl m Z m Z m Z ddlmZdZdZej$eZiZddd ed<d d d ed <d d d ed <ddd ed<iZddd ed<ddd ed<dZdZddZdZ ddZidfdZdZy)N)get_quadtree_depth abs2genomic natsorted)ThreadPoolExecutoriMeanmean)namevalueMinminMaxmaxzStandard DeviationstdzMin-MaxminMaxWhiskerwhiskercrtt|tdzDcgc]}d|z c}dddScc}w)Nr)ranger TILE_SIZE) chromsizesxs ^/var/www/html/software/conda/envs/higlass/lib/python3.12/site-packages/clodius/tiles/bigwig.pyget_zoom_resolutionsrs8!"4Z"Ka"OP QqAF QRVTVRV WW Qs 4ctj|}t|j}t j ||}|S)zj TODO: replace this with negspy Also, return NaNs from any missing chromosomes in bbi.fetch )bbirrkeyspdSeries)bwpathr chromosomes chrom_seriess rget_chromsizesr&s<'JJOO-.K99Z(5L c 4|7t|}g}|jD]\}}||t|ggz }n|}|Dcgc]}t|d}}t|t}dgt |gtd|zzt||t td}|Scc}w)a* Get the tileset info for a bigWig file Parameters ---------- bwpath: string The path to the bigwig file from which to retrieve data chromsizes: [[chrom, size],...] A list of chromosome sizes associated with this tileset. Typically passed in to specify in what order data from the bigwig should be returned. Returns ------- tileset_info: {'min_pos': [], 'max_pos': [], 'tile_size': 1024, 'max_zoom': 7 } rrr)min_posmax_pos max_width tile_sizemax_zoomraggregation_modes range_modes)r&itemsintrrsumr.r/)r#rchromsizes_listchromsizecr- tileset_infos rr7r7+s*#F+ %++- 4KE4 D 23 3O 4%)89Ac!A$i9 9!*i8H3 O$h.%." L :sBc.|\}}}}}}}}ttj||z |z } d} |dk(rd} |dk(rd} | dkDrtj| | fntj| } |j|} |j |} || ||g}| tj d}|dk(rOtj|it|d| ddd f<tj|it|d | dddf<n|dk(rtj|it|d| ddd f<tj|it|d | dddf<tj|it|d | dddf<tj|it|d | ddd f<n#tj|it||| dd|| k(r | |zd k7r| dd} | S#t$rtj | ddY| St$rtj | ddY| SwxYw)Nrrrrr)binsmissingr)summaryrrr rr) r1npceilzerosindexvaluesnanrfetchdict IndexErrorKeyError)ar#binsizeraggregation_mode range_modecidstartendn_binsn_dimrr4clenargskwargss r fetch_datarSYsSTPVWj"2JUC #+01 2F EXY%*QY&%!BHHV4DA  %  %uc* RVV4  !iiEfe)DEAadGiiEfe)DEAadG 9 $iiEfe)DEAadGiiEfe)DEAadGiiFff)EFAadGiiEfe)DEAadG99dMd6;K&LMAaD $;4'>Q.#2A H vv! H vv! H s2E"GH5HHcl| t|}t|}||}tt|||} t d5} t| j t | D cgc]} t|||||gt| z c} } dddtj Scc} w#1swY#xYw)N) max_workers) r&rlistrrmaprStupler= concatenate) r# zoom_level start_posend_posrrIrJ resolutionsrHcids_starts_endser6arrayss rget_bigwig_tilerbs#F+ &z2K*%GK IwGH  +  q EE .   *6F Sq'"      >>& !!   sB*#B%= B*%B**B3c (g}|D]z}|jddd}|jdd}|jd}ttt|dd} t |dkDr|dnd} | t vr| nd} | t vr| nd} t|D cgc]} | jdc} }|rE|Dcgc]}|d }}|Dcgc]}t|d}}tj|| }nd}d |vr|d }||vr||}nd}| d}| d}| t|}t|t}td ||z zz}||z}||z}t|||||| | }tj|}|||fgz }}|Scc} wcc}wcc}w) a  Generate tiles from a bigwig file. Parameters ---------- tileset: tilesets.models.Tileset object The tileset that the tile ids should be retrieved from tile_ids: [str,...] A list of tile_ids (e.g. xyx.0.0) identifying the tiles to be retrieved chromsizes_map: {uid: []} A set of chromsizes listings corresponding to the parameters of the tile_ids. To be used if a chromsizes id is passed in with the tile id with the `|cos:id` tag in the tile id chromsizes: [[chrom, size],...] A 2d array containing chromosome names and sizes. Overrides the chromsizes in chromsizes_map Returns ------- tile_list: [(tile_id, tile_data),...] A list of tile_id, tile_data tuples |rNr.r<r :)r@cosr)rIrJ)splitrWrXr1lenr.r/rDr!r"r&rrrbhgfoformat_dense_tile)r#tile_idschromsizes_maprgenerated_tilestile_idtile_option_partstile_no_options tile_id_parts tile_position return_valuerIrJo tile_optionsr6 chromnames chromlengthschromsizes_to_use chromsizes_idr[tile_pos max_depthr,r\r]dense tile_values rtilesrs0O13#MM#.qr2!--,Q/'--c2 SmAa&89: +.}+=+A}Q'v +7;L+Lrs ##==1    8 $%+f=&$)E:%$)E:%$85I% !*X> H"+i@ IX +\- j"B,.$K\Tr'