Numpy数组通过QBuffer到QGeometry

时间:2018-03-01 12:33:51

标签: python qt5 pyside qt3d pyside2

我的目标是创建一个numpy数组并将其字节数据转换为QBuffer。我想知道如何正确设置DataSizeByteStrideCount。请参阅下面的代码:

self.mesh = Qt3DRender.QGeometryRenderer()
self.mesh.setPrimitiveType(Qt3DRender.QGeometryRenderer.Points)
self.geometry = Qt3DRender.QGeometry(self.mesh)

vertex_data_buffer = Qt3DRender.QBuffer(Qt3DRender.QBuffer.VertexBuffer, self.geometry)
data = np.random.rand(1000, 3)
vertex_data_buffer.setData(QByteArray(data.tobytes()))

self.position_attribute = Qt3DRender.QAttribute()
self.position_attribute.setAttributeType(
    Qt3DRender.QAttribute.VertexAttribute)
self.position_attribute.setBuffer(vertex_data_buffer)
self.position_attribute.setDataType(Qt3DRender.QAttribute.Float)
self.position_attribute.setDataSize(3)  # ??
self.position_attribute.setByteOffset(0)
self.position_attribute.setByteStride(6)  # ??
self.position_attribute.setCount(1000)  # ??
self.position_attribute.setName(
    Qt3DRender.QAttribute.defaultPositionAttributeName())

self.geometry.addAttribute(self.position_attribute)

1 个答案:

答案 0 :(得分:0)

我们设法解决了这个问题。

首先,不推荐使用以下两行复制的行。它们可以被移除。

self.position_attribute.setDataType(Qt3DRender.QAttribute.Float)
self.position_attribute.setDataSize(3)

添加了一行:

self.position_attribute.setVertexSize(3)

ByteStride应设为123是坐标数,4float32的长度(以字节为单位)。请注意设置numpy' arraydata = np.random.rand(1000, 3).astype(np.float32)

self.position_attribute.setByteOffset(0)
self.position_attribute.setByteStride(3*4)
self.position_attribute.setCount(1000)