U
ñad‡h ã @ s~ d dl mZ d dlZd dlZd dlZd dlmZ d dlmZm Z m
Z
m Z zd dlm Z W n e
k
rx d dlm Z Y nX d dlmZmZ d dlmZ d dlmZmZ d d lmZ d d
lmZmZm Z d d l m Z m Z m Z d d l!m"Z" d
Z#dZ$dZ%dZ&dZ'dd„ Z(dd„ Z)dd„ Z*dd„ Z+dd„ Z,dd „ Z-d d „ Z.d d „ Z/d!d"„ Z0d#d$„ Z1d%d&„ Z2d'd(„ Z3d)d*„ Z4d+d,„ Z5d-d.„ Z6d/d0„ Z7dS )1é )Úwith_statementN)Ú defaultdict)ÚjoinÚexistsÚdirnameÚrealpath)Ú OrderedDict)ÚqutilsÚqconfig)Úget_aux_out_fpaths)Úcreate_minimap_output_dirÚ parse_cs_tag)Ú get_chr_lengths_from_fastafile)Úget_assembliesÚcheck_misassembled_blocksÚ Alignment)Úget_path_to_programÚis_non_empty_fileÚrelpath)ÚCOVERAGE_FACTORz
circos.pngg{®Gáz¤?g¸ …ëQ¸®?éPà c
C sž t | ¡ ƒ}d}t|dƒ}t|dƒH}| ¡ D ]8\}}| d dd||dt|ƒdg¡d ¡ t||ƒ}q.W 5 Q R X t|d
ƒ}t|dƒ}| d ¡ | d ¡ tj r¸|d
kr¸| d¡ n2|dkrÌ| d¡ n |dkrà| d¡ n
| d¡ | d¡ | d¡ | d¡ | d¡ | d¡ | d¡ | d¡ | d¡ | d ¡ | d ¡ | d ¡ | d ¡ | d ¡ | d!¡ | d"¡ | d#¡ W 5 Q R X |||fS )$Nr zreference.karyotype.txtÚwú Úchrú-Ú0ZlgreyÚ
z
ideogram.confz
z
é z
default = 0r
é zdefault = 0.005r
éd zdefault = 0.001r
zdefault = 0.0005r
zbreak = 0.005r
z
zthickness = 30p
zstroke_thickness = 2
zstroke_color = black
z fill = yes
zradius = 0.85r
úshow_label = no
zlabel_font = default
z-label_radius = dims(ideogram,radius) + 0.05r
zlabel_size = 36
zlabel_parallel = yes
zband_stroke_thickness = 2
zshow_bands = yes
zfill_bands = yes
z )
ÚlenÚkeysr ÚopenÚitemsÚwriteÚstrÚmaxr
Z
prokaryote) Ú chr_lengthsÚ
output_dirZnum_chromosomesÚmax_lenÚkaryotype_fpathÚout_fÚnameÚseq_lenÚideogram_fpath© r0 ú0lib/python3.8/site-packages/quast_libs/circos.pyÚcreate_ideogram$ sF
&
r2 c C sæ t |dƒ}t|dƒÆ}| d¡ | d¡ | d¡ | d¡ | d¡ | d¡ | d ¡ | d
¡ | d ¡ | d ¡ | d
¡ | d¡ | d¡ | d dkr¸d| d }d}n d| }d}| dt|ƒ d ¡ | d¡ | d¡ | d¡ | d¡ | d¡ | d ¡ | d ¡ | d¡ | d ¡ | d ¡ | d ¡ | d!¡ | d"¡ | d ¡ | d ¡ | d¡ | d#¡ | d ¡ | d$¡ | d!¡ | d%¡ | d&| ¡ | d ¡ | d ¡ | d'¡ W 5 Q R X |S )(Nz
ticks.confr zshow_ticks = yes
zshow_tick_labels = yes
zshow_grid = no
z
zskip_first_label = yes
zskip_last_label = no
z%radius = dims(ideogram,radius_outer)
ztick_separation = 2p
z min_label_distance_to_edge = 0p
zlabel_separation = 5p
zlabel_offset = 5p
zlabel_size = 12p
zthickness = 3p
é
é@B g ð?ZMbpZkbpzlabel_multiplier = r z
z
spacing = 1u
zcolor = dgrey
z size = 12p
r z format = %s
z
z
spacing = 5u
zcolor = black
z size = 18p
zshow_label = yes
zlabel_size = 24p
zspacing = 10u
z size = 24p
zlabel_size = 32p
zsuffix = " %s"
z)r r# r% r& )Ú chrom_unitsr) Ú ticks_fpathr, Zlabel_multiplierÚsuffixr0 r0 r1 Úcreate_ticks_confL s^
r8 c C sÊ t |dƒ}dddg}t|dƒ¢}tƒ }ddlm} |j ¡ D ]&\}}||krTg ||< || |¡ qÄ s z$parse_alignments..©NN)r Úlabel_from_fpath_for_fnamerU r r; r' r Ú
assemblies)Úcontigs_fpathsÚcontig_report_fpath_patternZlists_of_aligned_blocksZ
contigs_fpathrP rQ rR Z max_contigsr0 r0 r1 Úparse_alignments· s r] c C s( t || jd ƒ}|d }t|dƒü}d }| jD ]º}d|_|jrFd|_n |jrRd|_|r°|j|jkr°|j|jkr°t|j |j ƒt
|j |j ƒ |k r°t
|j |j ƒ|_ t|j |j ƒ|_ q.|rä| d |jt
|j ƒt
|j ƒd|j g¡d ¡ |}q.| d |jt
|j ƒt
|j ƒd|j g¡d ¡ W 5 Q R X |S )
Nz.confr r ZgreenZredZppurpler zcolor=r )r Úlabelr# Z
alignmentsZcolorZ misassembledrO rG r' rE ÚminrF r% r& )ÚassemblyÚref_lenr) Ú
conf_fpathZmax_gapr, Z
prev_alignZalignr0 r0 r1 Úcreate_alignment_plotsÊ s(
ÿ0:rc c C s| g }t | ƒ&}|D ]}| t| ¡ d ƒ¡ qW 5 Q R X t|ƒ}tt|ƒƒtt|ƒƒ }}t|dƒ}t
| |¡ ||||fS )Néÿÿÿÿzgc.txt) r# r; ÚfloatrK r! rN r_ r' r ÚshutilÚcopy) Úgc_fpathÚdata_dirZ gc_valuesÚfrS Ú
max_pointsÚmin_gcÚmax_gcZ dst_gc_fpathr0 r0 r1 Úcreate_gc_plotá s
"
rn c C sx d}| sd |fS t ƒ }t|dƒ}t| ¡ ƒ}t| ƒ¤}d}t|ƒD ]\} }
|
¡ } |
d¡r¨d}| d dd … } t| d ƒd }
||
}dd„ t || d ƒD ƒ|| < qBt| d ƒ}|| ||
|¡ |t 7 }qBW 5 Q R X t|d ƒ‚}| ¡ D ]r\} }t|ƒD ]`\}}|r t
|ƒt|ƒ nd}| d
| t|| ƒt|d | ƒt|ƒg¡d ¡ |d7 }qqòW 5 Q R X ||fS ) Nr z coverage.txtú#r c S s g | ]}g ‘qS r0 r0 )rV r@ r0 r0 r1 rW þ s z(create_coverage_plot..é rd r r r )Údictr rI Úvaluesr# r< rK Ú
startswithrN Úranger; r r$ Úsumr! r% r& )Ú cov_fpathÚ window_sizer( r) rk Z cov_by_chromÚcov_data_fpathrj ÚposrL rS Zfsr> Z chrom_orderZ chrom_lenZdepthr, Z
depth_listr@ ZdepthsZ avg_depthr0 r0 r1 Úcreate_coverage_plotí s4
4rz c s t | j¡}t|dtjƒ}tt|ƒ|ƒ}t|ƒ\}} }}t| ƒrFtj sJd S t||d ƒ}
t
‡ ‡fdd„ƒ} t | ƒ¬} | D ] }
t |
d¡d
¡ d ƒ}|
¡ d ¡ }|
¡ d ¡ }|}t|ƒD ]T}t|ƒd }| d
¡r| | t |ƒˆ d 7 < |d 7 }qÂ| d ¡sÂ||7 }qÂqxW 5 Q R X t |
d ƒæ}| ¡ D ]Ö\}}d
\}}t|ƒD ]\}}|dkrp|d ˆ }nn|rš| d |t|ƒt|ƒdg¡d ¡ | d |t|ˆ ƒt|d ˆ ƒt|ƒg¡d ¡ |d ˆ }d }qP|r8| d |t|ƒt|ƒdg¡d ¡ q8W 5 Q R X |
S )Nz..z.mismatches.txtc s dgˆ ˆ d S ©Nr r r0 r0 ©ra rw r0 r1 Ú ó z(create_mismatches_plot..ú|r é rd r Ú*ú+r )r r r r r )r rY Zfpathr r
Z detailed_contigs_reports_dirnamer r r Z show_snpsr r# rN rK rM r
r! rs r$ r< r% r& )r` rw ra Zroot_dirr) Zassembly_labelZaligner_dirpathZcoords_basenameÚ_Zcoords_filtered_fpathZmismatches_fpathZmismatch_density_by_chromZ coords_filerS Ús1r> ZcigarZref_posÚopZn_basesr, Z density_listrE rF r@ Údensityr0 r| r1 Úcreate_mismatches_plot
sF
$4 2r‡ c
s’ g }d}| s||fS | D ]n}t ||jd ƒ}t|jƒdkr>qd}t‡ ‡fdd„ƒ} t|dƒ}
|jD ]’} | jr„| j|jkr„| jn| j} | |jkrž|j| nd } | s¨qht | j
ˆ t | j ˆ d t| | ƒƒƒD ](}
|
t| | ƒk rÐ| | |
d7 < qÐqh|
¡ D ]Z\} }t|ƒD ]F\}
}|
d | t|
ˆ ƒt|
d ˆ ƒt|ƒg¡d ¡ |d7 }qqW 5 Q R X t|ƒr~| |¡ t||ƒ}q||fS ) Nr z.txtc s dgˆ ˆ d S r{ r0 r0 r| r0 r1 r} A r~ z#create_genes_plot..r r r r )r Úkindr! Ú region_listr r# Z
chromosomeZchr_names_dictZseqnamert rE r_ rF r$ r< r% r& r r; r' )Úfeatures_containersrw ra r) Úfeature_fpathsrk Úfeature_containerÚ
feature_fpathZ
num_pointsZgene_density_by_chromr, Zregionr> r@ Zgene_density_listr† r0 r| r1 Úcreate_genes_plot5 s:
ÿ,4
rŽ c
C sT t |dƒ}t|dƒ6}| ¡ D ]&\}}| d |dt|ƒg¡d ¡ q W 5 Q R X |S )Nz
genome.txtr r r r )r r# r$ r% r& )r( r) Z genome_fpathr, r- r. r0 r0 r1 Úcreate_genome_fileV s
*r c
C s t |dƒ}g }d}t|ƒD ]*\}} | dt|d ƒ |f¡ |d7 }q|D ]*}
t|
jƒdkrJ| |
j|f¡ |d7 }qJ|rˆ| d|f¡ t|dƒ4} | t |
¡ ƒd d d d d
„ |D ƒ¡ ¡ W 5 Q R X t |d ƒ} t| dƒ } | d t ||ƒ d
¡ W 5 Q R X | |fS )Nz
labels.txtr r` r Zcoverager z
0 0 null ú,c S s g | ]\}}d ||f ‘qS )z
track%d=%sr0 )rV r^ r@ r0 r0 r1 rW m s z!create_labels..z
label.confzRz = 10
type = text
label_size = 30p
label_font = bold
label_parallel = yes
file = zÍ
r0 = eval(sprintf("%fr+5p", conf(conf(., track_idx)_pos)))
r1 = eval(sprintf("%fr+500p", conf(conf(., track_idx)_pos)))
condition = 1
value = eval(var(conf(., track_idx)))
) r r< r; r& r! r‰ rˆ r# r% rI r" r )
r( rZ rŠ Úcoverage_fpathr) Zlabels_txt_fpathÚ track_labelsÚplot_idxr@ r` rŒ r, Zlabels_conf_fpathr0 r0 r1 Ú
create_labels^ s,
8
ûûr” c C s@ | dkrd}n.| dkr d}n | dkr*d}n| dkr8d}nd }|S )
Ni eÍ i N i £áé' i áõiˆ r4 éè r r0 r| r0 r0 r1 Úset_window_size s r— c
C s¸ t |dƒ}t|dƒš}| dtjr&dnd||f ¡ tjrD| d¡ | d¡ t| ƒD ] \}} | d|d | jf ¡ qV| d
¡ |r| d ¡ |rª| d |r¢d
nd ¡ W 5 Q R X |S )Nz
legend.txtr zn1) The outer circle represents reference sequence%s with GC (%%) heatmap [from %d%% (white) to %d%% (black)].
ÚsrB z=Color bars help to distinguish between different references.
z
2) Assembly tracks:
z assembly%d - %s
r zjAssembly tracks are combined with mismatches visualization: higher columns indicate larger mismatch rate.
zK
3) User-provided genes. A darker color indicates higher density of genes.
z:
%d) The inner circle represents read coverage histogram.
é é )r r# r% r
Úis_combined_refr< r^ )
rZ rl rm rŠ r‘ r) Z legend_fpathr, r@ r` r0 r0 r1 Ú
create_legend s
ÿ
rœ c ' sP t ˆdƒ‰ tˆ ƒs t ˆ ¡ t| ƒ}t|ˆ ƒ\} }
} | dkrBd} n| dkrPd} nd} t| ˆ ƒ}
t| ¡ ƒ‰t ˆƒ‰t
||ƒ\}}‡ ‡fdd„|D ƒ}|sœd S t |ˆ ƒ\}}}}t |ˆˆˆ ƒ\}}‡ ‡‡‡fdd„|D ƒ}t
|ˆ|ˆ ƒ\}}tt||||gƒ}t||||ˆ ƒ\}}t ˆd ƒ} d
} d } tgt|ƒ } |rPt| d < | tgt|ƒ 7 } |rht| d < | t¡ t| d < t| d
ƒ´} | d¡ | dt| ˆƒ ¡ | dt|
ˆƒ ¡ | dt|
ˆƒ ¡ | d| ¡ | d¡ | dttƒ d ¡ tt| ƒƒD ],}!| d|!| f ¡ | t8 } | | |! 8 } qþ| dt| ƒ| f ¡ | d¡ | dˆ ¡ | dt ¡ | d¡ | d¡ | d¡ | d ¡ | d ¡ | d ¡ | d ¡ | d ¡ t j r| d!¡ t |ˆ ƒ}"| d"¡ | dt|"ˆƒ ¡ | d#¡ | d$¡ | d%¡ | d&¡ | dt d'd(ƒ ¡ | d)| ¡ | d*t| ¡ ƒ ¡ | d+¡ | d,¡ |D ]@\}#}!| d-¡ | d.|! ¡ | dt|ˆƒ ¡ | d/¡ qjt |ƒD ]\}!}$| d-¡ | d0¡ | d1¡ | d2¡ | d3¡ | dt|$ˆƒ ¡ | d4t| ƒ d5 ¡ | d6t| ƒ d7 ¡ | d/¡ |rÀ||! rÀ| d-¡ | d8¡ | d9¡ | d:¡ | dt||! ˆƒ ¡ | d4t| ƒ d5 ¡ | d6t| ƒ d7 ¡ | d/¡ | d;7 } q´|D ]v}%| d-¡ | d<¡ | dt|%ˆƒ ¡ | d=¡ | d4t| ƒ d5 ¡ | d6t| ƒ d7 ¡ | d/¡ | d;7 } qÐ|rÈ| d-¡ | d8¡ | d9¡ | dt|ˆƒ ¡ | d>¡ | d4t| ƒ d5 ¡ | d6t| ƒ d7 ¡ | d/¡ | d;7 } | d-¡ | d<¡ | dt|ˆƒ ¡ | d?¡ | d@¡ | dA¡ | dB¡ | d/¡ | dC¡ W 5 Q R X t!|||||ˆƒ}&| |&fS )DNÚdatar4 i † r• r– c s g | ]}t |ˆˆ ƒ‘qS r0 )rc ©rV r` )ri ra r0 r1 rW ´ s z create_conf..c s g | ]}t |ˆˆˆˆ ƒ‘qS r0 )r‡ rž ©ri r) ra rw r0 r1 rW º s z circos.confgffffffî?r rd r z+<>
z<>
zkaryotype = %s
zchromosomes_units = %d
z"chromosomes_display_default = yes
ztrack_width = r ztrack%d_pos = %f
z
z dir = %s
z
file = %s
z
png = yes
z svg = no
zradius = 1500p
zangle_offset = -90
zauto_alpha_colors = yes
zauto_alpha_steps = 5
zbackground = white
z
z
z
zr0 = 1r - 50p
zr1 = 1r - 30p
z
z
Úetczhousekeeping.confzmax_points_per_track* = %d
zmax_ideograms* = %d
z
zlayers_overflow = collapse
z
ztrack_idx = track%d
z
z type = tile
zthickness = 50p
zstroke_thickness = 0
z layers = 1
z$r0 = eval(sprintf("%.3fr",conf(trackz _pos) - conf(track_width)))
z$r1 = eval(sprintf("%.3fr",conf(trackz_pos)))
ztype = histogram
zthickness = 1
zfill_color = vlyellow
r ztype = heatmap
zcolor = ylorbr-9
zfill_color = vlblue
zcolor = greys-6
zscale_log_base = 1.5
zr0 = 1r - 29p
z
r1 = 1r - 1p
z
)"r r ÚosÚmakedirsr r2 r8 ru rr r— r] rn rŽ rz r' Ú
MAX_POINTSr” ÚTRACK_INTERVALr! ÚBIG_TRACK_INTERVALr; r# r% r r& Ú TRACK_WIDTHrt Úcircos_png_fnamer
r› rA r" r< rœ )'Ú ref_fpathr[ r\ r) rh rŠ rv Úloggerr( r* r+ r/ r5 r6 rZ Z
contig_pointsZalignments_fpathsrl rm Z gc_pointsr‹ Z gene_pointsZmismatches_fpathsrx Z
cov_pointsrk Z labels_fpathr’ rb Zradiusr“ Ztrack_intervalsr, r@ r= r^ Zalignments_confr Úcircos_legend_fpathr0 rŸ r1 Ú create_conf¢ sþ
r« c C sÔ t |ƒst |¡ t| |||||||ƒ\}} tdƒ}
|
sV| d| d | d ¡ dS |
d|g} t|dƒ} t|dƒ}
t|tƒ}tj | t
| d ƒt
|
d ƒd
}|d kr²t |ƒr²|| fS | d | d
|
d ¡ dS d S )NZcircoszØCircos is not installed!
If you want to create Circos plots, install Circos as described at http://circos.ca/tutorials/lessons/configuration/distribution_and_installation and run the following command:
circos -conf z
The plot legend is saved to r rX z-confz
circos.logz
circos.errr )ÚstdoutÚstderrr z& Circos diagram was not created. See z and z for details) r r¡ r¢ r« r Zwarningr r§ r Zcall_subprocessr# r )r¨ r[ r\ rh rŠ rv r) r© rb rª Z circos_execZcmdlineZ log_fpathZ err_fpathZcircos_png_fpathZ return_coder0 r0 r1 Údo- s0
þþýý
r® )8Z
__future__r r¡ Úrerf Ú collectionsr Úos.pathr r r r r Ú ImportErrorZ%quast_libs.site_packages.ordered_dictr: r r
Z!quast_libs.ca_utils.align_contigsr Zquast_libs.ca_utils.miscr r
Zquast_libs.fastaparserr Zquast_libs.icarus_utilsr r r Zquast_libs.qutilsr r r Zquast_libs.reads_analyzerr r§ r¦ r¤ r¥ r£ r2 r8 rA rU r] rc rn rz r‡ rŽ r r” r— rœ r« r® r0 r0 r0 r1 Ú sL (3( (!#