/*
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 "portstatsfilterdialog.h"

PortStatsFilterDialog::PortStatsFilterDialog(QWidget *parent)
    : QDialog(parent)
{
    setupUi(this);

    mUnselected.setSortRole(PositionRole);

    lvUnselected->setModel(&mUnselected);
    lvSelected->setModel(&mSelected);
}

QList<uint> PortStatsFilterDialog::getItemList(bool* ok, 
    QAbstractItemModel *model, Qt::Orientation orientation,
    QList<uint> initial)
{
    QList<uint>    ret;

    uint count = (orientation == Qt::Vertical) ? 
        model->rowCount() : model->columnCount();

    *ok = false;

    mUnselected.clear();
    mSelected.clear();

    for (uint i = 0; i < count; i++)
    {
        QStandardItem    *item;
        
        item = new QStandardItem(model->headerData(i, orientation).toString());
        item->setData(i, PositionRole);
        item->setFlags(Qt::ItemIsSelectable
                | Qt::ItemIsDragEnabled
                //| Qt::ItemIsDropEnabled
                | Qt::ItemIsEnabled);

        if (initial.contains(i))
            mSelected.appendRow(item);
        else
            mUnselected.appendRow(item);
    }

    // No need to sort right now 'coz we have inserted items in order

    if (exec() == QDialog::Accepted)
    {
        uint count = mSelected.rowCount();
        for (uint i = 0; i < count; i++)
        {
            QModelIndex index = mSelected.index(i, 0, QModelIndex());
            QStandardItem *item = mSelected.itemFromIndex(index);
            ret.append(item->data(PositionRole).toInt());
        }
        *ok = true;
    }

    return ret;
}

void PortStatsFilterDialog::on_tbSelectIn_clicked()
{
    QStandardItem    *item;
    while (lvUnselected->selectionModel()->selectedIndexes().size())
    {
        item = mUnselected.takeItem(lvUnselected->selectionModel()->
                selectedIndexes().at(0).row());
        if (mUnselected.removeRow(lvUnselected->selectionModel()->
                selectedIndexes().at(0).row()))
            mSelected.appendRow(item);
    }
}

void PortStatsFilterDialog::on_tbSelectOut_clicked()
{
    QStandardItem    *item;

    while (lvSelected->selectionModel()->selectedIndexes().size())
    {
        item = mSelected.takeItem(lvSelected->selectionModel()->
                selectedIndexes().at(0).row());
        if (mSelected.removeRow(lvSelected->selectionModel()->
                selectedIndexes().at(0).row()))
        {
            mUnselected.appendRow(item);
            mUnselected.sort(0);
        }
    }
}

void PortStatsFilterDialog::on_lvUnselected_doubleClicked(const QModelIndex &index)
{
    QStandardItem    *item;

    item = mUnselected.takeItem(index.row());
    if (mUnselected.removeRow(index.row()))
        mSelected.appendRow(item);
}

void PortStatsFilterDialog::on_lvSelected_doubleClicked(const QModelIndex &index)
{
    QStandardItem    *item;

    item = mSelected.takeItem(index.row());
    if (mSelected.removeRow(index.row()))
    {
        mUnselected.appendRow(item);
        mUnselected.sort(0);
    }
}