Function PyGetKelvinTemperatureBuffer

trinity/Tr2KelvinColor_Blue.cpp:99–136

}

static PyObject* PyGetKelvinTemperatureBuffer( PyObject* self, PyObject* args )
{
	int width, height;
	int intWhiteBalance;
	if( !PyArg_ParseTuple( args, "iii", &width, &height, &intWhiteBalance ) )
	{
		return NULL;
	}

	Tr2StandardIlluminant stdIllum = (Tr2StandardIlluminant)intWhiteBalance;
	char* buffer = CCP_NEW( "PyGetKelvinTemperatureBuffer" ) char[3 * width * height];

	const float startK = MIN_KELVIN;
	const float deltaK = ( MAX_KELVIN - MIN_KELVIN ) / ( static_cast<float>( width ) - 1.0f );
	float curK = startK;

	// Render one row of the bitmap
	char* ptr = buffer;
	for( int i = 0; i < width; ++i )
	{
		Vector3d rgb = TriColorFromKelvin( curK, 0.5f, stdIllum );
		ptr[3 * i] = static_cast<char>( rgb.x * 255.0 );
		ptr[3 * i + 1] = static_cast<char>( rgb.y * 255.0 );
		ptr[3 * i + 2] = static_cast<char>( rgb.z * 255.0 );
		curK += deltaK;
	}

	// Copy the row into the rest of the buffer
	for( int i = 1; i < height; ++i )
	{
		memcpy( buffer + i * 3 * width, buffer, 3 * width );
	}

	PyObject* ret = NULL;
	ret = PyByteArray_FromStringAndSize( buffer, 3 * width * height );
	CCP_DELETE[] buffer;
	return ret;
}

static PyObject* PyGetKelvinTintBuffer( PyObject* self, PyObject* args )
{