git.sesse.net Git - casparcg/blob - shell/main.cpp

   1 /*\r
   2 * copyright (c) 2010 Sveriges Television AB <info@casparcg.com>\r
   3 *\r
   4 *  This file is part of CasparCG.\r
   5 *\r
   6 *    CasparCG is free software: you can redistribute it and/or modify\r
   7 *    it under the terms of the GNU General Public License as published by\r
   8 *    the Free Software Foundation, either version 3 of the License, or\r
   9 *    (at your option) any later version.\r
  10 *\r
  11 *    CasparCG is distributed in the hope that it will be useful,\r
  12 *    but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  13 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  14 *    GNU General Public License for more details.\r
  15 \r
  16 *    You should have received a copy of the GNU General Public License\r
  17 *    along with CasparCG.  If not, see <http://www.gnu.org/licenses/>.\r
  18 *\r
  19 */\r
  20 \r
  21 // tbbmalloc_proxy: \r
  22 // Replace the standard memory allocation routines in Microsoft* C/C++ RTL \r
  23 // (malloc/free, global new/delete, etc.) with the TBB memory allocator. \r
  24 #include <tbb/tbbmalloc_proxy.h>\r
  25 \r
  26 #include "resource.h"\r
  27 \r
  28 #include "server.h"\r
  29 \r
  30 #ifdef _DEBUG\r
  31         #define _CRTDBG_MAP_ALLOC\r
  32         #include <stdlib.h>\r
  33         #include <crtdbg.h>\r
  34 #endif\r
  35 \r
  36 #define NOMINMAX\r
  37 #define WIN32_LEAN_AND_MEAN\r
  38 \r
  39 #include <windows.h>\r
  40 #include <winnt.h>\r
  41 #include <mmsystem.h>\r
  42 #include <atlbase.h>\r
  43 \r
  44 #include <protocol/amcp/AMCPProtocolStrategy.h>\r
  45 \r
  46 #include <modules/bluefish/bluefish.h>\r
  47 #include <modules/decklink/decklink.h>\r
  48 #include <modules/flash/flash.h>\r
  49 #include <modules/ffmpeg/ffmpeg.h>\r
  50 #include <modules/image/image.h>\r
  51 \r
  52 #include <common/env.h>\r
  53 #include <common/exception/win32_exception.h>\r
  54 #include <common/exception/exceptions.h>\r
  55 #include <common/log/log.h>\r
  56 #include <common/gl/gl_check.h>\r
  57 #include <common/os/windows/current_version.h>\r
  58 #include <common/os/windows/system_info.h>\r
  59 \r
  60 #include <core/mixer/gpu/ogl_device.h>\r
  61 \r
  62 #include <tbb/task_scheduler_init.h>\r
  63 #include <tbb/task_scheduler_observer.h>\r
  64 \r
  65 #include <boost/property_tree/detail/file_parser_error.hpp>\r
  66 \r
  67 #include <algorithm>\r
  68 \r
  69 struct application_state\r
  70 {\r
  71         enum type\r
  72         {\r
  73                 running,\r
  74                 shutdown,\r
  75                 pause_and_shutdown\r
  76         };\r
  77 };\r
  78 \r
  79 boost::condition_variable       shutdown_cond;\r
  80 boost::mutex                            shutdown_mut;\r
  81 tbb::atomic<int>                        shutdown_event;\r
  82 \r
  83 // NOTE: This is needed in order to make CComObject work since this is not a real ATL project.\r
  84 CComModule _AtlModule;\r
  85 extern __declspec(selectany) CAtlModule* _pAtlModule = &_AtlModule;\r
  86 \r
  87 void change_icon( const HICON hNewIcon )\r
  88 {\r
  89    auto hMod = ::LoadLibrary(L"Kernel32.dll"); \r
  90    typedef DWORD(__stdcall *SCI)(HICON);\r
  91    auto pfnSetConsoleIcon = reinterpret_cast<SCI>(::GetProcAddress(hMod, "SetConsoleIcon")); \r
  92    pfnSetConsoleIcon(hNewIcon); \r
  93    ::FreeLibrary(hMod);\r
  94 }\r
  95 \r
  96 BOOL WINAPI HandlerRoutine(__in  DWORD dwCtrlType)\r
  97 {\r
  98         switch(dwCtrlType)\r
  99         {\r
 100         case CTRL_CLOSE_EVENT:\r
 101         case CTRL_SHUTDOWN_EVENT:\r
 102                 shutdown_event = application_state::shutdown;\r
 103                 shutdown_cond.notify_all();\r
 104                 return true;\r
 105         }\r
 106         return false;\r
 107 }\r
 108 \r
 109 void setup_console_window()\r
 110 {        \r
 111         auto hOut = GetStdHandle(STD_OUTPUT_HANDLE);\r
 112 \r
 113         // Disable close button in console to avoid shutdown without cleanup.\r
 114         //EnableMenuItem(GetSystemMenu(GetConsoleWindow(), FALSE), SC_CLOSE , MF_GRAYED);\r
 115         //DrawMenuBar(GetConsoleWindow());\r
 116         SetConsoleCtrlHandler(HandlerRoutine, true);\r
 117 \r
 118         // Configure console size and position.\r
 119         auto coord = GetLargestConsoleWindowSize(hOut);\r
 120         coord.X /= 2;\r
 121 \r
 122         SetConsoleScreenBufferSize(hOut, coord);\r
 123 \r
 124         SMALL_RECT DisplayArea = {0, 0, 0, 0};\r
 125         DisplayArea.Right = coord.X-1;\r
 126         DisplayArea.Bottom = (coord.Y-1)/2;\r
 127         SetConsoleWindowInfo(hOut, TRUE, &DisplayArea);\r
 128                  \r
 129         change_icon(::LoadIcon(GetModuleHandle(0), MAKEINTRESOURCE(101)));\r
 130 \r
 131         // Set console title.\r
 132         std::wstringstream str;\r
 133         str << "CasparCG Server " << caspar::env::version();\r
 134 #ifdef COMPILE_RELEASE\r
 135         str << " Release";\r
 136 #elif  COMPILE_PROFILE\r
 137         str << " Profile";\r
 138 #elif  COMPILE_DEVELOP\r
 139         str << " Develop";\r
 140 #elif  COMPILE_DEBUG\r
 141         str << " Debug";\r
 142 #endif\r
 143         SetConsoleTitle(str.str().c_str());\r
 144 }\r
 145 \r
 146 void print_info()\r
 147 {\r
 148         CASPAR_LOG(info) << L"Copyright (c) 2010 Sveriges Television AB, www.casparcg.com, <info@casparcg.com>";\r
 149         CASPAR_LOG(info) << L"Starting CasparCG Video and Graphics Playout Server " << caspar::env::version();\r
 150         CASPAR_LOG(info) << L"on " << caspar::get_win_product_name() << L" " << caspar::get_win_sp_version();\r
 151         CASPAR_LOG(info) << caspar::get_cpu_info();\r
 152         CASPAR_LOG(info) << caspar::get_system_product_name();\r
 153         CASPAR_LOG(info) << L"Flash " << caspar::flash::get_version();\r
 154         CASPAR_LOG(info) << L"Flash-Template-Host " << caspar::flash::get_cg_version();\r
 155         CASPAR_LOG(info) << L"FreeImage " << caspar::image::get_version();\r
 156         \r
 157         CASPAR_LOG(info) << L"Decklink " << caspar::decklink::get_version();\r
 158         auto deck = caspar::decklink::get_device_list();\r
 159         std::for_each(deck.begin(), deck.end(), [](const std::wstring& device)\r
 160         {\r
 161                 CASPAR_LOG(info) << device;\r
 162         });\r
 163                 \r
 164         CASPAR_LOG(info) << L"Bluefish " << caspar::bluefish::get_version();\r
 165         auto blue = caspar::bluefish::get_device_list();\r
 166         std::for_each(blue.begin(), blue.end(), [](const std::wstring& device)\r
 167         {\r
 168                 CASPAR_LOG(info) << device;\r
 169         });\r
 170         \r
 171         CASPAR_LOG(info) << L"FFMPEG-avcodec "  << caspar::ffmpeg::get_avcodec_version();\r
 172         CASPAR_LOG(info) << L"FFMPEG-avformat " << caspar::ffmpeg::get_avformat_version();\r
 173         CASPAR_LOG(info) << L"FFMPEG-avfilter " << caspar::ffmpeg::get_avfilter_version();\r
 174         CASPAR_LOG(info) << L"FFMPEG-avutil "   << caspar::ffmpeg::get_avutil_version();\r
 175         CASPAR_LOG(info) << L"FFMPEG-swscale "  << caspar::ffmpeg::get_swscale_version();\r
 176         //CASPAR_LOG(info) << L"OpenGL " << caspar::core::ogl_device::get_version() << "\n\n"; // Dead-locks for some reason?\r
 177 }\r
 178 \r
 179 LONG WINAPI UserUnhandledExceptionFilter(EXCEPTION_POINTERS* info)\r
 180 {\r
 181         try\r
 182         {\r
 183                 CASPAR_LOG(fatal) << L"#######################\n UNHANDLED EXCEPTION: \n" \r
 184                         << L"Adress:" << info->ExceptionRecord->ExceptionAddress << L"\n"\r
 185                         << L"Code:" << info->ExceptionRecord->ExceptionCode << L"\n"\r
 186                         << L"Flag:" << info->ExceptionRecord->ExceptionFlags << L"\n"\r
 187                         << L"Info:" << info->ExceptionRecord->ExceptionInformation << L"\n"\r
 188                         << L"Continuing execution. \n#######################";\r
 189         }\r
 190         catch(...){}\r
 191 \r
 192     return EXCEPTION_EXECUTE_HANDLER;\r
 193 }\r
 194 \r
 195 int main(int argc, wchar_t* argv[])\r
 196 {       \r
 197         static_assert(sizeof(void*) == 4, "64-bit code generation is not supported.");\r
 198         \r
 199         SetUnhandledExceptionFilter(UserUnhandledExceptionFilter);\r
 200 \r
 201         CASPAR_LOG(info) << L"Type \"q\" to close application";\r
 202         \r
 203         // Set debug mode.\r
 204         #ifdef _DEBUG\r
 205                 _CrtSetDbgFlag ( _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF | _CRTDBG_CHECK_ALWAYS_DF );\r
 206                 _CrtSetReportMode( _CRT_ERROR, _CRTDBG_MODE_DEBUG );\r
 207                 _CrtSetReportMode( _CRT_WARN, _CRTDBG_MODE_DEBUG );\r
 208                 _CrtSetReportMode( _CRT_ASSERT, _CRTDBG_MODE_DEBUG );\r
 209         #endif\r
 210 \r
 211         // Increase process priotity.\r
 212         SetPriorityClass(GetCurrentProcess(), ABOVE_NORMAL_PRIORITY_CLASS);\r
 213 \r
 214         // Install structured exception handler.\r
 215         caspar::win32_exception::install_handler();\r
 216 \r
 217         caspar::log::set_log_level(L"debug");\r
 218                         \r
 219         // Increase time precision. This will increase accuracy of function like Sleep(1) from 10 ms to 1 ms.\r
 220         struct inc_prec\r
 221         {\r
 222                 inc_prec(){timeBeginPeriod(1);}\r
 223                 ~inc_prec(){timeEndPeriod(1);}\r
 224         } inc_prec;     \r
 225 \r
 226         // Install unstructured exception handlers into all tbb threads.\r
 227         struct tbb_thread_installer : public tbb::task_scheduler_observer\r
 228         {\r
 229                 tbb_thread_installer(){observe(true);}\r
 230                 void on_scheduler_entry(bool is_worker)\r
 231                 {\r
 232                         //caspar::detail::SetThreadName(GetCurrentThreadId(), "tbb-worker-thread");\r
 233                         caspar::win32_exception::install_handler();\r
 234                 }\r
 235         } tbb_thread_installer;\r
 236 \r
 237         tbb::task_scheduler_init init;\r
 238         \r
 239         try \r
 240         {\r
 241                 {\r
 242                         // Configure environment properties from configuration.\r
 243                         caspar::env::configure("casparcg.config");\r
 244 \r
 245                 #ifdef _DEBUG\r
 246                         if(caspar::env::properties().get("configuration.debugging.remote", false))\r
 247                                 MessageBox(nullptr, TEXT("Now is the time to connect for remote debugging..."), TEXT("Debug"), MB_OK | MB_TOPMOST);\r
 248                 #endif   \r
 249 \r
 250                         // Start logging to file.\r
 251                         caspar::log::add_file_sink(caspar::env::log_folder());\r
 252                 \r
 253                         // Setup console window.\r
 254                         setup_console_window();\r
 255 \r
 256                         // Print environment information.\r
 257                         print_info();\r
 258                                 \r
 259                         // Create server object which initializes channels, protocols and controllers.\r
 260                         caspar::server caspar_server;\r
 261                                 \r
 262                         // Create a amcp parser for console commands.\r
 263                         caspar::protocol::amcp::AMCPProtocolStrategy amcp(caspar_server.get_channels());\r
 264 \r
 265                         // Create a dummy client which prints amcp responses to console.\r
 266                         auto console_client = std::make_shared<caspar::IO::ConsoleClientInfo>();\r
 267 \r
 268                         boost::thread input_thread([&]\r
 269                         {\r
 270                                 while(shutdown_event == application_state::running)\r
 271                                 {\r
 272                                         std::wstring wcmd;\r
 273                                         std::getline(std::wcin, wcmd); // TODO: It's blocking...\r
 274 \r
 275                                         try\r
 276                                         {\r
 277                                                 if(wcmd == L"exit" || wcmd == L"q")\r
 278                                                 {\r
 279                                                         shutdown_event = application_state::pause_and_shutdown;\r
 280                                                         shutdown_cond.notify_all();\r
 281                                                         return;\r
 282                                                 }\r
 283 \r
 284                                                 // This is just dummy code for testing.\r
 285                                                 if(wcmd.substr(0, 1) == L"1")\r
 286                                                         wcmd = L"LOADBG 1-1 " + wcmd.substr(1, wcmd.length()-1) + L" SLIDE 100 LOOP \r\nPLAY 1-1";\r
 287                                                 else if(wcmd.substr(0, 1) == L"2")\r
 288                                                         wcmd = L"MIXER 1-0 VIDEO IS_KEY 1";\r
 289                                                 else if(wcmd.substr(0, 1) == L"3")\r
 290                                                         wcmd = L"CG 1-2 ADD 1 BBTELEFONARE 1";\r
 291                                                 else if(wcmd.substr(0, 1) == L"4")\r
 292                                                         wcmd = L"PLAY 1-1 DV FILTER yadif=1:-1 LOOP";\r
 293                                                 else if(wcmd.substr(0, 1) == L"5")\r
 294                                                 {\r
 295                                                         auto file = wcmd.substr(2, wcmd.length()-1);\r
 296                                                         wcmd = L"PLAY 1-1 " + file + L" LOOP\r\n" \r
 297                                                                    L"PLAY 1-2 " + file + L" LOOP\r\n" \r
 298                                                                    L"PLAY 1-3 " + file + L" LOOP\r\n"\r
 299                                                                    L"PLAY 2-1 " + file + L" LOOP\r\n" \r
 300                                                                    L"PLAY 2-2 " + file + L" LOOP\r\n" \r
 301                                                                    L"PLAY 2-3 " + file + L" LOOP\r\n";\r
 302                                                 }\r
 303                                                 else if(wcmd.substr(0, 1) == L"X")\r
 304                                                 {\r
 305                                                         int num = 0;\r
 306                                                         std::wstring file;\r
 307                                                         try\r
 308                                                         {\r
 309                                                                 num = boost::lexical_cast<int>(wcmd.substr(1, 2));\r
 310                                                                 file = wcmd.substr(4, wcmd.length()-1);\r
 311                                                         }\r
 312                                                         catch(...)\r
 313                                                         {\r
 314                                                                 num = boost::lexical_cast<int>(wcmd.substr(1, 1));\r
 315                                                                 file = wcmd.substr(3, wcmd.length()-1);\r
 316                                                         }\r
 317 \r
 318                                                         int n = 0;\r
 319                                                         int num2 = num;\r
 320                                                         while(num2 > 0)\r
 321                                                         {\r
 322                                                                 num2 >>= 1;\r
 323                                                                 n++;\r
 324                                                         }\r
 325 \r
 326                                                         wcmd = L"MIXER 1 GRID " + boost::lexical_cast<std::wstring>(n);\r
 327 \r
 328                                                         for(int i = 1; i <= num; ++i)\r
 329                                                                 wcmd += L"\r\nPLAY 1-" + boost::lexical_cast<std::wstring>(i) + L" " + file + L" LOOP";// + L" SLIDE 100 LOOP";\r
 330                                                 }\r
 331 \r
 332                                                 wcmd += L"\r\n";\r
 333                                                 amcp.Parse(wcmd.c_str(), wcmd.length(), console_client);\r
 334                                         }\r
 335                                         catch(...)\r
 336                                         {\r
 337                                                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 338                                         }\r
 339                                 }\r
 340                         });\r
 341 \r
 342                         boost::unique_lock<boost::mutex> lock(shutdown_mut);\r
 343                         while(shutdown_event == application_state::running)                     \r
 344                                 shutdown_cond.wait(lock);                               \r
 345                 }       \r
 346                         \r
 347                 Sleep(200); // CAPSAR_LOG is asynchronous. Try to get text in correct order.'\r
 348 \r
 349                 if(shutdown_event == application_state::pause_and_shutdown)\r
 350                         system("pause");        \r
 351         }\r
 352         catch(boost::property_tree::file_parser_error&)\r
 353         {\r
 354                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 355                 CASPAR_LOG(fatal) << L"Unhandled configuration error in main thread. Please check the configuration file (casparcg.config) for errors.";\r
 356         }\r
 357         catch(caspar::gl::ogl_exception&)\r
 358         {\r
 359                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 360                 CASPAR_LOG(fatal) << L"Unhandled OpenGL Error in main thread. Please try to update graphics drivers for OpenGL 3.0+ Support.";\r
 361         }\r
 362         catch(...)\r
 363         {\r
 364                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 365                 CASPAR_LOG(fatal) << L"Unhandled exception in main thread. Please report this error on the CasparCG forums (www.casparcg.com/forum).";\r
 366         }       \r
 367         \r
 368         CASPAR_LOG(info) << "Successfully shutdown CasparCG Server.";\r
 369         return 0;\r
 370 }