source: trunk/src/tools/rcc/rcc.cpp@ 769

Last change on this file since 769 was 769, checked in by Dmitry A. Kuminov, 15 years ago

trunk: Merged in qt 4.6.3 sources from branches/vendor/nokia/qt.

File size: 32.5 KB
Line 
1/****************************************************************************
2**
3** Copyright (C) 2010 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 tools applications of the Qt Toolkit.
8**
9** $QT_BEGIN_LICENSE:LGPL$
10** Commercial Usage
11** Licensees holding valid Qt Commercial licenses may use this file in
12** accordance with the Qt Commercial License Agreement provided with the
13** Software or, alternatively, in accordance with the terms contained in
14** a written agreement between you and Nokia.
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** GNU General Public License Usage
29** Alternatively, this file may be used under the terms of the GNU
30** General Public License version 3.0 as published by the Free Software
31** Foundation and appearing in the file LICENSE.GPL included in the
32** packaging of this file. Please review the following information to
33** ensure the GNU General Public License version 3.0 requirements will be
34** met: http://www.gnu.org/copyleft/gpl.html.
35**
36** If you have questions regarding the use of this file, please contact
37** Nokia at qt-info@nokia.com.
38** $QT_END_LICENSE$
39**
40****************************************************************************/
41
42#include "rcc.h"
43
44#include <QtCore/QByteArray>
45#include <QtCore/QDateTime>
46#include <QtCore/QDebug>
47#include <QtCore/QDir>
48#include <QtCore/QDirIterator>
49#include <QtCore/QFile>
50#include <QtCore/QIODevice>
51#include <QtCore/QLocale>
52#include <QtCore/QStack>
53
54#include <QtXml/QDomDocument>
55
56QT_BEGIN_NAMESPACE
57
58enum {
59 CONSTANT_USENAMESPACE = 1,
60 CONSTANT_COMPRESSLEVEL_DEFAULT = -1,
61 CONSTANT_COMPRESSTHRESHOLD_DEFAULT = 70
62};
63
64
65#define writeString(s) write(s, sizeof(s))
66
67void RCCResourceLibrary::write(const char *str, int len)
68{
69 --len; // trailing \0 on string literals...
70 int n = m_out.size();
71 m_out.resize(n + len);
72 memcpy(m_out.data() + n, str, len);
73}
74
75void RCCResourceLibrary::writeByteArray(const QByteArray &other)
76{
77 m_out.append(other);
78}
79
80static inline QString msgOpenReadFailed(const QString &fname, const QString &why)
81{
82 return QString::fromUtf8("Unable to open %1 for reading: %2\n").arg(fname).arg(why);
83}
84
85
86///////////////////////////////////////////////////////////
87//
88// RCCFileInfo
89//
90///////////////////////////////////////////////////////////
91
92class RCCFileInfo
93{
94public:
95 enum Flags
96 {
97 NoFlags = 0x00,
98 Compressed = 0x01,
99 Directory = 0x02
100 };
101
102 RCCFileInfo(const QString &name = QString(), const QFileInfo &fileInfo = QFileInfo(),
103 QLocale::Language language = QLocale::C,
104 QLocale::Country country = QLocale::AnyCountry,
105 uint flags = NoFlags,
106 int compressLevel = CONSTANT_COMPRESSLEVEL_DEFAULT,
107 int compressThreshold = CONSTANT_COMPRESSTHRESHOLD_DEFAULT);
108 ~RCCFileInfo();
109
110 QString resourceName() const;
111
112public:
113 qint64 writeDataBlob(RCCResourceLibrary &lib, qint64 offset, QString *errorMessage);
114 qint64 writeDataName(RCCResourceLibrary &, qint64 offset);
115 void writeDataInfo(RCCResourceLibrary &lib);
116
117 int m_flags;
118 QString m_name;
119 QLocale::Language m_language;
120 QLocale::Country m_country;
121 QFileInfo m_fileInfo;
122 RCCFileInfo *m_parent;
123 QHash<QString, RCCFileInfo*> m_children;
124 int m_compressLevel;
125 int m_compressThreshold;
126
127 qint64 m_nameOffset;
128 qint64 m_dataOffset;
129 qint64 m_childOffset;
130};
131
132RCCFileInfo::RCCFileInfo(const QString &name, const QFileInfo &fileInfo,
133 QLocale::Language language, QLocale::Country country, uint flags,
134 int compressLevel, int compressThreshold)
135{
136 m_name = name;
137 m_fileInfo = fileInfo;
138 m_language = language;
139 m_country = country;
140 m_flags = flags;
141 m_parent = 0;
142 m_nameOffset = 0;
143 m_dataOffset = 0;
144 m_childOffset = 0;
145 m_compressLevel = compressLevel;
146 m_compressThreshold = compressThreshold;
147}
148
149RCCFileInfo::~RCCFileInfo()
150{
151 qDeleteAll(m_children);
152}
153
154QString RCCFileInfo::resourceName() const
155{
156 QString resource = m_name;
157 for (RCCFileInfo *p = m_parent; p; p = p->m_parent)
158 resource = resource.prepend(p->m_name + QLatin1Char('/'));
159 return QLatin1Char(':') + resource;
160}
161
162void RCCFileInfo::writeDataInfo(RCCResourceLibrary &lib)
163{
164 const bool text = (lib.m_format == RCCResourceLibrary::C_Code);
165 //some info
166 if (text) {
167 if (m_language != QLocale::C) {
168 lib.writeString(" // ");
169 lib.writeByteArray(resourceName().toLocal8Bit());
170 lib.writeString(" [");
171 lib.writeByteArray(QByteArray::number(m_country));
172 lib.writeString("::");
173 lib.writeByteArray(QByteArray::number(m_language));
174 lib.writeString("[\n ");
175 } else {
176 lib.writeString(" // ");
177 lib.writeByteArray(resourceName().toLocal8Bit());
178 lib.writeString("\n ");
179 }
180 }
181
182 //pointer data
183 if (m_flags & RCCFileInfo::Directory) {
184 // name offset
185 lib.writeNumber4(m_nameOffset);
186
187 // flags
188 lib.writeNumber2(m_flags);
189
190 // child count
191 lib.writeNumber4(m_children.size());
192
193 // first child offset
194 lib.writeNumber4(m_childOffset);
195 } else {
196 // name offset
197 lib.writeNumber4(m_nameOffset);
198
199 // flags
200 lib.writeNumber2(m_flags);
201
202 // locale
203 lib.writeNumber2(m_country);
204 lib.writeNumber2(m_language);
205
206 //data offset
207 lib.writeNumber4(m_dataOffset);
208 }
209 if (text)
210 lib.writeChar('\n');
211}
212
213qint64 RCCFileInfo::writeDataBlob(RCCResourceLibrary &lib, qint64 offset,
214 QString *errorMessage)
215{
216 const bool text = (lib.m_format == RCCResourceLibrary::C_Code);
217
218 //capture the offset
219 m_dataOffset = offset;
220
221 //find the data to be written
222 QFile file(m_fileInfo.absoluteFilePath());
223 if (!file.open(QFile::ReadOnly)) {
224 *errorMessage = msgOpenReadFailed(m_fileInfo.absoluteFilePath(), file.errorString());
225 return 0;
226 }
227 QByteArray data = file.readAll();
228
229#ifndef QT_NO_COMPRESS
230 // Check if compression is useful for this file
231 if (m_compressLevel != 0 && data.size() != 0) {
232 QByteArray compressed =
233 qCompress(reinterpret_cast<uchar *>(data.data()), data.size(), m_compressLevel);
234
235 int compressRatio = int(100.0 * (data.size() - compressed.size()) / data.size());
236 if (compressRatio >= m_compressThreshold) {
237 data = compressed;
238 m_flags |= Compressed;
239 }
240 }
241#endif // QT_NO_COMPRESS
242
243 // some info
244 if (text) {
245 lib.writeString(" // ");
246 lib.writeByteArray(m_fileInfo.absoluteFilePath().toLocal8Bit());
247 lib.writeString("\n ");
248 }
249
250 // write the length
251
252 lib.writeNumber4(data.size());
253 if (text)
254 lib.writeString("\n ");
255 offset += 4;
256
257 // write the payload
258 const char *p = data.constData();
259 if (text) {
260 for (int i = data.size(), j = 0; --i >= 0; --j) {
261 lib.writeHex(*p++);
262 if (j == 0) {
263 lib.writeString("\n ");
264 j = 16;
265 }
266 }
267 } else {
268 for (int i = data.size(); --i >= 0; )
269 lib.writeChar(*p++);
270 }
271 offset += data.size();
272
273 // done
274 if (text)
275 lib.writeString("\n ");
276 return offset;
277}
278
279qint64 RCCFileInfo::writeDataName(RCCResourceLibrary &lib, qint64 offset)
280{
281 const bool text = (lib.m_format == RCCResourceLibrary::C_Code);
282
283 // capture the offset
284 m_nameOffset = offset;
285
286 // some info
287 if (text) {
288 lib.writeString(" // ");
289 lib.writeByteArray(m_name.toLocal8Bit());
290 lib.writeString("\n ");
291 }
292
293 // write the length
294 lib.writeNumber2(m_name.length());
295 if (text)
296 lib.writeString("\n ");
297 offset += 2;
298
299 // write the hash
300 lib.writeNumber4(qHash(m_name));
301 if (text)
302 lib.writeString("\n ");
303 offset += 4;
304
305 // write the m_name
306 const QChar *unicode = m_name.unicode();
307 for (int i = 0; i < m_name.length(); ++i) {
308 lib.writeNumber2(unicode[i].unicode());
309 if (text && i % 16 == 0)
310 lib.writeString("\n ");
311 }
312 offset += m_name.length()*2;
313
314 // done
315 if (text)
316 lib.writeString("\n ");
317 return offset;
318}
319
320
321///////////////////////////////////////////////////////////
322//
323// RCCResourceLibrary
324//
325///////////////////////////////////////////////////////////
326
327RCCResourceLibrary::Strings::Strings() :
328 TAG_RCC(QLatin1String("RCC")),
329 TAG_RESOURCE(QLatin1String("qresource")),
330 TAG_FILE(QLatin1String("file")),
331 ATTRIBUTE_LANG(QLatin1String("lang")),
332 ATTRIBUTE_PREFIX(QLatin1String("prefix")),
333 ATTRIBUTE_ALIAS(QLatin1String("alias")),
334 ATTRIBUTE_THRESHOLD(QLatin1String("threshold")),
335 ATTRIBUTE_COMPRESS(QLatin1String("compress"))
336{
337}
338
339RCCResourceLibrary::RCCResourceLibrary()
340 : m_root(0),
341 m_format(C_Code),
342 m_verbose(false),
343 m_compressLevel(CONSTANT_COMPRESSLEVEL_DEFAULT),
344 m_compressThreshold(CONSTANT_COMPRESSTHRESHOLD_DEFAULT),
345 m_treeOffset(0),
346 m_namesOffset(0),
347 m_dataOffset(0),
348 m_useNameSpace(CONSTANT_USENAMESPACE),
349 m_errorDevice(0)
350{
351 m_out.reserve(30 * 1000 * 1000);
352}
353
354RCCResourceLibrary::~RCCResourceLibrary()
355{
356 delete m_root;
357}
358
359bool RCCResourceLibrary::interpretResourceFile(QIODevice *inputDevice,
360 const QString &fname, QString currentPath, bool ignoreErrors)
361{
362 Q_ASSERT(m_errorDevice);
363 const QChar slash = QLatin1Char('/');
364 if (!currentPath.isEmpty() && !currentPath.endsWith(slash))
365 currentPath += slash;
366
367 QDomDocument document;
368 {
369 QString errorMsg;
370 int errorLine = 0;
371 int errorColumn = 0;
372 if (!document.setContent(inputDevice, &errorMsg, &errorLine, &errorColumn)) {
373 if (ignoreErrors)
374 return true;
375 const QString msg = QString::fromUtf8("RCC Parse Error: '%1' Line: %2 Column: %3 [%4]\n").arg(fname).arg(errorLine).arg(errorColumn).arg(errorMsg);
376 m_errorDevice->write(msg.toUtf8());
377 return false;
378 }
379 }
380
381 QDomElement domRoot = document.firstChildElement(m_strings.TAG_RCC).toElement();
382 if (!domRoot.isNull() && domRoot.tagName() == m_strings.TAG_RCC) {
383 for (QDomNode node = domRoot.firstChild(); !node.isNull(); node = node.nextSibling()) {
384 if (!node.isElement())
385 continue;
386
387 QDomElement child = node.toElement();
388 if (!child.isNull() && child.tagName() == m_strings.TAG_RESOURCE) {
389 QLocale::Language language = QLocale::c().language();
390 QLocale::Country country = QLocale::c().country();
391
392 if (child.hasAttribute(m_strings.ATTRIBUTE_LANG)) {
393 QString attribute = child.attribute(m_strings.ATTRIBUTE_LANG);
394 QLocale lang = QLocale(attribute);
395 language = lang.language();
396 if (2 == attribute.length()) {
397 // Language only
398 country = QLocale::AnyCountry;
399 } else {
400 country = lang.country();
401 }
402 }
403
404 QString prefix;
405 if (child.hasAttribute(m_strings.ATTRIBUTE_PREFIX))
406 prefix = child.attribute(m_strings.ATTRIBUTE_PREFIX);
407 if (!prefix.startsWith(slash))
408 prefix.prepend(slash);
409 if (!prefix.endsWith(slash))
410 prefix += slash;
411
412 for (QDomNode res = child.firstChild(); !res.isNull(); res = res.nextSibling()) {
413 if (res.isElement() && res.toElement().tagName() == m_strings.TAG_FILE) {
414
415 QString fileName(res.firstChild().toText().data());
416 if (fileName.isEmpty()) {
417 const QString msg = QString::fromUtf8("RCC: Warning: Null node in XML of '%1'\n").arg(fname);
418 m_errorDevice->write(msg.toUtf8());
419 }
420 QString alias;
421 if (res.toElement().hasAttribute(m_strings.ATTRIBUTE_ALIAS))
422 alias = res.toElement().attribute(m_strings.ATTRIBUTE_ALIAS);
423 else
424 alias = fileName;
425
426 int compressLevel = m_compressLevel;
427 if (res.toElement().hasAttribute(m_strings.ATTRIBUTE_COMPRESS))
428 compressLevel = res.toElement().attribute(m_strings.ATTRIBUTE_COMPRESS).toInt();
429 int compressThreshold = m_compressThreshold;
430 if (res.toElement().hasAttribute(m_strings.ATTRIBUTE_THRESHOLD))
431 compressThreshold = res.toElement().attribute(m_strings.ATTRIBUTE_THRESHOLD).toInt();
432
433 // Special case for -no-compress. Overrides all other settings.
434 if (m_compressLevel == -2)
435 compressLevel = 0;
436
437 alias = QDir::cleanPath(alias);
438 while (alias.startsWith(QLatin1String("../")))
439 alias.remove(0, 3);
440 alias = QDir::cleanPath(m_resourceRoot) + prefix + alias;
441
442 QString absFileName = fileName;
443 if (QDir::isRelativePath(absFileName))
444 absFileName.prepend(currentPath);
445 QFileInfo file(absFileName);
446 if (!file.exists()) {
447 m_failedResources.push_back(absFileName);
448 const QString msg = QString::fromUtf8("RCC: Error in '%1': Cannot find file '%2'\n").arg(fname).arg(fileName);
449 m_errorDevice->write(msg.toUtf8());
450 if (ignoreErrors)
451 continue;
452 else
453 return false;
454 } else if (file.isFile()) {
455 const bool arc =
456 addFile(alias,
457 RCCFileInfo(alias.section(slash, -1),
458 file,
459 language,
460 country,
461 RCCFileInfo::NoFlags,
462 compressLevel,
463 compressThreshold)
464 );
465 if (!arc)
466 m_failedResources.push_back(absFileName);
467 } else {
468 QDir dir;
469 if (file.isDir()) {
470 dir.setPath(file.filePath());
471 } else {
472 dir.setPath(file.path());
473 dir.setNameFilters(QStringList(file.fileName()));
474 if (alias.endsWith(file.fileName()))
475 alias = alias.left(alias.length()-file.fileName().length());
476 }
477 if (!alias.endsWith(slash))
478 alias += slash;
479 QDirIterator it(dir, QDirIterator::FollowSymlinks|QDirIterator::Subdirectories);
480 while (it.hasNext()) {
481 it.next();
482 QFileInfo child(it.fileInfo());
483 if (child.fileName() != QLatin1String(".") && child.fileName() != QLatin1String("..")) {
484 const bool arc =
485 addFile(alias + child.fileName(),
486 RCCFileInfo(child.fileName(),
487 child,
488 language,
489 country,
490 RCCFileInfo::NoFlags,
491 compressLevel,
492 compressThreshold)
493 );
494 if (!arc)
495 m_failedResources.push_back(child.fileName());
496 }
497 }
498 }
499 }
500 }
501 }
502 }
503 }
504 if (m_root == 0) {
505 const QString msg = QString::fromUtf8("RCC: Warning: No resources in '%1'.\n").arg(fname);
506 m_errorDevice->write(msg.toUtf8());
507 if (!ignoreErrors && m_format == Binary) {
508 // create dummy entry, otherwise loading qith QResource will crash
509 m_root = new RCCFileInfo(QString(), QFileInfo(),
510 QLocale::C, QLocale::AnyCountry, RCCFileInfo::Directory);
511 }
512 }
513
514 return true;
515}
516
517bool RCCResourceLibrary::addFile(const QString &alias, const RCCFileInfo &file)
518{
519 Q_ASSERT(m_errorDevice);
520 if (file.m_fileInfo.size() > 0xffffffff) {
521 const QString msg = QString::fromUtf8("File too big: %1\n").arg(file.m_fileInfo.absoluteFilePath());
522 m_errorDevice->write(msg.toUtf8());
523 return false;
524 }
525 if (!m_root)
526 m_root = new RCCFileInfo(QString(), QFileInfo(), QLocale::C, QLocale::AnyCountry, RCCFileInfo::Directory);
527
528 RCCFileInfo *parent = m_root;
529 const QStringList nodes = alias.split(QLatin1Char('/'));
530 for (int i = 1; i < nodes.size()-1; ++i) {
531 const QString node = nodes.at(i);
532 if (node.isEmpty())
533 continue;
534 if (!parent->m_children.contains(node)) {
535 RCCFileInfo *s = new RCCFileInfo(node, QFileInfo(), QLocale::C, QLocale::AnyCountry, RCCFileInfo::Directory);
536 s->m_parent = parent;
537 parent->m_children.insert(node, s);
538 parent = s;
539 } else {
540 parent = parent->m_children[node];
541 }
542 }
543
544 const QString filename = nodes.at(nodes.size()-1);
545 RCCFileInfo *s = new RCCFileInfo(file);
546 s->m_parent = parent;
547 if (parent->m_children.contains(filename))
548 qWarning("potential duplicate alias detected: '%s'", qPrintable(filename));
549 parent->m_children.insertMulti(filename, s);
550 return true;
551}
552
553void RCCResourceLibrary::reset()
554{
555 if (m_root) {
556 delete m_root;
557 m_root = 0;
558 }
559 m_errorDevice = 0;
560 m_failedResources.clear();
561}
562
563
564bool RCCResourceLibrary::readFiles(bool ignoreErrors, QIODevice &errorDevice)
565{
566 reset();
567 m_errorDevice = &errorDevice;
568 //read in data
569 if (m_verbose) {
570 const QString msg = QString::fromUtf8("Processing %1 files [%2]\n")
571 .arg(m_fileNames.size()).arg(static_cast<int>(ignoreErrors));
572 m_errorDevice->write(msg.toUtf8());
573 }
574 for (int i = 0; i < m_fileNames.size(); ++i) {
575 QFile fileIn;
576 QString fname = m_fileNames.at(i);
577 QString pwd;
578 if (fname == QLatin1String("-")) {
579 fname = QLatin1String("(stdin)");
580 pwd = QDir::currentPath();
581 fileIn.setFileName(fname);
582 if (!fileIn.open(stdin, QIODevice::ReadOnly)) {
583 m_errorDevice->write(msgOpenReadFailed(fname, fileIn.errorString()).toUtf8());
584 return false;
585 }
586 } else {
587 pwd = QFileInfo(fname).path();
588 fileIn.setFileName(fname);
589 if (!fileIn.open(QIODevice::ReadOnly)) {
590 m_errorDevice->write(msgOpenReadFailed(fname, fileIn.errorString()).toUtf8());
591 return false;
592 }
593 }
594 if (m_verbose) {
595 const QString msg = QString::fromUtf8("Interpreting %1\n").arg(fname);
596 m_errorDevice->write(msg.toUtf8());
597 }
598
599 if (!interpretResourceFile(&fileIn, fname, pwd, ignoreErrors))
600 return false;
601 }
602 return true;
603}
604
605QStringList RCCResourceLibrary::dataFiles() const
606{
607 QStringList ret;
608 QStack<RCCFileInfo*> pending;
609
610 if (!m_root)
611 return ret;
612 pending.push(m_root);
613 while (!pending.isEmpty()) {
614 RCCFileInfo *file = pending.pop();
615 for (QHash<QString, RCCFileInfo*>::iterator it = file->m_children.begin();
616 it != file->m_children.end(); ++it) {
617 RCCFileInfo *child = it.value();
618 if (child->m_flags & RCCFileInfo::Directory)
619 pending.push(child);
620 ret.append(child->m_fileInfo.filePath());
621 }
622 }
623 return ret;
624}
625
626// Determine map of resource identifier (':/newPrefix/images/p1.png') to file via recursion
627static void resourceDataFileMapRecursion(const RCCFileInfo *m_root, const QString &path, RCCResourceLibrary::ResourceDataFileMap &m)
628{
629 typedef QHash<QString, RCCFileInfo*>::const_iterator ChildConstIterator;
630 const QChar slash = QLatin1Char('/');
631 const ChildConstIterator cend = m_root->m_children.constEnd();
632 for (ChildConstIterator it = m_root->m_children.constBegin(); it != cend; ++it) {
633 const RCCFileInfo *child = it.value();
634 QString childName = path;
635 childName += slash;
636 childName += child->m_name;
637 if (child->m_flags & RCCFileInfo::Directory) {
638 resourceDataFileMapRecursion(child, childName, m);
639 } else {
640 m.insert(childName, child->m_fileInfo.filePath());
641 }
642 }
643}
644
645RCCResourceLibrary::ResourceDataFileMap RCCResourceLibrary::resourceDataFileMap() const
646{
647 ResourceDataFileMap rc;
648 if (m_root)
649 resourceDataFileMapRecursion(m_root, QString(QLatin1Char(':')), rc);
650 return rc;
651}
652
653bool RCCResourceLibrary::output(QIODevice &outDevice, QIODevice &errorDevice)
654{
655 m_errorDevice = &errorDevice;
656 //write out
657 if (m_verbose)
658 m_errorDevice->write("Outputting code\n");
659 if (!writeHeader()) {
660 m_errorDevice->write("Could not write header\n");
661 return false;
662 }
663 if (m_root) {
664 if (!writeDataBlobs()) {
665 m_errorDevice->write("Could not write data blobs.\n");
666 return false;
667 }
668 if (!writeDataNames()) {
669 m_errorDevice->write("Could not write file names\n");
670 return false;
671 }
672 if (!writeDataStructure()) {
673 m_errorDevice->write("Could not write data tree\n");
674 return false;
675 }
676 }
677 if (!writeInitializer()) {
678 m_errorDevice->write("Could not write footer\n");
679 return false;
680 }
681 outDevice.write(m_out, m_out.size());
682 return true;
683}
684
685void RCCResourceLibrary::writeHex(quint8 tmp)
686{
687 const char * const digits = "0123456789abcdef";
688 writeChar('0');
689 writeChar('x');
690 if (tmp < 16) {
691 writeChar(digits[tmp]);
692 } else {
693 writeChar(digits[tmp >> 4]);
694 writeChar(digits[tmp & 0xf]);
695 }
696 writeChar(',');
697}
698
699void RCCResourceLibrary::writeNumber2(quint16 number)
700{
701 if (m_format == RCCResourceLibrary::Binary) {
702 writeChar(number >> 8);
703 writeChar(number);
704 } else {
705 writeHex(number >> 8);
706 writeHex(number);
707 }
708}
709
710void RCCResourceLibrary::writeNumber4(quint32 number)
711{
712 if (m_format == RCCResourceLibrary::Binary) {
713 writeChar(number >> 24);
714 writeChar(number >> 16);
715 writeChar(number >> 8);
716 writeChar(number);
717 } else {
718 writeHex(number >> 24);
719 writeHex(number >> 16);
720 writeHex(number >> 8);
721 writeHex(number);
722 }
723}
724
725bool RCCResourceLibrary::writeHeader()
726{
727 if (m_format == C_Code) {
728 writeString("/****************************************************************************\n");
729 writeString("** Resource object code\n");
730 writeString("**\n");
731 writeString("** Created: ");
732 writeByteArray(QDateTime::currentDateTime().toString().toLatin1());
733 writeString("\n** by: The Resource Compiler for Qt version ");
734 writeByteArray(QT_VERSION_STR);
735 writeString("\n**\n");
736 writeString("** WARNING! All changes made in this file will be lost!\n");
737 writeString( "*****************************************************************************/\n\n");
738 writeString("#include <QtCore/qglobal.h>\n\n");
739 } else if (m_format == Binary) {
740 writeString("qres");
741 writeNumber4(0);
742 writeNumber4(0);
743 writeNumber4(0);
744 writeNumber4(0);
745 }
746 return true;
747}
748
749bool RCCResourceLibrary::writeDataBlobs()
750{
751 Q_ASSERT(m_errorDevice);
752 if (m_format == C_Code)
753 writeString("static const unsigned char qt_resource_data[] = {\n");
754 else if (m_format == Binary)
755 m_dataOffset = m_out.size();
756 QStack<RCCFileInfo*> pending;
757
758 if (!m_root)
759 return false;
760
761 pending.push(m_root);
762 qint64 offset = 0;
763 QString errorMessage;
764 while (!pending.isEmpty()) {
765 RCCFileInfo *file = pending.pop();
766 for (QHash<QString, RCCFileInfo*>::iterator it = file->m_children.begin();
767 it != file->m_children.end(); ++it) {
768 RCCFileInfo *child = it.value();
769 if (child->m_flags & RCCFileInfo::Directory)
770 pending.push(child);
771 else {
772 offset = child->writeDataBlob(*this, offset, &errorMessage);
773 if (offset == 0)
774 m_errorDevice->write(errorMessage.toUtf8());
775 }
776 }
777 }
778 if (m_format == C_Code)
779 writeString("\n};\n\n");
780 return true;
781}
782
783bool RCCResourceLibrary::writeDataNames()
784{
785 if (m_format == C_Code)
786 writeString("static const unsigned char qt_resource_name[] = {\n");
787 else if (m_format == Binary)
788 m_namesOffset = m_out.size();
789
790 QHash<QString, int> names;
791 QStack<RCCFileInfo*> pending;
792
793 if (!m_root)
794 return false;
795
796 pending.push(m_root);
797 qint64 offset = 0;
798 while (!pending.isEmpty()) {
799 RCCFileInfo *file = pending.pop();
800 for (QHash<QString, RCCFileInfo*>::iterator it = file->m_children.begin();
801 it != file->m_children.end(); ++it) {
802 RCCFileInfo *child = it.value();
803 if (child->m_flags & RCCFileInfo::Directory)
804 pending.push(child);
805 if (names.contains(child->m_name)) {
806 child->m_nameOffset = names.value(child->m_name);
807 } else {
808 names.insert(child->m_name, offset);
809 offset = child->writeDataName(*this, offset);
810 }
811 }
812 }
813 if (m_format == C_Code)
814 writeString("\n};\n\n");
815 return true;
816}
817
818static bool qt_rcc_compare_hash(const RCCFileInfo *left, const RCCFileInfo *right)
819{
820 return qHash(left->m_name) < qHash(right->m_name);
821}
822
823bool RCCResourceLibrary::writeDataStructure()
824{
825 if (m_format == C_Code)
826 writeString("static const unsigned char qt_resource_struct[] = {\n");
827 else if (m_format == Binary)
828 m_treeOffset = m_out.size();
829 QStack<RCCFileInfo*> pending;
830
831 if (!m_root)
832 return false;
833
834 //calculate the child offsets (flat)
835 pending.push(m_root);
836 int offset = 1;
837 while (!pending.isEmpty()) {
838 RCCFileInfo *file = pending.pop();
839 file->m_childOffset = offset;
840
841 //sort by hash value for binary lookup
842 QList<RCCFileInfo*> m_children = file->m_children.values();
843 qSort(m_children.begin(), m_children.end(), qt_rcc_compare_hash);
844
845 //write out the actual data now
846 for (int i = 0; i < m_children.size(); ++i) {
847 RCCFileInfo *child = m_children.at(i);
848 ++offset;
849 if (child->m_flags & RCCFileInfo::Directory)
850 pending.push(child);
851 }
852 }
853
854 //write out the structure (ie iterate again!)
855 pending.push(m_root);
856 m_root->writeDataInfo(*this);
857 while (!pending.isEmpty()) {
858 RCCFileInfo *file = pending.pop();
859
860 //sort by hash value for binary lookup
861 QList<RCCFileInfo*> m_children = file->m_children.values();
862 qSort(m_children.begin(), m_children.end(), qt_rcc_compare_hash);
863
864 //write out the actual data now
865 for (int i = 0; i < m_children.size(); ++i) {
866 RCCFileInfo *child = m_children.at(i);
867 child->writeDataInfo(*this);
868 if (child->m_flags & RCCFileInfo::Directory)
869 pending.push(child);
870 }
871 }
872 if (m_format == C_Code)
873 writeString("\n};\n\n");
874
875 return true;
876}
877
878void RCCResourceLibrary::writeMangleNamespaceFunction(const QByteArray &name)
879{
880 if (m_useNameSpace) {
881 writeString("QT_MANGLE_NAMESPACE(");
882 writeByteArray(name);
883 writeChar(')');
884 } else {
885 writeByteArray(name);
886 }
887}
888
889void RCCResourceLibrary::writeAddNamespaceFunction(const QByteArray &name)
890{
891 if (m_useNameSpace) {
892 writeString("QT_PREPEND_NAMESPACE(");
893 writeByteArray(name);
894 writeChar(')');
895 } else {
896 writeByteArray(name);
897 }
898}
899
900bool RCCResourceLibrary::writeInitializer()
901{
902 if (m_format == C_Code) {
903 //write("\nQT_BEGIN_NAMESPACE\n");
904 QString initName = m_initName;
905 if (!initName.isEmpty()) {
906 initName.prepend(QLatin1Char('_'));
907 initName.replace(QRegExp(QLatin1String("[^a-zA-Z0-9_]")), QLatin1String("_"));
908 }
909
910 //init
911 if (m_useNameSpace)
912 writeString("QT_BEGIN_NAMESPACE\n\n");
913 if (m_root) {
914 writeString("extern Q_CORE_EXPORT bool qRegisterResourceData\n "
915 "(int, const unsigned char *, "
916 "const unsigned char *, const unsigned char *);\n\n");
917 writeString("extern Q_CORE_EXPORT bool qUnregisterResourceData\n "
918 "(int, const unsigned char *, "
919 "const unsigned char *, const unsigned char *);\n\n");
920 }
921 if (m_useNameSpace)
922 writeString("QT_END_NAMESPACE\n\n\n");
923 QString initResources = QLatin1String("qInitResources");
924 initResources += initName;
925 writeString("int ");
926 writeMangleNamespaceFunction(initResources.toLatin1());
927 writeString("()\n{\n");
928
929 if (m_root) {
930 writeString(" ");
931 writeAddNamespaceFunction("qRegisterResourceData");
932 writeString("\n (0x01, qt_resource_struct, "
933 "qt_resource_name, qt_resource_data);\n");
934 }
935 writeString(" return 1;\n");
936 writeString("}\n\n");
937 writeString("Q_CONSTRUCTOR_FUNCTION(");
938 writeMangleNamespaceFunction(initResources.toLatin1());
939 writeString(")\n\n");
940
941 //cleanup
942 QString cleanResources = QLatin1String("qCleanupResources");
943 cleanResources += initName;
944 writeString("int ");
945 writeMangleNamespaceFunction(cleanResources.toLatin1());
946 writeString("()\n{\n");
947 if (m_root) {
948 writeString(" ");
949 writeAddNamespaceFunction("qUnregisterResourceData");
950 writeString("\n (0x01, qt_resource_struct, "
951 "qt_resource_name, qt_resource_data);\n");
952 }
953 writeString(" return 1;\n");
954 writeString("}\n\n");
955 writeString("Q_DESTRUCTOR_FUNCTION(");
956 writeMangleNamespaceFunction(cleanResources.toLatin1());
957 writeString(")\n\n");
958 } else if (m_format == Binary) {
959 int i = 4;
960 char *p = m_out.data();
961 p[i++] = 0; // 0x01
962 p[i++] = 0;
963 p[i++] = 0;
964 p[i++] = 1;
965
966 p[i++] = (m_treeOffset >> 24) & 0xff;
967 p[i++] = (m_treeOffset >> 16) & 0xff;
968 p[i++] = (m_treeOffset >> 8) & 0xff;
969 p[i++] = (m_treeOffset >> 0) & 0xff;
970
971 p[i++] = (m_dataOffset >> 24) & 0xff;
972 p[i++] = (m_dataOffset >> 16) & 0xff;
973 p[i++] = (m_dataOffset >> 8) & 0xff;
974 p[i++] = (m_dataOffset >> 0) & 0xff;
975
976 p[i++] = (m_namesOffset >> 24) & 0xff;
977 p[i++] = (m_namesOffset >> 16) & 0xff;
978 p[i++] = (m_namesOffset >> 8) & 0xff;
979 p[i++] = (m_namesOffset >> 0) & 0xff;
980 }
981 return true;
982}
983
984QT_END_NAMESPACE
Note: See TracBrowser for help on using the repository browser.