o Rŀg]@s\dZddlmZddlmZddlmZejejejej ej ej ej dZ GdddZdS) zRepresent information for graphical display. Classes in this module are designed to hold information in a way that makes it easy to draw graphical figures. )colors)ChromosomeSegment)TelomereSegment))r)r)r)r)r )r )c@sXeZdZdZefddZdddZddd Zd d Zd d Z ddZ ddZ ddZ dS)ChromosomeCountsaRepresent a chromosome with count information. This is used to display information about counts along a chromosome. The segments are expected to have different count information, which will be displayed using a color scheme. I envision using this class when you think that certain regions of the chromosome will be especially abundant in the counts, and you want to pick those out. cCsL||_i|_i|_i|_|jD]}d|j|<d|j|<d|j|<q||_dS)aInitialize a representation of chromosome counts. Arguments: - segment_names - An ordered list of all segment names along the chromosome. The count and other information will be added to these. - color_scheme - A coloring scheme to use in the counts. This should be a dictionary mapping count ranges to colors (specified in reportlab.lib.colors). rNr)_names _count_info _label_info _scale_info _color_scheme)self segment_names color_schemenamerV/var/www/html/myenv/lib/python3.10/site-packages/Bio/Graphics/DisplayRepresentation.py__init__,s      zChromosomeCounts.__init__rcCs:z |j||7<WdStytd|ddw)a6Add counts to the given segment name. Arguments: - segment_name - The name of the segment we should add counts to. If the name is not present, a KeyError will be raised. - count - The counts to add the current segment. This defaults to a single count. Segment name not found.NrKeyError)r segment_namecountrrr add_countCs   zChromosomeCounts.add_countNcCs:z |j||<WdStytd|ddw)zDivide the counts for a segment by some kind of scale value. This is useful if segments aren't represented by raw counts, but are instead counts divided by some number. rrNr)rr scale_valuerrrscale_segment_valueRs  z$ChromosomeCounts.scale_segment_valuecCs(||jvr ||j|<dStd|d)zrAdd a label to a specific segment. Raises a KeyError is the specified segment name is not found. rrN)rr)rrlabelrrr add_label]s zChromosomeCounts.add_labelcCs(||jvr ||j|<dStd|d)zSet the scale for a specific chromosome segment. By default all segments have the same scale -- this allows scaling by the size of the segment. Raises a KeyError is the specified segment name is not found. rrN)rrr)rrscalerrr set_scalegs zChromosomeCounts.set_scalecCs.g}|jD]}||j||j|fq|S)a0Retrieve the color and label info about the segments. Returns a list consisting of two tuples specifying the counts and label name for each segment. The list is ordered according to the original listing of names. Labels are set as None if no label was specified. )rappendrr)r order_infoseg_namerrrget_segment_infots z!ChromosomeCounts.get_segment_infocCstt|jD]V}d}|dkrt}d}n|t|jdkr%td}d}nt}|j|}|j|dkr?||j|}||_|j|durL|j||_ |rRd|_ n|j ||_ | |q|S)aAdd the collected segment information to a chromosome for drawing. Arguments: - chromosome - A Chromosome graphics object that we can add chromosome segments to. This creates ChromosomeSegment (and TelomereSegment) objects to fill in the chromosome. The information is derived from the label and count information, with counts transformed to the specified color map. Returns the chromosome with all of the segments added. rrNr) rangelenrrrr_color_from_count fill_colorrr#r%radd)r chromosomeseg_numis_end_segment cur_segmentr)colorrrrfill_chromosomes(    z ChromosomeCounts.fill_chromosomecCsB|jD]\}}||kr||kr|j||fSqtd|d)zHTranslate the given count into a color using the color scheme (PRIVATE).z Count value z# was not found in the color scheme.)r ValueError)rr count_start count_endrrrr-s z"ChromosomeCounts._color_from_count)r)N) __name__ __module__ __qualname____doc__RAINBOW_COLORSrr r"r$r&r*r5r-rrrrr s      -r N)r< reportlab.librBio.Graphics.BasicChromosomerrvioletindigobluegreenyelloworangeredr=r rrrrs