Commit b9936fb9 authored by João Rodrigues's avatar João Rodrigues
Browse files

Ensure coords are always in the proper units and not converted to floats

parent b0b7548a
...@@ -1544,7 +1544,7 @@ class Modeller(object): ...@@ -1544,7 +1544,7 @@ class Modeller(object):
if residue.name == 'HOH': if residue.name == 'HOH':
for atom in residue.atoms(): for atom in residue.atoms():
if atom.element == elem.oxygen: if atom.element == elem.oxygen:
waterPos[residue] = modeller.positions[atom.index].value_in_unit(nanometer) waterPos[residue] = modeller.positions[atom.index]
# Calculate lipid Z boundaries # Calculate lipid Z boundaries
lipidNames = {res.name for res in patch.topology.residues() if res.name != 'HOH'} lipidNames = {res.name for res in patch.topology.residues() if res.name != 'HOH'}
...@@ -1564,7 +1564,7 @@ class Modeller(object): ...@@ -1564,7 +1564,7 @@ class Modeller(object):
waterResidues = list(waterPos) waterResidues = list(waterPos)
for wRes in waterResidues: for wRes in waterResidues:
waterZ = waterPos[wRes][2] waterZ = waterPos[wRes][2]
if lowerZBoundary < waterZ < upperZBoundary: if lowerZBoundary < waterZ.value_in_unit(nanometer) < upperZBoundary:
del waterPos[wRes] del waterPos[wRes]
self._addIons(forcefield, waterPos, positiveIon=positiveIon, negativeIon=negativeIon, ionicStrength=ionicStrength, neutralize=neutralize) self._addIons(forcefield, waterPos, positiveIon=positiveIon, negativeIon=negativeIon, ionicStrength=ionicStrength, neutralize=neutralize)
......
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