ó ¦žbc@s"ddlmZmZyddlmZWn!ek rMddlmZnXddlmZddl m Z m Z m Z m Z ddlmZdefd„ƒYZdefd „ƒYZd efd „ƒYZd efd „ƒYZedd„Zd„Zd„Zd„Zdd„ZdS(iÿÿÿÿ(theappushtheappop(t OrderedDict(tqconfig(tis_misassemblytexclude_internal_overlapst Misassemblyt$is_fragmented_ref_fake_translocation(tis_same_referencet ScoredSetcBseZdZd„ZRS(tscoretindexest uncoveredcCs||_||_||_dS(N(R R R (tselfR R R ((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyt__init__s  (sscoresindexess uncovered(t__name__t __module__t __slots__R(((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyR stPutativeBestSetcBsMeZd Zd„Zd„Zd„Zd„Zd„Zd„Zd „Z RS( R t score_dropR cCs||_||_||_dS(N(R RR (R R RR ((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyRs  cCs||k o||k S(N((R tother((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyt__eq__$scCs ||k S(N((R R((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyt__ne__'scCs"|j|jf|j|jfkS(N(RR (R R((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyt__lt__*scCs"|j|jf|j|jfkS(N(RR (R R((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyt__gt__-scCs||kp||kS(N((R R((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyt__ge__0scCs||kp||kS(N((R R((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyt__le__3s(sindexess score_drops uncovered( RRRRRRRRRR(((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyRs      tPSAcBsJeZd ZdZdZdZddd „Zd „Zd „Z d „Z RS(talignt num_sideststartt unique_endttotal_unique_lentend_overlap_penaltytstart_overlap_penaltyiicCs±||_||_|jƒ|_|dkr9|jƒnt|jƒ|ƒ|_d|_|dkrldntd|jƒ|ƒ}t||j |j ƒ|_ |j |_ dS(Ni( RRRtNonetendtminRR tmaxtoverlap_penalty_coefftmax_single_side_penaltyR!R"(R Rtctg_unique_endRt end_overlap((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyR>s  0 +cCs5|jt|jd|j|j|j|jƒkS(Ni(R R&tmin_unique_lenRR(R"R!(R ((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pytis_solidHscCs7|j|jd|j|j|j|j|jkS(Ni(RRR RR(R"R!(R ((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pytcould_be_solidMsc Cs¼|jtd|jt|jƒ|jdƒƒ7_t|jdt|j|jƒdƒƒ|_|jƒ|jkr¸td|jƒ|jdƒ}t||j|jƒ|_ndS(Nii( R R&RR$RR%R'R(R"(R Rt start_overlap((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyt intersectQs 5/ (saligns num_sidessstarts unique_endstotal_unique_lensend_overlap_penaltysstart_overlap_penaltyN( RRRR'R(R+R#RR,R-R/(((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyR7s  t SolidRegioncBs eZdZd„Zd„ZRS(RR$cCs"|jƒ|_|jƒ|_dS(N(RR$(R R((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyR\scCs(|j|jƒko'|jƒ|jkS(N(RR$(R R((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pytinclude`s(sstartsend(RRRRR1(((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyR0Ys c) Cs: tƒ}tdttjtt|dƒƒƒƒd|dlss: Too much alignments (%d), will use speed up heuristics iÿÿÿÿRis Found %d solid alignments: s Solid alignment %s sN Skipping alignments located inside solid regions since they are redundant: s" Skipping redundant alignment %s cSs|jƒ|jfS(N(R$R7(R8((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyR9³ss Nothing was skipped sG Looking for the best set of alignments (out of %d total alignments) cSs|jƒ|jfS(N(R$R7(R8((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyR9¾streversetallsl Ambiguity for this contig is not checked for speed up (too much alignments and --ambiguity-usage is "%s") (3tdictR&R%RtBSS_EXTENSIVE_PENALTYtinttroundtBSS_LOCAL_PENALTYtsortedtlentBSS_critical_number_of_alignstwriteRR(R't min_alignmentR+RtreversedR-R/R,tappendRR#RR"tstrtTrueR0R$tlisttpopt enumerateR1tFalset IndexErrorR R t get_scoreR R tambiguity_usagetambiguity_scoreRRRRtBSS_MAX_SETS_NUMBERRtitems()t sorted_alignstctg_lentstdout_ftseqtref_lenst is_cyclictregion_struct_variationst penaltiestsolidstcur_PSAR)Rtnothing_skippedt solid_regionst cur_regiontfiltered_alignstidxtall_scored_setst max_scoret cur_solid_idxtnext_solid_idxtlocal_max_scoretnew_scored_sett scored_settitcur_set_alignsR R tbest_settmax_allowed_score_dropt putative_setst best_setsRtambiguity_check_is_neededttoo_much_best_setst putative_settlocal_uncoveredtputative_predecessorst preceding_sett new_indext new_uncovered((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pytget_best_aligns_setsds0 33                        '  #     $      -      #'    cCs-tg|D]}|jD] }|^qq ƒS(N(tsetR (RoRltindex((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pytget_used_indexesscCsçd}||}|jƒt|jƒd|jƒƒ}d}x|jƒ|jƒkrÞ||jƒ|jƒ7}|d8}| t|ƒkrÚ|jƒ}||}|tdt||jƒƒ|jƒdƒ8}qBPqBW||S(Niþÿÿÿii(R$R&RRBR%(t set_alignst cur_aligntlast_align_idxt last_alignt added_rightt added_leftt prev_start((sT/home/psgendb/BIRCHDEV/install/quast-5.2.0/quast_libs/ca_utils/best_set_selection.pyt get_added_len!s )   3cCsÓt|ƒdkr¤|djƒ|djƒf\}} \|d<|ds  " " ¹   /