JobListModel.cpp

Go to the documentation of this file.

//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      GUI/View/Job/JobListModel.cpp
//! @brief     Implements class JobListModel
//!
//! @homepage  http://www.bornagainproject.org
//! @license   GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors   Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
//  ************************************************************************************************
 
#include "GUI/View/Job/JobListModel.h"
#include "GUI/Model/Job/JobItem.h"
#include "GUI/Model/Model/JobModel.h"
 
//==================================================================================================
// JobListModel
//==================================================================================================
 
//--------------------------------------------------------------------------------------------------
// public member functions
//--------------------------------------------------------------------------------------------------
 
JobListModel::JobListModel(JobModel* jobs, QObject* parent)
    : QAbstractListModel(parent)
    , m_jobs(jobs)
{
    for (JobItem* job : m_jobs->jobItems())
        enableJobNotification(job);
 
    connect(jobs, &QAbstractItemModel::rowsAboutToBeInserted, this,
            &JobListModel::onRowsAboutToBeInserted);
    connect(jobs, &QAbstractItemModel::rowsInserted, this, &JobListModel::onRowsInserted);
}
 
JobListModel::~JobListModel()
{
    for (JobItem* job : m_jobs->jobItems()) {
        disableJobNotification(job);
    }
}
 
int JobListModel::rowCount(const QModelIndex&) const
{
    return m_jobs->jobItems().size();
}
 
QVariant JobListModel::data(const QModelIndex& index, int role) const
{
    QVector<JobItem*> jobs = m_jobs->jobItems();
    if (!index.isValid() || index.row() >= jobs.size() || index.row() < 0)
        return {};
 
    JobItem* item = jobs[index.row()];
    if (role == Qt::DisplayRole)
        return item->jobName();
 
    return {};
}
 
JobItem* JobListModel::jobForIndex(const QModelIndex& index) const
{
    QVector<JobItem*> jobs = m_jobs->jobItems();
    if (index.row() >= 0 && index.row() < jobs.size())
        return jobs[index.row()];
    return nullptr;
}
 
QModelIndex JobListModel::indexForJob(JobItem* job)
{
    QVector<JobItem*> jobs = m_jobs->jobItems();
    int idx = jobs.indexOf(job);
    if (idx != -1)
        return index(idx, 0);
    return {};
}
 
void JobListModel::runJob(const QModelIndex& index)
{
    m_jobs->runJob(jobForIndex(index));
}
 
void JobListModel::removeJob(const QModelIndex& index)
{
    beginRemoveRows(QModelIndex(), index.row(), index.row());
    JobItem* job = jobForIndex(index);
    disableJobNotification(job);
    m_jobs->removeJob(job);
    endRemoveRows();
}
 
void JobListModel::cancelJob(const QModelIndex& index)
{
    m_jobs->cancelJob(jobForIndex(index));
}
 
void JobListModel::emitJobListModelChanged(JobItem* job)
{
    QVector<JobItem*> jobs = m_jobs->jobItems();
    int i = jobs.indexOf(job);
    if (i != -1) {
        QModelIndex idx = index(i, 0);
        emit dataChanged(idx, idx);
    }
}
 
//--------------------------------------------------------------------------------------------------
// private slots
//--------------------------------------------------------------------------------------------------
 
void JobListModel::onRowsAboutToBeInserted(const QModelIndex& parent, int start, int end)
{
    if (!parent.isValid())
        beginInsertRows(QModelIndex(), start, end);
}
 
void JobListModel::onRowsInserted(const QModelIndex& parent, int start, int end)
{
    if (!parent.isValid()) {
        endInsertRows();
        QVector<JobItem*> jobs = m_jobs->jobItems();
        for (int i = start; i <= end; i++)
            enableJobNotification(jobs.at(i));
    }
}
 
//--------------------------------------------------------------------------------------------------
// private member functions
//--------------------------------------------------------------------------------------------------
 
void JobListModel::enableJobNotification(JobItem* job)
{
    // name
    connect(job, &JobItem::jobNameChanged, this,
            [=](const QString&) { emitJobListModelChanged(job); });
 
    // status
    connect(job, &JobItem::jobStatusChanged, this,
            [=](const JobStatus) { emitJobListModelChanged(job); });
 
    // progress
    connect(job, &JobItem::jobProgressChanged, this, [=](int) { emitJobListModelChanged(job); });
}
 
void JobListModel::disableJobNotification(JobItem* job)
{
    disconnect(job, nullptr, this, nullptr);
}