MRUIRectAllocator.h

Go to the documentation of this file.

#pragma once
 
#include "exports.h"
#include "MRMesh/MRBox.h"
#include "MRMesh/MRphmap.h"
#include "MRMesh/MRVector2.h"
#include "MRMesh/MRViewportId.h"
#include "MRImGui.h"
 
namespace MR::UI
{
 
class RectAllocator
{
public:
    MRVIEWER_API RectAllocator();
 
    struct FindFreeRectResult
    {
        Box2f rect;
        bool ok = true; 
        [[nodiscard]] explicit operator bool() const { return ok; }
    };
 
    using FindPotentiallyOverlappingRects = std::function<void( Box2f target, std::function<void( const char* name, Box2f box )> overlaps )>;
 
    [[nodiscard]] MRVIEWER_API FindFreeRectResult findFreeRect(
        Box2f preferredRect,
        Box2f preferredBounds,
        FindPotentiallyOverlappingRects findOverlaps,
        ImVec2 axisWeights = ImVec2( 1, 1 )
    );
 
    [[nodiscard]] static bool rectRectOverlap( Box2f a, Box2f b )
    {
        return a.max.x > b.min.x && a.min.x < b.max.x && a.max.y > b.min.y && a.min.y < b.max.y;
    }
 
private:
    phmap::flat_hash_map<Vector2f, float> visitedCoords;
 
    struct CoordsToVisit
    {
        Vector2f pos;
        float cost = 0;
        std::array<float, 4> overlapWithBounds;
    };
    std::vector<CoordsToVisit> coordsToVisitHeap;
};
 
class WindowRectAllocator : public RectAllocator
{
public:
    MRVIEWER_API void setFreeNextWindowPos( const char* expectedWindowName, ImVec2 defaultPos, ImGuiCond cond = ImGuiCond_Appearing, ImVec2 pivot = ImVec2() );
 
    MRVIEWER_API void invalidateClosedWindows();
private:
    enum class AllocationState
    {
        None, 
        Requested, 
        WaitAppearing, 
        Set 
    };
 
    struct WindowEntry
    {
        AllocationState state_{ AllocationState::None };
    };
    phmap::flat_hash_map<std::string, WindowEntry> windows_;
};
[[nodiscard]] MRVIEWER_API WindowRectAllocator& getDefaultWindowRectAllocator();
 
class LabelRectAllocator : public RectAllocator
{
public:
    MRVIEWER_API ImVec2 createRect( ViewportId viewportId, std::string id, ImVec2 pos, ImVec2 size, bool forceExactPosition = false );
 
private:
    int lastFrameCount_ = -1;
 
    struct Entry
    {
        Box2f box;
        bool visitedThisFrame = true;
    };
    std::vector<phmap::flat_hash_map<std::string, Entry>> entries_;
};
[[nodiscard]] MRVIEWER_API LabelRectAllocator& getDefaultLabelRectAllocator();
 
}