gpu_py_buffer.hh

Go to the documentation of this file.

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

 
#pragma once
 
#include <Python.h>
 
#include "BLI_sys_types.h"
 
extern PyTypeObject BPyGPU_BufferType;
 
#define BPyGPU_Buffer_Check(v) (Py_TYPE(v) == &BPyGPU_BufferType)

struct BPyGPUBuffer {
  PyObject_VAR_HEAD
  PyObject *parent;
 
  int format;
  int shape_len;
  Py_ssize_t *shape;
 
  union {
    char *as_byte;
    int *as_int;
    uint *as_uint;
    float *as_float;
 
    void *as_void;
  } buf;
};
 
[[nodiscard]] size_t bpygpu_Buffer_size(BPyGPUBuffer *buffer);
[[nodiscard]] BPyGPUBuffer *BPyGPU_Buffer_CreatePyObject(int format,
                                                         const Py_ssize_t *shape,
                                                         int shape_len,
                                                         void *buffer);