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

   1 /*\r
   2 * Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>\r
   3 *\r
   4 * This file is part of CasparCG (www.casparcg.com).\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 * Author: Robert Nagy, ronag89@gmail.com\r
  20 */\r
  21 \r
  22 // tbbmalloc_proxy: \r
  23 // Replace the standard memory allocation routines in Microsoft* C/C++ RTL \r
  24 // (malloc/free, global new/delete, etc.) with the TBB memory allocator. \r
  25 \r
  26 #include "stdafx.h"\r
  27 \r
  28 #ifdef _DEBUG\r
  29         #define _CRTDBG_MAP_ALLOC\r
  30         #include <stdlib.h>\r
  31         #include <crtdbg.h>\r
  32 #else\r
  33         #include <tbb/tbbmalloc_proxy.h>\r
  34 #endif\r
  35 \r
  36 #include "resource.h"\r
  37 \r
  38 #include "server.h"\r
  39 \r
  40 #include <common/os/windows/windows.h>\r
  41 #include <winnt.h>\r
  42 #include <mmsystem.h>\r
  43 #include <atlbase.h>\r
  44 \r
  45 #include <protocol/amcp/AMCPProtocolStrategy.h>\r
  46 #include <protocol/osc/server.h>\r
  47 \r
  48 #include <modules/bluefish/bluefish.h>\r
  49 #include <modules/decklink/decklink.h>\r
  50 #include <modules/flash/flash.h>\r
  51 #include <modules/ffmpeg/ffmpeg.h>\r
  52 #include <modules/image/image.h>\r
  53 \r
  54 #include <common/env.h>\r
  55 #include <common/except.h>\r
  56 #include <common/except.h>\r
  57 #include <common/log.h>\r
  58 #include <common/gl/gl_check.h>\r
  59 #include <common/os/windows/current_version.h>\r
  60 #include <common/os/windows/system_info.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 #include <boost/property_tree/xml_parser.hpp>\r
  67 #include <boost/foreach.hpp>\r
  68 #include <boost/locale.hpp>\r
  69 \r
  70 #include <signal.h>\r
  71 \r
  72 using namespace caspar;\r
  73         \r
  74 // NOTE: This is needed in order to make CComObject work since this is not a real ATL project.\r
  75 CComModule _AtlModule;\r
  76 extern __declspec(selectany) CAtlModule* _pAtlModule = &_AtlModule;\r
  77 \r
  78 void change_icon( const HICON hNewIcon )\r
  79 {\r
  80    auto hMod = ::LoadLibrary(L"Kernel32.dll"); \r
  81    typedef DWORD(__stdcall *SCI)(HICON);\r
  82    auto pfnSetConsoleIcon = reinterpret_cast<SCI>(::GetProcAddress(hMod, "SetConsoleIcon")); \r
  83    pfnSetConsoleIcon(hNewIcon); \r
  84    ::FreeLibrary(hMod);\r
  85 }\r
  86 \r
  87 void setup_global_locale()\r
  88 {\r
  89         boost::locale::generator gen;\r
  90         gen.categories(boost::locale::codepage_facet);\r
  91 \r
  92         std::locale::global(gen(""));\r
  93 }\r
  94 \r
  95 void setup_console_window()\r
  96 {        \r
  97         auto hOut = GetStdHandle(STD_OUTPUT_HANDLE);\r
  98 \r
  99         // Disable close button in console to avoid shutdown without cleanup.\r
 100         EnableMenuItem(GetSystemMenu(GetConsoleWindow(), FALSE), SC_CLOSE , MF_GRAYED);\r
 101         DrawMenuBar(GetConsoleWindow());\r
 102         //SetConsoleCtrlHandler(HandlerRoutine, true);\r
 103 \r
 104         // Configure console size and position.\r
 105         auto coord = GetLargestConsoleWindowSize(hOut);\r
 106         coord.X /= 2;\r
 107 \r
 108         SetConsoleScreenBufferSize(hOut, coord);\r
 109 \r
 110         SMALL_RECT DisplayArea = {0, 0, 0, 0};\r
 111         DisplayArea.Right = coord.X-1;\r
 112         DisplayArea.Bottom = (coord.Y-1)/2;\r
 113         SetConsoleWindowInfo(hOut, TRUE, &DisplayArea);\r
 114                  \r
 115         change_icon(::LoadIcon(GetModuleHandle(0), MAKEINTRESOURCE(101)));\r
 116 \r
 117         // Set console title.\r
 118         std::wstringstream str;\r
 119         str << "CasparCG Server " << env::version() << L" x64 ";\r
 120 #ifdef COMPILE_RELEASE\r
 121         str << " Release";\r
 122 #elif  COMPILE_PROFILE\r
 123         str << " Profile";\r
 124 #elif  COMPILE_DEVELOP\r
 125         str << " Develop";\r
 126 #elif  COMPILE_DEBUG\r
 127         str << " Debug";\r
 128 #endif\r
 129         SetConsoleTitle(str.str().c_str());\r
 130 }\r
 131 \r
 132 void print_info()\r
 133 {\r
 134         CASPAR_LOG(info) << L"################################################################################";\r
 135         CASPAR_LOG(info) << L"Copyright (c) 2010 Sveriges Television AB, www.casparcg.com, <info@casparcg.com>";\r
 136         CASPAR_LOG(info) << L"################################################################################";\r
 137         CASPAR_LOG(info) << L"Starting CasparCG Video and Graphics Playout Server " << env::version();\r
 138         CASPAR_LOG(info) << L"on " << win_product_name() << L" " << win_sp_version();\r
 139         CASPAR_LOG(info) << cpu_info();\r
 140         CASPAR_LOG(info) << system_product_name();\r
 141         \r
 142         CASPAR_LOG(info) << L"Decklink " << decklink::version();\r
 143         BOOST_FOREACH(auto device, decklink::device_list())\r
 144                 CASPAR_LOG(info) << L" - " << device;   \r
 145                 \r
 146         CASPAR_LOG(info) << L"Bluefish " << bluefish::version();\r
 147         BOOST_FOREACH(auto device, bluefish::device_list())\r
 148                 CASPAR_LOG(info) << L" - " << device;   \r
 149         \r
 150         CASPAR_LOG(info) << L"Flash "                   << flash::version();\r
 151         CASPAR_LOG(info) << L"FreeImage "               << image::version();\r
 152         CASPAR_LOG(info) << L"FFMPEG-avcodec "  << ffmpeg::avcodec_version();\r
 153         CASPAR_LOG(info) << L"FFMPEG-avformat " << ffmpeg::avformat_version();\r
 154         CASPAR_LOG(info) << L"FFMPEG-avfilter " << ffmpeg::avfilter_version();\r
 155         CASPAR_LOG(info) << L"FFMPEG-avutil "   << ffmpeg::avutil_version();\r
 156         CASPAR_LOG(info) << L"FFMPEG-swscale "  << ffmpeg::swscale_version();\r
 157 }\r
 158 \r
 159 LONG WINAPI UserUnhandledExceptionFilter(EXCEPTION_POINTERS* info)\r
 160 {\r
 161         try\r
 162         {\r
 163                 CASPAR_LOG(fatal) << L"#######################\n UNHANDLED EXCEPTION: \n" \r
 164                         << L"Adress:" << info->ExceptionRecord->ExceptionAddress << L"\n"\r
 165                         << L"Code:" << info->ExceptionRecord->ExceptionCode << L"\n"\r
 166                         << L"Flag:" << info->ExceptionRecord->ExceptionFlags << L"\n"\r
 167                         << L"Info:" << info->ExceptionRecord->ExceptionInformation << L"\n"\r
 168                         << L"Continuing execution. \n#######################";\r
 169 \r
 170                 CASPAR_LOG_CALL_STACK();\r
 171         }\r
 172         catch(...){}\r
 173 \r
 174     return EXCEPTION_CONTINUE_EXECUTION;\r
 175 }\r
 176 \r
 177 void run()\r
 178 {\r
 179         // Create server object which initializes channels, protocols and controllers.\r
 180         server caspar_server;\r
 181                                 \r
 182         auto server = spl::make_shared<protocol::osc::server>(5253);\r
 183         caspar_server.subscribe(server);\r
 184                                                 \r
 185         //auto console_obs = reactive::make_observer([](const monitor::event& e)\r
 186         //{\r
 187         //      std::stringstream str;\r
 188         //      str << e;\r
 189         //      CASPAR_LOG(trace) << str.str().c_str();\r
 190         //});\r
 191 \r
 192         //caspar_server.subscribe(console_obs);\r
 193                                                 \r
 194         // Create a amcp parser for console commands.\r
 195         protocol::amcp::AMCPProtocolStrategy amcp(caspar_server.channels());\r
 196 \r
 197         // Create a dummy client which prints amcp responses to console.\r
 198         auto console_client = std::make_shared<IO::ConsoleClientInfo>();\r
 199                         \r
 200         std::wstring wcmd;\r
 201         while(true)\r
 202         {\r
 203                 std::getline(std::wcin, wcmd); // TODO: It's blocking...\r
 204                                 \r
 205                 boost::to_upper(wcmd);\r
 206 \r
 207                 if(wcmd == L"EXIT" || wcmd == L"Q" || wcmd == L"QUIT" || wcmd == L"BYE")\r
 208                         break;\r
 209                                 \r
 210                 // This is just dummy code for testing.\r
 211                 if(wcmd.substr(0, 1) == L"1")\r
 212                         wcmd = L"LOADBG 1-1 " + wcmd.substr(1, wcmd.length()-1) + L" SLIDE 100 LOOP \r\nPLAY 1-1";\r
 213                 else if(wcmd.substr(0, 1) == L"2")\r
 214                         wcmd = L"MIXER 1-0 VIDEO IS_KEY 1";\r
 215                 else if(wcmd.substr(0, 1) == L"3")\r
 216                         wcmd = L"CG 1-2 ADD 1 BBTELEFONARE 1";\r
 217                 else if(wcmd.substr(0, 1) == L"4")\r
 218                         wcmd = L"PLAY 1-1 DV FILTER yadif=1:-1 LOOP";\r
 219                 else if(wcmd.substr(0, 1) == L"5")\r
 220                 {\r
 221                         auto file = wcmd.substr(2, wcmd.length()-1);\r
 222                         wcmd = L"PLAY 1-1 " + file + L" LOOP\r\n" \r
 223                                         L"PLAY 1-2 " + file + L" LOOP\r\n" \r
 224                                         L"PLAY 1-3 " + file + L" LOOP\r\n"\r
 225                                         L"PLAY 2-1 " + file + L" LOOP\r\n" \r
 226                                         L"PLAY 2-2 " + file + L" LOOP\r\n" \r
 227                                         L"PLAY 2-3 " + file + L" LOOP\r\n";\r
 228                 }\r
 229                 else if(wcmd.substr(0, 1) == L"7")\r
 230                 {\r
 231                         wcmd = L"";\r
 232                         wcmd += L"CLEAR 1\r\n";\r
 233                         wcmd += L"MIXER 1 CLEAR\r\n";\r
 234                         wcmd += L"PLAY 1-0 GREEN\r\n";\r
 235                         wcmd += L"PLAY 1-1 BLUE\r\n";\r
 236                         wcmd += L"CG 1-2 ADD 1 ECS_TEST 1\r\n";\r
 237                         wcmd += L"MIXER 1-2 FILL 0 -1 1 2\r\n";\r
 238                 }\r
 239                 else if(wcmd.substr(0, 1) == L"8")\r
 240                 {\r
 241                         wcmd = L"";\r
 242                         wcmd += L"MIXER 1-1 FILL 0.0 0.5 1.0 1.0 500 linear DEFER\r\n";\r
 243                         wcmd += L"MIXER 1-2 FILL 0.0 0.0 1.0 1.0 500 linear DEFER\r\n";\r
 244                         wcmd += L"MIXER 1 COMMIT\r\n";\r
 245                 }\r
 246                 else if(wcmd.substr(0, 1) == L"X")\r
 247                 {\r
 248                         int num = 0;\r
 249                         std::wstring file;\r
 250                         try\r
 251                         {\r
 252                                 num = boost::lexical_cast<int>(wcmd.substr(1, 2));\r
 253                                 file = wcmd.substr(4, wcmd.length()-1);\r
 254                         }\r
 255                         catch(...)\r
 256                         {\r
 257                                 num = boost::lexical_cast<int>(wcmd.substr(1, 1));\r
 258                                 file = wcmd.substr(3, wcmd.length()-1);\r
 259                         }\r
 260 \r
 261                         int n = 0;\r
 262                         int num2 = num;\r
 263                         while(num2 > 0)\r
 264                         {\r
 265                                 num2 >>= 1;\r
 266                                 n++;\r
 267                         }\r
 268 \r
 269                         wcmd = L"MIXER 1 GRID " + boost::lexical_cast<std::wstring>(n);\r
 270 \r
 271                         for(int i = 1; i <= num; ++i)\r
 272                                 wcmd += L"\r\nPLAY 1-" + boost::lexical_cast<std::wstring>(i) + L" " + file + L" LOOP";// + L" SLIDE 100 LOOP";\r
 273                 }\r
 274 \r
 275                 wcmd += L"\r\n";\r
 276                 amcp.Parse(wcmd.c_str(), static_cast<int>(wcmd.length()), console_client);\r
 277         }       \r
 278         CASPAR_LOG(info) << "Successfully shutdown CasparCG Server.";\r
 279 }\r
 280 \r
 281 void on_abort(int)\r
 282 {\r
 283         CASPAR_THROW_EXCEPTION(invalid_operation() << msg_info("abort called"));\r
 284 }\r
 285 \r
 286 int main(int argc, wchar_t* argv[])\r
 287 {       \r
 288         SetUnhandledExceptionFilter(UserUnhandledExceptionFilter);\r
 289         signal(SIGABRT, on_abort);\r
 290 \r
 291         setup_global_locale();\r
 292 \r
 293         std::wcout << L"Type \"q\" to close application." << std::endl;\r
 294         \r
 295         // Set debug mode.\r
 296         #ifdef _DEBUG\r
 297                 HANDLE hLogFile;\r
 298                 hLogFile = CreateFile(L"crt_log.txt", GENERIC_WRITE, FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);\r
 299                 std::shared_ptr<void> crt_log(nullptr, [](HANDLE h){::CloseHandle(h);});\r
 300 \r
 301                 _CrtSetDbgFlag (_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF);\r
 302                 _CrtSetReportMode(_CRT_WARN,    _CRTDBG_MODE_FILE);\r
 303                 _CrtSetReportFile(_CRT_WARN,    hLogFile);\r
 304                 _CrtSetReportMode(_CRT_ERROR,   _CRTDBG_MODE_FILE);\r
 305                 _CrtSetReportFile(_CRT_ERROR,   hLogFile);\r
 306                 _CrtSetReportMode(_CRT_ASSERT,  _CRTDBG_MODE_FILE);\r
 307                 _CrtSetReportFile(_CRT_ASSERT,  hLogFile);\r
 308         #endif\r
 309 \r
 310         // Increase process priotity.\r
 311         SetPriorityClass(GetCurrentProcess(), ABOVE_NORMAL_PRIORITY_CLASS);\r
 312 \r
 313         // Install structured exception handler.\r
 314         win32_exception::install_handler();\r
 315                                 \r
 316         // Increase time precision. This will increase accuracy of function like Sleep(1) from 10 ms to 1 ms.\r
 317         struct inc_prec\r
 318         {\r
 319                 inc_prec(){timeBeginPeriod(1);}\r
 320                 ~inc_prec(){timeEndPeriod(1);}\r
 321         } inc_prec;     \r
 322 \r
 323         // Install SEH into all tbb threads.\r
 324         struct tbb_thread_installer : public tbb::task_scheduler_observer\r
 325         {\r
 326                 tbb_thread_installer(){observe(true);}\r
 327                 void on_scheduler_entry(bool is_worker)\r
 328                 {\r
 329                         //detail::SetThreadName(GetCurrentThreadId(), "tbb-worker-thread");\r
 330                         win32_exception::install_handler();\r
 331                 }\r
 332         } tbb_thread_installer;\r
 333 \r
 334         tbb::task_scheduler_init init;\r
 335         \r
 336         try \r
 337         {\r
 338                 // Configure environment properties from configuration.\r
 339                 env::configure(L"casparcg.config");\r
 340                                 \r
 341                 log::set_log_level(env::properties().get(L"configuration.log-level", L"debug"));\r
 342 \r
 343         #ifdef _DEBUG\r
 344                 if(env::properties().get(L"configuration.debugging.remote", false))\r
 345                         MessageBox(nullptr, TEXT("Now is the time to connect for remote debugging..."), TEXT("Debug"), MB_OK | MB_TOPMOST);\r
 346         #endif   \r
 347 \r
 348                 // Start logging to file.\r
 349                 log::add_file_sink(env::log_folder());                  \r
 350                 std::wcout << L"Logging [info] or higher severity to " << env::log_folder() << std::endl << std::endl;\r
 351                 \r
 352                 // Setup console window.\r
 353                 setup_console_window();\r
 354 \r
 355                 // Print environment information.\r
 356                 print_info();\r
 357                 \r
 358                 std::wstringstream str;\r
 359                 boost::property_tree::xml_writer_settings<wchar_t> w(' ', 3);\r
 360                 boost::property_tree::write_xml(str, env::properties(), w);\r
 361                 CASPAR_LOG(info) << L"casparcg.config:\n-----------------------------------------\n" << str.str().c_str() << L"-----------------------------------------";\r
 362                 \r
 363                 run();\r
 364                 \r
 365                 system("pause");        \r
 366         }\r
 367         catch(boost::property_tree::file_parser_error&)\r
 368         {\r
 369                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 370                 CASPAR_LOG(fatal) << L"Unhandled configuration error in main thread. Please check the configuration file (casparcg.config) for errors.";\r
 371                 system("pause");        \r
 372         }\r
 373         catch(...)\r
 374         {\r
 375                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 376                 CASPAR_LOG(fatal) << L"Unhandled exception in main thread. Please report this error on the CasparCG forums (www.casparcg.com/forum).";\r
 377                 Sleep(1000);\r
 378                 std::wcout << L"\n\nCasparCG will automatically shutdown. See the log file located at the configured log-file folder for more information.\n\n";\r
 379                 Sleep(4000);\r
 380         }               \r
 381         \r
 382         return 0;\r
 383 }