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

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

trunk: Merged in qt 4.6.1 sources.
File size: 32.4 KB
Line 
1/****************************************************************************
2**
3** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
4** All rights reserved.
5** Contact: Nokia Corporation ([email protected])
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 [email protected].
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 parent->m_children.insertMulti(filename, s);
548 return true;
549}
550
551void RCCResourceLibrary::reset()
552{
553 if (m_root) {
554 delete m_root;
555 m_root = 0;
556 }
557 m_errorDevice = 0;
558 m_failedResources.clear();
559}
560
561
562bool RCCResourceLibrary::readFiles(bool ignoreErrors, QIODevice &errorDevice)
563{
564 reset();
565 m_errorDevice = &errorDevice;
566 //read in data
567 if (m_verbose) {
568 const QString msg = QString::fromUtf8("Processing %1 files [%2]\n")
569 .arg(m_fileNames.size()).arg(static_cast<int>(ignoreErrors));
570 m_errorDevice->write(msg.toUtf8());
571 }
572 for (int i = 0; i < m_fileNames.size(); ++i) {
573 QFile fileIn;
574 QString fname = m_fileNames.at(i);
575 QString pwd;
576 if (fname == QLatin1String("-")) {
577 fname = QLatin1String("(stdin)");
578 pwd = QDir::currentPath();
579 fileIn.setFileName(fname);
580 if (!fileIn.open(stdin, QIODevice::ReadOnly)) {
581 m_errorDevice->write(msgOpenReadFailed(fname, fileIn.errorString()).toUtf8());
582 return false;
583 }
584 } else {
585 pwd = QFileInfo(fname).path();
586 fileIn.setFileName(fname);
587 if (!fileIn.open(QIODevice::ReadOnly)) {
588 m_errorDevice->write(msgOpenReadFailed(fname, fileIn.errorString()).toUtf8());
589 return false;
590 }
591 }
592 if (m_verbose) {
593 const QString msg = QString::fromUtf8("Interpreting %1\n").arg(fname);
594 m_errorDevice->write(msg.toUtf8());
595 }
596
597 if (!interpretResourceFile(&fileIn, fname, pwd, ignoreErrors))
598 return false;
599 }
600 return true;
601}
602
603QStringList RCCResourceLibrary::dataFiles() const
604{
605 QStringList ret;
606 QStack<RCCFileInfo*> pending;
607
608 if (!m_root)
609 return ret;
610 pending.push(m_root);
611 while (!pending.isEmpty()) {
612 RCCFileInfo *file = pending.pop();
613 for (QHash<QString, RCCFileInfo*>::iterator it = file->m_children.begin();
614 it != file->m_children.end(); ++it) {
615 RCCFileInfo *child = it.value();
616 if (child->m_flags & RCCFileInfo::Directory)
617 pending.push(child);
618 ret.append(child->m_fileInfo.filePath());
619 }
620 }
621 return ret;
622}
623
624// Determine map of resource identifier (':/newPrefix/images/p1.png') to file via recursion
625static void resourceDataFileMapRecursion(const RCCFileInfo *m_root, const QString &path, RCCResourceLibrary::ResourceDataFileMap &m)
626{
627 typedef QHash<QString, RCCFileInfo*>::const_iterator ChildConstIterator;
628 const QChar slash = QLatin1Char('/');
629 const ChildConstIterator cend = m_root->m_children.constEnd();
630 for (ChildConstIterator it = m_root->m_children.constBegin(); it != cend; ++it) {
631 const RCCFileInfo *child = it.value();
632 QString childName = path;
633 childName += slash;
634 childName += child->m_name;
635 if (child->m_flags & RCCFileInfo::Directory) {
636 resourceDataFileMapRecursion(child, childName, m);
637 } else {
638 m.insert(childName, child->m_fileInfo.filePath());
639 }
640 }
641}
642
643RCCResourceLibrary::ResourceDataFileMap RCCResourceLibrary::resourceDataFileMap() const
644{
645 ResourceDataFileMap rc;
646 if (m_root)
647 resourceDataFileMapRecursion(m_root, QString(QLatin1Char(':')), rc);
648 return rc;
649}
650
651bool RCCResourceLibrary::output(QIODevice &outDevice, QIODevice &errorDevice)
652{
653 m_errorDevice = &errorDevice;
654 //write out
655 if (m_verbose)
656 m_errorDevice->write("Outputting code\n");
657 if (!writeHeader()) {
658 m_errorDevice->write("Could not write header\n");
659 return false;
660 }
661 if (m_root) {
662 if (!writeDataBlobs()) {
663 m_errorDevice->write("Could not write data blobs.\n");
664 return false;
665 }
666 if (!writeDataNames()) {
667 m_errorDevice->write("Could not write file names\n");
668 return false;
669 }
670 if (!writeDataStructure()) {
671 m_errorDevice->write("Could not write data tree\n");
672 return false;
673 }
674 }
675 if (!writeInitializer()) {
676 m_errorDevice->write("Could not write footer\n");
677 return false;
678 }
679 outDevice.write(m_out, m_out.size());
680 return true;
681}
682
683void RCCResourceLibrary::writeHex(quint8 tmp)
684{
685 const char * const digits = "0123456789abcdef";
686 writeChar('0');
687 writeChar('x');
688 if (tmp < 16) {
689 writeChar(digits[tmp]);
690 } else {
691 writeChar(digits[tmp >> 4]);
692 writeChar(digits[tmp & 0xf]);
693 }
694 writeChar(',');
695}
696
697void RCCResourceLibrary::writeNumber2(quint16 number)
698{
699 if (m_format == RCCResourceLibrary::Binary) {
700 writeChar(number >> 8);
701 writeChar(number);
702 } else {
703 writeHex(number >> 8);
704 writeHex(number);
705 }
706}
707
708void RCCResourceLibrary::writeNumber4(quint32 number)
709{
710 if (m_format == RCCResourceLibrary::Binary) {
711 writeChar(number >> 24);
712 writeChar(number >> 16);
713 writeChar(number >> 8);
714 writeChar(number);
715 } else {
716 writeHex(number >> 24);
717 writeHex(number >> 16);
718 writeHex(number >> 8);
719 writeHex(number);
720 }
721}
722
723bool RCCResourceLibrary::writeHeader()
724{
725 if (m_format == C_Code) {
726 writeString("/****************************************************************************\n");
727 writeString("** Resource object code\n");
728 writeString("**\n");
729 writeString("** Created: ");
730 writeByteArray(QDateTime::currentDateTime().toString().toLatin1());
731 writeString("\n** by: The Resource Compiler for Qt version ");
732 writeByteArray(QT_VERSION_STR);
733 writeString("\n**\n");
734 writeString("** WARNING! All changes made in this file will be lost!\n");
735 writeString( "*****************************************************************************/\n\n");
736 writeString("#include <QtCore/qglobal.h>\n\n");
737 } else if (m_format == Binary) {
738 writeString("qres");
739 writeNumber4(0);
740 writeNumber4(0);
741 writeNumber4(0);
742 writeNumber4(0);
743 }
744 return true;
745}
746
747bool RCCResourceLibrary::writeDataBlobs()
748{
749 Q_ASSERT(m_errorDevice);
750 if (m_format == C_Code)
751 writeString("static const unsigned char qt_resource_data[] = {\n");
752 else if (m_format == Binary)
753 m_dataOffset = m_out.size();
754 QStack<RCCFileInfo*> pending;
755
756 if (!m_root)
757 return false;
758
759 pending.push(m_root);
760 qint64 offset = 0;
761 QString errorMessage;
762 while (!pending.isEmpty()) {
763 RCCFileInfo *file = pending.pop();
764 for (QHash<QString, RCCFileInfo*>::iterator it = file->m_children.begin();
765 it != file->m_children.end(); ++it) {
766 RCCFileInfo *child = it.value();
767 if (child->m_flags & RCCFileInfo::Directory)
768 pending.push(child);
769 else {
770 offset = child->writeDataBlob(*this, offset, &errorMessage);
771 if (offset == 0)
772 m_errorDevice->write(errorMessage.toUtf8());
773 }
774 }
775 }
776 if (m_format == C_Code)
777 writeString("\n};\n\n");
778 return true;
779}
780
781bool RCCResourceLibrary::writeDataNames()
782{
783 if (m_format == C_Code)
784 writeString("static const unsigned char qt_resource_name[] = {\n");
785 else if (m_format == Binary)
786 m_namesOffset = m_out.size();
787
788 QHash<QString, int> names;
789 QStack<RCCFileInfo*> pending;
790
791 if (!m_root)
792 return false;
793
794 pending.push(m_root);
795 qint64 offset = 0;
796 while (!pending.isEmpty()) {
797 RCCFileInfo *file = pending.pop();
798 for (QHash<QString, RCCFileInfo*>::iterator it = file->m_children.begin();
799 it != file->m_children.end(); ++it) {
800 RCCFileInfo *child = it.value();
801 if (child->m_flags & RCCFileInfo::Directory)
802 pending.push(child);
803 if (names.contains(child->m_name)) {
804 child->m_nameOffset = names.value(child->m_name);
805 } else {
806 names.insert(child->m_name, offset);
807 offset = child->writeDataName(*this, offset);
808 }
809 }
810 }
811 if (m_format == C_Code)
812 writeString("\n};\n\n");
813 return true;
814}
815
816static bool qt_rcc_compare_hash(const RCCFileInfo *left, const RCCFileInfo *right)
817{
818 return qHash(left->m_name) < qHash(right->m_name);
819}
820
821bool RCCResourceLibrary::writeDataStructure()
822{
823 if (m_format == C_Code)
824 writeString("static const unsigned char qt_resource_struct[] = {\n");
825 else if (m_format == Binary)
826 m_treeOffset = m_out.size();
827 QStack<RCCFileInfo*> pending;
828
829 if (!m_root)
830 return false;
831
832 //calculate the child offsets (flat)
833 pending.push(m_root);
834 int offset = 1;
835 while (!pending.isEmpty()) {
836 RCCFileInfo *file = pending.pop();
837 file->m_childOffset = offset;
838
839 //sort by hash value for binary lookup
840 QList<RCCFileInfo*> m_children = file->m_children.values();
841 qSort(m_children.begin(), m_children.end(), qt_rcc_compare_hash);
842
843 //write out the actual data now
844 for (int i = 0; i < m_children.size(); ++i) {
845 RCCFileInfo *child = m_children.at(i);
846 ++offset;
847 if (child->m_flags & RCCFileInfo::Directory)
848 pending.push(child);
849 }
850 }
851
852 //write out the structure (ie iterate again!)
853 pending.push(m_root);
854 m_root->writeDataInfo(*this);
855 while (!pending.isEmpty()) {
856 RCCFileInfo *file = pending.pop();
857
858 //sort by hash value for binary lookup
859 QList<RCCFileInfo*> m_children = file->m_children.values();
860 qSort(m_children.begin(), m_children.end(), qt_rcc_compare_hash);
861
862 //write out the actual data now
863 for (int i = 0; i < m_children.size(); ++i) {
864 RCCFileInfo *child = m_children.at(i);
865 child->writeDataInfo(*this);
866 if (child->m_flags & RCCFileInfo::Directory)
867 pending.push(child);
868 }
869 }
870 if (m_format == C_Code)
871 writeString("\n};\n\n");
872
873 return true;
874}
875
876void RCCResourceLibrary::writeMangleNamespaceFunction(const QByteArray &name)
877{
878 if (m_useNameSpace) {
879 writeString("QT_MANGLE_NAMESPACE(");
880 writeByteArray(name);
881 writeChar(')');
882 } else {
883 writeByteArray(name);
884 }
885}
886
887void RCCResourceLibrary::writeAddNamespaceFunction(const QByteArray &name)
888{
889 if (m_useNameSpace) {
890 writeString("QT_PREPEND_NAMESPACE(");
891 writeByteArray(name);
892 writeChar(')');
893 } else {
894 writeByteArray(name);
895 }
896}
897
898bool RCCResourceLibrary::writeInitializer()
899{
900 if (m_format == C_Code) {
901 //write("\nQT_BEGIN_NAMESPACE\n");
902 QString initName = m_initName;
903 if (!initName.isEmpty()) {
904 initName.prepend(QLatin1Char('_'));
905 initName.replace(QRegExp(QLatin1String("[^a-zA-Z0-9_]")), QLatin1String("_"));
906 }
907
908 //init
909 if (m_useNameSpace)
910 writeString("QT_BEGIN_NAMESPACE\n\n");
911 if (m_root) {
912 writeString("extern Q_CORE_EXPORT bool qRegisterResourceData\n "
913 "(int, const unsigned char *, "
914 "const unsigned char *, const unsigned char *);\n\n");
915 writeString("extern Q_CORE_EXPORT bool qUnregisterResourceData\n "
916 "(int, const unsigned char *, "
917 "const unsigned char *, const unsigned char *);\n\n");
918 }
919 if (m_useNameSpace)
920 writeString("QT_END_NAMESPACE\n\n\n");
921 QString initResources = QLatin1String("qInitResources");
922 initResources += initName;
923 writeString("int ");
924 writeMangleNamespaceFunction(initResources.toLatin1());
925 writeString("()\n{\n");
926
927 if (m_root) {
928 writeString(" ");
929 writeAddNamespaceFunction("qRegisterResourceData");
930 writeString("\n (0x01, qt_resource_struct, "
931 "qt_resource_name, qt_resource_data);\n");
932 }
933 writeString(" return 1;\n");
934 writeString("}\n\n");
935 writeString("Q_CONSTRUCTOR_FUNCTION(");
936 writeMangleNamespaceFunction(initResources.toLatin1());
937 writeString(")\n\n");
938
939 //cleanup
940 QString cleanResources = QLatin1String("qCleanupResources");
941 cleanResources += initName;
942 writeString("int ");
943 writeMangleNamespaceFunction(cleanResources.toLatin1());
944 writeString("()\n{\n");
945 if (m_root) {
946 writeString(" ");
947 writeAddNamespaceFunction("qUnregisterResourceData");
948 writeString("\n (0x01, qt_resource_struct, "
949 "qt_resource_name, qt_resource_data);\n");
950 }
951 writeString(" return 1;\n");
952 writeString("}\n\n");
953 writeString("Q_DESTRUCTOR_FUNCTION(");
954 writeMangleNamespaceFunction(cleanResources.toLatin1());
955 writeString(")\n\n");
956 } else if (m_format == Binary) {
957 int i = 4;
958 char *p = m_out.data();
959 p[i++] = 0; // 0x01
960 p[i++] = 0;
961 p[i++] = 0;
962 p[i++] = 1;
963
964 p[i++] = (m_treeOffset >> 24) & 0xff;
965 p[i++] = (m_treeOffset >> 16) & 0xff;
966 p[i++] = (m_treeOffset >> 8) & 0xff;
967 p[i++] = (m_treeOffset >> 0) & 0xff;
968
969 p[i++] = (m_dataOffset >> 24) & 0xff;
970 p[i++] = (m_dataOffset >> 16) & 0xff;
971 p[i++] = (m_dataOffset >> 8) & 0xff;
972 p[i++] = (m_dataOffset >> 0) & 0xff;
973
974 p[i++] = (m_namesOffset >> 24) & 0xff;
975 p[i++] = (m_namesOffset >> 16) & 0xff;
976 p[i++] = (m_namesOffset >> 8) & 0xff;
977 p[i++] = (m_namesOffset >> 0) & 0xff;
978 }
979 return true;
980}
981
982QT_END_NAMESPACE
Note: See TracBrowser for help on using the repository browser.