Function PySetItemElement

trinity/Tr2RuntimeInstanceData_Blue.cpp:324–396

}

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

	unsigned index;
	unsigned elementIndex;
	PyObject* element;
	if( !PyArg_ParseTuple( args, "IIO", &index, &elementIndex, &element ) )
	{
		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( elementIndex >= def.m_items.size() )
	{
		PyErr_SetString( PyExc_IndexError, "elementIndex out of range" );
		return nullptr;
	}

	uint8_t* dataRaw = reinterpret_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 + def.m_items[elementIndex].m_offset ) );

	unsigned size = def.GetDataTypeSizeInMembers( def.m_items[elementIndex].m_dataType );
	if( size == 1 )
	{
		if( !PyFloat_Check( element ) )
		{
			pThis->DestroyData();
			PyErr_SetString( PyExc_TypeError, "Argument 3 needs to be a float" );
			return nullptr;
		}
		*data++ = float( PyFloat_AsDouble( element ) );
	}
	else
	{
		if( !PyTuple_Check( element ) )
		{
			pThis->DestroyData();
			PyErr_SetString( PyExc_TypeError, "Argument 3 needs to be a tuple and it needs to match element layout" );
			return nullptr;
		}