src/Model_Progress.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2011 LoRd_MuldeR <MuldeR2@GMX.de>
   4 //
   5 // This program is free software; you can redistribute it and/or modify
   6 // it under the terms of the GNU General Public License as published by
   7 // the Free Software Foundation; either version 2 of the License, or
   8 // (at your option) any later version.
   9 //
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14 //
  15 // You should have received a copy of the GNU General Public License along
  16 // with this program; if not, write to the Free Software Foundation, Inc.,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 // http://www.gnu.org/licenses/gpl-2.0.txt
  20 ///////////////////////////////////////////////////////////////////////////////
  21
  22 #include "Model_Progress.h"
  23
  24 #include <QUuid>
  25
  26 ProgressModel::ProgressModel(void)
  27 :
  28         m_iconRunning(":/icons/media_play.png"),
  29         m_iconPaused(":/icons/control_pause_blue.png"),
  30         m_iconComplete(":/icons/tick.png"),
  31         m_iconFailed(":/icons/exclamation.png"),
  32         m_iconSystem(":/icons/computer.png")
  33 {
  34 }
  35
  36 ProgressModel::~ProgressModel(void)
  37 {
  38 }
  39
  40 int ProgressModel::columnCount(const QModelIndex &parent) const
  41 {
  42         return 2;
  43 }
  44
  45 int ProgressModel::rowCount(const QModelIndex &parent) const
  46 {
  47         return m_jobList.count();
  48 }
  49
  50 QVariant ProgressModel::data(const QModelIndex &index, int role) const
  51 {
  52         if(index.row() >= 0 && index.row() < m_jobList.count())
  53         {
  54                 if(role == Qt::DisplayRole)
  55                 {
  56                         switch(index.column())
  57                         {
  58                         case 0:
  59                                 return m_jobName.value(m_jobList.at(index.row()));
  60                                 break;
  61                         case 1:
  62                                 return m_jobStatus.value(m_jobList.at(index.row()));
  63                                 break;
  64                         default:
  65                                 return QVariant();
  66                                 break;
  67                         }
  68                 }
  69                 else if(role == Qt::DecorationRole && index.column() == 0)
  70                 {
  71                         switch(m_jobState.value(m_jobList.at(index.row())))
  72                         {
  73                         case JobRunning:
  74                                 return m_iconRunning;
  75                                 break;
  76                         case JobPaused:
  77                                 return m_iconPaused;
  78                                 break;
  79                         case JobComplete:
  80                                 return m_iconComplete;
  81                                 break;
  82                         case JobSystem:
  83                                 return m_iconSystem;
  84                                 break;
  85                         default:
  86                                 return m_iconFailed;
  87                                 break;
  88                         }
  89                 }
  90                 else if(role == Qt::TextAlignmentRole)
  91                 {
  92                         return (index.column() == 1) ? QVariant(Qt::AlignHCenter | Qt::AlignVCenter) : QVariant();
  93                 }
  94         }
  95
  96         return QVariant();
  97 }
  98
  99 QVariant ProgressModel::headerData(int section, Qt::Orientation orientation, int role) const
 100 {
 101         if(role == Qt::DisplayRole)
 102         {
 103                 if(orientation == Qt::Horizontal)
 104                 {
 105                         switch(section)
 106                         {
 107                         case 0:
 108                                 return tr("Job");
 109                                 break;
 110                         case 1:
 111                                 return tr("Status");
 112                                 break;
 113                         default:
 114                                 return QVariant();
 115                                 break;
 116                         }
 117                 }
 118                 if(orientation == Qt::Vertical)
 119                 {
 120                         return QString::number(section + 1);
 121                 }
 122         }
 123
 124         return QVariant();
 125 }
 126
 127 void ProgressModel::addJob(const QUuid &jobId, const QString &jobName, const QString &jobInitialStatus, int jobInitialState)
 128 {
 129         if(m_jobList.contains(jobId))
 130         {
 131                 return;
 132         }
 133
 134         int newIndex = m_jobList.count();
 135         beginInsertRows(QModelIndex(), newIndex, newIndex);
 136
 137         m_jobList.append(jobId);
 138         m_jobName.insert(jobId, jobName);
 139         m_jobStatus.insert(jobId, jobInitialStatus);
 140         m_jobState.insert(jobId, jobInitialState);
 141         m_jobLogFile.insert(jobId, QStringList());
 142
 143         endInsertRows();
 144 }
 145
 146 void ProgressModel::updateJob(const QUuid &jobId, const QString &newStatus, int newState)
 147 {
 148         int row = m_jobList.indexOf(jobId);
 149
 150         if(row < 0)
 151         {
 152                 return;
 153         }
 154
 155         if(!newStatus.isEmpty()) m_jobStatus.insert(jobId, newStatus);
 156         if(newState >= 0) m_jobState.insert(jobId, newState);
 157
 158         emit dataChanged(index(row, 0), index(row, 1));
 159 }
 160
 161 void ProgressModel::appendToLog(const QUuid &jobId, const QString &line)
 162 {
 163         if(m_jobList.contains(jobId))
 164         {
 165                 m_jobLogFile[jobId].append(line.split('\n'));
 166         }
 167 }
 168
 169 const QStringList &ProgressModel::getLogFile(const QModelIndex &index)
 170 {
 171         if(index.row() < m_jobList.count())
 172         {
 173                 QUuid id = m_jobList.at(index.row());
 174                 return m_jobLogFile[id];
 175         }
 176
 177         return *(reinterpret_cast<QStringList*>(NULL));
 178 }
 179
 180 void ProgressModel::addSystemMessage(const QString &text)
 181 {
 182         const QUuid &jobId = QUuid::createUuid();
 183
 184         if(m_jobList.contains(jobId))
 185         {
 186                 return;
 187         }
 188
 189         int newIndex = m_jobList.count();
 190         beginInsertRows(QModelIndex(), newIndex, newIndex);
 191
 192         m_jobList.append(jobId);
 193         m_jobName.insert(jobId, text);
 194         m_jobStatus.insert(jobId, QString());
 195         m_jobState.insert(jobId, JobSystem);
 196         m_jobLogFile.insert(jobId, QStringList());
 197
 198         endInsertRows();
 199 }