0_plane_case.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_sphere.hpp"
#include "simulation.hpp"
#include "wall_motion_control.hpp"
#include "wall_servo_control.hpp"
#include <iostream>
#include <unordered_map>
using namespace netdem;
using namespace std;
 
int PlaneCase(int argc, char **argv) {
  int self_rank, total_rank;
  MPI_Init(&argc, &argv);
  MPI_Comm_rank(MPI_COMM_WORLD, &self_rank);
  MPI_Comm_size(MPI_COMM_WORLD, &total_rank);
 
  Simulation *sim = new Simulation();
 
  sim->domain_manager.SetBound(-0.6, -0.6, -1.2, 0.6, 0.6, 1.2);
  sim->domain_manager.SetDecomposition(total_rank, 1, 1);
  sim->domain_manager.SetCellSpacing(0.3, 0.3, 0.3);
 
  LinearSpring cnt_model = LinearSpring(2.0e6, 1.0e6, 0.5, 0.0);
  sim->scene.InsertContactModel(&cnt_model);
  sim->scene.SetNumberOfMaterials(1);
  sim->scene.SetCollisionModel(0, 0, cnt_model.label);
 
  Sphere sphere_1 = Sphere(0.1);
  Sphere sphere_2 = Sphere(0.1);
  sim->scene.InsertShape(&sphere_1);
  sim->scene.InsertShape(&sphere_2);
 
  VecXT<Particle> particle_list = PackGenerator::GetGridPack(
      1, 1, 2, 0, 0, 0, 9, 9, 18, sim->scene.GetShapes());
  for (auto &p : particle_list) {
    p.damp_numerical = 0.7;
  }
  sim->scene.InsertParticle(particle_list);
 
  WallBoxPlane wall_box(1, 1, 2, 0, 0, 0);
  wall_box.ImportToScene(&(sim->scene));
 
  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);
 
  // servo control for top wall
  WallServoControl servo_top(2e6, 1.0);
  servo_top.label = "servo_top";
  servo_top.enable_auto_area = true;
  servo_top.Init(sim);
  servo_top.target_pressure = 1e2;
  servo_top.SetWall({5});
  auto tmp = sim->modifier_manager.Insert(&servo_top);
  sim->modifier_manager.Enable(servo_top.label);
  auto servo_top_ptr = static_cast<WallServoControl *>(tmp);
 
  sim->Run(2);
 
  // servo control for lateral walls
  servo_top_ptr->target_pressure = 1e5;
 
  WallServoControl servo_lateral(2e6, 1.0);
  servo_lateral.label = "servo_lateral";
  servo_lateral.enable_auto_area = true;
  servo_lateral.Init(sim);
  servo_lateral.target_pressure = 1e5;
  servo_lateral.SetWall({0, 1, 2, 3});
  tmp = sim->modifier_manager.Insert(&servo_lateral);
  sim->modifier_manager.Enable(servo_lateral.label);
  auto servo_lateral_ptr = static_cast<WallServoControl *>(tmp);
 
  // keep loading until target pressure is achieved
  do {
    sim->Run(1);
  } while (!servo_top_ptr->achieved | !servo_lateral.achieved);
 
  // disable servo control of the top wall
  sim->modifier_manager.Disable(servo_top.label);
  servo_lateral_ptr->enable_warning = true;
 
  // add deformation drived loading for top wall
  WallMotionControl disp_control;
  disp_control.Init(sim);
  disp_control.SetWall({5});
  disp_control.SetVelocity(0, 0, -0.1);
  sim->modifier_manager.Insert(&disp_control);
  sim->modifier_manager.Enable(disp_control.label);
 
  sim->Run(2);
 
  delete sim;
 
  MPI_Finalize();
  return 0;
}