CWE-ou-minkata/Sources/Plasma/FeatureLib/pfPython/pyGrassShader.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 "plSurface/plGrassShaderMod.h"

pyGrassShader::pyGrassShader()
{
	fShaderKey = nil;
}

pyGrassShader::pyGrassShader(plKey key)
{
	fShaderKey = key;
}

pyGrassShader::pyGrassShader(pyKey& key)
{
	fShaderKey = key.getKey();
}

void pyGrassShader::SetKey(plKey key)
{
	fShaderKey = key;
}

//////////////////////////////////////////////////////////////////////
// Setter functions
//////////////////////////////////////////////////////////////////////

void pyGrassShader::SetWaveDistortion(int waveNum, const std::vector<hsScalar> & distortion)
{
	if ((waveNum < 0)||(waveNum >= plGrassShaderMod::kNumWaves))
	{
		char errmsg[256];
		sprintf(errmsg,"setWaveDistortion expects the waveNum to be between 0 and %d (inclusive)", plGrassShaderMod::kNumWaves - 1);
		PyErr_SetString(PyExc_ValueError, errmsg);
		return;
	}
	if (distortion.size() != 3)
	{
		char errmsg[256];
		sprintf(errmsg,"setWaveDistortion expects the direction to be a three-element tuple only");
		PyErr_SetString(PyExc_TypeError, errmsg);
		return;
	}
	if (fShaderKey)
	{
		plGrassShaderMod* shader = plGrassShaderMod::ConvertNoRef(fShaderKey->ObjectIsLoaded());
		if (shader)
		{
			shader->fWaves[waveNum].fDistX = distortion[0];
			shader->fWaves[waveNum].fDistY = distortion[1];
			shader->fWaves[waveNum].fDistZ = distortion[2];
			shader->RefreshWaves();
		}
	}
}

void pyGrassShader::SetWaveDirection(int waveNum, const std::vector<hsScalar> & direction)
{
	if ((waveNum < 0)||(waveNum >= plGrassShaderMod::kNumWaves))
	{
		char errmsg[256];
		sprintf(errmsg,"setWaveDirection expects the waveNum to be between 0 and %d (inclusive)", plGrassShaderMod::kNumWaves - 1);
		PyErr_SetString(PyExc_ValueError, errmsg);
		return;
	}
	if (direction.size() != 2)
	{
		char errmsg[256];
		sprintf(errmsg,"setWaveDirection expects the direction to be a two-element tuple only");
		PyErr_SetString(PyExc_TypeError, errmsg);
		return;
	}
	if (fShaderKey)
	{
		plGrassShaderMod* shader = plGrassShaderMod::ConvertNoRef(fShaderKey->ObjectIsLoaded());
		if (shader)
		{
			shader->fWaves[waveNum].fDirX = direction[0];
			shader->fWaves[waveNum].fDirY = direction[1];
			shader->RefreshWaves();
		}
	}
}

void pyGrassShader::SetWaveSpeed(int waveNum, hsScalar speed)
{
	if ((waveNum < 0)||(waveNum >= plGrassShaderMod::kNumWaves))
	{
		char errmsg[256];
		sprintf(errmsg,"setWaveSpeed expects the waveNum to be between 0 and %d (inclusive)", plGrassShaderMod::kNumWaves - 1);
		PyErr_SetString(PyExc_ValueError, errmsg);
		return;
	}
	if (fShaderKey)
	{
		plGrassShaderMod* shader = plGrassShaderMod::ConvertNoRef(fShaderKey->ObjectIsLoaded());
		if (shader)
		{
			shader->fWaves[waveNum].fSpeed = speed;
			shader->RefreshWaves();
		}
	}
}

//////////////////////////////////////////////////////////////////////
// Getter functions
//////////////////////////////////////////////////////////////////////

std::vector<hsScalar> pyGrassShader::GetWaveDistortion(int waveNum) const
{
	std::vector<hsScalar> retVal;
	retVal.push_back(-1);
	retVal.push_back(-1);
	retVal.push_back(-1);
	if ((waveNum < 0)||(waveNum >= plGrassShaderMod::kNumWaves))
	{
		char errmsg[256];
		sprintf(errmsg,"getWaveDistortion expects the waveNum to be between 0 and %d (inclusive)", plGrassShaderMod::kNumWaves - 1);
		PyErr_SetString(PyExc_ValueError, errmsg);
		return retVal;
	}
	if (fShaderKey)
	{
		plGrassShaderMod* shader = plGrassShaderMod::ConvertNoRef(fShaderKey->ObjectIsLoaded());
		if (shader)
		{
			retVal[0] = shader->fWaves[waveNum].fDistX;
			retVal[1] = shader->fWaves[waveNum].fDistY;
			retVal[2] = shader->fWaves[waveNum].fDistZ;
			return retVal;
		}
	}
	return retVal;
}

std::vector<hsScalar> pyGrassShader::GetWaveDirection(int waveNum) const
{
	std::vector<hsScalar> retVal;
	retVal.push_back(-1);
	retVal.push_back(-1);
	if ((waveNum < 0)||(waveNum >= plGrassShaderMod::kNumWaves))
	{
		char errmsg[256];
		sprintf(errmsg,"getWaveDirection expects the waveNum to be between 0 and %d (inclusive)", plGrassShaderMod::kNumWaves - 1);
		PyErr_SetString(PyExc_ValueError, errmsg);
		return retVal;
	}
	if (fShaderKey)
	{
		plGrassShaderMod* shader = plGrassShaderMod::ConvertNoRef(fShaderKey->ObjectIsLoaded());
		if (shader)
		{
			retVal[0] = shader->fWaves[waveNum].fDirX;
			retVal[1] = shader->fWaves[waveNum].fDirY;
			return retVal;
		}
	}
	return retVal;
}

hsScalar pyGrassShader::GetWaveSpeed(int waveNum) const
{
	if ((waveNum < 0)||(waveNum >= plGrassShaderMod::kNumWaves))
	{
		char errmsg[256];
		sprintf(errmsg,"getWaveSpeed expects the waveNum to be between 0 and %d (inclusive)", plGrassShaderMod::kNumWaves - 1);
		PyErr_SetString(PyExc_ValueError, errmsg);
		return -1;
	}
	if (fShaderKey)
	{
		plGrassShaderMod* shader = plGrassShaderMod::ConvertNoRef(fShaderKey->ObjectIsLoaded());
		if (shader)
			return shader->fWaves[waveNum].fSpeed;
	}
	return -1;
}

//////////////////////////////////////////////////////////////////////
// Other functions
//////////////////////////////////////////////////////////////////////

void pyGrassShader::ResetWaves()
{
	if (fShaderKey)
	{
		plGrassShaderMod* shader = plGrassShaderMod::ConvertNoRef(fShaderKey->ObjectIsLoaded());
		if (shader)
			shader->ResetWaves();
	}
}