97 lines
2.5 KiB
C++
97 lines
2.5 KiB
C++
// Filename: pythonTask.h
|
|
// Created by: drose (16Sep08)
|
|
//
|
|
////////////////////////////////////////////////////////////////////
|
|
//
|
|
// 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."
|
|
//
|
|
////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef PYTHONTASK_H
|
|
#define PYTHONTASK_H
|
|
|
|
#include "pandabase.h"
|
|
|
|
#include "asyncTask.h"
|
|
|
|
#ifdef HAVE_PYTHON
|
|
////////////////////////////////////////////////////////////////////
|
|
// Class : PythonTask
|
|
// Description : This class exists to allow association of a Python
|
|
// function with the AsyncTaskManager.
|
|
////////////////////////////////////////////////////////////////////
|
|
class EXPCL_PANDA_PIPELINE PythonTask : public AsyncTask {
|
|
PUBLISHED:
|
|
PythonTask(PyObject *function = Py_None, const string &name = string());
|
|
virtual ~PythonTask();
|
|
ALLOC_DELETED_CHAIN(PythonTask);
|
|
|
|
void set_function(PyObject *function);
|
|
PyObject *get_function();
|
|
|
|
void set_args(PyObject *args, bool append_task);
|
|
PyObject *get_args();
|
|
|
|
void set_upon_death(PyObject *upon_death);
|
|
PyObject *get_upon_death();
|
|
|
|
void set_owner(PyObject *owner);
|
|
PyObject *get_owner();
|
|
|
|
int __setattr__(const string &attr_name, PyObject *v);
|
|
int __setattr__(const string &attr_name);
|
|
PyObject *__getattr__(const string &attr_name) const;
|
|
|
|
protected:
|
|
virtual bool is_runnable();
|
|
virtual DoneStatus do_task();
|
|
DoneStatus do_python_task();
|
|
virtual void upon_birth(AsyncTaskManager *manager);
|
|
virtual void upon_death(AsyncTaskManager *manager, bool clean_exit);
|
|
|
|
private:
|
|
void register_to_owner();
|
|
void unregister_from_owner();
|
|
void call_owner_method(const char *method_name);
|
|
void call_function(PyObject *function);
|
|
|
|
private:
|
|
PyObject *_function;
|
|
PyObject *_args;
|
|
bool _append_task;
|
|
PyObject *_upon_death;
|
|
PyObject *_owner;
|
|
PyObject *_dict;
|
|
|
|
PyObject *_generator;
|
|
|
|
public:
|
|
static TypeHandle get_class_type() {
|
|
return _type_handle;
|
|
}
|
|
static void init_type() {
|
|
AsyncTask::init_type();
|
|
register_type(_type_handle, "PythonTask",
|
|
AsyncTask::get_class_type());
|
|
}
|
|
virtual TypeHandle get_type() const {
|
|
return get_class_type();
|
|
}
|
|
virtual TypeHandle force_init_type() {init_type(); return get_class_type();}
|
|
|
|
private:
|
|
static TypeHandle _type_handle;
|
|
};
|
|
|
|
#include "pythonTask.I"
|
|
|
|
#endif // HAVE_PYTHON
|
|
|
|
#endif
|
|
|