ó ƒå˜[c@`s¾ddlmZmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddl Z ejeƒZdZddd„ƒYZdS(i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tTopological_sortgš™™™™™©?tCompact_graph_wholecB`sbeZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z RS( cC`sdS(N((tself((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyt__init__scC`s5tj|jƒƒ}x|D]}|jƒr4qntjdj|jƒƒƒt|j ƒƒ}t |ƒdkr|d}t |j ƒƒdkr»tjdj||jƒƒƒqntjdj||ƒƒ|j |j ƒƒ|j|jƒ|jƒƒ|j|j ƒ|gƒ|j|ƒqqWdS(Nucompact_unbranched( {} )iiu8prev node: {} id downward branched, so can't compact. {}u$linear compaction of nodes {} and {}(Rttopologically_sortt get_all_nodestis_deadtloggertdebugtformatttoStringtlisttget_prev_nodestlentget_next_nodestadd_transcriptstget_transcriptstset_seqtget_seqt add_edgest prune_node(Rttgrapht all_nodestnodet prev_nodest prev_node((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pytcompact_unbrancheds    c C`sÿt}d}xì|rú|d7}tjdj||ƒƒ|jƒt}tj|jƒƒ}t j rždj||ƒ}|j |ƒtjdj|ƒƒnxÿ|D]÷}|j ƒdkrÃq¥ntjdj||j ƒƒƒ|jƒ}t|ƒdkr?|j|ƒr?|j||ƒr?|j||ƒr?t}q?n|jƒ} t| ƒdkr¥|j| ƒr¥|j|| ƒr¥|j| |ƒrœt}qœq¥q¥W|rt j rçdj||ƒ}|j |ƒtjdj|ƒƒn|j|ƒqqWdS(Niiu) ### Num allowed variants: {}, Round: {}u8ladeda.compacted_whole.BEGIN.num_allowed_{}.round_{}.dotu-wrote dot: {}u5 ///////// R[{}] Exploring compaction at node: {} u6ladeda.compacted_whole.END.num_allowed_{}.round_{}.dot(tTrueR R R tclear_touch_settingstFalseRRR tTGLOBALStDEBUGt draw_graphtget_touched_valRRRt untouchedt$all_have_lower_topological_orderingstcompact_upwardRt%all_have_higher_topological_orderingstcompact_downwardR( RRtnum_allowed_variantstcompacted_flagtroundt sorted_nodest dot_filenameRRt next_nodes((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyt compact_graph8s<       3  3  cC`s+x$|D]}|jƒdkrtSqWtS(Ni(R%R!R(Rt node_listR((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyR&gs cC`sR|jƒ}|dkr'tdƒ‚nx$|D]}|jƒ|kr.tSq.WtS(Niu*Error, must run topological ordering first(tget_topological_ordert RuntimeErrorR!R(RRtother_nodes_listt topo_ordert other_node((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyR'os   cC`sR|jƒ}|dkr'tdƒ‚nx$|D]}|jƒ|kr.tSq.WtS(Niu*Error, must run topological ordering first(R3R4R!R(RRR5R6R7((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyR)zs   cC`s’tjdj||ƒƒt|ƒ}xftdt|ƒdƒD]K}xBt|dt|ƒƒD]'}|j|||||ƒr_tSq_Wq?WtS(Nu+compact_upward: {} max_allowed_variants: {}ii( R R R RtrangeRtcompact_upward_node_pairRR!(RRR+titj((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyR(ˆs    c C`stjdj|||ƒƒ|j|ƒs:|j|ƒrWtjdj||ƒƒtStjdj||ƒƒt|jƒƒt|jƒƒkr¤||}}n|jƒ}|jƒ}t|ƒ}|ddd…}|ddd…}d} x8td|ƒD]'} || || kr| d7} qqW| |kr^tjdj| |||ƒƒtS| |tkr”tjdj| |t||ƒƒtS|j ƒ} tjd j|j ƒƒƒtjd j|j ƒƒƒ|j |j ƒƒ| j |g|jƒƒ| j|g|jƒƒ|jƒ|jƒt|ƒ|kr†t|ƒ|} |d| !}|j|ƒ| j |g|gƒn&| j |jƒ|gƒ| j|ƒtjd j|j ƒƒƒtjd j|j ƒƒƒtjd j||ƒƒtS(Nu9compact_upward_node_pair({}, {}, num_allowed_variants: {}u6-not parallel paths. excluding compaction of {} and {}u-not ancestral: {}, {}iÿÿÿÿiiu.num_mm: {} exceeds {} for node pair: {} and {}uEnum_mm: {} / len: {} exceeds max mm rate: {} for node pair: {} and {}unode_A before mods: {}unode_B before mods: {}unode_A after mods: {}unode_B after mods: {}u% *** compacted nodes: {} and {} ***(R R R t is_ancestralR!RRR8t MAX_MM_RATEt get_graphRRRRRt prune_edgesttouchRRRR( Rtnode_Atnode_BR+tseqAtseqBt shorter_lentseqA_revtseqB_revtnum_mmR:Rtdelta((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyR9•sT$          cC`s’tjdj||ƒƒt|ƒ}xftdt|ƒdƒD]K}xBt|dt|ƒƒD]'}|j|||||ƒr_tSq_Wq?WtS(Nu-compact_downward: {} max_allowed_variants: {}ii( R R R RR8Rtcompact_downward_node_pairRR!(RR0R+R:R;((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyR*ìs    c C`s“|j|ƒs|j|ƒr;tjdj||ƒƒtSt|jƒƒt|jƒƒkro||}}n|jƒ}|jƒ}t|ƒ}d}x8td|ƒD]'}||||kr©|d7}q©q©W||krtjdj||||ƒƒtS||tkr9tjdj||t||ƒƒtS|j ƒ} tjdj|j ƒƒƒtjdj|j ƒƒƒ|j |j ƒƒ| j |jƒ|gƒ| j|jƒ|gƒ|jƒ|jƒt|ƒ|kr||}|j|ƒ| j |g|gƒn&| j |g|jƒƒ| j|ƒtjdj|j ƒƒƒtjd j|j ƒƒƒtjd j||ƒƒtS( Nu6-not parallel paths. excluding compaction of {} and {}iiu.num_mm: {} exceeds {} for node pair: {} and {}uEnum_mm: {} / len: {} exceeds max mm rate: {} for node pair: {} and {}unode_A before mods: {}unode_B before mods: {}unode_A after mods: {}unode_B after mods: {}u% *** compacted nodes: {} and {} ***(t is_descendantR R R R!RRR8R=R>RRRRRR?R@RRRR( RRARBR+RCRDRERHR:R((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyRJøsJ$          ( t__name__t __module__RRR1R&R'R)R(R9R*RJ(((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyRs   /   W ((t __future__RRRRtostsystretloggingtTGraphtTNodetNode_alignmentRR"t getLoggerRLR R=R(((sƒ/oak/stanford/groups/akundaje/marinovg/programs/trinityrnaseq-Trinity-v2.8.4/Analysis/SuperTranscripts/pylib/Compact_graph_whole.pyts"$