kitti_odometry_pair_dataset.hpp

Go to the documentation of this file.

#pragma once
 
#include <cpp-toolbox/io/dataset/dataset.hpp>
#include <cpp-toolbox/io/dataset/kitti_types.hpp>
#include <cpp-toolbox/io/dataloader/kitti_odometry_dataset.hpp>
 
#include <memory>
#include <string>
#include <optional>
 
namespace toolbox::io {
 
template<typename DataType = float>
class kitti_odometry_pair_dataset_t : 
    public dataset_t<kitti_odometry_pair_dataset_t<DataType>, 
                     kitti_odometry_frame_pair_t<DataType>> {
public:
    using pair_type = kitti_odometry_frame_pair_t<DataType>;
    using base_type = dataset_t<kitti_odometry_pair_dataset_t<DataType>, pair_type>;
    
    explicit kitti_odometry_pair_dataset_t(const std::string& sequence_path, 
                                          std::size_t skip = 1);
    
    void set_skip(std::size_t skip) { 
        skip_ = skip;
        // Clear cache as pairs have changed
        pair_cache_.clear();
    }
    
    [[nodiscard]] std::size_t get_skip() const { return skip_; }
    
    void enable_overlap_computation(bool enable) {
        compute_overlap_ = enable;
    }
    
    void set_cache_size(std::size_t size) {
        max_cache_size_ = size;
        // Trim cache if needed
        while (pair_cache_.size() > max_cache_size_) {
            pair_cache_.erase(pair_cache_.begin());
        }
    }
    
    // ==================== Dataset Interface ====================
    
    [[nodiscard]] std::size_t size_impl() const {
        const auto num_frames = base_dataset_.size();
        return (num_frames > skip_) ? (num_frames - skip_) : 0;
    }
    
    [[nodiscard]] std::optional<pair_type> at_impl(std::size_t index) const;
    
    // ==================== Additional Functionality ====================
    
    [[nodiscard]] const kitti_odometry_dataset_t<DataType>& 
    get_base_dataset() const { return base_dataset_; }
    
    [[nodiscard]] std::size_t get_source_frame_index(std::size_t pair_index) const {
        return pair_index;
    }
    
    [[nodiscard]] std::size_t get_target_frame_index(std::size_t pair_index) const {
        return pair_index + skip_;
    }
    
    void clear_cache() const {
        pair_cache_.clear();
        base_dataset_.clear_cache();
    }
    
    [[nodiscard]] std::vector<std::size_t> get_pairs_with_motion(
        DataType min_translation = 0.1,
        DataType min_rotation = 0.01) const;
 
private:
    DataType compute_overlap_ratio(
        const point_cloud_t<DataType>& source,
        const point_cloud_t<DataType>& target,
        DataType threshold = 0.1) const;
    
    // ==================== Member Variables ====================
    
    mutable kitti_odometry_dataset_t<DataType> base_dataset_;
    
    std::size_t skip_;
    
    mutable std::map<std::size_t, pair_type> pair_cache_;
    
    std::size_t max_cache_size_ = 50;
    
    bool compute_overlap_ = false;
};
 
} // namespace toolbox::io
 
// Include implementation
#include <cpp-toolbox/io/dataloader/impl/kitti_odometry_pair_dataset_impl.hpp>