:>"^¥$ã@sÞdZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl m Z dd lmZdd lmZGd d „d eƒZGd d„deƒZGdd„deƒZdS)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)ÚDrawingÚString)ÚVerticalBarChart)ÚBarChartProperties)ÚTypedPropertyCollection)Ú_writec@sUeZdZdZddd„Zdd„Zdd„Zd d „Zd d „Zd S)ÚDistributionPagez‰Display a grouping of distributions on a page. This organizes Distributions, and will display them nicely on a single page. ZpdfcCs1g|_d|_t|_d|_||_dS)z Initialize.ééN)Ú distributionsÚnumber_of_columnsrÚ page_sizeÚ title_sizeÚ output_format)Úselfr©rú>/tmp/pip-build-ww9dw3qa/biopython/Bio/Graphics/Distribution.pyÚ__init__'s     zDistributionPage.__init__c Csú|j\}}t||ƒ}|j||||ƒtd}|td}|dt}dt} ||t|jƒ} tjtt|j ƒƒt|jƒƒ} || | } |j ||| || | ƒ|j |dt|ƒt |||j ƒS)zêDraw 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_titlerÚfloatrÚmathÚceilÚlenrÚ_draw_distributionsÚ _draw_legendr r) rÚ output_fileÚtitleÚwidthÚheightÚ cur_drawingZ cur_x_posÚ end_x_posZ cur_y_posÚ end_y_posÚ x_pos_changeZ num_y_rowsÚ y_pos_changerrrÚdraw2s   %zDistributionPage.drawcCsIt|d|t|ƒ}d|_|j|_d|_|j|ƒdS)z5Add the title of the figure to the drawing (PRIVATE).ézHelvetica-BoldÚmiddleN)rrÚfontNamerÚfontSizeÚ textAnchorÚadd)rr"rr r!Ú title_stringrrrrRs    zDistributionPage._draw_titlecCsåxÞtt|ƒƒD]Ê}|d|jt|jƒkrVt|jƒ||j}n |j}x{t|ƒD]m} ||j| } |j| } || |} | |} |||}||}| j|| || |ƒqlWqWdS)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)ÚrangeÚintrrrr')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_yr rrrr†szDistributionPage._draw_legendN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rr'rrrrrrrr s  +r c@sIeZdZdZddd„Zdd„Zdd„Zd d „ZdS) ÚBarChartDistributionz6Display the distribution of values as a bunch of bars.NcCsL|dkrg}||_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_titleÚchart_title_sizeÚpadding_percent)rr:rrrr‘s      zBarChartDistribution.__init__c Cs¦tƒ}|jr1|j||j||||ƒ|j||||ƒ\}}} } ||_| |_t||ƒ|_t| | ƒ|_|j |_ t |j dƒ|j _ t|j dƒ|j _xo|j dd…D]Z} t | ƒ|j j krt | ƒ|j _ t| ƒ|j jkrßt| ƒ|j _qßWt|j ƒdkr•d|_ttƒ} d| _tj| _tj| d_| |_|j|ƒdS)z6Draw a bar chart with the info in the specified range.rr N)rr;rÚ_determine_positionÚxÚyÚabsr r!r:ÚdataÚminZ valueAxisZvalueMinÚmaxZvalueMaxrZ groupSpacingr rZ strokeWidthrZgreenZ strokeColorZ fillColorZbarsr-) rr"Ústart_xr2Úend_xÚend_yZ bar_chartZx_startZx_endZy_startZy_endZdata_setÚstylerrrr'¥s4          zBarChartDistribution.drawc Csl|||d}||j||d}t|||ƒ} d| _|j| _d| _|j| ƒdS)z5Add the title of the figure to the drawing (PRIVATE).r(zHelvetica-Boldr)N)r=rr*r<r+r,r-) rr"rrEr2rFrGZx_centerr1r.rrrrÑs   z BarChartDistribution._draw_titlec Csw|j||}|j||}||}||}|jrS|||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<) rrEr2rFrGZ x_paddingZ y_paddingZ new_x_startZ new_x_endZ new_y_startZ new_y_endrrrr>Üs     z(BarChartDistribution._determine_position)r3r4r5r6rr'rr>rrrrr7Žs  , r7c@s.eZdZdZdd„Zdd„ZdS)ÚLineDistributionzßDisplay 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)z Initialize.Nr)rrrrrüszLineDistribution.__init__cCsdS)z2Draw a line distribution into the current drawing.Nr)rr"rEr2rFrGrrrr'szLineDistribution.drawN)r3r4r5r6rr'rrrrrIôs  rI)r6rZreportlab.lib.pagesizesrZreportlab.lib.unitsrZ reportlab.librZreportlab.graphics.shapesrrZ#reportlab.graphics.charts.barchartsrrZreportlab.graphics.widgetbaser Z Bio.Graphicsr Úobjectr r7rIrrrrÚs nf