CatalogueEventsTableModel.cpp
185 lines
| 5.6 KiB
| text/x-c
|
CppLexer
r1130 | #include "Catalogue/CatalogueEventsTableModel.h" | |||
#include <Common/DateUtils.h> | ||||
r1133 | #include <Common/MimeTypesDef.h> | |||
r1130 | #include <DBEvent.h> | |||
#include <DBTag.h> | ||||
r1133 | #include <Data/SqpRange.h> | |||
#include <QMimeData> | ||||
#include <SqpApplication.h> | ||||
#include <Time/TimeController.h> | ||||
r1130 | ||||
struct CatalogueEventsTableModel::CatalogueEventsTableModelPrivate { | ||||
r1143 | QVector<std::shared_ptr<DBEvent> > m_Events; | |||
r1130 | ||||
enum class Column { Event, TStart, TEnd, Tags, Product, NbColumn }; | ||||
QStringList columnNames() | ||||
{ | ||||
return QStringList{tr("Event"), tr("TStart"), tr("TEnd"), tr("Tags"), tr("Product")}; | ||||
} | ||||
r1143 | QVariant eventData(int col, const std::shared_ptr<DBEvent> &event) const | |||
r1130 | { | |||
switch (static_cast<Column>(col)) { | ||||
case Column::Event: | ||||
r1143 | return event->getName(); | |||
r1130 | case Column::TStart: | |||
r1143 | return DateUtils::dateTime(event->getTStart()); | |||
r1130 | case Column::TEnd: | |||
r1143 | return DateUtils::dateTime(event->getTEnd()); | |||
r1130 | case Column::Product: | |||
r1143 | return event->getProduct(); | |||
r1130 | case Column::Tags: { | |||
QString tagList; | ||||
r1143 | auto tags = event->getTags(); | |||
r1130 | for (auto tag : tags) { | |||
tagList += tag.getName(); | ||||
tagList += ' '; | ||||
} | ||||
return tagList; | ||||
} | ||||
default: | ||||
break; | ||||
} | ||||
Q_ASSERT(false); | ||||
return QStringLiteral("Unknown Data"); | ||||
} | ||||
}; | ||||
CatalogueEventsTableModel::CatalogueEventsTableModel(QObject *parent) | ||||
: QAbstractTableModel(parent), | ||||
impl{spimpl::make_unique_impl<CatalogueEventsTableModelPrivate>()} | ||||
{ | ||||
} | ||||
r1143 | void CatalogueEventsTableModel::setEvents(const QVector<std::shared_ptr<DBEvent> > &events) | |||
r1130 | { | |||
beginResetModel(); | ||||
impl->m_Events = events; | ||||
endResetModel(); | ||||
} | ||||
r1143 | std::shared_ptr<DBEvent> CatalogueEventsTableModel::getEvent(int row) const | |||
r1130 | { | |||
return impl->m_Events.value(row); | ||||
} | ||||
r1143 | void CatalogueEventsTableModel::addEvent(const std::shared_ptr<DBEvent> &events) | |||
r1134 | { | |||
beginInsertRows(QModelIndex(), impl->m_Events.count() - 1, impl->m_Events.count() - 1); | ||||
// impl->m_Events.append(event); TODO | ||||
endInsertRows(); | ||||
} | ||||
r1143 | void CatalogueEventsTableModel::removeEvent(const std::shared_ptr<DBEvent> &events) | |||
r1134 | { | |||
// TODO | ||||
auto index = -1; // impl->m_Events.indexOf(event); | ||||
if (index >= 0) { | ||||
beginRemoveRows(QModelIndex(), index, index); | ||||
impl->m_Events.removeAt(index); | ||||
endRemoveRows(); | ||||
} | ||||
} | ||||
r1130 | int CatalogueEventsTableModel::rowCount(const QModelIndex &parent) const | |||
{ | ||||
int r = impl->m_Events.count(); | ||||
return r; | ||||
} | ||||
int CatalogueEventsTableModel::columnCount(const QModelIndex &parent) const | ||||
{ | ||||
int c = static_cast<int>(CatalogueEventsTableModelPrivate::Column::NbColumn); | ||||
return c; | ||||
} | ||||
Qt::ItemFlags CatalogueEventsTableModel::flags(const QModelIndex &index) const | ||||
{ | ||||
r1133 | return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled; | |||
r1130 | } | |||
QVariant CatalogueEventsTableModel::data(const QModelIndex &index, int role) const | ||||
{ | ||||
if (index.isValid()) { | ||||
r1133 | auto event = getEvent(index.row()); | |||
r1130 | ||||
switch (role) { | ||||
case Qt::DisplayRole: | ||||
return impl->eventData(index.column(), event); | ||||
break; | ||||
} | ||||
} | ||||
return QVariant{}; | ||||
} | ||||
QVariant CatalogueEventsTableModel::headerData(int section, Qt::Orientation orientation, | ||||
int role) const | ||||
{ | ||||
if (orientation == Qt::Horizontal && role == Qt::DisplayRole) { | ||||
return impl->columnNames().value(section); | ||||
} | ||||
return QVariant(); | ||||
} | ||||
void CatalogueEventsTableModel::sort(int column, Qt::SortOrder order) | ||||
{ | ||||
std::sort(impl->m_Events.begin(), impl->m_Events.end(), | ||||
[this, column, order](auto e1, auto e2) { | ||||
auto data1 = impl->eventData(column, e1); | ||||
auto data2 = impl->eventData(column, e2); | ||||
auto result = data1.toString() < data2.toString(); | ||||
return order == Qt::AscendingOrder ? result : !result; | ||||
}); | ||||
emit dataChanged(QModelIndex(), QModelIndex()); | ||||
} | ||||
r1133 | ||||
Qt::DropActions CatalogueEventsTableModel::supportedDragActions() const | ||||
{ | ||||
return Qt::CopyAction | Qt::MoveAction; | ||||
} | ||||
QStringList CatalogueEventsTableModel::mimeTypes() const | ||||
{ | ||||
return {MIME_TYPE_EVENT_LIST, MIME_TYPE_TIME_RANGE}; | ||||
} | ||||
QMimeData *CatalogueEventsTableModel::mimeData(const QModelIndexList &indexes) const | ||||
{ | ||||
auto mimeData = new QMimeData; | ||||
r1143 | QVector<std::shared_ptr<DBEvent> > eventList; | |||
r1133 | ||||
SqpRange firstTimeRange; | ||||
for (const auto &index : indexes) { | ||||
if (index.column() == 0) { // only the first column | ||||
auto event = getEvent(index.row()); | ||||
if (eventList.isEmpty()) { | ||||
// Gets the range of the first variable | ||||
r1143 | firstTimeRange.m_TStart = event->getTStart(); | |||
firstTimeRange.m_TEnd = event->getTEnd(); | ||||
r1133 | } | |||
eventList << event; | ||||
} | ||||
} | ||||
auto eventsEncodedData | ||||
= QByteArray{}; // sqpApp->catalogueController().->mimeDataForEvents(eventList); //TODO | ||||
mimeData->setData(MIME_TYPE_EVENT_LIST, eventsEncodedData); | ||||
if (eventList.count() == 1) { | ||||
// No time range MIME data if multiple events are dragged | ||||
auto timeEncodedData = TimeController::mimeDataForTimeRange(firstTimeRange); | ||||
mimeData->setData(MIME_TYPE_TIME_RANGE, timeEncodedData); | ||||
} | ||||
return mimeData; | ||||
} | ||||