solver_ann_pplane.hpp

Go to the documentation of this file.

#pragma once
 
#include "collision_geometries.hpp"
#include "collision_solver_pw.hpp"
#include "contact_model_factory.hpp"
#include "general_net.hpp"
#include "particle.hpp"
#include "regression_net.hpp"
#include "shape.hpp"
#include "shape_plane.hpp"
#include "wall.hpp"
#include <string>
 
namespace netdem {
 
class SolverANNPPlane : public CollisionSolverPW {
public:
  netdem::GeneralNet classifier;
 
  netdem::RegressionNet regressor;
 
  SolverANNPPlane();
 
  SolverANNPPlane(Particle *const p, Wall *const w);
 
  CollisionSolverPW *Clone() const override;
  void Init(std::string const &classifier_file,
            std::string const &regressor_file);
  void Init(Particle *const p, Wall *const w) override;
  bool Detect() override;
 
  void ResolveInit(ContactPW *const cnt, double timestep) override;
  void ResolveUpdate(ContactPW *const cnt, double timestep) override;
 
  void ResolveInit_LinearSpring(CollisionGeometries *const cnt_geoms,
                                double timestep);
  void ResolveUpdate_LinearSpring(CollisionGeometries *const cnt_geoms,
                                  double timestep);
  void ResolveInit_VolumeBased(CollisionGeometries *const cnt_geoms,
                               double timestep);
  void ResolveUpdate_VolumeBased(CollisionGeometries *const cnt_geoms,
                                 double timestep);
  void ResolveInit_PotentialBased(CollisionGeometries *const cnt_geoms,
                                  double timestep);
 
  void ResolveUpdate_PotentialBased(CollisionGeometries *const cnt_geoms,
                                    double timestep);
  // return: potential and cnt_pos
  std::tuple<double, Vec3d> GetPotential(double dist, Vec3d const &nn);
  // return: force, moment and pos
  std::tuple<double, Vec3d, Vec3d> EvaluateContactForces();
 
private:
  double bound_sphere_radius_1, dist_pc_to_plane, scale;
  Vec3d dir_n, dir_n_ref;
};
 
} // namespace netdem