src/declarative/particles/qsgimageparticle.cpp

   1 /****************************************************************************
   2 **
   3 ** Copyright (C) 2011 Nokia Corporation and/or its subsidiary(-ies).
   4 ** All rights reserved.
   5 ** Contact: Nokia Corporation (qt-info@nokia.com)
   6 **
   7 ** This file is part of the Declarative module of the Qt Toolkit.
   8 **
   9 ** $QT_BEGIN_LICENSE:LGPL$
  10 ** No Commercial Usage
  11 ** This file contains pre-release code and may not be distributed.
  12 ** You may use this file in accordance with the terms and conditions
  13 ** contained in the Technology Preview License Agreement accompanying
  14 ** this package.
  15 **
  16 ** GNU Lesser General Public License Usage
  17 ** Alternatively, this file may be used under the terms of the GNU Lesser
  18 ** General Public License version 2.1 as published by the Free Software
  19 ** Foundation and appearing in the file LICENSE.LGPL included in the
  20 ** packaging of this file.  Please review the following information to
  21 ** ensure the GNU Lesser General Public License version 2.1 requirements
  22 ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
  23 **
  24 ** In addition, as a special exception, Nokia gives you certain additional
  25 ** rights.  These rights are described in the Nokia Qt LGPL Exception
  26 ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
  27 **
  28 ** If you have questions regarding the use of this file, please contact
  29 ** Nokia at qt-info@nokia.com.
  30 **
  31 **
  32 **
  33 **
  34 **
  35 **
  36 **
  37 **
  38 ** $QT_END_LICENSE$
  39 **
  40 ****************************************************************************/
  41
  42 #include <private/qsgcontext_p.h>
  43 #include <private/qsgadaptationlayer_p.h>
  44 #include <qsgnode.h>
  45 #include <qsgtexturematerial.h>
  46 #include <qsgtexture.h>
  47 #include <QFile>
  48 #include "qsgimageparticle_p.h"
  49 #include "qsgparticleemitter_p.h"
  50 #include "qsgsprite_p.h"
  51 #include "qsgspriteengine_p.h"
  52 #include <QGLFunctions>
  53 #include <qsgengine.h>
  54
  55 QT_BEGIN_NAMESPACE
  56
  57 const float CONV = 0.017453292519943295;
  58 class UltraMaterial : public QSGMaterial
  59 {
  60 public:
  61     UltraMaterial(bool withSprites=false)
  62         : timestamp(0)
  63         , framecount(1)
  64         , animcount(1)
  65         , usesSprites(withSprites)
  66     {
  67         setFlag(Blending, true);
  68     }
  69
  70     ~UltraMaterial()
  71     {
  72         delete texture;
  73         delete colortable;
  74         delete sizetable;
  75         delete opacitytable;
  76     }
  77
  78     virtual QSGMaterialType *type() const { static QSGMaterialType type; return &type; }
  79     virtual QSGMaterialShader *createShader() const;
  80     virtual int compare(const QSGMaterial *other) const
  81     {
  82         return this - static_cast<const UltraMaterial *>(other);
  83     }
  84
  85     QSGTexture *texture;
  86     QSGTexture *colortable;
  87     QSGTexture *sizetable;
  88     QSGTexture *opacitytable;
  89
  90     qreal timestamp;
  91     int framecount;
  92     int animcount;
  93     bool usesSprites;
  94 };
  95 class UltraMaterialData : public QSGMaterialShader
  96 {
  97 public:
  98     UltraMaterialData(const char *vertexFile = 0, const char *fragmentFile = 0)
  99     {
 100         QFile vf(vertexFile ? vertexFile : ":defaultshaders/ultravertex.shader");
 101         vf.open(QFile::ReadOnly);
 102         m_vertex_code = vf.readAll();
 103
 104         QFile ff(fragmentFile ? fragmentFile : ":defaultshaders/ultrafragment.shader");
 105         ff.open(QFile::ReadOnly);
 106         m_fragment_code = ff.readAll();
 107
 108         Q_ASSERT(!m_vertex_code.isNull());
 109         Q_ASSERT(!m_fragment_code.isNull());
 110     }
 111
 112     void deactivate() {
 113         QSGMaterialShader::deactivate();
 114
 115         for (int i=0; i<8; ++i) {
 116             program()->setAttributeArray(i, GL_FLOAT, chunkOfBytes, 1, 0);
 117         }
 118     }
 119
 120     virtual void updateState(const RenderState &state, QSGMaterial *newEffect, QSGMaterial *)
 121     {
 122         UltraMaterial *m = static_cast<UltraMaterial *>(newEffect);
 123         state.context()->functions()->glActiveTexture(GL_TEXTURE1);
 124         m->colortable->bind();
 125         program()->setUniformValue(m_colortable_id, 1);
 126
 127         state.context()->functions()->glActiveTexture(GL_TEXTURE2);
 128         m->sizetable->bind();
 129         program()->setUniformValue(m_sizetable_id, 2);
 130
 131         state.context()->functions()->glActiveTexture(GL_TEXTURE3);
 132         m->opacitytable->bind();
 133         program()->setUniformValue(m_opacitytable_id, 3);
 134
 135         state.context()->functions()->glActiveTexture(GL_TEXTURE0);//Investigate why this screws up Text{} if placed before 1
 136         m->texture->bind();
 137
 138         program()->setUniformValue(m_opacity_id, state.opacity());
 139         program()->setUniformValue(m_timestamp_id, (float) m->timestamp);
 140         program()->setUniformValue(m_framecount_id, (float) m->framecount);
 141         program()->setUniformValue(m_animcount_id, (float) m->animcount);
 142
 143         if (state.isMatrixDirty())
 144             program()->setUniformValue(m_matrix_id, state.combinedMatrix());
 145     }
 146
 147     virtual void initialize() {
 148         m_colortable_id = program()->uniformLocation("colortable");
 149         m_sizetable_id = program()->uniformLocation("sizetable");
 150         m_opacitytable_id = program()->uniformLocation("opacitytable");
 151         m_matrix_id = program()->uniformLocation("matrix");
 152         m_opacity_id = program()->uniformLocation("opacity");
 153         m_timestamp_id = program()->uniformLocation("timestamp");
 154         m_framecount_id = program()->uniformLocation("framecount");
 155         m_animcount_id = program()->uniformLocation("animcount");
 156     }
 157
 158     virtual const char *vertexShader() const { return m_vertex_code.constData(); }
 159     virtual const char *fragmentShader() const { return m_fragment_code.constData(); }
 160
 161     virtual char const *const *attributeNames() const {
 162         static const char *attr[] = {
 163             "vPos",
 164             "vTex",
 165             "vData",
 166             "vVec",
 167             "vColor",
 168             "vDeformVec",
 169             "vRotation",
 170             "vAnimData",
 171             0
 172         };
 173         return attr;
 174     }
 175
 176     virtual bool isColorTable() const { return false; }
 177
 178     int m_matrix_id;
 179     int m_opacity_id;
 180     int m_timestamp_id;
 181     int m_colortable_id;
 182     int m_sizetable_id;
 183     int m_opacitytable_id;
 184     int m_framecount_id;
 185     int m_animcount_id;
 186
 187     QByteArray m_vertex_code;
 188     QByteArray m_fragment_code;
 189
 190     static float chunkOfBytes[1024];
 191 };
 192 float UltraMaterialData::chunkOfBytes[1024];
 193
 194 QSGMaterialShader *UltraMaterial::createShader() const
 195 {
 196     if (usesSprites)//TODO: Perhaps just swap the shaders, and don't mind the extra vector?
 197         return new UltraMaterialData;
 198     else
 199         return new UltraMaterialData;
 200 }
 201
 202
 203 class SimpleMaterial : public UltraMaterial
 204 {
 205     virtual QSGMaterialShader *createShader() const;
 206     virtual QSGMaterialType *type() const { static QSGMaterialType type; return &type; }
 207 };
 208
 209 class SimpleMaterialData : public QSGMaterialShader
 210 {
 211 public:
 212     SimpleMaterialData(const char *vertexFile = 0, const char *fragmentFile = 0)
 213     {
 214         QFile vf(vertexFile ? vertexFile : ":defaultshaders/simplevertex.shader");
 215         vf.open(QFile::ReadOnly);
 216         m_vertex_code = vf.readAll();
 217
 218         QFile ff(fragmentFile ? fragmentFile : ":defaultshaders/simplefragment.shader");
 219         ff.open(QFile::ReadOnly);
 220         m_fragment_code = ff.readAll();
 221
 222         Q_ASSERT(!m_vertex_code.isNull());
 223         Q_ASSERT(!m_fragment_code.isNull());
 224     }
 225
 226     void deactivate() {
 227         QSGMaterialShader::deactivate();
 228
 229         for (int i=0; i<8; ++i) {
 230             program()->setAttributeArray(i, GL_FLOAT, chunkOfBytes, 1, 0);
 231         }
 232     }
 233
 234     virtual void updateState(const RenderState &state, QSGMaterial *newEffect, QSGMaterial *)
 235     {
 236         UltraMaterial *m = static_cast<UltraMaterial *>(newEffect);
 237         state.context()->functions()->glActiveTexture(GL_TEXTURE0);
 238         m->texture->bind();
 239
 240         program()->setUniformValue(m_opacity_id, state.opacity());
 241         program()->setUniformValue(m_timestamp_id, (float) m->timestamp);
 242
 243         if (state.isMatrixDirty())
 244             program()->setUniformValue(m_matrix_id, state.combinedMatrix());
 245     }
 246
 247     virtual void initialize() {
 248         m_matrix_id = program()->uniformLocation("matrix");
 249         m_opacity_id = program()->uniformLocation("opacity");
 250         m_timestamp_id = program()->uniformLocation("timestamp");
 251     }
 252
 253     virtual const char *vertexShader() const { return m_vertex_code.constData(); }
 254     virtual const char *fragmentShader() const { return m_fragment_code.constData(); }
 255
 256     virtual char const *const *attributeNames() const {
 257         static const char *attr[] = {
 258             "vPos",
 259             "vTex",
 260             "vData",
 261             "vVec",
 262             0
 263         };
 264         return attr;
 265     }
 266
 267     virtual bool isColorTable() const { return false; }
 268
 269     int m_matrix_id;
 270     int m_opacity_id;
 271     int m_timestamp_id;
 272
 273     QByteArray m_vertex_code;
 274     QByteArray m_fragment_code;
 275
 276     static float chunkOfBytes[1024];
 277 };
 278 float SimpleMaterialData::chunkOfBytes[1024];
 279
 280 QSGMaterialShader *SimpleMaterial::createShader() const {
 281     return new SimpleMaterialData;
 282 }
 283
 284 QSGImageParticle::QSGImageParticle(QSGItem* parent)
 285     : QSGParticlePainter(parent)
 286     , m_do_reset(false)
 287     , m_color_variation(0.0)
 288     , m_rootNode(0)
 289     , m_material(0)
 290     , m_alphaVariation(0.0)
 291     , m_alpha(1.0)
 292     , m_redVariation(0.0)
 293     , m_greenVariation(0.0)
 294     , m_blueVariation(0.0)
 295     , m_rotation(0)
 296     , m_autoRotation(false)
 297     , m_xVector(0)
 298     , m_yVector(0)
 299     , m_rotationVariation(0)
 300     , m_rotationSpeed(0)
 301     , m_rotationSpeedVariation(0)
 302     , m_spriteEngine(0)
 303     , m_bloat(false)
 304     , perfLevel(Unknown)
 305     , m_lastLevel(Unknown)
 306 {
 307     setFlag(ItemHasContents);
 308 }
 309
 310 QDeclarativeListProperty<QSGSprite> QSGImageParticle::sprites()
 311 {
 312     return QDeclarativeListProperty<QSGSprite>(this, &m_sprites, spriteAppend, spriteCount, spriteAt, spriteClear);
 313 }
 314
 315 void QSGImageParticle::setImage(const QUrl &image)
 316 {
 317     if (image == m_image_name)
 318         return;
 319     m_image_name = image;
 320     emit imageChanged();
 321     reset();
 322 }
 323
 324
 325 void QSGImageParticle::setColortable(const QUrl &table)
 326 {
 327     if (table == m_colortable_name)
 328         return;
 329     m_colortable_name = table;
 330     emit colortableChanged();
 331     reset();
 332 }
 333
 334 void QSGImageParticle::setSizetable(const QUrl &table)
 335 {
 336     if (table == m_sizetable_name)
 337         return;
 338     m_sizetable_name = table;
 339     emit sizetableChanged();
 340     reset();
 341 }
 342
 343 void QSGImageParticle::setOpacitytable(const QUrl &table)
 344 {
 345     if (table == m_opacitytable_name)
 346         return;
 347     m_opacitytable_name = table;
 348     emit opacitytableChanged();
 349     reset();
 350 }
 351
 352 void QSGImageParticle::setColor(const QColor &color)
 353 {
 354     if (color == m_color)
 355         return;
 356     m_color = color;
 357     emit colorChanged();
 358     if (perfLevel < Colored)
 359         reset();
 360 }
 361
 362 void QSGImageParticle::setColorVariation(qreal var)
 363 {
 364     if (var == m_color_variation)
 365         return;
 366     m_color_variation = var;
 367     emit colorVariationChanged();
 368     if (perfLevel < Colored)
 369         reset();
 370 }
 371
 372 void QSGImageParticle::setAlphaVariation(qreal arg)
 373 {
 374     if (m_alphaVariation != arg) {
 375         m_alphaVariation = arg;
 376         emit alphaVariationChanged(arg);
 377     }
 378     if (perfLevel < Colored)
 379         reset();
 380 }
 381
 382 void QSGImageParticle::setAlpha(qreal arg)
 383 {
 384     if (m_alpha != arg) {
 385         m_alpha = arg;
 386         emit alphaChanged(arg);
 387     }
 388     if (perfLevel < Colored)
 389         reset();
 390 }
 391
 392 void QSGImageParticle::setRedVariation(qreal arg)
 393 {
 394     if (m_redVariation != arg) {
 395         m_redVariation = arg;
 396         emit redVariationChanged(arg);
 397     }
 398     if (perfLevel < Colored)
 399         reset();
 400 }
 401
 402 void QSGImageParticle::setGreenVariation(qreal arg)
 403 {
 404     if (m_greenVariation != arg) {
 405         m_greenVariation = arg;
 406         emit greenVariationChanged(arg);
 407     }
 408     if (perfLevel < Colored)
 409         reset();
 410 }
 411
 412 void QSGImageParticle::setBlueVariation(qreal arg)
 413 {
 414     if (m_blueVariation != arg) {
 415         m_blueVariation = arg;
 416         emit blueVariationChanged(arg);
 417     }
 418     if (perfLevel < Colored)
 419         reset();
 420 }
 421
 422 void QSGImageParticle::setRotation(qreal arg)
 423 {
 424     if (m_rotation != arg) {
 425         m_rotation = arg;
 426         emit rotationChanged(arg);
 427     }
 428     if (perfLevel < Deformable)
 429         reset();
 430 }
 431
 432 void QSGImageParticle::setRotationVariation(qreal arg)
 433 {
 434     if (m_rotationVariation != arg) {
 435         m_rotationVariation = arg;
 436         emit rotationVariationChanged(arg);
 437     }
 438     if (perfLevel < Deformable)
 439         reset();
 440 }
 441
 442 void QSGImageParticle::setRotationSpeed(qreal arg)
 443 {
 444     if (m_rotationSpeed != arg) {
 445         m_rotationSpeed = arg;
 446         emit rotationSpeedChanged(arg);
 447     }
 448     if (perfLevel < Deformable)
 449         reset();
 450 }
 451
 452 void QSGImageParticle::setRotationSpeedVariation(qreal arg)
 453 {
 454     if (m_rotationSpeedVariation != arg) {
 455         m_rotationSpeedVariation = arg;
 456         emit rotationSpeedVariationChanged(arg);
 457     }
 458     if (perfLevel < Deformable)
 459         reset();
 460 }
 461
 462 void QSGImageParticle::setAutoRotation(bool arg)
 463 {
 464     if (m_autoRotation != arg) {
 465         m_autoRotation = arg;
 466         emit autoRotationChanged(arg);
 467     }
 468     if (perfLevel < Deformable)
 469         reset();
 470 }
 471
 472 void QSGImageParticle::setXVector(QSGStochasticDirection* arg)
 473 {
 474     if (m_xVector != arg) {
 475         m_xVector = arg;
 476         emit xVectorChanged(arg);
 477     }
 478     if (perfLevel < Deformable)
 479         reset();
 480 }
 481
 482 void QSGImageParticle::setYVector(QSGStochasticDirection* arg)
 483 {
 484     if (m_yVector != arg) {
 485         m_yVector = arg;
 486         emit yVectorChanged(arg);
 487     }
 488     if (perfLevel < Deformable)
 489         reset();
 490 }
 491
 492 void QSGImageParticle::setBloat(bool arg)
 493 {
 494     if (m_bloat != arg) {
 495         m_bloat = arg;
 496         emit bloatChanged(arg);
 497     }
 498     if (perfLevel < 9999)
 499         reset();
 500 }
 501
 502 void QSGImageParticle::reset()
 503 {
 504     QSGParticlePainter::reset();
 505     m_pleaseReset = true;
 506     update();
 507 }
 508
 509 void QSGImageParticle::createEngine()
 510 {
 511     if (m_spriteEngine)
 512         delete m_spriteEngine;
 513     if (m_sprites.count())
 514         m_spriteEngine = new QSGSpriteEngine(m_sprites, this);
 515     else
 516         m_spriteEngine = 0;
 517     reset();
 518 }
 519
 520 static QSGGeometry::Attribute SimpleParticle_Attributes[] = {
 521     { 0, 2, GL_FLOAT },             // Position
 522     { 1, 2, GL_FLOAT },             // TexCoord
 523     { 2, 4, GL_FLOAT },             // Data
 524     { 3, 4, GL_FLOAT }             // Vectors
 525 };
 526
 527 static QSGGeometry::AttributeSet SimpleParticle_AttributeSet =
 528 {
 529     4, // Attribute Count
 530     (2 + 2 + 4 + 4 ) * sizeof(float),
 531     SimpleParticle_Attributes
 532 };
 533
 534 static QSGGeometry::Attribute UltraParticle_Attributes[] = {
 535     { 0, 2, GL_FLOAT },             // Position
 536     { 1, 2, GL_FLOAT },             // TexCoord
 537     { 2, 4, GL_FLOAT },             // Data
 538     { 3, 4, GL_FLOAT },             // Vectors
 539     { 4, 4, GL_UNSIGNED_BYTE },     // Colors
 540     { 5, 4, GL_FLOAT },             // DeformationVectors
 541     { 6, 3, GL_FLOAT },             // Rotation
 542     { 7, 4, GL_FLOAT }              // Anim Data
 543 };
 544
 545 static QSGGeometry::AttributeSet UltraParticle_AttributeSet =
 546 {
 547     8, // Attribute Count
 548     (2 + 2 + 4 + 4 + 4 + 4 + 3) * sizeof(float) + 4 * sizeof(uchar),
 549     UltraParticle_Attributes
 550 };
 551
 552 QSGGeometryNode* QSGImageParticle::buildSimpleParticleNodes()
 553 {
 554     perfLevel = Simple;//TODO: Intermediate levels
 555     QImage image = QImage(m_image_name.toLocalFile());
 556     if (image.isNull()) {
 557         printf("UltraParticle: loading image failed... '%s'\n", qPrintable(m_image_name.toLocalFile()));
 558         return 0;
 559     }
 560
 561     if (m_material) {
 562         delete m_material;
 563         m_material = 0;
 564     }
 565
 566     m_material = new SimpleMaterial();
 567     m_material->texture = sceneGraphEngine()->createTextureFromImage(image);
 568     m_material->texture->setFiltering(QSGTexture::Linear);
 569     m_material->framecount = 1;
 570
 571     foreach (const QString &str, m_particles){
 572         int gIdx = m_system->m_groupIds[str];
 573         int count = m_system->m_groupData[gIdx]->size();
 574
 575         QSGGeometryNode* node = new QSGGeometryNode();
 576         m_nodes.insert(gIdx, node);
 577         node->setMaterial(m_material);
 578
 579         int vCount = count * 4;
 580         int iCount = count * 6;
 581
 582         QSGGeometry *g = new QSGGeometry(SimpleParticle_AttributeSet, vCount, iCount);
 583         node->setGeometry(g);
 584         g->setDrawingMode(GL_TRIANGLES);
 585
 586         SimpleVertex *vertices = (SimpleVertex *) g->vertexData();
 587         for (int p=0; p < count; ++p){
 588             commit(gIdx, p);
 589             vertices[0].tx = 0;
 590             vertices[0].ty = 0;
 591
 592             vertices[1].tx = 1;
 593             vertices[1].ty = 0;
 594
 595             vertices[2].tx = 0;
 596             vertices[2].ty = 1;
 597
 598             vertices[3].tx = 1;
 599             vertices[3].ty = 1;
 600
 601             vertices += 4;
 602         }
 603
 604         quint16 *indices = g->indexDataAsUShort();
 605         for (int i=0; i < count; ++i) {
 606             int o = i * 4;
 607             indices[0] = o;
 608             indices[1] = o + 1;
 609             indices[2] = o + 2;
 610             indices[3] = o + 1;
 611             indices[4] = o + 3;
 612             indices[5] = o + 2;
 613             indices += 6;
 614         }
 615     }
 616
 617     foreach (QSGGeometryNode* node, m_nodes){
 618         if (node == *(m_nodes.begin()))
 619                 continue;
 620         (*(m_nodes.begin()))->appendChildNode(node);
 621     }
 622
 623     return *(m_nodes.begin());
 624 }
 625
 626 QSGGeometryNode* QSGImageParticle::buildParticleNodes()
 627 {
 628     if (m_count * 4 > 0xffff) {
 629         printf("UltraParticle: Too many particles... \n");//### Why is this here?
 630         return 0;
 631     }
 632
 633     if (count() <= 0)
 634         return 0;
 635
 636     if (!m_sprites.count() && !m_bloat
 637             && m_colortable_name.isEmpty()
 638             && m_sizetable_name.isEmpty()
 639             && m_opacitytable_name.isEmpty()
 640             && !m_autoRotation
 641             && !m_rotation && !m_rotationVariation
 642             && !m_rotationSpeed && !m_rotationSpeedVariation
 643             && !m_alphaVariation && m_alpha == 1.0
 644             && !m_redVariation && !m_blueVariation && !m_greenVariation
 645             && !m_color.isValid()
 646             )
 647         return buildSimpleParticleNodes();
 648     perfLevel = Sprites;//TODO: intermediate levels
 649     if (!m_color.isValid())//But we're in colored level (or higher)
 650         m_color = QColor(Qt::white);
 651
 652     QImage image;
 653     if (m_sprites.count()){
 654         if (!m_spriteEngine) {
 655             qWarning() << "UltraParticle: No sprite engine...";
 656             return 0;
 657         }
 658         image = m_spriteEngine->assembledImage();
 659         if (image.isNull())//Warning is printed in engine
 660             return 0;
 661     }else{
 662         image = QImage(m_image_name.toLocalFile());
 663         if (image.isNull()) {
 664             printf("UltraParticle: loading image failed... '%s'\n", qPrintable(m_image_name.toLocalFile()));
 665             return 0;
 666         }
 667     }
 668
 669     if (m_material) {
 670         delete m_material;
 671         m_material = 0;
 672     }
 673
 674     QImage colortable(m_colortable_name.toLocalFile());
 675     QImage sizetable(m_sizetable_name.toLocalFile());
 676     QImage opacitytable(m_opacitytable_name.toLocalFile());
 677     m_material = new UltraMaterial();
 678     if (colortable.isNull())
 679         colortable = QImage(":defaultshaders/identitytable.png");
 680     if (sizetable.isNull())
 681         sizetable = QImage(":defaultshaders/identitytable.png");
 682     if (opacitytable.isNull())
 683         opacitytable = QImage(":defaultshaders/defaultFadeInOut.png");
 684     Q_ASSERT(!colortable.isNull());
 685     Q_ASSERT(!sizetable.isNull());
 686     Q_ASSERT(!opacitytable.isNull());
 687     m_material->colortable = sceneGraphEngine()->createTextureFromImage(colortable);
 688     m_material->sizetable = sceneGraphEngine()->createTextureFromImage(sizetable);
 689     m_material->opacitytable = sceneGraphEngine()->createTextureFromImage(opacitytable);
 690
 691     m_material->texture = sceneGraphEngine()->createTextureFromImage(image);
 692     m_material->texture->setFiltering(QSGTexture::Linear);
 693
 694     m_material->framecount = 1;
 695     if (m_spriteEngine){
 696         m_material->framecount = m_spriteEngine->maxFrames();
 697         m_spriteEngine->setCount(m_count);
 698     }
 699
 700     foreach (const QString &str, m_particles){
 701         int gIdx = m_system->m_groupIds[str];
 702         int count = m_system->m_groupData[gIdx]->size();
 703         QSGGeometryNode* node = new QSGGeometryNode();
 704         node->setMaterial(m_material);
 705
 706         m_nodes.insert(gIdx, node);
 707         m_idxStarts.insert(gIdx, m_lastIdxStart);
 708         m_lastIdxStart += count;
 709
 710         //Create Particle Geometry
 711         int vCount = count * 4;
 712         int iCount = count * 6;
 713
 714         QSGGeometry *g = new QSGGeometry(UltraParticle_AttributeSet, vCount, iCount);
 715         node->setGeometry(g);
 716         g->setDrawingMode(GL_TRIANGLES);
 717
 718         UltraVertex *vertices = (UltraVertex *) g->vertexData();
 719         for (int p=0; p < count; ++p) {
 720             commit(gIdx, p);//commit sets geometry for the node
 721
 722             vertices[0].tx = 0;
 723             vertices[0].ty = 0;
 724
 725             vertices[1].tx = 1;
 726             vertices[1].ty = 0;
 727
 728             vertices[2].tx = 0;
 729             vertices[2].ty = 1;
 730
 731             vertices[3].tx = 1;
 732             vertices[3].ty = 1;
 733
 734             vertices += 4;
 735         }
 736
 737         quint16 *indices = g->indexDataAsUShort();
 738         for (int i=0; i < count; ++i) {
 739             int o = i * 4;
 740             indices[0] = o;
 741             indices[1] = o + 1;
 742             indices[2] = o + 2;
 743             indices[3] = o + 1;
 744             indices[4] = o + 3;
 745             indices[5] = o + 2;
 746             indices += 6;
 747         }
 748
 749     }
 750
 751     foreach (QSGGeometryNode* node, m_nodes){
 752         if (node == *(m_nodes.begin()))
 753             continue;
 754         (*(m_nodes.begin()))->appendChildNode(node);
 755     }
 756
 757     return *(m_nodes.begin());
 758 }
 759
 760 QSGNode *QSGImageParticle::updatePaintNode(QSGNode *, UpdatePaintNodeData *)
 761 {
 762     if (m_pleaseReset){
 763         m_lastLevel = perfLevel;
 764
 765         delete m_rootNode;//Automatically deletes children
 766         m_rootNode = 0;
 767         m_nodes.clear();
 768
 769         m_idxStarts.clear();
 770         m_lastIdxStart = 0;
 771
 772         if (m_material)
 773             delete m_material;
 774         m_material = 0;
 775
 776         m_pleaseReset = false;
 777     }
 778
 779     if (m_system && m_system->isRunning())
 780         prepareNextFrame();
 781     if (m_rootNode){
 782         update();
 783         //### Should I be using dirty geometry too/instead?
 784         foreach (QSGGeometryNode* node, m_nodes)
 785             node->markDirty(QSGNode::DirtyMaterial);
 786     }
 787
 788     return m_rootNode;
 789 }
 790
 791 void QSGImageParticle::prepareNextFrame()
 792 {
 793     if (m_rootNode == 0){//TODO: Staggered loading (as emitted)
 794         m_rootNode = buildParticleNodes();
 795         if (m_rootNode == 0)
 796             return;
 797         //qDebug() << "Feature level: " << perfLevel;
 798     }
 799     qint64 timeStamp = m_system->systemSync(this);
 800
 801     qreal time = timeStamp / 1000.;
 802     m_material->timestamp = time;
 803
 804     //Advance State
 805     if (m_spriteEngine){//perfLevel == Sprites?//TODO: use signals?
 806
 807         m_material->animcount = m_spriteEngine->spriteCount();
 808         m_spriteEngine->updateSprites(timeStamp);
 809         foreach (const QString &str, m_particles){
 810             int gIdx = m_system->m_groupIds[str];
 811             int count = m_system->m_groupData[gIdx]->size();
 812
 813             UltraVertices *particles = (UltraVertices *) m_nodes[gIdx]->geometry()->vertexData();
 814             for (int i=0; i < count; i++){
 815                 int spriteIdx = m_idxStarts[gIdx] + i;
 816                 UltraVertices &p = particles[i];
 817                 int curIdx = m_spriteEngine->spriteState(spriteIdx);
 818                 if (curIdx != p.v1.animIdx){
 819                     p.v1.animIdx = p.v2.animIdx = p.v3.animIdx = p.v4.animIdx = curIdx;
 820                     p.v1.animT = p.v2.animT = p.v3.animT = p.v4.animT = m_spriteEngine->spriteStart(spriteIdx)/1000.0;
 821                     p.v1.frameCount = p.v2.frameCount = p.v3.frameCount = p.v4.frameCount = m_spriteEngine->spriteFrames(spriteIdx);
 822                     p.v1.frameDuration = p.v2.frameDuration = p.v3.frameDuration = p.v4.frameDuration = m_spriteEngine->spriteDuration(spriteIdx);
 823                 }
 824             }
 825         }
 826     }else{
 827         m_material->animcount = 1;
 828     }
 829 }
 830
 831 void QSGImageParticle::reloadColor(const Color4ub &c, QSGParticleData* d)
 832 {
 833     d->color = c;
 834     //TODO: get index for reload - or make function take an index
 835 }
 836
 837 void QSGImageParticle::initialize(int gIdx, int pIdx)
 838 {
 839     Color4ub color;
 840     QSGParticleData* datum = m_system->m_groupData[gIdx]->data[pIdx];
 841     qreal redVariation = m_color_variation + m_redVariation;
 842     qreal greenVariation = m_color_variation + m_greenVariation;
 843     qreal blueVariation = m_color_variation + m_blueVariation;
 844     int spriteIdx = m_idxStarts[gIdx] + datum->index;
 845     switch (perfLevel){//Fall-through is intended on all of them
 846         case Sprites:
 847             // Initial Sprite State
 848             datum->animT = datum->t;
 849             datum->animIdx = 0;
 850             if (m_spriteEngine){
 851                 m_spriteEngine->startSprite(spriteIdx);
 852                 datum->frameCount = m_spriteEngine->spriteFrames(spriteIdx);
 853                 datum->frameDuration = m_spriteEngine->spriteDuration(spriteIdx);
 854             }else{
 855                 datum->frameCount = 1;
 856                 datum->frameDuration = 9999;
 857             }
 858         case Tabled:
 859         case Deformable:
 860             //Initial Rotation
 861             if (m_xVector){
 862                 const QPointF &ret = m_xVector->sample(QPointF(datum->x, datum->y));
 863                 datum->xx = ret.x();
 864                 datum->xy = ret.y();
 865             }
 866             if (m_yVector){
 867                 const QPointF &ret = m_yVector->sample(QPointF(datum->x, datum->y));
 868                 datum->yx = ret.x();
 869                 datum->yy = ret.y();
 870             }
 871             datum->rotation =
 872                     (m_rotation + (m_rotationVariation - 2*((qreal)rand()/RAND_MAX)*m_rotationVariation) ) * CONV;
 873             datum->rotationSpeed =
 874                     (m_rotationSpeed + (m_rotationSpeedVariation - 2*((qreal)rand()/RAND_MAX)*m_rotationSpeedVariation) ) * CONV;
 875             datum->autoRotate = m_autoRotation?1.0:0.0;
 876         case Colored:
 877             //Color initialization
 878             // Particle color
 879             color.r = m_color.red() * (1 - redVariation) + rand() % 256 * redVariation;
 880             color.g = m_color.green() * (1 - greenVariation) + rand() % 256 * greenVariation;
 881             color.b = m_color.blue() * (1 - blueVariation) + rand() % 256 * blueVariation;
 882             color.a = m_alpha * m_color.alpha() * (1 - m_alphaVariation) + rand() % 256 * m_alphaVariation;
 883             datum->color = color;
 884         default:
 885             break;
 886     }
 887 }
 888
 889 void QSGImageParticle::commit(int gIdx, int pIdx)
 890 {
 891     if (m_pleaseReset)
 892         return;
 893     QSGGeometryNode *node = m_nodes[gIdx];
 894     if (!node)
 895         return;
 896     QSGParticleData* datum = m_system->m_groupData[gIdx]->data[pIdx];
 897
 898     node->setFlag(QSGNode::OwnsGeometry, false);
 899     UltraVertex *ultraVertices = (UltraVertex *) node->geometry()->vertexData();
 900     SimpleVertex *simpleVertices = (SimpleVertex *) node->geometry()->vertexData();
 901     switch (perfLevel){
 902     case Sprites:
 903         ultraVertices += pIdx*4;
 904         for (int i=0; i<4; i++){
 905             ultraVertices[i].x = datum->x  - m_systemOffset.x();
 906             ultraVertices[i].y = datum->y  - m_systemOffset.y();
 907             ultraVertices[i].t = datum->t;
 908             ultraVertices[i].lifeSpan = datum->lifeSpan;
 909             ultraVertices[i].size = datum->size;
 910             ultraVertices[i].endSize = datum->endSize;
 911             ultraVertices[i].sx = datum->sx;
 912             ultraVertices[i].sy = datum->sy;
 913             ultraVertices[i].ax = datum->ax;
 914             ultraVertices[i].ay = datum->ay;
 915             ultraVertices[i].xx = datum->xx;
 916             ultraVertices[i].xy = datum->xy;
 917             ultraVertices[i].yx = datum->yx;
 918             ultraVertices[i].yy = datum->yy;
 919             ultraVertices[i].rotation = datum->rotation;
 920             ultraVertices[i].rotationSpeed = datum->rotationSpeed;
 921             ultraVertices[i].autoRotate = datum->autoRotate;
 922             ultraVertices[i].animIdx = datum->animIdx;
 923             ultraVertices[i].frameDuration = datum->frameDuration;
 924             ultraVertices[i].frameCount = datum->frameCount;
 925             ultraVertices[i].animT = datum->animT;
 926             ultraVertices[i].color.r = datum->color.r;
 927             ultraVertices[i].color.g = datum->color.g;
 928             ultraVertices[i].color.b = datum->color.b;
 929             ultraVertices[i].color.a = datum->color.a;
 930         }
 931         break;
 932     case Tabled://TODO: Us
 933     case Deformable:
 934     case Colored:
 935     case Simple:
 936         simpleVertices += pIdx*4;
 937         for (int i=0; i<4; i++){
 938             simpleVertices[i].x = datum->x - m_systemOffset.x();
 939             simpleVertices[i].y = datum->y - m_systemOffset.y();
 940             simpleVertices[i].t = datum->t;
 941             simpleVertices[i].lifeSpan = datum->lifeSpan;
 942             simpleVertices[i].size = datum->size;
 943             simpleVertices[i].endSize = datum->endSize;
 944             simpleVertices[i].sx = datum->sx;
 945             simpleVertices[i].sy = datum->sy;
 946             simpleVertices[i].ax = datum->ax;
 947             simpleVertices[i].ay = datum->ay;
 948         }
 949         break;
 950     default:
 951         break;
 952     }
 953
 954     node->setFlag(QSGNode::OwnsGeometry, true);
 955 }
 956
 957
 958
 959 QT_END_NAMESPACE