ó O_[c@sŽddlZddlZddlZddlZddlZddljZddlZddl m Z ddl Z ddl Z ddddddd d d d d dddddddddgZ idd6dd6dd6dd6dd6dd6d d6d d6d d6d d 6d d!6dd"6dd#6dd$6dd%6dd&6dd'6dd(6dd)6dd*6Zd+d1d,„ƒYZd-„Zd.„Zd/„Zd0„ZdS(2i˙˙˙˙N(tSettHtKtRtDtEtStTtNtQtAtVtLtItMtFtYtWtPtGtCtHIStLYStARGtASPtGLUtSERtTHRtASNtGLNtALAtVALtLEUtILEtMETtPHEtTYRtTRPtPROtGLYtCYStPDB_atomcBseZd„Zd„ZRS(c CsL||_||_||_||_||_||_||_||_dS(N(tatomtrestchain_IDtxtytztindextvalue( tselft atom_typeR+R,R-R.R/R0R1((sx/cstor/stanford/rbaltman/users/wtorng/DEEP_LEARNING/3DCNN/FSCNN_models/User_test/code/extract_all_res_defined_AA_type.pyt__init__s       cCs|j|jkS(N(t__dict__(R2tother((sx/cstor/stanford/rbaltman/users/wtorng/DEEP_LEARNING/3DCNN/FSCNN_models/User_test/code/extract_all_res_defined_AA_type.pyt__eq__s(t__name__t __module__R4R7(((sx/cstor/stanford/rbaltman/users/wtorng/DEEP_LEARNING/3DCNN/FSCNN_models/User_test/code/extract_all_res_defined_AA_type.pyR)s cCs:i}x-|D]%}|j|j|jg||j||jt| |||d|d |dd|dd ƒƒ|jt| |||d|d |dd|dd ƒƒ|d 7}n| ddkrCPqCqCW|| | g}|S(NitATOMi it iiiiiii%i&i-i.i5R3R+R,R-R.R/iR0R1tENDMDL(tsplittstriptinttfloatRBRAR>R)(t entry_listRHtall_post all_linest all_atom_typet PDB_entriest atom_indextmodel_IDtMODELStSEQRGtline1tlineR*R+R,tchaintres_notnew_postlR<tMODEL((sx/cstor/stanford/rbaltman/users/wtorng/DEEP_LEARNING/3DCNN/FSCNN_models/User_test/code/extract_all_res_defined_AA_type.pytgrab_PDB5sN       6  E  >: cCsźd|dGHd|d|d|d}t|ƒ}t|dƒ}x\|D]T}|jdƒd}|GH|d |jƒd } tjj| ƒr—t| ƒ} t| ƒ} t| ƒ} | \} }}|}t||||| ƒ\}}|rĄxŸ|D]”}|\}}}|d}|d }|d }|j |d t |ƒd t |ƒd t |ƒd |d |dd t |d ƒdƒqůWqĄqMdGH|GHqMW|j ƒd|GH|S(NsExtracting all s residues from PDBs in pdb_lists ../data/ptf/t_s.ptftwt.it/s.pdbiis s sPDB not found!s Writing extracting positions in ( topenRUtlowertostpathtisfiletlistRiRQtwritetstrtclose(tpdb_listtpdb_dirRDREtsitetptf_nametptf_fileRctPDB_IDt pdb_file_nametpdb_fileRgRhRHRaRGRFt if_patternRKtp_RNROtpR-R.R/((sx/cstor/stanford/rbaltman/users/wtorng/DEEP_LEARNING/3DCNN/FSCNN_models/User_test/code/extract_all_res_defined_AA_type.pytextract_all_res_defined_AA_typefs6          k   ((tmathtnumpyRptsysttheanot theano.tensorttensorRt collectionstsetsRtscipytgziptXR>R)R=RQRiR‚(((sx/cstor/stanford/rbaltman/users/wtorng/DEEP_LEARNING/3DCNN/FSCNN_models/User_test/code/extract_all_res_defined_AA_type.pyts        B’   1