Документация Python неофициальный перевод
Содержание страницы

Протокол буфераBuffer Protocol

Некоторые объекты в Python предоставляют доступ к лежащему в основе массиву памяти или буферу. К таким объектам относятся встроенные bytes и bytearray, а также некоторые типы расширений, например array.array. Сторонние библиотеки могут определять собственные типы для специальных целей, таких как обработка изображений или численный анализ.

Хотя каждый из этих типов имеет свою семантику, их объединяет общая характеристика: все они основаны на возможно большом буфере памяти. В некоторых ситуациях желательно получить прямой доступ к этому буферу без промежуточного копирования.

Python предоставляет такую возможность на уровне C в виде буферного протокола. У этого протокола есть две стороны:

  • На стороне производителя тип может экспортировать «буферный интерфейс», который позволяет объектам этого типа предоставлять информацию о своём базовом буфере. Этот интерфейс описан в разделе Buffer Object Structures;
  • На стороне потребителя доступно несколько способов получить указатель на необработанные базовые данные объекта (например, параметр метода).

Простые объекты, такие как bytes и bytearray, предоставляют свой нижележащий буфер в байт-ориентированной форме. Возможны и другие формы; например, элементы, предоставляемые array.array, могут быть многобайтовыми значениями.

Примером потребителя интерфейса буфера является метод write() файловых объектов: любой объект, способный экспортировать последовательность байтов через интерфейс буфера, может быть записан в файл. В то время как write() требует только доступа на чтение к внутреннему содержимому переданного ему объекта, другие методы, такие как readinto(), требуют доступа на запись к содержимому своего аргумента. Интерфейс буфера позволяет объектам выборочно разрешать или запрещать экспорт буферов для чтения-записи и только для чтения.

Существует два способа для потребителя буферного интерфейса получить буфер над целевым объектом:

В обоих случаях PyBuffer_Release() необходимо вызвать, когда буфер больше не нужен. Несоблюдение этого может привести к различным проблемам, таким как утечка ресурсов.

Структура буфераBuffer structure

Структуры буферов (или просто «буферы») полезны как способ предоставления бинарных данных из другого объекта программисту на Python. Их также можно использовать как механизм среза с нулевым копированием. Благодаря возможности ссылаться на блок памяти, можно довольно легко предоставить любые данные программисту на Python. Память может быть большим константным массивом в C-расширении, необработанным блоком памяти для манипуляций перед передачей в библиотеку операционной системы или использоваться для передачи структурированных данных в их родном формате в памяти.

В отличие от большинства типов данных, предоставляемых интерпретатором Python, буферы не являются указателями на PyObject, а представляют собой простые структуры C. Это позволяет создавать и копировать их очень просто. Когда требуется универсальная обёртка вокруг буфера, можно создать объект memoryview.

Краткие инструкции по написанию экспортирующего объекта см. в разделе Структуры объектов буфера. Для получения буфера см. PyObject_GetBuffer().

Py_buffer
void *buf

Указатель на начало логической структуры, описываемой полями буфера. Это может быть любое место в базовом физическом блоке памяти экспортёра. Например, при отрицательных шагах значение может указывать на конец блока памяти.

Для непрерывных массивов значение указывает на начало блока памяти.

void *obj

Новая ссылка на экспортирующий объект. Ссылка принадлежит потребителю и автоматически уменьшается на единицу и устанавливается в NULL функцией PyBuffer_Release(). Это поле эквивалентно возвращаемому значению любой стандартной функции C-API.

В особом случае для временных буферов, которые обёрнуты функциями PyMemoryView_FromBuffer() или PyBuffer_FillInfo(), это поле равно NULL. В общем случае экспортирующие объекты НЕ ДОЛЖНЫ использовать эту схему.

Py_ssize_t len

product(shape) * itemsize. Для непрерывных массивов это длина базового блока памяти. Для несплошных массивов это длина, которую имела бы логическая структура, если бы она была скопирована в непрерывное представление.

Доступ к ((char *)buf)[0] до ((char *)buf)[len-1] действителен только в том случае, если буфер был получен с помощью запроса, гарантирующего непрерывность. В большинстве случаев такой запрос будет PyBUF_SIMPLE или PyBUF_WRITABLE.

int readonly

Индикатор того, является ли буфер доступным только для чтения. Это поле управляется флагом PyBUF_WRITABLE.

Py_ssize_t itemsize

Размер одного элемента в байтах. То же, что значение struct.calcsize(), вызванной для значений format, не равных NULL.

Важное исключение: если потребитель запрашивает буфер без флага PyBUF_FORMAT, то format будет установлен в NULL, но itemsize по-прежнему содержит значение оригинального формата.

Если shape присутствует, равенство product(shape) * itemsize == len по-прежнему выполняется, и потребитель может использовать itemsize для навигации по буферу.

Если shape равен NULL в результате запроса PyBUF_SIMPLE или PyBUF_WRITABLE, потребитель должен игнорировать itemsize и считать itemsize == 1.

const char *format

Строка, завершающаяся NUL, в синтаксисе стиля модуля struct, описывающая содержимое одного элемента. Если это NULL, предполагается "B" (беззнаковые байты).

Это поле управляется флагом PyBUF_FORMAT.

int ndim

Количество измерений, которое память представляет в виде n-мерного массива. Если оно равно 0, buf указывает на единственный элемент, представляющий скаляр. В этом случае shape, strides и suboffsets ДОЛЖНЫ быть NULL.

Макрос PyBUF_MAX_NDIM ограничивает максимальное количество измерений до 64. Экспортёры ДОЛЖНЫ соблюдать это ограничение, потребители многомерных буферов должны быть способны обрабатывать до PyBUF_MAX_NDIM измерений.

Py_ssize_t *shape

Массив из Py_ssize_t длиной ndim, описывающий форму памяти как n-мерный массив. Обратите внимание, что shape[0] * ... * shape[ndim-1] * itemsize ДОЛЖНО быть равно len.

Значения shape ограничены shape[n] >= 0. Случай shape[n] == 0 требует особого внимания. См. complex arrays для получения дополнительной информации.

Массив формы доступен потребителю только для чтения.

Py_ssize_t *strides

Массив из Py_ssize_t длиной ndim, указывающий количество байтов, которое необходимо пропустить, чтобы перейти к новому элементу в каждом измерении.

Значения шага могут быть любым целым числом. Для обычных массивов шаги обычно положительны, но потребитель ДОЛЖЕН уметь обрабатывать случай strides[n] <= 0. См. сложные массивы для получения дополнительной информации.

Массив шагов доступен потребителю только для чтения.

Py_ssize_t *suboffsets

Массив из Py_ssize_t длиной ndim. Если suboffsets[n] >= 0, значения, хранящиеся вдоль n-го измерения, являются указателями, и значение suboffset определяет, сколько байтов добавить к каждому указателю после разыменования. Отрицательное значение suboffset указывает, что разыменование не должно выполняться (шаг в непрерывном блоке памяти).

Такой тип представления массива используется библиотекой Python Imaging Library (PIL). Дополнительную информацию о доступе к элементам такого массива см. в разделе сложные массивы.

Массив смещений подмассивов доступен потребителю только для чтения.

void *internal

Предназначено для внутреннего использования экспортирующим объектом. Например, экспортёр может переинтерпретировать это значение как целое число и использовать для хранения флагов о том, нужно ли освобождать массивы shape, strides и suboffsets при освобождении буфера. Потребитель НЕ ДОЛЖЕН изменять это значение.

Типы запросов буфераBuffer request types

Буферы обычно получают, отправляя запрос на буфер экспортирующему объекту через PyObject_GetBuffer(). Поскольку сложность логической структуры памяти может сильно различаться, потребитель использует аргумент flags для указания точного типа буфера, который он может обработать.

Все поля Py_buffer однозначно определяются типом запроса.

поля, не зависящие от запросаrequest-independent fields

Следующие поля не зависят от flags и должны всегда заполняться корректными значениями: obj, buf, len, itemsize, ndim.

только для чтения, форматreadonly, format

PyBUF_WRITABLE

Управляет полем readonly. Если установлен, экспортёр ДОЛЖЕН предоставить буфер для записи, иначе сообщить об ошибке. В противном случае экспортёр МОЖЕТ предоставить либо буфер только для чтения, либо для записи, но выбор ДОЛЖЕН быть согласован для всех потребителей.

PyBUF_FORMAT

Управляет полем format. Если установлен, это поле ДОЛЖНО быть заполнено корректно. В противном случае это поле ДОЛЖНО быть NULL.

PyBUF_WRITABLE может быть объединён операцией | с любыми флагами из следующего раздела. Поскольку PyBUF_SIMPLE определён как 0, PyBUF_WRITABLE может использоваться как самостоятельный флаг для запроса простого буфера для записи.

PyBUF_FORMAT может быть объединён операцией | с любыми флагами, кроме PyBUF_SIMPLE. Последний уже подразумевает формат B (беззнаковые байты).

форма, шаги, подсмещенияshape, strides, suboffsets

Флаги, управляющие логической структурой памяти, перечислены в порядке убывания сложности. Обратите внимание, что каждый флаг содержит все биты флагов, расположенных ниже него.

Запрос форма шаги подсмещения
PyBUF_INDIRECT
да да если необходимо
PyBUF_STRIDES
да да NULL
PyBUF_ND
да NULL NULL
PyBUF_SIMPLE
NULL NULL NULL

запросы непрерывностиcontiguity requests

Непрерывность по C или Fortran может быть явно запрошена, с информацией о шагах или без неё. Без информации о шагах буфер должен быть непрерывным по C.

Запрос форма шаги подсмещения непрерывность
PyBUF_C_CONTIGUOUS
да да NULL C
PyBUF_F_CONTIGUOUS
да да NULL F
PyBUF_ANY_CONTIGUOUS
да да NULL C или F
PyBUF_ND
да NULL NULL C

составные запросыcompound requests

Все возможные запросы полностью определяются некоторой комбинацией флагов из предыдущего раздела. Для удобства протокол буфера предоставляет часто используемые комбинации в виде отдельных флагов.

В следующей таблице U означает неопределённую смежность. Потребитель должен вызвать PyBuffer_IsContiguous(), чтобы определить смежность.

Запрос форма шаги подсмещения непрерывность только для чтения формат
PyBUF_FULL
да да если необходимо U 0 да
PyBUF_FULL_RO
да да если необходимо U 1 или 0 да
PyBUF_RECORDS
да да NULL U 0 да
PyBUF_RECORDS_RO
да да NULL U 1 или 0 да
PyBUF_STRIDED
да да NULL U 0 NULL
PyBUF_STRIDED_RO
да да NULL U 1 или 0 NULL
PyBUF_CONTIG
да NULL NULL C 0 NULL
PyBUF_CONTIG_RO
да NULL NULL C 1 или 0 NULL

Комплексные массивыComplex arrays

В стиле NumPy: форма и шагиNumPy-style: shape and strides

Логическая структура массивов в стиле NumPy определяется itemsize, ndim, shape и strides.

Если ndim == 0, то область памяти, на которую указывает buf, интерпретируется как скаляр размера itemsize. В таком случае и shape, и strides равны NULL.

Если strides равна NULL, массив интерпретируется как стандартный n-мерный C-массив. В противном случае потребитель должен обращаться к n-мерному массиву следующим образом:

ptr = (char *)buf + indices[0] * strides[0] + ... + indices[n-1] * strides[n-1] item = *((typeof(item) *)ptr);

Как отмечалось выше, buf может указывать на любое место внутри фактического блока памяти. Экспортёр может проверить корректность буфера с помощью этой функции:

def verify_structure(memlen, itemsize, ndim, shape, strides, offset):
    """Проверить, что параметры представляют допустимый массив в пределах
       границ выделенной памяти:
           char *mem: начало физического блока памяти
           memlen: длина физического блока памяти
           смещение: (char *)buf - mem
    """
    if offset % itemsize:
        return False
    if offset < 0 or offset+itemsize > memlen:
        return False
    if any(v % itemsize for v in strides):
        return False

    if ndim <= 0:
        return ndim == 0 and not shape and not strides
    if 0 in shape:
        return True

    imin = sum(strides[j]*(shape[j]-1) for j in range(ndim)
               if strides[j] <= 0)
    imax = sum(strides[j]*(shape[j]-1) for j in range(ndim)
               if strides[j] > 0)

    return 0 <= offset+imin and offset+imax+itemsize <= memlen

В стиле PIL: форма, шаги и suboffsetsPIL-style: shape, strides and suboffsets

В дополнение к обычным элементам, массивы в стиле PIL могут содержать указатели, по которым нужно перейти, чтобы добраться до следующего элемента в измерении. Например, обычный трёхмерный C-массив char v[2][2][3] также можно рассматривать как массив из 2 указателей на 2 двумерных массива: char (*v[2])[2][3]. В представлении со смещениями подмассивов эти два указателя могут быть встроены в начало buf, указывая на два массива char x[2][3], которые могут располагаться в любом месте памяти.

Ниже приведена функция, возвращающая указатель на элемент в N-мерном массиве, на который указывает N-мерный индекс, при наличии как ненулевых шагов (strides), так и субсмещений (suboffsets):

void *get_item_pointer(int ndim, void *buf, Py_ssize_t *strides,
                       Py_ssize_t *suboffsets, Py_ssize_t *indices) {
    char *pointer = (char*)buf;
    int i;
    for (i = 0; i < ndim; i++) {
        pointer += strides[i] * indices[i];
        if (suboffsets[i] >=0 ) {
            pointer = *((char**)pointer) + suboffsets[i];
        }
    }
    return (void*)pointer;
}