OSDN Git Service

Added multi-instance handling to main() function.
[x264-launcher/x264-launcher.git] / src / main.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Simple x264 Launcher
3 // Copyright (C) 2004-2014 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 "global.h"
23 #include "win_main.h"
24 #include "ipc.h"
25 #include "taskbar7.h"
26
27 //Qt includes
28 #include <QApplication>
29 #include <QDate>
30 #include <QPlastiqueStyle>
31
32 //Windows includes
33 #define NOMINMAX
34 #define WIN32_LEAN_AND_MEAN
35 #include <Windows.h>
36
37 //Forward declaration
38 void handleMultipleInstances(QStringList args, IPC *ipc);
39
40 ///////////////////////////////////////////////////////////////////////////////
41 // Main function
42 ///////////////////////////////////////////////////////////////////////////////
43
44 static int x264_main(int argc, char* argv[])
45 {
46         //Init console
47         x264_init_console(argc, argv);
48
49         //Print version info
50         qDebug("Simple x264 Launcher v%u.%02u.%u - use 64-Bit x264 with 32-Bit Avisynth", x264_version_major(), x264_version_minor(), x264_version_build());
51         qDebug("Copyright (c) 2004-%04d LoRd_MuldeR <mulder2@gmx.de>. Some rights reserved.", qMax(x264_version_date().year(),QDate::currentDate().year()));
52         qDebug("Built on %s at %s with %s for Win-%s.\n", x264_version_date().toString(Qt::ISODate).toLatin1().constData(), x264_version_time(), x264_version_compiler(), x264_version_arch());
53         
54         //print license info
55         qDebug("This program is free software: you can redistribute it and/or modify");
56         qDebug("it under the terms of the GNU General Public License <http://www.gnu.org/>.");
57         qDebug("Note that this program is distributed with ABSOLUTELY NO WARRANTY.\n");
58
59         //Print warning, if this is a "debug" build
60         if(X264_DEBUG)
61         {
62                 qWarning("---------------------------------------------------------");
63                 qWarning("DEBUG BUILD: DO NOT RELEASE THIS BINARY TO THE PUBLIC !!!");
64                 qWarning("---------------------------------------------------------\n"); 
65         }
66
67         //Get CLI arguments
68         const QStringList &arguments = x264_arguments();
69         
70         //Detect CPU capabilities
71         const x264_cpu_t cpuFeatures = x264_detect_cpu_features(arguments);
72         qDebug("   CPU vendor id  :  %s (Intel: %s)", cpuFeatures.vendor, X264_BOOL(cpuFeatures.intel));
73         qDebug("CPU brand string  :  %s", cpuFeatures.brand);
74         qDebug("   CPU signature  :  Family: %d, Model: %d, Stepping: %d", cpuFeatures.family, cpuFeatures.model, cpuFeatures.stepping);
75         qDebug("CPU capabilities  :  MMX=%s, MMXEXT=%s, SSE=%s, SSE2=%s, SSE3=%s, SSSE3=%s, X64=%s", X264_BOOL(cpuFeatures.mmx), X264_BOOL(cpuFeatures.mmx2), X264_BOOL(cpuFeatures.sse), X264_BOOL(cpuFeatures.sse2), X264_BOOL(cpuFeatures.sse3), X264_BOOL(cpuFeatures.ssse3), X264_BOOL(cpuFeatures.x64));
76         qDebug(" Number of CPU's  :  %d\n", cpuFeatures.count);
77
78         //Initialize the IPC handler class
79         bool firstInstance = false;
80         IPC *ipc = new IPC();
81         if(!ipc->initialize(firstInstance))
82         {
83                 if(!firstInstance)
84                 {
85                         handleMultipleInstances(arguments, ipc);
86                         X264_DELETE(ipc);
87                         return 0;
88                 }
89         }
90         else
91         {
92                 qWarning("IPC initialization has failed!");
93         }
94
95         //Initialize Qt
96         if(!x264_init_qt(argc, argv))
97         {
98                 return -1;
99         }
100         
101         //Running in portable mode?
102         if(x264_portable())
103         {
104                 qDebug("Application is running in portable mode!\n");
105         }
106
107         //Taskbar init
108         WinSevenTaskbar::init();
109
110         //Set style
111         if(!qApp->arguments().contains("--no-style", Qt::CaseInsensitive))
112         {
113                 qApp->setStyle(new QPlastiqueStyle());
114         }
115
116         //Create Main Window
117         MainWindow *mainWin = new MainWindow(&cpuFeatures);
118         mainWin->show();
119
120         //Run application
121         int ret = qApp->exec();
122
123         //Taskbar uninit
124         WinSevenTaskbar::init();
125         
126         X264_DELETE(mainWin);
127         return ret;
128 }
129
130 ///////////////////////////////////////////////////////////////////////////////
131 // Multi-instance handler
132 ///////////////////////////////////////////////////////////////////////////////
133
134 void handleMultipleInstances(QStringList args, IPC *ipc)
135 {
136         bool commandSent = false;
137
138         //Process all command-line arguments
139         while(!args.isEmpty())
140         {
141                 const QString current = args.takeFirst();
142                 if((current.compare("--add", Qt::CaseInsensitive) == 0) || (current.compare("--add-file", Qt::CaseInsensitive) == 0))
143                 {
144                         if(!args.isEmpty())
145                         {
146                                 commandSent = true;
147                                 if(!ipc->sendAsync(IPC::IPC_OPCODE_ADD_FILE, QStringList() << args.takeFirst()))
148                                 {
149                                         break;
150                                 }
151                         }
152                         else
153                         {
154                                 qWarning("Argument for '--add-file' is missing!");
155                         }
156                 }
157                 else if(current.compare("--add-job", Qt::CaseInsensitive) == 0)
158                 {
159                         if(args.size() >= 3)
160                         {
161                                 commandSent = true;
162                                 if(!ipc->sendAsync(IPC::IPC_OPCODE_ADD_JOB, QStringList() << args.takeFirst() << args.takeFirst() << args.takeFirst()))
163                                 {
164                                         break;
165                                 }
166                         }
167                         else
168                         {
169                                 qWarning("Argument(s) for '--add-job' are missing!");
170                                 args.clear();
171                         }
172                 }
173         }
174
175         //If no argument has been sent yet, send a ping!
176         if(!commandSent)
177         {
178                 ipc->sendAsync(IPC::IPC_OPCODE_PING, QStringList());
179         }
180 }
181
182 ///////////////////////////////////////////////////////////////////////////////
183 // Applicaton entry point
184 ///////////////////////////////////////////////////////////////////////////////
185
186 LONG WINAPI x264_exception_handler(__in struct _EXCEPTION_POINTERS *ExceptionInfo);
187 void x264_invalid_param_handler(const wchar_t*, const wchar_t*, const wchar_t*, unsigned int, uintptr_t);
188
189 static int _main(int argc, char* argv[])
190 {
191         if(X264_DEBUG)
192         {
193                 int iResult = -1;
194                 qInstallMsgHandler(x264_message_handler);
195                 X264_MEMORY_CHECK(x264_main, iResult, argc, argv);
196                 x264_finalization();
197                 return iResult;
198         }
199         else
200         {
201                 int iResult = -1;
202                 try
203                 {
204                         qInstallMsgHandler(x264_message_handler);
205                         iResult = x264_main(argc, argv);
206                         x264_finalization();
207                 }
208                 catch(char *error)
209                 {
210                         fflush(stdout);
211                         fflush(stderr);
212                         fprintf(stderr, "\nGURU MEDITATION !!!\n\nException error message: %s\n", error);
213                         x264_fatal_exit(L"Unhandeled C++ exception error, application will exit!");
214                 }
215                 catch(int error)
216                 {
217                         fflush(stdout);
218                         fflush(stderr);
219                         fprintf(stderr, "\nGURU MEDITATION !!!\n\nException error code: 0x%X\n", error);
220                         x264_fatal_exit(L"Unhandeled C++ exception error, application will exit!");
221                 }
222                 catch(...)
223                 {
224                         fflush(stdout);
225                         fflush(stderr);
226                         fprintf(stderr, "\nGURU MEDITATION !!!\n");
227                         x264_fatal_exit(L"Unhandeled C++ exception error, application will exit!");
228                 }
229                 return iResult;
230         }
231 }
232
233 int main(int argc, char* argv[])
234 {
235         if(X264_DEBUG)
236         {
237                 return _main(argc, argv);
238         }
239         else
240         {
241                 __try
242                 {
243                         SetUnhandledExceptionFilter(x264_exception_handler);
244                         _set_invalid_parameter_handler(x264_invalid_param_handler);
245                         return _main(argc, argv);
246                 }
247                 __except(1)
248                 {
249                         fflush(stdout);
250                         fflush(stderr);
251                         fprintf(stderr, "\nGURU MEDITATION !!!\n\nUnhandeled structured exception error! [code: 0x%X]\n", GetExceptionCode());
252                         x264_fatal_exit(L"Unhandeled structured exception error, application will exit!");
253                 }
254         }
255 }