02_ev_morphology.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 "shape_polybezier.hpp"
#include "shape_sphere.hpp"
#include "shape_trimesh.hpp"
#include "simulation.hpp"
#include <iostream>
#include <unordered_map>
 
using namespace netdem;
using namespace std;
 
void EvaluateMorphology(int argc, char **argv) {
  int num_particles = atof(argv[2]);
  int num_patches = atof(argv[3]);
 
  stringbuf buf;
  ostream os(&buf);
  int os_width = 24;
  os.setf(ios::scientific);
  os.precision(15);
 
  for (int i = 0; i < num_particles; i++) {
    STLModel tmp_stl;
    tmp_stl.InitFromSTL("local/polybezier_model/random_generation/n" +
                        to_string(num_patches) + "/p" + to_string(i) + ".stl");
    cout << "local/polybezier_model/n" + to_string(num_patches) + "/p" +
                to_string(i) + ".stl"
         << endl;
 
    tmp_stl.Standardize();
 
    auto [bound_min, bound_max] = tmp_stl.GetBoundAABB();
    auto bound_len = bound_max - bound_min;
    auto len_max = max(bound_len[0], max(bound_len[1], bound_len[2]));
    auto len_min = min(bound_len[0], min(bound_len[1], bound_len[2]));
    auto len_mid =
        bound_len[0] + bound_len[1] + bound_len[2] - len_max - len_min;
 
    auto flatness = len_min / len_mid;
    auto elongation = len_mid / len_max;
    auto sphericity =
        pow(36.0 * Math::PI * tmp_stl.GetVolume() * tmp_stl.GetVolume(),
            1.0 / 3.0) /
        tmp_stl.GetSurfaceArea();
 
    os.width(os_width);
    os << flatness << ", ";
 
    os.width(os_width);
    os << elongation << ", ";
 
    os.width(os_width);
    os << sphericity << endl;
  }
 
  ofstream outfile;
  auto filename = "local/polybezier_model/random_generation/n" +
                  to_string(num_patches) + ".txt";
  outfile.open(filename);
  if (!outfile) {
    cout << "cannot open file: " << filename << endl;
  }
  outfile << buf.str();
  outfile.close();
  cout << "data saved to: " << filename << endl;
}