open_toontown_panda3d/dtool/src/interrogate/functionWriterPtrFromPython...

118 lines
4.2 KiB
C++

// Filename: functionWriterPtrFromPython.cxx
// Created by: drose (14Sep01)
//
////////////////////////////////////////////////////////////////////
//
// PANDA 3D SOFTWARE
// Copyright (c) 2001 - 2004, Disney Enterprises, Inc. All rights reserved
//
// All use of this software is subject to the terms of the Panda 3d
// Software license. You should have received a copy of this license
// along with this source code; you will also find a current copy of
// the license at http://etc.cmu.edu/panda3d/docs/license/ .
//
// To contact the maintainers of this program write to
// panda3d-general@lists.sourceforge.net .
//
////////////////////////////////////////////////////////////////////
#include "functionWriterPtrFromPython.h"
#include "typeManager.h"
#include "interrogateBuilder.h"
#include "interrogate.h"
#include "cppPointerType.h"
////////////////////////////////////////////////////////////////////
// Function: FunctionWriterPtrFromPython::Constructor
// Access: Public
// Description:
////////////////////////////////////////////////////////////////////
FunctionWriterPtrFromPython::
FunctionWriterPtrFromPython(CPPType *type) {
_type = TypeManager::unwrap_const(TypeManager::unwrap_pointer(type));
_name =
"from_python_" +
InterrogateBuilder::clean_identifier(_type->get_local_name(&parser));
_pointer_type = new CPPPointerType(_type);
}
////////////////////////////////////////////////////////////////////
// Function: FunctionWriterPtrFromPython::Destructor
// Access: Public
// Description:
////////////////////////////////////////////////////////////////////
FunctionWriterPtrFromPython::
~FunctionWriterPtrFromPython() {
delete _pointer_type;
}
////////////////////////////////////////////////////////////////////
// Function: FunctionWriterPtrFromPython::write_prototype
// Access: Public, Virtual
// Description: Outputs the prototype for the function.
////////////////////////////////////////////////////////////////////
void FunctionWriterPtrFromPython::
write_prototype(ostream &out) {
CPPType *ppointer = new CPPPointerType(_pointer_type);
out << "static int " << _name << "(PyObject *obj, ";
ppointer->output_instance(out, "addr", &parser);
out << ");\n";
delete ppointer;
}
////////////////////////////////////////////////////////////////////
// Function: FunctionWriterPtrFromPython::write_code
// Access: Public, Virtual
// Description: Outputs the code for the function.
////////////////////////////////////////////////////////////////////
void FunctionWriterPtrFromPython::
write_code(ostream &out) {
CPPType *ppointer = new CPPPointerType(_pointer_type);
out << "static int\n"
<< _name << "(PyObject *obj, ";
ppointer->output_instance(out, "addr", &parser);
out << ") {\n"
<< " if (obj != (PyObject *)NULL && PyInstance_Check(obj)) {\n"
// << " PyClassObject *in_class = ((PyInstanceObject *)obj)->in_class;\n"
<< " PyObject *in_dict = ((PyInstanceObject *)obj)->in_dict;\n"
<< " if (in_dict != (PyObject *)NULL && PyDict_Check(in_dict)) {\n"
<< " PyObject *thisobj = PyDict_GetItemString(in_dict, \"this\");\n"
<< " if (thisobj != (PyObject *)NULL && PyInt_Check(thisobj)) {\n"
<< " (*addr) = ("
<< _pointer_type->get_local_name(&parser) << ")PyInt_AsLong(thisobj);\n"
<< " return 1;\n"
<< " }\n"
<< " }\n"
<< " }\n"
<< " return 0;\n"
<< "}\n\n";
delete ppointer;
}
////////////////////////////////////////////////////////////////////
// Function: FunctionWriterPtrFromPython::get_type
// Access: Public
// Description: Returns the type that represents the actual data type.
////////////////////////////////////////////////////////////////////
CPPType *FunctionWriterPtrFromPython::
get_type() const {
return _type;
}
////////////////////////////////////////////////////////////////////
// Function: FunctionWriterPtrFromPython::get_pointer_type
// Access: Public
// Description: Returns the type that represents a pointer to the
// data type.
////////////////////////////////////////////////////////////////////
CPPType *FunctionWriterPtrFromPython::
get_pointer_type() const {
return _pointer_type;
}