110 lines
2.7 KiB
C++
110 lines
2.7 KiB
C++
/**
|
|
* 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."
|
|
*
|
|
* @file texturePool_ext.cxx
|
|
* @author Derzsi Daniel
|
|
* @date 2020-06-24
|
|
*/
|
|
|
|
#include "texturePool_ext.h"
|
|
|
|
#ifdef HAVE_PYTHON
|
|
|
|
#include "pythonTexturePoolFilter.h"
|
|
#include "texturePoolFilter.h"
|
|
|
|
/**
|
|
* Registers a texture pool filter that is implemented in Python.
|
|
*/
|
|
bool Extension<TexturePool>::
|
|
register_filter(PyObject *tex_filter) {
|
|
if (find_existing_filter(tex_filter) != nullptr) {
|
|
PyObject *repr = PyObject_Repr(tex_filter);
|
|
|
|
gobj_cat->warning()
|
|
<< "Attempted to register Python texture filter " << PyUnicode_AsUTF8(repr)
|
|
<< " more than once.\n";
|
|
return false;
|
|
}
|
|
|
|
PythonTexturePoolFilter *filter = new PythonTexturePoolFilter();
|
|
|
|
if (filter->init(tex_filter)) {
|
|
return _this->ns_register_filter(filter);
|
|
}
|
|
|
|
delete filter;
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* If the given Python texture pool filter is registered, unregisters it.
|
|
*/
|
|
bool Extension<TexturePool>::
|
|
unregister_filter(PyObject *tex_filter) {
|
|
// Keep looping until we've removed all instances of it.
|
|
bool unregistered = false;
|
|
|
|
while (true) {
|
|
TexturePoolFilter *filter = find_existing_filter(tex_filter);
|
|
|
|
if (filter == nullptr) {
|
|
// Last filter has been removed.
|
|
break;
|
|
}
|
|
|
|
unregistered = _this->ns_unregister_filter(filter);
|
|
}
|
|
|
|
if (!unregistered) {
|
|
PyObject *repr = PyObject_Repr(tex_filter);
|
|
|
|
gobj_cat->warning()
|
|
<< "Attempted to unregister Python texture filter " << PyUnicode_AsUTF8(repr)
|
|
<< " which was not registered.\n";
|
|
}
|
|
|
|
return unregistered;
|
|
}
|
|
|
|
/**
|
|
* If the given Python texture pool filter is registered, returns true.
|
|
*/
|
|
bool Extension<TexturePool>::
|
|
is_filter_registered(PyObject *tex_filter) {
|
|
return find_existing_filter(tex_filter) != nullptr;
|
|
}
|
|
|
|
/**
|
|
* Looks for a texture pool filter that is using the
|
|
* given Python implementation of the texture filter.
|
|
*
|
|
* Returns nullptr if none has been found.
|
|
*/
|
|
TexturePoolFilter *Extension<TexturePool>::
|
|
find_existing_filter(PyObject *tex_filter) {
|
|
size_t num_filters = _this->get_num_filters();
|
|
|
|
for (size_t i = 0; i < num_filters; ++i) {
|
|
TexturePoolFilter *filter = _this->get_filter(i);
|
|
|
|
if (filter != nullptr && filter->is_of_type(PythonTexturePoolFilter::get_class_type())) {
|
|
PythonTexturePoolFilter *py_filter = (PythonTexturePoolFilter *)filter;
|
|
|
|
if (py_filter->_entry_point == tex_filter) {
|
|
return filter;
|
|
}
|
|
}
|
|
}
|
|
|
|
// No filter found.
|
|
return nullptr;
|
|
}
|
|
|
|
#endif
|