shape_coded_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 CodedNetSDF : public Shape {
public:
  RegressionNet net_sdf_model;
 
  VecXT<double> latent_code;
 
  CodedNetSDF();
 
  void InitFromModelFiles(std::string const &net_sdf_file,
                          std::string const &latent_code_file, int id);
 
  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;
 
  ~CodedNetSDF();
 
  void Print() override;
 
private:
  int num_latent_code{0};
 
  STLModel surface_stl;
};
 
} // namespace netdem