U ñadÈOã@sŒddlZddlZddlZzddlmZWn ek rHddlmZYnXddlmZe  ¡ddl m Z e ej   ejd¡ƒddlmZmZmZmZmZmZddlmZmZmZddlmZmZmZmZmZdd lm Z m!Z!m"Z"m#Z#dd l$m%Z%e%ej&ƒZ'e' (¡dd d „Z)ddd„Z*dd„Z+e,dkrˆze+ej-dd…ƒZ.e/e.ƒWn>e0k r†e 1¡\Z2Z3Z2e' 4e3¡e'j5ddddYnXdS)éN)Ú OrderedDict)Úqconfig)Ú addsitedirZ site_packages)ÚAssemblyÚcorrect_meta_referencesÚcorrect_assembliesÚ#get_downloaded_refs_with_alignmentsÚpartition_contigsÚcalculate_ave_read_support)Ú parse_optionsÚremove_from_quast_py_argsÚprepare_regular_quast_args)Úcontigs_analyzerÚsearch_references_metaÚ plotter_dataÚqutilsÚ run_busco)ÚcleanupÚ check_dirpathÚ is_python2Ú run_parallel)Ú get_loggerFc  Cs’|dd…}| dd„|Dƒ¡|r8| d¡| |¡|rP| d¡| |¡| d¡dd„} | | d d d„|Dƒ¡ƒ¡d dl} zd dl} |  | ¡Wn ttfk r¾t| ƒYnX| jj d t j d |s|rît j  t |¡¡nd} | jj|| d |r&t dt j  |t jd¡¡|  |¡} |rT| j ¡}tttt||ƒƒƒ}|rŠdd„t jDƒ|dd…<dd„t jDƒ|dd…<| |fS)NcSsg|] }|j‘qS©)Úfpath©Ú.0Zasmrrúopt/quast-5.2.0/metaquast.pyÚ (sz%_start_quast_main..z-Rz-oz--labelscSsd|krd|}|S)Nú z"%s"r)ÚlinerrrÚquote4sz _start_quast_main..quotez, cSsg|] }|j‘qSr)Úlabelrrrrr9sré)Z indent_valÚdebug)Úis_parallel_runÚref_namez (logging to ú.log)cSsg|]}tj|‘qSr)rÚassembly_labels_by_fpath©rrrrrrNscSsg|]}t|tj|ƒ‘qSr)rrr'r(rrrrOs)ÚextendÚappendÚjoinÚquastÚ importlibÚreloadÚ ImportErrorÚAttributeErrorÚloggerÚset_up_console_handlerrr#ÚosÚpathÚbasenamerÚname_from_fpathZset_up_metaquastZ info_to_fileÚLOGGER_DEFAULT_NAMEÚmainZget_numbers_of_notificationsÚlistÚmapÚsumÚzipZassemblies_fpaths)ÚargsÚ assembliesÚreference_fpathÚoutput_dirpathÚnum_notifications_tupleÚlabelsÚrun_regular_quastÚis_combined_refr$r r,r-Zreference_nameÚ return_codeZcur_num_notificationsrrrÚ_start_quast_main$sB      ÿ  rFc Csüt |¡}|s~t d|d¡tj ||¡}tj |¡sDt |¡d}t j rtddl m } m } |  |t j|¡| j}|||fStj ||¡}d|} t d| dtj |t j¡d¡t||||||d \} }d}t j rîdd l m } | j}|||fSdS) Nz* No contigs were aligned to the reference z , skipping..r)Ú html_saverÚ json_saverzfor the contigs aligned to z Starting quast.py ú... (logging to r&)r>r?r@rAr$©rH)rr6r1Ú main_infor3r4r+ÚisdirÚmakedirsrÚ html_reportÚquast_libs.html_saverrGrHZsave_empty_reportZ min_contigÚ json_textr7rF) Ú quast_py_argsÚoutput_dirpath_per_refÚ ref_fpathÚref_assembliesÚtotal_num_notificationsr$r%r@rPrGrHÚrun_namerErrrÚ_run_quast_per_refTsB     ÿÿÿû  rWc: s$ ttjdttjƒdddd|s>tjtjdt d¡tj   t ¡g}t t ||ƒ\‰}tjtjtj}}}tj}tj}tj|d| dd \}}}tj  |tj¡} t t ¡t t_ d d lm} zd dl} |  | ¡Wn ttfk rüt| ƒYnXd d lm} tj  | ¡r"t  !| ¡t "| ¡|rVt  #¡t  #d ¡t$|| ƒ\} }}}t  #¡t  #d¡dt_%t&|||ƒ\}}|s’t  'd¡dStj(r¤ˆdg7‰tj)dkrˆdttj*ƒg7‰tj+rÐd}nd}d}|sÆt  #¡tj,d krþt  -d¡nÈtj.rt  #d¡n t  #d¡tj  |tj/¡}tj  |¡sDt "|¡tj  |tj¡} t0 1|||| tj.¡}|rªdt0_2tj.s€d}t  #¡t  #d¡t$|| dd\} }}}n|rÆ|sÆt j'dddd|st  #¡t  -d¡dd„|Dƒ}t3ˆ||ddtd ƒtj  |tj4¡}|t_tj5r8ˆdg7‰ˆtj5g7‰ˆd g7‰tj6sRtjr°t7j8r°d!d„|Dƒ}ˆd"g7‰ˆd# d$d„|Dƒ¡g7‰ˆd%g7‰ˆd# d&d„|Dƒ¡g7‰d'}t  #¡t  #d(|d)¡d }d }d }|||f}tjrþd d*l9m:}g}nd}tj;rg}nd+d,g}t3ˆ||||||dd-\}}|dk rL| <|j=¡dt0_2tj  |tj4d.¡} tj  | d/¡}!tj  >|!¡sÖt  #d0¡|s–d1}"nd2}"t  #d3|"¡t  #d0¡t?| ƒt  #d4¡t j@tA|ƒ|d5S|rÚ|d krÚt  #¡t  #d6¡tB|!||ƒ}#|#rº|#|krºt  #¡t  #d7¡t C|¡tDƒtE_Ft$|#| ƒ\} }}}t&|||ƒ\}}d8}t  #¡t  #d(|d)¡t3ˆ||||||dd-\}}|dk rÚ|dd9…}| <|j=¡n |#|krÐt  #d:¡n t  #d;¡|d krt  #d4¡t j@tA|ƒ|d5StjGrtH||ƒtIˆ||ƒt  #¡t  #d<¡tJ|| | tj  |tjKd=¡|ƒ\}$}%tjLr~tLjMt tjNtjOd>}&|&s~tPˆd?ƒtj  |tjQ¡‰tjRs€tS|$ƒtS|ƒkr€tS|ƒtjTkr€t  #¡t  #d@¡tUdtjTtS|$ƒƒ}'ˆdAg7‰ˆdBt|'ƒg7‰dC‰‡‡‡fdDd„|$Dƒ}(tVtW|(tjTddE\}})}*tXtYtZt[|*Žƒƒ}+tXtYtZt[||+ƒƒƒ}|dk rd| \|)¡ˆ CdA¡tPˆdBt|'ƒƒ‰nRg}|$D]H\},}-tWˆˆ|,|-|ƒ\}.}/}|.s°qˆ| <|.¡|dk rˆ| <|/¡qˆd}0|%D]4}1tj  ]|1j^¡rÚt _|1j^¡j`d krÚd}0qqÚdF}t  #¡|0r6t  #dG|dH¡n˜t  #d(|dItj  |tjatjbdJ¡¡t3ˆdBttjTƒg|%tj  |tja¡|dK\}}|dLkr¤t  'dM¡n|dkr²d}0|0sÎ|dk rÎ| <|j=¡| röt  c¡t  #dN¡tj  |tjd¡}2tj  |2¡ st e|2¡| r2| r2d dOl9mf}3|3 g|¡}4nd}4d dPlmh}5| jijj}6| jijk| jijl| jijm| jijng}7|0 r|dQd„| Dƒ}8ndRd„| Dƒtjag}8|5 1|4|2|ˆ|6|7|8¡| rö| rö|3jo||t7j8ddStjp rê|3 q||¡}9t  #dT|9¡|3 r||¡t?| ƒt  #d0¡t  #d4¡t j@tA|ƒ|d5S)UNzYou are trying to run it from z. z.)r>r@rCz--sv-bedz--combined-refcSsg|]}tj|j‘qSr)rÚdict_color_and_lsr!rrrrrèsz--colorsú,cSsg|] }|d‘qS)rr©rZstylerrrrêsz--lscSsg|] }|d‘qS)r"rrbrrrrìszfor the combined referencezStarting quast.py z...rJz--ambiguity-usageÚall)rBr>r?r@rArDZ genome_statszgenome_info.txtÚz1Try to restart MetaQUAST with another references.zjTry to use option --max-ref-number to change maximum number of references (per each assembly) to download.z4Failed aligning the contigs for all the references. zMetaQUAST finished.)ZnumbersZ check_testzPExcluding downloaded references with low genome fraction from further analysis..zFiltered reference(s):z$for the corrected combined referenceéÿÿÿÿzSAll downloaded references have genome fraction more than 10%. Nothing was excluded.zQAll downloaded references have low genome fraction. Nothing was excluded for now.z:Partitioning contigs into bins aligned to each reference..zalignments_%s.tsv)Ú is_fungusz--conservativez/Run QUAST on different references in parallel..z--memory-efficientz-t)rrrcs g|]\}}ˆˆ||ˆdf‘qS)Tr)rrSrT©Znum_notificationsrRrQrrrUsÿ)Zfilter_resultsz$for the contigs not aligned anywherezSkipping quast.py z (everything is aligned!)rIr&)r>r@rA)rr\z;Error running quast.py for the contigs not aligned anywherezSummarizing results...)rG)Úcreate_meta_summarycSsg|]}t |¡‘qSr©rr6©rrSrrrr—scSsg|]}t |¡‘qSrrirjrrrr™s)Úmetaz( Icarus (contig browser) is saved to %s)srrZ QUAST_HOMEÚstrZusageÚsysÚstderrÚexitr3r4ÚrealpathÚ__file__r r1r@Z referencerBrNZtestrZset_up_output_dirr+Zcorrected_dirnameZset_max_threadsÚ quast_libsrZr-r.r/r0r[rLÚshutilZrmtreeÚmkdirrKrZ no_check_metarÚerrorZ gene_findingZmin_IDYZ META_MIN_IDYÚreuse_combined_alignmentsZmax_referencesZnoticeZreferences_txtZdownloaded_dirnamerZdoZis_quast_first_runrFZcombined_output_nameZbedZ draw_plotsrr`rOrHZunique_mappingr*rPÚexistsrZ finish_upÚtuplerÚremoverrZref_labels_by_chromosomesZcalculate_read_supportr r r Z detailed_contigs_reports_dirnamerZ download_dbZ prokaryoterfr Zper_ref_dirnameZmemory_efficientÚlenZ max_threadsÚmaxrrWr9r:r;r<r)ÚisfilerÚstatÚst_sizeZnot_aligned_namer7Zprint_timestampZmeta_summary_dirrMrGZinit_meta_reportrhZFieldsZ main_metricsZMIS_RELOCATIONZMIS_TRANSLOCATIONZ MIS_INVERTIONZMIS_ISTRANSLOCATIONSZ save_colorsZcreate_icarus_htmlZcreate_meta_icarusZcreate_meta_report):r=Zmetaquast_pathZcontigs_fpathsr@Z ref_fpathsrBrNZ test_modeÚ_Zcorrected_dirpathrZr-r[Zcorrected_ref_fpathsZcombined_ref_fpathZchromosomes_by_refsZ ref_namesr>rvr]Zdownloaded_dirpathZcombined_output_dirpathZ colors_and_lsrVZtotal_num_noticesZtotal_num_warningsZtotal_num_nf_errorsrUrHZ json_textsZambiguity_optsrEZgenome_info_dirpathZgenome_info_fpathÚmsgZcorr_ref_fpathsZassemblies_by_referenceZnot_aligned_assembliesZ db_dirpathZthreads_per_refZparallel_run_argsZref_json_textsZref_notificationsZper_ref_num_notificationsrSrTr%rPZno_unaligned_contigsZassemblyZsummary_output_dirpathrGZhtml_summary_report_fpathrhZmetrics_for_plotsZmisassembly_metricsZfull_ref_namesZicarus_html_fpathrrgrr8ts"ÿÿ þ       ÿ            ÿ  ÿ        úÿ         ÿ úÿ          þ ÿ ÿ  ÿÿ    ÿ  $ ÿý            ÿ ÿ     r8Ú__main__r"zexception caught!T)r_r^)NNNNFFF)F)6rmr3rsÚ collectionsrr/Z%quast_libs.site_packages.ordered_dictrrrZcheck_python_versionZsiterr4r+Z LIBS_LOCATIONZquast_libs.metautilsrrrrr r Zquast_libs.options_parserr r r rrrrrZquast_libs.qutilsrrrrZquast_libs.logrZLOGGER_META_NAMEr1r2rFrWr8Ú__name__ÚargvrEroÚ ExceptionÚexc_inforÚ exc_valueZ exceptionrurrrrÚ sF     ÿ 0 7