Bump version + updated changelog.
[simple-x264-launcher.git] / src / model_jobList.cpp
blob5acc1d9a935046f776100ed39b123795afe5d15a
1 ///////////////////////////////////////////////////////////////////////////////
2 // Simple x264 Launcher
3 // Copyright (C) 2004-2018 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.
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.
19 // http://www.gnu.org/licenses/gpl-2.0.txt
20 ///////////////////////////////////////////////////////////////////////////////
22 //Internal
23 #include "global.h"
24 #include "model_jobList.h"
25 #include "thread_encode.h"
26 #include "encoder_factory.h"
27 #include "model_options.h"
28 #include "model_preferences.h"
29 #include "resource.h"
31 //MUtils
32 #include <MUtils/Sound.h>
34 //Qt
35 #include <QIcon>
36 #include <QFileInfo>
37 #include <QSettings>
39 static const char *KEY_ENTRY_COUNT = "entry_count";
40 static const char *KEY_SOURCE_FILE = "source_file";
41 static const char *KEY_OUTPUT_FILE = "output_file";
42 static const char *KEY_ENC_OPTIONS = "enc_options";
44 static const char *JOB_TEMPLATE = "job_%08x";
46 #define VALID_INDEX(INDEX) ((INDEX).isValid() && ((INDEX).row() >= 0) && ((INDEX).row() < m_jobs.count()))
48 JobListModel::JobListModel(PreferencesModel *preferences)
50 m_preferences = preferences;
53 JobListModel::~JobListModel(void)
55 while(!m_jobs.isEmpty())
57 QUuid id = m_jobs.takeFirst();
58 EncodeThread *thread = m_threads.value(id, NULL);
59 LogFileModel *logFile = m_logFile.value(id, NULL);
60 MUTILS_DELETE(thread);
61 MUTILS_DELETE(logFile);
65 ///////////////////////////////////////////////////////////////////////////////
66 // Model interface
67 ///////////////////////////////////////////////////////////////////////////////
69 int JobListModel::columnCount(const QModelIndex &parent) const
71 return 4;
74 int JobListModel::rowCount(const QModelIndex &parent) const
76 return m_jobs.count();
79 QVariant JobListModel::headerData(int section, Qt::Orientation orientation, int role) const
81 if((orientation == Qt::Horizontal) && (role == Qt::DisplayRole))
83 switch(section)
85 case 0:
86 return QVariant::fromValue<QString>(tr("Job"));
87 break;
88 case 1:
89 return QVariant::fromValue<QString>(tr("Status"));
90 break;
91 case 2:
92 return QVariant::fromValue<QString>(tr("Progress"));
93 break;
94 case 3:
95 return QVariant::fromValue<QString>(tr("Details"));
96 break;
97 default:
98 return QVariant();
99 break;
103 return QVariant();
106 QModelIndex JobListModel::index(int row, int column, const QModelIndex &parent) const
108 return createIndex(row, column, NULL);
111 QModelIndex JobListModel::parent(const QModelIndex &index) const
113 return QModelIndex();
116 QVariant JobListModel::data(const QModelIndex &index, int role) const
118 if(role == Qt::DisplayRole)
120 if(index.row() >= 0 && index.row() < m_jobs.count())
122 switch(index.column())
124 case 0:
125 return m_name.value(m_jobs.at(index.row()));
126 break;
127 case 1:
128 switch(m_status.value(m_jobs.at(index.row())))
130 case JobStatus_Enqueued:
131 return QVariant::fromValue<QString>(tr("Enqueued."));
132 break;
133 case JobStatus_Starting:
134 return QVariant::fromValue<QString>(tr("Starting..."));
135 break;
136 case JobStatus_Indexing:
137 return QVariant::fromValue<QString>(tr("Indexing..."));
138 break;
139 case JobStatus_Running:
140 return QVariant::fromValue<QString>(tr("Running..."));
141 break;
142 case JobStatus_Running_Pass1:
143 return QVariant::fromValue<QString>(tr("Running... (Pass 1)"));
144 break;
145 case JobStatus_Running_Pass2:
146 return QVariant::fromValue<QString>(tr("Running... (Pass 2)"));
147 break;
148 case JobStatus_Completed:
149 return QVariant::fromValue<QString>(tr("Completed."));
150 break;
151 case JobStatus_Failed:
152 return QVariant::fromValue<QString>(tr("Failed!"));
153 break;
154 case JobStatus_Pausing:
155 return QVariant::fromValue<QString>(tr("Pausing..."));
156 break;
157 case JobStatus_Paused:
158 return QVariant::fromValue<QString>(tr("Paused."));
159 break;
160 case JobStatus_Resuming:
161 return QVariant::fromValue<QString>(tr("Resuming..."));
162 break;
163 case JobStatus_Aborting:
164 return QVariant::fromValue<QString>(tr("Aborting..."));
165 break;
166 case JobStatus_Aborted:
167 return QVariant::fromValue<QString>(tr("Aborted!"));
168 break;
169 default:
170 return QVariant::fromValue<QString>(tr("(Unknown)"));
171 break;
173 break;
174 case 2:
175 return QString().sprintf("%d%%", m_progress.value(m_jobs.at(index.row())));
176 break;
177 case 3:
178 return m_details.value(m_jobs.at(index.row()));
179 break;
180 default:
181 return QVariant();
182 break;
186 else if(role == Qt::DecorationRole)
188 if(index.row() >= 0 && index.row() < m_jobs.count() && index.column() == 0)
190 switch(m_status.value(m_jobs.at(index.row())))
192 case JobStatus_Enqueued:
193 return QIcon(":/buttons/hourglass.png");
194 break;
195 case JobStatus_Starting:
196 return QIcon(":/buttons/lightning.png");
197 break;
198 case JobStatus_Indexing:
199 return QIcon(":/buttons/find.png");
200 break;
201 case JobStatus_Running:
202 case JobStatus_Running_Pass1:
203 case JobStatus_Running_Pass2:
204 return QIcon(":/buttons/play.png");
205 break;
206 case JobStatus_Completed:
207 return QIcon(":/buttons/accept.png");
208 break;
209 case JobStatus_Failed:
210 return QIcon(":/buttons/exclamation.png");
211 break;
212 case JobStatus_Pausing:
213 return QIcon(":/buttons/clock_pause.png");
214 break;
215 case JobStatus_Paused:
216 return QIcon(":/buttons/suspended.png");
217 break;
218 case JobStatus_Resuming:
219 return QIcon(":/buttons/clock_play.png");
220 break;
221 case JobStatus_Aborting:
222 return QIcon(":/buttons/clock_stop.png");
223 break;
224 case JobStatus_Aborted:
225 return QIcon(":/buttons/error.png");
226 break;
227 default:
228 return QVariant();
229 break;
234 return QVariant();
237 ///////////////////////////////////////////////////////////////////////////////
238 // Public interface
239 ///////////////////////////////////////////////////////////////////////////////
241 QModelIndex JobListModel::insertJob(EncodeThread *thread)
243 const QUuid id = thread->getId();
244 if(m_jobs.contains(id))
246 return QModelIndex();
249 const AbstractEncoderInfo &encoderInfo = EncoderFactory::getEncoderInfo(thread->options()->encType());
250 const QStringList encoderNameParts = encoderInfo.getName().simplified().split(' ', QString::SkipEmptyParts);
251 QString config = encoderNameParts.isEmpty() ? encoderInfo.getName() : encoderNameParts.first();
252 switch(encoderInfo.rcModeToType(thread->options()->rcMode()))
254 case AbstractEncoderInfo::RC_TYPE_QUANTIZER:
255 config.append(QString(", %1@%2").arg(encoderInfo.rcModeToString(thread->options()->rcMode()), QString::number(qRound(thread->options()->quantizer()))));
256 break;
257 case AbstractEncoderInfo::RC_TYPE_RATE_KBPS:
258 case AbstractEncoderInfo::RC_TYPE_MULTIPASS:
259 config.append(QString(", %1@%2").arg(encoderInfo.rcModeToString(thread->options()->rcMode()), QString::number(thread->options()->bitrate())));
260 break;
263 int n = 2;
264 QString jobName = QString("%1 [%2]").arg(QFileInfo(thread->sourceFileName()).completeBaseName().simplified(), config);
265 forever
267 bool unique = true;
268 for(int i = 0; i < m_jobs.count(); i++)
270 if(m_name.value(m_jobs.at(i)).compare(jobName, Qt::CaseInsensitive) == 0)
272 unique = false;
273 break;
276 if(!unique)
278 jobName = QString("%1 %2 [%3]").arg(QFileInfo(thread->sourceFileName()).completeBaseName().simplified(), QString::number(n++), config);
279 continue;
281 break;
284 LogFileModel *logFile = new LogFileModel(thread->sourceFileName(), thread->outputFileName(), config);
286 beginInsertRows(QModelIndex(), m_jobs.count(), m_jobs.count());
287 m_jobs.append(id);
288 m_name.insert(id, jobName);
289 m_status.insert(id, JobStatus_Enqueued);
290 m_progress.insert(id, 0);
291 m_threads.insert(id, thread);
292 m_logFile.insert(id, logFile);
293 m_details.insert(id, tr("Not started yet."));
294 endInsertRows();
296 connect(thread, SIGNAL(statusChanged(QUuid, JobStatus)), this, SLOT(updateStatus(QUuid, JobStatus)), Qt::QueuedConnection);
297 connect(thread, SIGNAL(progressChanged(QUuid, unsigned int)), this, SLOT(updateProgress(QUuid, unsigned int)), Qt::QueuedConnection);
298 connect(thread, SIGNAL(messageLogged(QUuid, qint64, QString)), logFile, SLOT(addLogMessage(QUuid, qint64, QString)), Qt::QueuedConnection);
299 connect(thread, SIGNAL(detailsChanged(QUuid, QString)), this, SLOT(updateDetails(QUuid, QString)), Qt::QueuedConnection);
301 return createIndex(m_jobs.count() - 1, 0, NULL);
304 bool JobListModel::startJob(const QModelIndex &index)
306 if(VALID_INDEX(index))
308 QUuid id = m_jobs.at(index.row());
309 if(m_status.value(id) == JobStatus_Enqueued)
311 updateStatus(id, JobStatus_Starting);
312 updateDetails(id, tr("Starting up, please wait..."));
313 m_threads.value(id)->start();
314 return true;
318 return false;
321 bool JobListModel::pauseJob(const QModelIndex &index)
323 if(VALID_INDEX(index))
325 QUuid id = m_jobs.at(index.row());
326 JobStatus status = m_status.value(id);
327 if((status == JobStatus_Indexing) || (status == JobStatus_Running) ||
328 (status == JobStatus_Running_Pass1) || (status == JobStatus_Running_Pass2))
330 updateStatus(id, JobStatus_Pausing);
331 m_threads.value(id)->pauseJob();
332 return true;
336 return false;
339 bool JobListModel::resumeJob(const QModelIndex &index)
341 if(VALID_INDEX(index))
343 QUuid id = m_jobs.at(index.row());
344 JobStatus status = m_status.value(id);
345 if(status == JobStatus_Paused)
347 updateStatus(id, JobStatus_Resuming);
348 m_threads.value(id)->resumeJob();
349 return true;
353 return false;
356 bool JobListModel::abortJob(const QModelIndex &index)
358 if(VALID_INDEX(index))
360 QUuid id = m_jobs.at(index.row());
361 if(m_status.value(id) == JobStatus_Indexing || m_status.value(id) == JobStatus_Running ||
362 m_status.value(id) == JobStatus_Running_Pass1 || JobStatus_Running_Pass2)
364 updateStatus(id, JobStatus_Aborting);
365 m_threads.value(id)->abortJob();
366 return true;
370 return false;
373 bool JobListModel::deleteJob(const QModelIndex &index)
375 if(VALID_INDEX(index))
377 QUuid id = m_jobs.at(index.row());
378 if(m_status.value(id) == JobStatus_Completed || m_status.value(id) == JobStatus_Failed ||
379 m_status.value(id) == JobStatus_Aborted || m_status.value(id) == JobStatus_Enqueued)
381 int idx = index.row();
382 QUuid id = m_jobs.at(idx);
383 EncodeThread *thread = m_threads.value(id, NULL);
384 LogFileModel *logFile = m_logFile.value(id, NULL);
385 if((thread == NULL) || (!thread->isRunning()))
388 beginRemoveRows(QModelIndex(), idx, idx);
389 m_jobs.removeAt(index.row());
390 m_name.remove(id);
391 m_threads.remove(id);
392 m_status.remove(id);
393 m_progress.remove(id);
394 m_logFile.remove(id);
395 m_details.remove(id);
396 endRemoveRows();
397 MUTILS_DELETE(thread);
398 MUTILS_DELETE(logFile);
399 return true;
404 return false;
407 bool JobListModel::moveJob(const QModelIndex &index, const int &direction)
409 if(VALID_INDEX(index))
411 if((direction == MOVE_UP) && (index.row() > 0))
413 beginMoveRows(QModelIndex(), index.row(), index.row(), QModelIndex(), index.row() - 1);
414 m_jobs.swap(index.row(), index.row() - 1);
415 endMoveRows();
416 return true;
418 if((direction == MOVE_DOWN) && (index.row() < m_jobs.size() - 1))
420 beginMoveRows(QModelIndex(), index.row(), index.row(), QModelIndex(), index.row() + 2);
421 m_jobs.swap(index.row(), index.row() + 1);
422 endMoveRows();
423 return true;
427 return false;
430 LogFileModel *JobListModel::getLogFile(const QModelIndex &index)
432 if(index.isValid() && index.row() >= 0 && index.row() < m_jobs.count())
434 return m_logFile.value(m_jobs.at(index.row()));
437 return NULL;
440 const QString &JobListModel::getJobSourceFile(const QModelIndex &index)
442 static QString nullStr;
444 if(index.isValid() && index.row() >= 0 && index.row() < m_jobs.count())
446 EncodeThread *thread = m_threads.value(m_jobs.at(index.row()));
447 return (thread != NULL) ? thread->sourceFileName() : nullStr;
450 return nullStr;
453 const QString &JobListModel::getJobOutputFile(const QModelIndex &index)
455 static QString nullStr;
457 if(index.isValid() && index.row() >= 0 && index.row() < m_jobs.count())
459 EncodeThread *thread = m_threads.value(m_jobs.at(index.row()));
460 return (thread != NULL) ? thread->outputFileName() : nullStr;
463 return nullStr;
466 JobStatus JobListModel::getJobStatus(const QModelIndex &index)
468 if(index.isValid() && index.row() >= 0 && index.row() < m_jobs.count())
470 return m_status.value(m_jobs.at(index.row()));
473 return static_cast<JobStatus>(-1);
476 unsigned int JobListModel::getJobProgress(const QModelIndex &index)
478 if(index.isValid() && index.row() >= 0 && index.row() < m_jobs.count())
480 return m_progress.value(m_jobs.at(index.row()));
483 return 0;
486 const OptionsModel *JobListModel::getJobOptions(const QModelIndex &index)
488 static QString nullStr;
490 if(index.isValid() && index.row() >= 0 && index.row() < m_jobs.count())
492 EncodeThread *thread = m_threads.value(m_jobs.at(index.row()));
493 return (thread != NULL) ? thread->options() : NULL;
496 return NULL;
499 QModelIndex JobListModel::getJobIndexById(const QUuid &id)
501 if(m_jobs.contains(id))
503 return createIndex(m_jobs.indexOf(id), 0);
506 return QModelIndex();
509 ///////////////////////////////////////////////////////////////////////////////
510 // Slots
511 ///////////////////////////////////////////////////////////////////////////////
513 void JobListModel::updateStatus(const QUuid &jobId, JobStatus newStatus)
515 int index = -1;
517 if((index = m_jobs.indexOf(jobId)) >= 0)
519 m_status.insert(jobId, newStatus);
520 emit dataChanged(createIndex(index, 0), createIndex(index, 1));
522 if(m_preferences->getEnableSounds())
524 switch(newStatus)
526 case JobStatus_Completed:
527 MUtils::Sound::play_sound("tada", true);
528 break;
529 case JobStatus_Aborted:
530 MUtils::Sound::play_sound("shattering", true);
531 break;
532 case JobStatus_Failed:
533 MUtils::Sound::play_sound("failure", true);
534 break;
540 void JobListModel::updateProgress(const QUuid &jobId, unsigned int newProgress)
542 int index = -1;
544 if((index = m_jobs.indexOf(jobId)) >= 0)
546 m_progress.insert(jobId, qBound(0U, newProgress, 100U));
547 emit dataChanged(createIndex(index, 2), createIndex(index, 2));
551 void JobListModel::updateDetails(const QUuid &jobId, const QString &details)
553 int index = -1;
555 if((index = m_jobs.indexOf(jobId)) >= 0)
557 m_details.insert(jobId, details);
558 emit dataChanged(createIndex(index, 3), createIndex(index, 3));
562 size_t JobListModel::saveQueuedJobs(void)
564 const QString appDir = x264_data_path();
565 QSettings settings(QString("%1/queue.ini").arg(appDir), QSettings::IniFormat);
567 settings.clear();
568 settings.setValue(KEY_ENTRY_COUNT, 0);
569 size_t jobCounter = 0;
571 for(QList<QUuid>::ConstIterator iter = m_jobs.constBegin(); iter != m_jobs.constEnd(); iter++)
573 if(m_status.value(*iter) == JobStatus_Enqueued)
575 if(const EncodeThread *thread = m_threads.value(*iter))
577 settings.beginGroup(QString().sprintf(JOB_TEMPLATE, jobCounter++));
578 settings.setValue(KEY_SOURCE_FILE, thread->sourceFileName());
579 settings.setValue(KEY_OUTPUT_FILE, thread->outputFileName());
581 settings.beginGroup(KEY_ENC_OPTIONS);
582 OptionsModel::saveOptions(thread->options(), settings);
584 settings.endGroup();
585 settings.endGroup();
587 settings.setValue(KEY_ENTRY_COUNT, jobCounter);
592 settings.sync();
593 return jobCounter;
596 size_t JobListModel::loadQueuedJobs(const SysinfoModel *sysinfo)
598 const QString appDir = x264_data_path();
599 QSettings settings(QString("%1/queue.ini").arg(appDir), QSettings::IniFormat);
601 bool ok = false;
602 const size_t jobCounter = settings.value(KEY_ENTRY_COUNT, 0).toUInt(&ok);
604 if((!ok) || (jobCounter < 1))
606 return 0;
609 const QStringList groups = settings.childGroups();
610 for(size_t i = 0; i < jobCounter; i++)
612 if(!groups.contains(QString().sprintf(JOB_TEMPLATE, i)))
614 return 0;
618 size_t jobsCreated = 0;
619 for(size_t i = 0; i < jobCounter; i++)
621 settings.beginGroup(QString().sprintf(JOB_TEMPLATE, i));
622 const QString sourceFileName = settings.value(KEY_SOURCE_FILE, QString()).toString().trimmed();
623 const QString outputFileName = settings.value(KEY_OUTPUT_FILE, QString()).toString().trimmed();
625 if(sourceFileName.isEmpty() || outputFileName.isEmpty())
627 settings.endGroup();
628 continue;
631 settings.beginGroup(KEY_ENC_OPTIONS);
632 OptionsModel options(sysinfo);
633 const bool okay = OptionsModel::loadOptions(&options, settings);
635 settings.endGroup();
636 settings.endGroup();
638 if(okay)
640 EncodeThread *thread = new EncodeThread(sourceFileName, outputFileName, &options, sysinfo, m_preferences);
641 insertJob(thread);
642 jobsCreated++;
646 return jobsCreated;
649 void JobListModel::clearQueuedJobs(void)
651 const QString appDir = x264_data_path();
652 QSettings settings(QString("%1/queue.ini").arg(appDir), QSettings::IniFormat);
653 settings.clear();
654 settings.setValue(KEY_ENTRY_COUNT, 0);
655 settings.sync();