#pragma once

#include <pdfengine/layout_tree.hpp>
#include <pdfengine/geometry/rect.hpp>

#include <vector>
#include <memory>
#include <string>

namespace pdfengine {

class MultiIndexSpatialIndex {
public:
    MultiIndexSpatialIndex() = default;

    void buildFromTree(const PhysicalLayoutTree& tree);
    void clear();

    [[nodiscard]] std::vector<std::shared_ptr<LayoutBlock>> queryBlocksAtPoint(float x, float y) const;
    [[nodiscard]] std::vector<std::shared_ptr<LayoutBlock>> queryBlocksInRect(const geometry::Rect& rect) const;
    [[nodiscard]] std::vector<std::shared_ptr<LayoutBlock>> queryBlocksByType(LayoutBlockType type) const;

    [[nodiscard]] size_t totalIndexedBlocks() const noexcept { return m_indexedBlocks.size(); }

private:
    struct SpatialEntry {
        geometry::Rect bounds;
        std::shared_ptr<LayoutBlock> block;
    };

    std::vector<SpatialEntry> m_indexedBlocks;
    std::vector<SpatialEntry> m_indexedLines;
};

} // namespace pdfengine