92_nut_in_bolt.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 "gen_wall_box_plate.hpp"
#include "gravity.hpp"
#include "model_linear_spring.hpp"
#include "model_volume_based.hpp"
#include "shape_sphere.hpp"
#include "shape_trimesh.hpp"
#include "simulation.hpp"
#include "solver_sdf_pw.hpp"
#include <filesystem>
#include <iostream>
#include <unordered_map>
using namespace netdem;
using namespace std;
 
void NutInBolt(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.3, 0.3, 0.3);
 
  sim->dem_solver.contact_solver_factory.settings.sdf_potential_type = 1;
  LinearSpring contact_model = LinearSpring(2.0e3, 1.0e4, 0.0, 0.0);
  sim->scene.InsertContactModel(&contact_model);
  sim->scene.SetNumberOfMaterials(1);
  sim->scene.SetCollisionModel(0, 0, contact_model.label);
 
  TriMesh tri_mesh;
  tri_mesh.InitFromSTL("data/copyleft/bolt_nut/bolt.stl");
  tri_mesh.AlignAxes();
  double scale = 0.3 / tri_mesh.GetSize();
  tri_mesh.SetSize(tri_mesh.GetSize() * scale);
  sim->scene.InsertShape(&tri_mesh);
 
  tri_mesh.InitFromSTL("data/copyleft/bolt_nut/nut.stl");
  tri_mesh.AlignAxes();
  tri_mesh.SetSize(tri_mesh.GetSize() * scale);
  sim->scene.InsertShape(&tri_mesh);
 
  Wall bolt(sim->scene.GetShapes()[1]);
  bolt.SetRodrigues(Math::PI / 2, 0.0, 1.0, 0.0);
 
  SolverSDFPW cnt_solver;
  Particle nut(sim->scene.GetShapes()[0]);
  nut.SetDensity(7850);
  double pos_z_2 = 0.2;
  for (int i = 0; i < 10000; i++) {
    nut.SetPosition(0.0, 0.0, pos_z_2);
    cnt_solver.Init(&nut, &bolt);
    if (cnt_solver.Detect()) {
      pos_z_2 += 0.00001;
    } else {
      cout << "found no contact postion for the nut" << endl;
      break;
    }
  }
 
  sim->scene.InsertWall(bolt);
  sim->scene.InsertParticle(nut);
 
  WallBoxPlane wall_box(1, 1, 1, 0, 0, 0);
  wall_box.ImportToScene(&(sim->scene));
 
  Gravity grav;
  grav.Init(sim);
  sim->modifier_manager.Insert(&grav);
  sim->modifier_manager.Enable(grav.label);
 
  string root_dir = "tmp/examples/sdf_dem/nut_in_bolt/";
  filesystem::create_directories(root_dir);
 
  DataDumper data_dumper;
  data_dumper.Init(sim);
  data_dumper.SetRootPath(root_dir);
  data_dumper.SetSaveByCycles(100);
  data_dumper.SaveShapeInfoAsSTL();
  data_dumper.dump_contact_info = true;
  sim->modifier_manager.Insert(&data_dumper);
  sim->modifier_manager.Enable(data_dumper.label);
 
  sim->dem_solver.timestep = 1.0e-4;
  sim->Run(6.0);
 
  delete sim;
}