libavdevice/decklink_common.cpp

   1 /*
   2  * Blackmagic DeckLink output
   3  * Copyright (c) 2013-2014 Ramiro Polla, Luca Barbato, Deti Fliegl
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 /* Include internal.h first to avoid conflict between winsock.h (used by
  23  * DeckLink headers) and winsock2.h (used by libavformat) in MSVC++ builds */
  24 extern "C" {
  25 #include "libavformat/internal.h"
  26 }
  27
  28 #include <DeckLinkAPI.h>
  29 #ifdef _WIN32
  30 #include <DeckLinkAPI_i.c>
  31 #else
  32 /* The file provided by the SDK is known to be missing prototypes, which doesn't
  33    cause issues with GCC since the warning doesn't apply to C++ files.  However
  34    Clang does complain (and warnings are treated as errors), so suppress the
  35    warning just for this one file */
  36 #ifdef __clang__
  37 #pragma clang diagnostic push
  38 #pragma clang diagnostic ignored "-Wmissing-prototypes"
  39 #endif
  40 #include <DeckLinkAPIDispatch.cpp>
  41 #ifdef __clang__
  42 #pragma clang diagnostic pop
  43 #endif
  44 #endif
  45
  46 extern "C" {
  47 #include "libavformat/avformat.h"
  48 #include "libavutil/imgutils.h"
  49 #include "libavutil/intreadwrite.h"
  50 #include "libavutil/bswap.h"
  51 #include "avdevice.h"
  52 }
  53
  54 #include "decklink_common.h"
  55
  56 static IDeckLinkIterator *decklink_create_iterator(AVFormatContext *avctx)
  57 {
  58     IDeckLinkIterator *iter;
  59
  60 #ifdef _WIN32
  61     if (CoInitialize(NULL) < 0) {
  62         av_log(avctx, AV_LOG_ERROR, "COM initialization failed.\n");
  63         return NULL;
  64     }
  65
  66     if (CoCreateInstance(CLSID_CDeckLinkIterator, NULL, CLSCTX_ALL,
  67                          IID_IDeckLinkIterator, (void**) &iter) != S_OK) {
  68         iter = NULL;
  69     }
  70 #else
  71     iter = CreateDeckLinkIteratorInstance();
  72 #endif
  73     if (!iter) {
  74         av_log(avctx, AV_LOG_ERROR, "Could not create DeckLink iterator. "
  75                                     "Make sure you have DeckLink drivers " BLACKMAGIC_DECKLINK_API_VERSION_STRING " or newer installed.\n");
  76     } else {
  77         IDeckLinkAPIInformation *api;
  78         int64_t version;
  79 #ifdef _WIN32
  80         if (CoCreateInstance(CLSID_CDeckLinkAPIInformation, NULL, CLSCTX_ALL,
  81                              IID_IDeckLinkAPIInformation, (void**) &api) != S_OK) {
  82             api = NULL;
  83         }
  84 #else
  85         api = CreateDeckLinkAPIInformationInstance();
  86 #endif
  87         if (api && api->GetInt(BMDDeckLinkAPIVersion, &version) == S_OK) {
  88             if (version < BLACKMAGIC_DECKLINK_API_VERSION)
  89                 av_log(avctx, AV_LOG_WARNING, "Installed DeckLink drivers are too old and may be incompatible with the SDK this module was built against. "
  90                                               "Make sure you have DeckLink drivers " BLACKMAGIC_DECKLINK_API_VERSION_STRING " or newer installed.\n");
  91         } else {
  92             av_log(avctx, AV_LOG_ERROR, "Failed to check installed DeckLink API version.\n");
  93         }
  94         if (api)
  95             api->Release();
  96     }
  97
  98     return iter;
  99 }
 100
 101 static int decklink_get_attr_string(IDeckLink *dl, BMDDeckLinkAttributeID cfg_id, const char **s)
 102 {
 103     DECKLINK_STR tmp;
 104     HRESULT hr;
 105     IDeckLinkProfileAttributes *attr;
 106     *s = NULL;
 107     if (dl->QueryInterface(IID_IDeckLinkProfileAttributes, (void **)&attr) != S_OK)
 108         return AVERROR_EXTERNAL;
 109     hr = attr->GetString(cfg_id, &tmp);
 110     attr->Release();
 111     if (hr == S_OK) {
 112         *s = DECKLINK_STRDUP(tmp);
 113         DECKLINK_FREE(tmp);
 114         if (!*s)
 115             return AVERROR(ENOMEM);
 116     } else if (hr == E_FAIL) {
 117         return AVERROR_EXTERNAL;
 118     }
 119     return 0;
 120 }
 121
 122 static int decklink_select_input(AVFormatContext *avctx, BMDDeckLinkConfigurationID cfg_id)
 123 {
 124     struct decklink_cctx *cctx = (struct decklink_cctx *)avctx->priv_data;
 125     struct decklink_ctx *ctx = (struct decklink_ctx *)cctx->ctx;
 126     BMDDeckLinkAttributeID attr_id = (cfg_id == bmdDeckLinkConfigAudioInputConnection) ? BMDDeckLinkAudioInputConnections : BMDDeckLinkVideoInputConnections;
 127     int64_t bmd_input              = (cfg_id == bmdDeckLinkConfigAudioInputConnection) ? (int64_t)ctx->audio_input : (int64_t)ctx->video_input;
 128     const char *type_name          = (cfg_id == bmdDeckLinkConfigAudioInputConnection) ? "audio" : "video";
 129     int64_t supported_connections = 0;
 130     HRESULT res;
 131
 132     if (bmd_input) {
 133         res = ctx->attr->GetInt(attr_id, &supported_connections);
 134         if (res != S_OK) {
 135             av_log(avctx, AV_LOG_ERROR, "Failed to query supported %s inputs.\n", type_name);
 136             return AVERROR_EXTERNAL;
 137         }
 138         if ((supported_connections & bmd_input) != bmd_input) {
 139             av_log(avctx, AV_LOG_ERROR, "Device does not support selected %s input.\n", type_name);
 140             return AVERROR(ENOSYS);
 141         }
 142         res = ctx->cfg->SetInt(cfg_id, bmd_input);
 143         if (res != S_OK) {
 144             av_log(avctx, AV_LOG_ERROR, "Failed to select %s input.\n", type_name);
 145             return AVERROR_EXTERNAL;
 146         }
 147     }
 148     return 0;
 149 }
 150
 151 static DECKLINK_BOOL field_order_eq(enum AVFieldOrder field_order, BMDFieldDominance bmd_field_order)
 152 {
 153     if (field_order == AV_FIELD_UNKNOWN)
 154         return true;
 155     if ((field_order == AV_FIELD_TT || field_order == AV_FIELD_TB) && bmd_field_order == bmdUpperFieldFirst)
 156         return true;
 157     if ((field_order == AV_FIELD_BB || field_order == AV_FIELD_BT) && bmd_field_order == bmdLowerFieldFirst)
 158         return true;
 159     if (field_order == AV_FIELD_PROGRESSIVE && (bmd_field_order == bmdProgressiveFrame || bmd_field_order == bmdProgressiveSegmentedFrame))
 160         return true;
 161     return false;
 162 }
 163
 164 int ff_decklink_set_configs(AVFormatContext *avctx,
 165                             decklink_direction_t direction) {
 166     struct decklink_cctx *cctx = (struct decklink_cctx *)avctx->priv_data;
 167     struct decklink_ctx *ctx = (struct decklink_ctx *)cctx->ctx;
 168     HRESULT res;
 169
 170     if (ctx->duplex_mode) {
 171         DECKLINK_BOOL duplex_supported = false;
 172
 173 #if BLACKMAGIC_DECKLINK_API_VERSION >= 0x0b000000
 174         IDeckLinkProfileManager *manager = NULL;
 175         if (ctx->dl->QueryInterface(IID_IDeckLinkProfileManager, (void **)&manager) == S_OK)
 176             duplex_supported = true;
 177 #else
 178         if (ctx->attr->GetFlag(BMDDeckLinkSupportsDuplexModeConfiguration, &duplex_supported) != S_OK)
 179             duplex_supported = false;
 180 #endif
 181
 182         if (duplex_supported) {
 183 #if BLACKMAGIC_DECKLINK_API_VERSION >= 0x0b000000
 184             IDeckLinkProfile *profile = NULL;
 185             BMDProfileID bmd_profile_id;
 186
 187             if (ctx->duplex_mode < 0 || ctx->duplex_mode >= FF_ARRAY_ELEMS(decklink_profile_id_map))
 188                 return EINVAL;
 189             bmd_profile_id = decklink_profile_id_map[ctx->duplex_mode];
 190             res = manager->GetProfile(bmd_profile_id, &profile);
 191             if (res == S_OK) {
 192                 res = profile->SetActive();
 193                 profile->Release();
 194             }
 195             manager->Release();
 196 #else
 197             res = ctx->cfg->SetInt(bmdDeckLinkConfigDuplexMode, ctx->duplex_mode == 2 ? bmdDuplexModeFull : bmdDuplexModeHalf);
 198 #endif
 199             if (res != S_OK)
 200                 av_log(avctx, AV_LOG_WARNING, "Setting duplex mode failed.\n");
 201             else
 202                 av_log(avctx, AV_LOG_VERBOSE, "Successfully set duplex mode to %s duplex.\n", ctx->duplex_mode == 2 || ctx->duplex_mode == 4 ? "full" : "half");
 203         } else {
 204             av_log(avctx, AV_LOG_WARNING, "Unable to set duplex mode, because it is not supported.\n");
 205         }
 206     }
 207     if (direction == DIRECTION_IN) {
 208         int ret;
 209         ret = decklink_select_input(avctx, bmdDeckLinkConfigAudioInputConnection);
 210         if (ret < 0)
 211             return ret;
 212         ret = decklink_select_input(avctx, bmdDeckLinkConfigVideoInputConnection);
 213         if (ret < 0)
 214             return ret;
 215     }
 216     if (direction == DIRECTION_OUT && cctx->timing_offset != INT_MIN) {
 217         res = ctx->cfg->SetInt(bmdDeckLinkConfigReferenceInputTimingOffset, cctx->timing_offset);
 218         if (res != S_OK)
 219             av_log(avctx, AV_LOG_WARNING, "Setting timing offset failed.\n");
 220     }
 221
 222     if (direction == DIRECTION_OUT && ctx->link > 0) {
 223         res = ctx->cfg->SetInt(bmdDeckLinkConfigSDIOutputLinkConfiguration, ctx->link);
 224         if (res != S_OK)
 225             av_log(avctx, AV_LOG_WARNING, "Setting link configuration failed.\n");
 226         else
 227             av_log(avctx, AV_LOG_VERBOSE, "Successfully set link configuration: 0x%x.\n", ctx->link);
 228         if (ctx->link == bmdLinkConfigurationQuadLink && cctx->sqd >= 0) {
 229             res = ctx->cfg->SetFlag(bmdDeckLinkConfigQuadLinkSDIVideoOutputSquareDivisionSplit, cctx->sqd);
 230             if (res != S_OK)
 231                 av_log(avctx, AV_LOG_WARNING, "Setting SquareDivisionSplit failed.\n");
 232             else
 233                 av_log(avctx, AV_LOG_VERBOSE, "Successfully set SquareDivisionSplit.\n");
 234         }
 235     }
 236
 237     if (direction == DIRECTION_OUT && cctx->level_a >= 0) {
 238         DECKLINK_BOOL level_a_supported = false;
 239
 240         if (ctx->attr->GetFlag(BMDDeckLinkSupportsSMPTELevelAOutput, &level_a_supported) != S_OK)
 241             level_a_supported = false;
 242
 243         if (level_a_supported) {
 244             res = ctx->cfg->SetFlag(bmdDeckLinkConfigSMPTELevelAOutput, cctx->level_a);
 245             if (res != S_OK)
 246                 av_log(avctx, AV_LOG_WARNING, "Setting SMPTE levelA failed.\n");
 247             else
 248                 av_log(avctx, AV_LOG_VERBOSE, "Successfully set SMPTE levelA.\n");
 249         } else {
 250             av_log(avctx, AV_LOG_WARNING, "Unable to set SMPTE levelA mode, because it is not supported.\n");
 251         }
 252     }
 253
 254     return 0;
 255 }
 256
 257 int ff_decklink_set_format(AVFormatContext *avctx,
 258                                int width, int height,
 259                                int tb_num, int tb_den,
 260                                enum AVFieldOrder field_order,
 261                                decklink_direction_t direction)
 262 {
 263     struct decklink_cctx *cctx = (struct decklink_cctx *)avctx->priv_data;
 264     struct decklink_ctx *ctx = (struct decklink_ctx *)cctx->ctx;
 265 #if BLACKMAGIC_DECKLINK_API_VERSION >= 0x0b000000
 266     DECKLINK_BOOL support;
 267 #else
 268     BMDDisplayModeSupport support;
 269 #endif
 270     IDeckLinkDisplayModeIterator *itermode;
 271     IDeckLinkDisplayMode *mode;
 272     int i = 1;
 273     HRESULT res;
 274
 275     av_log(avctx, AV_LOG_DEBUG, "Trying to find mode for frame size %dx%d, frame timing %d/%d, field order %d, direction %d, format code %s\n",
 276         width, height, tb_num, tb_den, field_order, direction, cctx->format_code ? cctx->format_code : "(unset)");
 277
 278     if (direction == DIRECTION_IN) {
 279         res = ctx->dli->GetDisplayModeIterator (&itermode);
 280     } else {
 281         res = ctx->dlo->GetDisplayModeIterator (&itermode);
 282     }
 283
 284     if (res!= S_OK) {
 285             av_log(avctx, AV_LOG_ERROR, "Could not get Display Mode Iterator\n");
 286             return AVERROR(EIO);
 287     }
 288
 289     char format_buf[] = "    ";
 290     if (cctx->format_code)
 291         memcpy(format_buf, cctx->format_code, FFMIN(strlen(cctx->format_code), sizeof(format_buf)));
 292     BMDDisplayMode target_mode = (BMDDisplayMode)AV_RB32(format_buf);
 293     AVRational target_tb = av_make_q(tb_num, tb_den);
 294     ctx->bmd_mode = bmdModeUnknown;
 295     while ((ctx->bmd_mode == bmdModeUnknown) && itermode->Next(&mode) == S_OK) {
 296         BMDTimeValue bmd_tb_num, bmd_tb_den;
 297         int bmd_width  = mode->GetWidth();
 298         int bmd_height = mode->GetHeight();
 299         BMDDisplayMode bmd_mode = mode->GetDisplayMode();
 300         BMDFieldDominance bmd_field_dominance = mode->GetFieldDominance();
 301
 302         mode->GetFrameRate(&bmd_tb_num, &bmd_tb_den);
 303         AVRational mode_tb = av_make_q(bmd_tb_num, bmd_tb_den);
 304
 305         if ((bmd_width == width &&
 306              bmd_height == height &&
 307              !av_cmp_q(mode_tb, target_tb) &&
 308              field_order_eq(field_order, bmd_field_dominance))
 309              || target_mode == bmd_mode) {
 310             ctx->bmd_mode   = bmd_mode;
 311             ctx->bmd_width  = bmd_width;
 312             ctx->bmd_height = bmd_height;
 313             ctx->bmd_tb_den = bmd_tb_den;
 314             ctx->bmd_tb_num = bmd_tb_num;
 315             ctx->bmd_field_dominance = bmd_field_dominance;
 316             av_log(avctx, AV_LOG_INFO, "Found Decklink mode %d x %d with rate %.2f%s\n",
 317                 bmd_width, bmd_height, 1/av_q2d(mode_tb),
 318                 (ctx->bmd_field_dominance==bmdLowerFieldFirst || ctx->bmd_field_dominance==bmdUpperFieldFirst)?"(i)":"");
 319         }
 320
 321         mode->Release();
 322         i++;
 323     }
 324
 325     itermode->Release();
 326
 327     if (ctx->bmd_mode == bmdModeUnknown)
 328         return -1;
 329
 330 #if BLACKMAGIC_DECKLINK_API_VERSION >= 0x0b050000
 331     if (direction == DIRECTION_IN) {
 332         BMDDisplayMode actualMode = ctx->bmd_mode;
 333         if (ctx->dli->DoesSupportVideoMode(ctx->video_input, ctx->bmd_mode, ctx->raw_format,
 334                                            bmdNoVideoInputConversion, bmdSupportedVideoModeDefault,
 335                                            &actualMode, &support) != S_OK || !support || ctx->bmd_mode != actualMode)
 336             return -1;
 337     } else {
 338         BMDDisplayMode actualMode = ctx->bmd_mode;
 339         if (ctx->dlo->DoesSupportVideoMode(bmdVideoConnectionUnspecified, ctx->bmd_mode, ctx->raw_format,
 340                                            bmdNoVideoOutputConversion, bmdSupportedVideoModeDefault,
 341                                            &actualMode, &support) != S_OK || !support || ctx->bmd_mode != actualMode)
 342             return -1;
 343     }
 344     return 0;
 345 #elif BLACKMAGIC_DECKLINK_API_VERSION >= 0x0b000000
 346     if (direction == DIRECTION_IN) {
 347         if (ctx->dli->DoesSupportVideoMode(ctx->video_input, ctx->bmd_mode, ctx->raw_format,
 348                                            bmdSupportedVideoModeDefault,
 349                                            &support) != S_OK)
 350             return -1;
 351     } else {
 352         BMDDisplayMode actualMode = ctx->bmd_mode;
 353         if (ctx->dlo->DoesSupportVideoMode(bmdVideoConnectionUnspecified, ctx->bmd_mode, ctx->raw_format,
 354                                            bmdSupportedVideoModeDefault,
 355                                            &actualMode, &support) != S_OK || !support || ctx->bmd_mode != actualMode) {
 356             return -1;
 357         }
 358
 359     }
 360     if (support)
 361         return 0;
 362 #else
 363     if (direction == DIRECTION_IN) {
 364         if (ctx->dli->DoesSupportVideoMode(ctx->bmd_mode, ctx->raw_format,
 365                                            bmdVideoOutputFlagDefault,
 366                                            &support, NULL) != S_OK)
 367             return -1;
 368     } else {
 369         if (!ctx->supports_vanc || ctx->dlo->DoesSupportVideoMode(ctx->bmd_mode, ctx->raw_format,
 370                                                                   bmdVideoOutputVANC,
 371                                                                   &support, NULL) != S_OK || support != bmdDisplayModeSupported) {
 372             /* Try without VANC enabled */
 373             if (ctx->dlo->DoesSupportVideoMode(ctx->bmd_mode, ctx->raw_format,
 374                                                bmdVideoOutputFlagDefault,
 375                                                &support, NULL) != S_OK) {
 376                 return -1;
 377             }
 378             ctx->supports_vanc = 0;
 379         }
 380
 381     }
 382     if (support == bmdDisplayModeSupported)
 383         return 0;
 384 #endif
 385
 386     return -1;
 387 }
 388
 389 int ff_decklink_set_format(AVFormatContext *avctx, decklink_direction_t direction) {
 390     return ff_decklink_set_format(avctx, 0, 0, 0, 0, AV_FIELD_UNKNOWN, direction);
 391 }
 392
 393 void ff_decklink_packet_queue_init(AVFormatContext *avctx, DecklinkPacketQueue *q, int64_t queue_size)
 394 {
 395     memset(q, 0, sizeof(DecklinkPacketQueue));
 396     pthread_mutex_init(&q->mutex, NULL);
 397     pthread_cond_init(&q->cond, NULL);
 398     q->avctx = avctx;
 399     q->max_q_size = queue_size;
 400 }
 401
 402 void ff_decklink_packet_queue_flush(DecklinkPacketQueue *q)
 403 {
 404     AVPacket pkt;
 405
 406     pthread_mutex_lock(&q->mutex);
 407     while (avpriv_packet_list_get(&q->pkt_list, &pkt) == 0) {
 408         av_packet_unref(&pkt);
 409     }
 410     q->nb_packets = 0;
 411     q->size       = 0;
 412     pthread_mutex_unlock(&q->mutex);
 413 }
 414
 415 void ff_decklink_packet_queue_end(DecklinkPacketQueue *q)
 416 {
 417     ff_decklink_packet_queue_flush(q);
 418     pthread_mutex_destroy(&q->mutex);
 419     pthread_cond_destroy(&q->cond);
 420 }
 421
 422 unsigned long long ff_decklink_packet_queue_size(DecklinkPacketQueue *q)
 423 {
 424     unsigned long long size;
 425     pthread_mutex_lock(&q->mutex);
 426     size = q->size;
 427     pthread_mutex_unlock(&q->mutex);
 428     return size;
 429 }
 430
 431 int ff_decklink_packet_queue_put(DecklinkPacketQueue *q, AVPacket *pkt)
 432 {
 433     int pkt_size = pkt->size;
 434     int ret;
 435
 436     // Drop Packet if queue size is > maximum queue size
 437     if (ff_decklink_packet_queue_size(q) > (uint64_t)q->max_q_size) {
 438         av_packet_unref(pkt);
 439         av_log(q->avctx, AV_LOG_WARNING,  "Decklink input buffer overrun!\n");
 440         return -1;
 441     }
 442     /* ensure the packet is reference counted */
 443     if (av_packet_make_refcounted(pkt) < 0) {
 444         av_packet_unref(pkt);
 445         return -1;
 446     }
 447
 448     pthread_mutex_lock(&q->mutex);
 449
 450     ret = avpriv_packet_list_put(&q->pkt_list, pkt, NULL, 0);
 451     if (ret == 0) {
 452         q->nb_packets++;
 453         q->size += pkt_size + sizeof(AVPacket);
 454         pthread_cond_signal(&q->cond);
 455     } else {
 456         av_packet_unref(pkt);
 457     }
 458
 459     pthread_mutex_unlock(&q->mutex);
 460     return ret;
 461 }
 462
 463 int ff_decklink_packet_queue_get(DecklinkPacketQueue *q, AVPacket *pkt, int block)
 464 {
 465     int ret;
 466
 467     pthread_mutex_lock(&q->mutex);
 468
 469     for (;; ) {
 470         ret = avpriv_packet_list_get(&q->pkt_list, pkt);
 471         if (ret == 0) {
 472             q->nb_packets--;
 473             q->size -= pkt->size + sizeof(AVPacket);
 474             ret = 1;
 475             break;
 476         } else if (!block) {
 477             ret = 0;
 478             break;
 479         } else {
 480             pthread_cond_wait(&q->cond, &q->mutex);
 481         }
 482     }
 483     pthread_mutex_unlock(&q->mutex);
 484     return ret;
 485 }
 486
 487 int64_t ff_decklink_packet_queue_peekpts(DecklinkPacketQueue *q)
 488 {
 489     PacketListEntry *pkt1;
 490     int64_t pts = -1;
 491
 492     pthread_mutex_lock(&q->mutex);
 493     pkt1 = q->pkt_list.head;
 494     if (pkt1) {
 495         pts = pkt1->pkt.pts;
 496     }
 497     pthread_mutex_unlock(&q->mutex);
 498
 499     return pts;
 500 }
 501
 502
 503 int ff_decklink_list_devices(AVFormatContext *avctx,
 504                              struct AVDeviceInfoList *device_list,
 505                              int show_inputs, int show_outputs)
 506 {
 507     IDeckLink *dl = NULL;
 508     IDeckLinkIterator *iter = decklink_create_iterator(avctx);
 509     int ret = 0;
 510
 511     if (!iter)
 512         return AVERROR(EIO);
 513
 514     while (ret == 0 && iter->Next(&dl) == S_OK) {
 515         IDeckLinkOutput *output_config;
 516         IDeckLinkInput *input_config;
 517         const char *display_name = NULL;
 518         const char *unique_name = NULL;
 519         AVDeviceInfo *new_device = NULL;
 520         int add = 0;
 521
 522         ret = decklink_get_attr_string(dl, BMDDeckLinkDisplayName, &display_name);
 523         if (ret < 0)
 524             goto next;
 525         ret = decklink_get_attr_string(dl, BMDDeckLinkDeviceHandle, &unique_name);
 526         if (ret < 0)
 527             goto next;
 528
 529         if (show_outputs) {
 530             if (dl->QueryInterface(IID_IDeckLinkOutput, (void **)&output_config) == S_OK) {
 531                 output_config->Release();
 532                 add = 1;
 533             }
 534         }
 535
 536         if (show_inputs) {
 537             if (dl->QueryInterface(IID_IDeckLinkInput, (void **)&input_config) == S_OK) {
 538                 input_config->Release();
 539                 add = 1;
 540             }
 541         }
 542
 543         if (add == 1) {
 544             new_device = (AVDeviceInfo *) av_mallocz(sizeof(AVDeviceInfo));
 545             if (!new_device) {
 546                 ret = AVERROR(ENOMEM);
 547                 goto next;
 548             }
 549
 550             new_device->device_name = av_strdup(unique_name ? unique_name : display_name);
 551             new_device->device_description = av_strdup(display_name);
 552
 553             if (!new_device->device_name ||
 554                 !new_device->device_description ||
 555                 av_dynarray_add_nofree(&device_list->devices, &device_list->nb_devices, new_device) < 0) {
 556                 ret = AVERROR(ENOMEM);
 557                 av_freep(&new_device->device_name);
 558                 av_freep(&new_device->device_description);
 559                 av_freep(&new_device);
 560                 goto next;
 561             }
 562         }
 563
 564     next:
 565         av_freep(&display_name);
 566         av_freep(&unique_name);
 567         dl->Release();
 568     }
 569     iter->Release();
 570     return ret;
 571 }
 572
 573 /* This is a wrapper around the ff_decklink_list_devices() which dumps the
 574    output to av_log() and exits (for backward compatibility with the
 575    "-list_devices" argument). */
 576 void ff_decklink_list_devices_legacy(AVFormatContext *avctx,
 577                                      int show_inputs, int show_outputs)
 578 {
 579     struct AVDeviceInfoList *device_list = NULL;
 580     int ret;
 581
 582     device_list = (struct AVDeviceInfoList *) av_mallocz(sizeof(AVDeviceInfoList));
 583     if (!device_list)
 584         return;
 585
 586     ret = ff_decklink_list_devices(avctx, device_list, show_inputs, show_outputs);
 587     if (ret == 0) {
 588         av_log(avctx, AV_LOG_INFO, "Blackmagic DeckLink %s devices:\n",
 589                show_inputs ? "input" : "output");
 590         for (int i = 0; i < device_list->nb_devices; i++) {
 591             av_log(avctx, AV_LOG_INFO, "\t'%s'\n", device_list->devices[i]->device_description);
 592         }
 593     }
 594     avdevice_free_list_devices(&device_list);
 595 }
 596
 597 int ff_decklink_list_formats(AVFormatContext *avctx, decklink_direction_t direction)
 598 {
 599     struct decklink_cctx *cctx = (struct decklink_cctx *)avctx->priv_data;
 600     struct decklink_ctx *ctx = (struct decklink_ctx *)cctx->ctx;
 601     IDeckLinkDisplayModeIterator *itermode;
 602     IDeckLinkDisplayMode *mode;
 603     uint32_t format_code;
 604     HRESULT res;
 605
 606     if (direction == DIRECTION_IN) {
 607         int ret;
 608         ret = decklink_select_input(avctx, bmdDeckLinkConfigAudioInputConnection);
 609         if (ret < 0)
 610             return ret;
 611         ret = decklink_select_input(avctx, bmdDeckLinkConfigVideoInputConnection);
 612         if (ret < 0)
 613             return ret;
 614         res = ctx->dli->GetDisplayModeIterator (&itermode);
 615     } else {
 616         res = ctx->dlo->GetDisplayModeIterator (&itermode);
 617     }
 618
 619     if (res!= S_OK) {
 620             av_log(avctx, AV_LOG_ERROR, "Could not get Display Mode Iterator\n");
 621             return AVERROR(EIO);
 622     }
 623
 624     av_log(avctx, AV_LOG_INFO, "Supported formats for '%s':\n\tformat_code\tdescription",
 625                avctx->url);
 626     while (itermode->Next(&mode) == S_OK) {
 627         BMDTimeValue tb_num, tb_den;
 628         mode->GetFrameRate(&tb_num, &tb_den);
 629         format_code = av_bswap32(mode->GetDisplayMode());
 630         av_log(avctx, AV_LOG_INFO, "\n\t%.4s\t\t%ldx%ld at %d/%d fps",
 631                 (char*) &format_code, mode->GetWidth(), mode->GetHeight(),
 632                 (int) tb_den, (int) tb_num);
 633         switch (mode->GetFieldDominance()) {
 634         case bmdLowerFieldFirst:
 635         av_log(avctx, AV_LOG_INFO, " (interlaced, lower field first)"); break;
 636         case bmdUpperFieldFirst:
 637         av_log(avctx, AV_LOG_INFO, " (interlaced, upper field first)"); break;
 638         }
 639         mode->Release();
 640     }
 641     av_log(avctx, AV_LOG_INFO, "\n");
 642
 643     itermode->Release();
 644
 645     return 0;
 646 }
 647
 648 void ff_decklink_cleanup(AVFormatContext *avctx)
 649 {
 650     struct decklink_cctx *cctx = (struct decklink_cctx *)avctx->priv_data;
 651     struct decklink_ctx *ctx = (struct decklink_ctx *)cctx->ctx;
 652
 653     if (ctx->dli)
 654         ctx->dli->Release();
 655     if (ctx->dlo)
 656         ctx->dlo->Release();
 657     if (ctx->attr)
 658         ctx->attr->Release();
 659     if (ctx->cfg)
 660         ctx->cfg->Release();
 661     if (ctx->dl)
 662         ctx->dl->Release();
 663 }
 664
 665 int ff_decklink_init_device(AVFormatContext *avctx, const char* name)
 666 {
 667     struct decklink_cctx *cctx = (struct decklink_cctx *)avctx->priv_data;
 668     struct decklink_ctx *ctx = (struct decklink_ctx *)cctx->ctx;
 669     IDeckLink *dl = NULL;
 670     IDeckLinkIterator *iter = decklink_create_iterator(avctx);
 671     if (!iter)
 672         return AVERROR_EXTERNAL;
 673
 674     while (iter->Next(&dl) == S_OK) {
 675         const char *display_name = NULL;
 676         const char *unique_name = NULL;
 677         decklink_get_attr_string(dl, BMDDeckLinkDisplayName, &display_name);
 678         decklink_get_attr_string(dl, BMDDeckLinkDeviceHandle, &unique_name);
 679         if (display_name && !strcmp(name, display_name) || unique_name && !strcmp(name, unique_name)) {
 680             av_free((void *)unique_name);
 681             av_free((void *)display_name);
 682             ctx->dl = dl;
 683             break;
 684         }
 685         av_free((void *)display_name);
 686         av_free((void *)unique_name);
 687         dl->Release();
 688     }
 689     iter->Release();
 690     if (!ctx->dl)
 691         return AVERROR(ENXIO);
 692
 693     if (ctx->dl->QueryInterface(IID_IDeckLinkConfiguration, (void **)&ctx->cfg) != S_OK) {
 694         av_log(avctx, AV_LOG_ERROR, "Could not get configuration interface for '%s'\n", name);
 695         ff_decklink_cleanup(avctx);
 696         return AVERROR_EXTERNAL;
 697     }
 698
 699     if (ctx->dl->QueryInterface(IID_IDeckLinkProfileAttributes, (void **)&ctx->attr) != S_OK) {
 700         av_log(avctx, AV_LOG_ERROR, "Could not get attributes interface for '%s'\n", name);
 701         ff_decklink_cleanup(avctx);
 702         return AVERROR_EXTERNAL;
 703     }
 704
 705     return 0;
 706 }