OSDN Git Service

Fixed IPC initialization + some code refactoring.
[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         //Clean up
127         X264_DELETE(mainWin);
128         X264_DELETE(ipc);
129         return ret;
130 }
131
132 ///////////////////////////////////////////////////////////////////////////////
133 // Multi-instance handler
134 ///////////////////////////////////////////////////////////////////////////////
135
136 void handleMultipleInstances(QStringList args, IPC *ipc)
137 {
138         bool commandSent = false;
139
140         //Process all command-line arguments
141         while(!args.isEmpty())
142         {
143                 const QString current = args.takeFirst();
144                 if(X264_STRCMP(current, "--add") || X264_STRCMP(current, "--add-file"))
145                 {
146                         if(!args.isEmpty())
147                         {
148                                 commandSent = true;
149                                 if(!ipc->sendAsync(IPC::IPC_OPCODE_ADD_FILE, QStringList() << args.takeFirst()))
150                                 {
151                                         break;
152                                 }
153                         }
154                         else
155                         {
156                                 qWarning("Argument for '--add-file' is missing!");
157                         }
158                 }
159                 else if(X264_STRCMP(current, "--add-job"))
160                 {
161                         if(args.size() >= 3)
162                         {
163                                 commandSent = true;
164                                 if(!ipc->sendAsync(IPC::IPC_OPCODE_ADD_JOB, QStringList() << args.takeFirst() << args.takeFirst() << args.takeFirst()))
165                                 {
166                                         break;
167                                 }
168                         }
169                         else
170                         {
171                                 qWarning("Argument(s) for '--add-job' are missing!");
172                                 args.clear();
173                         }
174                 }
175                 else
176                 {
177                         if(!current.startsWith("--"))
178                         {
179                                 qWarning("Unknown argument: %s", current.toUtf8().constData());
180                                 break;
181                         }
182                 }
183         }
184
185         //If no argument has been sent yet, send a ping!
186         if(!commandSent)
187         {
188                 ipc->sendAsync(IPC::IPC_OPCODE_PING, QStringList());
189         }
190 }
191
192 ///////////////////////////////////////////////////////////////////////////////
193 // Applicaton entry point
194 ///////////////////////////////////////////////////////////////////////////////
195
196 LONG WINAPI x264_exception_handler(__in struct _EXCEPTION_POINTERS *ExceptionInfo);
197 void x264_invalid_param_handler(const wchar_t*, const wchar_t*, const wchar_t*, unsigned int, uintptr_t);
198
199 static int _main(int argc, char* argv[])
200 {
201         if(X264_DEBUG)
202         {
203                 int iResult = -1;
204                 qInstallMsgHandler(x264_message_handler);
205                 X264_MEMORY_CHECK(x264_main, iResult, argc, argv);
206                 x264_finalization();
207                 return iResult;
208         }
209         else
210         {
211                 int iResult = -1;
212                 try
213                 {
214                         qInstallMsgHandler(x264_message_handler);
215                         iResult = x264_main(argc, argv);
216                         x264_finalization();
217                 }
218                 catch(char *error)
219                 {
220                         fflush(stdout);
221                         fflush(stderr);
222                         fprintf(stderr, "\nGURU MEDITATION !!!\n\nException error message: %s\n", error);
223                         x264_fatal_exit(L"Unhandeled C++ exception error, application will exit!");
224                 }
225                 catch(int error)
226                 {
227                         fflush(stdout);
228                         fflush(stderr);
229                         fprintf(stderr, "\nGURU MEDITATION !!!\n\nException error code: 0x%X\n", error);
230                         x264_fatal_exit(L"Unhandeled C++ exception error, application will exit!");
231                 }
232                 catch(...)
233                 {
234                         fflush(stdout);
235                         fflush(stderr);
236                         fprintf(stderr, "\nGURU MEDITATION !!!\n");
237                         x264_fatal_exit(L"Unhandeled C++ exception error, application will exit!");
238                 }
239                 return iResult;
240         }
241 }
242
243 int main(int argc, char* argv[])
244 {
245         if(X264_DEBUG)
246         {
247                 return _main(argc, argv);
248         }
249         else
250         {
251                 __try
252                 {
253                         SetUnhandledExceptionFilter(x264_exception_handler);
254                         _set_invalid_parameter_handler(x264_invalid_param_handler);
255                         return _main(argc, argv);
256                 }
257                 __except(1)
258                 {
259                         fflush(stdout);
260                         fflush(stderr);
261                         fprintf(stderr, "\nGURU MEDITATION !!!\n\nUnhandeled structured exception error! [code: 0x%X]\n", GetExceptionCode());
262                         x264_fatal_exit(L"Unhandeled structured exception error, application will exit!");
263                 }
264         }
265 }