def modifyTransmute(self, elem, force = False, atomType=None): """ This method was originally a method of class mode and selectMode. Transmute selected atoms into <elem> and with an optional <atomType>. <elem> is an element number that selected atoms will be transmuted to. <force>: boolean variable meaning keeping existing bond or not. <atomType>: the optional hybrid bond type if the element support hybrid. --Huaicai[9/1/05] """ # now change selected atoms to the specified element # [bruce 041215: this should probably be made available for any modes # in which "selected atoms" are permitted, not just Select modes. #e] from model.elements import PeriodicTable if self.selatoms: dstElem = PeriodicTable.getElement(elem) for atm in self.selatoms.values(): atm.Transmute(dstElem, force = force, atomtype=atomType) # bruce 041215 fix bug 131 by replacing low-level mvElement call # with new higher-level method Transmute. Note that singlets # can't be selected, so the fact that Transmute does nothing to # them is not (presently) relevant. #e status message? # (Presently a.Transmute makes one per "error or refusal".) self.o.gl_update() if self.selmols: #bruce 060720 elif -> if, in case both atoms and chunks can be selected someday dstElem = PeriodicTable.getElement(elem) #bruce 060720 fix typo dstElm -> dstElem to fix bug 2149 # but we have to decide if we want the behavior this now gives us, of transmuting inside selected chunks. for mol in self.selmols[:]: for atm in mol.atoms.values(): atm.Transmute(dstElem, force = force, atomtype=atomType) # this might run on some killed singlets; should be ok self.o.gl_update() return
def update_hybrid_btngrp(self, buttonIndex = 0): """Update the buttons of the current element\'s hybridization types into hybrid_btngrp; select the specified one if provided""" elem = PeriodicTable.getElement(self.w.Element) # self.w.Element is atomic number atypes = elem.atomtypes if elem.name == 'Carbon': self.setup_C_hybrid_buttons() elif elem.name == 'Nitrogen': self.setup_N_hybrid_buttons() elif elem.name == 'Oxygen': self.setup_O_hybrid_buttons() elif elem.name == 'Sulfur': self.setup_S_hybrid_buttons() else: self.hide_hybrid_btngrp() self.elemGLPane.changeHybridType(None) return #if len(atypes) > 1: # Prequisite: w.hybridComboBox has been updated at this moment. b_name = self.bond_id2name[buttonIndex] self.elemGLPane.changeHybridType(b_name) self.elemGLPane.refreshDisplay(self.elm, self.displayMode) self.theHybridizations.button(buttonIndex).setChecked(True) self.set_hybrid_type(buttonIndex) # Added Atomic Hybrids label. Mark 2007-05-30 self.atomic_hybrids_label.setText("Atomic Hybrids for " + elem.name + " :") self.show_hybrid_btngrp()
def update_selection_filter_list(self): """Adds/removes the element selected in the MMKit to/from Atom Selection Filter based on what modifier key is pressed (if any). """ eltnum = self.w.Element if self.o.modkeys is None: self.w.filtered_elements = [] self.w.filtered_elements.append(PeriodicTable.getElement(eltnum)) if self.o.modkeys == 'Shift': if not PeriodicTable.getElement(eltnum) in self.w.filtered_elements[:]: self.w.filtered_elements.append(PeriodicTable.getElement(eltnum)) elif self.o.modkeys == 'Control': if PeriodicTable.getElement(eltnum) in self.w.filtered_elements[:]: self.w.filtered_elements.remove(PeriodicTable.getElement(eltnum)) self.update_selection_filter_list_widget()
def write_art_data(fileHandle): """ Writes the Atom Rendering Table (ART) data, which contains all the atom rendering properties needed by QuteMolX, to the file with the given fileHandle. Each atom is on a separate line. Lines starting with '#' are comment lines. """ fileHandle.write("""\ REMARK 8 REMARK 8 ;NanoEngineer-1 Atom Rendering Table (format version 0.1.0) REMARK 8 ;This table specifies the scene rendering scheme as employed by REMARK 8 ;NanoEngineer-1 (NE1) at the time this file was created. REMARK 8 REMARK 8 ;Note: All CPK radii were calculated using a CPK scaling factor that REMARK 8 ;can be modified by the user from "Preferences... | Atoms".\n""") fileHandle.write("REMARK 8 ;This table's CPK Scaling Factor: %2.3f" % env.prefs[cpkScaleFactor_prefs_key]) fileHandle.write(""" REMARK 8 ;To compute the original van der Waals radii, use the formula: REMARK 8 ; vdW Radius = CPK Radius / CPK Scaling Factor REMARK 8 REMARK 8 ;Atom Name NE1 Atom CPK Radius Ball and Stick Color (RGB) REMARK 8 ; Number Radius\n""") elementTable = PeriodicTable.getAllElements() for elementNumber, element in elementTable.items(): color = element.color r = int(color[0] * 255 + 0.5) g = int(color[1] * 255 + 0.5) b = int(color[2] * 255 + 0.5) # The following was distilled from chem.py: Atom.howdraw() # # "Render Radius" cpkRadius = \ element.rvdw * env.prefs[cpkScaleFactor_prefs_key] # "Covalent Radius" ballAndStickRadius = \ element.rvdw * 0.25 * env.prefs[diBALL_AtomRadius_prefs_key] #if element.symbol == 'Ax3': # ballAndStickRadius = 0.1 fileHandle.write \ ("REMARK 8 %-3s %-3d %3.3f %3.3f %3d %3d %3d\n" % (element.symbol, elementNumber, cpkRadius, ballAndStickRadius, r, g, b)) fileHandle.close() return
def write_art_data(fileHandle): """ Writes the Atom Rendering Table (ART) data, which contains all the atom rendering properties needed by QuteMolX, to the file with the given fileHandle. Each atom is on a separate line. Lines starting with '#' are comment lines. """ fileHandle.write("""\ REMARK 8 REMARK 8 ;NanoEngineer-1 Atom Rendering Table (format version 0.1.0) REMARK 8 ;This table specifies the scene rendering scheme as employed by REMARK 8 ;NanoEngineer-1 (NE1) at the time this file was created. REMARK 8 REMARK 8 ;Note: All CPK radii were calculated using a CPK scaling factor that REMARK 8 ;can be modified by the user from "Preferences... | Atoms".\n""") fileHandle.write("REMARK 8 ;This table's CPK Scaling Factor: %2.3f" % env.prefs[cpkScaleFactor_prefs_key]) fileHandle.write(""" REMARK 8 ;To compute the original van der Waals radii, use the formula: REMARK 8 ; vdW Radius = CPK Radius / CPK Scaling Factor REMARK 8 REMARK 8 ;Atom Name NE1 Atom CPK Radius Ball and Stick Color (RGB) REMARK 8 ; Number Radius\n""") elementTable = PeriodicTable.getAllElements() for elementNumber, element in elementTable.items(): color = element.color r = int(color[0] * 255 + 0.5) g = int(color[1] * 255 + 0.5) b = int(color[2] * 255 + 0.5) # The following was distilled from chem.py: Atom.howdraw() # # "Render Radius" cpkRadius = \ element.rvdw * env.prefs[cpkScaleFactor_prefs_key] # "Covalent Radius" ballAndStickRadius = \ element.rvdw * 0.25 * env.prefs[diBALL_AtomRadius_prefs_key] #if element.symbol == 'Ax3': # ballAndStickRadius = 0.1 fileHandle.write \ ("REMARK 8 %-3s %-3d %3.3f %3.3f %3d %3d %3d\n" % (element.symbol, elementNumber, cpkRadius, ballAndStickRadius, r, g, b)) fileHandle.close() return
def _readpdb(assy, filename, isInsert = False, showProgressDialog = False, chainId = None): """ Read a Protein DataBank-format file into a single new chunk, which is returned unless there are no atoms in the file, in which case a warning is printed and None is returned. (The new chunk (if returned) is in assy, but is not yet added into any Group or Part in assy -- caller must do that.) Unless isInsert = True, set assy.filename to match the file we read, even if we return None. @param assy: The assembly. @type assy: L{assembly} @param filename: The PDB filename to read. @type filename: string @param isInsert: If True, the PDB file will be inserted into the current assembly. If False (default), the PDB is opened as the assembly. @param isInsert: boolean @param showProgressDialog: if True, display a progress dialog while reading a file. @type showProgressDialog: boolean @return: A chunk containing the contents of the PDB file. @rtype: L{Chunk} @see: U{B{PDB File Format}<http://www.wwpdb.org/documentation/format23/v2.3.html>} """ fi = open(filename,"rU") lines = fi.readlines() fi.close() dir, nodename = os.path.split(filename) if not isInsert: assy.filename = filename ndix = {} mol = Chunk(assy, nodename) numconects = 0 atomname_exceptions = { "HB":"H", #k these are all guesses -- I can't find this documented # anywhere [bruce 070410] ## "HE":"H", ### REVIEW: I'm not sure about this one -- ### leaving it out means it's read as Helium, # but including it erroneously might prevent reading an actual Helium # if that was intended. # Guess for now: include it for ATOM but not HETATM. (So it's # specialcased below, rather than being included in this table.) # (Later: can't we use the case of the 'E' to distinguish it from He?) "HN":"H", } # Create and display a Progress dialog while reading the MMP file. # One issue with this implem is that QProgressDialog always displays # a "Cancel" button, which is not hooked up. I think this is OK for now, # but later we should either hook it up or create our own progress # dialog that doesn't include a "Cancel" button. --mark 2007-12-06 if showProgressDialog: _progressValue = 0 _progressFinishValue = len(lines) win = env.mainwindow() win.progressDialog.setLabelText("Reading file...") win.progressDialog.setRange(0, _progressFinishValue) _progressDialogDisplayed = False _timerStart = time.time() for card in lines: key = card[:6].lower().replace(" ", "") if key in ["atom", "hetatm"]: ## sym = capitalize(card[12:14].replace(" ", "").replace("_", "")) # bruce 080508 revision (guess at a bugfix for reading NE1-saved # pdb files): # get a list of atomnames to try; use the first one we recognize. # Note that full atom name is in columns 13-16 i.e. card[12:16]; # see http://www.wwpdb.org/documentation/format2.3-0108-us.pdf, # page 156. The old code only looked at two characters, # card[12:14] == columns 13-14, and discarded ' ' and '_', # and capitalized (the first character only). The code as I revised # it on 070410 also discarded digits, and handled HB, HE, HN # (guesses) using the atomname_exceptions dict. name4 = card[12:16].replace(" ", "").replace("_", "") name3 = card[12:15].replace(" ", "").replace("_", "") name2 = card[12:14].replace(" ", "").replace("_", "") def nodigits(name): for bad in "0123456789": name = name.replace(bad, "") return name atomnames_to_try = [ name4, # as seems best according to documentation name3, name2, # like old code nodigits(name4), nodigits(name3), nodigits(name2) # like code as revised on 070410 ] foundit = False for atomname in atomnames_to_try: atomname = atomname_exceptions.get(atomname, atomname) if atomname == "HE" and key == "atom": atomname = "H" # see comment in atomname_exceptions sym = capitalize(atomname) # turns either 'he' or 'HE' into 'He' try: PeriodicTable.getElement(sym) except: # note: this typically fails with AssertionError # (not e.g. KeyError) [bruce 050322] continue else: foundit = True break pass if not foundit: msg = "Warning: Pdb file: will use Carbon in place of unknown element %s in: %s" \ % (name4, card) print msg #bruce 070410 added this print env.history.message( redmsg( msg )) ##e It would probably be better to create a fake atom, so the # CONECT records would still work. #bruce 080508 let's do that: sym = "C" # Better still might be to create a fake element, # so we could write out the pdb file again # (albeit missing lots of info). [bruce 070410 comment] # Note: an advisor tells us: # PDB files sometimes encode atomtypes, # using C_R instead of C, for example, to represent sp2 # carbons. # That particular case won't trigger this exception, since we # only look at 2 characters [eventually, after trying more, as of 080508], # i.e. C_ in that case. It would be better to realize this means # sp2 and set the atomtype here (and perhaps then use it when # inferring bonds, which we do later if the file doesn't have # any bonds). [bruce 060614/070410 comment] # Now the element name is in sym. xyz = map(float, [card[30:38], card[38:46], card[46:54]] ) n = int(card[6:11]) a = Atom(sym, A(xyz), mol) ndix[n] = a elif key == "conect": try: a1 = ndix[int(card[6:11])] except: #bruce 050322 added this level of try/except and its message; # see code below for at least two kinds of errors this might # catch, but we don't try to distinguish these here. BTW this # also happens as a consequence of not finding the element # symbol, above, since atoms with unknown elements are not # created. env.history.message( redmsg( "Warning: Pdb file: can't find first atom in CONECT record: %s" % (card,) )) else: for i in range(11, 70, 5): try: a2 = ndix[int(card[i:i+5])] except ValueError: # bruce 050323 comment: # we assume this is from int('') or int(' ') etc; # this is the usual way of ending this loop. break except KeyError: #bruce 050322-23 added history warning for this, # assuming it comes from ndix[] lookup. env.history.message( redmsg( "Warning: Pdb file: can't find atom %s in: %s" % (card[i:i+5], card) )) continue bond_atoms(a1, a2) numconects += 1 if showProgressDialog: # Update the progress dialog. _progressValue += 1 if _progressValue >= _progressFinishValue: win.progressDialog.setLabelText("Building model...") elif _progressDialogDisplayed: win.progressDialog.setValue(_progressValue) else: _timerDuration = time.time() - _timerStart if _timerDuration > 0.25: # Display progress dialog after 0.25 seconds win.progressDialog.setValue(_progressValue) _progressDialogDisplayed = True if showProgressDialog: # Make the progress dialog go away. win.progressDialog.setValue(_progressFinishValue) #bruce 050322 part of fix for bug 433: don't return an empty chunk if not mol.atoms: env.history.message( redmsg( "Warning: Pdb file contained no atoms")) return None if numconects == 0: msg = orangemsg("PDB file has no bond info; inferring bonds") env.history.message(msg) # let user see message right away (bond inference can take significant # time) [bruce 060620] env.history.h_update() inferBonds(mol) return mol
def _get_atomlist_from_gms_outfile(assy, filename): """ Read the atoms from a GAMESS OUT file into an atom list, which is returned, unless there are no atoms in the file, in which case a warning is printed and None is returned. """ fi = open(filename, "rU") lines = fi.readlines() fi.close() dir, nodename = os.path.split(filename) mol = Chunk(assy, nodename) newAtomList = [] countdown = 0 equilibruim_found = False atoms_found = False for card in lines: if failpat.search( card): # GAMESS Aborted. No atom data will be found. print card env.history.message(redmsg(card)) break if noconvpat.search(card): # Geometry search is not converged. print card env.history.message(redmsg(card)) break # If this card is found: # "1 ***** EQUILIBRIUM GEOMETRY LOCATED *****\n" # we know we have a successfully optimized structure/set of atoms. # If this card is not found, the optimization failed for some reason. # Atom positions begin soon after this card. if card == "1 ***** EQUILIBRIUM GEOMETRY LOCATED *****\n": equilibruim_found = True continue # The atom positions we want ALWAYS begin 2 lines after this card: # " COORDINATES OF ALL ATOMS ARE (ANGS)\n" # which follows the previous card. # This is one way to fix the problem mentioned above. # I've commented the code below out since it needs further work to do what # we need, and there is a chance we will not need this if GAMESS-US has # the same number of lines (6) after the "EQUILIBRIUM" card above. # # P.S. The reason we do not just look for this card by itself is that there # can be many of them. There is only one "EQUILIBRIUM" card, and the # good atoms follow that card. # 050624 Mark if equilibruim_found: if card == " COORDINATES OF ALL ATOMS ARE (ANGS)\n": atoms_found = True reading_atoms = True countdown = 2 continue if not equilibruim_found or not atoms_found: continue if countdown: countdown -= 1 # print countdown, card # for debugging only. continue # The current card contains atom type and position. n = 0 if reading_atoms: # print "_get_atomlist_from_gms_outfile:", card if len(card) < 10: reading_atoms = False # Finished reading atoms. break m = irecpat.match(card) sym = capitalize(m.group(1)) try: PeriodicTable.getElement(sym) except: env.history.message( redmsg( "Warning: GAMESS OUT file: unknown element %s in: %s" % (sym, card))) else: xyz = map(float, (m.group(2), m.group(3), m.group(4))) a = Atom(sym, A(xyz), mol) newAtomList += [a] # Let caller handle history msgs. Mark 050712 # if not newAtomList: # msg = "Warning: GAMESS file contains no equilibrium geometry. No atoms read into part." # env.history.message( redmsg(msg)) # return None return newAtomList
def __init__(self, win): qt4todo('what to do with all those options?') ## ElementColorsDialog.__init__(self, win, None, 0, ## Qt.WStyle_Customize | Qt.WStyle_NormalBorder | ## Qt.WStyle_Title | Qt.WStyle_SysMenu) QDialog.__init__(self, win) self.setupUi(self) self.connect(self.okButton, SIGNAL("clicked()"), self.ok) self.connect(self.loadColorsPB, SIGNAL("clicked()"), self.read_element_rgb_table) self.connect(self.saveColorsPB, SIGNAL("clicked()"), self.write_element_rgb_table) self.connect(self.cancelButton, SIGNAL("clicked()"), self.reject) self.connect(self.defaultButton, SIGNAL("clicked()"), self.loadDefaultProp) self.connect(self.alterButton, SIGNAL("clicked()"), self.loadAlterProp) self.connect(self.elementButtonGroup, SIGNAL("clicked(int)"), self.setElementInfo) self.connect(self.previewPB, SIGNAL("clicked()"), self.preview_color_change) self.connect(self.restorePB, SIGNAL("clicked()"), self.restore_current_color) self.w = win self.fileName = None self.isElementModified = False self.isFileSaved = False self.oldTable = PeriodicTable.deepCopy() self.elemTable = PeriodicTable self.displayMode = self._displayList[0] # The next line fixes a bug. Thumbview expects self.gridLayout on # line 117 of Thumbview.py. Mark 2007-10-19. self.gridLayout = self.gridlayout self.elemGLPane = ElementView(self, "element glPane", self.w.glpane) # Put the GL widget inside the frame flayout = QVBoxLayout(self.elementFrame) flayout.setMargin(1) flayout.setSpacing(1) flayout.addWidget(self.elemGLPane, 1) def elementId(symbol): return PeriodicTable.getElement(symbol).eltnum self.toolButton6.setChecked(True) self.elementButtonGroup.setId(self.toolButton6, elementId("C")) self.elementButtonGroup.setId(self.toolButton8, elementId("O")) self.elementButtonGroup.setId(self.toolButton10, elementId("Ne")) self.elementButtonGroup.setId(self.toolButton9, elementId("F")) self.elementButtonGroup.setId(self.toolButton13, elementId("Al")) self.elementButtonGroup.setId(self.toolButton17, elementId("Cl")) self.elementButtonGroup.setId(self.toolButton5, elementId("B")) self.elementButtonGroup.setId(self.toolButton10_2, elementId("Ar")) self.elementButtonGroup.setId(self.toolButton15, elementId("P")) self.elementButtonGroup.setId(self.toolButton16, elementId("S")) self.elementButtonGroup.setId(self.toolButton14, elementId("Si")) self.elementButtonGroup.setId(self.toolButton33, elementId("As")) self.elementButtonGroup.setId(self.toolButton34, elementId("Se")) self.elementButtonGroup.setId(self.toolButton35, elementId("Br")) self.elementButtonGroup.setId(self.toolButton36, elementId("Kr")) self.elementButtonGroup.setId(self.toolButton32, elementId("Ge")) self.elementButtonGroup.setId(self.toolButton7, elementId("N")) self.elementButtonGroup.setId(self.toolButton2, elementId("He")) self.elementButtonGroup.setId(self.toolButton1, elementId("H")) self.elementButtonGroup.setId(self.toolButton0, elementId("X")) self.connect(self.toolButton6, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton8, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton10, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton9, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton13, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton17, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton5, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton10_2, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton15, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton16, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton14, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton33, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton34, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton35, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton36, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton32, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton7, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton2, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton1, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton0, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connectChangingControls() self.saveColorsPB.setWhatsThis( """Save the current color settings for elements in a text file.""") self.defaultButton.setWhatsThis( """Restore current element colors to the default colors.""") self.loadColorsPB.setWhatsThis( """Load element colors from an external text file.""") self.alterButton.setWhatsThis( """Set element colors to the alternate color set.""")
def _readpdb_new(assy, filename, isInsert = False, showProgressDialog = False, chainId = None): """ Read a Protein DataBank-format file into a single new chunk, which is returned unless there are no atoms in the file, in which case a warning is printed and None is returned. (The new chunk (if returned) is in assy, but is not yet added into any Group or Part in assy -- caller must do that.) Unless isInsert = True, set assy.filename to match the file we read, even if we return None. @param assy: The assembly. @type assy: L{assembly} @param filename: The PDB filename to read. @type filename: string @param isInsert: If True, the PDB file will be inserted into the current assembly. If False (default), the PDB is opened as the assembly. @param isInsert: boolean @param showProgressDialog: if True, display a progress dialog while reading a file. @type showProgressDialog: boolean @return: A chunk containing the contents of the PDB file. @rtype: L{Chunk} @see: U{B{PDB File Format}<http://www.wwpdb.org/documentation/format23/v2.3.html>} """ from protein.model.Protein import is_water def _finish_molecule(): """ Perform some operations after reading entire PDB chain: - rebuild (infer) bonds - rename molecule to reflect a chain ID - delete protein object if this is not a protein - append the molecule to the molecule list """ if mol == water: # Skip water, to be added explicitly at the end. return if mol.atoms: ###print "READING PDB ", (mol, numconects, chainId) mol.name = pdbid.lower() + chainId ###idzialprint "SEQ = ", mol.protein.get_sequence_string() ###print "SEC = ", mol.protein.get_secondary_structure_string() if mol.protein.count_c_alpha_atoms() == 0: # If there is no C-alpha atoms, consider the chunk # as a non-protein. But! Split it into individual # hetero groups. res_list = mol.protein.get_amino_acids() assy.part.ensure_toplevel_group() hetgroup = Group("Heteroatoms", assy, assy.part.topnode) for res in res_list: hetmol = Chunk(assy, res.get_three_letter_code().replace(" ", "") + \ "[" + str(res.get_id()) + "]") for atom in res.get_atom_list(): newatom = Atom(atom.element.symbol, atom.posn(), hetmol) # New chunk - infer the bonds anyway (this is not # correct, should first check connectivity read from # the PDB file CONECT records). inferBonds(hetmol) hetgroup.addchild(hetmol) mollist.append(hetgroup) else: #if numconects == 0: # msg = orangemsg("PDB file has no bond info; inferring bonds") # env.history.message(msg) # # let user see message right away (bond inference can take significant # # time) [bruce 060620] # env.history.h_update() # For protein - infer the bonds anyway. inferBonds(mol) mol.protein.set_chain_id(chainId) mol.protein.set_pdb_id(pdbid) if mol.atoms: mollist.append(mol) else: env.history.message( redmsg( "Warning: Pdb file contained no atoms")) env.history.h_update() fi = open(filename,"rU") lines = fi.readlines() fi.close() mollist = [] # Lists of secondary structure tuples (res_id, chain_id) helix = [] sheet = [] turn = [] dir, nodename = os.path.split(filename) if not isInsert: assy.filename = filename ndix = {} mol = Chunk(assy, nodename) mol.protein = Protein() # Create a chunk for water molecules. water = Chunk(assy, nodename) numconects = 0 comment_text = "" _read_rosetta_info = False # Create a temporary PDB ID - it should be later extracted from the # file header. pdbid = nodename.replace(".pdb","").lower() atomname_exceptions = { "HB":"H", #k these are all guesses -- I can't find this documented # anywhere [bruce 070410] "CA":"C", #k these are all guesses -- I can't find this documented ## "HE":"H", ### REVIEW: I'm not sure about this one -- ### leaving it out means it's read as Helium, # but including it erroneously might prevent reading an actual Helium # if that was intended. # Guess for now: include it for ATOM but not HETATM. (So it's # specialcased below, rather than being included in this table.) # (Later: can't we use the case of the 'E' to distinguish it from He?) "HN":"H", } # Create and display a Progress dialog while reading the MMP file. # One issue with this implem is that QProgressDialog always displays # a "Cancel" button, which is not hooked up. I think this is OK for now, # but later we should either hook it up or create our own progress # dialog that doesn't include a "Cancel" button. --mark 2007-12-06 if showProgressDialog: _progressValue = 0 _progressFinishValue = len(lines) win = env.mainwindow() win.progressDialog.setLabelText("Reading file...") win.progressDialog.setRange(0, _progressFinishValue) _progressDialogDisplayed = False _timerStart = time.time() for card in lines: key = card[:6].lower().replace(" ", "") if key in ["atom", "hetatm"]: ## sym = capitalize(card[12:14].replace(" ", "").replace("_", "")) # bruce 080508 revision (guess at a bugfix for reading NE1-saved # pdb files): # get a list of atomnames to try; use the first one we recognize. # Note that full atom name is in columns 13-16 i.e. card[12:16]; # see http://www.wwpdb.org/documentation/format2.3-0108-us.pdf, # page 156. The old code only looked at two characters, # card[12:14] == columns 13-14, and discarded ' ' and '_', # and capitalized (the first character only). The code as I revised # it on 070410 also discarded digits, and handled HB, HE, HN # (guesses) using the atomname_exceptions dict. name4 = card[12:16].replace(" ", "").replace("_", "") name3 = card[12:15].replace(" ", "").replace("_", "") name2 = card[12:14].replace(" ", "").replace("_", "") chainId = card[21] resIdStr = card[22:26].replace(" ", "") if resIdStr != "": resId = int(resIdStr) else: resId = 0 resName = card[17:20] sym = card[77:78] alt = card[16] # Alternate location indicator if alt != ' ' and \ alt != 'A': # Skip non-standard alternate location # This is not very safe test, it should preserve # the remaining atoms. piotr 080715 continue ###ATOM 131 CB ARG A 18 104.359 32.924 58.573 1.00 36.93 C def nodigits(name): for bad in "0123456789": name = name.replace(bad, "") return name atomnames_to_try = [ name4, # as seems best according to documentation name3, name2, # like old code nodigits(name4), nodigits(name3), nodigits(name2) # like code as revised on 070410 ] # First, look at 77-78 field - it should include an element symbol. foundit = False try: PeriodicTable.getElement(sym) except: pass else: foundit = True if not foundit: for atomname in atomnames_to_try: atomname = atomname_exceptions.get(atomname, atomname) if atomname[0] == 'H' and key == "atom": atomname = "H" # see comment in atomname_exceptions sym = capitalize(atomname) # turns either 'he' or 'HE' into 'He' try: PeriodicTable.getElement(sym) except: # note: this typically fails with AssertionError # (not e.g. KeyError) [bruce 050322] continue else: foundit = True break pass if not foundit: msg = "Warning: Pdb file: will use Carbon in place of unknown element %s in: %s" \ % (name4, card) print msg #bruce 070410 added this print env.history.message( redmsg( msg )) ##e It would probably be better to create a fake atom, so the # CONECT records would still work. #bruce 080508 let's do that: sym = "C" # Better still might be to create a fake element, # so we could write out the pdb file again # (albeit missing lots of info). [bruce 070410 comment] # Note: an advisor tells us: # PDB files sometimes encode atomtypes, # using C_R instead of C, for example, to represent sp2 # carbons. # That particular case won't trigger this exception, since we # only look at 2 characters [eventually, after trying more, as of 080508], # i.e. C_ in that case. It would be better to realize this means # sp2 and set the atomtype here (and perhaps then use it when # inferring bonds, which we do later if the file doesn't have # any bonds). [bruce 060614/070410 comment] _is_water = is_water(resName, name4) if _is_water: tmpmol = mol mol = water # Now the element name is in sym. xyz = map(float, [card[30:38], card[38:46], card[46:54]] ) n = int(card[6:11]) a = Atom(sym, A(xyz), mol) ndix[n] = a if not _is_water: mol.protein.add_pdb_atom(a, name4, resId, resName) # Assign secondary structure. if (resId, chainId) in helix: mol.protein.assign_helix(resId) if (resId, chainId) in sheet: mol.protein.assign_strand(resId) if (resId, chainId) in turn: mol.protein.assign_turn(resId) if mol == water: mol = tmpmol elif key == "conect": try: a1 = ndix[int(card[6:11])] except: #bruce 050322 added this level of try/except and its message; # see code below for at least two kinds of errors this might # catch, but we don't try to distinguish these here. BTW this # also happens as a consequence of not finding the element # symbol, above, since atoms with unknown elements are not # created. env.history.message( redmsg( "Warning: Pdb file: can't find first atom in CONECT record: %s" % (card,) )) else: for i in range(11, 70, 5): try: a2 = ndix[int(card[i:i+5])] except ValueError: # bruce 050323 comment: # we assume this is from int('') or int(' ') etc; # this is the usual way of ending this loop. break except KeyError: #bruce 050322-23 added history warning for this, # assuming it comes from ndix[] lookup. env.history.message( redmsg( "Warning: Pdb file: can't find atom %s in: %s" % (card[i:i+5], card) )) continue bond_atoms(a1, a2) numconects += 1 elif key == "ter": # Finish the current molecule. _finish_molecule() # Discard the original molecule and create a new one. mol = Chunk(assy, nodename) mol.protein = Protein() numconects = 0 elif key == "header": # Extract PDB ID from the header string. pdbid = card[62:66].lower() comment_text += card elif key == "compnd": comment_text += card elif key == "remark": comment_text += card elif key == "model": # Check out the MODEL record, ignore everything other than MODEL 1. # This behavior has to be optional and set via User Preference. # piotr 080714 model_id = int(card[6:20]) if model_id > 1: # Skip remaining part of the file. break elif key in ["helix", "sheet", "turn"]: # Read secondary structure information. if key == "helix": begin = int(card[22:25]) end = int(card[34:37]) chainId = card[19] for s in range(begin, end+1): helix.append((s, chainId)) elif key == "sheet": begin = int(card[23:26]) end = int(card[34:37]) chainId = card[21] for s in range(begin, end+1): sheet.append((s, chainId)) elif key == "turn": begin = int(card[23:26]) end = int(card[34:37]) chainId = card[19] for s in range(begin, end+1): turn.append((s, chainId)) else: if card[7:15] == "ntrials:": _read_rosetta_info = True comment_text += "Rosetta Scoring Analysis\n" if _read_rosetta_info: comment_text += card if showProgressDialog: # Update the progress dialog. _progressValue += 1 if _progressValue >= _progressFinishValue: win.progressDialog.setLabelText("Building model...") elif _progressDialogDisplayed: win.progressDialog.setValue(_progressValue) else: _timerDuration = time.time() - _timerStart if _timerDuration > 0.25: # Display progress dialog after 0.25 seconds win.progressDialog.setValue(_progressValue) _progressDialogDisplayed = True if showProgressDialog: # Make the progress dialog go away. win.progressDialog.setValue(_progressFinishValue) _finish_molecule() if water.atoms: # Check if there are any water molecules water.name = "Solvent" # The water should be hidden by default. water.hide() mollist.append(water) return (mollist, comment_text)
def _readgms(assy, filename, isInsert=False): """ Read the atoms from a GAMESS DAT file into a single new chunk, which is returned, unless there are no atoms in the file, in which case a warning is printed and None is returned. (The new chunk (if returned) is in assy, but is not yet added into any Group or Part in assy -- caller must do that.) """ fi = open(filename,"rU") lines = fi.readlines() fi.close() dir, nodename = os.path.split(filename) ndix = {} mol = Chunk(assy, nodename) countdown = 0 equilibruim_found = False atoms_found = False for card in lines: if failpat.search(card): # GAMESS Aborted. No atom data will be found. print card break # If this card is found: # "1 ***** EQUILIBRIUM GEOMETRY LOCATED *****\n" # we know we have a successfully optimized structure/set of atoms. # If this card is not found, the optimization failed for some reason. # Atom positions begin soon after this card. if card == "1 ***** EQUILIBRIUM GEOMETRY LOCATED *****\n": equilibruim_found = True continue # The atom positions we want ALWAYS begin 2 lines after this card: # " COORDINATES OF ALL ATOMS ARE (ANGS)\n" # which follows the previous card. # This is one way to fix the problem mentioned above. # I've commented the code below out since it needs further work to do what # we need, and there is a chance we will not need this if GAMESS-US has # the same number of lines (6) after the "EQUILIBRIUM" card above. # # P.S. The reason we do not just look for this card by itself is that there # can be many of them. There is only one "EQUILIBRIUM" card, and the # good atoms follow that card. # 050624 Mark if equilibruim_found: if card == " COORDINATES OF ALL ATOMS ARE (ANGS)\n": atoms_found = True reading_atoms = True countdown = 2 continue if not equilibruim_found or not atoms_found: continue if countdown: countdown -= 1 # print countdown, card # for debugging only. continue # The current card contains atom type and position. n = 0 if reading_atoms: if len(card)<10: reading_atoms = False # Finished reading atoms. break m=irecpat.match(card) sym = capitalize(m.group(1)) try: PeriodicTable.getElement(sym) except: env.history.message( redmsg( "Warning: GAMESS DAT file: unknown element %s in: %s" % (sym,card) )) else: xyz = map(float, (m.group(2),m.group(3), m.group(4))) a = Atom(sym, A(xyz), mol) ndix[n] = a n += 1 # Don't return an empty chunk. if not mol.atoms: msg = "Warning: GAMESS file contains no equilibrium geometry. No atoms read into part." env.history.message( redmsg(msg)) return None # Need to compute and add bonds for this chunk. I'll ask Bruce how to best accomplish this. # In the meantime, let's warn the user that no bonds have been formed since it # is impossible to see this in vdW display mode. # Mark 050623. msg = "Warning: Equilibrium geometry found. Atoms read into part, but there are no bonds." env.history.message( orangemsg(msg)) return mol
from utilities.exception_classes import PluginBug from utilities.constants import gensym from utilities.prefs_constants import dnaDefaultStrand1Color_prefs_key from utilities.prefs_constants import dnaDefaultStrand2Color_prefs_key from utilities.prefs_constants import dnaDefaultSegmentColor_prefs_key from dna.model.Dna_Constants import getDuplexBasesPerTurn ##from dna.updater.dna_updater_prefs import pref_dna_updater_convert_to_PAM3plus5 from simulation.sim_commandruns import adjustSinglet from model.elements import PeriodicTable from model.Line import Line from model.chem import Atom_prekill_prep Element_Ae3 = PeriodicTable.getElement('Ae3') from dna.model.Dna_Constants import basesDict, dnaDict from dna.model.dna_model_constants import LADDER_END0 basepath_ok, basepath = find_plugin_dir("DNA") if not basepath_ok: env.history.message(orangemsg("The cad/plugins/DNA directory is missing.")) RIGHT_HANDED = -1 LEFT_HANDED = 1 from geometry.VQT import V, Q, norm, cross from geometry.VQT import vlen from Numeric import dot
def build_struct(self, name, params, position, mol=None, createPrinted=False): """ Build a nanotube from the parameters in the Property Manger dialog. """ length, n, m, bond_length, zdist, xydist, \ twist, bend, members, endings, numwalls, spacing = params # This can take a few seconds. Inform the user. # 100 is a guess on my part. Mark 051103. if not createPrinted: # If it's a multi-wall tube, only print the "Creating" message once. if length > 100.0: env.history.message(self.cmd + "This may take a moment...") self.chirality = Chirality(n, m, bond_length) PROFILE = False if PROFILE: sw = Stopwatch() sw.start() xyz = self.chirality.xyz if mol == None: mol = Chunk(self.win.assy, name) atoms = mol.atoms mlimits = self.chirality.mlimits # populate the tube with some extra carbons on the ends # so that we can trim them later self.chirality.populate(mol, length + 4 * self.chirality.maxlen, members != 0) # Apply twist and distortions. Bends probably would come # after this point because they change the direction for the # length. I'm worried about Z distortion because it will work # OK for stretching, but with compression it can fail. BTW, # "Z distortion" is a misnomer, we're stretching in the Y # direction. for atm in atoms.values(): # twist x, y, z = atm.posn() twistRadians = twist * z c, s = cos(twistRadians), sin(twistRadians) x, y = x * c + y * s, -x * s + y * c atm.setposn(V(x, y, z)) for atm in atoms.values(): # z distortion x, y, z = atm.posn() z *= (zdist + length) / length atm.setposn(V(x, y, z)) length += zdist for atm in atoms.values(): # xy distortion x, y, z = atm.posn() radius = self.chirality.R x *= (radius + 0.5 * xydist) / radius y *= (radius - 0.5 * xydist) / radius atm.setposn(V(x, y, z)) # Judgement call: because we're discarding carbons with funky # valences, we will necessarily get slightly more ragged edges # on nanotubes. This is a parameter we can fiddle with to # adjust the length. My thought is that users would prefer a # little extra length, because it's fairly easy to trim the # ends, but much harder to add new atoms on the end. LENGTH_TWEAK = bond_length # trim all the carbons that fall outside our desired length # by doing this, we are introducing new singlets for atm in atoms.values(): x, y, z = atm.posn() if (z > .5 * (length + LENGTH_TWEAK) or z < -.5 * (length + LENGTH_TWEAK)): atm.kill() # Apply bend. Equations are anomalous for zero bend. if abs(bend) > pi / 360: R = length / bend for atm in atoms.values(): x, y, z = atm.posn() theta = z / R x, z = R - (R - x) * cos(theta), (R - x) * sin(theta) atm.setposn(V(x, y, z)) def trimCarbons(): # trim all the carbons that only have one carbon neighbor for i in range(2): for atm in atoms.values(): if not atm.is_singlet() and len(atm.realNeighbors()) == 1: atm.kill() trimCarbons() # if we're not picky about endings, we don't need to trim carbons if endings == "Capped": # buckyball endcaps addEndcap(mol, length, self.chirality.R) if endings == "Hydrogen": # hydrogen terminations for atm in atoms.values(): atm.Hydrogenate() elif endings == "Nitrogen": # nitrogen terminations dstElem = PeriodicTable.getElement('N') atomtype = dstElem.find_atomtype('sp2') for atm in atoms.values(): if len(atm.realNeighbors()) == 2: atm.Transmute(dstElem, force=True, atomtype=atomtype) # Translate structure to desired position for atm in atoms.values(): v = atm.posn() atm.setposn(v + position) if PROFILE: t = sw.now() env.history.message(greenmsg("%g seconds to build %d atoms" % (t, len(atoms.values())))) if numwalls > 1: n += int(spacing * 3 + 0.5) # empirical tinkering params = (length, n, m, bond_length, zdist, xydist, twist, bend, members, endings, numwalls-1, spacing) self.build_struct(name, params, position, mol=mol, createPrinted=True) return mol
def populate(self, mol, height, width, z, bond_length, endings, position): """ Create a graphene sheet chunk. """ def add(element, x, y, atomtype='sp2'): atm = Atom(element, V(x, y, z), mol) atm.set_atomtype_but_dont_revise_singlets(atomtype) return atm num_atoms = len(mol.atoms) bond_dict = {} i = j = 0 y = -0.5 * height - 2 * bond_length while y < 0.5 * height + 2 * bond_length: i = 0 x = -0.5 * width - 2 * bond_length while x < 0.5 * width + 2 * bond_length: lst = [] for x1, y1 in quartet: atm = add("C", x + x1 * bond_length, y + y1 * bond_length) lst.append(atm) bond_dict[(i, j)] = lst bond_atoms(lst[0], lst[1], bond_constants.V_GRAPHITE) bond_atoms(lst[1], lst[2], bond_constants.V_GRAPHITE) bond_atoms(lst[2], lst[3], bond_constants.V_GRAPHITE) i += 1 x += 3 * bond_length j += 1 y += sqrt3 * bond_length imax, jmax = i, j for i in range(imax): for j in range(jmax - 1): lst1 = bond_dict[(i, j)] lst2 = bond_dict[(i, j + 1)] bond_atoms(lst1[0], lst2[1], bond_constants.V_GRAPHITE) bond_atoms(lst1[3], lst2[2], bond_constants.V_GRAPHITE) for i in range(imax - 1): for j in range(jmax): lst1 = bond_dict[(i, j)] lst2 = bond_dict[(i + 1, j)] bond_atoms(lst1[3], lst2[0], bond_constants.V_GRAPHITE) # trim to dimensions atoms = mol.atoms for atm in atoms.values(): x, y, z = atm.posn() xdim, ydim = width + bond_length, height + bond_length # xdim, ydim = width + 0.5 * bond_length, height + 0.5 * bond_length if (x < -0.5 * xdim or x > 0.5 * xdim or y < -0.5 * ydim or y > 0.5 * ydim): atm.kill() def trimCarbons(): """Trim all the carbons that only have one carbon neighbor. """ for i in range(2): for atm in atoms.values(): if not atm.is_singlet() and len(atm.realNeighbors()) == 1: atm.kill() if TOROIDAL: # This is for making electrical inductors. What would be # really good here would be to break the bonds that are # stretched by this and put back the bondpoints. angstromsPerTurn = 6.0 for atm in atoms.values(): x, y, z = atm.posn() r = (x**2 + y**2)**.5 if 0.25 * width <= r <= 0.5 * width: angle = atan2(y, x) zdisp = (angstromsPerTurn * angle) / (2 * pi) atm.setposn(V(x, y, z + zdisp)) else: atm.kill() if endings == 1: # hydrogen terminations trimCarbons() for atm in atoms.values(): atm.Hydrogenate() elif endings == 2: # nitrogen terminations trimCarbons() dstElem = PeriodicTable.getElement('N') atomtype = dstElem.find_atomtype('sp2') for atm in atoms.values(): if len(atm.realNeighbors()) == 2: atm.Transmute(dstElem, force=True, atomtype=atomtype) for atm in atoms.values(): atm.setposn(atm.posn() + position) if num_atoms == len(mol.atoms): raise Exception("Graphene sheet too small - no atoms added")
def _readgms(assy, filename, isInsert=False): """ Read the atoms from a GAMESS DAT file into a single new chunk, which is returned, unless there are no atoms in the file, in which case a warning is printed and None is returned. (The new chunk (if returned) is in assy, but is not yet added into any Group or Part in assy -- caller must do that.) """ fi = open(filename, "rU") lines = fi.readlines() fi.close() dir, nodename = os.path.split(filename) ndix = {} mol = Chunk(assy, nodename) countdown = 0 equilibruim_found = False atoms_found = False for card in lines: if failpat.search( card): # GAMESS Aborted. No atom data will be found. print card break # If this card is found: # "1 ***** EQUILIBRIUM GEOMETRY LOCATED *****\n" # we know we have a successfully optimized structure/set of atoms. # If this card is not found, the optimization failed for some reason. # Atom positions begin soon after this card. if card == "1 ***** EQUILIBRIUM GEOMETRY LOCATED *****\n": equilibruim_found = True continue # The atom positions we want ALWAYS begin 2 lines after this card: # " COORDINATES OF ALL ATOMS ARE (ANGS)\n" # which follows the previous card. # This is one way to fix the problem mentioned above. # I've commented the code below out since it needs further work to do what # we need, and there is a chance we will not need this if GAMESS-US has # the same number of lines (6) after the "EQUILIBRIUM" card above. # # P.S. The reason we do not just look for this card by itself is that there # can be many of them. There is only one "EQUILIBRIUM" card, and the # good atoms follow that card. # 050624 Mark if equilibruim_found: if card == " COORDINATES OF ALL ATOMS ARE (ANGS)\n": atoms_found = True reading_atoms = True countdown = 2 continue if not equilibruim_found or not atoms_found: continue if countdown: countdown -= 1 # print countdown, card # for debugging only. continue # The current card contains atom type and position. n = 0 if reading_atoms: if len(card) < 10: reading_atoms = False # Finished reading atoms. break m = irecpat.match(card) sym = capitalize(m.group(1)) try: PeriodicTable.getElement(sym) except: env.history.message( redmsg( "Warning: GAMESS DAT file: unknown element %s in: %s" % (sym, card))) else: xyz = map(float, (m.group(2), m.group(3), m.group(4))) a = Atom(sym, A(xyz), mol) ndix[n] = a n += 1 # Don't return an empty chunk. if not mol.atoms: msg = "Warning: GAMESS file contains no equilibrium geometry. No atoms read into part." env.history.message(redmsg(msg)) return None # Need to compute and add bonds for this chunk. I'll ask Bruce how to best accomplish this. # In the meantime, let's warn the user that no bonds have been formed since it # is impossible to see this in vdW display mode. # Mark 050623. msg = "Warning: Equilibrium geometry found. Atoms read into part, but there are no bonds." env.history.message(orangemsg(msg)) return mol
def fix_deprecated_elements(changed_atoms): """ scan for deprecated elements, and fix them """ fix_PAM3 = pref_fix_deprecated_PAM3_atoms() fix_PAM5 = pref_fix_deprecated_PAM5_atoms() deprecated_atoms = [] for atom in changed_atoms.itervalues(): deprecated_to = atom.element.deprecated_to # an element symbol, or None, or 'remove' if deprecated_to: pam = atom.element.pam assert pam in (MODEL_PAM3, MODEL_PAM5) if pam == MODEL_PAM3: fix = fix_PAM3 elif pam == MODEL_PAM5: fix = fix_PAM5 else: fix = False if fix: deprecated_atoms.append(atom) elif debug_flags.DEBUG_DNA_UPDATER: print "dna updater: debug_pref says don't alter deprecated atom %r" % ( atom, ) continue for atom in deprecated_atoms: deprecated_to = atom.element.deprecated_to # an element symbol, or 'remove' if atom.display != diDEFAULT: # Atoms of deprecated elements sometimes have funny display modes # set by the DNA Duplex Generator. Remove these here. # (This may be needed even after we fix the generator, # due to old mmp files. REVIEW: can it ever cause harm?) atom.setDisplay(diDEFAULT) if deprecated_to == 'remove' or deprecated_to == 'X': # (Atom.kill might be unideal behavior for 'remove', # but that's only on Pl3 which never occurs AFAIK, so nevermind) # Kill the atom (and make sure that new bondpoints get into a good enough position). # REVIEW: does atom.kill make new bps immediately # (review whether ok if more than one needs making on one base atom) # or later # (review whether it's still going to happen in the current master_updater call)? #### if debug_flags.DEBUG_DNA_UPDATER_VERBOSE: print "dna updater: kill deprecated atom %r" % (atom, ) summary_format = \ "Warning: dna updater killed [N] deprecated %s pseudoatom(s)" % \ (atom.element.symbol,) env.history.deferred_summary_message(orangemsg(summary_format)) atom.kill() # TODO: worry about atom being a hotspot, or having a bondpoint which is a hotspot? else: # Transmute atom to a new element symbol -- assume its position, # bonds, and bondpoints are all ok and need no changes. # (Should be true of as 071119, since this is used only # to make Ax and Ss PAM atoms from variant atomtypes # used to mark them as being in special situations.) # # Use mvElement to avoid remaking existing bondpoints. elt = PeriodicTable.getElement(deprecated_to) if debug_flags.DEBUG_DNA_UPDATER_VERBOSE: print "dna updater: transmute deprecated atom %r to element %s" % \ (atom, elt.symbol) if dna_updater_warn_when_transmuting_deprecated_elements(): summary_format = \ "Warning: dna updater transmuted [N] %s to %s pseudoatom(s)" % \ (atom.element.symbol, elt.symbol ) env.history.deferred_summary_message(orangemsg(summary_format)) # todo: refactor so orangemsg is replaced with a warning option atom.mvElement(elt) atom.make_enough_bondpoints() # REVIEW: do this later, if atom classes should be corrected first # to help this properly position bondpoints # (or perhaps, first set correct atom classes, then do this, # making sure it sets correct bondpoint classes, # or that we correct them separately afterwards) continue return # from fix_deprecated_elements
def elementId(symbol): return PeriodicTable.getElement(symbol).eltnum
def __init__(self, win): qt4todo('what to do with all those options?') ## ElementColorsDialog.__init__(self, win, None, 0, ## Qt.WStyle_Customize | Qt.WStyle_NormalBorder | ## Qt.WStyle_Title | Qt.WStyle_SysMenu) QDialog.__init__(self, win) self.setupUi(self) self.connect(self.okButton, SIGNAL("clicked()"), self.ok) self.connect(self.loadColorsPB, SIGNAL("clicked()"), self.read_element_rgb_table) self.connect(self.saveColorsPB, SIGNAL("clicked()"), self.write_element_rgb_table) self.connect(self.cancelButton, SIGNAL("clicked()"), self.reject) self.connect(self.defaultButton, SIGNAL("clicked()"), self.loadDefaultProp) self.connect(self.alterButton, SIGNAL("clicked()"), self.loadAlterProp) self.connect(self.elementButtonGroup, SIGNAL("clicked(int)"), self.setElementInfo) self.connect(self.previewPB, SIGNAL("clicked()"), self.preview_color_change) self.connect(self.restorePB, SIGNAL("clicked()"), self.restore_current_color) self.w = win self.fileName = None self.isElementModified = False self.isFileSaved = False self.oldTable = PeriodicTable.deepCopy() self.elemTable = PeriodicTable self.displayMode = self._displayList[0] # The next line fixes a bug. Thumbview expects self.gridLayout on # line 117 of Thumbview.py. Mark 2007-10-19. self.gridLayout = self.gridlayout self.elemGLPane = ElementView(self, "element glPane", self.w.glpane) # Put the GL widget inside the frame flayout = QVBoxLayout(self.elementFrame) flayout.setMargin(1) flayout.setSpacing(1) flayout.addWidget(self.elemGLPane, 1) def elementId(symbol): return PeriodicTable.getElement(symbol).eltnum self.toolButton6.setChecked(True) self.elementButtonGroup.setId(self.toolButton6, elementId("C")) self.elementButtonGroup.setId(self.toolButton8, elementId("O")) self.elementButtonGroup.setId(self.toolButton10, elementId("Ne")) self.elementButtonGroup.setId(self.toolButton9, elementId("F")) self.elementButtonGroup.setId(self.toolButton13, elementId("Al")) self.elementButtonGroup.setId(self.toolButton17, elementId("Cl")) self.elementButtonGroup.setId(self.toolButton5, elementId("B")) self.elementButtonGroup.setId(self.toolButton10_2, elementId("Ar")) self.elementButtonGroup.setId(self.toolButton15, elementId("P")) self.elementButtonGroup.setId(self.toolButton16, elementId("S")) self.elementButtonGroup.setId(self.toolButton14, elementId("Si")) self.elementButtonGroup.setId(self.toolButton33, elementId("As")) self.elementButtonGroup.setId(self.toolButton34, elementId("Se")) self.elementButtonGroup.setId(self.toolButton35, elementId("Br")) self.elementButtonGroup.setId(self.toolButton36, elementId("Kr")) self.elementButtonGroup.setId(self.toolButton32, elementId("Ge")) self.elementButtonGroup.setId(self.toolButton7, elementId("N")) self.elementButtonGroup.setId(self.toolButton2, elementId("He")) self.elementButtonGroup.setId(self.toolButton1, elementId("H")) self.elementButtonGroup.setId(self.toolButton0, elementId("X")) self.connect(self.toolButton6, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton8, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton10, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton9, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton13, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton17, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton5, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton10_2, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton15, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton16, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton14, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton33, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton34, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton35, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton36, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton32, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton7, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton2, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton1, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connect(self.toolButton0, SIGNAL("clicked()"), self.updateElemColorDisplay) self.connectChangingControls() self.saveColorsPB.setWhatsThis( """Save the current color settings for elements in a text file.""") self.defaultButton.setWhatsThis( """Restore current element colors to the default colors.""") self.loadColorsPB.setWhatsThis( """Load element colors from an external text file.""") self.alterButton.setWhatsThis( """Set element colors to the alternate color set.""")
def elementId(symbol): return PeriodicTable.getElement(symbol).eltnum
def fix_deprecated_elements( changed_atoms): """ scan for deprecated elements, and fix them """ fix_PAM3 = pref_fix_deprecated_PAM3_atoms() fix_PAM5 = pref_fix_deprecated_PAM5_atoms() deprecated_atoms = [] for atom in changed_atoms.itervalues(): deprecated_to = atom.element.deprecated_to # an element symbol, or None, or 'remove' if deprecated_to: pam = atom.element.pam assert pam in (MODEL_PAM3, MODEL_PAM5) if pam == MODEL_PAM3: fix = fix_PAM3 elif pam == MODEL_PAM5: fix = fix_PAM5 else: fix = False if fix: deprecated_atoms.append(atom) elif debug_flags.DEBUG_DNA_UPDATER: print "dna updater: debug_pref says don't alter deprecated atom %r" % (atom,) continue for atom in deprecated_atoms: deprecated_to = atom.element.deprecated_to # an element symbol, or 'remove' if atom.display != diDEFAULT: # Atoms of deprecated elements sometimes have funny display modes # set by the DNA Duplex Generator. Remove these here. # (This may be needed even after we fix the generator, # due to old mmp files. REVIEW: can it ever cause harm?) atom.setDisplay(diDEFAULT) if deprecated_to == 'remove' or deprecated_to == 'X': # (Atom.kill might be unideal behavior for 'remove', # but that's only on Pl3 which never occurs AFAIK, so nevermind) # Kill the atom (and make sure that new bondpoints get into a good enough position). # REVIEW: does atom.kill make new bps immediately # (review whether ok if more than one needs making on one base atom) # or later # (review whether it's still going to happen in the current master_updater call)? #### if debug_flags.DEBUG_DNA_UPDATER_VERBOSE: print "dna updater: kill deprecated atom %r" % (atom,) summary_format = \ "Warning: dna updater killed [N] deprecated %s pseudoatom(s)" % \ (atom.element.symbol,) env.history.deferred_summary_message( orangemsg(summary_format) ) atom.kill() # TODO: worry about atom being a hotspot, or having a bondpoint which is a hotspot? else: # Transmute atom to a new element symbol -- assume its position, # bonds, and bondpoints are all ok and need no changes. # (Should be true of as 071119, since this is used only # to make Ax and Ss PAM atoms from variant atomtypes # used to mark them as being in special situations.) # # Use mvElement to avoid remaking existing bondpoints. elt = PeriodicTable.getElement(deprecated_to) if debug_flags.DEBUG_DNA_UPDATER_VERBOSE: print "dna updater: transmute deprecated atom %r to element %s" % \ (atom, elt.symbol) if dna_updater_warn_when_transmuting_deprecated_elements(): summary_format = \ "Warning: dna updater transmuted [N] %s to %s pseudoatom(s)" % \ (atom.element.symbol, elt.symbol ) env.history.deferred_summary_message( orangemsg(summary_format) ) # todo: refactor so orangemsg is replaced with a warning option atom.mvElement(elt) atom.make_enough_bondpoints() # REVIEW: do this later, if atom classes should be corrected first # to help this properly position bondpoints # (or perhaps, first set correct atom classes, then do this, # making sure it sets correct bondpoint classes, # or that we correct them separately afterwards) continue return # from fix_deprecated_elements
# - Remove Crossover needs to be offered when correct to do so, not otherwise # - Pl position is wrong after either op, esp. Remove # - Undo and Feature Help cmdnames are wrong (not working) from utilities.constants import noop, average_value from model.bond_constants import V_SINGLE from model.bond_constants import atoms_are_bonded, find_bond from model.bonds import bond_atoms_faster, bond_direction ##, bond_atoms from utilities.Log import redmsg, greenmsg, quote_html ##, orangemsg ##from debug_prefs import debug_pref, Choice import foundation.env as env from utilities.GlobalPreferences import dna_updater_is_enabled from model.elements import PeriodicTable Element_Sj5 = PeriodicTable.getElement('Sj5') Element_Ss5 = PeriodicTable.getElement('Ss5') def crossover_menu_spec(atom, selatoms): """ Make a crossover-related menu_spec list for the two atoms in the selatoms dict (atom.key -> atom), both Pl, for use in atom's context menu (which must be one of the atoms in selatoms). If no menu commands are needed, return [] (a valid empty menu_spec) or None. Should be reasonably fast, but needn't be super-fast -- called once whenever we construct a context menu for exactly two selected Pl atoms. """ assert len(selatoms) == 2 atoms = selatoms.values()
def _init(): global _is_initialized if (_is_initialized): return AMBER_AtomTypes["C"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CA"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CB"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CC"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CD"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CK"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CM"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CN"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CQ"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CR"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CT"] = PeriodicTable.getElement("C").find_atomtype("sp3") AMBER_AtomTypes["CV"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CW"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["C*"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CY"] = PeriodicTable.getElement("C").find_atomtype("sp") AMBER_AtomTypes["CZ"] = PeriodicTable.getElement("C").find_atomtype("sp") AMBER_AtomTypes["C0"] = PeriodicTable.getElement("Ca").find_atomtype("?") AMBER_AtomTypes["H"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H0"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HC"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H1"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H2"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H3"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HA"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H4"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H5"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HO"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HS"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HW"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HP"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HZ"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["F"] = PeriodicTable.getElement("F").find_atomtype("?") AMBER_AtomTypes["Cl"] = PeriodicTable.getElement("Cl").find_atomtype("?") AMBER_AtomTypes["Br"] = PeriodicTable.getElement("Br").find_atomtype("?") AMBER_AtomTypes["I"] = PeriodicTable.getElement("I").find_atomtype("?") AMBER_AtomTypes["IM"] = PeriodicTable.getElement("Cl").find_atomtype("?") AMBER_AtomTypes["IB"] = PeriodicTable.getElement("Na").find_atomtype("?") AMBER_AtomTypes["MG"] = PeriodicTable.getElement("Mg").find_atomtype("?") AMBER_AtomTypes["N"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["NA"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["NB"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["NC"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["N2"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["N3"] = PeriodicTable.getElement("N").find_atomtype("sp3") AMBER_AtomTypes["NT"] = PeriodicTable.getElement("N").find_atomtype("sp3") AMBER_AtomTypes["N*"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["NY"] = PeriodicTable.getElement("N").find_atomtype("sp") AMBER_AtomTypes["O"] = PeriodicTable.getElement("O").find_atomtype("sp2") AMBER_AtomTypes["O2"] = PeriodicTable.getElement("O").find_atomtype("sp2") AMBER_AtomTypes["OW"] = PeriodicTable.getElement("O").find_atomtype("sp3") AMBER_AtomTypes["OH"] = PeriodicTable.getElement("O").find_atomtype("sp3") AMBER_AtomTypes["OS"] = PeriodicTable.getElement("O").find_atomtype("sp3") AMBER_AtomTypes["P"] = PeriodicTable.getElement("P").find_atomtype("sp3(p)") #sp3(p) is 'sp3(phosphate) AMBER_AtomTypes["S"] = PeriodicTable.getElement("S").find_atomtype("sp3") # ? AMBER_AtomTypes["SH"] = PeriodicTable.getElement("S").find_atomtype("sp3") # ? AMBER_AtomTypes["CU"] = PeriodicTable.getElement("Cu").find_atomtype("?") AMBER_AtomTypes["FE"] = PeriodicTable.getElement("Fe").find_atomtype("?") AMBER_AtomTypes["Li"] = PeriodicTable.getElement("Li").find_atomtype("?") AMBER_AtomTypes["IP"] = PeriodicTable.getElement("Na").find_atomtype("?") AMBER_AtomTypes["Na"] = PeriodicTable.getElement("Na").find_atomtype("?") AMBER_AtomTypes["K"] = PeriodicTable.getElement("K").find_atomtype("?") #AMBER_AtomTypes["Rb"] = PeriodicTable.getElement("Rb").find_atomtype("?") #AMBER_AtomTypes["Cs"] = PeriodicTable.getElement("Cs").find_atomtype("?") AMBER_AtomTypes["Zn"] = PeriodicTable.getElement("Zn").find_atomtype("?") _is_initialized = True
from utilities.exception_classes import PluginBug from utilities.constants import gensym from utilities.prefs_constants import dnaDefaultStrand1Color_prefs_key from utilities.prefs_constants import dnaDefaultStrand2Color_prefs_key from utilities.prefs_constants import dnaDefaultSegmentColor_prefs_key from dna.model.Dna_Constants import getDuplexBasesPerTurn ##from dna.updater.dna_updater_prefs import pref_dna_updater_convert_to_PAM3plus5 from simulation.sim_commandruns import adjustSinglet from model.elements import PeriodicTable from model.Line import Line from model.chem import Atom_prekill_prep Element_Ae3 = PeriodicTable.getElement('Ae3') from dna.model.Dna_Constants import basesDict, dnaDict from dna.model.dna_model_constants import LADDER_END0 basepath_ok, basepath = find_plugin_dir("DNA") if not basepath_ok: env.history.message(orangemsg("The cad/plugins/DNA directory is missing.")) RIGHT_HANDED = -1 LEFT_HANDED = 1 from geometry.VQT import V, Q, norm, cross from geometry.VQT import vlen from Numeric import dot
def _get_atomlist_from_gms_outfile(assy, filename): """ Read the atoms from a GAMESS OUT file into an atom list, which is returned, unless there are no atoms in the file, in which case a warning is printed and None is returned. """ fi = open(filename,"rU") lines = fi.readlines() fi.close() dir, nodename = os.path.split(filename) mol = Chunk(assy, nodename) newAtomList = [] countdown = 0 equilibruim_found = False atoms_found = False for card in lines: if failpat.search(card): # GAMESS Aborted. No atom data will be found. print card env.history.message( redmsg( card )) break if noconvpat.search(card): # Geometry search is not converged. print card env.history.message( redmsg( card )) break # If this card is found: # "1 ***** EQUILIBRIUM GEOMETRY LOCATED *****\n" # we know we have a successfully optimized structure/set of atoms. # If this card is not found, the optimization failed for some reason. # Atom positions begin soon after this card. if card == "1 ***** EQUILIBRIUM GEOMETRY LOCATED *****\n": equilibruim_found = True continue # The atom positions we want ALWAYS begin 2 lines after this card: # " COORDINATES OF ALL ATOMS ARE (ANGS)\n" # which follows the previous card. # This is one way to fix the problem mentioned above. # I've commented the code below out since it needs further work to do what # we need, and there is a chance we will not need this if GAMESS-US has # the same number of lines (6) after the "EQUILIBRIUM" card above. # # P.S. The reason we do not just look for this card by itself is that there # can be many of them. There is only one "EQUILIBRIUM" card, and the # good atoms follow that card. # 050624 Mark if equilibruim_found: if card == " COORDINATES OF ALL ATOMS ARE (ANGS)\n": atoms_found = True reading_atoms = True countdown = 2 continue if not equilibruim_found or not atoms_found: continue if countdown: countdown -= 1 # print countdown, card # for debugging only. continue # The current card contains atom type and position. n = 0 if reading_atoms: # print "_get_atomlist_from_gms_outfile:", card if len(card)<10: reading_atoms = False # Finished reading atoms. break m=irecpat.match(card) sym = capitalize(m.group(1)) try: PeriodicTable.getElement(sym) except: env.history.message( redmsg( "Warning: GAMESS OUT file: unknown element %s in: %s" % (sym,card) )) else: xyz = map(float, (m.group(2),m.group(3), m.group(4))) a = Atom(sym, A(xyz), mol) newAtomList += [a] # Let caller handle history msgs. Mark 050712 # if not newAtomList: # msg = "Warning: GAMESS file contains no equilibrium geometry. No atoms read into part." # env.history.message( redmsg(msg)) # return None return newAtomList
def populate(self, mol, height, width, z, bond_length, endings, position): """ Create a graphene sheet chunk. """ def add(element, x, y, atomtype='sp2'): atm = Atom(element, V(x, y, z), mol) atm.set_atomtype_but_dont_revise_singlets(atomtype) return atm num_atoms = len(mol.atoms) bond_dict = { } i = j = 0 y = -0.5 * height - 2 * bond_length while y < 0.5 * height + 2 * bond_length: i = 0 x = -0.5 * width - 2 * bond_length while x < 0.5 * width + 2 * bond_length: lst = [ ] for x1, y1 in quartet: atm = add("C", x + x1 * bond_length, y + y1 * bond_length) lst.append(atm) bond_dict[(i, j)] = lst bonds.bond_atoms(lst[0], lst[1], bond_constants.V_GRAPHITE) bonds.bond_atoms(lst[1], lst[2], bond_constants.V_GRAPHITE) bonds.bond_atoms(lst[2], lst[3], bond_constants.V_GRAPHITE) i += 1 x += 3 * bond_length j += 1 y += sqrt3 * bond_length imax, jmax = i, j for i in range(imax): for j in range(jmax - 1): lst1 = bond_dict[(i, j)] lst2 = bond_dict[(i, j+1)] bonds.bond_atoms(lst1[0], lst2[1], bond_constants.V_GRAPHITE) bonds.bond_atoms(lst1[3], lst2[2], bond_constants.V_GRAPHITE) for i in range(imax - 1): for j in range(jmax): lst1 = bond_dict[(i, j)] lst2 = bond_dict[(i+1, j)] bonds.bond_atoms(lst1[3], lst2[0], bond_constants.V_GRAPHITE) # trim to dimensions atoms = mol.atoms for atm in atoms.values(): x, y, z = atm.posn() xdim, ydim = width + bond_length, height + bond_length # xdim, ydim = width + 0.5 * bond_length, height + 0.5 * bond_length if (x < -0.5 * xdim or x > 0.5 * xdim or y < -0.5 * ydim or y > 0.5 * ydim): atm.kill() def trimCarbons(): """Trim all the carbons that only have one carbon neighbor. """ for i in range(2): for atm in atoms.values(): if not atm.is_singlet() and len(atm.realNeighbors()) == 1: atm.kill() if TOROIDAL: # This is for making electrical inductors. What would be # really good here would be to break the bonds that are # stretched by this and put back the bondpoints. angstromsPerTurn = 6.0 for atm in atoms.values(): x, y, z = atm.posn() r = (x**2 + y**2) ** .5 if 0.25 * width <= r <= 0.5 * width: angle = atan2(y, x) zdisp = (angstromsPerTurn * angle) / (2 * pi) atm.setposn(V(x, y, z + zdisp)) else: atm.kill() if endings == 1: # hydrogen terminations trimCarbons() for atm in atoms.values(): atm.Hydrogenate() elif endings == 2: # nitrogen terminations trimCarbons() dstElem = PeriodicTable.getElement('N') atomtype = dstElem.find_atomtype('sp2') for atm in atoms.values(): if len(atm.realNeighbors()) == 2: atm.Transmute(dstElem, force=True, atomtype=atomtype) for atm in atoms.values(): atm.setposn(atm.posn() + position) if num_atoms == len(mol.atoms): raise Exception("Graphene sheet too small - no atoms added")
def build_struct(self, name, params, position, mol=None, createPrinted=False): """ Build a nanotube from the parameters in the Property Manger dialog. """ length, n, m, bond_length, zdist, xydist, \ twist, bend, members, endings, numwalls, spacing = params # This can take a few seconds. Inform the user. # 100 is a guess on my part. Mark 051103. if not createPrinted: # If it's a multi-wall tube, only print the "Creating" message once. if length > 100.0: env.history.message(self.cmd + "This may take a moment...") self.chirality = Chirality(n, m, bond_length) PROFILE = False if PROFILE: sw = Stopwatch() sw.start() xyz = self.chirality.xyz if mol == None: mol = Chunk(self.win.assy, name) atoms = mol.atoms mlimits = self.chirality.mlimits # populate the tube with some extra carbons on the ends # so that we can trim them later self.chirality.populate(mol, length + 4 * self.chirality.maxlen, members != 0) # Apply twist and distortions. Bends probably would come # after this point because they change the direction for the # length. I'm worried about Z distortion because it will work # OK for stretching, but with compression it can fail. BTW, # "Z distortion" is a misnomer, we're stretching in the Y # direction. for atm in atoms.values(): # twist x, y, z = atm.posn() twistRadians = twist * z c, s = cos(twistRadians), sin(twistRadians) x, y = x * c + y * s, -x * s + y * c atm.setposn(V(x, y, z)) for atm in atoms.values(): # z distortion x, y, z = atm.posn() z *= (zdist + length) / length atm.setposn(V(x, y, z)) length += zdist for atm in atoms.values(): # xy distortion x, y, z = atm.posn() radius = self.chirality.R x *= (radius + 0.5 * xydist) / radius y *= (radius - 0.5 * xydist) / radius atm.setposn(V(x, y, z)) # Judgement call: because we're discarding carbons with funky # valences, we will necessarily get slightly more ragged edges # on nanotubes. This is a parameter we can fiddle with to # adjust the length. My thought is that users would prefer a # little extra length, because it's fairly easy to trim the # ends, but much harder to add new atoms on the end. LENGTH_TWEAK = bond_length # trim all the carbons that fall outside our desired length # by doing this, we are introducing new singlets for atm in atoms.values(): x, y, z = atm.posn() if (z > .5 * (length + LENGTH_TWEAK) or z < -.5 * (length + LENGTH_TWEAK)): atm.kill() # Apply bend. Equations are anomalous for zero bend. if abs(bend) > pi / 360: R = length / bend for atm in atoms.values(): x, y, z = atm.posn() theta = z / R x, z = R - (R - x) * cos(theta), (R - x) * sin(theta) atm.setposn(V(x, y, z)) def trimCarbons(): # trim all the carbons that only have one carbon neighbor for i in range(2): for atm in atoms.values(): if not atm.is_singlet() and len(atm.realNeighbors()) == 1: atm.kill() trimCarbons() # if we're not picky about endings, we don't need to trim carbons if endings == "Capped": # buckyball endcaps addEndcap(mol, length, self.chirality.R) if endings == "Hydrogen": # hydrogen terminations for atm in atoms.values(): atm.Hydrogenate() elif endings == "Nitrogen": # nitrogen terminations dstElem = PeriodicTable.getElement('N') atomtype = dstElem.find_atomtype('sp2') for atm in atoms.values(): if len(atm.realNeighbors()) == 2: atm.Transmute(dstElem, force=True, atomtype=atomtype) # Translate structure to desired position for atm in atoms.values(): v = atm.posn() atm.setposn(v + position) if PROFILE: t = sw.now() env.history.message( greenmsg("%g seconds to build %d atoms" % (t, len(atoms.values())))) if numwalls > 1: n += int(spacing * 3 + 0.5) # empirical tinkering params = (length, n, m, bond_length, zdist, xydist, twist, bend, members, endings, numwalls - 1, spacing) self.build_struct(name, params, position, mol=mol, createPrinted=True) return mol
def _init(): global _is_initialized if (_is_initialized): return AMBER_AtomTypes["C"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CA"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CB"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CC"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CD"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CK"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CM"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CN"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CQ"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CR"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CT"] = PeriodicTable.getElement("C").find_atomtype("sp3") AMBER_AtomTypes["CV"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CW"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["C*"] = PeriodicTable.getElement("C").find_atomtype("sp2") AMBER_AtomTypes["CY"] = PeriodicTable.getElement("C").find_atomtype("sp") AMBER_AtomTypes["CZ"] = PeriodicTable.getElement("C").find_atomtype("sp") AMBER_AtomTypes["C0"] = PeriodicTable.getElement("Ca").find_atomtype("?") AMBER_AtomTypes["H"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H0"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HC"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H1"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H2"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H3"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HA"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H4"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["H5"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HO"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HS"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HW"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HP"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["HZ"] = PeriodicTable.getElement("H").find_atomtype("?") AMBER_AtomTypes["F"] = PeriodicTable.getElement("F").find_atomtype("?") AMBER_AtomTypes["Cl"] = PeriodicTable.getElement("Cl").find_atomtype("?") AMBER_AtomTypes["Br"] = PeriodicTable.getElement("Br").find_atomtype("?") AMBER_AtomTypes["I"] = PeriodicTable.getElement("I").find_atomtype("?") AMBER_AtomTypes["IM"] = PeriodicTable.getElement("Cl").find_atomtype("?") AMBER_AtomTypes["IB"] = PeriodicTable.getElement("Na").find_atomtype("?") AMBER_AtomTypes["MG"] = PeriodicTable.getElement("Mg").find_atomtype("?") AMBER_AtomTypes["N"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["NA"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["NB"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["NC"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["N2"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["N3"] = PeriodicTable.getElement("N").find_atomtype("sp3") AMBER_AtomTypes["NT"] = PeriodicTable.getElement("N").find_atomtype("sp3") AMBER_AtomTypes["N*"] = PeriodicTable.getElement("N").find_atomtype("sp2") AMBER_AtomTypes["NY"] = PeriodicTable.getElement("N").find_atomtype("sp") AMBER_AtomTypes["O"] = PeriodicTable.getElement("O").find_atomtype("sp2") AMBER_AtomTypes["O2"] = PeriodicTable.getElement("O").find_atomtype("sp2") AMBER_AtomTypes["OW"] = PeriodicTable.getElement("O").find_atomtype("sp3") AMBER_AtomTypes["OH"] = PeriodicTable.getElement("O").find_atomtype("sp3") AMBER_AtomTypes["OS"] = PeriodicTable.getElement("O").find_atomtype("sp3") AMBER_AtomTypes["P"] = PeriodicTable.getElement("P").find_atomtype("sp3(p)") #sp3(p) is 'sp3(phosphate) AMBER_AtomTypes["S"] = PeriodicTable.getElement("S").find_atomtype("sp3") # ? AMBER_AtomTypes["SH"] = PeriodicTable.getElement("S").find_atomtype("sp3") # ? AMBER_AtomTypes["CU"] = PeriodicTable.getElement("Cu").find_atomtype("?") AMBER_AtomTypes["FE"] = PeriodicTable.getElement("Fe").find_atomtype("?") AMBER_AtomTypes["Li"] = PeriodicTable.getElement("Li").find_atomtype("?") AMBER_AtomTypes["IP"] = PeriodicTable.getElement("Na").find_atomtype("?") AMBER_AtomTypes["Na"] = PeriodicTable.getElement("Na").find_atomtype("?") AMBER_AtomTypes["K"] = PeriodicTable.getElement("K").find_atomtype("?") #AMBER_AtomTypes["Rb"] = PeriodicTable.getElement("Rb").find_atomtype("?") #AMBER_AtomTypes["Cs"] = PeriodicTable.getElement("Cs").find_atomtype("?") AMBER_AtomTypes["Zn"] = PeriodicTable.getElement("Zn").find_atomtype("?") _is_initialized = True
# - Remove Crossover needs to be offered when correct to do so, not otherwise # - Pl position is wrong after either op, esp. Remove # - Undo and Feature Help cmdnames are wrong (not working) from utilities.constants import noop, average_value from model.bond_constants import V_SINGLE from model.bond_constants import atoms_are_bonded, find_bond from model.bonds import bond_atoms_faster, bond_direction ##, bond_atoms from utilities.Log import redmsg, greenmsg, quote_html ##, orangemsg ##from debug_prefs import debug_pref, Choice import foundation.env as env from utilities.GlobalPreferences import dna_updater_is_enabled from model.elements import PeriodicTable Element_Sj5 = PeriodicTable.getElement('Sj5') Element_Ss5 = PeriodicTable.getElement('Ss5') def crossover_menu_spec(atom, selatoms): """ Make a crossover-related menu_spec list for the two atoms in the selatoms dict (atom.key -> atom), both Pl, for use in atom's context menu (which must be one of the atoms in selatoms). If no menu commands are needed, return [] (a valid empty menu_spec) or None. Should be reasonably fast, but needn't be super-fast -- called once whenever we construct a context menu for exactly two selected Pl atoms. """ assert len(selatoms) == 2 atoms = selatoms.values()
def build(self, name, assy, position, mol=None, createPrinted=False): """ Build a Peptide from the parameters in the Property Manger dialog. """ endPoint1, endPoint2 = self.getEndPoints() cntAxis = endPoint2 - endPoint1 length = vlen(cntAxis) # This can take a few seconds. Inform the user. # 100 is a guess. --Mark 051103. if not createPrinted: # If it's a multi-wall tube, only print the "Creating" message once. if length > 100.0: env.history.message("This may take a moment...") PROFILE = False if PROFILE: sw = Stopwatch() sw.start() xyz = self.xyz if mol == None: mol = Chunk(assy, name) atoms = mol.atoms mlimits = self.mlimits # populate the tube with some extra carbons on the ends # so that we can trim them later self.populate(mol, length + 4 * self.maxlen) # Apply twist and distortions. Bends probably would come # after this point because they change the direction for the # length. I'm worried about Z distortion because it will work # OK for stretching, but with compression it can fail. BTW, # "Z distortion" is a misnomer, we're stretching in the Y # direction. for atm in atoms.values(): # twist x, y, z = atm.posn() twistRadians = self.twist * z c, s = cos(twistRadians), sin(twistRadians) x, y = x * c + y * s, -x * s + y * c atm.setposn(V(x, y, z)) for atm in atoms.values(): # z distortion x, y, z = atm.posn() z *= (self.zdist + length) / length atm.setposn(V(x, y, z)) length += self.zdist for atm in atoms.values(): # xy distortion x, y, z = atm.posn() radius = self.getRadius() x *= (radius + 0.5 * self.xydist) / radius y *= (radius - 0.5 * self.xydist) / radius atm.setposn(V(x, y, z)) # Judgement call: because we're discarding carbons with funky # valences, we will necessarily get slightly more ragged edges # on Peptides. This is a parameter we can fiddle with to # adjust the length. My thought is that users would prefer a # little extra length, because it's fairly easy to trim the # ends, but much harder to add new atoms on the end. LENGTH_TWEAK = self.getBondLength() # trim all the carbons that fall outside our desired length # by doing this, we are introducing new singlets for atm in atoms.values(): x, y, z = atm.posn() if z > 0.5 * (length + LENGTH_TWEAK) or z < -0.5 * (length + LENGTH_TWEAK): atm.kill() # Apply bend. Equations are anomalous for zero bend. if abs(self.bend) > pi / 360: R = length / self.bend for atm in atoms.values(): x, y, z = atm.posn() theta = z / R x, z = R - (R - x) * cos(theta), (R - x) * sin(theta) atm.setposn(V(x, y, z)) def trimCarbons(): """ Trim all the carbons that only have one carbon neighbor. """ for i in range(2): for atm in atoms.values(): if not atm.is_singlet() and len(atm.realNeighbors()) == 1: atm.kill() trimCarbons() # If we're not picky about endings, we don't need to trim carbons if self.endings == "Capped": # buckyball endcaps addEndcap(mol, length, self.getRadius()) if self.endings == "Hydrogen": # hydrogen terminations for atm in atoms.values(): atm.Hydrogenate() elif self.endings == "Nitrogen": # nitrogen terminations. # This option has been removed from the "Endings" combo box # in the PM. 2008-05-02 --mark dstElem = PeriodicTable.getElement("N") atomtype = dstElem.find_atomtype("sp2") for atm in atoms.values(): if len(atm.realNeighbors()) == 2: atm.Transmute(dstElem, force=True, atomtype=atomtype) # Translate structure to desired position for atm in atoms.values(): v = atm.posn() atm.setposn(v + position) if PROFILE: t = sw.now() env.history.message(greenmsg("%g seconds to build %d atoms" % (t, len(atoms.values())))) if self.numwalls > 1: n += int(self.spacing * 3 + 0.5) # empirical tinkering self.build(name, assy, endPoint1, endPoint2, position, mol=mol, createPrinted=True) # Orient the Peptide. if self.numwalls == 1: # This condition ensures that MWCTs get oriented only once. self._orient(mol, endPoint1, endPoint2) return mol