| 1 | /**************************************************************************** | 
| 2 | ** | 
| 3 | ** Copyright (C) 2014 Gunnar Sletta <gunnar@sletta.org> | 
| 4 | ** Contact: https://www.qt.io/licensing/ | 
| 5 | ** | 
| 6 | ** This file is part of the examples of the Qt Toolkit. | 
| 7 | ** | 
| 8 | ** $QT_BEGIN_LICENSE:BSD$ | 
| 9 | ** Commercial License Usage | 
| 10 | ** Licensees holding valid commercial Qt licenses may use this file in | 
| 11 | ** accordance with the commercial license agreement provided with the | 
| 12 | ** Software or, alternatively, in accordance with the terms contained in | 
| 13 | ** a written agreement between you and The Qt Company. For licensing terms | 
| 14 | ** and conditions see https://www.qt.io/terms-conditions. For further | 
| 15 | ** information use the contact form at https://www.qt.io/contact-us. | 
| 16 | ** | 
| 17 | ** BSD License Usage | 
| 18 | ** Alternatively, you may use this file under the terms of the BSD license | 
| 19 | ** as follows: | 
| 20 | ** | 
| 21 | ** "Redistribution and use in source and binary forms, with or without | 
| 22 | ** modification, are permitted provided that the following conditions are | 
| 23 | ** met: | 
| 24 | **   * Redistributions of source code must retain the above copyright | 
| 25 | **     notice, this list of conditions and the following disclaimer. | 
| 26 | **   * Redistributions in binary form must reproduce the above copyright | 
| 27 | **     notice, this list of conditions and the following disclaimer in | 
| 28 | **     the documentation and/or other materials provided with the | 
| 29 | **     distribution. | 
| 30 | **   * Neither the name of The Qt Company Ltd nor the names of its | 
| 31 | **     contributors may be used to endorse or promote products derived | 
| 32 | **     from this software without specific prior written permission. | 
| 33 | ** | 
| 34 | ** | 
| 35 | ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | 
| 36 | ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | 
| 37 | ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | 
| 38 | ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | 
| 39 | ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | 
| 40 | ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | 
| 41 | ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | 
| 42 | ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | 
| 43 | ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | 
| 44 | ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 
| 45 | ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE." | 
| 46 | ** | 
| 47 | ** $QT_END_LICENSE$ | 
| 48 | ** | 
| 49 | ****************************************************************************/ | 
| 50 |  | 
| 51 | #include "xorblender.h" | 
| 52 |  | 
| 53 | #include <QtCore/QPointer> | 
| 54 |  | 
| 55 | #include <QtGui/QOpenGLContext> | 
| 56 | #include <QtGui/QOpenGLFunctions> | 
| 57 |  | 
| 58 | #include <QtQuick/QSGMaterial> | 
| 59 | #include <QtQuick/QSGTexture> | 
| 60 | #include <QtQuick/QSGGeometryNode> | 
| 61 | #include <QtQuick/QSGTextureProvider> | 
| 62 |  | 
| 63 | /* This example could just as well have been implemented all in QML using | 
| 64 |  * a ShaderEffect, and for 90% of all usecases, using a ShaderEffect will | 
| 65 |  * be sufficient. This example exists to illustrate how to consume | 
| 66 |  * texture providers from C++ and how to use multiple texture sources in | 
| 67 |  * a custom material. | 
| 68 |  */ | 
| 69 |  | 
| 70 | class XorBlendMaterial : public QSGMaterial | 
| 71 | { | 
| 72 | public: | 
| 73 |     XorBlendMaterial(); | 
| 74 |     QSGMaterialType *type() const override; | 
| 75 |     QSGMaterialShader *createShader() const override; | 
| 76 |     int compare(const QSGMaterial *other) const override; | 
| 77 |  | 
| 78 |     struct { | 
| 79 |         QSGTexture *texture1 = nullptr; | 
| 80 |         QSGTexture *texture2 = nullptr; | 
| 81 |     } state; | 
| 82 | }; | 
| 83 |  | 
| 84 | class XorBlendShader : public QSGMaterialShader // for when the scenegraph is using OpenGL directly | 
| 85 | { | 
| 86 | public: | 
| 87 |     XorBlendShader(); | 
| 88 |     void initialize() override; | 
| 89 |     char const *const *attributeNames() const override; | 
| 90 |     void updateState(const RenderState &state, QSGMaterial *newEffect, QSGMaterial *oldEffect) override; | 
| 91 |  | 
| 92 | private: | 
| 93 |     int m_matrix_id; | 
| 94 |     int m_opacity_id; | 
| 95 | }; | 
| 96 |  | 
| 97 | class XorBlendRhiShader : public QSGMaterialRhiShader // for when the scenegraph is using QRhi | 
| 98 | { | 
| 99 | public: | 
| 100 |     XorBlendRhiShader(); | 
| 101 |     bool updateUniformData(RenderState &state, | 
| 102 |                            QSGMaterial *newMaterial, QSGMaterial *oldMaterial) override; | 
| 103 |     void updateSampledImage(RenderState &state, int binding, QSGTexture **texture, | 
| 104 |                             QSGMaterial *newMaterial, QSGMaterial *oldMaterial) override; | 
| 105 | }; | 
| 106 |  | 
| 107 | XorBlendMaterial::XorBlendMaterial() | 
| 108 | { | 
| 109 |     setFlag(flags: SupportsRhiShader); | 
| 110 |     setFlag(flags: Blending); | 
| 111 | } | 
| 112 |  | 
| 113 | QSGMaterialShader *XorBlendMaterial::createShader() const | 
| 114 | { | 
| 115 |     if (flags().testFlag(flag: RhiShaderWanted)) | 
| 116 |         return new XorBlendRhiShader; | 
| 117 |     else | 
| 118 |         return new XorBlendShader; | 
| 119 | } | 
| 120 |  | 
| 121 | QSGMaterialType *XorBlendMaterial::type() const | 
| 122 | { | 
| 123 |     static QSGMaterialType type; | 
| 124 |     return &type; | 
| 125 | } | 
| 126 |  | 
| 127 | int XorBlendMaterial::compare(const QSGMaterial *o) const | 
| 128 | { | 
| 129 |     Q_ASSERT(o && type() == o->type()); | 
| 130 |     const XorBlendMaterial *other = static_cast<const XorBlendMaterial *>(o); | 
| 131 |  | 
| 132 |     if (!state.texture1 || !other->state.texture1) | 
| 133 |         return state.texture1 ? 1 : -1; | 
| 134 |  | 
| 135 |     if (!state.texture2 || !other->state.texture2) | 
| 136 |         return state.texture2 ? -1 : 1; | 
| 137 |  | 
| 138 |     if (int diff = state.texture1->comparisonKey() - other->state.texture1->comparisonKey()) | 
| 139 |         return diff; | 
| 140 |  | 
| 141 |     if (int diff = state.texture2->comparisonKey() - other->state.texture2->comparisonKey()) | 
| 142 |         return diff; | 
| 143 |  | 
| 144 |     return 0; | 
| 145 | } | 
| 146 |  | 
| 147 | XorBlendShader::XorBlendShader() | 
| 148 | { | 
| 149 |     setShaderSourceFile(type: QOpenGLShader::Vertex, sourceFile: QLatin1String(":/scenegraph/twotextureproviders/shaders/xorblender.vert" )); | 
| 150 |     setShaderSourceFile(type: QOpenGLShader::Fragment, sourceFile: QLatin1String(":/scenegraph/twotextureproviders/shaders/xorblender.frag" )); | 
| 151 | } | 
| 152 |  | 
| 153 | void XorBlendShader::initialize() | 
| 154 | { | 
| 155 |     m_matrix_id = program()->uniformLocation(name: "qt_Matrix" ); | 
| 156 |     m_opacity_id  = program()->uniformLocation(name: "qt_Opacity" ); | 
| 157 |     // The texture units never change, only the textures we bind to them so | 
| 158 |     // we set these once and for all here. | 
| 159 |     program()->setUniformValue(name: "uSource1" , value: 0); // GL_TEXTURE0 | 
| 160 |     program()->setUniformValue(name: "uSource2" , value: 1); // GL_TEXTURE1 | 
| 161 | } | 
| 162 |  | 
| 163 | char const *const *XorBlendShader::attributeNames() const | 
| 164 | { | 
| 165 |     static char const *const attr[] = { "aVertex" , "aTexCoord" , nullptr }; | 
| 166 |     return attr; | 
| 167 | } | 
| 168 |  | 
| 169 | void XorBlendShader::updateState(const RenderState &state, QSGMaterial *newEffect, QSGMaterial *) | 
| 170 | { | 
| 171 |     XorBlendMaterial *material = static_cast<XorBlendMaterial *>(newEffect); | 
| 172 |  | 
| 173 |     if (state.isMatrixDirty()) | 
| 174 |         program()->setUniformValue(location: m_matrix_id, value: state.combinedMatrix()); | 
| 175 |  | 
| 176 |     if (state.isOpacityDirty()) | 
| 177 |         program()->setUniformValue(location: m_opacity_id, value: state.opacity()); | 
| 178 |  | 
| 179 |     QOpenGLFunctions *f = QOpenGLContext::currentContext()->functions(); | 
| 180 |     // We bind the textures in inverse order so that we leave the updateState | 
| 181 |     // function with GL_TEXTURE0 as the active texture unit. This is maintain | 
| 182 |     // the "contract" that updateState should not mess up the GL state beyond | 
| 183 |     // what is needed for this material. | 
| 184 |     f->glActiveTexture(GL_TEXTURE1); | 
| 185 |     material->state.texture2->bind(); | 
| 186 |     f->glActiveTexture(GL_TEXTURE0); | 
| 187 |     material->state.texture1->bind(); | 
| 188 | } | 
| 189 |  | 
| 190 | XorBlendRhiShader::XorBlendRhiShader() | 
| 191 | { | 
| 192 |     setShaderFileName(stage: VertexStage, filename: QLatin1String(":/scenegraph/twotextureproviders/shaders/+qsb/xorblender.vert" )); | 
| 193 |     setShaderFileName(stage: FragmentStage, filename: QLatin1String(":/scenegraph/twotextureproviders/shaders/+qsb/xorblender.frag" )); | 
| 194 | } | 
| 195 |  | 
| 196 | bool XorBlendRhiShader::updateUniformData(RenderState &state, QSGMaterial *, QSGMaterial *) | 
| 197 | { | 
| 198 |     bool changed = false; | 
| 199 |     QByteArray *buf = state.uniformData(); | 
| 200 |     Q_ASSERT(buf->size() >= 68); | 
| 201 |  | 
| 202 |     if (state.isMatrixDirty()) { | 
| 203 |         const QMatrix4x4 m = state.combinedMatrix(); | 
| 204 |         memcpy(dest: buf->data(), src: m.constData(), n: 64); | 
| 205 |         changed = true; | 
| 206 |     } | 
| 207 |  | 
| 208 |     if (state.isOpacityDirty()) { | 
| 209 |         const float opacity = state.opacity(); | 
| 210 |         memcpy(dest: buf->data() + 64, src: &opacity, n: 4); | 
| 211 |         changed = true; | 
| 212 |     } | 
| 213 |  | 
| 214 |     return changed; | 
| 215 | } | 
| 216 |  | 
| 217 | void XorBlendRhiShader::updateSampledImage(RenderState &state, int binding, QSGTexture **texture, | 
| 218 |                                            QSGMaterial *newMaterial, QSGMaterial *) | 
| 219 | { | 
| 220 |     Q_UNUSED(state); | 
| 221 |  | 
| 222 |     XorBlendMaterial *mat = static_cast<XorBlendMaterial *>(newMaterial); | 
| 223 |     switch (binding) { // the binding for the sampler2Ds in the fragment shader | 
| 224 |     case 1: | 
| 225 |         *texture = mat->state.texture1; | 
| 226 |         break; | 
| 227 |     case 2: | 
| 228 |         *texture = mat->state.texture2; | 
| 229 |         break; | 
| 230 |     default: | 
| 231 |         return; | 
| 232 |     } | 
| 233 | } | 
| 234 |  | 
| 235 | /* The rendering is split into two nodes. The top-most node is not actually | 
| 236 |  * rendering anything, but is responsible for managing the texture providers. | 
| 237 |  * The XorNode also has a geometry node internally which it uses to render | 
| 238 |  * the texture providers using the XorBlendShader when all providers and | 
| 239 |  * textures are all present. | 
| 240 |  * | 
| 241 |  * The texture providers are updated solely on the render thread (when rendering | 
| 242 |  * is happening on a separate thread). This is why we are using preprocess | 
| 243 |  * and direct signals between the the texture providers and the node rather | 
| 244 |  * than updating state in updatePaintNode. | 
| 245 |  */ | 
| 246 | class XorNode : public QObject, public QSGNode | 
| 247 | { | 
| 248 |     Q_OBJECT | 
| 249 | public: | 
| 250 |     XorNode(QSGTextureProvider *p1, QSGTextureProvider *p2) | 
| 251 |         : m_provider1(p1) | 
| 252 |         , m_provider2(p2) | 
| 253 |     { | 
| 254 |         setFlag(QSGNode::UsePreprocess, true); | 
| 255 |  | 
| 256 |         // Set up material so it is all set for later.. | 
| 257 |         m_material = new XorBlendMaterial; | 
| 258 |         m_node.setMaterial(m_material); | 
| 259 |         m_node.setFlag(QSGNode::OwnsMaterial); | 
| 260 |  | 
| 261 |         // Set up geometry, actual vertices will be initialized in updatePaintNode | 
| 262 |         m_node.setGeometry(new QSGGeometry(QSGGeometry::defaultAttributes_TexturedPoint2D(), 4)); | 
| 263 |         m_node.setFlag(QSGNode::OwnsGeometry); | 
| 264 |  | 
| 265 |         // If this node is used as in a shader effect source, we need to propegate | 
| 266 |         // changes that will occur in this node outwards. | 
| 267 |         connect(sender: m_provider1.data(), signal: &QSGTextureProvider::textureChanged, receiver: this, slot: &XorNode::textureChange, type: Qt::DirectConnection); | 
| 268 |         connect(sender: m_provider2.data(), signal: &QSGTextureProvider::textureChanged, receiver: this, slot: &XorNode::textureChange, type: Qt::DirectConnection); | 
| 269 |     } | 
| 270 |  | 
| 271 |     void preprocess() override { | 
| 272 |         // Update the textures from the providers, calling into QSGDynamicTexture if required | 
| 273 |         if (m_provider1) { | 
| 274 |             m_material->state.texture1 = m_provider1->texture(); | 
| 275 |             if (QSGDynamicTexture *dt1 = qobject_cast<QSGDynamicTexture *>(object: m_material->state.texture1)) | 
| 276 |                 dt1->updateTexture(); | 
| 277 |         } | 
| 278 |         if (m_provider2) { | 
| 279 |             m_material->state.texture2 = m_provider2->texture(); | 
| 280 |             if (QSGDynamicTexture *dt2 = qobject_cast<QSGDynamicTexture *>(object: m_material->state.texture2)) | 
| 281 |                 dt2->updateTexture(); | 
| 282 |         } | 
| 283 |  | 
| 284 |         // Remove node from the scene graph if it is there and either texture is missing... | 
| 285 |         if (m_node.parent() && (!m_material->state.texture1 || !m_material->state.texture2)) | 
| 286 |             removeChildNode(node: &m_node); | 
| 287 |  | 
| 288 |         // Add it if it is not already there and both textures are present.. | 
| 289 |         else if (!m_node.parent() && m_material->state.texture1 && m_material->state.texture2) | 
| 290 |             appendChildNode(node: &m_node); | 
| 291 |     } | 
| 292 |  | 
| 293 |     void setRect(const QRectF &rect) { | 
| 294 |         // Update geometry if it has changed and mark the change in the scene graph. | 
| 295 |         if (m_rect != rect) { | 
| 296 |             m_rect = rect; | 
| 297 |             QSGGeometry::updateTexturedRectGeometry(g: m_node.geometry(), rect: m_rect, sourceRect: QRectF(0, 0, 1, 1)); | 
| 298 |             m_node.markDirty(bits: QSGNode::DirtyGeometry); | 
| 299 |         } | 
| 300 |     } | 
| 301 |  | 
| 302 | public slots: | 
| 303 |     void textureChange() { | 
| 304 |         // When our sources change, we will look different, so signal the change to the | 
| 305 |         // scene graph. | 
| 306 |         markDirty(bits: QSGNode::DirtyMaterial); | 
| 307 |     } | 
| 308 |  | 
| 309 | private: | 
| 310 |     QRectF m_rect; | 
| 311 |     XorBlendMaterial *m_material; | 
| 312 |     QSGGeometryNode m_node; | 
| 313 |     QPointer<QSGTextureProvider> m_provider1; | 
| 314 |     QPointer<QSGTextureProvider> m_provider2; | 
| 315 | }; | 
| 316 |  | 
| 317 | XorBlender::XorBlender(QQuickItem *parent) | 
| 318 |     : QQuickItem(parent) | 
| 319 |     , m_source1(nullptr) | 
| 320 |     , m_source2(nullptr) | 
| 321 |     , m_source1Changed(false) | 
| 322 |     , m_source2Changed(false) | 
| 323 | { | 
| 324 |     setFlag(flag: ItemHasContents, enabled: true); | 
| 325 | } | 
| 326 |  | 
| 327 | void XorBlender::setSource1(QQuickItem *i) | 
| 328 | { | 
| 329 |     if (i == m_source1) | 
| 330 |         return; | 
| 331 |     m_source1 = i; | 
| 332 |     emit source1Changed(item: m_source1); | 
| 333 |     m_source1Changed = true; | 
| 334 |     update(); | 
| 335 | } | 
| 336 |  | 
| 337 | void XorBlender::setSource2(QQuickItem *i) | 
| 338 | { | 
| 339 |     if (i == m_source2) | 
| 340 |         return; | 
| 341 |     m_source2 = i; | 
| 342 |     emit source2Changed(item: m_source2); | 
| 343 |     m_source2Changed = true; | 
| 344 |     update(); | 
| 345 | } | 
| 346 |  | 
| 347 | QSGNode *XorBlender::updatePaintNode(QSGNode *old, UpdatePaintNodeData *) | 
| 348 | { | 
| 349 |     // Check if our input is valid and abort if not, deleting the old node. | 
| 350 |     bool abort = false; | 
| 351 |     if (!m_source1 || !m_source1->isTextureProvider()) { | 
| 352 |         qDebug() << "source1 is missing or not a texture provider" ; | 
| 353 |         abort = true; | 
| 354 |     } | 
| 355 |     if (!m_source2 || !m_source2->isTextureProvider()) { | 
| 356 |         qDebug() << "source2 is missing or not a texture provider" ; | 
| 357 |         abort = true; | 
| 358 |     } | 
| 359 |     if (abort) { | 
| 360 |         delete old; | 
| 361 |         return nullptr; | 
| 362 |     } | 
| 363 |  | 
| 364 |     XorNode *node = static_cast<XorNode *>(old); | 
| 365 |  | 
| 366 |     // If the sources have changed, recreate the nodes | 
| 367 |     if (m_source1Changed || m_source2Changed) { | 
| 368 |         delete node; | 
| 369 |         node = nullptr; | 
| 370 |         m_source1Changed = false; | 
| 371 |         m_source2Changed = false; | 
| 372 |     } | 
| 373 |  | 
| 374 |     // Create a new XorNode for us to render with. | 
| 375 |     if (!node) | 
| 376 |         node = new XorNode(m_source1->textureProvider(), m_source2->textureProvider()); | 
| 377 |  | 
| 378 |     // Update the geometry of the node to match the new bounding rect | 
| 379 |     node->setRect(boundingRect()); | 
| 380 |  | 
| 381 |     return node; | 
| 382 | } | 
| 383 |  | 
| 384 | #include "xorblender.moc" | 
| 385 |  |