diff options
Diffstat (limited to 'src/extras/geometries/qplanegeometry.cpp')
| -rw-r--r-- | src/extras/geometries/qplanegeometry.cpp | 529 |
1 files changed, 529 insertions, 0 deletions
diff --git a/src/extras/geometries/qplanegeometry.cpp b/src/extras/geometries/qplanegeometry.cpp new file mode 100644 index 000000000..a0c0ffbc0 --- /dev/null +++ b/src/extras/geometries/qplanegeometry.cpp @@ -0,0 +1,529 @@ +/**************************************************************************** +** +** Copyright (C) 2015 Klaralvdalens Datakonsult AB (KDAB). +** Contact: https://www.qt.io/licensing/ +** +** This file is part of the Qt3D module of the Qt Toolkit. +** +** $QT_BEGIN_LICENSE:LGPL$ +** Commercial License Usage +** Licensees holding valid commercial Qt licenses may use this file in +** accordance with the commercial license agreement provided with the +** Software or, alternatively, in accordance with the terms contained in +** a written agreement between you and The Qt Company. For licensing terms +** and conditions see https://www.qt.io/terms-conditions. For further +** information use the contact form at https://www.qt.io/contact-us. +** +** GNU Lesser General Public License Usage +** Alternatively, this file may be used under the terms of the GNU Lesser +** General Public License version 3 as published by the Free Software +** Foundation and appearing in the file LICENSE.LGPL3 included in the +** packaging of this file. Please review the following information to +** ensure the GNU Lesser General Public License version 3 requirements +** will be met: https://www.gnu.org/licenses/lgpl-3.0.html. +** +** GNU General Public License Usage +** Alternatively, this file may be used under the terms of the GNU +** General Public License version 2.0 or (at your option) the GNU General +** Public license version 3 or any later version approved by the KDE Free +** Qt Foundation. The licenses are as published by the Free Software +** Foundation and appearing in the file LICENSE.GPL2 and LICENSE.GPL3 +** included in the packaging of this file. Please review the following +** information to ensure the GNU General Public License requirements will +** be met: https://www.gnu.org/licenses/gpl-2.0.html and +** https://www.gnu.org/licenses/gpl-3.0.html. +** +** $QT_END_LICENSE$ +** +****************************************************************************/ + +#include "qplanegeometry.h" +#include "qplanegeometry_p.h" +#include <Qt3DRender/qattribute.h> +#include <Qt3DRender/qbuffer.h> +#include <Qt3DRender/qbufferdatagenerator.h> +#include <limits> + +QT_BEGIN_NAMESPACE + +using namespace Qt3DRender; + +namespace Qt3DExtras { + +namespace { + +QByteArray createPlaneVertexData(float w, float h, const QSize &resolution) +{ + Q_ASSERT(w > 0.0f); + Q_ASSERT(h > 0.0f); + Q_ASSERT(resolution.width() >= 2); + Q_ASSERT(resolution.height() >= 2); + + const int nVerts = resolution.width() * resolution.height(); + + // Populate a buffer with the interleaved per-vertex data with + // vec3 pos, vec2 texCoord, vec3 normal, vec4 tangent + const quint32 elementSize = 3 + 2 + 3 + 4; + const quint32 stride = elementSize * sizeof(float); + QByteArray bufferBytes; + bufferBytes.resize(stride * nVerts); + float* fptr = reinterpret_cast<float*>(bufferBytes.data()); + + const float x0 = -w / 2.0f; + const float z0 = -h / 2.0f; + const float dx = w / (resolution.width() - 1); + const float dz = h / (resolution.height() - 1); + const float du = 1.0 / (resolution.width() - 1); + const float dv = 1.0 / (resolution.height() - 1); + + // Iterate over z + for (int j = 0; j < resolution.height(); ++j) { + const float z = z0 + static_cast<float>(j) * dz; + const float v = static_cast<float>(j) * dv; + + // Iterate over x + for (int i = 0; i < resolution.width(); ++i) { + const float x = x0 + static_cast<float>(i) * dx; + const float u = static_cast<float>(i) * du; + + // position + *fptr++ = x; + *fptr++ = 0.0; + *fptr++ = z; + + // texture coordinates + *fptr++ = u; + *fptr++ = v; + + // normal + *fptr++ = 0.0f; + *fptr++ = 1.0f; + *fptr++ = 0.0f; + + // tangent + *fptr++ = 1.0f; + *fptr++ = 0.0f; + *fptr++ = 0.0f; + *fptr++ = 1.0f; + } + } + + return bufferBytes; +} + +QByteArray createPlaneIndexData(const QSize &resolution) +{ + // Create the index data. 2 triangles per rectangular face + const int faces = 2 * (resolution.width() - 1) * (resolution.height() - 1); + const int indices = 3 * faces; + Q_ASSERT(indices < std::numeric_limits<quint16>::max()); + QByteArray indexBytes; + indexBytes.resize(indices * sizeof(quint16)); + quint16* indexPtr = reinterpret_cast<quint16*>(indexBytes.data()); + + // Iterate over z + for (int j = 0; j < resolution.height() - 1; ++j) { + const int rowStartIndex = j * resolution.width(); + const int nextRowStartIndex = (j + 1) * resolution.width(); + + // Iterate over x + for (int i = 0; i < resolution.width() - 1; ++i) { + // Split quad into two triangles + *indexPtr++ = rowStartIndex + i; + *indexPtr++ = nextRowStartIndex + i; + *indexPtr++ = rowStartIndex + i + 1; + + *indexPtr++ = nextRowStartIndex + i; + *indexPtr++ = nextRowStartIndex + i + 1; + *indexPtr++ = rowStartIndex + i + 1; + } + } + + return indexBytes; +} + +} // anonymous + +class PlaneVertexBufferFunctor : public QBufferDataGenerator +{ +public: + explicit PlaneVertexBufferFunctor(float w, float h, const QSize &resolution) + : m_width(w) + , m_height(h) + , m_resolution(resolution) + {} + + ~PlaneVertexBufferFunctor() {} + + QByteArray operator()() Q_DECL_FINAL + { + return createPlaneVertexData(m_width, m_height, m_resolution); + } + + bool operator ==(const QBufferDataGenerator &other) const Q_DECL_FINAL + { + const PlaneVertexBufferFunctor *otherFunctor = functor_cast<PlaneVertexBufferFunctor>(&other); + if (otherFunctor != Q_NULLPTR) + return (otherFunctor->m_width == m_width && + otherFunctor->m_height == m_height && + otherFunctor->m_resolution == m_resolution); + return false; + } + + QT3D_FUNCTOR(PlaneVertexBufferFunctor) + + private: + float m_width; + float m_height; + QSize m_resolution; +}; + +class PlaneIndexBufferFunctor : public QBufferDataGenerator +{ +public: + explicit PlaneIndexBufferFunctor(const QSize &resolution) + : m_resolution(resolution) + {} + + ~PlaneIndexBufferFunctor() {} + + QByteArray operator()() Q_DECL_FINAL + { + return createPlaneIndexData(m_resolution); + } + + bool operator ==(const QBufferDataGenerator &other) const Q_DECL_FINAL + { + const PlaneIndexBufferFunctor *otherFunctor = functor_cast<PlaneIndexBufferFunctor>(&other); + if (otherFunctor != Q_NULLPTR) + return (otherFunctor->m_resolution == m_resolution); + return false; + } + + QT3D_FUNCTOR(PlaneIndexBufferFunctor) + + private: + QSize m_resolution; +}; + +/*! + * \qmltype PlaneGeometry + * \instantiates Qt3DRender::QPlaneGeometry + * \inqmlmodule Qt3D.Render + */ + +/*! + * \qmlproperty float PlaneGeometry::width + * + * Holds the plane width. + */ + +/*! + * \qmlproperty float PlaneGeometry::height + * + * Holds the plane height. + */ + +/*! + * \qmlproperty size PlaneGeometry::resolution + * + * Holds the plane resolution. + */ + +/*! + * \qmlproperty Attribute PlaneGeometry::positionAttribute + * + * Holds the geometry position attribute. + */ + +/*! + * \qmlproperty Attribute PlaneGeometry::normalAttribute + * + * Holds the geometry normal attribute. + */ + +/*! + * \qmlproperty Attribute PlaneGeometry::texCoordAttribute + * + * Holds the geometry texture coordinate attribute. + */ + +/*! + * \qmlproperty Attribute PlaneGeometry::tangentAttribute + * + * Holds the geometry tangent attribute. + */ + +/*! + * \qmlproperty Attribute PlaneGeometry::indexAttribute + * + * Holds the geometry index attribute. + */ + +/*! + * \class Qt3DRender::QPlaneGeometry + * \inmodule Qt3DRender + * + * \inherits Qt3DRender::QGeometry + */ + +/*! + * Constructs a new QPlaneGeometry with \a parent. + */ +QPlaneGeometry::QPlaneGeometry(QPlaneGeometry::QNode *parent) + : QGeometry(*new QPlaneGeometryPrivate(), parent) +{ + Q_D(QPlaneGeometry); + d->init(); +} + +/*! + * \internal + */ +QPlaneGeometry::QPlaneGeometry(QPlaneGeometryPrivate &dd, QNode *parent) + : QGeometry(dd, parent) +{ + Q_D(QPlaneGeometry); + d->init(); +} + +/*! + * Destroys this geometry. + */ +QPlaneGeometry::~QPlaneGeometry() +{ + QGeometry::cleanup(); +} + +/*! + * Updates vertices based on resolution. + */ +void QPlaneGeometry::updateVertices() +{ + Q_D(QPlaneGeometry); + const int nVerts = d->m_meshResolution.width() * d->m_meshResolution.height(); + + d->m_positionAttribute->setCount(nVerts); + d->m_normalAttribute->setCount(nVerts); + d->m_texCoordAttribute->setCount(nVerts); + d->m_tangentAttribute->setCount(nVerts); + d->m_vertexBuffer->setDataGenerator(QBufferDataGeneratorPtr(new PlaneVertexBufferFunctor(d->m_width, d->m_height, d->m_meshResolution))); +} + +/*! + * Updates indices based on resolution. + */ +void QPlaneGeometry::updateIndices() +{ + Q_D(QPlaneGeometry); + const int faces = 2 * (d->m_meshResolution.width() - 1) * (d->m_meshResolution.height() - 1); + // Each primitive has 3 vertices + d->m_indexAttribute->setCount(faces * 3); + d->m_indexBuffer->setDataGenerator(QBufferDataGeneratorPtr(new PlaneIndexBufferFunctor(d->m_meshResolution))); + +} + +void QPlaneGeometry::setResolution(const QSize &resolution) +{ + Q_D(QPlaneGeometry); + if (d->m_meshResolution == resolution) + return; + d->m_meshResolution = resolution; + updateVertices(); + updateIndices(); + emit resolutionChanged(resolution); +} + +void QPlaneGeometry::setWidth(float width) +{ + Q_D(QPlaneGeometry); + if (width == d->m_width) + return; + d->m_width = width; + updateVertices(); + emit widthChanged(width); +} + +void QPlaneGeometry::setHeight(float height) +{ + Q_D(QPlaneGeometry); + if (height == d->m_height) + return; + d->m_height = height; + updateVertices(); + emit heightChanged(height); +} + +/*! + * \property QPlaneGeometry::resolution + * + * Holds the plane resolution. + */ +QSize QPlaneGeometry::resolution() const +{ + Q_D(const QPlaneGeometry); + return d->m_meshResolution; +} + +/*! + * \property QPlaneGeometry::width + * + * Holds the plane width. + */ +float QPlaneGeometry::width() const +{ + Q_D(const QPlaneGeometry); + return d->m_width; +} + +/*! + * \property QPlaneGeometry::height + * + * Holds the plane height. + */ +float QPlaneGeometry::height() const +{ + Q_D(const QPlaneGeometry); + return d->m_height; +} + +/*! + * \property QPlaneGeometry::positionAttribute + * + * Holds the geometry position attribute. + */ +QAttribute *QPlaneGeometry::positionAttribute() const +{ + Q_D(const QPlaneGeometry); + return d->m_positionAttribute; +} + +/*! + * \property QPlaneGeometry::normalAttribute + * + * Holds the geometry normal attribute. + */ +QAttribute *QPlaneGeometry::normalAttribute() const +{ + Q_D(const QPlaneGeometry); + return d->m_normalAttribute; +} + +/*! + * \property QPlaneGeometry::texCoordAttribute + * + * Holds the geometry texture coordinate attribute. + */ +QAttribute *QPlaneGeometry::texCoordAttribute() const +{ + Q_D(const QPlaneGeometry); + return d->m_texCoordAttribute; +} + +/*! + * \property QPlaneGeometry::tangentAttribute + * + * Holds the geometry tangent attribute. + */ +QAttribute *QPlaneGeometry::tangentAttribute() const +{ + Q_D(const QPlaneGeometry); + return d->m_tangentAttribute; +} + +/*! + * \property QPlaneGeometry::indexAttribute + * + * Holds the geometry index attribute. + */ +QAttribute *QPlaneGeometry::indexAttribute() const +{ + Q_D(const QPlaneGeometry); + return d->m_indexAttribute; +} + +QPlaneGeometryPrivate::QPlaneGeometryPrivate() + : QGeometryPrivate() + , m_width(1.0f) + , m_height(1.0f) + , m_meshResolution(QSize(2, 2)) + , m_positionAttribute(Q_NULLPTR) + , m_normalAttribute(Q_NULLPTR) + , m_texCoordAttribute(Q_NULLPTR) + , m_tangentAttribute(Q_NULLPTR) + , m_indexAttribute(Q_NULLPTR) + , m_vertexBuffer(Q_NULLPTR) + , m_indexBuffer(Q_NULLPTR) +{ +} + +void QPlaneGeometryPrivate::init() +{ + Q_Q(QPlaneGeometry); + m_positionAttribute = new QAttribute(q); + m_normalAttribute = new QAttribute(q); + m_texCoordAttribute = new QAttribute(q); + m_tangentAttribute = new QAttribute(q); + m_indexAttribute = new QAttribute(q); + m_vertexBuffer = new Qt3DRender::QBuffer(Qt3DRender::QBuffer::VertexBuffer, q); + m_indexBuffer = new Qt3DRender::QBuffer(Qt3DRender::QBuffer::IndexBuffer, q); + + const int nVerts = m_meshResolution.width() * m_meshResolution.height(); + const int stride = (3 + 2 + 3 + 4) * sizeof(float); + const int faces = 2 * (m_meshResolution.width() - 1) * (m_meshResolution.height() - 1); + + m_positionAttribute->setName(QAttribute::defaultPositionAttributeName()); + m_positionAttribute->setDataType(QAttribute::Float); + m_positionAttribute->setDataSize(3); + m_positionAttribute->setAttributeType(QAttribute::VertexAttribute); + m_positionAttribute->setBuffer(m_vertexBuffer); + m_positionAttribute->setByteStride(stride); + m_positionAttribute->setCount(nVerts); + + m_texCoordAttribute->setName(QAttribute::defaultTextureCoordinateAttributeName()); + m_texCoordAttribute->setDataType(QAttribute::Float); + m_texCoordAttribute->setDataSize(2); + m_texCoordAttribute->setAttributeType(QAttribute::VertexAttribute); + m_texCoordAttribute->setBuffer(m_vertexBuffer); + m_texCoordAttribute->setByteStride(stride); + m_texCoordAttribute->setByteOffset(3 * sizeof(float)); + m_texCoordAttribute->setCount(nVerts); + + m_normalAttribute->setName(QAttribute::defaultNormalAttributeName()); + m_normalAttribute->setDataType(QAttribute::Float); + m_normalAttribute->setDataSize(3); + m_normalAttribute->setAttributeType(QAttribute::VertexAttribute); + m_normalAttribute->setBuffer(m_vertexBuffer); + m_normalAttribute->setByteStride(stride); + m_normalAttribute->setByteOffset(5 * sizeof(float)); + m_normalAttribute->setCount(nVerts); + + m_tangentAttribute->setName(QAttribute::defaultTangentAttributeName()); + m_tangentAttribute->setDataType(QAttribute::Float); + m_tangentAttribute->setDataSize(4); + m_tangentAttribute->setAttributeType(QAttribute::VertexAttribute); + m_tangentAttribute->setBuffer(m_vertexBuffer); + m_tangentAttribute->setByteStride(stride); + m_tangentAttribute->setByteOffset(8 * sizeof(float)); + m_tangentAttribute->setCount(nVerts); + + m_indexAttribute->setAttributeType(QAttribute::IndexAttribute); + m_indexAttribute->setDataType(QAttribute::UnsignedShort); + m_indexAttribute->setBuffer(m_indexBuffer); + + // Each primitive has 3 vertives + m_indexAttribute->setCount(faces * 3); + + m_vertexBuffer->setDataGenerator(QBufferDataGeneratorPtr(new PlaneVertexBufferFunctor(m_width, m_height, m_meshResolution))); + m_indexBuffer->setDataGenerator(QBufferDataGeneratorPtr(new PlaneIndexBufferFunctor(m_meshResolution))); + + q->addAttribute(m_positionAttribute); + q->addAttribute(m_texCoordAttribute); + q->addAttribute(m_normalAttribute); + q->addAttribute(m_tangentAttribute); + q->addAttribute(m_indexAttribute); +} + +} // Qt3DExtras + +QT_END_NAMESPACE |