/* SPDX-FileCopyrightText: 2021 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */

/** \file
 * \ingroup draw
 */

#include "extract_mesh.hh"

#include "GPU_index_buffer.hh"

#include "draw_subdivision.hh"

namespace blender::draw {
/* ---------------------------------------------------------------------- */
/** \name Extract Edit UV Triangles Indices
 * \{ */

struct MeshExtract_EditUvElem_Data {
  GPUIndexBufBuilder elb;
  Span<int> corner_tri_faces;
  bool sync_selection;
};

static void extract_edituv_tris_init(const MeshRenderData &mr,
                                     MeshBatchCache & /*cache*/,
                                     void * /*ibo*/,
                                     void *tls_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(tls_data);
  GPU_indexbuf_init(&data->elb, GPU_PRIM_TRIS, mr.corner_tris_num, mr.corners_num);
  if (mr.extract_type == MR_EXTRACT_MESH) {
    data->corner_tri_faces = mr.mesh->corner_tri_faces();
  }
  data->sync_selection = (mr.toolsettings->uv_flag & UV_SYNC_SELECTION) != 0;
}

BLI_INLINE void edituv_tri_add(
    MeshExtract_EditUvElem_Data *data, bool hidden, bool selected, int v1, int v2, int v3)
{
  if (!hidden && (data->sync_selection || selected)) {
    GPU_indexbuf_add_tri_verts(&data->elb, v1, v2, v3);
  }
}

static void extract_edituv_tris_iter_looptri_bm(const MeshRenderData & /*mr*/,
                                                BMLoop **elt,
                                                const int /*elt_index*/,
                                                void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  edituv_tri_add(data,
                 BM_elem_flag_test(elt[0]->f, BM_ELEM_HIDDEN),
                 BM_elem_flag_test(elt[0]->f, BM_ELEM_SELECT),
                 BM_elem_index_get(elt[0]),
                 BM_elem_index_get(elt[1]),
                 BM_elem_index_get(elt[2]));
}

static void extract_edituv_tris_iter_corner_tri_mesh(const MeshRenderData &mr,
                                                     const int3 &tri,
                                                     const int elt_index,
                                                     void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  const int face_i = data->corner_tri_faces[elt_index];
  const BMFace *efa = bm_original_face_get(mr, face_i);
  const bool mp_hidden = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_HIDDEN) : true;
  const bool mp_select = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_SELECT) : false;

  edituv_tri_add(data, mp_hidden, mp_select, tri[0], tri[1], tri[2]);
}

static void extract_edituv_tris_finish(const MeshRenderData & /*mr*/,
                                       MeshBatchCache & /*cache*/,
                                       void *buf,
                                       void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  gpu::IndexBuf *ibo = static_cast<gpu::IndexBuf *>(buf);
  GPU_indexbuf_build_in_place(&data->elb, ibo);
}

static void extract_edituv_tris_init_subdiv(const DRWSubdivCache &subdiv_cache,
                                            const MeshRenderData &mr,
                                            MeshBatchCache & /*cache*/,
                                            void * /*buf*/,
                                            void *tls_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(tls_data);
  GPU_indexbuf_init(
      &data->elb, GPU_PRIM_TRIS, subdiv_cache.num_subdiv_triangles, subdiv_cache.num_subdiv_loops);
  data->sync_selection = (mr.toolsettings->uv_flag & UV_SYNC_SELECTION) != 0;
}

static void extract_edituv_tris_iter_subdiv_bm(const DRWSubdivCache & /*subdiv_cache*/,
                                               const MeshRenderData & /*mr*/,
                                               void *_data,
                                               uint subdiv_quad_index,
                                               const BMFace *coarse_quad)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  const uint loop_idx = subdiv_quad_index * 4;

  edituv_tri_add(data,
                 BM_elem_flag_test(coarse_quad, BM_ELEM_HIDDEN) != 0,
                 BM_elem_flag_test(coarse_quad, BM_ELEM_SELECT) != 0,
                 loop_idx,
                 loop_idx + 1,
                 loop_idx + 2);

  edituv_tri_add(data,
                 BM_elem_flag_test(coarse_quad, BM_ELEM_HIDDEN) != 0,
                 BM_elem_flag_test(coarse_quad, BM_ELEM_SELECT) != 0,
                 loop_idx,
                 loop_idx + 2,
                 loop_idx + 3);
}

static void extract_edituv_tris_iter_subdiv_mesh(const DRWSubdivCache & /*subdiv_cache*/,
                                                 const MeshRenderData &mr,
                                                 void *_data,
                                                 uint subdiv_quad_index,
                                                 const int coarse_quad_index)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  const uint loop_idx = subdiv_quad_index * 4;

  const BMFace *efa = bm_original_face_get(mr, coarse_quad_index);
  const bool mp_hidden = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_HIDDEN) : true;
  const bool mp_select = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_SELECT) : false;

  edituv_tri_add(data, mp_hidden, mp_select, loop_idx, loop_idx + 1, loop_idx + 2);
  edituv_tri_add(data, mp_hidden, mp_select, loop_idx, loop_idx + 2, loop_idx + 3);
}

static void extract_edituv_tris_finish_subdiv(const DRWSubdivCache & /*subdiv_cache*/,
                                              const MeshRenderData & /*mr*/,
                                              MeshBatchCache & /*cache*/,
                                              void *buf,
                                              void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  gpu::IndexBuf *ibo = static_cast<gpu::IndexBuf *>(buf);
  GPU_indexbuf_build_in_place(&data->elb, ibo);
}

constexpr MeshExtract create_extractor_edituv_tris()
{
  MeshExtract extractor = {nullptr};
  extractor.init = extract_edituv_tris_init;
  extractor.iter_looptri_bm = extract_edituv_tris_iter_looptri_bm;
  extractor.iter_corner_tri_mesh = extract_edituv_tris_iter_corner_tri_mesh;
  extractor.finish = extract_edituv_tris_finish;
  extractor.init_subdiv = extract_edituv_tris_init_subdiv;
  extractor.iter_subdiv_bm = extract_edituv_tris_iter_subdiv_bm;
  extractor.iter_subdiv_mesh = extract_edituv_tris_iter_subdiv_mesh;
  extractor.finish_subdiv = extract_edituv_tris_finish_subdiv;
  extractor.data_type = MR_DATA_NONE;
  extractor.data_size = sizeof(MeshExtract_EditUvElem_Data);
  extractor.use_threading = false;
  extractor.mesh_buffer_offset = offsetof(MeshBufferList, ibo.edituv_tris);
  return extractor;
}

/** \} */

/* ---------------------------------------------------------------------- */
/** \name Extract Edit UV Line Indices around faces
 * \{ */

static void extract_edituv_lines_init(const MeshRenderData &mr,
                                      MeshBatchCache & /*cache*/,
                                      void * /*ibo*/,
                                      void *tls_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(tls_data);
  GPU_indexbuf_init(&data->elb, GPU_PRIM_LINES, mr.corners_num, mr.corners_num);
  data->sync_selection = (mr.toolsettings->uv_flag & UV_SYNC_SELECTION) != 0;
}

BLI_INLINE void edituv_edge_add(
    MeshExtract_EditUvElem_Data *data, bool hidden, bool selected, int v1, int v2)
{
  if (!hidden && (data->sync_selection || selected)) {
    GPU_indexbuf_add_line_verts(&data->elb, v1, v2);
  }
}

static void extract_edituv_lines_iter_face_bm(const MeshRenderData & /*mr*/,
                                              const BMFace *f,
                                              const int /*f_index*/,
                                              void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  BMLoop *l_iter, *l_first;
  l_iter = l_first = BM_FACE_FIRST_LOOP(f);
  do {
    const int l_index = BM_elem_index_get(l_iter);

    edituv_edge_add(data,
                    BM_elem_flag_test_bool(f, BM_ELEM_HIDDEN),
                    BM_elem_flag_test_bool(f, BM_ELEM_SELECT),
                    l_index,
                    BM_elem_index_get(l_iter->next));
  } while ((l_iter = l_iter->next) != l_first);
}

static void extract_edituv_lines_iter_face_mesh(const MeshRenderData &mr,
                                                const int face_index,
                                                void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  const IndexRange face = mr.faces[face_index];

  bool mp_hidden, mp_select;
  if (mr.bm) {
    const BMFace *efa = bm_original_face_get(mr, face_index);
    mp_hidden = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_HIDDEN) : true;
    mp_select = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_SELECT) : false;
  }
  else {
    mp_hidden = mr.hide_poly.is_empty() ? false : mr.hide_poly[face_index];
    mp_select = !mr.select_poly.is_empty() && mr.select_poly[face_index];
  }

  for (const int corner : face) {
    const int edge = mr.corner_edges[corner];
    const int corner_next = bke::mesh::face_corner_next(face, corner);

    const bool real_edge = (mr.orig_index_edge == nullptr ||
                            mr.orig_index_edge[edge] != ORIGINDEX_NONE);
    edituv_edge_add(data, mp_hidden || !real_edge, mp_select, corner, corner_next);
  }
}

static void extract_edituv_lines_finish(const MeshRenderData & /*mr*/,
                                        MeshBatchCache & /*cache*/,
                                        void *buf,
                                        void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  gpu::IndexBuf *ibo = static_cast<gpu::IndexBuf *>(buf);
  GPU_indexbuf_build_in_place(&data->elb, ibo);
}

static void extract_edituv_lines_init_subdiv(const DRWSubdivCache &subdiv_cache,
                                             const MeshRenderData &mr,
                                             MeshBatchCache & /*cache*/,
                                             void * /*buf*/,
                                             void *tls_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(tls_data);
  GPU_indexbuf_init(
      &data->elb, GPU_PRIM_LINES, subdiv_cache.num_subdiv_loops, subdiv_cache.num_subdiv_loops);
  data->sync_selection = (mr.toolsettings->uv_flag & UV_SYNC_SELECTION) != 0;
}

static void extract_edituv_lines_iter_subdiv_bm(const DRWSubdivCache &subdiv_cache,
                                                const MeshRenderData &mr,
                                                void *_data,
                                                uint subdiv_quad_index,
                                                const BMFace *coarse_face)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  int *subdiv_loop_edge_index = (int *)GPU_vertbuf_get_data(subdiv_cache.edges_orig_index);

  const bool mp_hidden = BM_elem_flag_test_bool(coarse_face, BM_ELEM_HIDDEN);
  const bool mp_select = BM_elem_flag_test_bool(coarse_face, BM_ELEM_SELECT);

  uint start_loop_idx = subdiv_quad_index * 4;
  uint end_loop_idx = (subdiv_quad_index + 1) * 4;
  for (uint loop_idx = start_loop_idx; loop_idx < end_loop_idx; loop_idx++) {
    const int edge_origindex = subdiv_loop_edge_index[loop_idx];
    const bool real_edge = (edge_origindex != -1 &&
                            (mr.orig_index_edge == nullptr ||
                             mr.orig_index_edge[edge_origindex] != ORIGINDEX_NONE));
    edituv_edge_add(data,
                    mp_hidden || !real_edge,
                    mp_select,
                    loop_idx,
                    (loop_idx + 1 == end_loop_idx) ? start_loop_idx : (loop_idx + 1));
  }
}

static void extract_edituv_lines_iter_subdiv_mesh(const DRWSubdivCache &subdiv_cache,
                                                  const MeshRenderData &mr,
                                                  void *_data,
                                                  uint subdiv_quad_index,
                                                  const int coarse_face_index)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  int *subdiv_loop_edge_index = (int *)GPU_vertbuf_get_data(subdiv_cache.edges_orig_index);
  bool mp_hidden, mp_select;
  if (mr.bm) {
    const BMFace *efa = bm_original_face_get(mr, coarse_face_index);
    mp_hidden = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_HIDDEN) : true;
    mp_select = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_SELECT) : false;
  }
  else {
    mp_hidden = mr.hide_poly.is_empty() ? false : mr.hide_poly[coarse_face_index];
    mp_select = !mr.select_poly.is_empty() && mr.select_poly[coarse_face_index];
  }

  uint start_loop_idx = subdiv_quad_index * 4;
  uint end_loop_idx = (subdiv_quad_index + 1) * 4;
  for (uint loop_idx = start_loop_idx; loop_idx < end_loop_idx; loop_idx++) {
    const int edge_origindex = subdiv_loop_edge_index[loop_idx];
    const bool real_edge = (edge_origindex != -1 &&
                            (mr.orig_index_edge == nullptr ||
                             mr.orig_index_edge[edge_origindex] != ORIGINDEX_NONE));
    edituv_edge_add(data,
                    mp_hidden || !real_edge,
                    mp_select,
                    loop_idx,
                    (loop_idx + 1 == end_loop_idx) ? start_loop_idx : (loop_idx + 1));
  }
}

static void extract_edituv_lines_finish_subdiv(const DRWSubdivCache & /*subdiv_cache*/,
                                               const MeshRenderData & /*mr*/,
                                               MeshBatchCache & /*cache*/,
                                               void *buf,
                                               void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  gpu::IndexBuf *ibo = static_cast<gpu::IndexBuf *>(buf);
  GPU_indexbuf_build_in_place(&data->elb, ibo);
}

constexpr MeshExtract create_extractor_edituv_lines()
{
  MeshExtract extractor = {nullptr};
  extractor.init = extract_edituv_lines_init;
  extractor.iter_face_bm = extract_edituv_lines_iter_face_bm;
  extractor.iter_face_mesh = extract_edituv_lines_iter_face_mesh;
  extractor.finish = extract_edituv_lines_finish;
  extractor.init_subdiv = extract_edituv_lines_init_subdiv;
  extractor.iter_subdiv_bm = extract_edituv_lines_iter_subdiv_bm;
  extractor.iter_subdiv_mesh = extract_edituv_lines_iter_subdiv_mesh;
  extractor.finish_subdiv = extract_edituv_lines_finish_subdiv;
  extractor.data_type = MR_DATA_NONE;
  extractor.data_size = sizeof(MeshExtract_EditUvElem_Data);
  extractor.use_threading = false;
  extractor.mesh_buffer_offset = offsetof(MeshBufferList, ibo.edituv_lines);
  return extractor;
}

/** \} */

/* ---------------------------------------------------------------------- */
/** \name Extract Edit UV Points Indices
 * \{ */

static void extract_edituv_points_init(const MeshRenderData &mr,
                                       MeshBatchCache & /*cache*/,
                                       void * /*ibo*/,
                                       void *tls_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(tls_data);
  GPU_indexbuf_init(&data->elb, GPU_PRIM_POINTS, mr.corners_num, mr.corners_num);
  data->sync_selection = (mr.toolsettings->uv_flag & UV_SYNC_SELECTION) != 0;
}

BLI_INLINE void edituv_point_add(MeshExtract_EditUvElem_Data *data,
                                 bool hidden,
                                 bool selected,
                                 int v1)
{
  if (!hidden && (data->sync_selection || selected)) {
    GPU_indexbuf_add_point_vert(&data->elb, v1);
  }
}

static void extract_edituv_points_iter_face_bm(const MeshRenderData & /*mr*/,
                                               const BMFace *f,
                                               const int /*f_index*/,
                                               void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  BMLoop *l_iter, *l_first;
  l_iter = l_first = BM_FACE_FIRST_LOOP(f);
  do {
    const int l_index = BM_elem_index_get(l_iter);

    edituv_point_add(
        data, BM_elem_flag_test(f, BM_ELEM_HIDDEN), BM_elem_flag_test(f, BM_ELEM_SELECT), l_index);
  } while ((l_iter = l_iter->next) != l_first);
}

static void extract_edituv_points_iter_face_mesh(const MeshRenderData &mr,
                                                 const int face_index,
                                                 void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);

  const BMFace *efa = bm_original_face_get(mr, face_index);
  const bool mp_hidden = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_HIDDEN) : true;
  const bool mp_select = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_SELECT) : false;

  for (const int corner : mr.faces[face_index]) {
    const int vert = mr.corner_verts[corner];

    const bool real_vert = !mr.orig_index_vert || mr.orig_index_vert[vert] != ORIGINDEX_NONE;
    edituv_point_add(data, mp_hidden || !real_vert, mp_select, corner);
  }
}

static void extract_edituv_points_finish(const MeshRenderData & /*mr*/,
                                         MeshBatchCache & /*cache*/,
                                         void *buf,
                                         void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  gpu::IndexBuf *ibo = static_cast<gpu::IndexBuf *>(buf);
  GPU_indexbuf_build_in_place(&data->elb, ibo);
}

static void extract_edituv_points_init_subdiv(const DRWSubdivCache &subdiv_cache,
                                              const MeshRenderData &mr,
                                              MeshBatchCache & /*cache*/,
                                              void * /*buf*/,
                                              void *tls_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(tls_data);
  GPU_indexbuf_init(
      &data->elb, GPU_PRIM_POINTS, subdiv_cache.num_subdiv_loops, subdiv_cache.num_subdiv_loops);
  data->sync_selection = (mr.toolsettings->uv_flag & UV_SYNC_SELECTION) != 0;
}

static void extract_edituv_points_iter_subdiv_bm(const DRWSubdivCache &subdiv_cache,
                                                 const MeshRenderData & /*mr*/,
                                                 void *_data,
                                                 uint subdiv_quad_index,
                                                 const BMFace *coarse_quad)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  int *subdiv_loop_vert_index = (int *)GPU_vertbuf_get_data(subdiv_cache.verts_orig_index);

  uint start_loop_idx = subdiv_quad_index * 4;
  uint end_loop_idx = (subdiv_quad_index + 1) * 4;
  for (uint i = start_loop_idx; i < end_loop_idx; i++) {
    const int vert_origindex = subdiv_loop_vert_index[i];
    edituv_point_add(data,
                     (BM_elem_flag_test(coarse_quad, BM_ELEM_HIDDEN) || vert_origindex == -1),
                     BM_elem_flag_test(coarse_quad, BM_ELEM_SELECT) != 0,
                     i);
  }
}

static void extract_edituv_points_iter_subdiv_mesh(const DRWSubdivCache &subdiv_cache,
                                                   const MeshRenderData &mr,
                                                   void *_data,
                                                   uint subdiv_quad_index,
                                                   const int coarse_quad_index)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  int *subdiv_loop_vert_index = (int *)GPU_vertbuf_get_data(subdiv_cache.verts_orig_index);

  const BMFace *efa = bm_original_face_get(mr, coarse_quad_index);
  const bool mp_hidden = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_HIDDEN) : true;
  const bool mp_select = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_SELECT) : false;

  uint start_loop_idx = subdiv_quad_index * 4;
  uint end_loop_idx = (subdiv_quad_index + 1) * 4;
  for (uint i = start_loop_idx; i < end_loop_idx; i++) {
    const int vert_origindex = subdiv_loop_vert_index[i];
    const bool real_vert = !mr.orig_index_vert ||
                           (vert_origindex != -1 &&
                            mr.orig_index_vert[vert_origindex] != ORIGINDEX_NONE);
    edituv_point_add(data, mp_hidden || !real_vert, mp_select, i);
  }
}

static void extract_edituv_points_finish_subdiv(const DRWSubdivCache & /*subdiv_cache*/,
                                                const MeshRenderData & /*mr*/,
                                                MeshBatchCache & /*cache*/,
                                                void *buf,
                                                void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  gpu::IndexBuf *ibo = static_cast<gpu::IndexBuf *>(buf);
  GPU_indexbuf_build_in_place(&data->elb, ibo);
}

constexpr MeshExtract create_extractor_edituv_points()
{
  MeshExtract extractor = {nullptr};
  extractor.init = extract_edituv_points_init;
  extractor.iter_face_bm = extract_edituv_points_iter_face_bm;
  extractor.iter_face_mesh = extract_edituv_points_iter_face_mesh;
  extractor.finish = extract_edituv_points_finish;
  extractor.init_subdiv = extract_edituv_points_init_subdiv;
  extractor.iter_subdiv_bm = extract_edituv_points_iter_subdiv_bm;
  extractor.iter_subdiv_mesh = extract_edituv_points_iter_subdiv_mesh;
  extractor.finish_subdiv = extract_edituv_points_finish_subdiv;
  extractor.data_type = MR_DATA_NONE;
  extractor.data_size = sizeof(MeshExtract_EditUvElem_Data);
  extractor.use_threading = false;
  extractor.mesh_buffer_offset = offsetof(MeshBufferList, ibo.edituv_points);
  return extractor;
}

/** \} */

/* ---------------------------------------------------------------------- */
/** \name Extract Edit UV Face-dots Indices
 * \{ */

static void extract_edituv_fdots_init(const MeshRenderData &mr,
                                      MeshBatchCache & /*cache*/,
                                      void * /*ibo*/,
                                      void *tls_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(tls_data);
  GPU_indexbuf_init(&data->elb, GPU_PRIM_POINTS, mr.faces_num, mr.faces_num);
  data->sync_selection = (mr.toolsettings->uv_flag & UV_SYNC_SELECTION) != 0;
}

BLI_INLINE void edituv_facedot_add(MeshExtract_EditUvElem_Data *data,
                                   bool hidden,
                                   bool selected,
                                   int face_index)
{
  if (!hidden && (data->sync_selection || selected)) {
    GPU_indexbuf_set_point_vert(&data->elb, face_index, face_index);
  }
  else {
    GPU_indexbuf_set_point_restart(&data->elb, face_index);
  }
}

static void extract_edituv_fdots_iter_face_bm(const MeshRenderData & /*mr*/,
                                              const BMFace *f,
                                              const int f_index,
                                              void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  edituv_facedot_add(data,
                     BM_elem_flag_test_bool(f, BM_ELEM_HIDDEN),
                     BM_elem_flag_test_bool(f, BM_ELEM_SELECT),
                     f_index);
}

static void extract_edituv_fdots_iter_face_mesh(const MeshRenderData &mr,
                                                const int face_index,
                                                void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);

  const BMFace *efa = bm_original_face_get(mr, face_index);
  const bool mp_hidden = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_HIDDEN) : true;
  const bool mp_select = (efa) ? BM_elem_flag_test_bool(efa, BM_ELEM_SELECT) : false;

  if (mr.use_subsurf_fdots) {
    const BitSpan facedot_tags = mr.mesh->runtime->subsurf_face_dot_tags;

    for (const int corner : mr.faces[face_index]) {
      const int vert = mr.corner_verts[corner];

      const bool real_fdot = !mr.orig_index_face ||
                             (mr.orig_index_face[face_index] != ORIGINDEX_NONE);
      const bool subd_fdot = facedot_tags[vert];
      edituv_facedot_add(data, mp_hidden || !real_fdot || !subd_fdot, mp_select, face_index);
    }
  }
  else {
    const bool real_fdot = !mr.orig_index_face ||
                           (mr.orig_index_face[face_index] != ORIGINDEX_NONE);
    edituv_facedot_add(data, mp_hidden || !real_fdot, mp_select, face_index);
  }
}

static void extract_edituv_fdots_finish(const MeshRenderData & /*mr*/,
                                        MeshBatchCache & /*cache*/,
                                        void *buf,
                                        void *_data)
{
  MeshExtract_EditUvElem_Data *data = static_cast<MeshExtract_EditUvElem_Data *>(_data);
  gpu::IndexBuf *ibo = static_cast<gpu::IndexBuf *>(buf);
  GPU_indexbuf_build_in_place(&data->elb, ibo);
}

constexpr MeshExtract create_extractor_edituv_fdots()
{
  MeshExtract extractor = {nullptr};
  extractor.init = extract_edituv_fdots_init;
  extractor.iter_face_bm = extract_edituv_fdots_iter_face_bm;
  extractor.iter_face_mesh = extract_edituv_fdots_iter_face_mesh;
  extractor.finish = extract_edituv_fdots_finish;
  extractor.data_type = MR_DATA_NONE;
  extractor.data_size = sizeof(MeshExtract_EditUvElem_Data);
  extractor.use_threading = false;
  extractor.mesh_buffer_offset = offsetof(MeshBufferList, ibo.edituv_fdots);
  return extractor;
}

/** \} */

const MeshExtract extract_edituv_tris = create_extractor_edituv_tris();
const MeshExtract extract_edituv_lines = create_extractor_edituv_lines();
const MeshExtract extract_edituv_points = create_extractor_edituv_points();
const MeshExtract extract_edituv_fdots = create_extractor_edituv_fdots();

}  // namespace blender::draw
