B b$@sdZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl m Z dd lmZdd lmZGd d d ZGd ddZGdddZdS)asDisplay information distributed across a Chromosome-like object. These classes are meant to show the distribution of some kind of information as it changes across any kind of segment. It was designed with chromosome distributions in mind, but could also work for chromosome regions, BAC clones or anything similar. Reportlab is used for producing the graphical output. N)letter)inch)colors)DrawingString)VerticalBarChart)BarChartProperties)TypedPropertyCollection)_writec@s:eZdZdZdddZddZddZd d Zd d Zd S)DistributionPagezDisplay a grouping of distributions on a page. This organizes Distributions, and will display them nicely on a single page. pdfcCs"g|_d|_t|_d|_||_dS)zInitialize the class.N) distributionsnumber_of_columnsr page_size title_size output_format)selfrr8lib/python3.7/site-packages/Bio/Graphics/Distribution.py__init__'s zDistributionPage.__init__c Cs|j\}}t||}|||||td}|td}|dt}dt} ||t|j} ttt|j t|j} || | } | ||| || | | |dt|t |||j S)zDraw out the distribution information. Arguments: - output_file - The name of the file to output the information to, or a handle to write to. - title - A title to display on the graphic. g?g?g@)rr _draw_titlerfloatrmathZceillenr_draw_distributions _draw_legendr r) rZ output_filetitlewidthheight cur_drawingZ cur_x_pos end_x_posZ cur_y_pos end_y_pos x_pos_changeZ num_y_rows y_pos_changerrrdraw2s     zDistributionPage.drawcCs6t|d|t|}d|_|j|_d|_||dS)z5Add the title of the figure to the drawing (PRIVATE).zHelvetica-BoldmiddleN)rrfontNamerfontSize textAnchoradd)rr!rrr title_stringrrrrRs zDistributionPage._draw_titlec Csxtt|D]}|d|jt|jkr@t|j||j}n|j}xbt|D]V} ||j| } |j| } || |} | |} |||}||}| || || |qPWqWdS)aDraw all of the distributions on the page (PRIVATE). Arguments: - cur_drawing - The drawing we are working with. - start_x_pos - The x position on the page to start drawing at. - x_pos_change - The change in x position between each figure. - start_y_pos - The y position on the page to start drawing at. - y_pos_change - The change in y position between each figure. - num_y_drawings - The number of drawings we'll have in the y (up/down) direction. r N)rangeintrrrr&)rr!Z start_x_posr$Z start_y_posr%Znum_y_drawingsZ y_drawingZnum_x_drawingsZ x_drawingZdist_numZcur_distributionZx_posr"r#y_posrrrr[s   z$DistributionPage._draw_distributionscCsdS)zqAdd a legend to the figure (PRIVATE). Subclasses can implement to provide a specialized legend. Nr)rr!start_yrrrrrszDistributionPage._draw_legendN)r ) __name__ __module__ __qualname____doc__rr&rrrrrrrr s    +r c@s2eZdZdZd ddZddZddZd d ZdS) BarChartDistributionz6Display the distribution of values as a bunch of bars.NcCs4|dkr g}||_d|_d|_d|_d|_d|_dS)a'Initialize a Bar Chart display of distribution info. Attributes: - display_info - the information to be displayed in the distribution. This should be ordered as a list of lists, where each internal list is a data set to display in the bar chart. N g333333?) display_infoZ x_axis_titleZ y_axis_title chart_titlechart_title_sizepadding_percent)rr9rrrrs zBarChartDistribution.__init__c Cs*t}|jr"|||j|||||||||\}}} } ||_| |_t|||_t| | |_|j |_ t |j d|j _ t|j d|j _xN|j ddD]<} t | |j j krt | |j _ t| |j jkrt| |j _qWt|j dkrd|_tt} d| _tj| _tj| d_| |_||dS)z6Draw a bar chart with the info in the specified range.rr N)rr:r_determine_positionxyabsrr r9dataminZ valueAxisZvalueMinmaxZvalueMaxrZ groupSpacingr rZ strokeWidthrZgreenZ strokeColorZ fillColorZbarsr,) rr!start_xr1end_xend_yZ bar_chartZx_startZx_endZy_startZy_endZdata_setZstylerrrr&s4  zBarChartDistribution.drawc CsT|||d}||j||d}t|||} d| _|j| _d| _|| dS)z5Add the title of the figure to the drawing (PRIVATE).r'zHelvetica-Boldr(N)r<rr)r;r*r+r,) rr!rrDr1rErFZx_centerr0r-rrrrs z BarChartDistribution._draw_titlec Cs^|j||}|j||}||}||}|jrB|||j} n||} ||} ||| | fS)a-Calculate the position of the chart with blank space (PRIVATE). This uses some padding around the chart, and takes into account whether the chart has a title. It returns 4 values, which are, in order, the x_start, x_end, y_start and y_end of the chart itself. )r<r:r;) rrDr1rErFZ x_paddingZ y_paddingZ new_x_startZ new_x_endZ new_y_startZ new_y_endrrrr=sz(BarChartDistribution._determine_position)N)r2r3r4r5rr&rr=rrrrr6s  , r6c@s eZdZdZddZddZdS)LineDistributionzDisplay the distribution of values as connected lines. This distribution displays the change in values across the object as lines. This also allows multiple distributions to be displayed on a single graph. cCsdS)zInitialize the class.Nr)rrrrrszLineDistribution.__init__cCsdS)z2Draw a line distribution into the current drawing.Nr)rr!rDr1rErFrrrr&szLineDistribution.drawN)r2r3r4r5rr&rrrrrGsrG)r5rZreportlab.lib.pagesizesrZreportlab.lib.unitsrZ reportlab.librZreportlab.graphics.shapesrrZ#reportlab.graphics.charts.barchartsrrZreportlab.graphics.widgetbaser Z Bio.Graphicsr r r6rGrrrrs       nf