CSVSaver Class Reference

#include <csv_saver.h>

Inheritance diagram for CSVSaver:

Public Member Functions
KisImportExportErrorCode	buildAnimation (QIODevice *io)
	CSVSaver (KisDocument *doc, bool batchMode)
KisImageSP	image ()
	~CSVSaver () override

Private Slots
void	cancel ()

Private Member Functions
QString	convertToBlending (const QString &)
void	createTempImage (KisDocument *)
KisImportExportErrorCode	encode (QIODevice *io)
KisImportExportErrorCode	getLayer (CSVLayerRecord *, KisDocument *, KisKeyframeSP, const QString &, int, int)

Private Attributes
bool	m_batchMode
KisDocument *	m_doc
KisImageSP	m_image
bool	m_stop

Detailed Description

Definition at line 19 of file csv_saver.h.

Constructor & Destructor Documentation

CSVSaver()

CSVSaver::CSVSaver	(	KisDocument *	doc,
		bool	batchMode )

Definition at line 40 of file csv_saver.cpp.

    : m_image(doc->savingImage())
    , m_doc(doc)
    , m_batchMode(batchMode)
    , m_stop(false)
{
}

~CSVSaver()

CSVSaver::~CSVSaver ( )

override

Definition at line 48 of file csv_saver.cpp.

{
}

Member Function Documentation

buildAnimation()

KisImportExportErrorCode CSVSaver::buildAnimation

(

QIODevice *

io

)

Definition at line 458 of file csv_saver.cpp.

{
    KIS_ASSERT_RECOVER_RETURN_VALUE(m_image, ImportExportCodes::InternalError);
    return encode(io);
}

References encode(), ImportExportCodes::InternalError, KIS_ASSERT_RECOVER_RETURN_VALUE, and m_image.

cancel

void CSVSaver::cancel ( )

privateslot

Definition at line 464 of file csv_saver.cpp.

{
    m_stop = true;
}

References m_stop.

convertToBlending()

QString CSVSaver::convertToBlending ( const QString & opid )

private

Definition at line 343 of file csv_saver.cpp.

{
    if (opid == COMPOSITE_OVER) return "Color";
    if (opid == COMPOSITE_BEHIND) return "Behind";
    if (opid == COMPOSITE_ERASE) return "Erase";
    // "Shade"
    if (opid == COMPOSITE_LINEAR_LIGHT) return "Light";
    if (opid == COMPOSITE_COLORIZE) return "Colorize";
    if (opid == COMPOSITE_HUE) return "Hue";
    if ((opid == COMPOSITE_ADD) ||
        (opid == COMPOSITE_LINEAR_DODGE)) return "Add";
    if (opid == COMPOSITE_INVERSE_SUBTRACT) return "Sub";
    if (opid == COMPOSITE_MULT) return "Multiply";
    if (opid == COMPOSITE_SCREEN) return "Screen";
    // "Replace"
    // "Substitute"
    if (opid == COMPOSITE_DIFF) return "Difference";
    if (opid == COMPOSITE_DIVIDE) return "Divide";
    if (opid == COMPOSITE_OVERLAY) return "Overlay";
    if (opid == COMPOSITE_DODGE) return "Light2";
    if (opid == COMPOSITE_BURN) return "Shade2";
    if (opid == COMPOSITE_HARD_LIGHT) return "HardLight";
    if ((opid == COMPOSITE_SOFT_LIGHT_PHOTOSHOP) ||
        (opid == COMPOSITE_SOFT_LIGHT_SVG)) return "SoftLight";
    if (opid == COMPOSITE_GRAIN_EXTRACT) return "GrainExtract";
    if (opid == COMPOSITE_GRAIN_MERGE) return "GrainMerge";
    if (opid == COMPOSITE_SUBTRACT) return "Sub2";
    if (opid == COMPOSITE_DARKEN) return "Darken";
    if (opid == COMPOSITE_LIGHTEN) return "Lighten";
    if (opid == COMPOSITE_SATURATION) return "Saturation";
 
    return "Color";
}

References COMPOSITE_ADD, COMPOSITE_BEHIND, COMPOSITE_BURN, COMPOSITE_COLORIZE, COMPOSITE_DARKEN, COMPOSITE_DIFF, COMPOSITE_DIVIDE, COMPOSITE_DODGE, COMPOSITE_ERASE, COMPOSITE_GRAIN_EXTRACT, COMPOSITE_GRAIN_MERGE, COMPOSITE_HARD_LIGHT, COMPOSITE_HUE, COMPOSITE_INVERSE_SUBTRACT, COMPOSITE_LIGHTEN, COMPOSITE_LINEAR_DODGE, COMPOSITE_LINEAR_LIGHT, COMPOSITE_MULT, COMPOSITE_OVER, COMPOSITE_OVERLAY, COMPOSITE_SATURATION, COMPOSITE_SCREEN, COMPOSITE_SOFT_LIGHT_PHOTOSHOP, COMPOSITE_SOFT_LIGHT_SVG, and COMPOSITE_SUBTRACT.

createTempImage()

void CSVSaver::createTempImage ( KisDocument * exportDoc )

private

Definition at line 441 of file csv_saver.cpp.

{
    exportDoc->setInfiniteAutoSaveInterval();
    exportDoc->setFileBatchMode(true);
 
    KisImageSP exportImage = new KisImage(exportDoc->createUndoStore(),
                                           m_image->width(), m_image->height(), m_image->colorSpace(),
                                           QString());
 
    exportImage->setResolution(m_image->xRes(), m_image->yRes());
    exportDoc->setCurrentImage(exportImage);
 
    KisPaintLayer* paintLayer = new KisPaintLayer(exportImage, "paint device", OPACITY_OPAQUE_U8);
    exportImage->addNode(paintLayer, exportImage->rootLayer(), KisLayerSP(0));
}

References KisNodeFacade::addNode(), KisImage::colorSpace(), KisDocument::createUndoStore(), KisImage::height(), m_image, OPACITY_OPAQUE_U8, KisImage::rootLayer(), KisDocument::setCurrentImage(), KisDocument::setFileBatchMode(), KisDocument::setInfiniteAutoSaveInterval(), KisImage::setResolution(), KisImage::width(), KisImage::xRes(), and KisImage::yRes().

encode()

KisImportExportErrorCode CSVSaver::encode ( QIODevice * io )

private

Definition at line 57 of file csv_saver.cpp.

{
    int idx;
    int start, end;
    KisNodeSP node;
    QByteArray ba;
//    KisTimeKeyframePair keyframeEntry;
    KisKeyframeSP keyframe;
    QVector<CSVLayerRecord*> layers;
 
    KisImageAnimationInterface *animation = m_image->animationInterface();
 
    KisCursorOverrideLock cursorLock(Qt::WaitCursor);
 
// XXX: Stream was unused?
//    //DataStream instead of TextStream for correct line endings
//    QDataStream stream(&f);
 
    //Using the original local path
    QString path = m_doc->localFilePath();
 
    if (path.right(4).toUpper() == ".CSV")
        path = path.left(path.size() - 4);
    else {
        // something is wrong: the local file name is not .csv!
        // trying the given (probably temporary) filename as well
 
        KIS_SAFE_ASSERT_RECOVER(0 && "Wrong extension of the saved file!") {
            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("/");
 
    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("KisLayer")) {
            KisLayer* paintLayer = qobject_cast<KisLayer*>(node.data());
            CSVLayerRecord* layerRecord = new CSVLayerRecord();
            layers.prepend(layerRecord); //reverse order!
 
            layerRecord->name = paintLayer->name();
            layerRecord->name.replace(QRegularExpression("[\"\\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->original()->keyframeChannel();
            layerRecord->last = "";
            layerRecord->frame = 0;
            idx++;
        }
        node = node->nextSibling();
    }
 
    KisTimeSpan range = animation->documentPlaybackRange();
 
    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;
        int keyframeTime;
 
        for (idx = 0; idx < layers.size(); idx++) {
            KisRasterKeyframeChannel *channel = layers.at(idx)->channel;
 
            if (channel) {
                keyframeTime = channel->lastKeyframeTime();
 
                if ( (channel->keyframeAt(keyframeTime)) && (keyframeTime > end) )
                    end = keyframeTime;
            }
        }
    }
 
    //create temporary doc for exporting
    QScopedPointer<KisDocument> exportDoc(KisPart::instance()->createDocument());
    createTempImage(exportDoc.data());
 
    KisImportExportErrorCode retval= ImportExportCodes::OK;
 
    if (!m_batchMode) {
        // TODO: use other systems of progress reporting (KisViewManager::createUnthreadedUpdater()
        //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 = ImportExportCodes::Cancelled;
            break;
        }
 
        switch(step) {
 
        case 0 :    //first row
            if (io->write("UTF-8, TVPaint, \"CSV 1.0\"\r\n") < 0) {
                retval = ImportExportCodes::Failure;
            }
            break;
 
        case 1 :    //scene header names
            if (io->write("Project Name, Width, Height, Frame Count, Layer Count, Frame Rate, Pixel Aspect Ratio, Field Mode\r\n") < 0) {
                retval = ImportExportCodes::Failure;
            }
            break;
 
        case 2 :    //scene header values
            ba = QString("\"%1\", ").arg(m_image->objectName()).toUtf8();
            if (io->write(ba.data()) < 0) {
                retval = ImportExportCodes::Failure;
                break;
            }
            ba = QString("%1, %2, ").arg(m_image->width()).arg(m_image->height()).toUtf8();
            if (io->write(ba.data()) < 0) {
                retval = ImportExportCodes::Failure;
                break;
            }
 
            ba = QString("%1, %2, ").arg(end - start + 1).arg(layers.size()).toUtf8();
            if (io->write(ba.data()) < 0) {
                retval = ImportExportCodes::Failure;
                break;
            }
            //the framerate is an integer here
            ba = QString("%1, ").arg((double)(animation->framerate()),0,'f',6).toUtf8();
            if (io->write(ba.data()) < 0) {
                retval = ImportExportCodes::Failure;
                break;
            }
            ba = QString("%1, Progressive\r\n").arg((double)(m_image->xRes() / m_image->yRes()),0,'f',6).toUtf8();
            if (io->write(ba.data()) < 0) {
                retval = ImportExportCodes::Failure;
                break;
            }
            break;
 
        case 3 :    //layer header values
            if (io->write("#Layers") < 0) {          //Layers
                retval = ImportExportCodes::Failure;
                break;
            }
 
            for (idx = 0; idx < layers.size(); idx++) {
                ba = QString(", \"%1\"").arg(layers.at(idx)->name).toUtf8();
                if (io->write(ba.data()) < 0)
                    break;
            }
            break;
 
        case 4 :
            if (io->write("\r\n#Density") < 0) {     //Density
                retval = ImportExportCodes::Failure;
                break;
            }
            for (idx = 0; idx < layers.size(); idx++) {
                ba = QString(", %1").arg((double)(layers.at(idx)->density), 0, 'f', 6).toUtf8();
                if (io->write(ba.data()) < 0)
                    break;
            }
            break;
 
        case 5 :
            if (io->write("\r\n#Blending") < 0) {     //Blending
                retval = ImportExportCodes::Failure;
                break;
            }
            for (idx = 0; idx < layers.size(); idx++) {
                ba = QString(", \"%1\"").arg(layers.at(idx)->blending).toUtf8();
                if (io->write(ba.data()) < 0)
                    break;
            }
            break;
 
        case 6 :
            if (io->write("\r\n#Visible") < 0) {     //Visible
                retval = ImportExportCodes::Failure;
                break;
            }
            for (idx = 0; idx < layers.size(); idx++) {
                ba = QString(", %1").arg(layers.at(idx)->visible).toUtf8();
                if (io->write(ba.data()) < 0)
                    break;
            }
            if (idx < layers.size()) {
                retval = ImportExportCodes::Failure;
            }
            break;
 
        default :    //frames
 
            if (frame > end) {
                if (io->write("\r\n") < 0)
                    retval = ImportExportCodes::Failure;
 
                step = 8;
                break;
            }
 
            ba = QString("\r\n#%1").arg(frame, 5, 10, QChar('0')).toUtf8();
            if (io->write(ba.data()) < 0) {
                retval = ImportExportCodes::Failure;
                break;
            }
 
            for (idx = 0; idx < layers.size(); idx++) {
                CSVLayerRecord *layer = layers.at(idx);
                KisRasterKeyframeChannel *channel = layer->channel;
 
                if (channel) {
                    if (frame == start) {
                        keyframe = channel->activeKeyframeAt(frame);
                    } else {
                        keyframe = channel->keyframeAt(frame); //TODO: Ugly...
                    }
                } else {
                    keyframe.clear(); // without animation
                }
 
                if ( keyframe || (frame == start) ) {
 
                    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.isOk())
                        break;
                }
                ba = QString(", \"%1\"").arg(layer->last).toUtf8();
 
                if (io->write(ba.data()) < 0)
                    break;
            }
            if (idx < layers.size())
                retval = ImportExportCodes::Failure;
 
            frame++;
            step = 6; //keep step here
            break;
        }
        step++;
    } while((retval.isOk()) && (step < 8));
 
    qDeleteAll(layers);
 
    // io->close();  it seems this is not required anymore
 
    if (!m_batchMode) {
        //disconnect(m_doc, SIGNAL(sigProgressCanceled()), this, SLOT(cancel()));
        //emit m_doc->sigProgress(100);
        //emit m_doc->clearStatusBarMessage();
    }
 
    return retval;
}

References KisKeyframeChannel::activeKeyframeAt(), KisImage::animationInterface(), CSVLayerRecord::blending, ImportExportCodes::Cancelled, CSVLayerRecord::channel, KisBaseNode::compositeOpId(), convertToBlending(), createDocument(), createTempImage(), KisSharedPtr< T >::data(), CSVLayerRecord::density, KisImageAnimationInterface::documentPlaybackRange(), KisTimeSpan::end(), ImportExportCodes::Failure, KisNode::firstChild(), CSVLayerRecord::frame, KisImageAnimationInterface::framerate(), getLayer(), KisImage::height(), KisPart::instance(), KisTimeSpan::isInfinite(), KisImportExportErrorCode::isOk(), KisTimeSpan::isValid(), KisKeyframeChannel::keyframeAt(), KisPaintDevice::keyframeChannel(), KIS_SAFE_ASSERT_RECOVER, CSVLayerRecord::last, KisKeyframeChannel::lastKeyframeTime(), CSVLayerRecord::layer, KisDocument::localFilePath(), m_batchMode, m_doc, m_image, m_stop, KisBaseNode::name(), CSVLayerRecord::name, KisNode::nextSibling(), ImportExportCodes::OK, KisBaseNode::opacity(), OPACITY_OPAQUE_U8, KisLayer::original(), KisImage::rootLayer(), KisTimeSpan::start(), KisBaseNode::visible(), CSVLayerRecord::visible, KisImage::width(), KisImage::xRes(), and KisImage::yRes().

getLayer()

KisImportExportErrorCode CSVSaver::getLayer ( CSVLayerRecord * layer, KisDocument * exportDoc, KisKeyframeSP keyframe, const QString & path, int frame, int idx )

private

Definition at line 377 of file csv_saver.cpp.

{
    //render to the temp layer
    KisImageSP image = exportDoc->savingImage();
 
    if (!image) image= exportDoc->image();
 
    KisPaintDeviceSP device = image->rootLayer()->firstChild()->projection();
 
    if (!keyframe.isNull()) {
        KisRasterKeyframeSP rasterKeyframe = keyframe.dynamicCast<KisRasterKeyframe>();
        if (rasterKeyframe) {
            rasterKeyframe->writeFrameToDevice(device);
        }
    } else {
        device->makeCloneFrom(layer->layer->projection(),image->bounds()); // without animation
    }
    QRect bounds = device->exactBounds();
 
    if (bounds.isEmpty()) {
        layer->last = "";                   //empty frame
        return ImportExportCodes::OK;
    }
    layer->last = QString("frame%1-%2.png").arg(idx + 1,5,10,QChar('0')).arg(frame,5,10,QChar('0'));
 
    QString filename = path;
    filename.append(layer->last);
 
    //save to PNG
    KisSequentialConstIterator it(device, image->bounds());
    const KoColorSpace* cs = device->colorSpace();
 
    bool isThereAlpha = false;
    while (it.nextPixel()) {
        if (cs->opacityU8(it.oldRawData()) != OPACITY_OPAQUE_U8) {
            isThereAlpha = true;
            break;
        }
    }
 
    if (!KisPNGConverter::isColorSpaceSupported(cs)) {
        device = new KisPaintDevice(*device.data());
        device->convertTo(KoColorSpaceRegistry::instance()->rgb8());
    }
    KisPNGOptions options;
 
    options.alpha = isThereAlpha;
    options.interlace = false;
    options.compression = 8;
    options.tryToSaveAsIndexed = false;
    options.transparencyFillColor = QColor(0,0,0);
    options.saveSRGBProfile = true;                 //TVPaint can use only sRGB
    options.forceSRGB = false;
 
    KisPNGConverter kpc(exportDoc);
 
    KisImportExportErrorCode result = kpc.buildFile(filename, image->bounds(),
                                                  image->xRes(), image->yRes(), device,
                                                  image->beginAnnotations(), image->endAnnotations(),
                                                  options, (KisMetaData::Store* )0 );
 
    return result;
}

References KisPNGOptions::alpha, KisImage::beginAnnotations(), KisImage::bounds(), bounds, KisPNGConverter::buildFile(), KisPaintDevice::colorSpace(), KisPNGOptions::compression, KisPaintDevice::convertTo(), KisSharedPtr< T >::data(), KisImage::endAnnotations(), KisPaintDevice::exactBounds(), KisNode::firstChild(), KisPNGOptions::forceSRGB, KisDocument::image, image(), KoColorSpaceRegistry::instance(), KisPNGOptions::interlace, KisPNGConverter::isColorSpaceSupported(), CSVLayerRecord::last, CSVLayerRecord::layer, KisPaintDevice::makeCloneFrom(), KisSequentialIteratorBase< IteratorPolicy, SourcePolicy, ProgressPolicy >::nextPixel(), ImportExportCodes::OK, KisSequentialIteratorBase< IteratorPolicy, SourcePolicy, ProgressPolicy >::oldRawData(), OPACITY_OPAQUE_U8, KoColorSpace::opacityU8(), KisBaseNode::projection(), KisLayer::projection(), KisImage::rootLayer(), KisPNGOptions::saveSRGBProfile, KisDocument::savingImage, KisPNGOptions::transparencyFillColor, KisPNGOptions::tryToSaveAsIndexed, KisImage::xRes(), and KisImage::yRes().

image()

KisImageSP CSVSaver::image

(

)

Definition at line 52 of file csv_saver.cpp.

{
    return m_image;
}

References m_image.

Member Data Documentation

m_batchMode

bool CSVSaver::m_batchMode

private

Definition at line 42 of file csv_saver.h.

m_doc

KisDocument* CSVSaver::m_doc

private

Definition at line 41 of file csv_saver.h.

m_image

KisImageSP CSVSaver::m_image

private

Definition at line 40 of file csv_saver.h.

m_stop

bool CSVSaver::m_stop

private

Definition at line 43 of file csv_saver.h.

---

The documentation for this class was generated from the following files:

• plugins/impex/csv/csv_saver.h
• plugins/impex/csv/csv_saver.cpp