Skip to content

Commit 583e70b

Browse files
committed
Decouple Sire seed from that used for OpenMM.
1 parent 9b11039 commit 583e70b

1 file changed

Lines changed: 22 additions & 22 deletions

File tree

wrapper/Tools/OpenMMMD.py

Lines changed: 22 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -508,11 +508,14 @@ def setupMoves(system, debug_seed, GPUS):
508508
moves = WeightedMoves()
509509
moves.add(mdmove, 1)
510510

511-
if debug_seed != 0:
512-
debug_seed = RanGenerator().randInt(100000, 1000000)
513-
print("Generated debugging seed number %d " % debug_seed)
511+
# Choose a random seed for Sire if a debugging seed hasn't been set.
512+
if debug_seed == 0:
513+
seed = RanGenerator().randInt(100000, 1000000)
514+
else:
515+
seed = debug_seed
516+
print("Using debugging seed number %d " % debug_seed)
514517

515-
moves.setGenerator(RanGenerator(debug_seed))
518+
moves.setGenerator(RanGenerator(seed))
516519

517520
return moves
518521

@@ -1203,10 +1206,17 @@ def setupMovesFreeEnergy(system, debug_seed, GPUS, lam_val):
12031206
Integrator_OpenMM.setMCBarostat(barostat.val)
12041207
Integrator_OpenMM.setMCBarostatFrequency(barostat_frequency.val)
12051208

1209+
# Choose a random seed for Sire if a debugging seed hasn't been set.
1210+
if debug_seed == 0:
1211+
seed = RanGenerator().randInt(100000, 1000000)
1212+
else:
1213+
seed = debug_seed
1214+
print("Using debugging seed number %d " % debug_seed)
1215+
12061216
#This calls the OpenMMFrEnergyST initialise function
12071217
Integrator_OpenMM.initialise()
12081218
velocity_generator = MaxwellBoltzmann(temperature.val)
1209-
velocity_generator.setGenerator(RanGenerator(debug_seed))
1219+
velocity_generator.setGenerator(RanGenerator(seed))
12101220

12111221
mdmove = MolecularDynamics(molecules, Integrator_OpenMM, timestep.val,
12121222
{"velocity generator":velocity_generator})
@@ -1216,11 +1226,7 @@ def setupMovesFreeEnergy(system, debug_seed, GPUS, lam_val):
12161226
moves = WeightedMoves()
12171227
moves.add(mdmove, 1)
12181228

1219-
if debug_seed != 0:
1220-
debug_seed = RanGenerator().randInt(100000, 1000000)
1221-
print("Generated debugging seed number %d " % debug_seed)
1222-
1223-
moves.setGenerator(RanGenerator(debug_seed))
1229+
moves.setGenerator(RanGenerator(seed))
12241230

12251231
return moves
12261232

@@ -1429,13 +1435,10 @@ def run():
14291435

14301436
system = setupForcefields(system, space)
14311437

1432-
if debug_seed.val == 0:
1433-
ranseed = debug_seed.val
1434-
else:
1435-
ranseed = RanGenerator().randInt(100000, 1000000)
1436-
print("Setting up the simulation with debugging seed %s" % ranseed)
1438+
if debug_seed.val != 0:
1439+
print("Setting up the simulation with debugging seed %s" % debug_seed.val)
14371440

1438-
moves = setupMoves(system, ranseed, gpu.val)
1441+
moves = setupMoves(system, debug_seed.val, gpu.val)
14391442

14401443
print("Saving restart")
14411444
Sire.Stream.save([system, moves], restart_file.val)
@@ -1586,13 +1589,10 @@ def runFreeNrg():
15861589

15871590
system = setupForceFieldsFreeEnergy(system, space)
15881591

1589-
if debug_seed.val == 0:
1590-
ranseed = debug_seed.val
1591-
else:
1592-
ranseed = RanGenerator().randInt(100000, 1000000)
1593-
print("Setting up the simulation with debugging seed %s" % ranseed)
1592+
if debug_seed.val != 0:
1593+
print("Setting up the simulation with debugging seed %s" % debug_seed.val)
15941594

1595-
moves = setupMovesFreeEnergy(system, ranseed, gpu.val, lambda_val.val)
1595+
moves = setupMovesFreeEnergy(system, debug_seed.val, gpu.val, lambda_val.val)
15961596

15971597
print("Saving restart")
15981598
Sire.Stream.save([system, moves], restart_file.val)

0 commit comments

Comments
 (0)