DUf`ddlZddlmZddlZddlmZddlZ ddl Z ddl m Z ddlmZdZdZdd gZd eeefd efd Z ddZ ddZ ddZdS)N)Union)to_rgb)arropsskybluedimgrayplot_intervalsto_ucsc_colorstringcolorreturnctj|rdSddt|DS)a] Convert any matplotlib color identifier into a UCSC itemRgb color string. Parameters ---------- color : str or tuple Any valid matplotlib color representation (e.g. 'red', 'tomato', '#ff0000', '#ff00', "#ff000055", (1, 0, 0), (1, 0, 0, 0.5)) Returns ------- str A UCSC itemRgb colorstring of the form "r,g,b" where r, g, and b are integers between 0 and 255, inclusive. Notes ----- The alpha (opacity) channel is ignored if represented in the input. Null values are converted to "0", which is shorthand for "0,0,0" (black). Note that BED9+ files with uninformative itemRgb values should use "0" as the itemRgb value on every data line. Examples -------- >>> to_ucsc_colorstring("red") '255,0,0' >>> to_ucsc_colorstring("tomato") '255,99,71' >>> df["itemRgb"] = df["color"].apply(to_ucsc_colorstring) >>> df chrom start end color itemRgb chr1 0 10 red 255,0,0 chr1 10 20 blue 0,0,255 chr2 0 10 green 0,128,0 chr2 10 20 None 0 0,c3TK|]#}tt|dzV$dS)N)strint).0xs I/var/www/html/software/conda/lib/python3.11/site-packages/bioframe/vis.py z&to_ucsc_colorstring..;s4AAaCCLL))AAAAAA)pdisnulljoinr)r s rr r sDL yBsxxAA6%==AAAAAAr333333?c |tn|}|tn|}|tjn|}|t j|||dz z f||z |||dS)N) facecolor edgecolor)DEFAULT_FACECOLORDEFAULT_EDGECOLORpltgca add_patchmplpatches Rectangle)startendlevelrr heightaxs r_plot_intervalr.>s&/%6!!II%.%6!!IIjbBLL  EFQJ& ' %K    rF rc tj|}tj|}|tj||t j||}ntj|}t |ts|tj |g}ntj |}t |ts|tj |g}ntj |}tt|||||D]?\}\} } } } } t| | | | | tj | | zdz | | dd@tjdtj|dz|+tjdtj|dzn!tj|d |d tjd tjd tjg|rdStjgdS) a Plot a collection of intervals. Parameters ---------- starts, ends : np.ndarray A collection of intervals. levels : iterable or None The level of each interval, i.e. the y-coordinate at which the interval must be plotted. If None, it will be determined automatically. labels : str or iterable or None The label of each interval. colors : str or iterable or None. The color of each interval. xlim : (float, float) or None The x-span of the plot. show_coords : bool If True, plot x-ticks. figsize : (float, float) or None. The size of the figure. If None, plot within the current figure. N)figsize)rrcenter)horizontalalignmentverticalalignmentgg?rrF)npasarrayr#figurerstack_intervals isinstancer itertoolscycle enumeratezipr.textylimmaxxlimr$ set_aspect set_frame_onyticksxticks)startsendslevelslabelscolorsrB show_coordsr2ir)r*r+r labels rplot_intervals_arrrOPs"LZ  F :d  D 7#### ~'55F##&#)6>&**((&#)6>&**((09 FD&&&1111  , ,E3ue uc5E::::   H!$,"*     HT26&>>C'((( | rvd||c)**** a$q'"""GIIGII5!!!JrNNN  2rc |dd}|ddd}|D]\}} |j|} t |t t jtj frtj || } n$|d} ntdt|t |t t jtj frtj || } n$|d} ntdt|t |t t jtj frtj || } n9|t |tr|} ntdt|t| j| j| | | ||| t#j|dS) a Plot a collection of intervals, one plot per chromosome. Parameters ---------- df : pandas.DataFrame A collection of intervals. levels : iterable or None The level of each interval, i.e. the y-coordinate at which the interval must be plotted. If None, it will be determined automatically. labels : str or iterable or None The label of each interval. colors : str or iterable or None. The color of each interval. xlim : (float, float) or None The x-span of the plot. show_coords : bool If True, plot x-ticks. figsize : (float, float) or None. The size of the figure. If None, plot within the current figure. chromT)observed)dropNzUnknown type of levels: zUnknown type of labels: zUnknown type of colors: )rIrJrKrBrLr2)groupby reset_indexgroupsto_numpyr:listrSeriesr6ndarrayr7 ValueErrortyperrOr)r*r#title)dfrIrJrKrBrLr2chrom_gbrQchrom_df chrom_indices chrom_levels chrom_labels chrom_colorss rr r sJzz'Dz11H~~4~((0040HHH#!!x .7799 ftRY ; < < H:f--mrpsF$$$$$$ 2 3)BuS%Z0)BS)B)B)B)BZGK*     TTTTr     HHHHHHr