File indexing completed on 2024-05-12 16:01:54

 /*
  *  SPDX-FileCopyrightText: 2016 Boudewijn Rempt <boud@valdyas.org>
  *
  *  SPDX-License-Identifier: GPL-2.0-or-later
  */
 #include "KisSaveGroupVisitor.h"
 
 #include <KisDocument.h>
 #include <kis_painter.h>
 #include <kis_paint_layer.h>
 #include <KisPart.h>
 
 KisSaveGroupVisitor::KisSaveGroupVisitor(KisImageWSP image,
                                          bool saveInvisible,
                                          bool saveTopLevelOnly,
                                          const QString &path,
                                          const QString &baseName,
                                          const QString &extension,
                                          const QString &mimeFilter)
     : m_image(image)
     , m_saveInvisible(saveInvisible)
     , m_saveTopLevelOnly(saveTopLevelOnly)
     , m_path(path)
     , m_baseName(baseName)
     , m_extension(extension)
     , m_mimeFilter(mimeFilter)
 {
 }
 
 KisSaveGroupVisitor::~KisSaveGroupVisitor()
 {
 }
 
 bool KisSaveGroupVisitor::visit(KisNode* ) {
     return true;
 }
 
 bool KisSaveGroupVisitor::visit(KisPaintLayer *) {
     return true;
 }
 
 bool KisSaveGroupVisitor::visit(KisAdjustmentLayer *) {
     return true;
 }
 
 bool KisSaveGroupVisitor::visit(KisExternalLayer *) {
     return true;
 }
 
 
 bool KisSaveGroupVisitor::visit(KisCloneLayer *) {
     return true;
 }
 
 
 bool KisSaveGroupVisitor::visit(KisFilterMask *) {
     return true;
 }
 
 bool KisSaveGroupVisitor::visit(KisTransformMask *) {
     return true;
 }
 
 bool KisSaveGroupVisitor::visit(KisTransparencyMask *) {
     return true;
 }
 
 bool KisSaveGroupVisitor::visit(KisGeneratorLayer * ) {
     return true;
 }
 
 bool KisSaveGroupVisitor::visit(KisSelectionMask* ) {
     return true;
 }
 
 bool KisSaveGroupVisitor::visit(KisColorizeMask* ) {
     return true;
 }
 
 bool KisSaveGroupVisitor::visit(KisGroupLayer *layer)
 {
     if (layer == m_image->rootLayer()) {
         KisLayerSP child = qobject_cast<KisLayer*>(layer->firstChild().data());
         while (child) {
             child->accept(*this);
             child = qobject_cast<KisLayer*>(child->nextSibling().data());
         }
 
     }
     else if (layer->visible() || m_saveInvisible) {
 
         QRect r = m_image->bounds();
 
         KisDocument *exportDocument = KisPart::instance()->createDocument();
         { // make sure dst is deleted before calling 'delete exportDocument',
           // since KisDocument checks that its image is properly deref()'d.
           KisImageSP dst = new KisImage(exportDocument->createUndoStore(), r.width(), r.height(), m_image->colorSpace(), layer->name());
           dst->setResolution(m_image->xRes(), m_image->yRes());
           exportDocument->setCurrentImage(dst);
           KisPaintLayer* paintLayer = new KisPaintLayer(dst, "projection", layer->opacity());
           KisPainter gc(paintLayer->paintDevice());
           gc.bitBlt(QPoint(0, 0), layer->projection(), r);
           dst->addNode(paintLayer, dst->rootLayer(), KisLayerSP(0));
 
           dst->refreshGraph();
         }
 
         QString path = m_path + "/" + m_baseName + "_" + layer->name().replace(' ', '_') + '.' + m_extension;
 
         exportDocument->setFileBatchMode(true);
         exportDocument->exportDocumentSync(path, m_mimeFilter.toLatin1());
 
         if (!m_saveTopLevelOnly) {
             KisGroupLayerSP child = dynamic_cast<KisGroupLayer*>(layer->firstChild().data());
             while (child) {
                 child->accept(*this);
                 child = dynamic_cast<KisGroupLayer*>(child->nextSibling().data());
             }
         }
         delete exportDocument;
     }
 
     return true;
 }