ProjectDB.hpp

Go to the documentation of this file.

#pragma once
#include "reusex/types.hpp"
 
#include <array>
#include <filesystem>
#include <map>
#include <memory>
#include <optional>
#include <opencv2/core/mat.hpp>
#include <pcl/PolygonMesh.h>
#include <pcl/TextureMesh.h>
#include <pcl/point_cloud.h>
#include <pcl/point_types.h>
#include <string>
#include <string_view>
#include <vector>
 
namespace reusex {
 
// Forward declarations
namespace core {
struct MaterialPassport;
struct MaterialPassportMetadata;
struct SensorIntrinsics;
} // namespace core
namespace geometry {
struct BuildingComponent;
enum class ComponentType;
} // namespace geometry
 
class ProjectDB {
    public:
  explicit ProjectDB(std::filesystem::path dbPath, bool readOnly = false);
  ~ProjectDB();
 
  // Non-copyable (RAII resource management)
  ProjectDB(const ProjectDB &) = delete;
  ProjectDB &operator=(const ProjectDB &) = delete;
 
  // Movable
  ProjectDB(ProjectDB &&) noexcept;
  ProjectDB &operator=(ProjectDB &&) noexcept;
 
  // --- Core Database Operations ---
 
  bool is_open() const noexcept;
  const std::filesystem::path &path() const noexcept;
  int schema_version() const;
  void validate_schema() const;
 
  // --- Sensor Frame Operations ---
 
  void save_sensor_frame(int nodeId, const cv::Mat &colorImage);
 
  void save_sensor_frame(int nodeId, const cv::Mat &color, const cv::Mat &depth,
                         const cv::Mat &confidence,
                         const std::array<double, 16> &worldPose,
                         const core::SensorIntrinsics &intrinsics,
                         double timestamp = -1.0);
 
  std::vector<int> sensor_frame_ids() const;
  cv::Mat sensor_frame_image(int nodeId) const;
  cv::Mat sensor_frame_depth(int nodeId) const;
  cv::Mat sensor_frame_confidence(int nodeId) const;
  std::array<double, 16> sensor_frame_pose(int nodeId) const;
  core::SensorIntrinsics sensor_frame_intrinsics(int nodeId) const;
  bool has_sensor_frame(int nodeId) const;

  double sensor_frame_timestamp(int nodeId) const;

  int nearest_sensor_frame_by_timestamp(double timestamp) const;
 
  // --- Panoramic Image Operations ---
 
  struct PanoramicImage {
    int id;
    std::string filename;
    double timestamp; // -1.0 if unknown
    int node_id;      // -1 if unmatched
  };
 
  void save_panoramic_image(const std::string &filename,
                            const std::vector<uint8_t> &jpeg_data,
                            double timestamp = -1.0, int nodeId = -1);
 
  cv::Mat panoramic_image(int id) const;
  cv::Mat panoramic_image(std::string_view filename) const;
  bool has_panoramic_image(std::string_view filename) const;
  void delete_panoramic_image(int id);
  void delete_panoramic_image(std::string_view filename);
  std::vector<PanoramicImage> list_panoramic_images() const;
  int panoramic_image_count() const;
 
  // --- Segmentation Image Operations ---
 
  bool has_segmentation_image(int nodeId) const;
  cv::Mat segmentation_image(int nodeId) const;
  std::vector<int> segmentation_image_ids() const;
  void save_segmentation_image(int nodeId, const cv::Mat &labels);
  void save_segmentation_images(const std::vector<int> &nodeIds,
                                const std::vector<cv::Mat> &labels);
 
  // --- Point Cloud Operations ---
 
  void save_point_cloud(std::string_view name, const Cloud &cloud,
                        std::string_view stage = "",
                        std::string_view paramsJson = "");
 
  void save_point_cloud(std::string_view name, const CloudN &cloud,
                        std::string_view stage = "",
                        std::string_view paramsJson = "");
 
  void save_point_cloud(std::string_view name, const CloudL &cloud,
                        std::string_view stage = "",
                        std::string_view paramsJson = "");
 
  void save_point_cloud(std::string_view name,
                        const pcl::PointCloud<pcl::PointXYZ> &cloud,
                        std::string_view stage = "",
                        std::string_view paramsJson = "");
 
  CloudPtr point_cloud_xyzrgb(std::string_view name) const;
  CloudNPtr point_cloud_normal(std::string_view name) const;
  CloudLPtr point_cloud_label(std::string_view name) const;
  pcl::PointCloud<pcl::PointXYZ>::Ptr
  point_cloud_xyz(std::string_view name) const;
 
  bool has_point_cloud(std::string_view name) const;
  void delete_point_cloud(std::string_view name);
  std::vector<std::string> list_point_clouds() const;
  std::string point_cloud_type(std::string_view name) const;
 
  // --- Label Definitions ---
 
  void save_label_definitions(std::string_view cloudName,
                              const std::map<int, std::string> &labelMap);
 
  std::map<int, std::string>
  label_definitions(std::string_view cloudName) const;
 
  // --- Mesh Operations ---
 
  void save_mesh(std::string_view name, const pcl::PolygonMesh &mesh,
                 std::string_view stage = "", std::string_view paramsJson = "");
 
  void save_mesh(std::string_view name, const pcl::TextureMesh &mesh,
                 std::string_view stage = "", std::string_view paramsJson = "");
 
  pcl::PolygonMesh::Ptr mesh(std::string_view name) const;
  pcl::TextureMesh::Ptr texture_mesh(std::string_view name) const;
  bool has_mesh(std::string_view name) const;
  std::vector<std::string> list_meshes() const;
  std::string mesh_format(std::string_view name) const;
 
  struct MeshMetadata {
    std::string name, format, stage, parameters, created_at;
    int vertex_count = 0, face_count = 0;
  };
  MeshMetadata mesh_metadata(std::string_view name) const;
 
  std::vector<uint8_t> mesh_data_blob(std::string_view name) const;
 
  struct MeshTextureInfo {
    std::string tex_name, format;
    std::vector<uint8_t> image_data;
    int width, height;
  };
  std::vector<MeshTextureInfo> mesh_texture_blobs(std::string_view name) const;
 
  struct MeshTextureMetadata {
    std::string tex_name, format;
    int width, height;
  };
  std::vector<MeshTextureMetadata>
  mesh_texture_metadata(std::string_view name) const;
 
  // --- Building Component Operations ---
 
  void save_building_component(const geometry::BuildingComponent &component);
  geometry::BuildingComponent building_component(std::string_view name) const;
  bool has_building_component(std::string_view name) const;
  void delete_building_component(std::string_view name);
  std::vector<std::string> list_building_components() const;
  std::vector<std::string>
  list_building_components(geometry::ComponentType type) const;
  int building_component_count() const;
 
  // --- Pipeline Log ---
 
  int log_pipeline_start(std::string_view stage,
                         std::string_view paramsJson = "");
 
  void log_pipeline_end(int logId, bool success,
                        std::string_view errorMsg = "");
 
  struct PipelineLogEntry {
    int id;
    std::string stage;
    std::string started_at;
    std::string finished_at; // Empty if still running
    std::string parameters;  // JSON string
    std::string status;      // "running", "success", "failed"
    std::string error_msg;   // Empty if no error
  };
 
  std::vector<PipelineLogEntry> pipeline_log(int limit = 0) const;
 
  // --- Project Summary ---
 
  struct ProjectSummary {
    struct CloudInfo {
      std::string name;
      std::string type; // "PointXYZRGB", "Normal", "Label", "PointXYZ"
      size_t point_count;
      size_t width;
      size_t height;
      bool organized;                    // height > 1
      std::map<int, std::string> labels; // Only for Label clouds
    };
 
    struct MeshInfo {
      std::string name;
      int vertex_count;
      int face_count;
    };
 
    struct SensorFrameInfo {
      int total_count;
      int width;           // 0 if no frames
      int height;          // 0 if no frames
      int segmented_count; // frames with segmentation
    };
 
    struct PanoramicInfo {
      int total_count = 0;
      int matched_count = 0; // images linked to a sensor frame
    };
 
    struct ComponentInfo {
      int total_count = 0;
      std::map<std::string, int> count_by_type;
    };
 
    struct MaterialInfo {
      std::string id;
      std::string guid;
      int property_count = 0;
      std::string created_at;     // ISO 8601 timestamp
      std::string version_number; // Semver (e.g., "0.1.1")
    };
 
    struct ProjectInfo {
      std::string id;
      std::string name;
      std::string building_address;
      int year_of_construction = 0; // 0 = not set
      std::string survey_date;
      std::string survey_organisation;
      std::string notes;
    };
 
    std::filesystem::path path;
    int schema_version;
    std::vector<ProjectInfo> projects;
    std::vector<CloudInfo> clouds;
    std::vector<MeshInfo> meshes;
    SensorFrameInfo sensor_frames;
    PanoramicInfo panoramic_images;
    ComponentInfo components;
    std::vector<MaterialInfo> materials;
  };
 
  ProjectSummary project_summary() const;
 
  // --- Material Passport Operations ---
 
  core::MaterialPassport material_passport(std::string_view documentGuid) const;
  std::vector<core::MaterialPassport> all_material_passports() const;
 
  void add_material_passport(const core::MaterialPassport &passport,
                             std::string_view projectId);

  void add_material_passport(const core::MaterialPassport &passport,
                             std::string_view projectId, std::string_view id);

  void delete_material_passport(std::string_view documentGuid);

  std::vector<std::string> list_passport_guids() const;

  std::map<std::string, std::string>
  passport_stored_properties(std::string_view documentGuid) const;

  std::string passport_property_value(std::string_view documentGuid,
                                      std::string_view fieldName) const;

  std::optional<int>
  passport_linked_node_id(std::string_view documentGuid) const;

  core::MaterialPassportMetadata
  passport_metadata(std::string_view documentGuid) const;

  void set_passport_metadata_field(std::string_view documentGuid,
                                   std::string_view column,
                                   std::string_view value);

  void set_passport_property(std::string_view documentGuid,
                             std::string_view fieldName,
                             std::string_view value);

  void delete_passport_property(std::string_view documentGuid,
                                std::string_view fieldName);
 
  // --- Project Metadata Operations ---
 
  struct ProjectMetadata {
    std::string id;
    std::string name;
    std::string building_address;
    int year_of_construction = 0; // 0 = not set
    std::string survey_date;
    std::string survey_organisation;
    std::string notes;
  };

  ProjectMetadata get_project_metadata(std::string_view projectId) const;

  void update_project_metadata(const ProjectMetadata &metadata);

  std::vector<std::string> list_project_ids() const;
 
    private:
  class Impl;
  std::unique_ptr<Impl> impl_;
};
} // namespace reusex