CWE-ou-minkata/Sources/Plasma/FeatureLib/pfPython/pyGrassShaderGlue.cpp

/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
#include "pyGrassShader.h"

#include <python.h>

// glue functions
PYTHON_CLASS_DEFINITION(ptGrassShader, pyGrassShader);

PYTHON_DEFAULT_NEW_DEFINITION(ptGrassShader, pyGrassShader)
PYTHON_DEFAULT_DEALLOC_DEFINITION(ptGrassShader)

PYTHON_INIT_DEFINITION(ptGrassShader, args, keywords)
{
    PyObject *keyObject = NULL;
    if (!PyArg_ParseTuple(args, "O", &keyObject))
    {
        PyErr_SetString(PyExc_TypeError, "init expects a ptKey");
        PYTHON_RETURN_INIT_ERROR;
    }
    if (!pyKey::Check(keyObject))
    {
        PyErr_SetString(PyExc_TypeError, "init expects a ptKey");
        PYTHON_RETURN_INIT_ERROR;
    }

    pyKey *key = pyKey::ConvertFrom(keyObject);
    self->fThis->SetKey(key->getKey());

    PYTHON_RETURN_INIT_OK;
}

PYTHON_METHOD_DEFINITION(ptGrassShader, setWaveDistortion, args)
{
    int waveNum;
    PyObject *tupleObject = NULL;
    if (!PyArg_ParseTuple(args, "iO", &waveNum, &tupleObject))
    {
        PyErr_SetString(PyExc_TypeError, "setWaveDistortion expects a integer and tuple of floats");
        PYTHON_RETURN_ERROR;
    }
    if (!PyTuple_Check(tupleObject))
    {
        PyErr_SetString(PyExc_TypeError, "setWaveDistortion expects a integer and tuple of floats");
        PYTHON_RETURN_ERROR;
    }

    int len = PyTuple_Size(tupleObject);
    std::vector<hsScalar> vecArgs;
    for (int curArg = 0; curArg < len; curArg++)
    {
        PyObject *arg = PyTuple_GetItem(tupleObject, curArg);
        if (!PyFloat_Check(arg))
        {
            PyErr_SetString(PyExc_TypeError, "setWaveDistortion expects a integer and tuple of floats");
            PYTHON_RETURN_ERROR;
        }
        vecArgs.push_back((hsScalar)PyFloat_AsDouble(arg));
    }

    self->fThis->SetWaveDistortion(waveNum, vecArgs);
    PYTHON_RETURN_NONE;
}

PYTHON_METHOD_DEFINITION(ptGrassShader, setWaveDirection, args)
{
    int waveNum;
    PyObject *tupleObject = NULL;
    if (!PyArg_ParseTuple(args, "iO", &waveNum, &tupleObject))
    {
        PyErr_SetString(PyExc_TypeError, "setWaveDirection expects a integer and tuple of floats");
        PYTHON_RETURN_ERROR;
    }
    if (!PyTuple_Check(tupleObject))
    {
        PyErr_SetString(PyExc_TypeError, "setWaveDirection expects a integer and tuple of floats");
        PYTHON_RETURN_ERROR;
    }

    int len = PyTuple_Size(tupleObject);
    std::vector<hsScalar> vecArgs;
    for (int curArg = 0; curArg < len; curArg++)
    {
        PyObject *arg = PyTuple_GetItem(tupleObject, curArg);
        if (!PyFloat_Check(arg))
        {
            PyErr_SetString(PyExc_TypeError, "setWaveDirection expects a integer and tuple of floats");
            PYTHON_RETURN_ERROR;
        }
        vecArgs.push_back((hsScalar)PyFloat_AsDouble(arg));
    }

    self->fThis->SetWaveDirection(waveNum, vecArgs);
    PYTHON_RETURN_NONE;
}

PYTHON_METHOD_DEFINITION(ptGrassShader, setWaveSpeed, args)
{
    int waveNum;
    float speed;
    if (!PyArg_ParseTuple(args, "if", &waveNum, &speed))
    {
        PyErr_SetString(PyExc_TypeError, "setWaveSpeed expects an integer and a float");
        PYTHON_RETURN_ERROR;
    }
    self->fThis->SetWaveSpeed(waveNum, speed);
    PYTHON_RETURN_NONE;
}

PYTHON_METHOD_DEFINITION(ptGrassShader, getWaveDistortion, args)
{
    int waveNum;
    if (!PyArg_ParseTuple(args, "i", &waveNum))
    {
        PyErr_SetString(PyExc_TypeError, "getWaveDistortion expects an integer");
        PYTHON_RETURN_ERROR;
    }

    std::vector<hsScalar> vecArgs = self->fThis->GetWaveDistortion(waveNum);
    PyObject *retVal = PyTuple_New(vecArgs.size());
    for (int curArg = 0; curArg < vecArgs.size(); curArg++)
        PyTuple_SetItem(retVal, curArg, PyFloat_FromDouble((double)vecArgs[curArg]));
    return retVal;
}

PYTHON_METHOD_DEFINITION(ptGrassShader, getWaveDirection, args)
{
    int waveNum;
    if (!PyArg_ParseTuple(args, "i", &waveNum))
    {
        PyErr_SetString(PyExc_TypeError, "getWaveDirection expects an integer");
        PYTHON_RETURN_ERROR;
    }

    std::vector<hsScalar> vecArgs = self->fThis->GetWaveDirection(waveNum);
    PyObject *retVal = PyTuple_New(vecArgs.size());
    for (int curArg = 0; curArg < vecArgs.size(); curArg++)
        PyTuple_SetItem(retVal, curArg, PyFloat_FromDouble((double)vecArgs[curArg]));
    return retVal;
}

PYTHON_METHOD_DEFINITION(ptGrassShader, getWaveSpeed, args)
{
    int waveNum;
    if (!PyArg_ParseTuple(args, "i", &waveNum))
    {
        PyErr_SetString(PyExc_TypeError, "getWaveDirection expects an integer");
        PYTHON_RETURN_ERROR;
    }
    return PyFloat_FromDouble((double)self->fThis->GetWaveSpeed(waveNum));
}

PYTHON_BASIC_METHOD_DEFINITION(ptGrassShader, resetWaves, ResetWaves)

PYTHON_START_METHODS_TABLE(ptGrassShader)
    PYTHON_METHOD(ptGrassShader, setWaveDistortion, "Params: waveNum, distortion\nSets the wave waveNum's distortion as a tuple of x,y,z. waveNum must be between 0 and plGrassShaderMod::kNumWaves-1 (currently 3) inclusive"),
    PYTHON_METHOD(ptGrassShader, setWaveDirection, "Params: waveNum, direction\nSets the wave waveNum's direction as a tuple of x,y. waveNum must be between 0 and plGrassShaderMod::kNumWaves-1 (currently 3) inclusive"),
    PYTHON_METHOD(ptGrassShader, setWaveSpeed, "Params: waveNum, speed\nSets the wave waveNum's speed as a float. waveNum must be between 0 and plGrassShaderMod::kNumWaves-1 (currently 3) inclusive"),

    PYTHON_METHOD(ptGrassShader, getWaveDistortion, "Params: waveNum\nGets the wave waveNum's distortion as a tuple of x,y,z. waveNum must be between 0 and plGrassShaderMod::kNumWaves-1 (currently 3) inclusive"),
    PYTHON_METHOD(ptGrassShader, getWaveDirection, "Params: waveNum\nGets the wave waveNum's direction as a tuple of x,y. waveNum must be between 0 and plGrassShaderMod::kNumWaves-1 (currently 3) inclusive"),
    PYTHON_METHOD(ptGrassShader, getWaveSpeed, "Params: waveNum\nGets the wave waveNum's speed as a float. waveNum must be between 0 and plGrassShaderMod::kNumWaves-1 (currently 3) inclusive"),

    PYTHON_BASIC_METHOD(ptGrassShader, resetWaves, "Resets wave data to 0"),
PYTHON_END_METHODS_TABLE;

// Type structure definition
PLASMA_DEFAULT_TYPE(ptGrassShader, "Params: key\nPlasma Grass Shader class");

// required functions for PyObject interoperability
PyObject *pyGrassShader::New(plKey key)
{
    ptGrassShader *newObj = (ptGrassShader*)ptGrassShader_type.tp_new(&ptGrassShader_type, NULL, NULL);
    newObj->fThis->SetKey(key);
    return (PyObject*)newObj;
}

PYTHON_CLASS_CHECK_IMPL(ptGrassShader, pyGrassShader)
PYTHON_CLASS_CONVERT_FROM_IMPL(ptGrassShader, pyGrassShader)

///////////////////////////////////////////////////////////////////////////
//
// AddPlasmaClasses - the python module definitions
//
void pyGrassShader::AddPlasmaClasses(PyObject *m)
{
    PYTHON_CLASS_IMPORT_START(m);
    PYTHON_CLASS_IMPORT(m, ptGrassShader);
    PYTHON_CLASS_IMPORT_END(m);
}