xmlprojectelementcollectionitem.cpp

Go to the documentation of this file.

/*
        Copyright 2006-2019 The QElectroTech Team
        This file is part of QElectroTech.

        QElectroTech is free software: you can redistribute it and/or modify
        it under the terms of the GNU General Public License as published by
        the Free Software Foundation, either version 2 of the License, or
        (at your option) any later version.

        QElectroTech is distributed in the hope that it will be useful,
        but WITHOUT ANY WARRANTY; without even the implied warranty of
        MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
        GNU General Public License for more details.

        You should have received a copy of the GNU General Public License
        along with QElectroTech. If not, see <http://www.gnu.org/licenses/>.
*/
#include "qetproject.h"
#include "xmlelementcollection.h"
#include "qeticons.h"
#include "xmlprojectelementcollectionitem.h"

XmlProjectElementCollectionItem::XmlProjectElementCollectionItem()
{}

bool XmlProjectElementCollectionItem::isDir() const
{
    if (m_dom_element.tagName() == "category") return true;
    else return false;
}

bool XmlProjectElementCollectionItem::isElement() const
{
    if (m_dom_element.tagName() == "element") return true;
    else return false;
}

QString XmlProjectElementCollectionItem::localName()
{
    if (!text().isNull())
        return text();

    if (isCollectionRoot()) {
        if (m_project->title().isEmpty())
            setText(QObject::tr("Projet sans titre"));
        else
            setText(m_project->title());
    }
    else {
        ElementsLocation location (embeddedPath(), m_project);
        setText(location.name());
    }

    return text();
}

QString XmlProjectElementCollectionItem::name() const
{
    return m_dom_element.attribute("name");
}

QString XmlProjectElementCollectionItem::collectionPath() const
{
    ElementsLocation loc (embeddedPath(), m_project);
    QString p = loc.projectCollectionPath();
    if (p.isEmpty())
        p = QObject::tr("Collection");
    return p;
}

QString XmlProjectElementCollectionItem::embeddedPath() const
{
    if (isCollectionRoot())
        return "embed://";
    else if (parent()){
        XmlProjectElementCollectionItem *xpeci = static_cast<XmlProjectElementCollectionItem *>(parent());

        if (xpeci->isCollectionRoot())
            return xpeci->embeddedPath() + name();
        else
            return xpeci->embeddedPath() + "/" + name();
    }
    else
        return QString();
}

bool XmlProjectElementCollectionItem::isCollectionRoot() const
{
    if (!parent())
        return true;
    else if (parent()->type() != XmlProjectElementCollectionItem::Type)
        return true;
    else
        return false;
}

void XmlProjectElementCollectionItem::addChildAtPath(const QString &collection_name)
{
    if (collection_name.isEmpty())
        return;

    QString str (collection_name.endsWith(".elmt")? "element" : "category");
    QDomElement child_element = m_dom_element.firstChildElement(str);

    while (!child_element.isNull()) {
        if (child_element.attribute("name") == collection_name) {
            XmlProjectElementCollectionItem *xpeci = new XmlProjectElementCollectionItem ();
            insertRow(rowForInsertItem(collection_name), xpeci);
            xpeci->setXmlElement(child_element, m_project);
            xpeci->setUpData();
            return;
        }
        else
            child_element = child_element.nextSiblingElement(str);
    }
}

QETProject *XmlProjectElementCollectionItem::project() const
{
    return m_project;
}

void XmlProjectElementCollectionItem::setProject(QETProject *project, bool set_data, bool hide_element)
{
    if (m_project)
        return;

    m_project = project;
    m_dom_element = project->embeddedElementCollection()->root();
    populate(set_data, hide_element);
}

void XmlProjectElementCollectionItem::setUpData()
{
        //Setup the displayed name
    localName();

    if (isDir()) {
        setFlags(Qt::ItemIsSelectable | Qt::ItemIsDragEnabled | Qt::ItemIsDropEnabled | Qt::ItemIsEnabled);
    }
    else
    {
        setFlags(Qt::ItemIsSelectable | Qt::ItemIsDragEnabled | Qt::ItemIsEnabled);
            
            //Set the local name and all informations of the element
            //in the data Qt::UserRole+1, these data will be use for search.
        ElementsLocation location(embeddedPath(), m_project);
        DiagramContext context = location.elementInformations();
        QStringList search_list;
        for (QString key : context.keys()) {
            search_list.append(context.value(key).toString());
        }
        search_list.append(localName());
        setData(search_list.join(" "));
    }

    setToolTip(collectionPath());
}

void XmlProjectElementCollectionItem::setUpIcon()
{
    if (!icon().isNull())
        return;

    if (isCollectionRoot())
        setIcon(QET::Icons::ProjectFileGP);
    else if (isDir())
        setIcon(QET::Icons::Folder);
    else {
        ElementsLocation loc(embeddedPath(), m_project);
        setIcon(loc.icon());
    }
}

void XmlProjectElementCollectionItem::populate(bool set_data, bool hide_element)
{
    QList <QDomElement> dom_category = m_project->embeddedElementCollection()->directories(m_dom_element);
    std::sort(dom_category.begin(), dom_category.end(), [](QDomElement a, QDomElement b){return (a.attribute("name") < b.attribute("name"));});

    foreach (QDomElement element, dom_category)
    {
        XmlProjectElementCollectionItem *xpeci = new XmlProjectElementCollectionItem();
        appendRow(xpeci);
        xpeci->setXmlElement(element, m_project, set_data, hide_element);
        if (set_data)
            xpeci->setUpData();
    }

    if (hide_element)
        return;

    QList <QDomElement> dom_elements = m_project->embeddedElementCollection()->elements(m_dom_element);
    std::sort(dom_elements.begin(), dom_elements.end(), [](QDomElement a, QDomElement b){return (a.attribute("name") < b.attribute("name"));});

    foreach (QDomElement element, dom_elements)
    {
        XmlProjectElementCollectionItem *xpeci = new XmlProjectElementCollectionItem();
        appendRow(xpeci);
        xpeci->setXmlElement(element, m_project, set_data);
        if (set_data)
            xpeci->setUpData();
    }
}

void XmlProjectElementCollectionItem::setXmlElement(const QDomElement& element, QETProject *project, bool set_data, bool hide_element)
{
    m_dom_element = element;
    m_project = project;
    populate(set_data, hide_element);
}