ostinato/client/packetmodel.cpp

/*
Copyright (C) 2010 Srivats P.

This file is part of "Ostinato"

This 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 3 of the License, or
(at your option) any later version.

This program 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 this program.  If not, see <http://www.gnu.org/licenses/>
*/

#include <QHostAddress>

#include "packetmodel.h"
#include "../common/protocollistiterator.h"
#include "../common/abstractprotocol.h"

PacketModel::PacketModel(QObject *parent)
    : QAbstractItemModel(parent)
{
}

void PacketModel::setSelectedProtocols(ProtocolListIterator &iter)
{
    QList<const AbstractProtocol*> currentProtocols;

    iter.toFront();
    while (iter.hasNext())
        currentProtocols.append(iter.next());

    if (mSelectedProtocols != currentProtocols)
    {
        mSelectedProtocols = currentProtocols;
        reset();
    }
    else
    {
        emit layoutAboutToBeChanged();
        emit layoutChanged();
    }
}

int PacketModel::rowCount(const QModelIndex &parent) const
{
    IndexId        parentId;

    // qDebug("in %s", __FUNCTION__);

    // Parent == Invalid i.e. Invisible Root.
    // ==> Children are Protocol (Top Level) Items
    if (!parent.isValid())
        return mSelectedProtocols.size();

    // Parent - Valid Item
    parentId.w = parent.internalId();
    switch(parentId.ws.type)
    {
    case  ITYP_PROTOCOL:
        return mSelectedProtocols.at(parentId.ws.protocol)->frameFieldCount();
    case  ITYP_FIELD: 
        return 0;
    default:
        qWarning("%s: Unhandled ItemType", __FUNCTION__);
    }

    Q_ASSERT(1 == 0);    // Unreachable code
    qWarning("%s: Catch all - need to investigate", __FUNCTION__);
    return 0; // catch all
}

int PacketModel::columnCount(const QModelIndex &/*parent*/) const
{
    return 1;
}

QModelIndex PacketModel::index(int row, int col, const QModelIndex &parent) const
{
    QModelIndex    index;
    IndexId    id, parentId;

    if (!hasIndex(row, col, parent))
        goto _exit;

    // Parent is Invisible Root
    // Request for a Protocol Item
    if (!parent.isValid())
    {
        id.w = 0;
        id.ws.type = ITYP_PROTOCOL;
        id.ws.protocol = row;
        index = createIndex(row, col, id.w);
        goto _exit;
    }

    // Parent is a Valid Item
    parentId.w = parent.internalId();
    id.w = parentId.w;
    switch(parentId.ws.type)
    {
    case  ITYP_PROTOCOL:
        id.ws.type = ITYP_FIELD;
        index = createIndex(row, col, id.w);
        goto _exit;

    case  ITYP_FIELD: 
        Q_ASSERT(1 == 0);    // Unreachable code
        goto _exit;

    default:
        qWarning("%s: Unhandled ItemType", __FUNCTION__);
    }

    Q_ASSERT(1 == 0);    // Unreachable code

_exit:
    return index;
}

QModelIndex PacketModel::parent(const QModelIndex &index) const
{
    QModelIndex    parentIndex;
    IndexId        id, parentId;

    if (!index.isValid())
        return QModelIndex();

    id.w = index.internalId();
    parentId.w = id.w;
    switch(id.ws.type)
    {
    case  ITYP_PROTOCOL:
        // return invalid index for invisible root
        goto _exit;

    case  ITYP_FIELD: 
        parentId.ws.type = ITYP_PROTOCOL;
        parentIndex = createIndex(id.ws.protocol, 0, parentId.w);
        goto _exit;

    default:
        qWarning("%s: Unhandled ItemType", __FUNCTION__);
    }

    Q_ASSERT(1 == 1); // Unreachable code

_exit:
    return parentIndex;
}

QVariant PacketModel::data(const QModelIndex &index, int role) const
{
    IndexId        id;
    int            fieldIdx = 0;    

    if (!index.isValid())
        return QVariant();

    id.w = index.internalId();

    if (id.ws.type == ITYP_FIELD)
    {
        const AbstractProtocol *p = mSelectedProtocols.at(id.ws.protocol);
        int n = index.row() + 1;

        while (n)
        {
            if (p->fieldFlags(fieldIdx).testFlag(AbstractProtocol::FrameField))
                n--;
            fieldIdx++;
        }
        fieldIdx--;
    }

    // FIXME(HI): Relook at this completely
    if (role == Qt::UserRole)
    {
        switch(id.ws.type)
        {
        case  ITYP_PROTOCOL:
            qDebug("*** %d/%d", id.ws.protocol, mSelectedProtocols.size());
            return    mSelectedProtocols.at(id.ws.protocol)->
                protocolFrameValue();

        case  ITYP_FIELD: 
            return    mSelectedProtocols.at(id.ws.protocol)->fieldData(
                fieldIdx, AbstractProtocol::FieldFrameValue);

        default:
            qWarning("%s: Unhandled ItemType", __FUNCTION__);
        }
        return QByteArray(); 
    }

    // FIXME: Use a new enum here instead of UserRole
    if (role == (Qt::UserRole+1))
    {
        switch(id.ws.type)
        {
        case  ITYP_PROTOCOL:
            return    mSelectedProtocols.at(id.ws.protocol)->
                protocolFrameValue().size();

        case  ITYP_FIELD: 
            return    mSelectedProtocols.at(id.ws.protocol)->fieldData(
                fieldIdx, AbstractProtocol::FieldBitSize);

        default:
            qWarning("%s: Unhandled ItemType", __FUNCTION__);
        }
        return QVariant(); 
    }

    if (role != Qt::DisplayRole)
        return QVariant();

    switch(id.ws.type)
    {
    case  ITYP_PROTOCOL:
        return QString("%1 (%2)")
            .arg(mSelectedProtocols.at(id.ws.protocol)->shortName())
            .arg(mSelectedProtocols.at(id.ws.protocol)->name());

    case  ITYP_FIELD: 
        return    mSelectedProtocols.at(id.ws.protocol)->fieldData(fieldIdx,
                AbstractProtocol::FieldName).toString() + QString(" : ") +
                mSelectedProtocols.at(id.ws.protocol)->fieldData(fieldIdx,
                AbstractProtocol::FieldTextValue).toString();

    default:
        qWarning("%s: Unhandled ItemType", __FUNCTION__);
    }

    Q_ASSERT(1 == 1); // Unreachable code

    return QVariant();
}