KisImportExportFilter::ConversionStatus KisAnimationImporter::import(QStringList files, int firstFrame, int step)
{
Q_ASSERT(step > 0);
bool batchMode;
if (m_d->document) {
batchMode= m_d->document->fileBatchMode();
if (!batchMode) {
emit m_d->document->statusBarMessage(i18n("Import frames"));
emit m_d->document->sigProgress(0);
connect(m_d->document, SIGNAL(sigProgressCanceled()), this, SLOT(cancel()));
}
} else batchMode = false;
m_d->image->lock();
KisUndoAdapter *undo = m_d->image->undoAdapter();
undo->beginMacro(kundo2_i18n("Import animation"));
QScopedPointer<KisDocument> importDoc(KisPart::instance()->createDocument());
importDoc->setFileBatchMode(true);
KisImportExportFilter::ConversionStatus status = KisImportExportFilter::OK;
int frame = firstFrame;
KisRasterKeyframeChannel *contentChannel = 0;
Q_FOREACH(QString file, files) {
bool successfullyLoaded = importDoc->openUrl(QUrl::fromLocalFile(file), KisDocument::OPEN_URL_FLAG_DO_NOT_ADD_TO_RECENT_FILES);
if (!successfullyLoaded) {
status = KisImportExportFilter::InternalError;
break;
}
if (m_d->stop) {
status = KisImportExportFilter::ProgressCancelled;
break;
}
if (frame == firstFrame) {
const KoColorSpace *cs = importDoc->image()->colorSpace();
KisPaintLayerSP paintLayer = new KisPaintLayer(m_d->image, m_d->image->nextLayerName(), OPACITY_OPAQUE_U8, cs);
undo->addCommand(new KisImageLayerAddCommand(m_d->image, paintLayer, m_d->image->rootLayer(), m_d->image->rootLayer()->childCount()));
paintLayer->enableAnimation();
contentChannel = qobject_cast<KisRasterKeyframeChannel*>(paintLayer->getKeyframeChannel(KisKeyframeChannel::Content.id()));
}
if (!batchMode) {
emit m_d->document->sigProgress((frame - firstFrame) * 100 / (step * files.size()));
}
contentChannel->importFrame(frame, importDoc->image()->projection(), NULL);
frame += step;
}
void KisAnimationExporter::stopExport()
{
if (!m_d->exporting) return;
m_d->exporting = false;
disconnect(m_d->image->animationInterface(), 0, this, 0);
if (!m_d->batchMode) {
disconnect(m_d->document, SIGNAL(sigProgressCanceled()), this, SLOT(cancel()));
emit m_d->document->sigProgress(100);
emit m_d->document->clearStatusBarMessage();
}
emit sigFinished();
}
KisImportExportFilter::ConversionStatus KisAnimationExporter::exportAnimation()
{
if (!m_d->batchMode) {
emit m_d->document->statusBarMessage(i18n("Export frames"));
emit m_d->document->sigProgress(0);
connect(m_d->document, SIGNAL(sigProgressCanceled()), this, SLOT(cancel()));
}
m_d->status = KisImportExportFilter::OK;
m_d->exporting = true;
m_d->currentFrame = m_d->firstFrame;
connect(m_d->image->animationInterface(), SIGNAL(sigFrameReady(int)), this, SLOT(frameReadyToCopy(int)), Qt::DirectConnection);
m_d->image->animationInterface()->requestFrameRegeneration(m_d->currentFrame, m_d->image->bounds());
QEventLoop loop;
loop.connect(this, SIGNAL(sigFinished()), SLOT(quit()));
loop.exec();
return m_d->status;
}
KisImageBuilder_Result CSVSaver::encode(const QUrl &uri,const QString &filename)
{
int idx;
int start, end;
KisNodeSP node;
QByteArray ba;
KisKeyframeSP keyframe;
QVector<CSVLayerRecord*> layers;
KisImageAnimationInterface *animation = m_image->animationInterface();
//open the csv file for writing
QFile f(uri.toLocalFile());
if (!f.open(QIODevice::WriteOnly)) {
return KisImageBuilder_RESULT_NOT_LOCAL;
}
QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
//DataStream instead of TextStream for correct line endings
QDataStream stream(&f);
QString path = filename;
if (path.right(4).toUpper() == ".CSV")
path = path.left(path.size() - 4);
path.append(".frames");
//create directory
QDir dir(path);
if (!dir.exists()) {
dir.mkpath(".");
}
//according to the QT docs, the slash is a universal directory separator
path.append("/");
m_image->lock();
node = m_image->rootLayer()->firstChild();
//TODO: correct handling of the layer tree.
//for now, only top level paint layers are saved
idx = 0;
while (node) {
if (node->inherits("KisPaintLayer")) {
KisPaintLayer* paintLayer = dynamic_cast<KisPaintLayer*>(node.data());
CSVLayerRecord* layerRecord = new CSVLayerRecord();
layers.prepend(layerRecord); //reverse order!
layerRecord->name = paintLayer->name();
layerRecord->name.replace(QRegExp("[\"\\r\\n]"), "_");
if (layerRecord->name.isEmpty())
layerRecord->name= QString("Unnamed-%1").arg(idx);
layerRecord->visible = (paintLayer->visible()) ? 1 : 0;
layerRecord->density = (float)(paintLayer->opacity()) / OPACITY_OPAQUE_U8;
layerRecord->blending = convertToBlending(paintLayer->compositeOpId());
layerRecord->layer = paintLayer;
layerRecord->channel = paintLayer->projection()->keyframeChannel();
layerRecord->last = "";
layerRecord->frame = 0;
idx++;
}
node = node->nextSibling();
}
KisTimeRange range = animation->fullClipRange();
start = (range.isValid()) ? range.start() : 0;
if (!range.isInfinite()) {
end = range.end();
if (end < start) end = start;
} else {
//undefined length, searching for the last keyframe
end = start;
for (idx = 0; idx < layers.size(); idx++) {
keyframe = layers.at(idx)->channel->lastKeyframe();
if ( (!keyframe.isNull()) && (keyframe->time() > end) )
end = keyframe->time();
}
}
//create temporary doc for exporting
QScopedPointer<KisDocument> exportDoc(KisPart::instance()->createDocument());
createTempImage(exportDoc.data());
KisImageBuilder_Result retval= KisImageBuilder_RESULT_OK;
if (!m_batchMode) {
emit m_doc->statusBarMessage(i18n("Saving CSV file..."));
emit m_doc->sigProgress(0);
connect(m_doc, SIGNAL(sigProgressCanceled()), this, SLOT(cancel()));
}
int frame = start;
int step = 0;
do {
qApp->processEvents();
if (m_stop) {
retval = KisImageBuilder_RESULT_CANCEL;
break;
}
switch(step) {
case 0 : //first row
if (f.write("UTF-8, TVPaint, \"CSV 1.0\"\r\n") < 0) {
retval = KisImageBuilder_RESULT_FAILURE;
}
break;
case 1 : //scene header names
if (f.write("Project Name, Width, Height, Frame Count, Layer Count, Frame Rate, Pixel Aspect Ratio, Field Mode\r\n") < 0) {
retval = KisImageBuilder_RESULT_FAILURE;
}
break;
case 2 : //scene header values
ba = QString("\"%1\", ").arg(m_image->objectName()).toUtf8();
if (f.write(ba.data()) < 0) {
retval = KisImageBuilder_RESULT_FAILURE;
break;
}
ba = QString("%1, %2, ").arg(m_image->width()).arg(m_image->height()).toUtf8();
if (f.write(ba.data()) < 0) {
retval = KisImageBuilder_RESULT_FAILURE;
break;
}
ba = QString("%1, %2, ").arg(end - start + 1).arg(layers.size()).toUtf8();
if (f.write(ba.data()) < 0) {
retval = KisImageBuilder_RESULT_FAILURE;
break;
}
//the framerate is an integer here
ba = QString("%1, ").arg((double)(animation->framerate()),0,'f',6).toUtf8();
if (f.write(ba.data()) < 0) {
retval = KisImageBuilder_RESULT_FAILURE;
break;
}
ba = QString("%1, Progressive\r\n").arg((double)(m_image->xRes() / m_image->yRes()),0,'f',6).toUtf8();
if (f.write(ba.data()) < 0) {
retval = KisImageBuilder_RESULT_FAILURE;
break;
}
break;
case 3 : //layer header values
if (f.write("#Layers") < 0) { //Layers
retval = KisImageBuilder_RESULT_FAILURE;
break;
}
for (idx = 0; idx < layers.size(); idx++) {
ba = QString(", \"%1\"").arg(layers.at(idx)->name).toUtf8();
if (f.write(ba.data()) < 0)
break;
}
break;
case 4 :
if (f.write("\r\n#Density") < 0) { //Density
retval = KisImageBuilder_RESULT_FAILURE;
break;
}
for (idx = 0; idx < layers.size(); idx++) {
ba = QString(", %1").arg((double)(layers.at(idx)->density), 0, 'f', 6).toUtf8();
if (f.write(ba.data()) < 0)
break;
}
break;
case 5 :
if (f.write("\r\n#Blending") < 0) { //Blending
retval = KisImageBuilder_RESULT_FAILURE;
break;
}
for (idx = 0; idx < layers.size(); idx++) {
ba = QString(", \"%1\"").arg(layers.at(idx)->blending).toUtf8();
if (f.write(ba.data()) < 0)
break;
}
break;
case 6 :
if (f.write("\r\n#Visible") < 0) { //Visible
retval = KisImageBuilder_RESULT_FAILURE;
break;
}
for (idx = 0; idx < layers.size(); idx++) {
ba = QString(", %1").arg(layers.at(idx)->visible).toUtf8();
if (f.write(ba.data()) < 0)
break;
}
if (idx < layers.size()) {
retval = KisImageBuilder_RESULT_FAILURE;
}
break;
default : //frames
if (frame > end) {
if (f.write("\r\n") < 0)
retval = KisImageBuilder_RESULT_FAILURE;
step = 8;
break;
}
ba = QString("\r\n#%1").arg(frame, 5, 10, QChar('0')).toUtf8();
if (f.write(ba.data()) < 0) {
retval = KisImageBuilder_RESULT_FAILURE;
break;
}
for (idx = 0; idx < layers.size(); idx++) {
CSVLayerRecord *layer = layers.at(idx);
keyframe = layer->channel->keyframeAt(frame);
if (!keyframe.isNull()) {
if (!m_batchMode) {
emit m_doc->sigProgress(((frame - start) * layers.size() + idx) * 100 /
((end - start) * layers.size()));
}
retval = getLayer(layer, exportDoc.data(), keyframe, path, frame, idx);
if (retval != KisImageBuilder_RESULT_OK)
break;
}
ba = QString(", \"%1\"").arg(layer->last).toUtf8();
if (f.write(ba.data()) < 0)
break;
}
if (idx < layers.size())
retval = KisImageBuilder_RESULT_FAILURE;
frame++;
step = 6; //keep step here
break;
}
step++;
} while((retval == KisImageBuilder_RESULT_OK) && (step < 8));
m_image->unlock();
qDeleteAll(layers);
f.close();
if (!m_batchMode) {
disconnect(m_doc, SIGNAL(sigProgressCanceled()), this, SLOT(cancel()));
emit m_doc->sigProgress(100);
emit m_doc->clearStatusBarMessage();
}
QApplication::restoreOverrideCursor();
return retval;
}
