diff --git a/examples/dem.py b/examples/dem.py index c035c6f2b3aab348efff6ef136ae7d48e1a3d4e3..b6123d86ecbdd202a9ab03e831ef2eab8a070cea 100644 --- a/examples/dem.py +++ b/examples/dem.py @@ -86,7 +86,7 @@ def euler(i): def gravity(i): volume = (4.0 / 3.0) * pi * radius[i] * radius[i] * radius[i] - force[i][2] -= densityParticle_SI - densityFluid_SI * volume * gravity_SI + force[i][2] += -(densityParticle_SI - densityFluid_SI) * volume * gravity_SI cmd = sys.argv[0] diff --git a/runtime/vtk.hpp b/runtime/vtk.hpp index ddfb659d915cdb299d4fbffa98567e13ae40eba2..349b66c01e1bc8ee199452bece8e703c8894ec74 100644 --- a/runtime/vtk.hpp +++ b/runtime/vtk.hpp @@ -13,6 +13,7 @@ void vtk_write_data(PairsSimulation *ps, const char *filename, int start, int en auto masses = ps->getAsFloatProperty(ps->getPropertyByName("mass")); auto positions = ps->getAsVectorProperty(ps->getPropertyByName("position")); const int n = end - start; + const int prec = 8; std::ostringstream filename_oss; if(frequency != 0 && timestep % frequency != 0) { @@ -38,9 +39,9 @@ void vtk_write_data(PairsSimulation *ps, const char *filename, int start, int en out_file << "POINTS " << n << " double\n"; for(int i = start; i < end; i++) { - out_file << std::fixed << std::setprecision(4) << positions(i, 0) << " "; - out_file << std::fixed << std::setprecision(4) << positions(i, 1) << " "; - out_file << std::fixed << std::setprecision(4) << positions(i, 2) << "\n"; + out_file << std::fixed << std::setprecision(prec) << positions(i, 0) << " "; + out_file << std::fixed << std::setprecision(prec) << positions(i, 1) << " "; + out_file << std::fixed << std::setprecision(prec) << positions(i, 2) << "\n"; } out_file << "\n\n"; @@ -60,7 +61,7 @@ void vtk_write_data(PairsSimulation *ps, const char *filename, int start, int en out_file << "SCALARS mass double\n"; out_file << "LOOKUP_TABLE default\n"; for(int i = start; i < end; i++) { - out_file << std::fixed << std::setprecision(4) << masses(i) << "\n"; + out_file << std::fixed << std::setprecision(prec) << masses(i) << "\n"; } out_file << "\n\n";