94_angle_repose_trimesh.cpp

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

#include "contact_solver_factory.hpp"
#include "data_dumper.hpp"
#include "gen_pack.hpp"
#include "gen_wall_box_plane.hpp"
#include "gravity.hpp"
#include "model_linear_spring.hpp"
#include "shape_plane.hpp"
#include "shape_spherical_harmonics.hpp"
#include "simulation.hpp"
#include "stl_model.hpp"
#include "wall_motion_control.hpp"
#include <iostream>
#include <unordered_map>
using namespace netdem;
using namespace std;
 
void AngleRepose(bool need_make_convex, double friction, string root_dir) {
  // new simulation
  Simulation *sim = new Simulation();
 
  // domain settings
  sim->domain_manager.SetBound(-1, -1, -1, 1, 1, 1);
  auto domain = sim->domain_manager.GetSelfSubDomain();
  domain->cell_manager.SetSpacing(0.02, 0.02, 0.02);
 
  // define contact model
  LinearSpring cnt_model = LinearSpring(1.0e2, 1.0e4, 0.7, friction);
  sim->scene.InsertContactModel(&cnt_model);
  sim->scene.SetNumberOfMaterials(1);
  sim->scene.SetCollisionModel(0, 0, cnt_model.label);
 
  // shape template
  STLModel stl_model;
  stl_model.InitFromSTL("data/mms_soil.stl");
  if (need_make_convex) {
    stl_model.MakeConvex();
    stl_model.SmoothMesh(3);
    stl_model.Refine(2);
  }
  SphericalHarmonics sh;
  sh.InitFromSTL(stl_model);
  sh.SetSize(0.005);
  sim->scene.InsertShape(&sh);
 
  // get a pack of particles
  VecXT<Particle> particle_list = PackGenerator::GetGridPack(
      0.04, 0.04, 0.04, 0, 0, 0.10, 5, 5, 5, sim->scene.GetShapes());
  for (auto &p : particle_list) {
    p.SetDensity(265000);
    p.damp_numerical = 0.7;
  }
 
  // get a box of wall with plane shapes, and insert the wall and shapes into
  // the scene
  WallBoxPlane wall_box(0.04, 0.04, 0.16, 0, 0, 0.04);
  wall_box.ImportToScene(&(sim->scene));
 
  // add a gravity modifier
  Gravity grav;
  grav.Init(sim);
  sim->modifier_manager.Insert(&grav);
  sim->modifier_manager.Enable(grav.label);
 
  // add a data dumper modifier
  DataDumper data_dumper;
  data_dumper.Init(sim);
  data_dumper.SetRootPath(root_dir);
  data_dumper.SetSaveByCycles(1000);
  data_dumper.SaveShapeInfoAsSTL();
  data_dumper.dump_wall_info = true;
  data_dumper.dump_contact_info = true;
  sim->modifier_manager.Insert(&data_dumper);
  sim->modifier_manager.Enable(data_dumper.label);
 
  // random packing
  sim->dem_solver.timestep = 1e-4;
  for (int i = 0; i < 10; i++) {
    sim->scene.InsertParticle(particle_list);
    sim->Run(0.1);
  }
  sim->scene.gravity_coef[2] = -0.0981;
  sim->Run(0.5);
 
  // remove the side and top walls
  Plane *plane_ptr;
 
  plane_ptr = static_cast<Plane *>(sim->scene.wall_list[2]->shape);
  plane_ptr->SetExtent(1.0);
  sim->scene.wall_list[2]->UpdateBound();
 
  plane_ptr = static_cast<Plane *>(sim->scene.wall_list[3]->shape);
  plane_ptr->SetExtent(1.0);
  sim->scene.wall_list[3]->UpdateBound();
 
  plane_ptr = static_cast<Plane *>(sim->scene.wall_list[4]->shape);
  plane_ptr->SetExtent(1.0);
  sim->scene.wall_list[4]->UpdateBound();
 
  sim->scene.RemoveWall(sim->scene.wall_list[1]);
  sim->scene.RemoveWall(sim->scene.wall_list[5]);
 
  // reposing process
  sim->Run(10);
 
  delete sim;
 
  MPI_Finalize();
}