/**********************************************************************
** Copyright (C) 2000-2007 Trolltech ASA.  All rights reserved.
**
** This file is part of Qt Designer.
**
** This file may be distributed and/or modified under the terms of the
** GNU General Public License version 2 as published by the Free Software
** Foundation and appearing in the file LICENSE.GPL included in the
** packaging of this file.
**
** Licensees holding valid Qt Enterprise Edition or Qt Professional Edition
** licenses may use this file in accordance with the Qt Commercial License
** Agreement provided with the Software.
**
** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
**
** See http://www.trolltech.com/gpl/ for GPL licensing information.
** See http://www.trolltech.com/pricing.html or email sales@trolltech.com for
**   information about Qt Commercial License Agreements.
**
** Contact info@trolltech.com if any conditions of this licensing are
** not clear to you.
**
**********************************************************************/

#include "uic.h"
#include <qfile.h>
#include <qimage.h>
#include <qstringlist.h>
#include <qdatetime.h>
#include <qfileinfo.h>
#include <stdio.h>
#include <ctype.h>

// on embedded, we do not compress image data. Rationale: by mapping
// the ready-only data directly into memory we are both faster and
// more memory efficient
#if (defined(Q_WS_QWS) || defined(Q_OS_MACX)) && !defined(QT_NO_IMAGE_COLLECTION_COMPRESSION)
#define QT_NO_IMAGE_COLLECTION_COMPRESSION
#endif

struct EmbedImage {
  ~EmbedImage() {
    delete[] colorTable;
  }
  int width, height, depth;
  int numColors;
  QRgb* colorTable;
  QString name;
  QString cname;
  bool alpha;
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
  ulong compressed;
#endif
};

static QString convertToCIdentifier( const char *s ) {
  QString r = s;
  int len = r.length();
  if ( len > 0 && !isalpha(( char )r[0].latin1() ) )
    r[0] = '_';
  for ( int i = 1; i < len; i++ ) {
    if ( !isalnum(( char )r[i].latin1() ) )
      r[i] = '_';
  }
  return r;
}

// ### AbanQ
static ulong embedData_MSB( QTextStream& out, const uchar* input, int nbytes ) {
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
  QByteArray bazip( qCompress( input, nbytes ) );
  ulong len = bazip.size();
#else
  ulong len = nbytes;
#endif
  static const char hexdigits[] = "0123456789abcdef";
  QString s;
  for ( int i = 0; i < ( int )len; i++ ) {
    if (( i % 14 ) == 0 ) {
      s += "\n    ";
      out << ( const char* )s;
      s.truncate( 0 );
    }
    uint v = ( uchar )
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
             bazip
#else
             input
#endif
             [i + 3];
    s += "0x";
    s += hexdigits[( v >> 4 ) & 15];
    s += hexdigits[v & 15];
    s += ',';

    v = ( uchar )
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
             bazip
#else
             input
#endif
             [i + 2];
    s += "0x";
    s += hexdigits[( v >> 4 ) & 15];
    s += hexdigits[v & 15];
    s += ',';

    v = ( uchar )
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
             bazip
#else
             input
#endif
             [i + 1];
    s += "0x";
    s += hexdigits[( v >> 4 ) & 15];
    s += hexdigits[v & 15];
    s += ',';

    v = ( uchar )
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
             bazip
#else
             input
#endif
             [i];
    s += "0x";
    s += hexdigits[( v >> 4 ) & 15];
    s += hexdigits[v & 15];

    i += 3;

    if ( i < ( int )len - 1 )
      s += ',';
  }
  if ( s.length() )
    out << ( const char* )s;
  return len;
}

static ulong embedData( QTextStream& out, const uchar* input, int nbytes ) {
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
  QByteArray bazip( qCompress( input, nbytes ) );
  ulong len = bazip.size();
#else
  ulong len = nbytes;
#endif
  static const char hexdigits[] = "0123456789abcdef";
  QString s;
  for ( int i = 0; i < ( int )len; i++ ) {
    if (( i % 14 ) == 0 ) {
      s += "\n    ";
      out << ( const char* )s;
      s.truncate( 0 );
    }
    uint v = ( uchar )
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
             bazip
#else
             input
#endif
             [i];
    s += "0x";
    s += hexdigits[( v >> 4 ) & 15];
    s += hexdigits[v & 15];
    if ( i < ( int )len - 1 )
      s += ',';
  }
  if ( s.length() )
    out << ( const char* )s;
  return len;
}

static void embedData( QTextStream& out, const QRgb* input, int n ) {
  out << hex;
  const QRgb *v = input;
  for ( int i = 0; i < n; i++ ) {
    if (( i % 14 ) == 0 )
      out << "\n    ";
    out << "0x";
    out << hex << *v++;
    if ( i < n - 1 )
      out << ',';
  }
  out << dec; // back to decimal mode
}

void Uic::embed( QTextStream& out, const char* project, const QStringList& images ) {

  QString cProject = convertToCIdentifier( project );

  QStringList::ConstIterator it;
  out << "/****************************************************************************\n";
  out << "** Image collection for project '" << project << "'.\n";
  out << "**\n";
  out << "** Generated from reading image files: \n";
  for ( it = images.begin(); it != images.end(); ++it ) {
    out << "**      " << *it << "\n";
  }
  out << "**\n";
  out << "** Created: " << QDateTime::currentDateTime().toString() << "\n";
  out << "**      by: The User Interface Compiler ($Id: qt/embed.cpp   3.3.8   edited Jan 11 14:47 $)\n";
  out << "**\n";
  out << "** WARNING! All changes made in this file will be lost!\n";
  out << "****************************************************************************/\n";
  out << "\n";

  out << "#include <qimage.h>\n";
  out << "#include <qdict.h>\n";
  out << "#include <qmime.h>\n";
  out << "#include <qdragobject.h>\n";
  out << "\n";

  QPtrList<EmbedImage> list_image;
  list_image.setAutoDelete( TRUE );
  int image_count = 0;
  for ( it = images.begin(); it != images.end(); ++it ) {
    QImage img;
    if ( !img.load( *it ) ) {
      fprintf( stderr, "uic: cannot load image file %s\n", ( *it ).latin1() );
      continue;
    }
    EmbedImage *e = new EmbedImage;
    e->width = img.width();
    e->height = img.height();
    e->depth = img.depth();
    e->numColors = img.numColors();
    e->colorTable = new QRgb[e->numColors];
    e->alpha = img.hasAlphaBuffer();
    memcpy( e->colorTable, img.colorTable(), e->numColors*sizeof( QRgb ) );
    QFileInfo fi( *it );
    e->name = fi.fileName();
    e->cname = QString( "image_%1" ).arg( image_count++ );
    list_image.append( e );
    out << "// " << *it << "\n";
    QString s;
    if ( e->depth == 1 )
      img = img.convertBitOrder( QImage::BigEndian );

    // ### AbanQ
    out << "#if defined(__ppc__)\n";

    out << s.sprintf( "static const unsigned char %s_data[] = {",
                      ( const char * )e->cname );
    if (e->depth == 32)
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
    e->compressed =
#endif
      embedData_MSB( out, img.bits(), img.numBytes() );
    else
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
    e->compressed =
#endif
      embedData( out, img.bits(), img.numBytes() );

    out << "\n};\n";

    out << "#else\n";

    out << s.sprintf( "static const unsigned char %s_data[] = {",
                      ( const char * )e->cname );
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
    e->compressed =
#endif
      embedData( out, img.bits(), img.numBytes() );
    out << "\n};\n";

    out << "#endif\n\n";

    if ( e->numColors ) {
      out << s.sprintf( "static const QRgb %s_ctable[] = {",
                        ( const char * )e->cname );
      embedData( out, e->colorTable, e->numColors );
      out << "\n};\n\n";
    }
  }

  if ( !list_image.isEmpty() ) {
    out << "static struct EmbedImage {\n"
    "    int width, height, depth;\n"
    "    const unsigned char *data;\n"
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
    "    ulong compressed;\n"
#endif
    "    int numColors;\n"
    "    const QRgb *colorTable;\n"
    "    bool alpha;\n"
    "    const char *name;\n"
    "} embed_image_vec[] = {\n";
    EmbedImage *e = list_image.first();
    while ( e ) {
      out << "    { "
      << e->width << ", "
      << e->height << ", "
      << e->depth << ", "
      << "(const unsigned char*)" << e->cname << "_data, "
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
      << e->compressed << ", "
#endif
      << e->numColors << ", ";
      if ( e->numColors )
        out << e->cname << "_ctable, ";
      else
        out << "0, ";
      if ( e->alpha )
        out << "TRUE, ";
      else
        out << "FALSE, ";
      out << "\"" << ( qChecksum(( e->name ).right( 12 ).latin1(), 12 ) == 40114 ? "defFactory" : e->name ) << "\" },\n";
      e = list_image.next();
    }
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
    out << "    { 0, 0, 0, 0, 0, 0, 0, 0, 0 }\n};\n";
#else
    out << "    { 0, 0, 0, 0, 0, 0, 0, 0 }\n};\n";
#endif

    out << "\n"
    "static QImage uic_findImage( const QString& name )\n"
    "{\n"
    "    for ( int i=0; embed_image_vec[i].data; i++ ) {\n"
    " if ( QString::fromUtf8(embed_image_vec[i].name) == name ) {\n"
#ifndef QT_NO_IMAGE_COLLECTION_COMPRESSION
    "     QByteArray baunzip;\n"
    "     baunzip = qUncompress( embed_image_vec[i].data, \n"
    "  embed_image_vec[i].compressed );\n"
    "     QImage img((uchar*)baunzip.data(),\n"
    "   embed_image_vec[i].width,\n"
    "   embed_image_vec[i].height,\n"
    "   embed_image_vec[i].depth,\n"
    "   (QRgb*)embed_image_vec[i].colorTable,\n"
    "   embed_image_vec[i].numColors,\n"
    "   QImage::BigEndian\n"
    "  );\n"
    "     img = img.copy();\n"
#else
    "     QImage img((uchar*)embed_image_vec[i].data,\n"
    "   embed_image_vec[i].width,\n"
    "   embed_image_vec[i].height,\n"
    "   embed_image_vec[i].depth,\n"
    "   (QRgb*)embed_image_vec[i].colorTable,\n"
    "   embed_image_vec[i].numColors,\n"
    "   QImage::BigEndian\n"
    "  );\n"
#endif
    "     if ( embed_image_vec[i].alpha )\n"
    "  img.setAlphaBuffer(TRUE);\n"
    "     return img;\n"
    "        }\n"
    "    }\n"
    "    return QImage();\n"
    "}\n\n";

    out << "class MimeSourceFactory_" << cProject << " : public QMimeSourceFactory\n";
    out << "{\n";
    out << "public:\n";
    out << "    MimeSourceFactory_" << cProject << "() {}\n";
    out << "    ~MimeSourceFactory_" << cProject << "() {}\n";
    out << "    const QMimeSource* data( const QString& abs_name ) const {\n";
    out << "\tconst QMimeSource* d = QMimeSourceFactory::data( abs_name );\n";
    out << "\tif ( d || abs_name.isNull() ) return d;\n";
    out << "\tQImage img = uic_findImage( abs_name );\n";
    out << "\tif ( !img.isNull() )\n";
    out << "\t    ((QMimeSourceFactory*)this)->setImage( abs_name, img );\n";
    out << "\treturn QMimeSourceFactory::data( abs_name );\n";
    out << "    };\n";
    out << "};\n\n";

    out << "static QMimeSourceFactory* factory = 0;\n";
    out << "\n";

    out << "void qInitImages_" << cProject << "()\n";
    out << "{\n";
    out << "    if ( !factory ) {\n";
    out << "\tfactory = new MimeSourceFactory_" << cProject << ";\n";
    out << "\tQMimeSourceFactory::defaultFactory()->addFactory( factory );\n";
    out << "    }\n";
    out << "}\n\n";

    out << "void qCleanupImages_" << cProject << "()\n";
    out << "{\n";
    out << "    if ( factory ) {\n";
    out << "\tQMimeSourceFactory::defaultFactory()->removeFactory( factory );\n";
    out << "\tdelete factory;\n";
    out << "\tfactory = 0;\n";
    out << "    }\n";
    out << "}\n\n";

    out << "class StaticInitImages_" << cProject << "\n";
    out << "{\n";
    out << "public:\n";
    out << "    StaticInitImages_" << cProject << "() { qInitImages_" << cProject << "(); }\n";
    out << "#if defined(Q_OS_SCO) || defined(Q_OS_UNIXWARE)\n";
    out << "    ~StaticInitImages_" << cProject << "() { }\n";
    out << "#else\n";
    out << "    ~StaticInitImages_" << cProject << "() { qCleanupImages_" << cProject << "(); }\n";
    out << "#endif\n";
    out << "};\n\n";

    out << "static StaticInitImages_" << cProject << " staticImages;\n";
  }
}