Skip to content

GitLab

Commit 813772e8 authored by egallicc's avatar egallicc Committed by peastman
Browse files

Implementation of OPLS through DMS reader (#2199) 

* Implementation of OPLS through DMS reader

* Improved Implementation of OPLS through DMS reader

* raise errors for invalid implicit solvent settings
parent 934e06db
Show whitespace changes
Inline Side-by-side
Showing with 737 additions and 255 deletions
wrappers/python/simtk/openmm/app/desmonddmsfile.py
View file @ 813772e8
...@@ -3,8 +3,7 @@ desmonddmsfile.py: Load Desmond dms files ...@@ -3,8 +3,7 @@ desmonddmsfile.py: Load Desmond dms files
Portions copyright (c) 2013 Stanford University and the Authors Portions copyright (c) 2013 Stanford University and the Authors
Authors: Robert McGibbon Authors: Robert McGibbon
Contributors: Contributors: Emilio Gallicchio, Baofeng Zhang, Tony Zhao
Permission is hereby granted, free of charge, to any person obtaining a Permission is hereby granted, free of charge, to any person obtaining a
copy of this software and associated documentation files (the "Software"), copy of this software and associated documentation files (the "Software"),
...@@ -25,6 +24,7 @@ OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE ...@@ -25,6 +24,7 @@ OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
USE OR OTHER DEALINGS IN THE SOFTWARE. USE OR OTHER DEALINGS IN THE SOFTWARE.
""" """
from __future__ import absolute_import from __future__ import absolute_import
from __future__ import print_function
import os import os
import math import math
...@@ -35,7 +35,11 @@ from simtk.openmm.app import Element, Topology, PDBFile ...@@ -35,7 +35,11 @@ from simtk.openmm.app import Element, Topology, PDBFile
from simtk.openmm.app.element import hydrogen from simtk.openmm.app.element import hydrogen
from simtk.unit import (nanometer, angstrom, dalton, radian, from simtk.unit import (nanometer, angstrom, dalton, radian,
kilocalorie_per_mole, kilojoule_per_mole, kilocalorie_per_mole, kilojoule_per_mole,
degree, elementary_charge) degree, elementary_charge, femtosecond)
from simtk.openmm.app.amberprmtopfile import HCT
from simtk.openmm.app.internal.customgbforces import GBSAHCTForce
import simtk.unit as u
class DesmondDMSFile(object): class DesmondDMSFile(object):
...@@ -43,14 +47,17 @@ class DesmondDMSFile(object): ...@@ -43,14 +47,17 @@ class DesmondDMSFile(object):
constructs a topology and (optionally) an OpenMM System from it constructs a topology and (optionally) an OpenMM System from it
""" """
def __init__(self, file): def __init__(self, file, verbose=False):
"""Load a DMS file """Load a DMS file
Parameters Parameters
---------- ----------
file : string file : string (for one file)
the name of the file to load or
file : list of strings (multiple files, each containing a molecule)
the name(s) of the file to load
""" """
# sqlite3 is included in the standard lib, but at python # sqlite3 is included in the standard lib, but at python
# compile time, you can disable support (I think), so it's # compile time, you can disable support (I think), so it's
# not *guarenteed* to be available. Doing the import here # not *guarenteed* to be available. Doing the import here
...@@ -58,43 +65,61 @@ class DesmondDMSFile(object): ...@@ -58,43 +65,61 @@ class DesmondDMSFile(object):
# this class, so the module can be safely imported # this class, so the module can be safely imported
import sqlite3 import sqlite3
self._open = False self._verbose = verbose
self._tables = None self._file = []
if not os.path.exists(str(file)): self._open = []
raise IOError("No such file or directory: '%s'" % str(file)) self._tables = []
self._conn = sqlite3.connect(file) self._conn = []
self._open = True self._provenance = []
self._readSchemas() self._natoms = [] #number of atoms in each file
if len(self._tables) == 0: #constructs list of files even if only one file
raise IOError('DMS file was not loaded sucessfully. No tables found') if isinstance(file, list):
if 'nbtype' not in self._tables['particle']: self._file = file
raise ValueError('No nonbonded parameters associated with this ' else:
'DMS file. You can add a forcefield with the ' self._file.append(file)
'viparr command line tool distributed with desmond')
#open sqlite connectors for each of the files
# build the provenance string nfiles = len(self._file)
provenance = [] for (fcounter,f) in zip(range(0,nfiles),self._file):
q = """SELECT id, user, timestamp, version, workdir, cmdline, executable self._open.append(False)
FROM provenance""" if not os.path.exists(str(f)):
#for id, user, timestamp, version, workdir, cmdline, executable in self._conn.execute(q): raise IOError("No such file or directory: '%s'" % str(f))
for row in self._conn.execute('SELECT * FROM provenance'): conn = sqlite3.connect(f)
rowdict = dict(zip(self._tables['provenance'], row)) self._conn.append(conn)
provenance.append('%(id)d) %(timestamp)s: %(user)s\n version: %(version)s\n ' self._open[fcounter] = True
'cmdline: %(cmdline)s\n executable: %(executable)s\n' % rowdict) tables = self._readSchemas(conn)
self.provenance = ''.join(provenance) if len(tables) == 0:
raise IOError('DMS file %s was not loaded sucessfully. No tables found' % str(f))
if 'nbtype' not in tables['particle']:
raise ValueError('No nonbonded parameters associated with '
'DMS file %s. You can add a forcefield with the '
'viparr command line tool distributed with desmond' % str(f))
self._tables.append(tables)
self._natoms.append(0)
# Build the topology # Build the topology
self.topology, self.positions = self._createTopology() self.topology, self.positions, self.velocities = self._createTopology()
self._topologyAtoms = list(self.topology.atoms()) self._topologyAtoms = list(self.topology.atoms())
self._atomBonds = [{} for x in range(len(self._topologyAtoms))] self._atomBonds = [{} for x in range(len(self._topologyAtoms))]
self._angleConstraints = [{} for x in range(len(self._topologyAtoms))] self._angleConstraints = [{} for x in range(len(self._topologyAtoms))]
#offset to each molecule/file in the topology
self._offset = self._prefixsum(self._natoms)
#retrieve provenance
self._createProvenance()
def getPositions(self): def getPositions(self):
"""Get the positions of each atom in the system """Get the positions of each atom in the system
""" """
return self.positions return self.positions
def getVelocities(self):
"""Get the positions of each atom in the system
"""
return self.velocities
def getTopology(self): def getTopology(self):
"""Get the topology of the system """Get the topology of the system
""" """
...@@ -103,27 +128,38 @@ class DesmondDMSFile(object): ...@@ -103,27 +128,38 @@ class DesmondDMSFile(object):
def getProvenance(self): def getProvenance(self):
"""Get the provenance string of this system """Get the provenance string of this system
""" """
return self.provenance if len(self._conn) == 1:
return self._provenance[0]
else:
return self._provenance
def _createTopology(self): def _createTopology(self):
"""Build the topology of the system """Build the topology of the system
""" """
top = Topology() top = Topology()
positions = [] positions = []
velocities = []
boxVectors = [] boxVectors = []
for x, y, z in self._conn.execute('SELECT x, y, z FROM global_cell'):
#assume cell dimensions are set in the first file
#the other molecules inherit the same cell
conn = self._conn[0]
for x, y, z in conn.execute('SELECT x, y, z FROM global_cell'):
boxVectors.append(mm.Vec3(x, y, z)) boxVectors.append(mm.Vec3(x, y, z))
unitCellDimensions = [boxVectors[0][0], boxVectors[1][1], boxVectors[2][2]] unitCellDimensions = [boxVectors[0][0], boxVectors[1][1], boxVectors[2][2]]
top.setUnitCellDimensions(unitCellDimensions*angstrom) top.setUnitCellDimensions(unitCellDimensions*angstrom)
#process each file
nfiles = len(self._conn)
for (fcounter, conn, tables) in zip(range(0,nfiles),self._conn,self._tables):
atoms = {} atoms = {}
lastChain = None lastChain = None
lastResId = None lastResId = None
c = top.addChain() c = top.addChain()
q = """SELECT id, name, anum, resname, resid, chain, x, y, z q = """SELECT id, name, anum, resname, resid, chain, x, y, z, vx, vy, vz
FROM particle""" FROM particle ORDER BY id"""
for (atomId, atomName, atomNumber, resName, resId, chain, x, y, z) in self._conn.execute(q): for (atomId, atomName, atomNumber, resName, resId, chain, x, y, z, vx, vy, vz) in conn.execute(q):
newChain = False newChain = False
if chain != lastChain: if chain != lastChain:
lastChain = chain lastChain = chain
...@@ -150,14 +186,124 @@ class DesmondDMSFile(object): ...@@ -150,14 +186,124 @@ class DesmondDMSFile(object):
atoms[atomId] = top.addAtom(atomName, elem, r) atoms[atomId] = top.addAtom(atomName, elem, r)
positions.append(mm.Vec3(x, y, z)) positions.append(mm.Vec3(x, y, z))
for p0, p1 in self._conn.execute('SELECT p0, p1 FROM bond'): velocities.append(mm.Vec3(vx, vy, vz))
self._natoms[fcounter] = len(atoms)
for p0, p1 in conn.execute('SELECT p0, p1 FROM bond'):
top.addBond(atoms[p0], atoms[p1]) top.addBond(atoms[p0], atoms[p1])
positions = positions*angstrom positions = positions*angstrom
return top, positions velocities = velocities*angstrom/femtosecond
return top, positions, velocities
def setPositions(self, positions):
"""Update atomic positions in attached DMS files
"""
q = """UPDATE particle SET x = ?1, y = ?2, z = ?3 WHERE id == ?4"""
#file counter
iat = 0 #global atom counter
for (fcounter,conn,tables,offset) in self._localVars():
natoms = self._natoms[fcounter]
for iat_in_file in range(0,natoms):
vec = positions[iat]
(x, y , z) = vec.value_in_unit(angstrom)
conn.execute(q, (x,y,z,iat_in_file))
iat += 1
conn.commit()
return iat
def setVelocities(self, velocities):
"""Update atomic velocities in attached DMS files
"""
q = """UPDATE particle SET vx = ?1, vy = ?2, vz = ?3 WHERE id == ?4"""
#file counter
iat = 0 #global atom counter
for (fcounter,conn,tables,offset) in self._localVars():
natoms = self._natoms[fcounter]
for iat_in_file in range(0,natoms):
vec = velocities[iat]
(vx, vy , vz) = vec.value_in_unit(angstrom/femtosecond)
conn.execute(q, (vx,vy,vz,iat_in_file))
iat += 1
conn.commit()
return iat
def _get_gb_params(self):
"""
get charge, radius, screened_radius from hct table in the .dms files
--radiusN=radius-0.009 # Offset radius, the radius needs to be converted to nanometer unit
--screenN=screened_radius*radiusN #Scaled offset radius
"""
length_conv = u.angstrom.conversion_factor_to(u.nanometer)
q="""SELECT charge,radius,screened_radius FROM hct ORDER BY id"""
gb_p = []
try:
for (fcounter,conn,tables,offset) in self._localVars():
for charge,radius,screened_radius in conn.execute(q):
chargeN = charge
radiusN = radius
screenN = screened_radius
radiusN *=length_conv
radiusN -=0.009
screenN *=radiusN
gb_p.append((chargeN,radiusN,screenN))
return gb_p
except:
return None
def _get_agbnp2_params(self):
"""
get charge, radius, etc. from AGBNP2 table in the .dms file and computes AGBNP2 parameters for each atom
"""
length_conv = u.angstrom.conversion_factor_to(u.nanometer)
en_conv = u.kilocalorie_per_mole.conversion_factor_to(u.kilojoule_per_mole)
gamma_conv = en_conv/(length_conv*length_conv)
alpha_conv = en_conv*length_conv*length_conv*length_conv
# gather AGBNP2 parameters from agbnp2 table
# atomic number and charge from particle table matching atom id
# sigma and epsilon from nonbonded_param table matching hbtype from particle
q="""SELECT anum,charge,radius,igamma,ialpha,idelta,sgamma,salpha,sdelta,hbtype,hbw,sigma,epsilon from particle INNER JOIN agbnp2 ON particle.id==agbnp2.id INNER JOIN nonbonded_param ON particle.nbtype==nonbonded_param.id ORDER BY particle.id"""
gb_p = []
try:
for (fcounter,conn,tables,offset) in self._localVars():
for anum,charge,radius,igamma,ialpha,idelta,sgamma,salpha,sdelta,hbtype,hbw,sigma,epsilon in conn.execute(q):
if anum == 1:
ishydrogenN = 1
else:
ishydrogenN = 0
radiusN = length_conv*radius
chargeN = charge
gammaN = gamma_conv*(igamma+0.*sgamma)#AGBNP must have only one gamma
alphaN = alpha_conv*(ialpha+salpha)
# delta parameter is ignored
hbtypeN = hbtype
hbwN = en_conv * hbw
gb_p.append([radiusN,chargeN,gammaN,alphaN,hbtype,hbwN,ishydrogenN])
return gb_p
except:
return None
def _add_agbnp2_ct(self,gb):
"""
adds connection table information to AGBNP3 force
"""
q = """SELECT p0,p1 FROM bond"""
for (fcounter,conn,tables,offset) in self._localVars():
for p0, p1 in conn.execute(q):
gb.addParticleConnection(p0+offset,p1+offset)
def createSystem(self, nonbondedMethod=ff.NoCutoff, nonbondedCutoff=1.0*nanometer, def createSystem(self, nonbondedMethod=ff.NoCutoff, nonbondedCutoff=1.0*nanometer,
ewaldErrorTolerance=0.0005, removeCMMotion=True, hydrogenMass=None): ewaldErrorTolerance=0.0005, removeCMMotion=True, hydrogenMass=None,
OPLS=False, implicitSolvent=None, AGBNPVersion=1):
"""Construct an OpenMM System representing the topology described by this """Construct an OpenMM System representing the topology described by this
DMS file DMS file
...@@ -176,12 +322,19 @@ class DesmondDMSFile(object): ...@@ -176,12 +322,19 @@ class DesmondDMSFile(object):
The mass to use for hydrogen atoms bound to heavy atoms. Any mass The mass to use for hydrogen atoms bound to heavy atoms. Any mass
added to a hydrogen is subtracted from the heavy atom to keep their added to a hydrogen is subtracted from the heavy atom to keep their
total mass the same. total mass the same.
OPLS : boolean=False
If True, forces OPLS combining rules
implicitSolvent: string=None
If not None, creates implicit solvent force of the given name
Allowed values are: HCT and 'AGBNP'
(the corresponding tables must be present in the DMS file)
AGBNPVersion: int=1
AGBNP implicit solvent version
""" """
self._checkForUnsupportedTerms() self._checkForUnsupportedTerms()
sys = mm.System() sys = mm.System()
# Buld the box dimensions # Build the box dimensions
sys = mm.System()
boxSize = self.topology.getUnitCellDimensions() boxSize = self.topology.getUnitCellDimensions()
if boxSize is not None: if boxSize is not None:
sys.setDefaultPeriodicBoxVectors((boxSize[0], 0, 0), (0, boxSize[1], 0), (0, 0, boxSize[2])) sys.setDefaultPeriodicBoxVectors((boxSize[0], 0, 0), (0, boxSize[1], 0), (0, 0, boxSize[2]))
...@@ -189,18 +342,20 @@ class DesmondDMSFile(object): ...@@ -189,18 +342,20 @@ class DesmondDMSFile(object):
raise ValueError('Illegal nonbonded method for a non-periodic system') raise ValueError('Illegal nonbonded method for a non-periodic system')
# Create all of the particles # Create all of the particles
for mass in self._conn.execute('SELECT mass from particle'): for (fcounter,conn,tables,offset) in self._localVars():
for mass in conn.execute('SELECT mass FROM particle ORDER BY id'):
sys.addParticle(mass[0]*dalton) sys.addParticle(mass[0]*dalton)
# Add all of the forces # Add all of the forces
self._addBondsToSystem(sys) self._addBondsToSystem(sys)
self._addAnglesToSystem(sys) self._addAnglesToSystem(sys)
self._addConstraintsToSystem(sys) self._addConstraintsToSystem(sys)
self._addPeriodicTorsionsToSystem(sys) self._addPeriodicTorsionsToSystem(sys, OPLS)
self._addImproperHarmonicTorsionsToSystem(sys) self._addImproperHarmonicTorsionsToSystem(sys)
self._addCMAPToSystem(sys) self._addCMAPToSystem(sys)
self._addVirtualSitesToSystem(sys) self._addVirtualSitesToSystem(sys)
nb = self._addNonbondedForceToSystem(sys) self._addPositionalHarmonicRestraints(sys)
nb, cnb = self._addNonbondedForceToSystem(sys, OPLS)
# Finish configuring the NonbondedForce. # Finish configuring the NonbondedForce.
methodMap = {ff.NoCutoff:mm.NonbondedForce.NoCutoff, methodMap = {ff.NoCutoff:mm.NonbondedForce.NoCutoff,
...@@ -212,10 +367,107 @@ class DesmondDMSFile(object): ...@@ -212,10 +367,107 @@ class DesmondDMSFile(object):
nb.setNonbondedMethod(methodMap[nonbondedMethod]) nb.setNonbondedMethod(methodMap[nonbondedMethod])
nb.setCutoffDistance(nonbondedCutoff) nb.setCutoffDistance(nonbondedCutoff)
nb.setEwaldErrorTolerance(ewaldErrorTolerance) nb.setEwaldErrorTolerance(ewaldErrorTolerance)
if cnb is not None:
nb.setUseDispersionCorrection(False)
if nonbondedMethod in (ff.CutoffPeriodic, ff.Ewald, ff.PME, ff.LJPME):
cnb.setNonbondedMethod(methodMap[ff.CutoffPeriodic])
cnb.setCutoffDistance(nonbondedCutoff)
elif nonbondedMethod == ff.CutoffNonPeriodic:
cnb.setNonbondedMethod(methodMap[ff.CutoffNonPeriodic])
cnb.setCutoffDistance(nonbondedCutoff)
else:
cnb.setNonbondedMethod(methodMap[ff.NoCutoff])
cnb.setUseSwitchingFunction(False)
cnb.setUseLongRangeCorrection(False)
#add implicit solvent model.
if implicitSolvent is not None:
if not (implicitSolvent in (HCT, 'AGBNP', 'GVolSA', 'AGBNP3')):
raise ValueError('Illegal implicit solvent method')
if self._verbose:
print('Adding implicit solvent ...')
#with implicit solvent turn off native reaction field
#However note that this does not affect the shifted Coulomb potential of the Nonbonded force
#(it affects the only the energy, not the forces and equation of motion)
nb.setReactionFieldDielectric(1.0)
if implicitSolvent is HCT:
gb_parms = self._get_gb_params()
if gb_parms:
if self._verbose:
print('Adding HCT GB force ...')
gb = GBSAHCTForce(SA='ACE')
for i in range(len(gb_parms)):
gb.addParticle(list(gb_parms[i]))
gb.finalize()
sys.addForce(gb)
else:
raise IOError("No HCT parameters found in DMS file")
if implicitSolvent is 'AGBNP3':
#load AGBNP3 plugin if available
try:
from AGBNP3plugin import AGBNP3Force
except ImportError:
raise NotImplementedError('AGBNP3 is not supported in this version')
#sets up AGBNP3
gb_parms = self._get_agbnp2_params()
if gb_parms:
if self._verbose:
print('Adding AGBNP3 force ...')
gb = AGBNP3Force()
# add particles
for i in range(len(gb_parms)):
p = gb_parms[i]
gb.addParticle(p[0],p[1],p[2],p[3],p[4],p[5],p[6])
# connection table (from bonds)
self._add_agbnp2_ct(gb)
sys.addForce(gb)
else:
raise IOError("No AGBNP parameters found in DMS file")
if implicitSolvent is 'GVolSA':
#implemented as AGBNP version 0
implicitSolvent = 'AGBNP'
AGBNPVersion = 0
if self._verbose:
print('Using GVolSA')
if implicitSolvent is 'AGBNP':
#load AGBNP plugin if available
try:
from AGBNPplugin import AGBNPForce
except ImportError:
raise NotImplementedError('AGBNP is not supported in this version')
#sets up AGBNP
gb_parms = self._get_agbnp2_params()
if gb_parms:
gb = AGBNPForce()
gb.setNonbondedMethod(methodMap[nonbondedMethod])
gb.setCutoffDistance(nonbondedCutoff)
gb.setVersion(AGBNPVersion)
if self._verbose:
print('Using AGBNP force version %d ...' % AGBNPVersion)
# add particles
for i in range(len(gb_parms)):
[radiusN,chargeN,gammaN,alphaN,hbtype,hbwN,ishydrogenN] = gb_parms[i]
h_flag = ishydrogenN > 0
gb.addParticle(radiusN, gammaN, alphaN, chargeN, h_flag)
sys.addForce(gb)
self.gb_parms = gb_parms
self.agbnp = gb
else:
raise IOError("No AGBNP parameters found in DMS file")
# Adjust masses. # Adjust masses.
if hydrogenMass is not None: if hydrogenMass is not None:
for atom1, atom2 in self.topology.bonds(): for atom1, atom2 in self.topology.bonds():
if atom1.element == hydrogen: if atom1.element == hydrogen:
(atom1, atom2) = (atom2, atom1) (atom1, atom2) = (atom2, atom1)
if atom2.element == hydrogen and atom1.element not in (hydrogen, None): if atom2.element == hydrogen and atom1.element not in (hydrogen, None):
...@@ -238,7 +490,10 @@ class DesmondDMSFile(object): ...@@ -238,7 +490,10 @@ class DesmondDMSFile(object):
q = """SELECT p0, p1, r0, fc, constrained q = """SELECT p0, p1, r0, fc, constrained
FROM stretch_harm_term INNER JOIN stretch_harm_param FROM stretch_harm_term INNER JOIN stretch_harm_param
ON stretch_harm_term.param=stretch_harm_param.id""" ON stretch_harm_term.param=stretch_harm_param.id"""
for p0, p1, r0, fc, constrained in self._conn.execute(q): for (fcounter,conn,tables,offset) in self._localVars():
for p0, p1, r0, fc, constrained in conn.execute(q):
p0 += offset
p1 += offset
if constrained: if constrained:
sys.addConstraint(p0, p1, r0*angstrom) sys.addConstraint(p0, p1, r0*angstrom)
else: else:
...@@ -262,7 +517,11 @@ class DesmondDMSFile(object): ...@@ -262,7 +517,11 @@ class DesmondDMSFile(object):
q = """SELECT p0, p1, p2, theta0, fc, constrained q = """SELECT p0, p1, p2, theta0, fc, constrained
FROM angle_harm_term INNER JOIN angle_harm_param FROM angle_harm_term INNER JOIN angle_harm_param
ON angle_harm_term.param=angle_harm_param.id""" ON angle_harm_term.param=angle_harm_param.id"""
for p0, p1, p2, theta0, fc, constrained in self._conn.execute(q): for (fcounter,conn,tables,offset) in self._localVars():
for p0, p1, p2, theta0, fc, constrained in conn.execute(q):
p0 += offset
p1 += offset
p2 += offset
if constrained: if constrained:
l1 = self._atomBonds[p1][p0] l1 = self._atomBonds[p1][p0]
l2 = self._atomBonds[p1][p2] l2 = self._atomBonds[p1][p2]
...@@ -282,83 +541,133 @@ class DesmondDMSFile(object): ...@@ -282,83 +541,133 @@ class DesmondDMSFile(object):
added by the bonds table, but we want to make sure that there's added by the bonds table, but we want to make sure that there's
no extra information in the constraints table that we're not no extra information in the constraints table that we're not
including in the system""" including in the system"""
for term_table in [n for n in self._tables.keys() if n.startswith('constraint_a') and n.endswith('term')]:
for (fcounter,conn,tables,offset) in self._localVars():
for term_table in [n for n in list(tables.keys()) if n.startswith('constraint_a') and n.endswith('term')]:
param_table = term_table.replace('term', 'param') param_table = term_table.replace('term', 'param')
q = """SELECT p0, p1, r1 q = """SELECT p0, p1, r1
FROM %(term)s INNER JOIN %(param)s FROM %(term)s INNER JOIN %(param)s
ON %(term)s.param=%(param)s.id""" % \ ON %(term)s.param=%(param)s.id""" % \
{'term': term_table, 'param': param_table} {'term': term_table, 'param': param_table}
for p0, p1, r1 in self._conn.execute(q): for p0, p1, r1 in conn.execute(q):
p0 += offset
p1 += offset
if not p1 in self._atomBonds[p0]: if not p1 in self._atomBonds[p0]:
sys.addConstraint(p0, p1, r1*angstrom) sys.addConstraint(p0, p1, r1*angstrom)
self._atomBonds[p0][p1] = r1*angstrom self._atomBonds[p0][p1] = r1*angstrom
self._atomBonds[p1][p0] = r1*angstrom self._atomBonds[p1][p0] = r1*angstrom
if 'constraint_hoh_term' in self._tables: if 'constraint_hoh_term' in tables:
degToRad = math.pi/180 degToRad = math.pi/180
q = """SELECT p0, p1, p2, r1, r2, theta q = """SELECT p0, p1, p2, r1, r2, theta
FROM constraint_hoh_term INNER JOIN constraint_hoh_param FROM constraint_hoh_term INNER JOIN constraint_hoh_param
ON constraint_hoh_term.param=constraint_hoh_param.id""" ON constraint_hoh_term.param=constraint_hoh_param.id"""
for p0, p1, p2, r1, r2, theta in self._conn.execute(q): for p0, p1, p2, r1, r2, theta in conn.execute(q):
p0 += offset
p1 += offset
p2 += offset
# Here, p0 is the heavy atom and p1 and p2 are the H1 and H2 # Here, p0 is the heavy atom and p1 and p2 are the H1 and H2
# wihth O-H1 and O-H2 distances r1 and r2 # wihth O-H1 and O-H2 distances r1 and r2
if not (self._angleConstraints[p0].get(p1, None) == p2): if not (self._angleConstraints[p0].get(p1, None) == p2):
length = (r1*r1 + r2*r2 - 2*r1*r2*math.cos(theta*degToRad)).sqrt() length = (r1*r1 + r2*r2 - 2*r1*r2*math.cos(theta*degToRad)).sqrt()
sys.addConstraint(p1, p2, length) sys.addConstraint(p1, p2, length)
def _addPeriodicTorsionsToSystem(self, sys): def _addPeriodicTorsionsToSystem(self, sys, OPLS):
"""Create the torsion terms """Create the torsion terms
""" """
if OPLS:
periodic = mm.CustomTorsionForce('f * cos(n * theta - phi0)')
periodic.addPerTorsionParameter('n')
periodic.addPerTorsionParameter('phi0')
periodic.addPerTorsionParameter('f')
else:
periodic = mm.PeriodicTorsionForce() periodic = mm.PeriodicTorsionForce()
sys.addForce(periodic) sys.addForce(periodic)
q = """SELECT p0, p1, p2, p3, phi0, fc0, fc1, fc2, fc3, fc4, fc5, fc6 q = """SELECT p0, p1, p2, p3, phi0, fc0, fc1, fc2, fc3, fc4, fc5, fc6
FROM dihedral_trig_term INNER JOIN dihedral_trig_param FROM dihedral_trig_term INNER JOIN dihedral_trig_param
ON dihedral_trig_term.param=dihedral_trig_param.id""" ON dihedral_trig_term.param=dihedral_trig_param.id"""
for p0, p1, p2, p3, phi0, fc0, fc1, fc2, fc3, fc4, fc5, fc6 in self._conn.execute(q):
for (fcounter,conn,tables,offset) in self._localVars():
for p0, p1, p2, p3, phi0, fc0, fc1, fc2, fc3, fc4, fc5, fc6 in conn.execute(q):
p0 += offset
p1 += offset
p2 += offset
p3 += offset
for order, fc in enumerate([fc0, fc1, fc2, fc3, fc4, fc5, fc6]): for order, fc in enumerate([fc0, fc1, fc2, fc3, fc4, fc5, fc6]):
if fc == 0: if fc == 0:
continue continue
if OPLS:
periodic.addTorsion(p0, p1, p2, p3, [order, phi0*degree, fc*kilocalorie_per_mole])
else:
periodic.addTorsion(p0, p1, p2, p3, order, phi0*degree, fc*kilocalorie_per_mole) periodic.addTorsion(p0, p1, p2, p3, order, phi0*degree, fc*kilocalorie_per_mole)
def _addImproperHarmonicTorsionsToSystem(self, sys): def _addImproperHarmonicTorsionsToSystem(self, sys):
"""Create the improper harmonic torsion terms """Create the improper harmonic torsion terms
""" """
if not self._hasTable('improper_harm_term'):
return
harmonicTorsion = mm.CustomTorsionForce('k*(theta-theta0)^2') harmonicTorsion = mm.CustomTorsionForce('k*(theta-theta0)^2')
harmonicTorsion.addPerTorsionParameter('theta0') harmonicTorsion.addPerTorsionParameter('theta0')
harmonicTorsion.addPerTorsionParameter('k') harmonicTorsion.addPerTorsionParameter('k')
go = []
for (fcounter,conn,tables,offset) in self._localVars():
if not self._hasTable('improper_harm_term', tables):
go.append(False)
else:
go.append(True)
if any(go):
sys.addForce(harmonicTorsion) sys.addForce(harmonicTorsion)
else:
return
q = """SELECT p0, p1, p2, p3, phi0, fc q = """SELECT p0, p1, p2, p3, phi0, fc
FROM improper_harm_term INNER JOIN improper_harm_param FROM improper_harm_term INNER JOIN improper_harm_param
ON improper_harm_term.param=improper_harm_param.id""" ON improper_harm_term.param=improper_harm_param.id"""
for p0, p1, p2, p3, phi0, fc in self._conn.execute(q):
for (fcounter,conn,tables,offset) in self._localVars():
if not go[fcounter]:
continue
for p0, p1, p2, p3, phi0, fc in conn.execute(q):
p0 += offset
p1 += offset
p2 += offset
p3 += offset
harmonicTorsion.addTorsion(p0, p1, p2, p3, [phi0*degree, fc*kilocalorie_per_mole]) harmonicTorsion.addTorsion(p0, p1, p2, p3, [phi0*degree, fc*kilocalorie_per_mole])
def _addCMAPToSystem(self, sys): def _addCMAPToSystem(self, sys):
"""Create the CMAP terms """Create the CMAP terms
""" """
if not self._hasTable('torsiontorsion_cmap_term'): go = []
return
for (fcounter,conn,tables,offset) in self._localVars():
if not self._hasTable('torsiontorsion_cmap_term', tables):
go.append(False)
else:
go.append(True)
if any(go):
# Create CMAP torsion terms # Create CMAP torsion terms
cmap = mm.CMAPTorsionForce() cmap = mm.CMAPTorsionForce()
sys.addForce(cmap) sys.addForce(cmap)
else:
return
cmap_indices = {} cmap_indices = {}
for name in [k for k in self._tables.keys() if k.startswith('cmap')]: for (fcounter,conn,tables,offset) in self._localVars():
size2 = self._conn.execute('SELECT COUNT(*) FROM %s' % name).fetchone()[0] if not go[fcounter]:
continue
for name in [k for k in list(tables.keys()) if k.startswith('cmap')]:
size2 = conn.execute('SELECT COUNT(*) FROM %s' % name).fetchone()[0]
fsize = math.sqrt(size2) fsize = math.sqrt(size2)
if fsize != int(fsize): if fsize != int(fsize):
raise ValueError('Non-square CMAPs are not supported') raise ValueError('Non-square CMAPs are not supported')
size = int(fsize) size = int(fsize)
map = [0 for i in range(size2)] map = [0 for i in range(size2)]
for phi, psi, energy in self._conn.execute("SELECT phi, psi, energy FROM %s" % name): for phi, psi, energy in conn.execute("SELECT phi, psi, energy FROM %s" % name):
i = int((phi % 360) / (360.0 / size)) i = int((phi % 360) / (360.0 / size))
j = int((psi % 360) / (360.0 / size)) j = int((psi % 360) / (360.0 / size))
map[i+size*j] = energy map[i+size*j] = energy
...@@ -368,32 +677,65 @@ class DesmondDMSFile(object): ...@@ -368,32 +677,65 @@ class DesmondDMSFile(object):
q = """SELECT p0, p1, p2, p3, p4, p5, p6, p7, cmapid q = """SELECT p0, p1, p2, p3, p4, p5, p6, p7, cmapid
FROM torsiontorsion_cmap_term INNER JOIN torsiontorsion_cmap_param FROM torsiontorsion_cmap_term INNER JOIN torsiontorsion_cmap_param
ON torsiontorsion_cmap_term.param=torsiontorsion_cmap_param.id""" ON torsiontorsion_cmap_term.param=torsiontorsion_cmap_param.id"""
for p0, p1, p2, p3, p4, p5, p6, p7, cmapid in self._conn.execute(q): for p0, p1, p2, p3, p4, p5, p6, p7, cmapid in conn.execute(q):
p0 += offset
p1 += offset
p2 += offset
p3 += offset
p4 += offset
p5 += offset
p6 += offset
p7 += offset
cmap.addTorsion(cmap_indices[cmapid], p0, p1, p2, p3, p4, p5, p6, p7) cmap.addTorsion(cmap_indices[cmapid], p0, p1, p2, p3, p4, p5, p6, p7)
def _addNonbondedForceToSystem(self, sys): def _addNonbondedForceToSystem(self, sys, OPLS):
"""Create the nonbonded force """Create the nonbonded force
""" """
cnb = None
nb = mm.NonbondedForce() nb = mm.NonbondedForce()
sys.addForce(nb) sys.addForce(nb)
if OPLS:
cnb = mm.CustomNonbondedForce("4.0*epsilon12*((sigma12/r)^12 - (sigma12/r)^6); sigma12=sqrt(sigma1*sigma2); epsilon12=sqrt(epsilon1*epsilon2)")
cnb.addPerParticleParameter("sigma")
cnb.addPerParticleParameter("epsilon")
sys.addForce(cnb)
if OPLS:
q = """SELECT sigma, epsilon
FROM particle INNER JOIN nonbonded_param
ON particle.nbtype=nonbonded_param.id ORDER BY particle.id"""
for (fcounter,conn,tables,offset) in self._localVars():
for sigma, epsilon in conn.execute(q):
cnb.addParticle([sigma*angstrom, epsilon*kilocalorie_per_mole])
q = """SELECT charge, sigma, epsilon q = """SELECT charge, sigma, epsilon
FROM particle INNER JOIN nonbonded_param FROM particle INNER JOIN nonbonded_param
ON particle.nbtype=nonbonded_param.id""" ON particle.nbtype=nonbonded_param.id ORDER BY particle.id"""
for charge, sigma, epsilon in self._conn.execute(q): for (fcounter,conn,tables,offset) in self._localVars():
for charge, sigma, epsilon in conn.execute(q):
if OPLS:
epsilon = 0
nb.addParticle(charge, sigma*angstrom, epsilon*kilocalorie_per_mole) nb.addParticle(charge, sigma*angstrom, epsilon*kilocalorie_per_mole)
for p0, p1 in self._conn.execute('SELECT p0, p1 FROM exclusion'): for (fcounter,conn,tables,offset) in self._localVars():
for p0, p1 in conn.execute('SELECT p0, p1 FROM exclusion'):
p0 += offset
p1 += offset
nb.addException(p0, p1, 0.0, 1.0, 0.0) nb.addException(p0, p1, 0.0, 1.0, 0.0)
if OPLS:
cnb.addExclusion(p0, p1)
q = """SELECT p0, p1, aij, bij, qij q = """SELECT p0, p1, aij, bij, qij
FROM pair_12_6_es_term INNER JOIN pair_12_6_es_param FROM pair_12_6_es_term INNER JOIN pair_12_6_es_param
ON pair_12_6_es_term.param=pair_12_6_es_param.id;""" ON pair_12_6_es_term.param=pair_12_6_es_param.id"""
for p0, p1, a_ij, b_ij, q_ij in self._conn.execute(q): for (fcounter,conn,tables,offset) in self._localVars():
for p0, p1, a_ij, b_ij, q_ij in conn.execute(q):
p0 += offset
p1 += offset
a_ij = (a_ij*kilocalorie_per_mole*(angstrom**12)).in_units_of(kilojoule_per_mole*(nanometer**12)) a_ij = (a_ij*kilocalorie_per_mole*(angstrom**12)).in_units_of(kilojoule_per_mole*(nanometer**12))
b_ij = (b_ij*kilocalorie_per_mole*(angstrom**6)).in_units_of(kilojoule_per_mole*(nanometer**6)) b_ij = (b_ij*kilocalorie_per_mole*(angstrom**6)).in_units_of(kilojoule_per_mole*(nanometer**6))
q_ij = q_ij*elementary_charge**2 q_ij = q_ij*elementary_charge**2
if (b_ij._value == 0.0) or (a_ij._value == 0.0): if (b_ij._value == 0.0) or (a_ij._value == 0.0):
new_epsilon = 0 new_epsilon = 0
new_sigma = 1 new_sigma = 1
...@@ -402,107 +744,217 @@ class DesmondDMSFile(object): ...@@ -402,107 +744,217 @@ class DesmondDMSFile(object):
new_sigma = (a_ij / b_ij)**(1.0/6.0) new_sigma = (a_ij / b_ij)**(1.0/6.0)
nb.addException(p0, p1, q_ij, new_sigma, new_epsilon, True) nb.addException(p0, p1, q_ij, new_sigma, new_epsilon, True)
n_total = self._conn.execute("""SELECT COUNT(*) FROM pair_12_6_es_term""").fetchone() n_total = conn.execute("""SELECT COUNT(*) FROM pair_12_6_es_term""").fetchone()
n_in_exclusions = self._conn.execute("""SELECT COUNT(*) n_in_exclusions = conn.execute("""SELECT COUNT(*)
FROM exclusion INNER JOIN pair_12_6_es_term FROM exclusion INNER JOIN pair_12_6_es_term
ON exclusion.p0==pair_12_6_es_term.p0 AND exclusion.p1==pair_12_6_es_term.p1""").fetchone() ON ( ( exclusion.p0==pair_12_6_es_term.p0 AND exclusion.p1==pair_12_6_es_term.p1)
OR ( exclusion.p0==pair_12_6_es_term.p1 AND exclusion.p1==pair_12_6_es_term.p0)
)""").fetchone()
if not n_total == n_in_exclusions: if not n_total == n_in_exclusions:
raise NotImplementedError('All pair_12_6_es_terms must have a corresponding exclusion') raise NotImplementedError('All pair_12_6_es_terms must have a corresponding exclusion')
return nb return nb, cnb
def _addVirtualSitesToSystem(self, sys): def _addVirtualSitesToSystem(self, sys):
"""Create any virtual sites in the systempy """Create any virtual sites in the system
""" """
if not any(t.startswith('virtual_') for t in self._tables.keys()): go = []
for (fcounter,conn,tables,offset) in self._localVars():
if not any(t.startswith('virtual_') for t in list(tables.keys())):
go.append(False)
else:
go.append(True)
if not any(go):
return return
if 'virtual_lc2_term' in self._tables: for (fcounter,conn,tables,offset) in self._localVars():
if not go[fcounter]:
continue
if 'virtual_lc2_term' in tables:
q = """SELECT p0, p1, p2, c1 q = """SELECT p0, p1, p2, c1
FROM virtual_lc2_term INNER JOIN virtual_lc2_param FROM virtual_lc2_term INNER JOIN virtual_lc2_param
ON virtual_lc2_term.param=virtual_lc2_param.id""" ON virtual_lc2_term.param=virtual_lc2_param.id"""
for p0, p1, p2, c1 in self._conn.execute(q): for p0, p1, p2, c1 in conn.execute(q):
p0 += offset
p1 += offset
p2 += offset
vsite = mm.TwoParticleAverageSite(p1, p2, (1-c1), c1) vsite = mm.TwoParticleAverageSite(p1, p2, (1-c1), c1)
sys.setVirtualSite(p0, vsite) sys.setVirtualSite(p0, vsite)
if 'virtual_lc3_term' in self._tables: for (fcounter,conn,tables,offset) in self._localVars():
if not go[fcounter]:
continue
if 'virtual_lc3_term' in tables:
q = """SELECT p0, p1, p2, p3, c1, c2 q = """SELECT p0, p1, p2, p3, c1, c2
FROM virtual_lc3_term INNER JOIN virtual_lc3_param FROM virtual_lc3_term INNER JOIN virtual_lc3_param
ON virtual_lc3_term.param=virtual_lc3_param.id""" ON virtual_lc3_term.param=virtual_lc3_param.id"""
for p0, p1, p2, p3, c1, c2 in self._conn.execute(q): for p0, p1, p2, p3, c1, c2 in conn.execute(q):
p0 += offset
p1 += offset
p2 += offset
p3 += offset
vsite = mm.ThreeParticleAverageSite(p1, p2, p3, (1-c1-c2), c1, c2) vsite = mm.ThreeParticleAverageSite(p1, p2, p3, (1-c1-c2), c1, c2)
sys.setVirtualSite(p0, vsite) sys.setVirtualSite(p0, vsite)
if 'virtual_out3_term' in self._tables: for (fcounter,conn,tables,offset) in self._localVars():
if not go[fcounter]:
continue
if 'virtual_out3_term' in tables:
q = """SELECT p0, p1, p2, p3, c1, c2, c3 q = """SELECT p0, p1, p2, p3, c1, c2, c3
FROM virtual_out3_term INNER JOIN virtual_out3_param FROM virtual_out3_term INNER JOIN virtual_out3_param
ON virtual_out3_term.param=virtual_out3_param.id""" ON virtual_out3_term.param=virtual_out3_param.id"""
for p0, p1, p2, p3, c1, c2, c3 in self._conn.execute(q): for p0, p1, p2, p3, c1, c2, c3 in conn.execute(q):
p0 += offset
p1 += offset
p2 += offset
p3 += offset
vsite = mm.OutOfPlaneSite(p1, p2, p3, c1, c2, c3) vsite = mm.OutOfPlaneSite(p1, p2, p3, c1, c2, c3)
sys.setVirtualSite(p0, vsite) sys.setVirtualSite(p0, vsite)
if 'virtual_fdat3_term' in self._tables: for (fcounter,conn,tables,offset) in self._localVars():
if not go[fcounter]:
continue
if 'virtual_fdat3_term' in tables:
raise NotImplementedError('OpenMM does not currently support ' raise NotImplementedError('OpenMM does not currently support '
'fdat3-style virtual sites') 'fdat3-style virtual sites')
def _addPositionalHarmonicRestraints(self, sys):
def _hasTable(self, table_name): go = []
"""Does our DMS file contain this table?
""" for (fcounter,conn,tables,offset) in self._localVars():
return table_name in self._tables if not self._hasTable('posre_harm_term',tables):
go.append(False)
else:
go.append(True)
if go[fcounter] and (not self._hasTable('posre_harm_param',tables)):
raise IOError('DMS file lacks posre_harm_param table even though posre_harm_term table is present.')
return
if not any(go):
return
if self._verbose:
print("Using positional harmonic restraints.")
def _readSchemas(self): force = mm.CustomExternalForce("hkx*(x-x0)^2+hky*(y-y0)^2+hkz*(z-z0)^2")
"""Read the schemas of each of the tables in the dms file, populating force.addPerParticleParameter("x0")
the `_tables` instance attribute force.addPerParticleParameter("y0")
force.addPerParticleParameter("z0")
force.addPerParticleParameter("hkx")
force.addPerParticleParameter("hky")
force.addPerParticleParameter("hkz")
sys.addForce(force)
q = """SELECT p0, x0, y0, z0, fcx, fcy, fcz FROM posre_harm_term INNER JOIN posre_harm_param ON posre_harm_term.param=posre_harm_param.id"""
for (fcounter,conn,tables,offset) in self._localVars():
if not go[fcounter]:
continue
for p0, x0, y0, z0, fcx, fcy, fcz in conn.execute(q):
p0 += offset
x0d = (x0*angstrom).value_in_unit(nanometer)
y0d = (y0*angstrom).value_in_unit(nanometer)
z0d = (z0*angstrom).value_in_unit(nanometer)
hfcxd = (0.5*fcx*kilocalorie_per_mole/angstrom**2).value_in_unit(kilojoule_per_mole/(nanometer**2))
hfcyd = (0.5*fcy*kilocalorie_per_mole/angstrom**2).value_in_unit(kilojoule_per_mole/(nanometer**2))
hfczd = (0.5*fcz*kilocalorie_per_mole/angstrom**2).value_in_unit(kilojoule_per_mole/(nanometer**2))
force.addParticle(p0,[ x0d, y0d, z0d, hfcxd, hfcyd, hfczd])
def _hasTable(self, table_name, tables):
"""check existence of a table
""" """
return table_name in tables
def _readSchemas(self, conn):
"""Read and return the schemas of each of the tables in the dms file connection 'conn'"""
tables = {} tables = {}
for table in self._conn.execute("SELECT name FROM sqlite_master WHERE type='table'"): for table in conn.execute("SELECT name FROM sqlite_master WHERE type='table'"):
names = [] names = []
for e in self._conn.execute('PRAGMA table_info(%s)' % table): for e in conn.execute('PRAGMA table_info(%s)' % table):
names.append(str(e[1])) names.append(str(e[1]))
tables[str(table[0])] = names tables[str(table[0])] = names
self._tables = tables return tables
def _checkForUnsupportedTerms(self): def _checkForUnsupportedTerms(self):
"""Check the file for forcefield terms that are not currenty supported, """Check the file for forcefield terms that are not currenty supported,
raising a NotImplementedError raising a NotImplementedError
""" """
if 'posre_harm_term' in self._tables:
raise NotImplementedError('Position restraints are not implemented.')
flat_bottom_potential_terms = ['stretch_fbhw_term', 'angle_fbhw_term', flat_bottom_potential_terms = ['stretch_fbhw_term', 'angle_fbhw_term',
'improper_fbhw_term', 'posre_fbhw_term'] 'improper_fbhw_term', 'posre_fbhw_term']
if any((t in self._tables) for t in flat_bottom_potential_terms):
for (fcounter,conn,tables,offset) in self._localVars():
if any((t in tables) for t in flat_bottom_potential_terms):
raise NotImplementedError('Flat bottom potential terms ' raise NotImplementedError('Flat bottom potential terms '
'are not implemeneted') 'are not implemeneted')
nbinfo = dict(zip(tables['nonbonded_info'],
nbinfo = dict(zip(self._tables['nonbonded_info'], conn.execute('SELECT * FROM nonbonded_info').fetchone()))
self._conn.execute('SELECT * FROM nonbonded_info').fetchone())) self._opls_combining_rules = False
if nbinfo['vdw_funct'] != u'vdw_12_6': if nbinfo['vdw_funct'] != u'vdw_12_6':
raise NotImplementedError('Only Leonard-Jones van der Waals ' raise NotImplementedError('Only Lennard-Jones van der Waals '
'interactions are currently supported') 'interactions are currently supported')
if nbinfo['vdw_rule'] != u'arithmetic/geometric': if (nbinfo['vdw_rule'] != u'arithmetic') and (nbinfo['vdw_rule'] != u'geometric') and (nbinfo['vdw_rule'] != u'arithmetic/geometric'):
raise NotImplementedError('Only Lorentz-Berthelot nonbonded ' raise NotImplementedError('Unknown vdw combination rule')
'combining rules are currently supported') if nbinfo['vdw_rule'] == u'geometric':
#need to generalize this
if 'nonbonded_combined_param' in self._tables: self._opls_combining_rules = True
if nbinfo['vdw_rule'] == u'arithmetic':
self._opls_combining_rules = False
"""
if 'nonbonded_combined_param' in tables:
raise NotImplementedError('nonbonded_combined_param interactions ' raise NotImplementedError('nonbonded_combined_param interactions '
'are not currently supported') 'are not currently supported')
"""
if 'alchemical_particle' in self._tables: if 'alchemical_particle' in tables:
raise NotImplementedError('Alchemical particles are not supported') raise NotImplementedError('Alchemical particles are not supported')
if 'alchemical_stretch_harm' in self._tables: if 'alchemical_stretch_harm' in tables:
raise NotImplementedError('Alchemical bonds are not supported') raise NotImplementedError('Alchemical bonds are not supported')
if 'polar_term' in tables:
if 'polar_term' in self._tables: if conn.execute("SELECT COUNT(*) FROM polar_term").fetchone()[0] != 0:
if self._conn.execute("SELECT COUNT(*) FROM polar_term").fetchone()[0] != 0:
raise NotImplementedError('Drude particles are not currently supported') raise NotImplementedError('Drude particles are not currently supported')
def _createProvenance(self):
for (fcounter, conn, tables, offset) in self._localVars():
try:
# build the provenance string
provenance = []
q = """SELECT id, user, timestamp, version, workdir, cmdline, executable
FROM provenance"""
for id, user, timestamp, version, workdir, cmdline, executable in conn.execute(q):
for row in conn.execute('SELECT * FROM provenance'):
rowdict = dict(list(zip(tables['provenance'], row)))
provenance.append('%(id)d) %(timestamp)s: %(user)s\n version: %(version)s\n '
'cmdline: %(cmdline)s\n executable: %(executable)s\n' % rowdict)
self._provenance.append(''.join(provenance))
except:
self._provenance.append('')
if self._verbose:
print('Warning: unable to retrieve provenance information from DMS file %s' % str(f))
def _prefixsum(self, values):
""" exclusive prefix sum of 'values' """
total =0
sum = [total]
for v in values:
total += v
sum.append(total)
return sum
def _localVars(self):
counter = range(0,len(self._conn))
return zip(counter, self._conn, self._tables, self._offset)
def close(self): def close(self):
"""Close the SQL connection """Close the SQL connections
""" """
if self._open: for (fcounter,conn,tables,offset) in self._localVars():
self._conn.close() if self._open[fcounter]:
conn.close()
def __del__(self): def __del__(self):
self.close() self.close()
wrappers/python/tests/TestDesmondDMSFile.py
View file @ 813772e8
...@@ -13,6 +13,15 @@ class TestDesmondDMSFile(unittest.TestCase): ...@@ -13,6 +13,15 @@ class TestDesmondDMSFile(unittest.TestCase):
path = os.path.join(os.path.dirname(__file__), 'systems/alanine-dipeptide-explicit-amber99SBILDN-tip3p.dms') path = os.path.join(os.path.dirname(__file__), 'systems/alanine-dipeptide-explicit-amber99SBILDN-tip3p.dms')
self.dms = DesmondDMSFile(path) self.dms = DesmondDMSFile(path)
#nabumetone OPLS force field
path = os.path.join(os.path.dirname(__file__), 'systems/bcd-nabumetone_lig.dms')
self.dms_opls1 = DesmondDMSFile(path)
#beta-cyclodextrin/nabumetone complex OPLS force field
path1 = os.path.join(os.path.dirname(__file__), 'systems/bcd-nabumetone_lig.dms')
path2 = os.path.join(os.path.dirname(__file__), 'systems/bcd-nabumetone_rcpt.dms')
self.dms_opls2 = DesmondDMSFile([path1,path2])
def test_NonbondedMethod(self): def test_NonbondedMethod(self):
"""Test all six options for the nonbondedMethod parameter.""" """Test all six options for the nonbondedMethod parameter."""
...@@ -76,3 +85,24 @@ class TestDesmondDMSFile(unittest.TestCase): ...@@ -76,3 +85,24 @@ class TestDesmondDMSFile(unittest.TestCase):
totalMass1 = sum([system1.getParticleMass(i) for i in range(system1.getNumParticles())]).value_in_unit(amu) totalMass1 = sum([system1.getParticleMass(i) for i in range(system1.getNumParticles())]).value_in_unit(amu)
totalMass2 = sum([system2.getParticleMass(i) for i in range(system2.getNumParticles())]).value_in_unit(amu) totalMass2 = sum([system2.getParticleMass(i) for i in range(system2.getNumParticles())]).value_in_unit(amu)
self.assertAlmostEqual(totalMass1, totalMass2) self.assertAlmostEqual(totalMass1, totalMass2)
def test_OPLS1(self):
"""Test OPLS potential energy of nabumetone """
system = self.dms_opls1.createSystem(nonbondedMethod=NoCutoff, OPLS = True)
integrator = LangevinIntegrator(300*kelvin, 1.0/picosecond, 0.0005*picoseconds)
context = Context(system, integrator,Platform.getPlatformByName('Reference'))
context.setPositions(self.dms_opls1.positions)
ene = context.getState(getEnergy=True).getPotentialEnergy()
self.assertAlmostEqual(ene.value_in_unit(kilojoules_per_mole), 94.21, places=2)
def test_OPLS2(self):
"""Test OPLS potential energy of beta-cyclodextrin/nabumetone complex """
system = self.dms_opls2.createSystem(nonbondedMethod=NoCutoff, OPLS = True)
integrator = LangevinIntegrator(300*kelvin, 1.0/picosecond, 0.0005*picoseconds)
context = Context(system, integrator,Platform.getPlatformByName('Reference'))
context.setPositions(self.dms_opls2.positions)
ene = context.getState(getEnergy=True).getPotentialEnergy()
self.assertAlmostEqual(ene.value_in_unit(kilojoules_per_mole), 903.04, places=2)
if __name__ == '__main__':
unittest.main()
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