1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
/* $Id$
-----------------------------------------------------------------------------
This source file is part of blendTex library
Copyright (c) 2007 The Zdeno Ash Miklas
This program is free software; you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the Free Software
Foundation; either version 2 of the License, or (at your option) any later
version.
This program is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License along with
this program; if not, write to the Free Software Foundation, Inc., 59 Temple
Place - Suite 330, Boston, MA 02111-1307, USA, or go to
http://www.gnu.org/copyleft/lesser.txt.
-----------------------------------------------------------------------------
*/
#include "PyTypeList.h"
#include <memory>
#include <vector>
#include <PyObjectPlus.h>
/// check, if type is in list
bool PyTypeList::in (PyTypeObject * type)
{
// if list exists
if (m_list.get() != NULL)
// iterate items in list
for (PyTypeListType::iterator it = m_list->begin(); it != m_list->end(); ++it)
// if item is found, return with success
if ((*it)->getType() == type) return true;
// otherwise return not found
return false;
}
/// add type to list
void PyTypeList::add (PyTypeObject * type, const char * name)
{
PyTypeListItem * typeItem;
// if list doesn't exist, create it
if (m_list.get() == NULL)
m_list.reset(new PyTypeListType());
if (!in(type))
// add new item to list
m_list->push_back(new PyTypeListItem(type, name));
}
/// prepare types
bool PyTypeList::ready (void)
{
// if list exists
if (m_list.get() != NULL)
// iterate items in list
for (PyTypeListType::iterator it = m_list->begin(); it != m_list->end(); ++it)
// if preparation failed, report it
if (PyType_Ready((*it)->getType()) < 0) return false;
// success
return true;
}
/// register types to module
void PyTypeList::reg (PyObject * module)
{
// if list exists
if (m_list.get() != NULL)
// iterate items in list
for (PyTypeListType::iterator it = m_list->begin(); it != m_list->end(); ++it)
{
// increase ref count
Py_INCREF((*it)->getType());
// add type to module
PyModule_AddObject(module, (char*)(*it)->getName(), (PyObject*)(*it)->getType());
}
}
|