DUfddlZejejddlmZddlZddlZddl Z ddl m Z ddl m Z dZe dd efd ZdS) N)level)partial)is_cooler_balanced)pool_decoratorcg}|rd}nd}||dd||}|d|dg<|d|dg<|d |d g<|gd d |}d dd|g|_|jddkr||||dd||}|d|dg<|d|dg<|d |dg<|gdd |}d dd|g|_|jddkr||t|dkrtj d dd|gStj |dS)NbalancedcountT)balance as_pixelsjoinrchrom2start2rend2)chrom1start1end1)observedchromstartendrrr)rrr)columns ignore_index) matrixfetchgroupbymean reset_indexrshapeappendlenpd DataFrameconcat)rclrclr_weight_name viewpoint to_returncolnamepxls1pxls2s T/var/www/html/software/conda/lib/python3.11/site-packages/cooltools/api/virtual4c.py_extract_profiler/sI JJ$TJ J J P P y  E"! E8*!! E8*lE6(O  222T BB7K   gug6EM {1~ JJ$TJ J J P P5  E"! E8*!! E8*lE6(O 222T BB7K   gug6EM {1~ 9~~|Wgug$FGGGGy6666weightrc|dvr= t||d}n&#t$r}td|d|d}~wwxYwd}nd}tjj|}tt||| }t|||j } tj | d } | j d d krLtjd ||ddgd } t"j| |<n?| dd| d<| dt(| d<| dt(| d<tj| tj|jt"j| j| d|d k| d|dkz| d|dkz|f<tj|ddgd | gd dd} | S)a Generate genome-wide contact profile for a given viewpoint. Extract all contacts of a given viewpoint from a cooler file. Parameters ---------- clr : cooler.Cooler A cooler with balanced Hi-C data. viewpoint : tuple or str Coordinates of the viewpoint. clr_weight_name : str Name of the column in the bin table with weight nproc : int, optional How many processes to use for calculation. Ignored if map_functor is passed. map_functor : callable, optional Map function to dispatch the matrix chunks to workers. If left unspecified, pool_decorator applies the following defaults: if nproc>1 this defaults to multiprocess.Pool; If nproc=1 this defaults the builtin map. Returns ------- v4C_table : pandas.DataFrame A table containing the interaction frequency of the viewpoint with the rest of the genome Note ---- Note: this is a new (experimental) function, the interface or output might change in a future version. )NFT) raise_errorsz#provided cooler is not balanced or z is missingNr r )r'r(r)rrz No contacts found for viewpoint )rrrrcategoryrr)view_dfrrleft)onhow)drop)r Exception ValueErrorbioframecore stringops parse_regionrr/list chromnamesr$r&r!loggingwarnbinsnpnanastypeint sort_bedframemake_viewframe chromsizeslocmergedrop_duplicatesr ) r'r)r(nproc map_functor_er+fcountsv4cs r. virtual4crV:s|Lm++ "3dKKKAA   RoRRR   '44Y??Ic?i   A++a00 1 1F )F . . .C y|q C CCDDDhhjjm5556vG 7|**:66G 7|**3//G Z&&s++E  +CN;;     F  \Yq\ )7|y|+ -5zYq\) +    H 111 7778,,,     _   [d[ # #  Js <7<)rB basicConfigINFO functoolsrnumpyrEpandasr$r< lib.checksr lib.commonrr/maprVr0r.r`s','''',+++++'''''''7'7'7R  WWWWWWr0