Function PySetItem

trinity/Tr2RuntimeInstanceData_Blue.cpp:198–275

}

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

	unsigned index;
	PyObject* item;
	if( !PyArg_ParseTuple( args, "IO!", &index, &PyTuple_Type, &item ) )
	{
		return nullptr;
	}

	if( pThis->GetCount() <= index )
	{
		PyErr_SetString( PyExc_IndexError, "Index out of range" );
		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;
	}

	if( PyTuple_GET_SIZE( item ) != def.m_items.size() )
	{
		pThis->DestroyData();
		PyErr_SetString( PyExc_TypeError, "Argument 2 needs to be a tuple and it needs to match element layout" );
		return nullptr;
	}

	uint8_t* dataRaw = static_cast<uint8_t*>( pThis->GetData( pThis->GetCount() ) );
	if( !dataRaw )
	{
		PyErr_SetString( PyExc_RuntimeError, "Could not lock data" );
		return nullptr;
	}
	float* data = static_cast<float*>( static_cast<void*>( dataRaw + pThis->GetStride() * index ) );


	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 2 needs to be a tuple and it needs to match element layout" );
				return nullptr;
			}
			*data++ = float( PyFloat_AsDouble( element ) );
		}
		else
		{
			if( !PyTuple_Check( element ) )
			{