shape_netsdf.hpp

Go to the documentation of this file.

#pragma once
 
#include "igl_wrapper.hpp"
#include "regression_net.hpp"
#include "sdf_calculator.hpp"
#include "shape.hpp"
#include "stl_model.hpp"
#include "stl_reader.hpp"
#include <iostream>
#include <string>
 
namespace netdem {
 
class NetSDF : public Shape {
public:
  // ann for netsdf
  RegressionNet net_sdf_model;
  RegressionNet net_spf_model;
 
  // for reconstructing surface from point cloud
  double alpha{0.7};
 
  NetSDF();
 
  void InitFromModelFiles(std::string const &net_sdf_file,
                          std::string const &net_spf_file);
 
  void Init();
 
  nlohmann::json PackJson() override;
 
  void InitFromJson(nlohmann::json const &js) override;
 
  void UpdateSurfaceNodes() override;
 
  void UpdateShapeProperties() override;
 
  void SetSize(double d) override;
 
  Shape *Clone() const override;
 
  STLModel GetSTLModel(int num_nodes = 200) override;
 
  double SignedDistance(Vec3d const &pos) const override;
 
  Vec3d SurfacePoint(Vec3d const &pos) override;
 
  ~NetSDF();
 
  void Print() override;
 
private:
  STLModel surface_stl;
};
 
} // namespace netdem