Skip to content
Snippets Groups Projects
Commit aa66a723 authored by Chichi Lalescu's avatar Chichi Lalescu
Browse files

run basic interpolation test

parent 69b6b36d
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
bfps/base.py
+ 1
− 0
View file @ aa66a723
...@@ -25,6 +25,7 @@ ...@@ -25,6 +25,7 @@
import os import os
import numpy as np
import h5py import h5py
import bfps import bfps
......
tests/test_interpolation.py
+ 127
− 1
View file @ aa66a723
...@@ -27,6 +27,7 @@ ...@@ -27,6 +27,7 @@
import numpy as np import numpy as np
from test_base import * from test_base import *
import matplotlib.pyplot as plt
class test_interpolation(bfps.NavierStokes): class test_interpolation(bfps.NavierStokes):
def __init__( def __init__(
...@@ -39,6 +40,121 @@ class test_interpolation(bfps.NavierStokes): ...@@ -39,6 +40,121 @@ class test_interpolation(bfps.NavierStokes):
simname = simname, simname = simname,
name = name) name = name)
return None return None
def add_particles(
self,
neighbours = 1,
smoothness = 1,
integration_steps = 2,
kcut = 'fs->kM'):
self.parameters['neighbours{0}'.format(self.particle_species)] = neighbours
self.parameters['smoothness{0}'.format(self.particle_species)] = smoothness
self.parameters['kcut{0}'.format(self.particle_species)] = kcut
self.parameters['integration_steps{0}'.format(self.particle_species)] = integration_steps
self.particle_variables += 'tracers<{0}> *ps{1};\n'.format(self.C_dtype, self.particle_species)
grow_template = """
//begincpp
temp_string = (std::string("/particles/") +
std::string(ps{0}->name) +
std::string("/{1}"));
Cdset = H5Dopen(stat_file, temp_string.c_str(), H5P_DEFAULT);
Cspace = H5Dget_space(Cdset);
ndims = H5Sget_simple_extent_dims(Cspace, dims, NULL);
dims[0] += niter_todo/niter_part;
H5Dset_extent(Cdset, dims);
H5Sclose(Cspace);
H5Dclose(Cdset);
//endcpp
"""
self.file_datasets_grow += grow_template.format(self.particle_species, 'state')
self.file_datasets_grow += grow_template.format(self.particle_species, 'rhs')
self.file_datasets_grow += grow_template.format(self.particle_species, 'velocity')
self.file_datasets_grow += grow_template.format(self.particle_species, 'acceleration')
#self.particle_definitions
if kcut == 'fs->kM':
if self.particle_species == 0:
update_field = ('fs->compute_velocity(fs->cvorticity);\n' +
'fs->ift_velocity();\n')
else:
update_field = ''
else:
update_field = ('fs->compute_velocity(fs->cvorticity);\n' +
'fs->low_pass_Fourier(fs->cvelocity, 3, {0});\n'.format(kcut) +
'fs->ift_velocity();\n')
update_field += 'ps{0}->update_field();\n'.format(self.particle_species)
if self.dtype == np.float32:
FFTW = 'fftwf'
elif self.dtype == np.float64:
FFTW = 'fftw'
compute_acc = ('{0} *acc_field = {1}_alloc_real(ps{2}->buffered_field_descriptor->local_size);\n' +
'{0} *acc_field_tmp = {1}_alloc_real(fs->rd->local_size);\n' +
'fs->compute_Lagrangian_acceleration(acc_field_tmp);\n' +
'ps{2}->rFFTW_to_buffered(acc_field_tmp, acc_field);\n' +
'ps{2}->sample_vec_field(acc_field, acceleration);\n' +
'{1}_free(acc_field_tmp);\n' +
'{1}_free(acc_field);\n').format(self.C_dtype, FFTW, self.particle_species)
output_vel_acc = """
//begincpp
{{
double *acceleration = new double[ps{0}->array_size];
double *velocity = new double[ps{0}->array_size];
{1}
ps{0}->sample_vec_field(ps{0}->data, velocity);
{2}
if (ps{0}->fs->rd->myrank == 0)
{{
//VELOCITY begin
std::string temp_string = (std::string("/particles/") +
std::string(ps{0}->name) +
std::string("/velocity"));
hid_t Cdset = H5Dopen(stat_file, temp_string.c_str(), H5P_DEFAULT);
hid_t mspace, wspace;
int ndims;
hsize_t count[3], offset[3];
wspace = H5Dget_space(Cdset);
ndims = H5Sget_simple_extent_dims(wspace, count, NULL);
count[0] = 1;
offset[0] = ps{0}->iteration / ps{0}->traj_skip;
offset[1] = 0;
offset[2] = 0;
mspace = H5Screate_simple(ndims, count, NULL);
H5Sselect_hyperslab(wspace, H5S_SELECT_SET, offset, NULL, count, NULL);
H5Dwrite(Cdset, H5T_NATIVE_DOUBLE, mspace, wspace, H5P_DEFAULT, velocity);
H5Dclose(Cdset);
//VELOCITY end
//ACCELERATION begin
temp_string = (std::string("/particles/") +
std::string(ps{0}->name) +
std::string("/acceleration"));
Cdset = H5Dopen(stat_file, temp_string.c_str(), H5P_DEFAULT);
H5Dwrite(Cdset, H5T_NATIVE_DOUBLE, mspace, wspace, H5P_DEFAULT, acceleration);
H5Sclose(mspace);
H5Sclose(wspace);
H5Dclose(Cdset);
//ACCELERATION end
}}
delete[] acceleration;
delete[] velocity;
}}
//endcpp
""".format(self.particle_species, update_field, compute_acc)
self.particle_start += ('sprintf(fname, "tracers{1}");\n' +
'ps{1} = new tracers<{0}>(\n' +
'fname, fs,\n' +
'nparticles,\n' +
'neighbours{1}, smoothness{1}, niter_part, integration_steps{1},\n' +
'fs->ru);\n' +
'ps{1}->dt = dt;\n' +
'ps{1}->iteration = iteration;\n' +
update_field +
'ps{1}->read(stat_file);\n').format(self.C_dtype, self.particle_species)
self.particle_loop += ((update_field +
'if (ps{0}->iteration % niter_part == 0)\n' +
'ps{0}->write(stat_file, false);\n').format(self.particle_species) +
output_vel_acc)
self.particle_end += ('ps{0}->write(stat_file);\n' +
'delete ps{0};\n').format(self.particle_species)
self.particle_species += 1
return None
if __name__ == '__main__': if __name__ == '__main__':
opt = parser.parse_args() opt = parser.parse_args()
...@@ -46,7 +162,7 @@ if __name__ == '__main__': ...@@ -46,7 +162,7 @@ if __name__ == '__main__':
c.pars_from_namespace(opt) c.pars_from_namespace(opt)
c.add_particles( c.add_particles(
integration_steps = 1, integration_steps = 1,
neighbours = 1, neighbours = 4,
smoothness = 1) smoothness = 1)
c.fill_up_fluid_code() c.fill_up_fluid_code()
c.finalize_code() c.finalize_code()
...@@ -54,6 +170,7 @@ if __name__ == '__main__': ...@@ -54,6 +170,7 @@ if __name__ == '__main__':
c.write_par() c.write_par()
pos = np.zeros((c.parameters['nparticles'], 3), dtype = np.float64) pos = np.zeros((c.parameters['nparticles'], 3), dtype = np.float64)
pos[:, 0] = np.linspace(0, 2*np.pi, pos.shape[0]) pos[:, 0] = np.linspace(0, 2*np.pi, pos.shape[0])
c.generate_vector_field(write_to_file = True, spectra_slope = 5./3)
c.generate_tracer_state( c.generate_tracer_state(
species = 0, species = 0,
write_to_file = False, write_to_file = False,
...@@ -61,4 +178,13 @@ if __name__ == '__main__': ...@@ -61,4 +178,13 @@ if __name__ == '__main__':
c.set_host_info({'type' : 'pc'}) c.set_host_info({'type' : 'pc'})
if opt.run: if opt.run:
c.run(ncpu = opt.ncpu) c.run(ncpu = opt.ncpu)
df = c.get_data_file()
x = c.get_coord('x')
fig = plt.figure()
a = fig.add_subplot(111)
a.plot(x, df['statistics/xlines/velocity'][1])
a.plot(df['particles/tracers0/state'][0, :, 0],
df['particles/tracers0/velocity'][0],
dashes = (1, 1))
fig.savefig('test.pdf')
tox_interpolation.ini
+ 3
− 2
View file @ aa66a723
...@@ -12,9 +12,10 @@ commands = ...@@ -12,9 +12,10 @@ commands =
python tests/test_interpolation.py \ python tests/test_interpolation.py \
-n 32 --ncpu 2 \ -n 32 --ncpu 2 \
--initialize \ --initialize \
--nparticles 17 \ --nparticles 65 \
--niter_todo 1 \ --niter_todo 1 \
--niter_stat 1 \ --niter_stat 1 \
--niter_part 1 \ --niter_part 1 \
--niter_out 1 --niter_out 1 \
--run
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment