qelectrotech-source-mirror/sources/ElementsCollection/xmlprojectelementcollectionitem.cpp

/*
        Copyright 2006-2015 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 "xmlprojectelementcollectionitem.h"
#include "qetproject.h"
#include "qeticons.h"
#include "xmlelementcollection.h"
#include "nameslist.h"
#include "qetapp.h"
#include <algorithm>

/**
 * @brief XmlProjectElementCollectionItem::XmlProjectElementCollectionItem
 * Default constructor.
 * @param project -project for this item
 * @param parent -paretn item
 */
XmlProjectElementCollectionItem::XmlProjectElementCollectionItem(QETProject *project, ElementCollectionItem *parent) :
	ElementCollectionItem(parent),
	m_project(project)
{
	m_dom_element = project->embeddedElementCollection()->root();
	populate();
}

/**
 * @brief XmlProjectElementCollectionItem::XmlProjectElementCollectionItem
 * Private constructor
 * @param project -project for this item
 * @param dom_element: the dom_element must represent this item
 * @param parent
 */
XmlProjectElementCollectionItem::XmlProjectElementCollectionItem(QETProject *project, const QDomElement &dom_element, ElementCollectionItem *parent) :
	ElementCollectionItem(parent),
	m_project(project),
	m_dom_element(dom_element)
{
	populate();
}

/**
 * @brief XmlProjectElementCollectionItem::~XmlProjectElementCollectionItem
 */
XmlProjectElementCollectionItem::~XmlProjectElementCollectionItem()
{}

/**
 * @brief XmlProjectElementCollectionItem::data
 * The data used by the view who display this item through the model
 * @param column
 * @param role
 * @return
 */
QVariant XmlProjectElementCollectionItem::data(int column, int role)
{
	if (column > 0)
		return QVariant();

    switch (role)
    {
        case Qt::DisplayRole:
            return name();
            break;
        case Qt::DecorationRole:
			if (isCollectionRoot())
				return QIcon(QET::Icons::ProjectFileGP);
			else if (isDir())
				return QET::Icons::Folder;
			else
				return QET::Icons::Element;
            break;
		case Qt::ToolTipRole:
			if (isCollectionRoot())
				return m_project->filePath();
			else
				return collectionPath();
			break;
        default:
            return QVariant();
	}
}

/**
 * @brief XmlProjectElementCollectionItem::mimeData
 * @return The mimedata of this item
 */
QMimeData *XmlProjectElementCollectionItem::mimeData()
{
	QMimeData *mime_data = new QMimeData();
	mime_data->setText(collectionPath());

	if (isElement())
		mime_data->setData("application/x-qet-element-uri", collectionPath().toLatin1());
	else
		mime_data->setData("application/x-qet-category-uri", collectionPath().toLatin1());

	return mime_data;
}

/**
 * @brief XmlProjectElementCollectionItem::flags
 * @return The flags of this item
 */
Qt::ItemFlags XmlProjectElementCollectionItem::flags()
{
	if (isDir())
		return (Qt::ItemIsSelectable | Qt::ItemIsDragEnabled | Qt::ItemIsDropEnabled | Qt::ItemIsEnabled);
	else
		return (Qt::ItemIsSelectable | Qt::ItemIsDragEnabled | Qt::ItemIsEnabled);
}

/**
 * @brief XmlProjectElementCollectionItem::isCollectionRoot
 * @return True if this item represent the root collection of a project
 */
bool XmlProjectElementCollectionItem::isCollectionRoot() const
{
    if (!m_parent_item) return true;
    else if (m_parent_item->type() != XmlProjectElementCollectionItem::Type) return true;
    else return false;
}

/**
 * @brief XmlProjectElementCollectionItem::name
 * @return The name of this item, name is notably use for Qt::DisplayRole data
 */
QString XmlProjectElementCollectionItem::name()
{
    if (!m_name.isNull()) return m_name;

    if (isCollectionRoot())
    {
		if (m_project->title().isEmpty())
            return QString("Projet sans titre");
        else
			return m_project->title();
    }
	else
	{
		NamesList nl;
		if (isDir())
		{
			nl.fromXml(m_dom_element);
			if (nl.name().isEmpty())
				m_name = m_dom_element.attribute("name");
			else
				m_name = nl.name();
		}
		else
		{
			nl.fromXml(m_dom_element.firstChildElement("definition"));
			if (nl.name().isEmpty())
				m_name = m_dom_element.attribute("name");
			else
				m_name = nl.name();
		}
		return m_name;
	}
}

/**
 * @brief XmlProjectElementCollectionItem::isValid
 * @return Always true
 */
bool XmlProjectElementCollectionItem::isValid() const {
	return true;
}

/**
 * @brief XmlProjectElementCollectionItem::project
 * @return The project for this collection item
 */
QETProject *XmlProjectElementCollectionItem::project() const {
	return m_project;
}

/**
 * @brief XmlProjectElementCollectionItem::isDir
 * @return true if this item represent a directory
 */
bool XmlProjectElementCollectionItem::isDir() const
{
	if (m_dom_element.tagName() == "category") return true;
	else return false;
}

/**
 * @brief XmlProjectElementCollectionItem::isElement
 * @return true if this item represent an element
 */
bool XmlProjectElementCollectionItem::isElement() const
{
	if (m_dom_element.tagName() == "element") return true;
	else return false;
}

/**
 * @brief XmlProjectElementCollectionItem::collectionPath
 * @return The collection path of this item
 */
QString XmlProjectElementCollectionItem::collectionPath() const
{
	if (isCollectionRoot())
	{
		QString path;
		return path + "project" + QString::number(QETApp::projectId(m_project)) + "+embed://";
	}
	else
	{
		XmlProjectElementCollectionItem *parent = static_cast<XmlProjectElementCollectionItem *>(m_parent_item);

		if (parent->isCollectionRoot())
			return parent->collectionPath() + m_dom_element.attribute("name");
		else
			return parent->collectionPath() + "/" + m_dom_element.attribute("name");
	}
}

/**
 * @brief XmlProjectElementCollectionItem::populate
 * Populate this item
 */
void XmlProjectElementCollectionItem::populate()
{
	QList <QDomElement> dom_category = m_project->embeddedElementCollection()->directory(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(m_project, element, this);
		this->appendChild(xpeci);
	}

	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(m_project, element, this);
		this->appendChild(xpeci);
	}
}