shape_polybezier.hpp

Go to the documentation of this file.

#pragma once
 
#include "shape.hpp"
#include "stl_model.hpp"
#include "stl_reader.hpp"
#include <iostream>
#include <string>
 
namespace netdem {
 
class Polybezier : public Shape {
public:
  // num_patches have to be even
  int num_patches{30}, order{2}; // only second order implemented for now
 
  // [num of faces] by num of knots (e.g., 6 for 2nd order) by 3
  VecXT<VecXT<Vec3d>> face_patch_knots_list;
 
  // [num of faces] by 3 vertices by 3
  VecXT<VecXT<Vec3d>> face_patch_normals_list;
 
  // [num of faces*3/2] by num of knots (e.g., 6 for 2nd order) by 3
  VecXT<VecXT<Vec3d>> edge_patch_knots_list;
 
  // [num of faces] by 3, used to find the edge patches for a face patch
  VecXT<Vec3i> linked_edges_list;
 
  // [6*num_cells^2] by [num of patches]
  VecXT<VecXT<int>> linked_patches_list;
  int num_cells{4};
 
  Polybezier();
 
  nlohmann::json PackJson() override;
 
  void InitFromJson(nlohmann::json const &js) override;
 
  Shape *Clone() const override;
 
  void InitByRandom();
 
  void InitFromKernelSTL(std::string const &file);
 
  void InitFromKernelSTL(STLModel const &stl_model);
 
  void Init();
 
  void UpdateShapeProperties() override;
 
  void SetSize(double d) override;
 
  STLModel GetSTLModel(int num_nodes = 200) override;
 
  STLModel GetSTLModel_v2(int num_nodes = 200);
 
  void SaveNormalPatchesSpherical(std::string const &file);
 
  void SaveNormalPatchesCubic(std::string const &file);
 
  Vec3d SupportPoint(Vec3d const &dir) override;
 
  VecXT<Vec3d> SupportPoints(Vec3d const &dir) override;
 
  void Print() override;
 
private:
  void AlignAxes();
 
  void UpdateLinkedPatches();
 
  VecXT<VecNT<Vec3d, 3>> mat_du_list, mat_dv_list;
 
  void UpdataMatDuDv();
 
  Vec3d GetEdgeKnot(Vec3d const &v0, Vec3d const &v1, Vec3d const &n01);
 
  void GetUniqueEdges(VecXT<Vec2i> *const edges,
                      VecXT<Vec3i> *const linked_list,
                      STLModel const &stl_model);
 
  Vec3d GetPatchNormal(Vec3d const &v0, Vec3d const &v1, Vec3d const &v2);
 
  void SortNormalPatchVertices(VecXT<Vec3d> *const normals);
 
  VecXT<Vec3d> GetCartesianProject(const VecXT<Vec3d> &normals);
 
  VecXT<Vec3d> GetCartesianProject(Vec3d const &v1, Vec3d const &v2);
 
  bool ContainCorner(Vec3d const &corner, const VecXT<Vec3d> &normals);
 
  STLModel GetSTLModel(const VecXT<Vec3d> &knots, int num_nodes_each);
 
  int GetSupportPatchID(Vec3d const &dir);
};
 
} // namespace netdem