#include "config.h"

#include "gskvulkanborderopprivate.h"

#include "gskvulkanprivate.h"
#include "gskvulkanshaderopprivate.h"
#include "gsk/gskroundedrectprivate.h"

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

typedef struct _GskVulkanBorderOp GskVulkanBorderOp;

struct _GskVulkanBorderOp
{
  GskVulkanShaderOp op;

  GskRoundedRect outline;
  float widths[4];
  GdkRGBA colors[4];
};

static void
gsk_vulkan_border_op_finish (GskVulkanOp *op)
{
}

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

  print_indent (string, indent);
  print_rounded_rect (string, &self->outline);
  g_string_append (string, "border ");
  print_rgba (string, &self->colors[0]);
  if (!gdk_rgba_equal (&self->colors[3], &self->colors[0]) ||
      !gdk_rgba_equal (&self->colors[2], &self->colors[0]) ||
      !gdk_rgba_equal (&self->colors[1], &self->colors[0]))
    {
      print_rgba (string, &self->colors[1]);
      print_rgba (string, &self->colors[2]);
      print_rgba (string, &self->colors[3]);
    }
  g_string_append_printf (string, "%g ", self->widths[0]);
  if (self->widths[0] != self->widths[1] ||
      self->widths[0] != self->widths[2] ||
      self->widths[0] != self->widths[3])
    g_string_append_printf (string, "%g %g %g ", self->widths[1], self->widths[2], self->widths[3]);

  print_newline (string);
}

static void
gsk_vulkan_border_op_collect_vertex_data (GskVulkanOp *op,
                                          guchar      *data)
{
  GskVulkanBorderOp *self = (GskVulkanBorderOp *) op;
  GskVulkanBorderInstance *instance = (GskVulkanBorderInstance *) (data + ((GskVulkanShaderOp *) op)->vertex_offset);
  guint i;

  gsk_rounded_rect_to_float (&self->outline, graphene_point_zero (), instance->rect);
  for (i = 0; i < 4; i++)
    {
      instance->border_widths[i] = self->widths[i];
      gsk_vulkan_rgba_to_float (&self->colors[i], (gpointer) &instance->border_colors[4 * i]);
    }
}

static void
gsk_vulkan_border_op_reserve_descriptor_sets (GskVulkanOp     *op,
                                              GskVulkanRender *render)
{
}

static GskVulkanOp *
gsk_vulkan_border_op_command (GskVulkanOp     *op,
                              GskVulkanRender *render,
                              VkRenderPass     render_pass,
                              VkCommandBuffer  command_buffer)
{
  return gsk_vulkan_shader_op_command_n (op, render, render_pass, command_buffer, 8);
}

static const GskVulkanShaderOpClass GSK_VULKAN_BORDER_OP_CLASS = {
  {
    GSK_VULKAN_OP_SIZE (GskVulkanBorderOp),
    GSK_VULKAN_STAGE_SHADER,
    gsk_vulkan_border_op_finish,
    gsk_vulkan_border_op_print,
    gsk_vulkan_shader_op_count_vertex_data,
    gsk_vulkan_border_op_collect_vertex_data,
    gsk_vulkan_border_op_reserve_descriptor_sets,
    gsk_vulkan_border_op_command
  },
  "border",
  0,
  &gsk_vulkan_border_info,
};

void
gsk_vulkan_border_op (GskVulkanRender         *render,
                      GskVulkanShaderClip      clip,
                      const GskRoundedRect    *outline,
                      const graphene_point_t  *offset,
                      const float              widths[4],
                      const GdkRGBA            colors[4])
{
  GskVulkanBorderOp *self;
  guint i;

  self = (GskVulkanBorderOp *) gsk_vulkan_shader_op_alloc (render, &GSK_VULKAN_BORDER_OP_CLASS, clip, NULL);

  self->outline = *outline;
  gsk_rounded_rect_offset (&self->outline, offset->x, offset->y);
  for (i = 0; i < 4; i++)
    {
      self->widths[i] = widths[i];
      self->colors[i] = colors[i];
    }
}