83 lines
2.8 KiB
C++
83 lines
2.8 KiB
C++
// Filename: virtualFileSystem_ext.cxx
|
|
// Created by: rdb (12Sep13)
|
|
//
|
|
////////////////////////////////////////////////////////////////////
|
|
//
|
|
// PANDA 3D SOFTWARE
|
|
// Copyright (c) Carnegie Mellon University. All rights reserved.
|
|
//
|
|
// All use of this software is subject to the terms of the revised BSD
|
|
// license. You should have received a copy of this license along
|
|
// with this source code in a file named "LICENSE."
|
|
//
|
|
////////////////////////////////////////////////////////////////////
|
|
|
|
#include "virtualFileSystem_ext.h"
|
|
|
|
#ifdef HAVE_PYTHON
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Function: VirtualFileSystem::read_file
|
|
// Access: Published
|
|
// Description: Convenience function; returns the entire contents of
|
|
// the indicated file as a string (or as a bytes object,
|
|
// in Python 3).
|
|
//
|
|
// This variant on read_file() is implemented directly
|
|
// for Python, as a small optimization, to avoid the
|
|
// double-construction of a string object that would be
|
|
// otherwise required for the return value.
|
|
////////////////////////////////////////////////////////////////////
|
|
PyObject *Extension<VirtualFileSystem>::
|
|
read_file(const Filename &filename, bool auto_unwrap) const {
|
|
pvector<unsigned char> pv;
|
|
bool okflag = _this->read_file(filename, pv, auto_unwrap);
|
|
nassertr(okflag, NULL);
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
if (pv.empty()) {
|
|
return PyBytes_FromStringAndSize("", 0);
|
|
} else {
|
|
return PyBytes_FromStringAndSize((const char *)&pv[0], pv.size());
|
|
}
|
|
#else
|
|
if (pv.empty()) {
|
|
return PyString_FromStringAndSize("", 0);
|
|
} else {
|
|
return PyString_FromStringAndSize((const char *)&pv[0], pv.size());
|
|
}
|
|
#endif
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Function: VirtualFileSystem::write_file
|
|
// Access: Published
|
|
// Description: Convenience function; writes the entire contents of
|
|
// the indicated file as a string.
|
|
//
|
|
// This variant on write_file() is implemented directly
|
|
// for Python, as a small optimization, to avoid the
|
|
// double-construction of a string object that would be
|
|
// otherwise required.
|
|
////////////////////////////////////////////////////////////////////
|
|
PyObject *Extension<VirtualFileSystem>::
|
|
write_file(const Filename &filename, PyObject *data, bool auto_wrap) {
|
|
char *buffer;
|
|
Py_ssize_t length;
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
if (PyBytes_AsStringAndSize(data, &buffer, &length) == -1) {
|
|
return NULL;
|
|
}
|
|
#else
|
|
if (PyString_AsStringAndSize(data, &buffer, &length) == -1) {
|
|
return NULL;
|
|
}
|
|
#endif
|
|
|
|
bool result = _this->write_file(filename, (const unsigned char *)buffer, length, auto_wrap);
|
|
return PyBool_FromLong(result);
|
|
}
|
|
|
|
#endif // HAVE_PYTHON
|