costpositionmodel.cpp

Go to the documentation of this file.

#include <QStringList>
#include <QDomDocument>
#include <QDebug>
#include <QFile>
#include "costpositionmodel.h"

CostPositionModel::CostPositionModel(QObject *parent, bool empty) : QAbstractItemModel(parent)
{
    qDebug()<<"Improvement needed: CostPositionModel -- better way for getting the CostPositionItem* from the pointer in the data-methode";

    QList<QVariant> rootData;
    rootData << tr("Kostenpositionen");
    rootItem = new CostPositionItem(rootData);

    if(!empty)
    {
        readCategories();
    }
}

CostPositionModel::~CostPositionModel()
{
    delete rootItem;
}

int CostPositionModel::columnCount(const QModelIndex &parent) const
{
    if(parent.isValid())
    {
        return static_cast<CostPositionItem*>(parent.internalPointer())->columnCount();
    }
    else
    {
        return rootItem->columnCount();
    }
}

QVariant CostPositionModel::data(const QModelIndex &index, int role) const
{
    if(!index.isValid())
    {
        return QVariant();
    }

    if(role != Qt::DisplayRole && role != 32)
    {
        return QVariant();
    }
    else if(role == 32)
    {
        CostPositionItem *item = getCostPositionItem(index.data(Qt::DisplayRole).toString(), index.parent().data(Qt::DisplayRole).toString());
        return item->itemInfo();
    }
    else
    {
        CostPositionItem *item = static_cast<CostPositionItem*>(index.internalPointer());
        return item->data(index.column());
    }
}

CostPositionItem* CostPositionModel::getCostPositionItem(QString name, QString secName) const
{
    CostPositionItem* root = rootItem;
    for(int i = 0; i != root->childCount(); i++)
    {
        CostPositionItem *category = root->child(i);
        if(category->data(0).toString() == secName)
        {
            for(int j=0; j!=category->childCount(); j++)
            {
                CostPositionItem *item = category->child(j);
                if(item->data(0).toString() == name)
                {
                    return item;
                }
            }
        }
    }
    return new CostPositionItem(QVariantList());
}

Qt::ItemFlags CostPositionModel::flags(const QModelIndex &index) const
{
    if(!index.isValid())
    {
        return 0;
    }

    return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
}

QVariant CostPositionModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if(orientation == Qt::Horizontal && role == Qt::DisplayRole)
    {
        return rootItem->data(section);
    }

    return QVariant();
}

QModelIndex CostPositionModel::index(int row, int column, const QModelIndex &parent) const
{
    if(!hasIndex(row, column, parent))
    {
        return QModelIndex();
    }

    CostPositionItem *parentItem;

    if(!parent.isValid())
    {
        parentItem = rootItem;
    }
    else
    {
        parentItem = static_cast<CostPositionItem*>(parent.internalPointer());
    }

    CostPositionItem *childItem = parentItem->child(row);
    if(childItem)
    {
        return createIndex(row, column, childItem);
    }
    else
    {
        return QModelIndex();
    }
}

QModelIndex CostPositionModel::parent(const QModelIndex &index) const
{
    if(!index.isValid())
    {
        return QModelIndex();
    }

    CostPositionItem *childItem = static_cast<CostPositionItem*>(index.internalPointer());
    CostPositionItem *parentItem = childItem->parent();

    if(parentItem == rootItem)
    {
        return QModelIndex();
    }

    return createIndex(parentItem->row(), 0, parentItem);
}

int CostPositionModel::rowCount(const QModelIndex &parent) const
{
    CostPositionItem *parentItem;
    if(parent.column() > 0)
        return 0;

    if(!parent.isValid())
    {
        parentItem = rootItem;
    }
    else
    {
        parentItem = static_cast<CostPositionItem*>(parent.internalPointer());
    }
    return parentItem->childCount();
}

void CostPositionModel::addCategory(uint sectionid, QString name, bool editable)
{
    QMap<QString, QVariant> d;
    QVariantList c;

    d.insert("sectionId", sectionid);
    d.insert("editable", editable);

    c<<name;                                                    //name of a category
    c<<d;                                                       //other information for category

    rootItem->appendChild(new CostPositionItem(c, rootItem));
    reset();            //NEEDED!!!! -- otherwise the view doesn't display anything of the new category!
}

void CostPositionModel::addCostPosition(QString desc, QString lang, int action, int costtype, QString secName, bool editable)
{
    if(existsCategory(secName))
    {
        if(!existsPositionInCategory(desc, getSectionId(secName)))
        {
            addPosition(desc, lang, action, costtype, getSectionId(secName), editable);
        }
    }
    else
    {
        int count = rootItem->childItems.count();
        addCategory(count++, secName, editable);
        addPosition(desc, lang, action, costtype, getSectionId(secName), editable);
    }
}

void CostPositionModel::addPosition(QString desc, QString lang, int action, int costtype, int sectionid, bool editable)
{
    for(int i=0; i < rootItem->childCount();i++)
    {
        CostPositionItem *cate = rootItem->child(i);
        if(cate->data(1).toMap().value("sectionId").toInt() == sectionid)
        {
            QMap<QString, QVariant> data;
            QVariantList columnData;

            data.insert("language", lang);
            data.insert("action", action);
            data.insert("costtype", costtype);
            data.insert("sectionId", sectionid);
            data.insert("editable", editable);

            columnData<<desc;
            columnData<<data;

            rootItem->childItems.value(i)->appendChild(new CostPositionItem(columnData, rootItem->childItems.value(i)));
            reset();            //NEEDED!!!! -- otherwise the view doesn't display anything of the new position!
        }
    }
}

int CostPositionModel::getSectionId(QString cat)
{
    int id = -1;
    for(int i=0; i!=rootItem->childCount(); i++)
    {
        CostPositionItem *cate = rootItem->child(i);
        if(cate->data(0).toString() == cat)
        {
            id = cate->data(1).toMap().value("sectionId").toInt();
        }
    }
    return id;
}

int CostPositionModel::getSectionIdFromPosition(QString positionName)
{
    for(int i=0; i < rootItem->childCount(); i++)
    {
        CostPositionItem *cat= rootItem->child(i);
        for(int j=0; j < cat->childCount(); j++)
        {
            CostPositionItem *item = cat->child(j);
            if(item->data(0).toString() == positionName)
            {
                return item->data(1).toMap().value("section-id").toInt();
            }
        }
    }
    return -1;
}

QString CostPositionModel::getCategoryNameFromSectionId(int sectionId)
{
    CostPositionItem *root = rootItem;
    QString name;
    for(int i=0; i!=root->childCount(); i++)
    {
        CostPositionItem *cat = root->child(i);
        if(cat->data(1).toMap().value("sectionId").toInt() == sectionId)
        {
            name = cat->data(0).toString();
        }
    }
    return name;
}

bool CostPositionModel::existsCategory(QString name)
{
    bool exist = false;
    for(int i=0; i!=rootItem->childCount(); i++)
    {
        CostPositionItem *cate = rootItem->child(i);
        if(cate->data(0).toString() == name)
        {
            exist = true;
        }
    }
    return exist;
}

bool CostPositionModel::existsPositionInCategory(QString posName, int sectionid)
{
    bool exist = false;
    for(int i=0; i!=rootItem->childCount(); i++)
    {
        CostPositionItem *cate = rootItem->child(i);
        if(cate->data(1).toMap().value("id").toInt() == sectionid)
        {
            for(int j=0; j!=cate->childCount(); j++)
            {
                CostPositionItem *pos = cate->child(j);
                if(pos->data(0).toString() == posName)
                {
                    exist = true;
                }
            }

        }
    }
    return exist;
}

bool CostPositionModel::isCategoryEditable(int sectionId)
{
    for(int i=0; i < rootItem->childCount(); i++)
    {
        CostPositionItem *cat = rootItem->child(i);
        if(cat->data(1).toMap().value("sectionId").toInt() == sectionId)
        {
            if(cat->data(1).toMap().value("editable").toBool() == true)
            {
                return true;
            }
        }
    }
    return false;
}

bool CostPositionModel::isPositionEditable(QString positionName, int sectionId)
{
    for(int i=0; i < rootItem->childCount(); i++)
    {
        CostPositionItem *cat = rootItem->child(i);
        qDebug()<<getSectionIdFromPosition(positionName);
        if(cat->data(1).toMap().value("sectionId").toInt() == sectionId)
        {
            for(int j=0; j < cat->childCount(); j++)
            {
                CostPositionItem *item = cat->child(j);
                if(item->data(0).toString() == positionName)
                {
                    if(item->data(1).toMap().value("editable").toBool() == true)
                    {
                        return true;
                    }
                }
            }
        }
    }
    return false;
}

void CostPositionModel::deletePosition(QString positionname, int sectionId)
{
    for(int i=0; i!=rootItem->childCount(); i++)
    {
        CostPositionItem *cat = rootItem->child(i);
        if(cat->data(1).toMap().value("sectionId").toInt() == sectionId)
        {
            for(int j=0; j!=cat->childCount(); j++)
            {
                CostPositionItem *item = cat->child(j);
                if(item->data(0).toString() == positionname)
                {
                    cat->removeChild(item);
                    reset();
                    return;
                }
            }
        }
    }
}

void CostPositionModel::deleteCategory(int sectionid)
{
    for(int i=0; i!=rootItem->childCount(); i++)
    {
        CostPositionItem *cat = rootItem->child(i);
        if(cat->data(1).toMap().value("sectionId").toInt() == sectionid)
        {
            qDebug()<<i<<" "<<rootItem->childItems.size();
            rootItem->removeChild(cat);
            reset();
            return;
        }
    }
}

void CostPositionModel::editCategory(QString oldCat, QString newCat)
{
    for(int i=0; i < rootItem->childCount(); i++)
    {
        CostPositionItem *cat = rootItem->child(i);
        if(cat->data(0).toString() == oldCat)
        {
           // qDebug()<<"oldCat: "<<cat->data(0).toString();

            QMap<QString, QVariant> d;
            QVariantList c;

            d.insert("sectionId", cat->data(1).toMap().value("sectionId"));
            d.insert("editable", cat->data(1).toMap().value("editable"));

            c<<newCat;                                                          //name of a category
            c<<d;
            CostPositionItem *cate = new CostPositionItem(c, rootItem);

            //qDebug()<<i<<" "<<cate->itemInfo();
            rootItem->childItems.replace(i, cate);
            // have to add all other position form the other category
            addOtherPositions(cate->data(1).toMap().value("sectionId").toInt(), cat);

            //qDebug()<<"newCat: "<<cat->data(0).toString();
            reset();
            return;
        }
    }
}

void CostPositionModel::addOtherPositions(int sectionId, CostPositionItem *&oldCat)
{
    // sectionId --> the place, where the oldPosition were added
    // oldCat --> all oldPositions
    for(int i=0; i!=rootItem->childCount(); i++)
    {
        CostPositionItem *cate = rootItem->child(i);
        if(cate->data(1).toMap().value("sectionId").toInt() == sectionId)
        {
            for(int j=0; j!= oldCat->childCount(); j++)
            {
                CostPositionItem *item = oldCat->child(j);
                QMap<QString, QVariant> d;
                QVariantList c;

                d.insert("sectionId", item->data(1).toMap().value("sectionId"));
                d.insert("editable", item->data(1).toMap().value("editable"));
                d.insert("action", item->data(1).toMap().value("action"));
                d.insert("costtype", item->data(1).toMap().value("costtype"));
                d.insert("language", item->data(1).toMap().value("language"));

                c<<item->data(0).toString();
                c<<d;

                CostPositionItem *position = new CostPositionItem(c, cate);
                cate->appendChild(position);
                reset();
            }
        }
    }
}

void CostPositionModel::editPosition(QMap<QString, QVariant> *changes)
{
    if(changes->value("newCategoryName").toString() != changes->value("oldCategoryName").toString())
    {
        if(!existsCategory(changes->value("newCategory").toString()))
        {
            addCategory(rootItem->childCount(), changes->value("newCategoryName").toString(), true);
        }
        CostPositionItem *item = getCostPositionItem(changes->value("oldPositionName").toString(), changes->value("oldCategoryName").toString());

        QVariantList list = item->itemInfo();
        CostPositionItem *temp = new CostPositionItem(list, rootItem->child(rootItem->childCount()-1));

        rootItem->child(rootItem->childCount()-1)->appendChild(temp);
        rootItem->child(getSectionId(changes->value("oldCategoryName").toString()))->removeChild(item);

    }
    if(changes->value("newPositionName").toString() != changes->value("oldPositionName").toString() || changes->value("newAction").toInt() != changes->value("oldAction").toInt() || changes->value("newCosttype").toInt() != changes->value("oldCosttype").toInt())
    {
        CostPositionItem *item = getCostPositionItem(changes->value("oldPositionName").toString(), changes->value("newCategoryName").toString());
        rootItem->child(getSectionId(changes->value("newCategoryName").toString()))->removeChild(item);
        addPosition(changes->value("newPositionName").toString(), "de", changes->value("newAction").toInt(), changes->value("newCosttype").toInt(), getSectionId(changes->value("newCategoryName").toString()), true);
    }
    reset();
}

// load the Xml file here (positions coming by default with the application), add them to the model
void CostPositionModel::readCategories()
{
    QFile file(":/data/costpositions");
    QDomDocument doc("Allevo");
    if(!file.open(QIODevice::ReadOnly))
    {
        emit costTypePositionsNotReadable();
        return;
    }
    if(!doc.setContent(&file))
    {
        emit costTypePositionsNotReadable();
        file.close();
        return;
    }
    file.close();

    // clears list
    clearPositions();

    // resets direct/fixed costs radio buttons -- false because we don't want to clear the list again
    emit clearCostPositions(false);

    QStringList documentinfo = QStringList();
    QStringList h = QStringList();
    QDomElement root = doc.documentElement();
    QDomNode node;
    node = root.firstChild();
    while(!node.isNull())
    {
        if(node.isElement() && node.nodeName() == "section")
        {
            QDomElement e = node.toElement();
            addCategory(e.attribute("id").toInt(), e.attribute("desc"), (e.attribute("editable")=="true")?true:false);

            QDomElement info = node.toElement();
            QDomNode n = info.firstChild();
            while(!n.isNull())
            {
                if(n.isElement())
                {
                    if(n.nodeName() == "category")
                    {
                        QDomElement element = n.toElement();
                        QDomElement e2 = node.toElement();

                        // add Position with theses attributes: desc, lang, action, costtype, section-id, editable
                        addCostPosition(element.attribute("name"), element.attribute("lang"), element.attribute("action").toInt(), element.attribute("costtype").toInt(), e.attribute("desc"), (element.attribute("editable")=="true")?true:false);
                    }
                    n = n.nextSibling();
                }
            }
            node = node.nextSibling();
        }
    }
}

QDomElement CostPositionModel::dataNode()
{
    QDomDocument doc;
    QDomElement categories = doc.createElement("categories");
    bool found = false;

    CostPositionItem *root = rootItem;
    for(int i = 0; i != root->childCount(); i++)
    {
        CostPositionItem *category = root->child(i);
        QDomElement section = doc.createElement("section");
        section.setAttribute("name", category->data(0).toString());
        section.setAttribute("lang", "de");
        section.setAttribute("sectionId", category->data(1).toMap().value("sectionId").toString());
        section.setAttribute("editable", category->data(1).toMap().value("editable").toString());
        categories.appendChild(section);
        for(int j = 0; j != category->childCount(); j++)
        {
            CostPositionItem *item = category->child(j);
            if(item->data(1).toMap().value("editable").toBool())
            {
                found = true;
                QDomElement position = doc.createElement("position");
                position.setAttribute("positionName", item->data(0).toString());
                position.setAttribute("action", item->data(1).toMap().value("action").toString());
                position.setAttribute("costtype", item->data(1).toMap().value("costtype").toString());
                position.setAttribute("lang", item->data(1).toMap().value("language").toString());
                position.setAttribute("editable", item->data(1).toMap().value("editable").toString());
                section.appendChild(position);
            }
        }
        if(!found)                              //don't add the category if there's no user created position in it
        {
            categories.removeChild(section);
        }
        else
        {
            found = false;
        }
    }
    return categories;
}

void CostPositionModel::clearPositions()
{
    int counter = 0;
    while(rootItem->childCount() > 0)
    {
        CostPositionItem *item = rootItem->child(counter);
        item->removeChilds();

        rootItem->removeChild(item);
    }
}

CostPositionItem::CostPositionItem(const QVariantList &data, CostPositionItem *parent)
{
    parentItem = parent;
    itemData = data;
}

CostPositionItem::~CostPositionItem()
{
    qDeleteAll(childItems);
}

void CostPositionItem::appendChild(CostPositionItem *item)
{
    childItems.append(item);
}

void CostPositionItem::removeChild(CostPositionItem *item)
{
    childItems.removeAll(item);
}

void CostPositionItem::removeChilds()
{
    childItems.clear();
}

CostPositionItem *CostPositionItem::child(int row)
{
    return childItems.value(row);
}

int CostPositionItem::childCount() const
{
    return childItems.count();
}

int CostPositionItem::columnCount() const
{
    return itemData.count();
}

QVariant CostPositionItem::data(int column) const
{
    return itemData.value(column);
}

QVariantList CostPositionItem::itemInfo() const
{
    return itemData;
}

QMap<QString, QVariant> CostPositionItem::itemInformation()
{
    return itemData.value(0).value<QMap<QString, QVariant> >();
}

CostPositionItem *CostPositionItem::parent()
{
    return parentItem;
}

int CostPositionItem::row() const
{
    if(parentItem)
    {
        return parentItem->childItems.indexOf(const_cast<CostPositionItem*>(this));
    }
    return 0;
}

---