mask_selection.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 Povl Filip Sonne-Frederiksen
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#pragma once

 
#include "reusex/core/ProjectDB.hpp"
 
#include <cstddef>
#include <cstdint>
#include <optional>
#include <stdexcept>
#include <string>
#include <vector>
 
namespace cv {
class Mat;
} // namespace cv
 
namespace reusex::core {

class MaskSelectionError : public std::runtime_error {
    public:
  using std::runtime_error::runtime_error;
};

struct MaskProjectionOptions {
  std::string cloud = "cloud";
  double depth_tolerance = 0.10;
  double max_depth = 8.0;
};

std::vector<std::size_t>
project_frame_mask(const ProjectDB &db, int frame_id, const cv::Mat &mask,
                   const MaskProjectionOptions &opts = {});

struct MaskSelectionOptions {
  std::string base_cloud = "cloud";
  std::string semantic_cloud = "labels"; // pipeline kDefaultSemanticCloud
  std::string instance_cloud = std::string(kDefaultInstanceCloud);
  std::string rooms_cloud = "rooms";
  std::optional<int64_t> type_id;
  int frame_id = -1;
};

struct MaskSelectionResult {
  std::string resource_code; 
  int64_t type_id = 0;       
  bool type_created = false; 
  uint32_t label_id = 0;     
  bool label_created = false;
  uint32_t instance_id = 0; 
  std::string instance_guid;
  int point_count = 0;
  std::vector<std::pair<uint32_t, int>> shrunk_instances;
};

MaskSelectionResult
apply_mask_selection(ProjectDB &db, const std::vector<std::size_t> &indices,
                     const std::string &class_name,
                     const MaskSelectionOptions &opts = {});
 
} // namespace reusex::core