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 #include <tbb/tbbmalloc_proxy.h>\r
  26 \r
  27 #include "resource.h"\r
  28 \r
  29 #include "server.h"\r
  30 \r
  31 #ifdef _DEBUG\r
  32         #define _CRTDBG_MAP_ALLOC\r
  33         #include <stdlib.h>\r
  34         #include <crtdbg.h>\r
  35 #endif\r
  36 \r
  37 #define NOMINMAX\r
  38 #define WIN32_LEAN_AND_MEAN\r
  39 \r
  40 #include <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 \r
  47 #include <modules/bluefish/bluefish.h>\r
  48 #include <modules/decklink/decklink.h>\r
  49 #include <modules/flash/flash.h>\r
  50 #include <modules/ffmpeg/ffmpeg.h>\r
  51 #include <modules/image/image.h>\r
  52 \r
  53 #include <common/env.h>\r
  54 #include <common/exception/win32_exception.h>\r
  55 #include <common/exception/exceptions.h>\r
  56 #include <common/log/log.h>\r
  57 #include <common/gl/gl_check.h>\r
  58 #include <common/os/windows/current_version.h>\r
  59 #include <common/os/windows/system_info.h>\r
  60 #include <common/utility/string.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 \r
  69 // NOTE: This is needed in order to make CComObject work since this is not a real ATL project.\r
  70 CComModule _AtlModule;\r
  71 extern __declspec(selectany) CAtlModule* _pAtlModule = &_AtlModule;\r
  72 \r
  73 void change_icon( const HICON hNewIcon )\r
  74 {\r
  75    auto hMod = ::LoadLibrary(L"Kernel32.dll"); \r
  76    typedef DWORD(__stdcall *SCI)(HICON);\r
  77    auto pfnSetConsoleIcon = reinterpret_cast<SCI>(::GetProcAddress(hMod, "SetConsoleIcon")); \r
  78    pfnSetConsoleIcon(hNewIcon); \r
  79    ::FreeLibrary(hMod);\r
  80 }\r
  81 \r
  82 void setup_console_window()\r
  83 {        \r
  84         auto hOut = GetStdHandle(STD_OUTPUT_HANDLE);\r
  85 \r
  86         // Disable close button in console to avoid shutdown without cleanup.\r
  87         EnableMenuItem(GetSystemMenu(GetConsoleWindow(), FALSE), SC_CLOSE , MF_GRAYED);\r
  88         DrawMenuBar(GetConsoleWindow());\r
  89         //SetConsoleCtrlHandler(HandlerRoutine, true);\r
  90 \r
  91         // Configure console size and position.\r
  92         auto coord = GetLargestConsoleWindowSize(hOut);\r
  93         coord.X /= 2;\r
  94 \r
  95         SetConsoleScreenBufferSize(hOut, coord);\r
  96 \r
  97         SMALL_RECT DisplayArea = {0, 0, 0, 0};\r
  98         DisplayArea.Right = coord.X-1;\r
  99         DisplayArea.Bottom = (coord.Y-1)/2;\r
 100         SetConsoleWindowInfo(hOut, TRUE, &DisplayArea);\r
 101                  \r
 102         change_icon(::LoadIcon(GetModuleHandle(0), MAKEINTRESOURCE(101)));\r
 103 \r
 104         // Set console title.\r
 105         std::stringstream str;\r
 106         str << "CasparCG Server " << caspar::env::version();\r
 107 #ifdef COMPILE_RELEASE\r
 108         str << " Release";\r
 109 #elif  COMPILE_PROFILE\r
 110         str << " Profile";\r
 111 #elif  COMPILE_DEVELOP\r
 112         str << " Develop";\r
 113 #elif  COMPILE_DEBUG\r
 114         str << " Debug";\r
 115 #endif\r
 116         SetConsoleTitle(caspar::u16(str.str()).c_str());\r
 117 }\r
 118 \r
 119 void print_info()\r
 120 {\r
 121         CASPAR_LOG(info) << "Copyright (c) 2010 Sveriges Television AB, www.casparcg.com, <info@casparcg.com>";\r
 122         CASPAR_LOG(info) << "Starting CasparCG Video and Graphics Playout Server " << caspar::env::version();\r
 123         CASPAR_LOG(info) << "on " << caspar::get_win_product_name() << " " << caspar::get_win_sp_version();\r
 124         CASPAR_LOG(info) << caspar::get_cpu_info();\r
 125         CASPAR_LOG(info) << caspar::get_system_product_name();\r
 126         \r
 127         CASPAR_LOG(info) << "Decklink " << caspar::decklink::get_version();\r
 128         BOOST_FOREACH(auto device, caspar::decklink::get_device_list())\r
 129                 CASPAR_LOG(info) << " - " << device;    \r
 130                 \r
 131         CASPAR_LOG(info) << "Bluefish " << caspar::bluefish::get_version();\r
 132         BOOST_FOREACH(auto device, caspar::bluefish::get_device_list())\r
 133                 CASPAR_LOG(info) << " - " << device;    \r
 134         \r
 135         CASPAR_LOG(info) << "Flash "                    << caspar::flash::get_version();\r
 136         CASPAR_LOG(info) << "FreeImage "                << caspar::image::get_version();\r
 137         CASPAR_LOG(info) << "FFMPEG-avcodec "  << caspar::ffmpeg::get_avcodec_version();\r
 138         CASPAR_LOG(info) << "FFMPEG-avformat " << caspar::ffmpeg::get_avformat_version();\r
 139         CASPAR_LOG(info) << "FFMPEG-avfilter " << caspar::ffmpeg::get_avfilter_version();\r
 140         CASPAR_LOG(info) << "FFMPEG-avutil "    << caspar::ffmpeg::get_avutil_version();\r
 141         CASPAR_LOG(info) << "FFMPEG-swscale "  << caspar::ffmpeg::get_swscale_version();\r
 142 }\r
 143 \r
 144 LONG WINAPI UserUnhandledExceptionFilter(EXCEPTION_POINTERS* info)\r
 145 {\r
 146         try\r
 147         {\r
 148                 CASPAR_LOG(fatal) << "#######################\n UNHANDLED EXCEPTION: \n" \r
 149                         << "Adress:" << info->ExceptionRecord->ExceptionAddress << "\n"\r
 150                         << "Code:" << info->ExceptionRecord->ExceptionCode << "\n"\r
 151                         << "Flag:" << info->ExceptionRecord->ExceptionFlags << "\n"\r
 152                         << "Info:" << info->ExceptionRecord->ExceptionInformation << "\n"\r
 153                         << "Continuing execution. \n#######################";\r
 154         }\r
 155         catch(...){}\r
 156 \r
 157     return EXCEPTION_EXECUTE_HANDLER;\r
 158 }\r
 159 \r
 160 int main(int argc, wchar_t* argv[])\r
 161 {       \r
 162         static_assert(sizeof(void*) == 4, "64-bit code generation is not supported.");\r
 163         \r
 164         SetUnhandledExceptionFilter(UserUnhandledExceptionFilter);\r
 165 \r
 166         std::wcout << "Type \"q\" to close application." << std::endl;\r
 167         \r
 168         // Set debug mode.\r
 169         #ifdef _DEBUG\r
 170                 _CrtSetDbgFlag ( _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF | _CRTDBG_CHECK_ALWAYS_DF );\r
 171                 _CrtSetReportMode( _CRT_ERROR, _CRTDBG_MODE_DEBUG );\r
 172                 _CrtSetReportMode( _CRT_WARN, _CRTDBG_MODE_DEBUG );\r
 173                 _CrtSetReportMode( _CRT_ASSERT, _CRTDBG_MODE_DEBUG );\r
 174         #endif\r
 175 \r
 176         // Increase process priotity.\r
 177         SetPriorityClass(GetCurrentProcess(), ABOVE_NORMAL_PRIORITY_CLASS);\r
 178 \r
 179         // Install structured exception handler.\r
 180         caspar::win32_exception::install_handler();\r
 181 \r
 182         caspar::log::set_log_level("debug");\r
 183                         \r
 184         // Increase time precision. This will increase accuracy of function like Sleep(1) from 10 ms to 1 ms.\r
 185         struct inc_prec\r
 186         {\r
 187                 inc_prec(){timeBeginPeriod(1);}\r
 188                 ~inc_prec(){timeEndPeriod(1);}\r
 189         } inc_prec;     \r
 190 \r
 191         // Install unstructured exception handlers into all tbb threads.\r
 192         struct tbb_thread_installer : public tbb::task_scheduler_observer\r
 193         {\r
 194                 tbb_thread_installer(){observe(true);}\r
 195                 void on_scheduler_entry(bool is_worker)\r
 196                 {\r
 197                         //caspar::detail::SetThreadName(GetCurrentThreadId(), "tbb-worker-thread");\r
 198                         caspar::win32_exception::install_handler();\r
 199                 }\r
 200         } tbb_thread_installer;\r
 201 \r
 202         tbb::task_scheduler_init init;\r
 203         \r
 204         try \r
 205         {\r
 206                 // Configure environment properties from configuration.\r
 207                 caspar::env::configure("casparcg.config");\r
 208                                 \r
 209         #ifdef _DEBUG\r
 210                 if(caspar::env::properties().get("configuration.debugging.remote", false))\r
 211                         MessageBox(nullptr, TEXT("Now is the time to connect for remote debugging..."), TEXT("Debug"), MB_OK | MB_TOPMOST);\r
 212         #endif   \r
 213 \r
 214                 // Start logging to file.\r
 215                 caspar::log::add_file_sink(caspar::env::log_folder());                  \r
 216                 std::cout << "Logging [info] or higher severity to " << caspar::env::log_folder() << std::endl << std::endl;\r
 217                 \r
 218                 // Setup console window.\r
 219                 setup_console_window();\r
 220 \r
 221                 // Print environment information.\r
 222                 print_info();\r
 223                         \r
 224                 std::stringstream str;\r
 225                 boost::property_tree::xml_writer_settings<char> w(' ', 3);\r
 226                 boost::property_tree::write_xml(str, caspar::env::properties(), w);\r
 227                 CASPAR_LOG(info) << "casparcg.config:\n-----------------------------------------\n" << str.str().c_str() << "-----------------------------------------";\r
 228                                 \r
 229                 {\r
 230                         // Create server object which initializes channels, protocols and controllers.\r
 231                         caspar::server caspar_server;\r
 232                                 \r
 233                         // Create a amcp parser for console commands.\r
 234                         caspar::protocol::amcp::AMCPProtocolStrategy amcp(caspar_server.get_channels());\r
 235 \r
 236                         // Create a dummy client which prints amcp responses to console.\r
 237                         auto console_client = std::make_shared<caspar::IO::ConsoleClientInfo>();\r
 238 \r
 239                         std::wstring wcmd;\r
 240                         while(true)\r
 241                         {\r
 242                                 std::getline(std::wcin, wcmd); // TODO: It's blocking...\r
 243                                 \r
 244                                 caspar::to_upper(wcmd);\r
 245 \r
 246                                 if(wcmd == L"EXIT" || wcmd == L"Q" || wcmd == L"QUIT" || wcmd == L"BYE")\r
 247                                         break;\r
 248                                 \r
 249                                 // This is just dummy code for testing.\r
 250                                 if(wcmd.substr(0, 1) == L"1")\r
 251                                         wcmd = L"LOADBG 1-1 L" + wcmd.substr(1, wcmd.length()-1) + L" SLIDE 100 LOOP \r\nPLAY 1-1";\r
 252                                 else if(wcmd.substr(0, 1) == L"2")\r
 253                                         wcmd = L"MIXER 1-0 VIDEO IS_KEY 1";\r
 254                                 else if(wcmd.substr(0, 1) == L"3")\r
 255                                         wcmd = L"CG 1-2 ADD 1 BBTELEFONARE 1";\r
 256                                 else if(wcmd.substr(0, 1) == L"4")\r
 257                                         wcmd = L"PLAY 1-1 DV FILTER yadif=1:-1 LOOP";\r
 258                                 else if(wcmd.substr(0, 1) == L"5")\r
 259                                 {\r
 260                                         auto file = wcmd.substr(2, wcmd.length()-1);\r
 261                                         wcmd = L"PLAY 1-1 L" + file + L" LOOP\r\n" \r
 262                                                    L"PLAY 1-2 L" + file + L" LOOP\r\n" \r
 263                                                    L"PLAY 1-3 L" + file + L" LOOP\r\n"\r
 264                                                    L"PLAY 2-1 L" + file + L" LOOP\r\n" \r
 265                                                    L"PLAY 2-2 L" + file + L" LOOP\r\n" \r
 266                                                    L"PLAY 2-3 L" + file + L" LOOP\r\n";\r
 267                                 }\r
 268                                 else if(wcmd.substr(0, 1) == L"X")\r
 269                                 {\r
 270                                         int num = 0;\r
 271                                         std::wstring file;\r
 272                                         try\r
 273                                         {\r
 274                                                 num = boost::lexical_cast<int>(wcmd.substr(1, 2));\r
 275                                                 file = wcmd.substr(4, wcmd.length()-1);\r
 276                                         }\r
 277                                         catch(...)\r
 278                                         {\r
 279                                                 num = boost::lexical_cast<int>(wcmd.substr(1, 1));\r
 280                                                 file = wcmd.substr(3, wcmd.length()-1);\r
 281                                         }\r
 282 \r
 283                                         int n = 0;\r
 284                                         int num2 = num;\r
 285                                         while(num2 > 0)\r
 286                                         {\r
 287                                                 num2 >>= 1;\r
 288                                                 n++;\r
 289                                         }\r
 290 \r
 291                                         wcmd = L"MIXER 1 GRID L" + boost::lexical_cast<std::wstring>(n);\r
 292 \r
 293                                         for(int i = 1; i <= num; ++i)\r
 294                                                 wcmd += L"\r\nPLAY 1-" + boost::lexical_cast<std::wstring>(i) + L" L" + file + L" LOOP";// + L" SLIDE 100 LOOP";\r
 295                                 }\r
 296 \r
 297                                 wcmd += L"\r\n";\r
 298                                 amcp.Parse(wcmd.c_str(), wcmd.length(), console_client);\r
 299                         }       \r
 300                 }\r
 301                 Sleep(500);\r
 302                 CASPAR_LOG(info) << "Successfully shutdown CasparCG Server.";\r
 303                 system("pause");        \r
 304         }\r
 305         catch(boost::property_tree::file_parser_error&)\r
 306         {\r
 307                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 308                 CASPAR_LOG(fatal) << "Unhandled configuration error in main thread. Please check the configuration file (casparcg.config) for errors.";\r
 309                 system("pause");        \r
 310         }\r
 311         catch(caspar::gl::ogl_exception&)\r
 312         {\r
 313                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 314                 CASPAR_LOG(fatal) << "Unhandled OpenGL Error in main thread. Please try to update graphics drivers for OpenGL 3.0+ Support.";\r
 315         }\r
 316         catch(...)\r
 317         {\r
 318                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 319                 CASPAR_LOG(fatal) << "Unhandled exception in main thread. Please report this error on the CasparCG forums (www.casparcg.com/forum).";\r
 320         }       \r
 321         \r
 322         return 0;\r
 323 }