blade_mixing_main.cpp

This is an example of how to use the netdem library.

#include "data_dumper.hpp"
#include "gen_pack.hpp"
#include "gen_wall_box_plane.hpp"
#include "gravity.hpp"
#include "model_linear_spring.hpp"
#include "shape_cylinder.hpp"
#include "shape_sphere.hpp"
#include "shape_triangle.hpp"
#include "simulation.hpp"
#include <iostream>
#include <unordered_map>
using namespace netdem;
using namespace std;
 
void ImportWall(Simulation *sim) {
  STLModel cylinder_stl;
  cylinder_stl.InitFromSTL("data/cylinder.stl");
  for (auto &vert : cylinder_stl.vertices) {
    vert[2] *= 0.5;
  }
  cylinder_stl.SaveAsSTL("tmp/tests/cylinder_outer.stl");
  for (auto &facet : cylinder_stl.facets) {
    Triangle triangle(cylinder_stl.vertices[facet[0]],
                      cylinder_stl.vertices[facet[1]],
                      cylinder_stl.vertices[facet[2]]);
    auto shape_ptr = sim->scene.InsertShape(&triangle);
    Wall wall(shape_ptr);
    sim->scene.InsertWall(wall);
  }
 
  for (auto &vert : cylinder_stl.vertices) {
    vert[0] *= 0.15;
    vert[1] *= 0.15;
    vert[2] *= 0.64;
    vert[2] -= 0.10;
  }
  cylinder_stl.SaveAsSTL("tmp/tests/cylinder_inner.stl");
  for (auto &facet : cylinder_stl.facets) {
    Triangle triangle(cylinder_stl.vertices[facet[0]],
                      cylinder_stl.vertices[facet[1]],
                      cylinder_stl.vertices[facet[2]]);
    auto shape_ptr = sim->scene.InsertShape(&triangle);
    Wall wall(shape_ptr);
    sim->scene.InsertWall(wall);
  }
 
  STLModel propeller_stl;
  propeller_stl.InitFromSTL("data/copyleft/propeller.stl");
  propeller_stl.Standardize();
  for (auto &vert : propeller_stl.vertices) {
    vert[0] *= 0.14;
    vert[1] *= 0.14;
    vert[2] *= 0.8;
    vert[2] -= 0.075;
  }
  propeller_stl.SaveAsSTL("tmp/tests/propeller.stl");
  for (auto &facet : propeller_stl.facets) {
    Triangle triangle(propeller_stl.vertices[facet[0]],
                      propeller_stl.vertices[facet[1]],
                      propeller_stl.vertices[facet[2]]);
    triangle.SetSkinFactor(0.5);
    auto shape_ptr = sim->scene.InsertShape(&triangle);
    Wall wall;
    wall.SetShape(shape_ptr);
    wall.SetVelocitySpin(0, 0, -2.0 * Math::PI);
    auto wall_ptr = sim->scene.InsertWall(wall);
 
    auto [bound_aabb_min, bound_aabb_max] = shape_ptr->GetBoundAABB();
    wall_ptr->pos = 0.5 * (bound_aabb_min + bound_aabb_max);
    shape_ptr->Translate(-1.0 * wall_ptr->pos);
  }
}
 
int main(int argc, char **argv) {
  Simulation *sim = new Simulation();
 
  sim->domain_manager.SetBound(-0.6, -0.6, -0.6, 0.6, 0.6, 0.6);
  sim->domain_manager.SetCellSpacing(0.2, 0.2, 0.2);
 
  LinearSpring cnt_model = LinearSpring(1.0e4, 1.0e4, 0.5, 0.0);
  sim->scene.InsertContactModel(&cnt_model);
  sim->scene.SetNumberOfMaterials(1);
  sim->scene.SetCollisionModel(0, 0, cnt_model.label);
 
  Sphere sphere = Sphere(0.02);
  auto shape_ptr = sim->scene.InsertShape(&sphere);
 
  VecXT<Particle> particle_list = PackGenerator::GetGridPack(
      0.7, 0.7, 0.1, 0, 0, 0.2, 25, 20, 2, sim->scene.GetShapes());
  for (auto &p : particle_list) {
    p.SetVelocity(0, 0, -2.0);
    p.damp_numerical = 0.7;
  }
 
  ImportWall(sim);
 
  Gravity grav;
  grav.Init(sim);
  sim->modifier_manager.Insert(&grav);
  sim->modifier_manager.Enable(grav.label);
 
  DataDumper data_dumper;
  data_dumper.Init(sim);
  data_dumper.SetRootPath("tmp/out/");
  data_dumper.SetSaveByCycles(100);
  data_dumper.dump_wall_info = true;
  data_dumper.dump_shape_info = true;
  sim->modifier_manager.Insert(&data_dumper);
  sim->modifier_manager.Enable(data_dumper.label);
 
  for (int i = 0; i < 20; i++) {
    sim->scene.InsertParticle(particle_list);
    sim->Run(0.1);
  }
  sim->Run(2.0);
 
  delete sim;
}