Function PySetData

trinity/Tr2RuntimeInstanceData_Blue.cpp:77–149

}

static PyObject* PySetData( PyObject* self, PyObject* args )
{
	Tr2RuntimeInstanceData* pThis = BluePythonCast<Tr2RuntimeInstanceData*>( self );

	PyObject* pyData;
	if( !PyArg_ParseTuple( args, "O!", &PyList_Type, &pyData ) )
	{
		return nullptr;
	}

	const Tr2VertexDefinition& def = pThis->GetLayout();
	if( def.m_items.empty() )
	{
		PyErr_SetString( PyExc_RuntimeError, "Need to set valid element layout with SetElementLayout before calling SetData" );
		return nullptr;
	}

	float* data = reinterpret_cast<float*>( pThis->GetData( unsigned( PyList_GET_SIZE( pyData ) ) ) );
	if( !data )
	{
		PyErr_SetString( PyExc_RuntimeError, "Could not lock data" );
		return nullptr;
	}


	for( int i = 0; i < PyList_GET_SIZE( pyData ); ++i )
	{
		PyObject* item = PyList_GetItem( pyData, i );
		if( !PyTuple_Check( item ) || PyTuple_GET_SIZE( item ) != def.m_items.size() )
		{
			pThis->DestroyData();
			PyErr_SetString( PyExc_TypeError, "Argument 1 needs to be a list of tuples and tuples need to match element layout" );
			return nullptr;
		}
		for( size_t j = 0; j < def.m_items.size(); ++j )
		{
			PyObject* element = PyTuple_GET_ITEM( item, j );
			unsigned size = def.GetDataTypeSizeInMembers( def.m_items[j].m_dataType );
			if( size == 1 )
			{
				if( !PyFloat_Check( element ) )
				{
					pThis->DestroyData();
					PyErr_SetString( PyExc_TypeError, "Argument 1 needs to be a list of tuples and tuples need to match element layout" );
					return nullptr;
				}
				*data++ = float( PyFloat_AsDouble( element ) );
			}
			else
			{
				if( !PyTuple_Check( element ) )
				{
					pThis->DestroyData();
					PyErr_SetString( PyExc_TypeError, "Argument 1 needs to be a list of tuples and tuples need to match element layout" );
					return nullptr;
				}
				for( unsigned k = 0; k < size; ++k )
				{