#include "config.h"

#include "gskvulkanbluropprivate.h"

#include "gskvulkanprivate.h"

#include "vulkan/resources/blur.vert.h"

typedef struct _GskVulkanBlurOp GskVulkanBlurOp;

struct _GskVulkanBlurOp
{
  GskVulkanOp op;

  GskVulkanImage *image;
  graphene_rect_t rect;
  graphene_rect_t tex_rect;
  float radius;

  guint32 image_descriptor;
};

static void
gsk_vulkan_blur_op_finish (GskVulkanOp *op)
{
  GskVulkanBlurOp *self = (GskVulkanBlurOp *) op;

  g_object_unref (self->image);
}

static void
gsk_vulkan_blur_op_print (GskVulkanOp *op,
                          GString     *string,
                          guint        indent)
{
  GskVulkanBlurOp *self = (GskVulkanBlurOp *) op;

  print_indent (string, indent);
  print_rect (string, &self->rect);
  g_string_append_printf (string, "blur %g",
                          self->radius);
  print_newline (string);
}

static void
gsk_vulkan_blur_op_collect_vertex_data (GskVulkanOp *op,
                                        guchar      *data)
{
  GskVulkanBlurOp *self = (GskVulkanBlurOp *) op;
  GskVulkanBlurInstance *instance = (GskVulkanBlurInstance *) (data + op->vertex_offset);

  gsk_vulkan_rect_to_float (&self->rect, instance->rect);
  gsk_vulkan_rect_to_float (&self->tex_rect, instance->tex_rect);
  instance->tex_id = self->image_descriptor;
  instance->radius = self->radius;
}

static void
gsk_vulkan_blur_op_reserve_descriptor_sets (GskVulkanOp     *op,
                                            GskVulkanRender *render)
{
  GskVulkanBlurOp *self = (GskVulkanBlurOp *) op;

  self->image_descriptor = gsk_vulkan_render_get_image_descriptor (render,
                                                                   self->image,
                                                                   GSK_VULKAN_SAMPLER_DEFAULT);
}

static const GskVulkanOpClass GSK_VULKAN_BLUR_OP_CLASS = {
  GSK_VULKAN_OP_SIZE (GskVulkanBlurOp),
  GSK_VULKAN_STAGE_COMMAND,
  "blur",
  &gsk_vulkan_blur_info,
  gsk_vulkan_blur_op_finish,
  gsk_vulkan_blur_op_print,
  gsk_vulkan_op_draw_upload,
  gsk_vulkan_op_draw_count_vertex_data,
  gsk_vulkan_blur_op_collect_vertex_data,
  gsk_vulkan_blur_op_reserve_descriptor_sets,
  gsk_vulkan_op_draw_command
};

void
gsk_vulkan_blur_op (GskVulkanRender         *render,
                    const char              *clip_type,
                    GskVulkanImage          *image,
                    const graphene_rect_t   *rect,
                    const graphene_point_t  *offset,
                    const graphene_rect_t   *tex_rect,
                    float                    radius)
{
  GskVulkanBlurOp *self;

  g_assert (radius > 0);

  self = (GskVulkanBlurOp *) gsk_vulkan_op_alloc (render, &GSK_VULKAN_BLUR_OP_CLASS);

  ((GskVulkanOp *) self)->clip_type = g_intern_string (clip_type);
  self->image = g_object_ref (image);
  graphene_rect_offset_r (rect, offset->x, offset->y, &self->rect);
  gsk_vulkan_normalize_tex_coords (&self->tex_rect, rect, tex_rect);
  self->radius = radius;
}