B d" @sddlmZmZmZddlmZddlmZddddddd d d d d dd ZiZ x8e D],\Z Z e dd dd ddZ e e e <qXWdZGdddeZeZGdddeZddZedZGdddeZd S)!)absolute_importdivisionprint_function)matrix)slots_getstate_setstateg(\?grh|?gtV?g-?gPn?gX9v?gn?g r?gK?gV-?gE?) zOP1 PzOP2 PzO5' PzC1' C2'zC1' O4'zC2' O2'zC2' C3'zC3' O3'zC3' C4'zC4' O4'zC4' C5'zC5' O5'OP1O1POP2O2P'*g rh?c@seZdZddZddZdS) _make_tablescCs d|_dS)N)cache)Orz/mnt/filia/a/genomebrowser/www/genomebrowser/fleming/tools/molprobity/modules/cctbx_project/iotbx/pdb/rna_dna_detection.py__init__ sz_make_tables.__init__cCsx|jdkrng|_d}tdd|D}xFdD]>}t|d}|j||f|j|tdd|Dfq,W|j|S)N)zC5'zC4'zO4'zC1'zC2'zC3'zO3'cSsg|]}|ddqS)r r )replace).0arrr 'sz)_make_tables.__call__..))zOP1 PzOP2 P)zO1P PzO2P P) zO5' PzC1' C2'zC2' C3'zC3' C4'zC3' O3'zC4' C5'zC4' O4'zC1' O4'zC5' O5'cSsg|]}|ddqS)r r )r)rbrrrr.s)rtuplesetappend)rindexZatoms_tZatoms_sZbonds_1Zbonds_trrr__call__#s  "z_make_tables.__call__N)__name__ __module__ __qualname__rrrrrrr sr c@s.eZdZdddddddgZdd Zd d Zd S) atom_name_analysissub_classificationrequired_atomsrequired_bondshave_o2c2_o2c1have_all_required_atomscCs@d|_d|_d|_d|_d|_d|_d|_d|kr6dSd|krPd|krJdSd}nd|krjd|krddSd}ndSd |krt|d \|_|_d |k|_d |_d |_n:d|krt|dd \|_|_d|k|_d|_d|_ndS|dkr|jdkrd|_n"|dkr|jd krd|_ndSx|jD]}||krdSqWd|_dS)NFPrr rrr zO5')rzO2'zC2' O2'zC1'zO5*zO2*zC2* O2*zC1*v3T)r!r"r#r$r%r&r'_tables)r atom_dictZtab_offs required_atomrrrr>sL    zatom_name_analysis.__init__cCs|jdkrtStS)Nr,)r!)bond_distance_ideal_by_bond_atom_names_v3)bond_distance_ideal_by_bond_atom_names_v2)rrrr&bond_distance_ideal_by_bond_atom_nameses z9atom_name_analysis.bond_distance_ideal_by_bond_atom_namesN)rrr __slots__rr2rrrrr 3s'r cCst|d}|jsdSd}t}xT|D]L}|j|jg}|d|}||jkr\||q$|j r$||j kr$d}q$Wt |t |jkrdS|rd|j Sd|j S)N)r.F TRNADNA) r r'r atom_id_1 atom_id_2sortjoinr#addr$r%lenr!)r. bond_listrZ rna_indicatorZ bonds_matchedbondpairrrrclassificationjs$       r@zC1' C2' C3' O3' C4' O4' C5' O5'c @s0eZdZddddddddd d d g Zdd dZdS)residue_analysisproblemsr.deoxy_ribo_atom_dictp_atomop_atomsc1p_outbound_candidateso2p_atomh_atomsis_terminus_with_o5pc1p_outbound_atomis_rna?cs^g|_|jddddd|_t|jt|kr8|jdg}i|_xBtD]:}|j|}|dkrp|d|qH||j|<|j|=qHWd|_xJt |j D]8}| dr|dkr|j||_|j|=q|jdqWg|_ xndD]f}x`|D].}|j|}|dk r|j ||j|=PqW|j d|jdk r|jd|d qW|jd |_ |j dk rp|jd =i|_xt |j D]}t|d kr|jd qd |d d kst|d krd |d d krd|d d kr|j||j|<|j|=qWi|_xt |j D]t}|dd krh|dkrX||j|<|j|=n |jdq&| ds| dr&|j||j|<|j|=q&Wd|_t|jd krt|d krt|jd kr|jddk rd|_n |j|fdd}|jdk rt|jj}x`tddgD]P\} } |j | }|dkrLq,t|j} || || s,|jd| d q,W|jd}|dk rt|j} |d|| s|jd|j dk r|jd}|dk rt|j}t|j j} |d|| s|jdxdD]} g} xt| dd | d!dgD]0}|j|}|dkrVP| t|jq:W|| f| s|jd"| d#d$qWd|_t|jd kr8|jd%}|dk r8t} t|j}xZ|jD]L\}}t|j} t|| }| |kr| |ks(|jdkr||_|} qWt|jd krNd|_|j dk |_dS)&NTF)Z strip_namesZ upper_namesZconvert_stars_to_primesZ%throw_runtime_error_if_duplicate_keysZ key_clashZmissing_r(Zother_P))rr)r r rzO2'Z blank_nameZHDr* 0123456789r zN2'Z other_primeNCzO5'cs(t||}t|}||kr$dSdS)NFT)absr0)key_pairsite_1site_2distance_modeldistance_ideal)distance_tolerancerrcheck_distances   z1residue_analysis.__init__..check_distancezOP1 PzOP2 Pzlong_distance_P_OP%dzO5' Pzlong_distance_P_O5'zC2'zC2' O2'zlong_distance_C2'_O2')zC1' C2'zC1' O4'zC2' C3'zC3' O3'zC3' C4'zC4' O4'zC4' C5'zC5' O5'Zlong_distance_r4_zC1')rB build_dictr.r<rrCdeoxy_ribo_atom_keysgetrDlistkeys startswithrErGrHfindrFrIextendrcolxyz enumeraterrJc1p_outbound_distance_estimateitemsrPrK)rZ residue_atomsrVZdeoxy_ribo_problemskeyatomr_rWrRirQrSsitesZclosest_distancedistancer)rVrrs                                                zresidue_analysis.__init__N)rL)rrrr3rrrrrrAsrAN) __future__rrrscitbxrlibtbxrr0r1rgrhvaluerrfobjectr r-r r@rsplitr\rArrrrs8   7