DUf ddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z e jejd d ejd d deddejdddddeejdddeeddejddedejdddd ejd!d"dd#d$%d&ZdS)'N)lock)coarsen_cooler)parse_cooler_uri)cliparse_field_paramcool_uri COOL_PATH)metavarz--factorz-kzyGridding factor. The contact matrix is coarsegrained by grouping each chromosomal contact block into k-by-k element tilesT)helptypedefault show_defaultz--nprocz-nz-pzcNumber of processes to use for batch processing chunks of pixels [default: 1, i.e. no process pool])rrrz --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'.)rrmultiplez--outz-ozOutput file or URI)requiredrz--appendz-aFz_Pass this flag to append the output cooler to an existing file instead of overwriting the file.)is_flagrrc t|\}}t|\} }tj|tj| k} t|rOd|D} t | \} }} }dt | | D} dt | |D}ndgdd}} } t |||||| | || rt nd|rdnd 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. c0g|]}t|dS)F)includes_colnumr ).0args O/var/www/html/software/conda/lib/python3.11/site-packages/cooler/cli/coarsen.py zcoarsen..Is3   >A c5 9 9 9   ci|] \}}||| SN)rcoldts r zcoarsen..MsPPPgc2#rrci|] \}}||| Srr)rr fs rr"zcoarsen..NsGGG&#qsArcountNaw) chunksizenproccolumnsdtypesaggrmode)roprealpathlenziprr)r factorr)r(fieldoutappendinfile_outfile same_filefield_specifiersr*r+r,s rcoarsenr; s&p!**IFA!#&&JGQ F##r{7';';;I 5zz 5  EJ   #&'7"8FCPPWf)=)=PPPGGC$5$5GGG!(y$   (TTD #SS      r)os.pathpathr.clickparallelrreducerutilrr_utilr commandargumentoptionintstrr;rrrrIs ############$$$$$$  K000  ?   )  5 CII   L gtd1EFFF    ' %%GF10\%%%r