o Nrf@sNddlZddlZddlmZddlZddlmZddlmZddl m Z m Z m Z m Z ddlmZdd lmZmZdd lmZd d Zeejd ddejdddddedejdddeedddejddddejdd dd!d"ejd#d$ed%ejd&d'd(dd)ejd*d+d,dejd-d.edd/ejd0d1dd!d"d2d3ZdS)4N)ceil)apilock)HIGLASS_TILE_DIMlegacy_zoomifypreferred_sequencezoomify_cooler)parse_cooler_uri)cli get_loggerparse_field_paramc Csddlm}tt}|durg}nt|}|d||D]I}|dt|}dt| vr4q | d|z |j |g|ddWq t yi}z|j }|durYd }|d kr_|WYd}~q d}~wwdS) Nr balanceBalancing args: z::resolutions/weightz#Balancing zoom level with bin size coolerargsZ prog_namer)rr__name__shlexsplitdebugstrrCoolerbinsinfomain SystemExitcode) r resolutionsoutfile balance_cmdloggerresurie exit_coder+[/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/cooler/cli/zoomify.pyinvoke_balances.   r-cool_uriZ COOL_PATH)metavarz--nprocz-nz-pzcNumber of processes to use for batch processing chunks of pixels [default: 1, i.e. no process pool])helpdefaulttypez --chunksizez-cz*Number of pixels allocated to each processgcAT)r0r2r1Z show_defaultz --resolutionsz-rakComma-separated list of target resolutions. Use suffixes B or N to specify a progression: B for binary (geometric steps of factor 2), N for nice (geometric steps of factor 10 interleaved with steps of 2 and 5). Examples: 1000B=1000,2000,4000,8000,... 1000N=1000,2000,5000,10000,... 5000N=5000,10000,25000,50000,... 4DN is an alias for 1000,2000,5000N [default: B])r0z --balancez3Apply balancing to each zoom level. Off by default.F)r0Zis_flagr1z--balance-argszAdditional arguments to pass to cooler balance. To deal with space ambiguity, use quotes to pass multiple arguments, e.g. ``--balance-args '--nproc 8 --ignore-diags 3'``. Note that nproc for balancing must be specified independently of zoomify arguments.)r0r2z --base-uriz-izAOne or more additional base coolers to aggregate from, if needed.)r0multiplez--outz-ozOutput file or URIz--fieldzSpecify the names of value columns to merge as ''. Repeat the ``--field`` option for each one. Use ':dtype=' to specify the dtype. Include ',agg=' to specify an aggregation function different from 'sum'.)r0r2r3z--legacyz5Use the legacy layout of integer-labeled zoom levels.c " Cstt} t|\} } | dur| dd} nt| \} } | d|d| d| d|rt|| ||td\}}|rdd lm}|durIg}nt |}| d |t t | D]V\}}| d t|}|t|krzd t|vrzq^| d |d|z |j|g|ddWq^ty}z|j}|durd}|dkr|WYd}~q^d}~wwdSdSt|}|jj}|jrtt|t}|j}n"|ddgdd}|d|d}tt||t}d}|durd}g|}}dd| dDD]u}d|vsd|vr||krtd|dkr*t||d}nM|dkr6t||d}nA|dkrFddgtd|d}n1| dr\t| dd}t||d}n| drrt| dd}t||d}nt|g}|!|qt"|rd d|D}t#|\}} } }!t |}d!d"t#|| D} d#d"t#||!D}!n d$gdd}} }!t$|gt || |||t|| |!d% |rt%||| dSdS)&zu Generate a multi-resolution cooler file by coarsening. COOL_PATH : Path to a COOL file or Cooler URI. Nz.coolz.mcoolzRecursively aggregating ""z Writing to "rr rrz::rzBalancing zoom level z , bin size rrrstartendbcSsg|]}|qSr+)striplower).0sr+r+r, szzoomify..,nzWMap is already < 256 x 256. Provide resolutions explicitly if you want to coarsen more.nicebinaryZ4dniiicSsg|]}t|ddqS)F)Zincludes_colnumr)r:argr+r+r,r<s cSi|] \}}|dur||qSNr+)r:coldtr+r+r, zzoomify..cSrBrCr+)r:rDfr+r+r,rFrGcount)nprocrcolumnsdtypesagg)&rrr replacerrrrrrrreversedlistitemsrrrrr r!r"Z chromsizesvaluessumZbinsizeintrrmeanwarningswarnr endswithextendlenzipr r-)"r.rJ chunksizer#rZ balance_argsfieldlegacyZbase_urioutr&infile_r$Zn_zoomsZ zoom_levelsr%levelr'r(r)r*ZclrZ genome_lengthZmaxresZcurresrZ mean_fragsizeZrstringrZfield_specifiersrKrLrMr+r+r,zoomify1sS                    rd)rrVmathrZclickrZparallelrreducerrr r utilr r rZ_utilrr-commandargumentoptionrTrrdr+r+r+r,s         "