Skip to content

GitLab

Commit 25d8559f authored by Robert McGibbon's avatar Robert McGibbon
Browse files

Fix to PR #152

parent e23c0022
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
wrappers/python/simtk/openmm/app/pdbfile.py
View file @ 25d8559f
...@@ -287,10 +287,14 @@ class PDBFile(object): ...@@ -287,10 +287,14 @@ class PDBFile(object):
else: else:
atomName = atom.name atomName = atom.name
coords = positions[posIndex] coords = positions[posIndex]
if atom.element is not None:
symbol = atom.element.symbol
else:
symbol = ' '
line = "ATOM %5d %-4s %3s %s%4d %s%s%s 1.00 0.00 %2s " % ( line = "ATOM %5d %-4s %3s %s%4d %s%s%s 1.00 0.00 %2s " % (
atomIndex%100000, atomName, resName, chainName, atomIndex%100000, atomName, resName, chainName,
(resIndex+1)%10000, _format_83(coords[0]), (resIndex+1)%10000, _format_83(coords[0]),
_format_83(coords[1]), _format_83(coords[2]), atom.element.symbol) _format_83(coords[1]), _format_83(coords[2]), symbol)
assert len(line) == 80, 'Fixed width overflow detected' assert len(line) == 80, 'Fixed width overflow detected'
print >>file, line print >>file, line
posIndex += 1 posIndex += 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment