bidsviewmodel.cpp

Go to the documentation of this file.

//=============================================================================================================
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "bidsviewmodel.h"
#include <iostream>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QtDebug>
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace DISPLIB;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
BidsViewModel::BidsViewModel(QObject* pParent)
: QStandardItemModel(pParent)
{
}
 
//=============================================================================================================
 
BidsViewModel::~BidsViewModel()
{
}
 
//=============================================================================================================
 
void BidsViewModel::addData(QModelIndex selectedItem,
                            QStandardItem* pNewItem,
                            int iDataType)
{
    switch (iDataType) {
        case BIDS_FUNCTIONALDATA:
        case BIDS_ANATOMICALDATA:
        case BIDS_BEHAVIORALDATA: {
            if (!selectedItem.isValid()) {
                addDataToSession(addSessionToSubject(addSubject("sub-01"), "ses-01"), pNewItem, iDataType);
            } else {
                if (itemFromIndex(selectedItem)->data(BIDS_ITEM_TYPE).value<int>() != BIDS_SUBJECT) {
                    addDataToSession(itemFromIndex(selectedItem)->data(BIDS_ITEM_SESSION).value<QModelIndex>(),
                                     pNewItem,
                                     iDataType);
                } else {
                    qDebug() << "[BidsViewModel::addData] Prompt user to select a session";
                }
            }
            break;
        }
        case BIDS_EVENT:
        case BIDS_AVERAGE: {
            if (!selectedItem.isValid()) {
                QStandardItem* pItem = new QStandardItem("Unknown");
                pItem->setEditable(false);
                pItem->setDragEnabled(true);
 
                addDataToSession(addSessionToSubject(addSubject("sub-01"), "ses-01"), pItem, BIDS_UNKNOWN);
                addToData(pNewItem,
                          indexFromItem(pItem),
                          iDataType);
            } else {
                addToData(pNewItem,
                          selectedItem,
                          iDataType);
            }
            break;
        }
    }
}
 
//=============================================================================================================
 
void BidsViewModel::addToData(QStandardItem* pNewAvgItem,
                              const QModelIndex& parentIndex,
                              int iDataType)
{
    QStandardItem* selectedData = itemFromIndex(parentIndex);
    selectedData->setChild(selectedData->rowCount(),
                           pNewAvgItem);
 
    pNewAvgItem->setData(itemFromIndex(parentIndex)->data(BIDS_ITEM_SUBJECT), BIDS_ITEM_SUBJECT);
    pNewAvgItem->setData(itemFromIndex(parentIndex)->data(BIDS_ITEM_SESSION), BIDS_ITEM_SESSION);
    pNewAvgItem->setData(QVariant::fromValue(iDataType), BIDS_ITEM_TYPE);
 
    emit newItemIndex(pNewAvgItem->index());
}
 
//=============================================================================================================
 
QModelIndex BidsViewModel::addSubject(const QString& sSubjectName)
{
    //Ensure subject name follow BIDS format
    QString sNewSubjectName;
 
    if (!sSubjectName.startsWith("sub-")) {
        sNewSubjectName = "sub-" + sSubjectName;
    } else {
        sNewSubjectName = sSubjectName;
    }
 
    sNewSubjectName.remove(" ");
 
    //Add subject to model
    QStandardItem* pSubjectItem = new QStandardItem(sNewSubjectName);
    appendRow(pSubjectItem);
 
    pSubjectItem->setData(QVariant::fromValue(BIDS_SUBJECT), BIDS_ITEM_TYPE);
    pSubjectItem->setData(QVariant::fromValue(pSubjectItem->index()), BIDS_ITEM_SUBJECT);
 
    emit newItemIndex(pSubjectItem->index());
 
    return pSubjectItem->index();
}
 
//=============================================================================================================
 
QModelIndex BidsViewModel::addSessionToSubject(const QString& sSubjectName,
                                               const QString& sSessionName)
{
    //Ensure session name follow BIDS format
    QString sNewSessionName;
 
    if (!sSessionName.startsWith("ses-")) {
        sNewSessionName = "ses-" + sSessionName;
    } else {
        sNewSessionName = sSessionName;
    }
 
    sNewSessionName.remove(" ");
 
    QList<QStandardItem*> listItems = findItems(sSubjectName);
    bool bRepeat = false;
 
    //Check for repeated names, no names
    if (listItems.size() > 1) {
        qWarning() << "[BidsViewModel::addSessionToSubject] Multiple subjects with same name";
        bRepeat = true;
    } else if (listItems.size() == 0) {
        qWarning() << "[BidsViewModel::addSessionToSubject] No Subject found with name:" << sSubjectName;
        return QModelIndex();
    }
 
    // Stays null when the loop body never runs, which the return below guards
    // against; dereferencing it unconditionally would be undefined.
    QStandardItem* pNewSessionItem = nullptr;
 
    //Add session to subjects with mathcing names. Renames them if multiple.
    for (int i = 0; i < listItems.size(); i++) {
        QStandardItem* pSubjectItem = listItems.at(i);
 
        if (bRepeat) {
            pSubjectItem->setText(pSubjectItem->text() + QString::number(i + 1));
        }
 
        pNewSessionItem = new QStandardItem(sNewSessionName);
        pSubjectItem->setChild(pSubjectItem->rowCount(),
                               pNewSessionItem);
 
        pNewSessionItem->setData(QVariant::fromValue(BIDS_SESSION), BIDS_ITEM_TYPE);
        pNewSessionItem->setData(QVariant::fromValue(pSubjectItem->index()), BIDS_ITEM_SUBJECT);
        pNewSessionItem->setData(QVariant::fromValue(pNewSessionItem->index()), BIDS_ITEM_SESSION);
 
        emit newItemIndex(pNewSessionItem->index());
    }
 
    if (!pNewSessionItem) {
        return QModelIndex();
    }
 
    return pNewSessionItem->index();
}
 
//=============================================================================================================
 
QModelIndex BidsViewModel::addSessionToSubject(QModelIndex subjectIndex,
                                               const QString& sSessionName)
{
    //Ensure session name follow BIDS format
    QString sNewSessionName;
 
    if (!sSessionName.startsWith("ses-")) {
        sNewSessionName = "ses-" + sSessionName;
    } else {
        sNewSessionName = sSessionName;
    }
 
    sNewSessionName.remove(" ");
 
    QStandardItem* pSubjectItem = itemFromIndex(subjectIndex);
    QStandardItem* pNewSessionItem = new QStandardItem(sNewSessionName);
    pSubjectItem->setChild(pSubjectItem->rowCount(),
                           pNewSessionItem);
 
    pNewSessionItem->setData(QVariant::fromValue(BIDS_SESSION), BIDS_ITEM_TYPE);
    pNewSessionItem->setData(QVariant::fromValue(subjectIndex), BIDS_ITEM_SUBJECT);
    pNewSessionItem->setData(QVariant::fromValue(pNewSessionItem->index()), BIDS_ITEM_SESSION);
 
    emit newItemIndex(pNewSessionItem->index());
 
    return pNewSessionItem->index();
}
 
//=============================================================================================================
 
QModelIndex BidsViewModel::addDataToSession(QModelIndex sessionIndex,
                                            QStandardItem* pNewItem,
                                            int iDataType)
{
    QStandardItem* pSessionItem = itemFromIndex(sessionIndex);
    bool bFolder = false;
    int iFolder;
 
    QString sFolderName;
 
    switch (iDataType) {
        case BIDS_FUNCTIONALDATA:
            sFolderName = "func";
            break;
        case BIDS_ANATOMICALDATA:
            sFolderName = "anat";
            break;
        case BIDS_BEHAVIORALDATA:
            sFolderName = "beh";
            break;
        default:
            sFolderName = "unknown";
    }
 
    for (iFolder = 0; iFolder < pSessionItem->rowCount(); iFolder++) {
        if (pSessionItem->child(iFolder)->text() == sFolderName) {
            bFolder = true;
            break;
        }
    }
 
    if (!bFolder) {
        QStandardItem* pFunctionalItem = new QStandardItem(sFolderName);
        pFunctionalItem->setData(QVariant::fromValue(BIDS_FOLDER), BIDS_ITEM_TYPE);
        pFunctionalItem->setData(QVariant::fromValue(sessionIndex), BIDS_ITEM_SESSION);
        pFunctionalItem->setData(itemFromIndex(sessionIndex)->data(BIDS_ITEM_SUBJECT), BIDS_ITEM_SUBJECT);
 
        pSessionItem->setChild(pSessionItem->rowCount(),
                               pFunctionalItem);
        pFunctionalItem->setChild(pFunctionalItem->rowCount(),
                                  pNewItem);
    } else {
        pSessionItem->child(iFolder)->setChild(pSessionItem->child(iFolder)->rowCount(),
                                               pNewItem);
    }
 
    pNewItem->setData(QVariant::fromValue(iDataType), BIDS_ITEM_TYPE);
    pNewItem->setData(QVariant::fromValue(sessionIndex), BIDS_ITEM_SESSION);
    pNewItem->setData(itemFromIndex(sessionIndex)->data(BIDS_ITEM_SUBJECT), BIDS_ITEM_SUBJECT);
 
    emit newItemIndex(pNewItem->index());
 
    return pNewItem->index();
}
 
//=============================================================================================================
 
QModelIndex BidsViewModel::moveSessionToSubject(QModelIndex subjectIndex,
                                                QModelIndex sessionIndex)
{
    beginResetModel();
 
    QStandardItem* subjectItem = itemFromIndex(subjectIndex);
    QStandardItem* sessionItem = itemFromIndex(sessionIndex);
 
    sessionItem->parent()->takeRow(sessionItem->row());
 
    subjectItem->setChild(subjectItem->rowCount(), sessionItem);
    subjectItem->setData(subjectIndex, BIDS_ITEM_SUBJECT);
 
    for (int i = 0; i < sessionItem->rowCount(); i++) {
        sessionItem->child(i)->setData(subjectIndex, BIDS_ITEM_SUBJECT);
        for (int j = 0; j < sessionItem->child(i)->rowCount(); j++) {
            sessionItem->child(i)->child(j)->setData(subjectIndex, BIDS_ITEM_SUBJECT);
        }
    }
 
    endResetModel();
 
    emit newItemIndex(sessionItem->index());
 
    return sessionItem->index();
}
 
//=============================================================================================================
 
QModelIndex BidsViewModel::moveDataToSession(QModelIndex sessionIndex,
                                             QModelIndex dataIndex)
{
    beginResetModel();
 
    QStandardItem* dataItem = itemFromIndex(dataIndex);
 
    if (dataItem->parent()->rowCount() < 2) {
        QStandardItem* parent = dataItem->parent();
 
        dataItem->parent()->takeRow(dataItem->row()).first();
        parent->parent()->takeRow(parent->row()).first();
    } else {
        dataItem->parent()->takeRow(dataItem->row()).first();
    }
 
    QModelIndex newIndex;
    //    switch(dataItem->data(ITEM_TYPE).value<int>()){
    //        case FUNCTIONALDATA:{
    //            newIndex = addDataToSession(sessionIndex,
    //                                        dataItem,
    //                                        FUNCTIONALDATA);
    //            break;
    //        }
    //        default:{
    //            qWarning() << "[BidsViewModel::moveDataToSession] Move not supported for this type of data";
    //        }
    //    }
 
    newIndex = addDataToSession(sessionIndex,
                                dataItem,
                                dataItem->data(BIDS_ITEM_TYPE).value<int>());
 
    endResetModel();
 
    emit newItemIndex(sessionIndex);
    emit newItemIndex(newIndex);
 
    return newIndex;
}
 
//=============================================================================================================
 
bool BidsViewModel::removeItem(QModelIndex itemIndex)
{
    if (!itemIndex.isValid()) {
        return false;
    }
 
    beginResetModel();
 
    QStandardItem* pItem = itemFromIndex(itemIndex);
 
    qInfo() << "Deleting" << pItem->text();
 
    switch (pItem->data(BIDS_ITEM_TYPE).value<int>()) {
        case BIDS_SUBJECT:
            if (removeRows(itemIndex.row(), 1, itemIndex.parent())) {
                endResetModel();
            }
            return true;
        case BIDS_SESSION:
            if (removeRows(itemIndex.row(), 1, itemIndex.parent())) {
                endResetModel();
            }
            return true;
        case BIDS_BEHAVIORALDATA:
        case BIDS_ANATOMICALDATA:
        case BIDS_FUNCTIONALDATA:
            if (removeRows(itemIndex.row(), 1, itemIndex.parent())) {
                endResetModel();
            }
            return true;
        case BIDS_AVERAGE:
        case BIDS_EVENT:
        case BIDS_DIPOLE:
            if (removeRows(itemIndex.row(), 1, itemIndex.parent())) {
                endResetModel();
            }
            return true;
        default:
            if (removeRows(itemIndex.row(), 1, itemIndex.parent())) {
                endResetModel();
            }
            return true;
    }
}