Форум программистов [Powered by Invision Power Board]

Код

QImage *LocationEditor::getEquirect()
{
  int i, j, f, w, h, k, l;
  float az, el, x, y;
  QImage *faces[6];
  w = 360;
  h = 180;

  QStringList cubeMapImages = getCubeMapImages();

  for (i = 0; i < cubeMapImages.count(); ++i)
  {
    QImage image;
    QString s = QString("%1/%2").arg( locationName ).arg( cubeMapImages.at(i) );

    if (!image.load( s, PathUtils::getExt(s).toUpper().toAscii().constData() ))
    {
      qDebug("Can't load image");
      return 0;
    }

    faces[i] = new QImage( image.convertToFormat( QImage::Format_RGB888 ) );
  }

  QImage *equirect = new QImage( w, h, QImage::Format_RGB888 );

  for (i = 0; i < w; ++i)
  {
    for (j = 0; j < h; ++j)
    {
      az = (i + 0.5f) / w * 2 * (float)M_PI;
      el = ((-0.5f - j) / h + 0.5f) * (float)M_PI;
      f = 0;

      while (az > M_PI / 4)
      {
        az -= M_PI / 2;
        f++;
      }

      if (f == 4)
        f = 0;

      x = tanf( az );
      y = -sqrtf( 1 + x * x ) * tanf( el );

      if (y < -1)
      {
        f = 4;
        az = (i + 0.5f) / w * 360 * (float)M_PI / 180;
        x = sinf( az ) / tanf( el );
        y = cosf( az ) / tanf( el );
      }
      else if (y > 1)
      {
        f = 5;
        az = (i + 0.5f) / w * 360 * (float)M_PI / 180;
        x = -sinf( az ) / tanf( el );
        y = cosf( az ) / tanf( el );
      }

      k = (int)floorf( (faces[f]->width() - 1) * (x + 1) / 2 + 0.5f );
      l = (int)floorf( (faces[f]->height() - 1) * (y + 1) / 2 + 0.5f );

      ((unsigned char *)(equirect->bits()))[ j * equirect->bytesPerLine() + 3 * i + 0 ] = ((unsigned char *)(faces[f]->bits()))[ l * faces[f]->bytesPerLine() + 3 * k + 0 ];
      ((unsigned char *)(equirect->bits()))[ j * equirect->bytesPerLine() + 3 * i + 1 ] = ((unsigned char *)(faces[f]->bits()))[ l * faces[f]->bytesPerLine() + 3 * k + 1 ];
      ((unsigned char *)(equirect->bits()))[ j * equirect->bytesPerLine() + 3 * i + 2 ] = ((unsigned char *)(faces[f]->bits()))[ l * faces[f]->bytesPerLine() + 3 * k + 2 ];
    }
  }

  for (int i = 0; i < cubeMapImages.count(); ++i)
  {
    delete faces[i];
    faces[i] = 0;
  }

  return equirect;
}