394 lines
11 KiB
C++
394 lines
11 KiB
C++
/*
|
|
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 "stream.h"
|
|
#include "streammodel.h"
|
|
#include "portgrouplist.h"
|
|
#include "qicon.h"
|
|
|
|
#include <QMimeData>
|
|
|
|
const QLatin1String kStreamsMimeType("application/vnd.ostinato.streams");
|
|
|
|
StreamModel::StreamModel(PortGroupList *p, QObject *parent)
|
|
: QAbstractTableModel(parent)
|
|
{
|
|
pgl = p;
|
|
mCurrentPort = NULL;
|
|
}
|
|
|
|
int StreamModel::rowCount(const QModelIndex &parent) const
|
|
{
|
|
if (parent.isValid())
|
|
return 0;
|
|
|
|
if (mCurrentPort)
|
|
return mCurrentPort->numStreams();
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
int StreamModel::columnCount(const QModelIndex &/*parent*/) const
|
|
{
|
|
int count = StreamMaxFields;
|
|
if (mCurrentPort &&
|
|
(mCurrentPort->transmitMode() == OstProto::kInterleavedTransmit))
|
|
count--;
|
|
|
|
return count;
|
|
}
|
|
|
|
Qt::ItemFlags StreamModel::flags(const QModelIndex &index) const
|
|
{
|
|
Qt::ItemFlags flags = QAbstractTableModel::flags(index);
|
|
|
|
|
|
switch (index.column())
|
|
{
|
|
case StreamIcon:
|
|
break;
|
|
case StreamName:
|
|
flags |= Qt::ItemIsEditable;
|
|
break;
|
|
case StreamStatus:
|
|
flags |= Qt::ItemIsUserCheckable;
|
|
break;
|
|
case StreamNextWhat:
|
|
flags |= Qt::ItemIsEditable;
|
|
break;
|
|
default:
|
|
//qFatal("Missed case in switch!");
|
|
break;
|
|
}
|
|
|
|
return flags;
|
|
}
|
|
|
|
QVariant StreamModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
// Check for a valid index
|
|
if (!index.isValid())
|
|
return QVariant();
|
|
|
|
// Check for row/column limits
|
|
if (index.row() >= mCurrentPort->numStreams())
|
|
return QVariant();
|
|
|
|
if (index.column() >= StreamMaxFields)
|
|
return QVariant();
|
|
|
|
if (mCurrentPort == NULL)
|
|
return QVariant();
|
|
|
|
// Return data based on field and role
|
|
switch(index.column())
|
|
{
|
|
case StreamIcon:
|
|
{
|
|
if (role == Qt::DecorationRole)
|
|
return QIcon(":/icons/stream_edit.png");
|
|
else
|
|
return QVariant();
|
|
break;
|
|
}
|
|
case StreamName:
|
|
{
|
|
if ((role == Qt::DisplayRole) || (role == Qt::EditRole))
|
|
return mCurrentPort->streamByIndex(index.row())->name();
|
|
else
|
|
return QVariant();
|
|
break;
|
|
}
|
|
case StreamStatus:
|
|
{
|
|
if (role == Qt::CheckStateRole)
|
|
{
|
|
if (mCurrentPort->streamByIndex(index.row())->isEnabled())
|
|
return Qt::Checked;
|
|
else
|
|
return Qt::Unchecked;
|
|
}
|
|
else
|
|
return QVariant();
|
|
break;
|
|
}
|
|
case StreamNextWhat:
|
|
{
|
|
int val = mCurrentPort->streamByIndex(index.row())->nextWhat();
|
|
|
|
if (role == Qt::DisplayRole)
|
|
return nextWhatOptionList().at(val);
|
|
else if (role == Qt::EditRole)
|
|
return val;
|
|
else
|
|
return QVariant();
|
|
|
|
break;
|
|
}
|
|
default:
|
|
qFatal("-------------UNHANDLED STREAM FIELD----------------");
|
|
}
|
|
|
|
return QVariant();
|
|
}
|
|
|
|
bool StreamModel::setData(const QModelIndex &index, const QVariant &value, int role)
|
|
{
|
|
if (mCurrentPort == NULL)
|
|
return false;
|
|
|
|
if (index.isValid())
|
|
{
|
|
switch (index.column())
|
|
{
|
|
// Edit Supported Fields
|
|
case StreamName:
|
|
mCurrentPort->mutableStreamByIndex(index.row())
|
|
->setName(value.toString());
|
|
emit(dataChanged(index, index));
|
|
return true;
|
|
|
|
case StreamStatus:
|
|
mCurrentPort->mutableStreamByIndex(index.row())
|
|
->setEnabled(value.toBool());
|
|
emit(dataChanged(index, index));
|
|
return true;
|
|
|
|
case StreamNextWhat:
|
|
if (role == Qt::EditRole)
|
|
{
|
|
mCurrentPort->mutableStreamByIndex(index.row())->setNextWhat(
|
|
(Stream::NextWhat)value.toInt());
|
|
emit(dataChanged(index, index));
|
|
return true;
|
|
}
|
|
else
|
|
return false;
|
|
|
|
// Edit Not Supported Fields
|
|
case StreamIcon:
|
|
return false;
|
|
|
|
// Unhandled Stream Field
|
|
default:
|
|
qDebug("-------------UNHANDLED STREAM FIELD----------------");
|
|
break;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
QVariant StreamModel::headerData(int section, Qt::Orientation orientation, int role) const
|
|
{
|
|
if (role != Qt::DisplayRole)
|
|
return QVariant();
|
|
|
|
if (orientation == Qt::Horizontal)
|
|
{
|
|
switch(section)
|
|
{
|
|
case StreamIcon:
|
|
return QString("");
|
|
break;
|
|
case StreamName:
|
|
return QString("Name");
|
|
break;
|
|
case StreamStatus:
|
|
return QString("");
|
|
break;
|
|
case StreamNextWhat:
|
|
return QString("Goto");
|
|
break;
|
|
default:
|
|
qDebug("-------------UNHANDLED STREAM FIELD----------------");
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
return QString("%1").arg(section+1);
|
|
|
|
return QVariant();
|
|
}
|
|
|
|
QStringList StreamModel::mimeTypes() const
|
|
{
|
|
return QStringList() << kStreamsMimeType;
|
|
}
|
|
|
|
QMimeData* StreamModel::mimeData(const QModelIndexList &indexes) const
|
|
{
|
|
using ::google::protobuf::uint8;
|
|
|
|
if (indexes.isEmpty())
|
|
return nullptr;
|
|
|
|
// indexes may include multiple columns for a row - but we are only
|
|
// interested in rows 'coz we have a single data for all columns
|
|
// XXX: use QMap instead of QSet to keep rows in sorted order
|
|
QMap<int, int> rows;
|
|
foreach(QModelIndex index, indexes)
|
|
rows.insert(index.row(), index.row());
|
|
|
|
OstProto::StreamConfigList streams;
|
|
streams.mutable_port_id()->set_id(mCurrentPort->id());
|
|
foreach(int row, rows) {
|
|
OstProto::Stream *stream = streams.add_stream();
|
|
mCurrentPort->streamByIndex(row)->protoDataCopyInto(*stream);
|
|
}
|
|
|
|
QByteArray data;
|
|
data.resize(streams.ByteSize());
|
|
streams.SerializeWithCachedSizesToArray((uint8*)data.data());
|
|
//qDebug("copy %s", streams.DebugString().c_str());
|
|
//TODO: copy DebugString as text/plain?
|
|
|
|
QMimeData *mimeData = new QMimeData();
|
|
mimeData->setData(kStreamsMimeType, data);
|
|
return mimeData; // XXX: caller is expected to take ownership and free!
|
|
}
|
|
|
|
bool StreamModel::dropMimeData(const QMimeData *data, Qt::DropAction action,
|
|
int row, int /*column*/, const QModelIndex &parent)
|
|
{
|
|
if (!data)
|
|
return false;
|
|
|
|
if (!data->hasFormat(kStreamsMimeType))
|
|
return false;
|
|
|
|
if (action != Qt::CopyAction)
|
|
return false;
|
|
|
|
OstProto::StreamConfigList streamsData;
|
|
QByteArray ba(data->data(kStreamsMimeType));
|
|
streamsData.ParseFromArray((void*)ba.constData(), ba.size());
|
|
//qDebug("paste %s", streamsData.DebugString().c_str());
|
|
|
|
QList<Stream*> streams;
|
|
for (int i = 0; i < streamsData.stream_size(); i++) {
|
|
Stream *stream = new Stream;
|
|
stream->protoDataCopyFrom(streamsData.stream(i));
|
|
streams.append(stream);
|
|
}
|
|
|
|
if ((row < 0) || (row > rowCount(parent)))
|
|
row = rowCount(parent);
|
|
|
|
// Delete rows that we are going to overwrite
|
|
if (row < rowCount(parent))
|
|
removeRows(row, qMin(rowCount() - row, streams.size()));
|
|
|
|
return insert(row, streams); // callee will free streams after insert
|
|
}
|
|
|
|
/*!
|
|
* Inserts streams before the given row
|
|
*
|
|
* StreamModel takes ownership of the passed streams; caller should
|
|
* not try to access them after calling this function
|
|
*/
|
|
bool StreamModel::insert(int row, QList<Stream*> &streams)
|
|
{
|
|
int count = streams.size();
|
|
qDebug("insert row = %d", row);
|
|
qDebug("insert count = %d", count);
|
|
beginInsertRows(QModelIndex(), row, row+count-1);
|
|
for (int i = 0; i < count; i++) {
|
|
OstProto::Stream s;
|
|
streams.at(i)->protoDataCopyInto(s);
|
|
mCurrentPort->newStreamAt(row+i, &s);
|
|
delete streams.at(i);
|
|
}
|
|
streams.clear();
|
|
endInsertRows();
|
|
|
|
return true;
|
|
}
|
|
|
|
bool StreamModel::insertRows(int row, int count, const QModelIndex &/*parent*/)
|
|
{
|
|
qDebug("insertRows() row = %d", row);
|
|
qDebug("insertRows() count = %d", count);
|
|
beginInsertRows(QModelIndex(), row, row+count-1);
|
|
for (int i = 0; i < count; i++)
|
|
mCurrentPort->newStreamAt(row);
|
|
endInsertRows();
|
|
|
|
return true;
|
|
}
|
|
|
|
bool StreamModel::removeRows(int row, int count, const QModelIndex &/*parent*/)
|
|
{
|
|
qDebug("removeRows() row = %d", row);
|
|
qDebug("removeRows() count = %d", count);
|
|
beginRemoveRows(QModelIndex(), row, row+count-1);
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
mCurrentPort->deleteStreamAt(row);
|
|
}
|
|
endRemoveRows();
|
|
|
|
return true;
|
|
}
|
|
|
|
// --------------------- SLOTS ------------------------
|
|
|
|
void StreamModel::setCurrentPortIndex(const QModelIndex ¤t)
|
|
{
|
|
beginResetModel();
|
|
if (!current.isValid() || !pgl->isPort(current))
|
|
{
|
|
qDebug("current is either invalid or not a port");
|
|
mCurrentPort = NULL;
|
|
}
|
|
else
|
|
{
|
|
qDebug("change to valid port");
|
|
// Disconnect any existing connection to avoid duplication
|
|
// Qt 4.6 has Qt::UniqueConnection, but we want to remain compatible
|
|
// with earlier Qt versions
|
|
if (mCurrentPort)
|
|
{
|
|
disconnect(mCurrentPort, SIGNAL(streamListChanged(int, int)),
|
|
this, SLOT(when_mCurrentPort_streamListChanged(int, int)));
|
|
}
|
|
quint16 pg = current.internalId() >> 16;
|
|
mCurrentPort = pgl->mPortGroups[pgl->indexOfPortGroup(pg)]->mPorts[current.row()];
|
|
connect(mCurrentPort, SIGNAL(streamListChanged(int, int)),
|
|
this, SLOT(when_mCurrentPort_streamListChanged(int, int)));
|
|
}
|
|
endResetModel();
|
|
}
|
|
|
|
void StreamModel::when_mCurrentPort_streamListChanged(int portGroupId,
|
|
int portId)
|
|
{
|
|
qDebug("In %s", __FUNCTION__);
|
|
if (mCurrentPort)
|
|
{
|
|
if ((quint32(portGroupId) == mCurrentPort->portGroupId())
|
|
&& (quint32(portId) == mCurrentPort->id()))
|
|
{
|
|
beginResetModel();
|
|
endResetModel();
|
|
}
|
|
}
|
|
}
|