o Nrf @sddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z e ejd d d ejd d dedddejdddddedejdddeedddejddeddejddddd ejd!d"dd#d$d%d&d'ZdS)(N)lock)coarsen_cooler)parse_cooler_uri)cliparse_field_paramcool_uriZ COOL_PATH)metavarz--factorz-kzyGridding factor. The contact matrix is coarsegrained by grouping each chromosomal contact block into k-by-k element tilesT)helptypedefaultZ show_defaultz--nprocz-nz-pzcNumber of processes to use for batch processing chunks of pixels [default: 1, i.e. no process pool])r rr z --chunksizez-cz*Number of pixels allocated to each processgcAz--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'.)r r Zmultiplez--outz-ozOutput file or URI)requiredr z--appendz-aFz_Pass this flag to append the output cooler to an existing file instead of overwriting the file.)Zis_flagrr c Cst|\}}t|\} }t|t| k} t|r>dd|D} t| \} }} }ddt| | D} ddt| |D}n dgdd} } }t|||||| | || rTtnd|rYdnd d dS) z Coarsen a cooler to a lower resolution. Works by pooling *k*-by-*k* neighborhoods of pixels and aggregating. Each chromosomal block is coarsened individually. COOL_PATH : Path to a COOL file or Cooler URI. cSsg|]}t|ddqS)F)Zincludes_colnumr).0argr[/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/cooler/cli/coarsen.py Is zcoarsen..cSi|] \}}|dur||qSNr)rcoldtrrr Mzcoarsen..cSrrr)rrfrrrrNrcountNaw) chunksizenproccolumnsdtypesaggrmode)roprealpathlenziprr)r factorr rfieldoutappendinfile_outfileZ same_fileZfield_specifiersr!r"r#rrrcoarsen s. 8    r0)Zos.pathpathr%ZclickZparallelrreducerutilrrZ_utilr commandargumentoptionintstrr0rrrrs`