#ifndef PYTHONTASK_H

#define PYTHONTASK_H

#include "pandabase.h"

#include "asyncTask.h"

#ifdef HAVE_PYTHON

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;

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);

void register_to_owner();

void unregister_from_owner();

void call_owner_method(const char *method_name);

void call_function(PyObject *function);

PyObject *_function;

PyObject *_args;

bool _append_task;

PyObject *_upon_death;

PyObject *_owner;

bool _registered_to_owner;

PyObject *_dict;

PyObject *_generator;

static TypeHandle get_class_type() {

return _type_handle;

}

static void init_type() {

}

virtual TypeHandle get_type() const {

return get_class_type();

}

virtual TypeHandle force_init_type() {init_type(); return get_class_type();}

static TypeHandle _type_handle;

};

#include "pythonTask.I"

#endif // HAVE_PYTHON

#endif