def test_virial_observable_CPU(self):
fname = os.path.join(self.dir, "test_observables_virial.h5")
sim = Simulation()
sim.set_kernel("CPU")
sim.box_size = common.Vec(13, 13, 13)
sim.register_particle_type("A", .1)
sim.register_potential_harmonic_repulsion("A", "A", 10., .5)
for _ in range(10000):
pos = common.Vec(*(13*np.random.random(size=3)-.5*13))
sim.add_particle("A", pos)
virials = []
def virial_callback(virial):
virials.append(np.ndarray((3,3), buffer=virial))
handle = sim.register_observable_virial(1, virial_callback)
with closing(io.File.create(fname)) as f:
handle.enable_write_to_file(f, u"virial", int(3))
sim.run_scheme_readdy(True).configure(.1).run(10)
handle.flush()
with h5py.File(fname, "r") as f2:
h5virials = f2["readdy/observables/virial/data"]
for v, v2 in zip(virials, h5virials):
np.testing.assert_almost_equal(v, v2)
def test_virial_observable_CPU(self):
fname = os.path.join(self.dir, "test_observables_virial.h5")
sim = Simulation()
sim.set_kernel("CPU")
sim.box_size = common.Vec(13, 13, 13)
sim.register_particle_type("A", .1)
sim.register_potential_harmonic_repulsion("A", "A", 10., .5)
for _ in range(10000):
pos = common.Vec(*(13 * np.random.random(size=3) - .5 * 13))
sim.add_particle("A", pos)
virials = []
def virial_callback(virial):
virials.append(np.ndarray((3, 3), buffer=virial))
handle = sim.register_observable_virial(1, virial_callback)
with closing(io.File.create(fname)) as f:
handle.enable_write_to_file(f, u"virial", int(3))
sim.run_scheme_readdy(True).configure(.1).run(10)
handle.flush()
with h5py.File(fname, "r") as f2:
h5virials = f2["readdy/observables/virial/data"]
for v, v2 in zip(virials, h5virials):
np.testing.assert_almost_equal(v, v2)
def test_histogram_along_axis_observable(self):
fname = os.path.join(self.dir, "test_observables_hist_along_axis.h5")
simulation = Simulation()
simulation.set_kernel("SingleCPU")
box_size = common.Vec(10, 10, 10)
simulation.kbt = 2
simulation.periodic_boundary = [True, True, True]
simulation.box_size = box_size
simulation.register_particle_type("A", .2)
simulation.register_particle_type("B", .2)
simulation.register_potential_harmonic_repulsion("A", "B", 10, 2.)
simulation.add_particle("A", common.Vec(-2.5, 0, 0))
simulation.add_particle("B", common.Vec(0, 0, 0))
bin_borders = np.arange(0, 5, .01)
n_time_steps = 50
callback_hist = []
def hist_callback(hist):
callback_hist.append(hist)
handle = simulation.register_observable_histogram_along_axis(2, bin_borders, 0, ["A", "B"], hist_callback)
with closing(io.File.create(fname)) as f:
handle.enable_write_to_file(f, u"hist_along_x_axis", int(3))
simulation.run(n_time_steps, 0.02)
handle.flush()
with h5py.File(fname, "r") as f2:
histogram = f2["readdy/observables/hist_along_x_axis/data"][:]
time_series = f2["readdy/observables/hist_along_x_axis/time"]
np.testing.assert_equal(time_series, np.array(range(0, n_time_steps+1))[::2])
for t in range(n_time_steps // 2):
np.testing.assert_equal(histogram[t], np.array(callback_hist[t]))
class TestInternalSimulationModule(ReaDDyTestCase):
def setUp(self):
super().setUp()
self.kernel_provider = KernelProvider.get()
self.kernel_provider.load_from_dir(platform_utils.get_readdy_plugin_dir())
self.simulation = Simulation()
def py_harmonic_repulsion_energy(self, x_ij):
dist = x_ij * x_ij
# if dist < sqrt(25): return energy with force constant 1
if dist < 25:
return (np.sqrt(dist) - 5) ** 2
else:
return 0
def py_harmonic_repulsion_force(self, x_ij):
dist = x_ij * x_ij
if dist < 25:
dist = np.sqrt(dist)
return (2 * (dist - 5) / dist) * x_ij
else:
return Vec(0, 0, 0)
def test_properties(self):
if not self.simulation.is_kernel_selected():
self.simulation.set_kernel('SingleCPU')
np.testing.assert_equal(self.simulation.is_kernel_selected(), True)
np.testing.assert_equal(self.simulation.get_selected_kernel_type(), "SingleCPU")
self.simulation.kbt = 5.0
np.testing.assert_equal(self.simulation.kbt, 5.0)
self.simulation.periodic_boundary = [True, False, True]
np.testing.assert_equal(self.simulation.periodic_boundary, (True, False, True))
self.simulation.box_size = Vec(1, 3.6, 7)
np.testing.assert_equal(self.simulation.box_size, Vec(1, 3.6, 7))
def py_position_observable_callback(self, positions):
_vec_sum = Vec(0, 0, 0)
for v in positions:
_vec_sum += v
mean = _vec_sum / float(len(positions))
print("mean=%s" % mean)
def test_potentials(self):
if not self.simulation.is_kernel_selected():
self.simulation.set_kernel("SingleCPU")
ida = self.simulation.register_particle_type("ParticleTypeA", 1.0)
idb = self.simulation.register_particle_type("ParticleTypeB", 3.0)
self.simulation.register_particle_type("ParticleTypeA_internal", 1.0)
self.simulation.register_particle_type("ParticleTypeB_internal", 3.0)
pot = Pot2(ida, idb, self.py_harmonic_repulsion_energy,
self.py_harmonic_repulsion_force)
self.simulation.register_potential_order_2(pot)
self.simulation.register_potential_harmonic_repulsion("ParticleTypeA_internal", "ParticleTypeB_internal", 1.0, 2.0)
self.simulation.add_particle("ParticleTypeA", Vec(0, 0, 0))
self.simulation.add_particle("ParticleTypeB", Vec(0.4, 0.4, 0.4))
self.simulation.run(100, 1)
def test_radial_distribution_observable(self):
common.set_logging_level("warn")
fname = os.path.join(self.dir,
"test_observables_radial_distribution.h5")
simulation = Simulation()
simulation.set_kernel("SingleCPU")
box_size = common.Vec(10, 10, 10)
simulation.kbt = 2
simulation.periodic_boundary = [True, True, True]
simulation.box_size = box_size
simulation.register_particle_type("A", .2, 1.)
simulation.register_particle_type("B", .2, 1.)
simulation.register_potential_harmonic_repulsion("A", "B", 10)
simulation.add_particle("A", common.Vec(-2.5, 0, 0))
simulation.add_particle("B", common.Vec(0, 0, 0))
bin_borders = np.arange(0, 5, .01)
density = 1. / (box_size[0] * box_size[1] * box_size[2])
n_time_steps = 50
callback_centers = []
callback_rdf = []
def rdf_callback(pair):
callback_centers.append(pair[0])
callback_rdf.append(pair[1])
handle = simulation.register_observable_radial_distribution(
1, bin_borders, ["A"], ["B"], density, rdf_callback)
with closing(
io.File(fname, io.FileAction.CREATE,
io.FileFlag.OVERWRITE)) as f:
handle.enable_write_to_file(f, u"radial_distribution", int(3))
simulation.run(n_time_steps, 0.02)
handle.flush()
with h5py.File(fname, "r") as f2:
bin_centers = f2[
"readdy/observables/radial_distribution/bin_centers"][:]
distribution = f2[
"readdy/observables/radial_distribution/distribution"][:]
for t in range(n_time_steps):
np.testing.assert_equal(bin_centers,
np.array(callback_centers[t]))
np.testing.assert_equal(distribution[t],
np.array(callback_rdf[t]))
def test_histogram_along_axis_observable(self):
common.set_logging_level("warn")
fname = os.path.join(self.dir, "test_observables_hist_along_axis.h5")
simulation = Simulation()
simulation.set_kernel("SingleCPU")
box_size = common.Vec(10, 10, 10)
simulation.kbt = 2
simulation.periodic_boundary = [True, True, True]
simulation.box_size = box_size
simulation.register_particle_type("A", .2, 1.)
simulation.register_particle_type("B", .2, 1.)
simulation.register_potential_harmonic_repulsion("A", "B", 10)
simulation.add_particle("A", common.Vec(-2.5, 0, 0))
simulation.add_particle("B", common.Vec(0, 0, 0))
bin_borders = np.arange(0, 5, .01)
n_time_steps = 50
callback_hist = []
def hist_callback(hist):
callback_hist.append(hist)
handle = simulation.register_observable_histogram_along_axis(
2, bin_borders, 0, ["A", "B"], hist_callback)
with closing(
io.File(fname, io.FileAction.CREATE,
io.FileFlag.OVERWRITE)) as f:
handle.enable_write_to_file(f, u"hist_along_x_axis", int(3))
simulation.run(n_time_steps, 0.02)
handle.flush()
with h5py.File(fname, "r") as f2:
histogram = f2["readdy/observables/hist_along_x_axis/data"][:]
time_series = f2["readdy/observables/hist_along_x_axis/time"]
np.testing.assert_equal(time_series,
np.array(range(0, n_time_steps + 1))[::2])
for t in range(n_time_steps // 2):
np.testing.assert_equal(histogram[t],
np.array(callback_hist[t]))
def test_radial_distribution_observable(self):
fname = os.path.join(self.dir, "test_observables_radial_distribution.h5")
simulation = Simulation()
simulation.set_kernel("SingleCPU")
box_size = common.Vec(10, 10, 10)
simulation.kbt = 2
simulation.periodic_boundary = [True, True, True]
simulation.box_size = box_size
simulation.register_particle_type("A", .2)
simulation.register_particle_type("B", .2)
simulation.register_potential_harmonic_repulsion("A", "B", 10, 2.)
simulation.add_particle("A", common.Vec(-2.5, 0, 0))
simulation.add_particle("B", common.Vec(0, 0, 0))
bin_borders = np.arange(0, 5, .01)
density = 1. / (box_size[0] * box_size[1] * box_size[2])
n_time_steps = 50
callback_centers = []
callback_rdf = []
def rdf_callback(pair):
callback_centers.append(pair[0])
callback_rdf.append(pair[1])
handle = simulation.register_observable_radial_distribution(1, bin_borders, ["A"], ["B"], density, rdf_callback)
with closing(io.File.create(fname)) as f:
handle.enable_write_to_file(f, u"radial_distribution", int(3))
simulation.run(n_time_steps, 0.02)
handle.flush()
with h5py.File(fname, "r") as f2:
bin_centers = f2["readdy/observables/radial_distribution/bin_centers"][:]
distribution = f2["readdy/observables/radial_distribution/distribution"][:]
for t in range(n_time_steps):
np.testing.assert_equal(bin_centers, np.array(callback_centers[t]))
np.testing.assert_equal(distribution[t], np.array(callback_rdf[t]))
def run(self, time_steps, out_file):
sim = Simulation()
sim.set_kernel(self.kernel)
sim.box_size = common.Vec(60, 20, 20)
sim.periodic_boundary = [True, True, True]
typeid_b = sim.register_particle_type("B", 1.0, 1.0,
ParticleTypeFlavor.NORMAL)
sim.register_particle_type("Topology A", .5, .5,
ParticleTypeFlavor.TOPOLOGY)
sim.register_potential_harmonic_repulsion("Topology A", "Topology A",
10)
sim.register_potential_harmonic_repulsion("Topology A", "B", 10)
sim.register_potential_harmonic_repulsion("B", "B", 10)
sim.configure_topology_bond_potential("Topology A", "Topology A", 10,
1.)
sim.configure_topology_angle_potential("Topology A", "Topology A",
"Topology A", 10, np.pi)
# sim.configure_topology_dihedral_potential("Topology A", "Topology A", "Topology A", "Topology A", 1, 1, -np.pi)
n_elements = 50.
particles = [
sim.create_topology_particle("Topology A",
common.Vec(-25. + i, 0, 0))
for i in range(int(n_elements))
]
topology = sim.add_topology(particles)
for i in range(int(n_elements - 1)):
topology.get_graph().add_edge(i, i + 1)
topology.add_reaction(self._get_decay_reaction(typeid_b))
topology.add_reaction(self._get_split_reaction())
traj_handle = sim.register_observable_flat_trajectory(1)
with closing(
io.File(out_file, io.FileAction.CREATE,
io.FileFlag.OVERWRITE)) as f:
traj_handle.enable_write_to_file(f, u"", 50)
sim.run_scheme_readdy(True)\
.evaluate_topology_reactions()\
.write_config_to_file(f)\
.configure_and_run(time_steps, self.time_step)
print("currently %s topologies" % len(sim.current_topologies()))
def run(self, time_steps, out_file):
sim = Simulation()
sim.set_kernel(self.kernel)
sim.box_size = common.Vec(60, 20, 20)
sim.periodic_boundary = [True, True, True]
typeid_b = sim.register_particle_type("B", 1.0, 1.0, ParticleTypeFlavor.NORMAL)
sim.register_particle_type("Topology A", .5, .5, ParticleTypeFlavor.TOPOLOGY)
sim.register_potential_harmonic_repulsion("Topology A", "Topology A", 10)
sim.register_potential_harmonic_repulsion("Topology A", "B", 10)
sim.register_potential_harmonic_repulsion("B", "B", 10)
sim.configure_topology_bond_potential("Topology A", "Topology A", 10, 1.)
sim.configure_topology_angle_potential("Topology A", "Topology A", "Topology A", 10, np.pi)
# sim.configure_topology_dihedral_potential("Topology A", "Topology A", "Topology A", "Topology A", 1, 1, -np.pi)
n_elements = 50.
particles = [sim.create_topology_particle("Topology A", common.Vec(-25. + i, 0, 0))
for i in range(int(n_elements))]
topology = sim.add_topology(particles)
for i in range(int(n_elements - 1)):
topology.get_graph().add_edge(i, i + 1)
topology.add_reaction(self._get_decay_reaction(typeid_b))
topology.add_reaction(self._get_split_reaction())
traj_handle = sim.register_observable_flat_trajectory(1)
with closing(io.File(out_file, io.FileAction.CREATE, io.FileFlag.OVERWRITE)) as f:
traj_handle.enable_write_to_file(f, u"", 50)
sim.run_scheme_readdy(True)\
.evaluate_topology_reactions()\
.write_config_to_file(f)\
.configure_and_run(time_steps, self.time_step)
print("currently %s topologies" % len(sim.current_topologies()))
n_calls += 1
if n_calls % 10000 == 0:
print("%s" % (10. * float(n_calls) / float(T)))
if __name__ == '__main__':
KernelProvider.get().load_from_dir(platform_utils.get_readdy_plugin_dir())
simulation = Simulation()
simulation.set_kernel("CPU")
box_size = Vec(10, 10, 10)
simulation.kbt = 2
simulation.periodic_boundary = [True, True, True]
simulation.box_size = box_size
simulation.register_particle_type("A", .2, 1.)
simulation.register_particle_type("B", .2, 1.)
simulation.register_potential_harmonic_repulsion("A", "B", 10)
simulation.add_particle("A", Vec(-2.5, 0, 0))
simulation.add_particle("B", Vec(0, 0, 0))
simulation.register_observable_radial_distribution(
10, np.arange(0, 5, .01), ["A"], ["B"],
1. / (box_size[0] * box_size[1] * box_size[2]), rdf_callback)
simulation.run(T, 0.02)
print("n_calls=%s" % n_calls)
fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(centers, rdf / n_calls)
plt.show()