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

/** \file
 * \ingroup bpygpu
 *
 * - Use `bpygpu_` for local API.
 * - Use `BPyGPU` for public API.
 */

#include <Python.h>

#include "GPU_capabilities.hh"

#include "gpu_py.hh"
#include "gpu_py_capabilities.hh" /* own include */

/* -------------------------------------------------------------------- */
/** \name Functions
 * \{ */

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_texture_size_get_doc,
    ".. function:: max_texture_size_get()\n"
    "\n"
    "   Get estimated maximum texture size to be able to handle.\n"
    "\n"
    "   :return: Texture size.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_texture_size_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_texture_size());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_texture_layers_get_doc,
    ".. function:: max_texture_layers_get()\n"
    "\n"
    "   Get maximum number of layers in texture.\n"
    "\n"
    "   :return: Number of layers.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_texture_layers_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_texture_layers());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_textures_get_doc,
    ".. function:: max_textures_get()\n"
    "\n"
    "   Get maximum supported texture image units used for\n"
    "   accessing texture maps from the vertex shader and the\n"
    "   fragment processor.\n"
    "\n"
    "   :return: Texture image units.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_textures_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_textures());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_textures_vert_get_doc,
    ".. function:: max_textures_vert_get()\n"
    "\n"
    "   Get maximum supported texture image units used for\n"
    "   accessing texture maps from the vertex shader.\n"
    "\n"
    "   :return: Texture image units.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_textures_vert_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_textures_vert());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_textures_geom_get_doc,
    ".. function:: max_textures_geom_get()\n"
    "\n"
    "   Get maximum supported texture image units used for\n"
    "   accessing texture maps from the geometry shader.\n"
    "\n"
    "   :return: Texture image units.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_textures_geom_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_textures_geom());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_textures_frag_get_doc,
    ".. function:: max_textures_frag_get()\n"
    "\n"
    "   Get maximum supported texture image units used for\n"
    "   accessing texture maps from the fragment shader.\n"
    "\n"
    "   :return: Texture image units.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_textures_frag_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_textures_frag());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_images_get_doc,
    ".. function:: max_images_get()\n"
    "\n"
    "   Get maximum supported number of image units.\n"
    "\n"
    "   :return: Number of image units.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_images_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_images());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_uniforms_vert_get_doc,
    ".. function:: max_uniforms_vert_get()\n"
    "\n"
    "   Get maximum number of values held in uniform variable\n"
    "   storage for a vertex shader.\n"
    "\n"
    "   :return: Number of values.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_uniforms_vert_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_uniforms_vert());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_uniforms_frag_get_doc,
    ".. function:: max_uniforms_frag_get()\n"
    "\n"
    "   Get maximum number of values held in uniform variable\n"
    "   storage for a fragment shader.\n"
    "\n"
    "   :return: Number of values.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_uniforms_frag_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_uniforms_frag());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_batch_indices_get_doc,
    ".. function:: max_batch_indices_get()\n"
    "\n"
    "   Get maximum number of vertex array indices.\n"
    "\n"
    "   :return: Number of indices.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_batch_indices_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_batch_indices());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_batch_vertices_get_doc,
    ".. function:: max_batch_vertices_get()\n"
    "\n"
    "   Get maximum number of vertex array vertices.\n"
    "\n"
    "   :return: Number of vertices.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_batch_vertices_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_batch_vertices());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_vertex_attribs_get_doc,
    ".. function:: max_vertex_attribs_get()\n"
    "\n"
    "   Get maximum number of vertex attributes accessible to\n"
    "   a vertex shader.\n"
    "\n"
    "   :return: Number of attributes.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_vertex_attribs_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_vertex_attribs());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_varying_floats_get_doc,
    ".. function:: max_varying_floats_get()\n"
    "\n"
    "   Get maximum number of varying variables used by\n"
    "   vertex and fragment shaders.\n"
    "\n"
    "   :return: Number of variables.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_varying_floats_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyLong_FromLong(GPU_max_varying_floats());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_extensions_get_doc,
    ".. function:: extensions_get()\n"
    "\n"
    "   Get supported extensions in the current context.\n"
    "\n"
    "   :return: Extensions.\n"
    "   :rtype: tuple[str]\n");
static PyObject *pygpu_extensions_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  int extensions_len = GPU_extensions_len();
  PyObject *ret = PyTuple_New(extensions_len);
  PyObject **ob_items = ((PyTupleObject *)ret)->ob_item;
  for (int i = 0; i < extensions_len; i++) {
    ob_items[i] = PyUnicode_FromString(GPU_extension_get(i));
  }

  return ret;
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_compute_shader_support_get_doc,
    ".. function:: compute_shader_support_get()\n"
    "\n"
    "   Are compute shaders supported.\n"
    "\n"
    "   :return: True when supported, False when not supported.\n"
    "   :rtype: bool\n");
static PyObject *pygpu_compute_shader_support_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  static bool deprecation_warning_issued = false;
  if (!deprecation_warning_issued) {
    PyErr_WarnEx(PyExc_DeprecationWarning,
                 "compute_shader_support_get is deprecated. All platforms have support for "
                 "compute shaders.",
                 1);
    deprecation_warning_issued = true;
  }

  return PyBool_FromLong(true);
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_shader_image_load_store_support_get_doc,
    ".. function:: shader_image_load_store_support_get()\n"
    "\n"
    "   Is image load/store supported.\n"
    "\n"
    "   :return: True when supported, False when not supported.\n"
    "   :rtype: bool\n");
static PyObject *pygpu_shader_image_load_store_support_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  static bool deprecation_warning_issued = false;
  if (!deprecation_warning_issued) {
    PyErr_WarnEx(
        PyExc_DeprecationWarning,
        "shader_image_load_store_support_get is deprecated. All platforms have support for "
        "image load store.",
        1);
    deprecation_warning_issued = true;
  }
  /* Now required to start Blender. */
  return PyBool_FromLong(true);
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_hdr_support_get_doc,
    ".. function:: hdr_support_get()\n"
    "\n"
    "  Return whether GPU backend supports High Dynamic range for viewport.\n"
    "\n"
    "   :return: HDR support available.\n"
    "   :rtype: bool\n");
static PyObject *pygpu_hdr_support_get(PyObject * /*self*/)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  return PyBool_FromLong(GPU_hdr_support());
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_work_group_count_get_doc,
    ".. function:: max_work_group_count_get(index)\n"
    "\n"
    "   Get maximum number of work groups that may be dispatched to a compute shader.\n"
    "\n"
    "   :arg index: Index of the dimension.\n"
    "   :type index: int\n"
    "   :return: Maximum number of work groups for the queried dimension.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_work_group_count_get(PyObject * /*self*/, PyObject *args)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  int index;
  if (!PyArg_ParseTuple(args, "i", &index)) {
    return nullptr;
  }

  const int max_work_group_count = GPU_max_work_group_count(index);

  return PyLong_FromLong(max_work_group_count);
}

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_max_work_group_size_get_doc,
    ".. function:: max_work_group_size_get(index)\n"
    "\n"
    "   Get maximum size of a work group that may be dispatched to a compute shader.\n"
    "\n"
    "   :arg index: Index of the dimension.\n"
    "   :type index: int\n"
    "   :return: Maximum size of a work group for the queried dimension.\n"
    "   :rtype: int\n");
static PyObject *pygpu_max_work_group_size_get(PyObject * /*self*/, PyObject *args)
{
  BPYGPU_IS_INIT_OR_ERROR_OBJ;

  int index;
  if (!PyArg_ParseTuple(args, "i", &index)) {
    return nullptr;
  }

  const int max_work_group_size = GPU_max_work_group_size(index);

  return PyLong_FromLong(max_work_group_size);
}

/** \} */

/* -------------------------------------------------------------------- */
/** \name Module
 * \{ */

#ifdef __GNUC__
#  ifdef __clang__
#    pragma clang diagnostic push
#    pragma clang diagnostic ignored "-Wcast-function-type"
#  else
#    pragma GCC diagnostic push
#    pragma GCC diagnostic ignored "-Wcast-function-type"
#  endif
#endif

static PyMethodDef pygpu_capabilities__tp_methods[] = {
    {"max_texture_size_get",
     (PyCFunction)pygpu_max_texture_size_get,
     METH_NOARGS,
     pygpu_max_texture_size_get_doc},
    {"max_texture_layers_get",
     (PyCFunction)pygpu_max_texture_layers_get,
     METH_NOARGS,
     pygpu_max_texture_layers_get_doc},
    {"max_textures_get",
     (PyCFunction)pygpu_max_textures_get,
     METH_NOARGS,
     pygpu_max_textures_get_doc},
    {"max_textures_vert_get",
     (PyCFunction)pygpu_max_textures_vert_get,
     METH_NOARGS,
     pygpu_max_textures_vert_get_doc},
    {"max_textures_geom_get",
     (PyCFunction)pygpu_max_textures_geom_get,
     METH_NOARGS,
     pygpu_max_textures_geom_get_doc},
    {"max_textures_frag_get",
     (PyCFunction)pygpu_max_textures_frag_get,
     METH_NOARGS,
     pygpu_max_textures_frag_get_doc},
    {"max_images_get", (PyCFunction)pygpu_max_images_get, METH_NOARGS, pygpu_max_images_get_doc},
    {"max_uniforms_vert_get",
     (PyCFunction)pygpu_max_uniforms_vert_get,
     METH_NOARGS,
     pygpu_max_uniforms_vert_get_doc},
    {"max_uniforms_frag_get",
     (PyCFunction)pygpu_max_uniforms_frag_get,
     METH_NOARGS,
     pygpu_max_uniforms_frag_get_doc},
    {"max_batch_indices_get",
     (PyCFunction)pygpu_max_batch_indices_get,
     METH_NOARGS,
     pygpu_max_batch_indices_get_doc},
    {"max_batch_vertices_get",
     (PyCFunction)pygpu_max_batch_vertices_get,
     METH_NOARGS,
     pygpu_max_batch_vertices_get_doc},
    {"max_vertex_attribs_get",
     (PyCFunction)pygpu_max_vertex_attribs_get,
     METH_NOARGS,
     pygpu_max_vertex_attribs_get_doc},
    {"max_varying_floats_get",
     (PyCFunction)pygpu_max_varying_floats_get,
     METH_NOARGS,
     pygpu_max_varying_floats_get_doc},
    {"extensions_get", (PyCFunction)pygpu_extensions_get, METH_NOARGS, pygpu_extensions_get_doc},

    {"compute_shader_support_get",
     (PyCFunction)pygpu_compute_shader_support_get,
     METH_NOARGS,
     pygpu_compute_shader_support_get_doc},
    {"shader_image_load_store_support_get",
     (PyCFunction)pygpu_shader_image_load_store_support_get,
     METH_NOARGS,
     pygpu_shader_image_load_store_support_get_doc},
    {"hdr_support_get",
     (PyCFunction)pygpu_hdr_support_get,
     METH_NOARGS,
     pygpu_hdr_support_get_doc},
    {
        "max_work_group_count_get",
        (PyCFunction)pygpu_max_work_group_count_get,
        METH_VARARGS,
        pygpu_max_work_group_count_get_doc,
    },
    {
        "max_work_group_size_get",
        (PyCFunction)pygpu_max_work_group_size_get,
        METH_VARARGS,
        pygpu_max_work_group_size_get_doc,
    },
    {nullptr, nullptr, 0, nullptr},
};

#ifdef __GNUC__
#  ifdef __clang__
#    pragma clang diagnostic pop
#  else
#    pragma GCC diagnostic pop
#  endif
#endif

PyDoc_STRVAR(
    /* Wrap. */
    pygpu_capabilities__tp_doc,
    "This module provides access to the GPU capabilities.");
static PyModuleDef pygpu_capabilities_module_def = {
    /*m_base*/ PyModuleDef_HEAD_INIT,
    /*m_name*/ "gpu.capabilities",
    /*m_doc*/ pygpu_capabilities__tp_doc,
    /*m_size*/ 0,
    /*m_methods*/ pygpu_capabilities__tp_methods,
    /*m_slots*/ nullptr,
    /*m_traverse*/ nullptr,
    /*m_clear*/ nullptr,
    /*m_free*/ nullptr,
};

PyObject *bpygpu_capabilities_init()
{
  PyObject *submodule;

  submodule = PyModule_Create(&pygpu_capabilities_module_def);

  return submodule;
}

/** \} */