#include "compute-renderpass.hpp"

#include "vk-api.hpp"

#include "vk-renderer.hpp"

void ComputeRenderpass::set_ouput_texture(Texture* output_texture) {
    this->output_texture = output_texture;

    // TODO: Remove hardcoded offset
    memcpy(constants.data() + 56, &output_texture->device_image.bindless_storage_index, 4);
}

void ComputeRenderpass::set_dispatch_size(size_t count_x, size_t count_y, size_t count_z) {
    group_count_x = count_x;
    group_count_y = count_y;
    group_count_z = count_z;
}

void ComputeRenderpass::set_constant(off_t offset, uint64_t* constant) {
    memcpy(constants.data() + offset, constant, sizeof(uint64_t));
}

void ComputeRenderpass::set_constant(off_t offset, Texture* texture) {
    memcpy(constants.data() + offset, (void*)&texture->device_image.bindless_sampled_index, 4);
}

void ComputeRenderpass::execute(VkCommandBuffer command_buffer) {
    vkCmdPushConstants(command_buffer, api.bindless_descriptor.pipeline_layout, VK_SHADER_STAGE_COMPUTE_BIT, 0, 64, (void*)&constants);

    api.run_compute_pipeline(command_buffer, pipeline, group_count_x, group_count_y, group_count_z);
}