Phoenix/src/wxpybuffer.h

//--------------------------------------------------------------------------
// Name:        src/wxpybuffer.h
// Purpose:     A simple class to hold a pointer and a size.  See 
//              wxpybuffer.sip for code that converts from a Python buffer
//              object as a MappedType.
//
// Author:      Robin Dunn
//
// Created:     26-Apr-2012
// Copyright:   (c) 2013 by Total Control Software
// Licence:     wxWindows license
//--------------------------------------------------------------------------

#ifndef WXPYBUFFER_H
#define WXPYBUFFER_H


class wxPyBuffer
{
public:
    wxPyBuffer() : m_ptr(NULL), m_len(0) {}

    // Initialize this object's pointer and length from a PyObject supporting
    // the Python buffer protocol.  Raises a TypeError if the object can not 
    // be used as a buffer.
    bool create(PyObject* obj) {
        int rv = PyObject_AsReadBuffer(obj, (const void**)&m_ptr, &m_len);
        return rv != -1;
    }
        
        
    // Ensure that the buffer's size is the expected size.  Raises a 
    // Python ValueError exception and returns false if not.
    bool checkSize(Py_ssize_t expectedSize) {
        if (m_len < expectedSize) {
            wxPyErr_SetString(PyExc_ValueError, "Invalid data buffer size.");
            return false;
        }
        return true;
    }

    // Make a simple C copy of the data buffer.  Malloc is used because 
    // the wxAPIs this is used with will later free() the pointer.  Raises 
    // a Python exception if there is an allocation error.
    void* copy() {
        void* ptr = malloc(m_len);
        if (ptr == NULL) {
            wxPyBLOCK_THREADS(PyErr_NoMemory());
            return NULL;
        }            
        memcpy(ptr, m_ptr, m_len);
        return ptr;
    }

    void*       m_ptr;
    Py_ssize_t  m_len;
};

#endif
//--------------------------------------------------------------------------