File indexing completed on 2024-05-12 15:59:02

 /*
  * SPDX-FileCopyrightText: 2016 Boudewijn Rempt <boud@valdyas.org>
  *
  * SPDX-License-Identifier: LGPL-2.0-or-later
  */
 
 #ifndef NodeTypeCheck_H
 #define NodeTypeCheck_H
 
 #include "KisExportCheckRegistry.h"
 #include <KoID.h>
 #include <klocalizedstring.h>
 #include <kis_image.h>
 #include <kis_count_visitor.h>
 
 class NodeTypeCheck : public KisExportCheckBase
 {
 public:
 
     NodeTypeCheck(const QString &nodeType, const QString &description, const QString &id, Level level, const QString &customWarning = QString())
         : KisExportCheckBase(id, level, customWarning, true)
         , m_nodeType(nodeType)
     {
         if (customWarning.isEmpty()) {
             m_warning = i18nc("image conversion warning", "The image contains layers of unsupported type <b>%1</b>. Only the rendered result will be saved.", description);
         }
     }
 
     bool checkNeeded(KisImageSP image) const override
     {
         QStringList nodetypes = QStringList() << m_nodeType;
         KoProperties props;
         KisCountVisitor v(nodetypes, props);
         image->rootLayer()->accept(v);
 
         // There is always one group layer, the root layer.
         if (m_nodeType == "KisGroupLayer") {
             return (v.count() > 1);
         }
         else {
             return (v.count() > 0);
         }
     }
 
     Level check(KisImageSP /*image*/) const override
     {
         return m_level;
     }
 
     QString m_nodeType;
 };
 
 class NodeTypeCheckFactory : public KisExportCheckFactory
 {
 public:
 
     NodeTypeCheckFactory(const QString &nodeType, const QString &description)
         : m_nodeType(nodeType)
         , m_description(description)
     {}
 
     ~NodeTypeCheckFactory() override {}
 
     KisExportCheckBase *create(KisExportCheckBase::Level level, const QString &customWarning) override
     {
         return new NodeTypeCheck(m_nodeType, m_description, id(), level, customWarning);
     }
 
     QString id() const override {
         return "NodeTypeCheck/" + m_nodeType;
     }
 
     QString m_nodeType;
     QString m_description;
 
 };
 
 #endif // NodeTypeCheck_H