#pragma once

#include <mln/gfx/drawable_data.hpp>
#include <mln/gfx/texture2d.hpp>
#include <mln/gfx/uniform_buffer.hpp>
#include <mln/tile/tile_id.hpp>
#include <mln/util/color.hpp>
#include <mln/util/containers.hpp>
#include <mln/util/identity.hpp>
#include <mln/util/monotonic_timer.hpp>
#include <mln/util/traits.hpp>
#include <mln/util/immutable.hpp>

#include <cstdint>
#include <cstddef>
#include <memory>
#include <optional>
#include <vector>

namespace mln {

class Bucket;
class Color;
class LayerTweaker;
class PaintParameters;
class PaintPropertyBindersBase;
enum class RenderPass : uint8_t;
class RenderTile;
class SegmentBase;

using LayerTweakerPtr = std::shared_ptr<LayerTweaker>;
using RenderTiles = std::shared_ptr<const std::vector<std::reference_wrapper<const RenderTile>>>;

namespace gfx {

class ColorMode;
class CullFaceMode;
enum class DepthMaskType : bool;
class DrawableTweaker;
class DrawMode;
class IndexVectorBase;
class ShaderProgramBase;
class VertexAttributeArray;

using DrawPriority = int64_t;
using DrawableTweakerPtr = std::shared_ptr<DrawableTweaker>;
using IndexVectorBasePtr = std::shared_ptr<IndexVectorBase>;
using ShaderProgramBasePtr = std::shared_ptr<ShaderProgramBase>;
using Texture2DPtr = std::shared_ptr<Texture2D>;
using VertexAttributeArrayPtr = std::shared_ptr<VertexAttributeArray>;

class Drawable {
public:
    /// @brief Array of textures to bind
    using Textures = std::array<gfx::Texture2DPtr, shaders::maxTextureCountPerShader>;

protected:
    Drawable(std::string name);

public:
    /// Drawable constructor
    virtual ~Drawable();

    struct DrawSegment;
    using UniqueDrawSegment = std::unique_ptr<DrawSegment>;

    /// Get drawable's ID
    const util::SimpleIdentity& getID() const { return uniqueID; }

    /// Draw the drawable
    virtual void draw(PaintParameters&) const = 0;

    /// Drawable name is used for debugging and troubleshooting
    const std::string& getName() const { return name; }

    /// Set drawable name
    void setName(std::string value) { name = std::move(value); }

    /// Which shader to use when rendering this drawable
    const gfx::ShaderProgramBasePtr& getShader() const { return shader; }

    /// Set the shader to be used
    virtual void setShader(gfx::ShaderProgramBasePtr value) { shader = std::move(value); }

    /// The pass on which we'll be rendered
    mln::RenderPass getRenderPass() const { return renderPass; }

    /// Sets the render passes
    void setRenderPass(mln::RenderPass value) { renderPass = value; }

    /// Test whether to draw this drawable in a given render pass.
    bool hasRenderPass(const mln::RenderPass value) const {
        return (mln::underlying_type(renderPass) & mln::underlying_type(value)) != 0;
    }

    /// Test whether to draw this drawable in a given render pass.
    /// If multiple render pass bits are set, all must be present.
    bool hasAllRenderPasses(const mln::RenderPass value) const {
        const auto underlying_value = mln::underlying_type(value);
        return (mln::underlying_type(renderPass) & underlying_value) == underlying_value;
    }

    /// Width for lines
    int32_t getLineWidth() const { return lineWidth; }

    /// Set line width
    virtual void setLineWidth(int32_t value) { lineWidth = value; }

    /// @brief Get the texture at the given internal ID.
    const gfx::Texture2DPtr& getTexture(size_t id) const;

    /// @brief Set the collection of textures bound to this drawable
    /// @param textures_ A Textures collection to set
    virtual void setTextures(const Textures& textures_) noexcept { textures = textures_; }
    virtual void setTextures(Textures&& textures_) noexcept { textures = std::move(textures_); }

    /// @brief Attach the given texture to this drawable at the given internal ID.
    /// @param texture Texture2D instance
    /// @param id Internal ID of the texture.
    virtual void setTexture(gfx::Texture2DPtr texture, size_t id);

    /// Whether the drawble should be drawn
    bool getEnabled() const { return enabled; }

    /// Sets whether the drawble should be drawn
    void setEnabled(bool value) { enabled = value; }

    /// Whether to render to the color target
    bool getEnableColor() const { return enableColor; }

    /// Set whether to render to the color target
    virtual void setEnableColor(bool value) { enableColor = value; }

    /// Whether to do stenciling (based on the Tile ID or 3D)
    bool getEnableStencil() const { return enableStencil; }

    /// Set stencil usage
    virtual void setEnableStencil(bool value) { enableStencil = value; }

    /// not used for anything yet
    DrawPriority getDrawPriority() const { return drawPriority; }
    void setDrawPriority(DrawPriority value) { drawPriority = value; }

    /// Whether to enable depth testing
    bool getEnableDepth() const { return enableDepth; }
    virtual void setEnableDepth(bool value) { enableDepth = value; }

    /// Determines depth range within the layer for 2D drawables
    int32_t getSubLayerIndex() const { return subLayerIndex; }

    /// Set sub-layer index
    virtual void setSubLayerIndex(int32_t value) { subLayerIndex = value; }

    /// Depth writability for 2D drawables
    DepthMaskType getDepthType() const { return depthType; }

    /// Set depth type
    virtual void setDepthType(DepthMaskType value) { depthType = value; }

    /// Uses 3D depth mode
    bool getIs3D() const { return is3D; }

    /// Set 3D mode
    void setIs3D(bool value) { is3D = value; }

    /// True if this is a custom drawable
    bool getIsCustom() const { return isCustom; }

    /// Sets custom status for this drawable
    void setIsCustom(bool value) { isCustom = value; }

    /// Get the ID of the tile that this drawable represents, if any
    const std::optional<OverscaledTileID>& getTileID() const { return tileID; }

    /// Set the ID of the tile that this drawable represents
    void setTileID(const OverscaledTileID& value) { tileID = value; }

    /// Get cull face mode
    const gfx::CullFaceMode& getCullFaceMode() const;

    /// Set cull face mode
    virtual void setCullFaceMode(const gfx::CullFaceMode&);

    /// Get color mode
    const gfx::ColorMode& getColorMode() const;

    /// Set color mode
    virtual void setColorMode(const gfx::ColorMode&);

    /// Get the vertex attributes that override default values in the shader program
    const gfx::VertexAttributeArrayPtr& getVertexAttributes() const noexcept { return vertexAttributes; }

    /// Set vertex attribute array
    void setVertexAttributes(gfx::VertexAttributeArrayPtr value) noexcept {
        vertexAttributes = std::move(value);
        // The attribute bindings need to be rebuilt, we can't rely on the update
        // time check as these new values may not have been modified recently.
        attributeUpdateTime.reset();
    }

    /// Update vertices, indices, and segments
    virtual void updateVertexAttributes(gfx::VertexAttributeArrayPtr,
                                        std::size_t vertexCount,
                                        gfx::DrawMode,
                                        gfx::IndexVectorBasePtr,
                                        const SegmentBase* segments,
                                        std::size_t segmentCount) = 0;

    /// Get the instance attributes
    const gfx::VertexAttributeArrayPtr& getInstanceAttributes() const noexcept { return instanceAttributes; }

    /// Set instance attribute array
    void setInstanceAttributes(gfx::VertexAttributeArrayPtr value) noexcept { instanceAttributes = std::move(value); }

    /// Provide raw data for vertices. Incompatible with adding primitives
    virtual void setVertices(std::vector<uint8_t>&&, std::size_t, AttributeDataType) = 0;

    /// Provide raw indexes and segments
    void setIndexData(std::vector<std::uint16_t> indexes, std::vector<UniqueDrawSegment>);

    /// Set shared indexes and segments
    virtual void setIndexData(gfx::IndexVectorBasePtr, std::vector<UniqueDrawSegment>) = 0;

    /// Get the tweakers attached to this drawable
    const std::vector<DrawableTweakerPtr>& getTweakers() const { return tweakers; }

    /// Add a tweaker to this drawable
    void addTweaker(DrawableTweakerPtr value) { tweakers.emplace_back(std::move(value)); }

    /// Set the entire tweaker collection
    void setTweakers(std::vector<DrawableTweakerPtr> value) { tweakers = std::move(value); }

    /// Clear the tweaker collection
    void clearTweakers() { tweakers.clear(); }

    /// Get the uniform buffers attached to this drawable
    virtual const gfx::UniformBufferArray& getUniformBuffers() const = 0;

    /// Get the mutable uniform buffer array
    virtual gfx::UniformBufferArray& mutableUniformBuffers() = 0;

    /// Get drawable data
    const UniqueDrawableData& getData() const { return drawableData; }

    /// Set drawable data
    void setData(UniqueDrawableData&& value) { drawableData = std::move(value); }

    /// Set drawable user-defined type
    void setType(std::size_t type_) { type = type_; }

    /// Get drawable user-defined type
    size_t getType() const { return type; }

    void setUBOIndex(uint32_t uboIndex_) { uboIndex = uboIndex_; }
    uint32_t getUBOIndex() const { return uboIndex; }

    /// Associate the drawable with a layer tweaker.  This is used to manage the lifetime of the tweaker.
    void setLayerTweaker(LayerTweakerPtr tweaker) { layerTweaker = std::move(tweaker); }
    const LayerTweakerPtr& getLayerTweaker() const { return layerTweaker; }

    /// Get origin point
    const std::optional<mln::Point<double>>& getOrigin() const { return origin; }

    /// Set origin point
    void setOrigin(std::optional<Point<double>> p) { origin = std::move(p); }

    /// Get the property binders used for property updates
    PaintPropertyBindersBase* getBinders();
    const PaintPropertyBindersBase* getBinders() const;

    /// Set the property binders used for property updates
    void setBinders(std::shared_ptr<Bucket>, PaintPropertyBindersBase*);

    const RenderTile* getRenderTile() const;
    const std::shared_ptr<Bucket>& getBucket() const;
    void setRenderTile(Immutable<std::vector<RenderTile>>, const RenderTile*);

    const std::chrono::duration<double> createTime = util::MonotonicTimer::now();
    std::optional<std::chrono::duration<double>> getAttributeUpdateTime() const { return attributeUpdateTime; }

protected:
    bool enabled = true;
    bool enableColor = true;
    bool enableStencil = false;
    bool enableDepth = true;
    bool is3D = false;
    bool isCustom = false;
    std::string name;
    const util::SimpleIdentity uniqueID;
    gfx::ShaderProgramBasePtr shader;
    mln::RenderPass renderPass;
    std::optional<OverscaledTileID> tileID;
    DrawPriority drawPriority = 0;
    int32_t lineWidth = 1;
    int32_t subLayerIndex = 0;
    DepthMaskType depthType; // = DepthMaskType::ReadOnly;
    UniqueDrawableData drawableData{};
    gfx::VertexAttributeArrayPtr vertexAttributes;
    gfx::VertexAttributeArrayPtr instanceAttributes;
    std::optional<std::chrono::duration<double>> attributeUpdateTime;

    struct Impl;
    std::unique_ptr<Impl> impl;

    Textures textures;
    std::vector<DrawableTweakerPtr> tweakers;
    LayerTweakerPtr layerTweaker;

    std::size_t type = 0;
    std::optional<mln::Point<double>> origin;
    uint32_t uboIndex = 0;
};

using DrawablePtr = std::shared_ptr<Drawable>;
using UniqueDrawable = std::unique_ptr<Drawable>;

/// Comparator for sorting drawable pointers primarily by draw priority
struct DrawableLessByPriority {
    DrawableLessByPriority(bool descending = false)
        : desc(descending) {}
    bool operator()(const Drawable& left, const Drawable& right) const {
        const auto& a = desc ? right : left;
        const auto& b = desc ? left : right;
        if (a.getDrawPriority() != b.getDrawPriority()) {
            return a.getDrawPriority() < b.getDrawPriority();
        }
        return a.getID() < b.getID();
    }
    bool operator()(const Drawable* left, const Drawable* right) const { return operator()(*left, *right); }
    bool operator()(const UniqueDrawable& left, const UniqueDrawable& right) const { return operator()(*left, *right); }
    bool operator()(const DrawablePtr& left, const DrawablePtr& right) const {
        const auto& a = desc ? right : left;
        const auto& b = desc ? left : right;
        // nulls are less than non-nulls
        return (a && b) ? operator()(*a, *b) : (!a && b);
    }

private:
    bool desc;
};

} // namespace gfx
} // namespace mln