You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tellico/src/translators/tellicozipexporter.cpp

134 lines
4.7 KiB

/***************************************************************************
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::tqfromLatin1("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::tqfromLatin1("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"