#pragma once

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

class Texture;
class Buffer;

class ComputeRenderpass : public Renderpass {
    public:

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

    ~ComputeRenderpass() {
        api.destroy_pipeline(pipeline);
    }


    void set_pipeline(std::string& shader_name);

    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);
    void set_constant(off_t offset, Buffer* buffer);

    // TODO: Do not use vulkan api type
    void execute(vkrenderer& renderer, 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;

    pipeline pipeline;
};