git.sesse.net Git - casparcg/blob - modules/decklink/decklink.cpp

   1 /*\r
   2 * Copyright 2013 Sveriges Television AB http://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 #include "stdafx.h"\r
  23 \r
  24 #include "decklink.h"\r
  25 #include "util/util.h"\r
  26 \r
  27 #include "consumer/decklink_consumer.h"\r
  28 #include "consumer/blocking_decklink_consumer.h"\r
  29 #include "producer/decklink_producer.h"\r
  30 \r
  31 #include <core/parameters/parameters.h>\r
  32 #include <core/consumer/frame_consumer.h>\r
  33 #include <core/producer/frame_producer.h>\r
  34 \r
  35 #include "interop/DeckLinkAPI_h.h"\r
  36 \r
  37 #pragma warning(push)\r
  38 #pragma warning(disable : 4996)\r
  39 \r
  40         #include <atlbase.h>\r
  41 \r
  42         #include <atlcom.h>\r
  43         #include <atlhost.h>\r
  44 \r
  45 #pragma warning(push)\r
  46 \r
  47 namespace caspar { namespace decklink {\r
  48 \r
  49 void init()\r
  50 {\r
  51         struct co_init\r
  52         {\r
  53                 co_init(){::CoInitialize(nullptr);}\r
  54                 ~co_init(){::CoUninitialize();}\r
  55         } init;\r
  56         \r
  57         CComPtr<IDeckLinkIterator> pDecklinkIterator;\r
  58         if(FAILED(pDecklinkIterator.CoCreateInstance(CLSID_CDeckLinkIterator)))         \r
  59                 return;\r
  60                 \r
  61         core::register_consumer_factory([](const core::parameters& params){return decklink::create_consumer(params);});\r
  62         core::register_consumer_factory([](const core::parameters& params){return decklink::create_blocking_consumer(params);});\r
  63         core::register_producer_factory(create_producer);\r
  64 }\r
  65 \r
  66 std::wstring get_version() \r
  67 {\r
  68         std::wstring version = L"Not found";\r
  69         \r
  70         struct co_init\r
  71         {\r
  72                 co_init(){::CoInitialize(nullptr);}\r
  73                 ~co_init(){::CoUninitialize();}\r
  74         } init;\r
  75 \r
  76         try\r
  77         {\r
  78                 CComPtr<IDeckLinkIterator> pDecklinkIterator;\r
  79                 if(SUCCEEDED(pDecklinkIterator.CoCreateInstance(CLSID_CDeckLinkIterator)))              \r
  80                         version = get_version(pDecklinkIterator);\r
  81         }\r
  82         catch(...){}\r
  83 \r
  84         return version;\r
  85 }\r
  86 \r
  87 std::vector<std::wstring> get_device_list()\r
  88 {\r
  89         std::vector<std::wstring> devices;\r
  90         \r
  91         struct co_init\r
  92         {\r
  93                 co_init(){::CoInitialize(nullptr);}\r
  94                 ~co_init(){::CoUninitialize();}\r
  95         } init;\r
  96 \r
  97         try\r
  98         {\r
  99                 CComPtr<IDeckLinkIterator> pDecklinkIterator;\r
 100                 if(SUCCEEDED(pDecklinkIterator.CoCreateInstance(CLSID_CDeckLinkIterator)))\r
 101                 {               \r
 102                         IDeckLink* decklink;\r
 103                         for(int n = 1; pDecklinkIterator->Next(&decklink) == S_OK; ++n) \r
 104                         {\r
 105                                 BSTR model_name = L"Unknown";\r
 106                                 decklink->GetModelName(&model_name);\r
 107                                 decklink->Release();\r
 108                                 devices.push_back(std::wstring(model_name) + L" [" + boost::lexical_cast<std::wstring>(n) + L"]");      \r
 109                         }\r
 110                 }\r
 111         }\r
 112         catch(...){}\r
 113 \r
 114         return devices;\r
 115 }\r
 116 \r
 117 }}