a
£žb‡h ã @ s| d dl mZ d dlZd dlZd dlZd dlmZ d dlmZm Z m
Z
m Z zd dlm Z W n e
yv d dlm Z Y n0 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ƒR}| ¡ D ]8\}}| d dd||dt|ƒdg¡d ¡ t||ƒ}q.W d ƒ n1 s|0 Y 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 d ƒ n1 s´0 Y |||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
Ú
prokaryote) Ú chr_lengthsÚ
output_dirZnum_chromosomesÚmax_lenÚkaryotype_fpathÚout_fÚnameÚseq_lenÚideogram_fpath© r1 ú;/home/psgendb/BIRCHDEV/pkg/quast-5.2.0/quast_libs/circos.pyÚcreate_ideogram$ sF
&*
*r3 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 d ƒ n1 sî0 Y |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Úsuffixr1 r1 r2 Úcreate_ticks_confL s^
*r9 c C sÜ t |dƒ}g d¢}t|dƒ¬}tƒ }ddlm} |j ¡ D ]&\}}||vrRg ||< || |¡ q:t| ¡ ƒD ]J\} \}}
|
D ]8}| d |dt
| | ƒd|| t |ƒ g¡d ¡ q~qnW d ƒ n1 sÎ0 Y |S )
Nzhighlights.txt)ÚorangeÚpurpleÚbluer r )Úcontigs_analyzerr r z fill_color=r ) r r# r Ú
quast_libsr= Úref_labels_by_chromosomesr$ ÚappendÚ enumerater% r& r! ) r) r* Úhighlights_fpathÚcolorsr- Z
chrom_by_refsr= ÚchromÚrefÚiZ chromosomesr1 r1 r2 Úcreate_meta_highlights s
VrG c C s g }t tƒ}t| ƒ`}d }d }d }d }d }d } d }
t|ƒD ](\} } | dd¡ d¡}
| dkr¨|
d¡}|
d¡}|
d¡}|
d¡}|
d ¡}|
d
¡} |
d ¡}
q<|
r¼|
d d kr¼qÄ ó z$parse_alignments..©NN)r Úlabel_from_fpath_for_fnamera r r@ r' r Ú
assemblies)Úcontigs_fpathsÚcontig_report_fpath_patternZlists_of_aligned_blocksÚ
contigs_fpathrX rY rZ Z max_contigsr1 r1 r2 Úparse_alignments· s rk c C s@ t || jd ƒ}|d }t|dƒ}d }| jD ]º}d|_|jrHd|_n |jrTd|_|r²|j|jkr²|j|jkr²t|j |j ƒt
|j |j ƒ |k r²t
|j |j ƒ|_ t|j |j ƒ|_ q0|ræ| d |jt
|j ƒt
|j ƒd|j g¡d ¡ |}q0| d |jt
|j ƒt
|j ƒd|j g¡d ¡ W d ƒ n1 s20 Y |S )
Nz.confr r ÚgreenÚredZppurpler zcolor=r )r Úlabelr# Z
alignmentsÚcolorZ misassembledrW rO r' rM ÚminrN r% r& )ÚassemblyÚref_lenr* Ú
conf_fpathZmax_gapr- Z
prev_alignÚalignr1 r1 r2 Úcreate_alignment_plotsÊ s(
ÿ0Pru c C s g }t | ƒ0}|D ]}| t| ¡ d ƒ¡ qW d ƒ n1 sB0 Y t|ƒ}tt|ƒƒtt|ƒƒ }}t|dƒ}t
| |¡ ||||fS )Néÿÿÿÿzgc.txt) r# r@ ÚfloatrS r! rV rp r' r ÚshutilÚcopy) Úgc_fpathÚdata_dirZ gc_valuesÚfr] Ú
max_pointsÚmin_gcÚmax_gcZ dst_gc_fpathr1 r1 r2 Úcreate_gc_plotá s
6
r€ c C s¤ 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 d ƒ n1 sè0 Y t|d ƒŽ}| ¡ D ]t\} }t|ƒD ]`\}}|r4t
|ƒt|ƒ nd}| d
| t|| ƒt|d | ƒt|ƒg¡d ¡ |d7 }qqW d ƒ n1 s’0 Y ||fS ) Nr z coverage.txtú#r c S s g | ]}g ‘qS r1 r1 )rb rF r1 r1 r2 rc þ rd z(create_coverage_plot..é rv r r r )Údictr rQ Úvaluesr# rA rS Ú
startswithrV Úranger@ r r$ Úsumr! r% r& )Ú cov_fpathÚ window_sizer) r* r} Z cov_by_chromÚcov_data_fpathr| ÚposrT r] ÚfsrD Z chrom_orderZ chrom_lenÚdepthr- Z
depth_listrF ÚdepthsZ avg_depthr1 r1 r2 Úcreate_coverage_plotí s4
( 40r c sJ 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 d ƒ n1 s00 Y t |
d ƒð}| ¡ D ]Ö\}}d
\}}t|ƒD ]\}}|dkr†|d ˆ }nn|r°| d |t|ƒt|ƒdg¡d ¡ | d |t|ˆ ƒt|d ˆ ƒt|ƒg¡d ¡ |d ˆ }d }qf|rN| d |t|ƒt|ƒdg¡d ¡ qNW d ƒ n1 s<0 Y |
S )Nz..z.mismatches.txtc s dgˆ ˆ d S ©Nr r r1 r1 ©rr r‰ r1 r2 Ú rd z(create_mismatches_plot..ú|r é rv r Ú*ú+r )r r r r r )r rf Úfpathr r
Ú detailed_contigs_reports_dirnamer r r Ú show_snpsr r# rV rS rU r
r! r… r$ rA r% r& )rq r‰ rr Úroot_dirr* Úassembly_labelÚaligner_dirpathZcoords_basenameÚ_Zcoords_filtered_fpathZmismatches_fpathZmismatch_density_by_chromZ coords_filer] Ús1rD ÚcigarÚref_posÚopZn_basesr- Z density_listrM rN rF Údensityr1 r‘ r2 Úcreate_mismatches_plot
sF
,
$4 Hr£ c
s¨ g }d}| s||fS | D ]„}t ||jd ƒ}t|jƒdkr>qd}t‡ ‡fdd„ƒ} t|dƒ}
|jD ]’} | jr„| j|jv r„| jn| j} | |jv rž|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 d ƒ n1 sv0 Y t|ƒr”| |¡ t||ƒ}q||fS ) Nr z.txtc s dgˆ ˆ d S r r1 r1 r‘ r1 r2 r’ A rd z#create_genes_plot..r r r r )r Úkindr! Ú region_listr r# Ú
chromosomeZchr_names_dictÚseqnamer† rM rp rN r$ rA r% r& r r@ r' )Úfeatures_containersr‰ rr r* Úfeature_fpathsr} Úfeature_containerÚ
feature_fpathZ
num_pointsZgene_density_by_chromr- ÚregionrD rF Zgene_density_listr¢ r1 r‘ r2 Úcreate_genes_plot5 s:
ÿ,40
r c
C sh t |dƒ}t|dƒ@}| ¡ D ]&\}}| d |dt|ƒg¡d ¡ q W d ƒ n1 sZ0 Y |S )Nz
genome.txtr r r r )r r# r$ r% r& )r) r* Z genome_fpathr- r. r/ r1 r1 r2 Úcreate_genome_fileV s
>r® c
C s6 t |dƒ}g }d}t|ƒD ]*\}} | dt|d ƒ |f¡ |d7 }q|D ]*}
t|
jƒdkrJ| |
j|f¡ |d7 }qJ|rˆ| d|f¡ t|dƒ>} | t |
¡ ƒd d d d d
„ |D ƒ¡ ¡ W d ƒ n1 sÖ0 Y t |d ƒ} t| dƒ(} | d t ||ƒ d
¡ W d ƒ n1 s$0 Y | |fS )Nz
labels.txtr rq r Úcoverager z
0 0 null ú,c S s g | ]\}}d ||f ‘qS )z
track%d=%sr1 )rb rn rF r1 r1 r2 rc m rd 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 rA r@ r& r! r¥ r¤ r# r% rQ r" r )
r) rg r¨ Úcoverage_fpathr* Zlabels_txt_fpathÚ track_labelsÚplot_idxrF rq rª r- Zlabels_conf_fpathr1 r1 r2 Ú
create_labels^ s,
L
ûû&r´ c C s@ | dkrd}n.| dkr d}n | dkr*d}n| dkr8d}nd }|S )
Ni eÍ i N i £áé' i áõiˆ r5 éè r r1 r‘ r1 r1 r2 Ú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 d ƒ n1 s¾0 Y |S )Nz
legend.txtr zn1) The outer circle represents reference sequence%s with GC (%%) heatmap [from %d%% (white) to %d%% (black)].
ÚsrH 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_refrA rn )
rg r~ r r¨ r± r* Z legend_fpathr- rF rq r1 r1 r2 Ú
create_legend s
ÿ
4r¼ c ' sf 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 d ƒ n1 sB0 Y t!|||||ˆƒ}&| |&fS )DNÚdatar5 i † rµ r¶ c s g | ]}t |ˆˆ ƒ‘qS r1 )ru ©rb rq )r{ rr r1 r2 rc ´ rd z create_conf..c s g | ]}t |ˆˆˆˆ ƒ‘qS r1 )r£ r¾ ©r{ r* rr r‰ r1 r2 rc º rd z circos.confgffffffî?r rv 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 r3 r9 r‡ r„ r· rk r€ r r r' Ú
MAX_POINTSr´ ÚTRACK_INTERVALr! ÚBIG_TRACK_INTERVALr@ r# r% r r& Ú TRACK_WIDTHr† Úcircos_png_fnamer
r» rG r" rA r¼ )'Ú ref_fpathrh ri r* rz r¨ rˆ Úloggerr) r+ r, r0 r6 r7 rg Z
contig_pointsZalignments_fpathsr~ r Z gc_pointsr© Z gene_pointsZmismatches_fpathsrŠ Z
cov_pointsr} Z labels_fpathr² rs Úradiusr³ Ztrack_intervalsr- rF rB rn Zalignments_confr« Úcircos_legend_fpathr1 r¿ r2 Ú 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 )NÚcircoszØ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 re 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 Úwarningr rÇ r Úcall_subprocessr# r )rÈ rh ri rz r¨ rˆ r* rÉ rs rË Z circos_execÚcmdlineÚ log_fpathÚ err_fpathÚcircos_png_fpathÚ return_coder1 r1 r2 Údo- s0
þþýý
r× )8Ú
__future__r rÁ Úrerx Ú collectionsr Úos.pathr r r r r Ú ImportErrorZ%quast_libs.site_packages.ordered_dictr> r r
Z!quast_libs.ca_utils.align_contigsr Úquast_libs.ca_utils.miscr r
Zquast_libs.fastaparserr Zquast_libs.icarus_utilsr r r Úquast_libs.qutilsr r r Zquast_libs.reads_analyzerr rÇ rÆ rÄ rÅ rà r3 r9 rG ra rk ru r€ r r£ r r® r´ r· r¼ rÌ r× r1 r1 r1 r2 Ú sL (3( (!#