1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
/***************************************************************************
copyright : (C) 2003-2006 by Robby Stephenson
email : robby@periapsis.org
***************************************************************************/
/***************************************************************************
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of version 2 of the GNU General Public License as *
* published by the Free Software Foundation; *
* *
***************************************************************************/
#include "tellicozipexporter.h"
#include "tellicoxmlexporter.h"
#include "../collection.h"
#include "../imagefactory.h"
#include "../image.h"
#include "../filehandler.h"
#include "../stringset.h"
#include "../tellico_debug.h"
#include "../progressmanager.h"
#include <klocale.h>
#include <kconfig.h>
#include <kzip.h>
#include <kapplication.h>
#include <tqdom.h>
#include <tqbuffer.h>
using Tellico::Export::TellicoZipExporter;
TQString TellicoZipExporter::formatString() const {
return i18n("Tellico Zip File");
}
TQString TellicoZipExporter::fileFilter() const {
return i18n("*.tc *.bc|Tellico Files (*.tc)") + TQChar('\n') + i18n("*|All Files");
}
bool TellicoZipExporter::exec() {
m_cancelled = false;
Data::CollPtr coll = collection();
if(!coll) {
return false;
}
// TODO: maybe need label?
ProgressItem& item = ProgressManager::self()->newProgressItem(this, TQString(), true);
item.setTotalSteps(100);
connect(&item, TQT_SIGNAL(signalCancelled(ProgressItem*)), TQT_SLOT(slotCancel()));
ProgressItem::Done done(this);
TellicoXMLExporter exp;
exp.setEntries(entries());
exp.setURL(url()); // needed in case of relative URL values
long opt = options();
opt |= Export::ExportUTF8; // always export to UTF-8
opt |= Export::ExportImages; // always list the images in the xml
opt &= ~Export::ExportProgress; // don't show progress for xml export
exp.setOptions(opt);
exp.setIncludeImages(false); // do not include the images themselves in XML
TQCString xml = exp.exportXML().toCString(); // encoded in utf-8
ProgressManager::self()->setProgress(this, 5);
TQByteArray data;
TQBuffer buf(data);
if(m_cancelled) {
return true; // intentionally cancelled
}
KZip zip(TQT_TQIODEVICE(&buf));
zip.open(IO_WriteOnly);
zip.writeFile(TQString::fromLatin1("tellico.xml"), TQString(), TQString(), xml.length(), xml);
if(m_includeImages) {
ProgressManager::self()->setProgress(this, 10);
// gonna be lazy and just increment progress every 3 images
// it might be less, might be more
uint j = 0;
const TQString imagesDir = TQString::fromLatin1("images/");
StringSet imageSet;
Data::FieldVec imageFields = coll->imageFields();
// already took 10%, only 90% left
const uint stepSize = TQMAX(1, (coll->entryCount() * imageFields.count()) / 90);
for(Data::EntryVec::ConstIterator it = entries().begin(); it != entries().end() && !m_cancelled; ++it) {
for(Data::FieldVec::Iterator fIt = imageFields.begin(); fIt != imageFields.end(); ++fIt, ++j) {
const TQString id = it->field(fIt);
if(id.isEmpty() || imageSet.has(id)) {
continue;
}
const Data::ImageInfo& info = ImageFactory::imageInfo(id);
if(info.linkOnly) {
myLog() << "TellicoZipExporter::exec() - not copying linked image: " << id << endl;
continue;
}
const Data::Image& img = ImageFactory::imageById(id);
// if no image, continue
if(img.isNull()) {
kdWarning() << "TellicoZipExporter::exec() - no image found for " << fIt->title() << " field" << endl;
kdWarning() << "...for the entry titled " << it->title() << endl;
continue;
}
TQByteArray ba = img.byteArray();
// myDebug() << "TellicoZipExporter::data() - adding image id = " << it->field(fIt) << endl;
zip.writeFile(imagesDir + id, TQString(), TQString(), ba.size(), ba);
imageSet.add(id);
if(j%stepSize == 0) {
ProgressManager::self()->setProgress(this, TQMIN(10+j/stepSize, 99));
kapp->processEvents();
}
}
}
} else {
ProgressManager::self()->setProgress(this, 80);
}
zip.close();
if(m_cancelled) {
return true;
}
bool success = FileHandler::writeDataURL(url(), data, options() & Export::ExportForce);
return success;
}
void TellicoZipExporter::slotCancel() {
m_cancelled = true;
}
#include "tellicozipexporter.moc"
|
