#pragma once

#include "vk-api.hpp"
#include <array>

class Texture;
class Renderpass {
    public:

    Renderpass(vkapi& api)
        :api(api)
    {};

    virtual ~Renderpass() {
        api.destroy_pipeline(pipeline);
    }

    void set_pipeline(pipeline pipeline) {
        this->pipeline = pipeline;
    }

    // TODO: Do not use vulkan api type
    virtual void execute(VkCommandBuffer command_buffer) = 0;

    protected:
    vkapi &api;

    pipeline pipeline;
};

class ComputeRenderpass : public Renderpass {
    public:

    ComputeRenderpass(vkapi& api)
        : Renderpass(api)
    {}

    void set_dispatch_size(size_t count_x, size_t count_y, size_t count_z);

    void set_ouput_texture(Texture* ouput_texture);

    void set_constant(off_t offset, uint64_t* constant);
    void set_constant(off_t offset, Texture* texture);

    // TODO: Do not use vulkan api type
    void execute(VkCommandBuffer command_buffer) override final;

    Texture* output_texture;

    private:

    std::vector<Texture*> input_textures;

    std::array<uint8_t, 64> constants;

    size_t group_count_x;
    size_t group_count_y;
    size_t group_count_z;
};