o RÅ€gèã@s²dZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd l m Z dd l m Z dd l mZdd lmZdd lmZddlmZGdd„dƒZdS)z÷Plots to compare information between different sources. This file contains high level plots which are designed to be used to compare different types of information. The most basic example is comparing two variables in a traditional scatter plot. é)ÚLinePlot)ÚmakeEmptyCircle)ÚmakeEmptySquare)ÚmakeFilledCircle)ÚmakeFilledDiamond)ÚmakeFilledSquare)Ú makeSmiley)ÚDrawing)ÚString)Úcolors)Úletter)Úinch)Ú_writec@sBeZdZdZddd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dS)ÚComparativeScatterPlota6Display a scatter-type plot comparing two different kinds of info. Attributes; - display_info - a 2D list of the information we'll be outputting. Each top level list is a different data type, and each data point is a two-tuple of the coordinates of a point. So if you had two distributions of points, it should look like:: display_info = [[(1, 2), (3, 4)], [(5, 6), (7, 8)]] If everything is just one set of points, display_info can look like:: display_info = [[(1, 2), (3, 4), (5, 6)]] ÚpdfcCsRd|_t|_d|_||_g|_tjtjtj tj tj tj g|_ ttttttg|_dS)zInitialize the class.ééN)Únumber_of_columnsr Ú page_sizeÚ title_sizeÚ output_formatÚ display_infor ÚredÚgreenÚblueÚyellowÚorangeÚblackÚ color_choicesrrrrrrÚ shape_choices)Úselfr©r!úL/var/www/html/myenv/lib/python3.10/site-packages/Bio/Graphics/Comparative.pyÚ__init__1s&ú  úzComparativeScatterPlot.__init__c Csl|j\}}t||ƒ}| ||||¡td}|td}|dt}dt} | ||| ||¡t|||jƒS)zéWrite the comparative plot to a file. 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ø?)rr Ú _draw_titler Ú_draw_scatter_plotrr) r Ú output_fileÚtitleÚwidthÚheightÚ cur_drawingÚstart_xÚend_xÚend_yÚstart_yr!r!r"Ú draw_to_fileOs   z#ComparativeScatterPlot.draw_to_filecCs6t|d|t|ƒ}d|_|j|_d|_| |¡dS)z4Add a title to the page we are outputting (PRIVATE).ézHelvetica-BoldÚmiddleN)r r ÚfontNamerÚfontSizeÚ textAnchorÚadd)r r*r'r(r)Ú title_stringr!r!r"r$es z"ComparativeScatterPlot._draw_titlec Cs¬tƒ}||_||_t||ƒ|_t||ƒ|_|j|_d|_|  |j¡\}}} } ||j _ ||j _ ||d|j _ | |j_ | |j_ | | d|j_ | ||j¡| |¡dS)zHDraw a scatter plot on the drawing with the given coordinates (PRIVATE).rg$@N)rÚxÚyÚabsr(r)rÚdataÚ joinedLinesÚ _find_min_maxÚ xValueAxisÚvalueMinÚvalueMaxÚ valueStepÚ yValueAxisÚ_set_colors_and_shapesr5) r r*Úx_startÚy_startÚx_endÚy_endÚ scatter_plotÚx_minÚx_maxÚy_minÚy_maxr!r!r"r%ns z)ComparativeScatterPlot._draw_scatter_plotcCsrtt|ƒƒD]0}|dt|jƒkr$|j||j|_|j||j|_q|jd|j|_|jd|j|_qdS)aNSet the colors and shapes of the points displayed (PRIVATE). By default this just sets all of the points according to the order of colors and shapes defined in self.color_choices and self.shape_choices. The first 5 shapes and colors are unique, the rest of them are just set to the same color and shape (since I ran out of shapes!). You can change how this function works by either changing the values of the color_choices and shape_choices attributes, or by inheriting from this class and overriding this function. réÿÿÿÿN)ÚrangeÚlenrÚlinesÚ strokeColorrÚsymbol)r rGrÚ value_numr!r!r"rBŠs  ÿöz-ComparativeScatterPlot._set_colors_and_shapesc Cs”|ddd}|ddd}|ddd}|ddd}|D]!}|D]\}}||kr0|}||kr6|}||kr<|}||krB|}q&q"||||fS)zEFind min and max for x and y coordinates in the given data (PRIVATE).rrr!) r ÚinforHrIrJrKÚ two_d_listr7r8r!r!r"r<£s" €ø z$ComparativeScatterPlot._find_min_maxN)r) Ú__name__Ú __module__Ú __qualname__Ú__doc__r#r/r$r%rBr<r!r!r!r"rs   rN)rXÚ#reportlab.graphics.charts.lineplotsrÚ!reportlab.graphics.charts.markersrrrrrrÚreportlab.graphics.shapesr r Ú reportlab.libr Úreportlab.lib.pagesizesr Úreportlab.lib.unitsr Ú Bio.Graphicsrrr!r!r!r"Ús