CatalogueEventsModel.cpp
442 lines
| 13.8 KiB
| text/x-c
|
CppLexer
r1182 | #include "Catalogue/CatalogueEventsModel.h" | |||
#include <Common/DateUtils.h> | ||||
#include <Common/MimeTypesDef.h> | ||||
#include <DBEvent.h> | ||||
#include <DBEventProduct.h> | ||||
#include <DBTag.h> | ||||
#include <Data/SqpRange.h> | ||||
#include <SqpApplication.h> | ||||
#include <Time/TimeController.h> | ||||
#include <list> | ||||
#include <unordered_map> | ||||
#include <QHash> | ||||
#include <QMimeData> | ||||
r1194 | Q_LOGGING_CATEGORY(LOG_CatalogueEventsModel, "CatalogueEventsModel") | |||
r1182 | const auto EVENT_ITEM_TYPE = 1; | |||
const auto EVENT_PRODUCT_ITEM_TYPE = 2; | ||||
r1183 | struct CatalogueEventsModel::CatalogueEventsModelPrivate { | |||
r1182 | QVector<std::shared_ptr<DBEvent> > m_Events; | |||
std::unordered_map<DBEvent *, QVector<std::shared_ptr<DBEventProduct> > > m_EventProducts; | ||||
r1194 | std::unordered_set<std::shared_ptr<DBEvent> > m_EventsWithChanges; | |||
r1182 | ||||
QStringList columnNames() | ||||
{ | ||||
r1194 | return QStringList{tr("Event"), tr("TStart"), tr("TEnd"), | |||
tr("Tags"), tr("Product"), tr("")}; | ||||
} | ||||
QVariant sortData(int col, const std::shared_ptr<DBEvent> &event) const | ||||
{ | ||||
if (col == (int)CatalogueEventsModel::Column::Validation) { | ||||
return m_EventsWithChanges.find(event) != m_EventsWithChanges.cend() ? true | ||||
: QVariant(); | ||||
} | ||||
return eventData(col, event); | ||||
r1182 | } | |||
QVariant eventData(int col, const std::shared_ptr<DBEvent> &event) const | ||||
{ | ||||
switch (static_cast<Column>(col)) { | ||||
r1194 | case CatalogueEventsModel::Column::Name: | |||
r1182 | return event->getName(); | |||
r1194 | case CatalogueEventsModel::Column::TStart: | |||
r1193 | return nbEventProducts(event) > 0 ? DateUtils::dateTime(event->getTStart()) | |||
: QVariant{}; | ||||
r1194 | case CatalogueEventsModel::Column::TEnd: | |||
r1193 | return nbEventProducts(event) > 0 ? DateUtils::dateTime(event->getTEnd()) | |||
: QVariant{}; | ||||
r1194 | case CatalogueEventsModel::Column::Product: | |||
r1193 | return QString::number(nbEventProducts(event)) + " product(s)"; | |||
r1194 | case CatalogueEventsModel::Column::Tags: { | |||
r1182 | QString tagList; | |||
auto tags = event->getTags(); | ||||
for (auto tag : tags) { | ||||
tagList += tag.getName(); | ||||
tagList += ' '; | ||||
} | ||||
return tagList; | ||||
} | ||||
r1194 | case CatalogueEventsModel::Column::Validation: | |||
return QVariant(); | ||||
r1182 | default: | |||
break; | ||||
} | ||||
Q_ASSERT(false); | ||||
return QStringLiteral("Unknown Data"); | ||||
} | ||||
void parseEventProduct(const std::shared_ptr<DBEvent> &event) | ||||
{ | ||||
for (auto product : event->getEventProducts()) { | ||||
m_EventProducts[event.get()].append(std::make_shared<DBEventProduct>(product)); | ||||
} | ||||
} | ||||
r1193 | int nbEventProducts(const std::shared_ptr<DBEvent> &event) const | |||
{ | ||||
auto eventProductsIt = m_EventProducts.find(event.get()); | ||||
if (eventProductsIt != m_EventProducts.cend()) { | ||||
return m_EventProducts.at(event.get()).count(); | ||||
} | ||||
else { | ||||
return 0; | ||||
} | ||||
} | ||||
r1182 | QVariant eventProductData(int col, const std::shared_ptr<DBEventProduct> &eventProduct) const | |||
{ | ||||
switch (static_cast<Column>(col)) { | ||||
r1194 | case CatalogueEventsModel::Column::Name: | |||
r1182 | return eventProduct->getProductId(); | |||
r1194 | case CatalogueEventsModel::Column::TStart: | |||
r1182 | return DateUtils::dateTime(eventProduct->getTStart()); | |||
r1194 | case CatalogueEventsModel::Column::TEnd: | |||
r1182 | return DateUtils::dateTime(eventProduct->getTEnd()); | |||
r1194 | case CatalogueEventsModel::Column::Product: | |||
r1182 | return eventProduct->getProductId(); | |||
r1194 | case CatalogueEventsModel::Column::Tags: | |||
r1182 | return QString(); | |||
r1194 | case CatalogueEventsModel::Column::Validation: | |||
return QVariant(); | ||||
r1182 | default: | |||
break; | ||||
} | ||||
Q_ASSERT(false); | ||||
return QStringLiteral("Unknown Data"); | ||||
} | ||||
}; | ||||
CatalogueEventsModel::CatalogueEventsModel(QObject *parent) | ||||
r1183 | : QAbstractItemModel(parent), impl{spimpl::make_unique_impl<CatalogueEventsModelPrivate>()} | |||
r1182 | { | |||
} | ||||
void CatalogueEventsModel::setEvents(const QVector<std::shared_ptr<DBEvent> > &events) | ||||
{ | ||||
beginResetModel(); | ||||
impl->m_Events = events; | ||||
r1183 | impl->m_EventProducts.clear(); | |||
r1194 | impl->m_EventsWithChanges.clear(); | |||
r1182 | for (auto event : events) { | |||
impl->parseEventProduct(event); | ||||
} | ||||
endResetModel(); | ||||
} | ||||
r1183 | std::shared_ptr<DBEvent> CatalogueEventsModel::getEvent(const QModelIndex &index) const | |||
{ | ||||
if (itemTypeOf(index) == CatalogueEventsModel::ItemType::Event) { | ||||
return impl->m_Events.value(index.row()); | ||||
} | ||||
else { | ||||
return nullptr; | ||||
} | ||||
} | ||||
std::shared_ptr<DBEvent> CatalogueEventsModel::getParentEvent(const QModelIndex &index) const | ||||
r1182 | { | |||
r1183 | if (itemTypeOf(index) == CatalogueEventsModel::ItemType::EventProduct) { | |||
return getEvent(index.parent()); | ||||
} | ||||
else { | ||||
return nullptr; | ||||
} | ||||
} | ||||
std::shared_ptr<DBEventProduct> | ||||
CatalogueEventsModel::getEventProduct(const QModelIndex &index) const | ||||
{ | ||||
if (itemTypeOf(index) == CatalogueEventsModel::ItemType::EventProduct) { | ||||
auto event = static_cast<DBEvent *>(index.internalPointer()); | ||||
return impl->m_EventProducts.at(event).value(index.row()); | ||||
} | ||||
else { | ||||
return nullptr; | ||||
} | ||||
r1182 | } | |||
void CatalogueEventsModel::addEvent(const std::shared_ptr<DBEvent> &event) | ||||
{ | ||||
beginInsertRows(QModelIndex(), impl->m_Events.count() - 1, impl->m_Events.count() - 1); | ||||
impl->m_Events.append(event); | ||||
impl->parseEventProduct(event); | ||||
endInsertRows(); | ||||
} | ||||
void CatalogueEventsModel::removeEvent(const std::shared_ptr<DBEvent> &event) | ||||
{ | ||||
auto index = impl->m_Events.indexOf(event); | ||||
if (index >= 0) { | ||||
beginRemoveRows(QModelIndex(), index, index); | ||||
impl->m_Events.removeAt(index); | ||||
impl->m_EventProducts.erase(event.get()); | ||||
r1194 | impl->m_EventsWithChanges.erase(event); | |||
r1182 | endRemoveRows(); | |||
} | ||||
} | ||||
r1194 | QVector<std::shared_ptr<DBEvent> > CatalogueEventsModel::events() const | |||
{ | ||||
return impl->m_Events; | ||||
} | ||||
r1182 | void CatalogueEventsModel::refreshEvent(const std::shared_ptr<DBEvent> &event) | |||
{ | ||||
r1194 | auto eventIndex = indexOf(event); | |||
if (eventIndex.isValid()) { | ||||
// Refreshes the event line | ||||
r1183 | auto colCount = columnCount(); | |||
r1194 | emit dataChanged(eventIndex, index(eventIndex.row(), colCount)); | |||
r1183 | ||||
r1194 | // Also refreshes its children event products | |||
r1183 | auto childCount = rowCount(eventIndex); | |||
emit dataChanged(index(0, 0, eventIndex), index(childCount, colCount, eventIndex)); | ||||
r1182 | } | |||
r1194 | else { | |||
qCWarning(LOG_CatalogueEventsModel()) << "refreshEvent: event not found."; | ||||
} | ||||
} | ||||
QModelIndex CatalogueEventsModel::indexOf(const std::shared_ptr<DBEvent> &event) const | ||||
{ | ||||
auto row = impl->m_Events.indexOf(event); | ||||
if (row >= 0) { | ||||
return index(row, 0); | ||||
} | ||||
return QModelIndex(); | ||||
} | ||||
void CatalogueEventsModel::setEventHasChanges(const std::shared_ptr<DBEvent> &event, | ||||
bool hasChanges) | ||||
{ | ||||
if (hasChanges) { | ||||
impl->m_EventsWithChanges.insert(event); | ||||
} | ||||
else { | ||||
impl->m_EventsWithChanges.erase(event); | ||||
} | ||||
} | ||||
bool CatalogueEventsModel::eventsHasChanges(const std::shared_ptr<DBEvent> &event) const | ||||
{ | ||||
return impl->m_EventsWithChanges.find(event) != impl->m_EventsWithChanges.cend(); | ||||
r1182 | } | |||
QModelIndex CatalogueEventsModel::index(int row, int column, const QModelIndex &parent) const | ||||
{ | ||||
if (!hasIndex(row, column, parent)) { | ||||
return QModelIndex(); | ||||
} | ||||
r1183 | switch (itemTypeOf(parent)) { | |||
case CatalogueEventsModel::ItemType::Root: | ||||
r1182 | return createIndex(row, column); | |||
r1183 | case CatalogueEventsModel::ItemType::Event: { | |||
auto event = getEvent(parent); | ||||
r1182 | return createIndex(row, column, event.get()); | |||
} | ||||
r1183 | case CatalogueEventsModel::ItemType::EventProduct: | |||
r1182 | break; | |||
default: | ||||
break; | ||||
} | ||||
return QModelIndex(); | ||||
} | ||||
QModelIndex CatalogueEventsModel::parent(const QModelIndex &index) const | ||||
{ | ||||
r1183 | switch (itemTypeOf(index)) { | |||
case CatalogueEventsModel::ItemType::EventProduct: { | ||||
r1182 | auto parentEvent = static_cast<DBEvent *>(index.internalPointer()); | |||
auto it | ||||
= std::find_if(impl->m_Events.cbegin(), impl->m_Events.cend(), | ||||
[parentEvent](auto event) { return event.get() == parentEvent; }); | ||||
if (it != impl->m_Events.cend()) { | ||||
return createIndex(it - impl->m_Events.cbegin(), 0); | ||||
} | ||||
else { | ||||
return QModelIndex(); | ||||
} | ||||
} | ||||
r1183 | case CatalogueEventsModel::ItemType::Root: | |||
r1182 | break; | |||
r1183 | case CatalogueEventsModel::ItemType::Event: | |||
r1182 | break; | |||
default: | ||||
break; | ||||
} | ||||
return QModelIndex(); | ||||
} | ||||
int CatalogueEventsModel::rowCount(const QModelIndex &parent) const | ||||
{ | ||||
if (parent.column() > 0) { | ||||
return 0; | ||||
} | ||||
r1183 | switch (itemTypeOf(parent)) { | |||
case CatalogueEventsModel::ItemType::Root: | ||||
r1182 | return impl->m_Events.count(); | |||
r1183 | case CatalogueEventsModel::ItemType::Event: { | |||
auto event = getEvent(parent); | ||||
r1182 | return impl->m_EventProducts[event.get()].count(); | |||
} | ||||
r1183 | case CatalogueEventsModel::ItemType::EventProduct: | |||
r1182 | break; | |||
default: | ||||
break; | ||||
} | ||||
return 0; | ||||
} | ||||
int CatalogueEventsModel::columnCount(const QModelIndex &parent) const | ||||
{ | ||||
r1194 | return static_cast<int>(CatalogueEventsModel::Column::NbColumn); | |||
r1182 | } | |||
Qt::ItemFlags CatalogueEventsModel::flags(const QModelIndex &index) const | ||||
{ | ||||
return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsDragEnabled; | ||||
} | ||||
QVariant CatalogueEventsModel::data(const QModelIndex &index, int role) const | ||||
{ | ||||
if (index.isValid()) { | ||||
r1183 | auto type = itemTypeOf(index); | |||
if (type == CatalogueEventsModel::ItemType::Event) { | ||||
auto event = getEvent(index); | ||||
r1182 | switch (role) { | |||
case Qt::DisplayRole: | ||||
return impl->eventData(index.column(), event); | ||||
break; | ||||
} | ||||
} | ||||
r1183 | else if (type == CatalogueEventsModel::ItemType::EventProduct) { | |||
auto product = getEventProduct(index); | ||||
r1182 | switch (role) { | |||
case Qt::DisplayRole: | ||||
return impl->eventProductData(index.column(), product); | ||||
break; | ||||
} | ||||
} | ||||
} | ||||
return QVariant{}; | ||||
} | ||||
QVariant CatalogueEventsModel::headerData(int section, Qt::Orientation orientation, int role) const | ||||
{ | ||||
if (orientation == Qt::Horizontal && role == Qt::DisplayRole) { | ||||
return impl->columnNames().value(section); | ||||
} | ||||
return QVariant(); | ||||
} | ||||
void CatalogueEventsModel::sort(int column, Qt::SortOrder order) | ||||
{ | ||||
std::sort(impl->m_Events.begin(), impl->m_Events.end(), | ||||
[this, column, order](auto e1, auto e2) { | ||||
r1194 | auto data1 = impl->sortData(column, e1); | |||
auto data2 = impl->sortData(column, e2); | ||||
r1182 | ||||
auto result = data1.toString() < data2.toString(); | ||||
return order == Qt::AscendingOrder ? result : !result; | ||||
}); | ||||
emit dataChanged(QModelIndex(), QModelIndex()); | ||||
r1194 | emit modelSorted(); | |||
r1182 | } | |||
Qt::DropActions CatalogueEventsModel::supportedDragActions() const | ||||
{ | ||||
return Qt::CopyAction | Qt::MoveAction; | ||||
} | ||||
QStringList CatalogueEventsModel::mimeTypes() const | ||||
{ | ||||
return {MIME_TYPE_EVENT_LIST, MIME_TYPE_TIME_RANGE}; | ||||
} | ||||
QMimeData *CatalogueEventsModel::mimeData(const QModelIndexList &indexes) const | ||||
{ | ||||
auto mimeData = new QMimeData; | ||||
r1184 | bool isFirst = true; | |||
r1182 | QVector<std::shared_ptr<DBEvent> > eventList; | |||
r1184 | QVector<std::shared_ptr<DBEventProduct> > eventProductList; | |||
r1182 | ||||
SqpRange firstTimeRange; | ||||
for (const auto &index : indexes) { | ||||
if (index.column() == 0) { // only the first column | ||||
r1184 | auto type = itemTypeOf(index); | |||
if (type == ItemType::Event) { | ||||
auto event = getEvent(index); | ||||
eventList << event; | ||||
if (isFirst) { | ||||
isFirst = false; | ||||
r1185 | firstTimeRange.m_TStart = event->getTStart(); | |||
firstTimeRange.m_TEnd = event->getTEnd(); | ||||
r1184 | } | |||
} | ||||
else if (type == ItemType::EventProduct) { | ||||
auto product = getEventProduct(index); | ||||
eventProductList << product; | ||||
if (isFirst) { | ||||
isFirst = false; | ||||
firstTimeRange.m_TStart = product->getTStart(); | ||||
firstTimeRange.m_TEnd = product->getTEnd(); | ||||
} | ||||
} | ||||
r1182 | } | |||
} | ||||
auto eventsEncodedData | ||||
= QByteArray{}; // sqpApp->catalogueController().->mimeDataForEvents(eventList); //TODO | ||||
mimeData->setData(MIME_TYPE_EVENT_LIST, eventsEncodedData); | ||||
r1184 | if (eventList.count() + eventProductList.count() == 1) { | |||
r1182 | // No time range MIME data if multiple events are dragged | |||
auto timeEncodedData = TimeController::mimeDataForTimeRange(firstTimeRange); | ||||
mimeData->setData(MIME_TYPE_TIME_RANGE, timeEncodedData); | ||||
} | ||||
return mimeData; | ||||
} | ||||
r1183 | ||||
CatalogueEventsModel::ItemType CatalogueEventsModel::itemTypeOf(const QModelIndex &index) const | ||||
{ | ||||
if (!index.isValid()) { | ||||
return ItemType::Root; | ||||
} | ||||
else if (index.internalPointer() == nullptr) { | ||||
return ItemType::Event; | ||||
} | ||||
else { | ||||
return ItemType::EventProduct; | ||||
} | ||||
} | ||||