#pragma once

#include <basic/defs.hpp>
#include <basic/math.hpp>
#include <vector>
#include <imgui.h>

enum REntityType : int32_t {
    REntityType_Rectangle,
    REntityType_Bitmap,
};

struct REntity_Rectangle {
    REntityType type;
    float x0;
    float y0;
    float x1;
    float y1;
    float z;
    Color color;
};

struct REntity_Bitmap {
    REntityType type;
    float x;
    float y;
    int32_t w;
    int32_t h;
    float z;
    void *data;
};

union REntity {
    REntityType type;
    REntity_Rectangle rect;
    REntity_Bitmap bitmap;
};

struct RSortEntry {
    RSortEntry(float z, size_t entity_index);
    float z;
    size_t entity_index;
};


class RenderGroup {
public:
    RenderGroup();
    void Clear(Color color);
    void Reset();

    void SetCameraSize(float width, float height);
    V2F32 ViewPosToScreenPos(V2F32 view_pos);
    V2F32 ViewSizeToScreenSize(V2F32 view_size);
    ImVec2 ViewPosToScreenPosImGui(V2F32 view_pos);
    ImVec2 ViewSizeToScreenSizeImGui(V2F32 view_size);
    float GetScale();


public:
    void PushRectangle(RectF32 rect, float z, Color color);
    void PushBitmap(V3F32 pos, int w, int h, void *bitmap);
    void Sort();


public:
    int32_t m_ScreenWidth;
    int32_t m_ScreenHeight;


private:
    friend class Renderer;

    float m_CameraWidth;
    float m_CameraHeight;


    Color m_ClearColor;

    std::vector<REntity> m_REntities;
    std::vector<RSortEntry> m_RSortEntries;
};