o RŀgU$@sdZddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd lmZdd lmZGd d d ZGdddZGdddZdS)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)BarChartProperties)VerticalBarChart)Drawing)String)TypedPropertyCollection)colors)letter)inch)_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)Initialize the class.N) distributionsnumber_of_columnsr page_size title_size output_format)selfrrM/var/www/html/myenv/lib/python3.10/site-packages/Bio/Graphics/Distribution.py__init__&s  zDistributionPage.__init__c Cs|j\}}t||}|||||td}|td}|dt}dt} |||j} tt|j|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_titler rmathceillenr_draw_distributions _draw_legendr r) r output_filetitlewidthheight cur_drawing cur_x_pos end_x_pos cur_y_pos end_y_pos x_pos_change num_y_rows y_pos_changerrrdraw1s     zDistributionPage.drawcCs6t|d|t|}d|_|j|_d|_||dSz5Add the title of the figure to the drawing (PRIVATE).zHelvetica-BoldmiddleN)rr fontNamerfontSize textAnchoradd)rr#r r!r" title_stringrrrrOs zDistributionPage._draw_titlec Cstt|D]L}|d|jt|jkrt|j||j}n|j}t|D]+} ||j| } |j| } || |} | |} |||}||}| || || |q&qdS)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. rN)rangeintrrrr+)rr# start_x_posr( start_y_posr*num_y_drawings y_drawingnum_x_drawings x_drawingdist_numcur_distributionx_posr%r'y_posrrrrXs    z$DistributionPage._draw_distributionscCdS)zqAdd a legend to the figure (PRIVATE). Subclasses can implement to provide a specialized legend. Nr)rr#start_yr!rrrrzDistributionPage._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|durg}||_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_info x_axis_title y_axis_title chart_titlechart_title_sizepadding_percent)rrJrrrrs  zBarChartDistribution.__init__c Cs$t}|jr|||j|||||||||\}}} } ||_| |_t|||_t| | |_|j |_ t |j d|j _ t|j d|j _|j ddD]} t | |j j kr^t | |j _ t| |j jkrlt| |j _qNt|j dkrd|_tt} d| _tj| _tj| d_| |_||dS)z6Draw a bar chart with the info in the specified range.rrN)rrMr_determine_positionxyabsr!r"rJdatamin valueAxisvalueMinmaxvalueMaxr groupSpacingrr strokeWidthrgreen strokeColor fillColorbarsr2) rr#start_xrAend_xend_y bar_chartx_startx_endy_starty_enddata_setstylerrrr+s:    zBarChartDistribution.drawc CsT|||d}||j||d}t|||} d| _|j| _d| _|| dSr,)rOrr/rNr0r1r2) rr#r r`rArarbx_centerr?r3rrrrs z BarChartDistribution._draw_titlec Cs^|j||}|j||}||}||}|jr!|||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. )rOrMrN) rr`rArarb x_padding y_padding new_x_start new_x_end new_y_start new_y_endrrrrPs z(BarChartDistribution._determine_position)N)rCrDrErFrr+rrPrrrrrGs  , rGc@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. cCr@)r Nr)rrrrrrBzLineDistribution.__init__cCr@)z2Draw a line distribution into the current drawing.Nr)rr#r`rArarbrrrr+rBzLineDistribution.drawN)rCrDrErFrr+rrrrrqs rq)rFr#reportlab.graphics.charts.barchartsrrreportlab.graphics.shapesrrreportlab.graphics.widgetbaser reportlab.librreportlab.lib.pagesizesrreportlab.lib.unitsr Bio.Graphicsr r rGrqrrrrs         kf