OSDN Git Service

Implement multi-theaded processing in progress dialog.
[lamexp/LameXP.git] / src / Dialog_Processing.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // LameXP - Audio Encoder Front-End
3 // Copyright (C) 2004-2010 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 "Dialog_Processing.h"
23
24 #include "Global.h"
25 #include "Model_FileList.h"
26 #include "Model_Progress.h"
27 #include "Thread_Process.h"
28
29 #include <QApplication>
30 #include <QRect>
31 #include <QDesktopWidget>
32 #include <QMovie>
33 #include <QMessageBox>
34 #include <QTimer>
35 #include <QCloseEvent>
36 #include <QDesktopServices>
37 #include <QUrl>
38 #include <Windows.h>
39
40 ////////////////////////////////////////////////////////////
41 // Constructor
42 ////////////////////////////////////////////////////////////
43
44 ProcessingDialog::ProcessingDialog(FileListModel *fileListModel)
45 {
46         //Init the dialog, from the .ui file
47         setupUi(this);
48         setWindowFlags(windowFlags() ^ Qt::WindowContextHelpButtonHint);
49         
50         //Setup version info
51         label_versionInfo->setText(QString().sprintf("v%d.%02d %s (Build %d)", lamexp_version_major(), lamexp_version_minor(), lamexp_version_release(), lamexp_version_build()));
52         label_versionInfo->installEventFilter(this);
53
54         //Register meta type
55         qRegisterMetaType<QUuid>("QUuid");
56
57         //Center window in screen
58         QRect desktopRect = QApplication::desktop()->screenGeometry();
59         QRect thisRect = this->geometry();
60         move((desktopRect.width() - thisRect.width()) / 2, (desktopRect.height() - thisRect.height()) / 2);
61         setMinimumSize(thisRect.width(), thisRect.height());
62
63         //Enable buttons
64         connect(button_AbortProcess, SIGNAL(clicked()), this, SLOT(abortEncoding()));
65         
66         //Init progress indicator
67         m_progressIndicator = new QMovie(":/images/Working.gif");
68         label_headerWorking->setMovie(m_progressIndicator);
69         progressBar->setValue(0);
70
71         //Init progress model
72         m_progressModel = new ProgressModel();
73         view_log->setModel(m_progressModel);
74         view_log->verticalHeader()->setResizeMode(QHeaderView::ResizeToContents);
75         view_log->verticalHeader()->hide();
76         view_log->horizontalHeader()->setResizeMode(QHeaderView::ResizeToContents);
77         view_log->horizontalHeader()->setResizeMode(0, QHeaderView::Stretch);
78
79         //Enque jobs
80         if(fileListModel)
81         {
82                 for(int i = 0; i < fileListModel->rowCount(); i++)
83                 {
84                         m_pendingJobs.append(fileListModel->getFile(fileListModel->index(i,0)));
85                 }
86         }
87
88         //Init other vars
89         m_runningThreads = 0;
90         m_userAborted = false;
91 }
92
93 ////////////////////////////////////////////////////////////
94 // Destructor
95 ////////////////////////////////////////////////////////////
96
97 ProcessingDialog::~ProcessingDialog(void)
98 {
99         view_log->setModel(NULL);
100         if(m_progressIndicator) m_progressIndicator->stop();
101         LAMEXP_DELETE(m_progressIndicator);
102         LAMEXP_DELETE(m_progressModel);
103
104         while(!m_threadList.isEmpty())
105         {
106                 delete m_threadList.takeFirst();
107         }
108 }
109
110 ////////////////////////////////////////////////////////////
111 // EVENTS
112 ////////////////////////////////////////////////////////////
113
114 void ProcessingDialog::showEvent(QShowEvent *event)
115 {
116         setCloseButtonEnabled(false);
117         button_closeDialog->setEnabled(false);
118         button_AbortProcess->setEnabled(false);
119
120         QTimer::singleShot(1000, this, SLOT(initEncoding()));
121 }
122
123 void ProcessingDialog::closeEvent(QCloseEvent *event)
124 {
125         if(!button_closeDialog->isEnabled()) event->ignore();
126 }
127
128 bool ProcessingDialog::eventFilter(QObject *obj, QEvent *event)
129 {
130         static QColor defaultColor = QColor();
131
132         if(obj == label_versionInfo)
133         {
134                 if(event->type() == QEvent::Enter)
135                 {
136                         QPalette palette = label_versionInfo->palette();
137                         defaultColor = palette.color(QPalette::Normal, QPalette::WindowText);
138                         palette.setColor(QPalette::Normal, QPalette::WindowText, Qt::red);
139                         label_versionInfo->setPalette(palette);
140                 }
141                 else if(event->type() == QEvent::Leave)
142                 {
143                         QPalette palette = label_versionInfo->palette();
144                         palette.setColor(QPalette::Normal, QPalette::WindowText, defaultColor);
145                         label_versionInfo->setPalette(palette);
146                 }
147                 else if(event->type() == QEvent::MouseButtonPress)
148                 {
149                         QUrl url("http://mulder.dummwiedeutsch.de/");
150                         QDesktopServices::openUrl(url);
151                 }
152         }
153
154         return false;
155 }
156
157 ////////////////////////////////////////////////////////////
158 // SLOTS
159 ////////////////////////////////////////////////////////////
160
161 void ProcessingDialog::initEncoding(void)
162 {
163         m_runningThreads = 0;
164         m_userAborted = false;
165         
166         label_progress->setText("Encoding files, please wait...");
167         m_progressIndicator->start();
168         
169         button_closeDialog->setEnabled(false);
170         button_AbortProcess->setEnabled(true);
171         progressBar->setRange(0, m_pendingJobs.count());
172
173         startNextJob();
174         startNextJob();
175 }
176
177 void ProcessingDialog::abortEncoding(void)
178 {
179         m_userAborted = true;
180         button_AbortProcess->setEnabled(false);
181         
182         for(int i = 0; i < m_threadList.count(); i++)
183         {
184                 m_threadList.at(i)->abort();
185         }
186 }
187
188 void ProcessingDialog::doneEncoding(void)
189 {
190         m_runningThreads--;
191
192         progressBar->setValue(progressBar->value() + 1);
193         label_progress->setText(QString("%1 files out of %2 completed, please wait...").arg(QString::number(progressBar->value()), QString::number(progressBar->maximum())));
194         
195         if(!m_pendingJobs.isEmpty() && !m_userAborted)
196         {
197                 startNextJob();
198                 qDebug("Running jobs: %u", m_runningThreads);
199                 return;
200         }
201         
202         if(m_runningThreads > 0)
203         {
204                 qDebug("Running jobs: %u", m_runningThreads);
205                 return;
206         }
207
208         qDebug("Running jobs: %u", m_runningThreads);
209
210         label_progress->setText(m_userAborted ? "Process aborted by user." : "Alle files completed.");
211         m_progressIndicator->stop();
212
213         setCloseButtonEnabled(true);
214         button_closeDialog->setEnabled(true);
215         button_AbortProcess->setEnabled(false);
216
217         progressBar->setValue(100);
218 }
219
220 ////////////////////////////////////////////////////////////
221 // Private Functions
222 ////////////////////////////////////////////////////////////
223
224 void ProcessingDialog::startNextJob(void)
225 {
226         if(m_pendingJobs.isEmpty())
227         {
228                 return;
229         }
230
231         ProcessThread *thread = new ProcessThread(m_pendingJobs.takeFirst());
232         m_threadList.append(thread);
233         connect(thread, SIGNAL(finished()), this, SLOT(doneEncoding()), Qt::QueuedConnection);
234         connect(thread, SIGNAL(processStateInitialized(QUuid,QString,QString,int)), m_progressModel, SLOT(addJob(QUuid,QString,QString,int)), Qt::QueuedConnection);
235         connect(thread, SIGNAL(processStateChanged(QUuid,QString,int)), m_progressModel, SLOT(updateJob(QUuid,QString,int)), Qt::QueuedConnection);
236         m_runningThreads++;
237         thread->start();
238 }
239
240 void ProcessingDialog::setCloseButtonEnabled(bool enabled)
241 {
242         HMENU hMenu = GetSystemMenu((HWND) winId(), FALSE);
243         EnableMenuItem(hMenu, SC_CLOSE, MF_BYCOMMAND | (enabled ? MF_ENABLED : MF_GRAYED));
244 }