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 #include "resource.h"\r
  21 \r
  22 #include "server.h"\r
  23 \r
  24 // tbbmalloc_proxy: \r
  25 // Replace the standard memory allocation routines in Microsoft* C/C++ RTL \r
  26 // (malloc/free, global new/delete, etc.) with the TBB memory allocator. \r
  27 #include <tbb/tbbmalloc_proxy.h>\r
  28 \r
  29 #ifdef _DEBUG\r
  30         #define _CRTDBG_MAP_ALLOC\r
  31         #include <stdlib.h>\r
  32         #include <crtdbg.h>\r
  33 #endif\r
  34 \r
  35 #define NOMINMAX\r
  36 \r
  37 #include <windows.h>\r
  38 #include <atlbase.h>\r
  39 \r
  40 #include <core/mixer/gpu/ogl_device.h>\r
  41 \r
  42 #include <protocol/amcp/AMCPProtocolStrategy.h>\r
  43 \r
  44 #include <modules/bluefish/bluefish.h>\r
  45 #include <modules/decklink/decklink.h>\r
  46 #include <modules/flash/flash.h>\r
  47 #include <modules/ffmpeg/ffmpeg.h>\r
  48 #include <modules/image/image.h>\r
  49 \r
  50 #include <common/env.h>\r
  51 #include <common/exception/win32_exception.h>\r
  52 #include <common/exception/exceptions.h>\r
  53 #include <common/log/log.h>\r
  54 #include <common/os/windows/current_version.h>\r
  55 #include <common/os/windows/system_info.h>\r
  56 #include <common/utility/assert.h>\r
  57 \r
  58 #include <tbb/task_scheduler_observer.h>\r
  59 #include <tbb/task_scheduler_init.h>\r
  60 \r
  61 #include <boost/foreach.hpp>\r
  62 \r
  63 // NOTE: This is needed in order to make CComObject work since this is not a real ATL project.\r
  64 CComModule _AtlModule;\r
  65 extern __declspec(selectany) CAtlModule* _pAtlModule = &_AtlModule;\r
  66 \r
  67 void change_icon( const HICON hNewIcon )\r
  68 {\r
  69    auto hMod = ::LoadLibrary(L"Kernel32.dll"); \r
  70    typedef DWORD(__stdcall *SCI)(HICON);\r
  71    auto pfnSetConsoleIcon = reinterpret_cast<SCI>(::GetProcAddress(hMod, "SetConsoleIcon")); \r
  72    pfnSetConsoleIcon(hNewIcon); \r
  73    ::FreeLibrary(hMod);\r
  74 }\r
  75 \r
  76 void setup_console_window()\r
  77 {        \r
  78         auto hOut = GetStdHandle(STD_OUTPUT_HANDLE);\r
  79 \r
  80         // Disable close button in console to avoid shutdown without cleanup.\r
  81         EnableMenuItem(GetSystemMenu(GetConsoleWindow(), FALSE), SC_CLOSE , MF_GRAYED);\r
  82         DrawMenuBar(GetConsoleWindow());\r
  83 \r
  84         // Configure console size and position.\r
  85         auto coord = GetLargestConsoleWindowSize(hOut);\r
  86         coord.X /= 2;\r
  87 \r
  88         SetConsoleScreenBufferSize(hOut, coord);\r
  89 \r
  90         SMALL_RECT DisplayArea = {0, 0, 0, 0};\r
  91         DisplayArea.Right = coord.X-1;\r
  92         DisplayArea.Bottom = (coord.Y-1)/2;\r
  93         SetConsoleWindowInfo(hOut, TRUE, &DisplayArea);\r
  94                  \r
  95         change_icon(::LoadIcon(GetModuleHandle(0), MAKEINTRESOURCE(101)));\r
  96 \r
  97         // Set console title.\r
  98         std::wstringstream str;\r
  99         str << "CasparCG Server " << caspar::env::version();\r
 100 #ifdef COMPILE_RELEASE\r
 101         str << " Release";\r
 102 #elif  COMPILE_PROFILE\r
 103         str << " Profile";\r
 104 #elif  COMPILE_DEVELOP\r
 105         str << " Develop";\r
 106 #elif  COMPILE_DEBUG\r
 107         str << " Debug";\r
 108 #endif\r
 109         SetConsoleTitle(str.str().c_str());\r
 110 }\r
 111 \r
 112 void print_info()\r
 113 {\r
 114         CASPAR_LOG(info) << L"Copyright (c) 2010 Sveriges Television AB, www.casparcg.com, <info@casparcg.com>";\r
 115         CASPAR_LOG(info) << L"Starting CasparCG Video and Graphics Playout Server " << caspar::env::version();\r
 116         CASPAR_LOG(info) << L"on " << caspar::get_win_product_name() << L" " << caspar::get_win_sp_version();\r
 117         CASPAR_LOG(info) << caspar::get_cpu_info();\r
 118         CASPAR_LOG(info) << caspar::get_system_product_name();\r
 119         CASPAR_LOG(info) << L"Flash " << caspar::get_flash_version();\r
 120         CASPAR_LOG(info) << L"Flash-Template-Host " << caspar::get_cg_version();\r
 121         CASPAR_LOG(info) << L"FreeImage " << caspar::get_image_version();\r
 122         \r
 123         CASPAR_LOG(info) << L"Decklink " << caspar::get_decklink_version();\r
 124         BOOST_FOREACH(auto& device, caspar::get_decklink_device_list())\r
 125                 CASPAR_LOG(info) << device;\r
 126                 \r
 127         CASPAR_LOG(info) << L"Bluefish " << caspar::get_bluefish_version();\r
 128         BOOST_FOREACH(auto& device, caspar::get_bluefish_device_list())\r
 129                 CASPAR_LOG(info) << device;\r
 130 \r
 131         CASPAR_LOG(info) << L"FFMPEG-avcodec "  << caspar::get_avcodec_version();\r
 132         CASPAR_LOG(info) << L"FFMPEG-avformat " << caspar::get_avformat_version();\r
 133         CASPAR_LOG(info) << L"FFMPEG-avfilter " << caspar::get_avfilter_version();\r
 134         CASPAR_LOG(info) << L"FFMPEG-avutil " << caspar::get_avutil_version();\r
 135         CASPAR_LOG(info) << L"FFMPEG-swscale "  << caspar::get_swscale_version();\r
 136         CASPAR_LOG(info) << L"OpenGL " << caspar::core::ogl_device::get_version() << "\n\n";\r
 137 }\r
 138  \r
 139 int main(int argc, wchar_t* argv[])\r
 140 {       \r
 141         static_assert(sizeof(void*) == 4, "64-bit code generation is not supported.");\r
 142         \r
 143         CASPAR_LOG(info) << L"THIS IS AN ALPHA BUILD";\r
 144 \r
 145         // Set debug mode.\r
 146         #ifdef _DEBUG\r
 147                 _CrtSetDbgFlag ( _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF | _CRTDBG_CHECK_ALWAYS_DF );\r
 148                 _CrtSetReportMode( _CRT_ERROR, _CRTDBG_MODE_DEBUG );\r
 149                 _CrtSetReportMode( _CRT_WARN, _CRTDBG_MODE_DEBUG );\r
 150                 _CrtSetReportMode( _CRT_ASSERT, _CRTDBG_MODE_DEBUG );\r
 151         #endif\r
 152 \r
 153         // Increase process priotity.\r
 154         SetPriorityClass(GetCurrentProcess(), ABOVE_NORMAL_PRIORITY_CLASS);\r
 155 \r
 156         // Install structured exception handler.\r
 157         caspar::win32_exception::install_handler();\r
 158                         \r
 159         // Increase time precision. This will increase accuracy of function like Sleep(1) from 10 ms to 1 ms.\r
 160         struct inc_prec\r
 161         {\r
 162                 inc_prec(){timeBeginPeriod(1);}\r
 163                 ~inc_prec(){timeEndPeriod(1);}\r
 164         } inc_prec;     \r
 165 \r
 166         // Install unstructured exception handlers into all tbb threads.\r
 167         struct tbb_thread_installer : public tbb::task_scheduler_observer\r
 168         {\r
 169                 tbb_thread_installer(){observe(true);}\r
 170                 void on_scheduler_entry(bool is_worker){caspar::win32_exception::install_handler();}\r
 171         } tbb_thread_installer;\r
 172         \r
 173         try \r
 174         {\r
 175                 // Configure environment properties from configuration.\r
 176                 caspar::env::configure("casparcg.config");\r
 177 \r
 178         #ifdef _DEBUG\r
 179                 if(caspar::env::properties().get("configuration.debugging.remote", false))\r
 180                         MessageBox(nullptr, TEXT("Now is the time to connect for remote debugging..."), TEXT("Debug"), MB_OK | MB_TOPMOST);\r
 181         #endif   \r
 182 \r
 183                 // Start logging to file.\r
 184                 caspar::log::add_file_sink(caspar::env::log_folder());\r
 185                 \r
 186                 // Setup console window.\r
 187                 setup_console_window();\r
 188 \r
 189                 // Print environment information.\r
 190                 print_info();\r
 191                                 \r
 192                 // Create server object which initializes channels, protocols and controllers.\r
 193                 caspar::server caspar_server;\r
 194                                 \r
 195                 // Create a amcp parser for console commands.\r
 196                 caspar::protocol::amcp::AMCPProtocolStrategy amcp(caspar_server.get_channels());\r
 197 \r
 198                 // Create a dummy client which prints amcp responses to console.\r
 199                 auto dummy = std::make_shared<caspar::IO::ConsoleClientInfo>();\r
 200 \r
 201                 bool is_running = true;\r
 202                 while(is_running)\r
 203                 {\r
 204                         std::wstring wcmd;\r
 205                         std::getline(std::wcin, wcmd); // TODO: It's blocking...\r
 206 \r
 207                         is_running = wcmd != L"exit" && wcmd != L"q";\r
 208                         if(wcmd.substr(0, 1) == L"1")\r
 209                                 wcmd = L"LOADBG 1-1 " + wcmd.substr(1, wcmd.length()-1) + L" SLIDE 100 LOOP \r\nPLAY 1-1";\r
 210                         else if(wcmd.substr(0, 1) == L"2")\r
 211                                 wcmd = L"MIXER 1-0 VIDEO IS_KEY 1";\r
 212                         else if(wcmd.substr(0, 1) == L"3")\r
 213                                 wcmd = L"CG 1-2 ADD 1 BBTELEFONARE 1";\r
 214                         else if(wcmd.substr(0, 1) == L"4")\r
 215                                 wcmd = L"PLAY 1-1 DV SEEK 350 FILTER yadif=1:-1 LOOP";\r
 216                         else if(wcmd.substr(0, 1) == L"X")\r
 217                         {\r
 218                                 int num = 0;\r
 219                                 std::wstring file;\r
 220                                 try\r
 221                                 {\r
 222                                         num = boost::lexical_cast<int>(wcmd.substr(1, 2));\r
 223                                         file = wcmd.substr(4, wcmd.length()-1);\r
 224                                 }\r
 225                                 catch(...)\r
 226                                 {\r
 227                                         num = boost::lexical_cast<int>(wcmd.substr(1, 1));\r
 228                                         file = wcmd.substr(3, wcmd.length()-1);\r
 229                                 }\r
 230 \r
 231                                 int n = 0;\r
 232                                 int num2 = num;\r
 233                                 while(num2 > 0)\r
 234                                 {\r
 235                                         num2 >>= 1;\r
 236                                         n++;\r
 237                                 }\r
 238 \r
 239                                 wcmd = L"MIXER 1 VIDEO GRID " + boost::lexical_cast<std::wstring>(n);\r
 240 \r
 241                                 for(int i = 1; i <= num; ++i)\r
 242                                         wcmd += L"\r\nPLAY 1-" + boost::lexical_cast<std::wstring>(i) + L" " + file + L" LOOP";// + L" SLIDE 100 LOOP";\r
 243                         }\r
 244 \r
 245                         wcmd += L"\r\n";\r
 246                         amcp.Parse(wcmd.c_str(), wcmd.length(), dummy);\r
 247                 }\r
 248         }\r
 249         catch(...)\r
 250         {\r
 251                 CASPAR_LOG(fatal) << "UNHANDLED EXCEPTION in main thread.";\r
 252                 CASPAR_LOG_CURRENT_EXCEPTION();\r
 253         }       \r
 254         \r
 255         CASPAR_LOG(info) << "Successfully shutdown CasparCG Server.";\r
 256         Sleep(100); // CAPSAR_LOG is asynchronous. Try to get text in correct order.\r
 257         system("pause");\r
 258         return 0;\r
 259 }