mirror of https://github.com/FFmpeg/FFmpeg.git
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>pull/86/head
Deti Fliegl
11 years ago
committed by
Michael Niedermayer
parent
56b8d10676
commit
a5e040ee3c
7 changed files with 380 additions and 193 deletions
@ -0,0 +1,229 @@ |
||||
/*
|
||||
* Blackmagic DeckLink output |
||||
* Copyright (c) 2013-2014 Ramiro Polla, Luca Barbato, Deti Fliegl |
||||
* |
||||
* This file is part of FFmpeg. |
||||
* |
||||
* FFmpeg is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* FFmpeg is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with FFmpeg; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#include <DeckLinkAPI.h> |
||||
#ifdef _WIN32 |
||||
#include <DeckLinkAPI_i.c> |
||||
#else |
||||
#include <DeckLinkAPIDispatch.cpp> |
||||
#endif |
||||
|
||||
#include <pthread.h> |
||||
#include <semaphore.h> |
||||
|
||||
extern "C" { |
||||
#include "libavformat/avformat.h" |
||||
#include "libavformat/internal.h" |
||||
#include "libavutil/imgutils.h" |
||||
} |
||||
|
||||
#include "decklink_common.h" |
||||
|
||||
#ifdef _WIN32 |
||||
IDeckLinkIterator *CreateDeckLinkIteratorInstance(void) |
||||
{ |
||||
IDeckLinkIterator *iter; |
||||
|
||||
if (CoInitialize(NULL) != S_OK) { |
||||
av_log(NULL, AV_LOG_ERROR, "COM initialization failed.\n"); |
||||
return NULL; |
||||
} |
||||
|
||||
if (CoCreateInstance(CLSID_CDeckLinkIterator, NULL, CLSCTX_ALL, |
||||
IID_IDeckLinkIterator, (void**) &iter) != S_OK) { |
||||
av_log(NULL, AV_LOG_ERROR, "DeckLink drivers not installed.\n"); |
||||
return NULL; |
||||
} |
||||
|
||||
return iter; |
||||
} |
||||
#endif |
||||
|
||||
#ifdef _WIN32 |
||||
static char *dup_wchar_to_utf8(wchar_t *w) |
||||
{ |
||||
char *s = NULL; |
||||
int l = WideCharToMultiByte(CP_UTF8, 0, w, -1, 0, 0, 0, 0); |
||||
s = (char *) av_malloc(l); |
||||
if (s) |
||||
WideCharToMultiByte(CP_UTF8, 0, w, -1, s, l, 0, 0); |
||||
return s; |
||||
} |
||||
#define DECKLINK_STR OLECHAR * |
||||
#define DECKLINK_STRDUP dup_wchar_to_utf8 |
||||
#else |
||||
#define DECKLINK_STR const char * |
||||
#define DECKLINK_STRDUP av_strdup |
||||
#endif |
||||
|
||||
HRESULT ff_decklink_get_display_name(IDeckLink *This, const char **displayName) |
||||
{ |
||||
DECKLINK_STR tmpDisplayName; |
||||
HRESULT hr = This->GetDisplayName(&tmpDisplayName); |
||||
if (hr != S_OK) |
||||
return hr; |
||||
*displayName = DECKLINK_STRDUP(tmpDisplayName); |
||||
/* free() is needed for a string returned by the DeckLink SDL. */ |
||||
free((void *) tmpDisplayName); |
||||
return hr; |
||||
} |
||||
|
||||
int ff_decklink_set_format(AVFormatContext *avctx, |
||||
int width, int height, |
||||
int tb_num, int tb_den, |
||||
decklink_direction_t direction, int num) |
||||
{ |
||||
struct decklink_cctx *cctx = (struct decklink_cctx *) avctx->priv_data; |
||||
struct decklink_ctx *ctx = (struct decklink_ctx *)cctx->ctx; |
||||
BMDDisplayModeSupport support; |
||||
IDeckLinkDisplayModeIterator *itermode; |
||||
IDeckLinkDisplayMode *mode; |
||||
int i = 1; |
||||
HRESULT res; |
||||
|
||||
if (direction == DIRECTION_IN) { |
||||
res = ctx->dli->GetDisplayModeIterator (&itermode); |
||||
} else { |
||||
res = ctx->dlo->GetDisplayModeIterator (&itermode); |
||||
} |
||||
|
||||
if (res!= S_OK) { |
||||
av_log(avctx, AV_LOG_ERROR, "Could not get Display Mode Iterator\n"); |
||||
return AVERROR(EIO); |
||||
} |
||||
|
||||
|
||||
if (tb_num == 1) { |
||||
tb_num *= 1000; |
||||
tb_den *= 1000; |
||||
} |
||||
ctx->bmd_mode = bmdModeUnknown; |
||||
while ((ctx->bmd_mode == bmdModeUnknown) && itermode->Next(&mode) == S_OK) { |
||||
BMDTimeValue bmd_tb_num, bmd_tb_den; |
||||
int bmd_width = mode->GetWidth(); |
||||
int bmd_height = mode->GetHeight(); |
||||
|
||||
mode->GetFrameRate(&bmd_tb_num, &bmd_tb_den); |
||||
|
||||
if ((bmd_width == width && bmd_height == height && |
||||
bmd_tb_num == tb_num && bmd_tb_den == tb_den) || i == num) { |
||||
ctx->bmd_mode = mode->GetDisplayMode(); |
||||
ctx->bmd_width = bmd_width; |
||||
ctx->bmd_height = bmd_height; |
||||
ctx->bmd_tb_den = bmd_tb_den; |
||||
ctx->bmd_tb_num = bmd_tb_num; |
||||
ctx->bmd_field_dominance = mode->GetFieldDominance(); |
||||
av_log(avctx, AV_LOG_INFO, "Found Decklink mode %d x %d with rate %.2f%s\n", |
||||
bmd_width, bmd_height, (float)bmd_tb_den/(float)bmd_tb_num, |
||||
(ctx->bmd_field_dominance==bmdLowerFieldFirst || ctx->bmd_field_dominance==bmdUpperFieldFirst)?"(i)":""); |
||||
} |
||||
|
||||
mode->Release(); |
||||
i++; |
||||
} |
||||
|
||||
itermode->Release(); |
||||
|
||||
if (ctx->bmd_mode == bmdModeUnknown) |
||||
return -1; |
||||
if (direction == DIRECTION_IN) { |
||||
if (ctx->dli->DoesSupportVideoMode(ctx->bmd_mode, bmdFormat8BitYUV, |
||||
bmdVideoOutputFlagDefault, |
||||
&support, NULL) != S_OK) |
||||
return -1; |
||||
} else { |
||||
if (ctx->dlo->DoesSupportVideoMode(ctx->bmd_mode, bmdFormat8BitYUV, |
||||
bmdVideoOutputFlagDefault, |
||||
&support, NULL) != S_OK) |
||||
return -1; |
||||
} |
||||
if (support == bmdDisplayModeSupported) |
||||
return 0; |
||||
|
||||
return -1; |
||||
} |
||||
|
||||
int ff_decklink_set_format(AVFormatContext *avctx, decklink_direction_t direction, int num) { |
||||
return ff_decklink_set_format(avctx, 0, 0, 0, 0, direction, num); |
||||
} |
||||
|
||||
int ff_decklink_list_devices(AVFormatContext *avctx) |
||||
{ |
||||
IDeckLink *dl = NULL; |
||||
IDeckLinkIterator *iter = CreateDeckLinkIteratorInstance(); |
||||
if (!iter) { |
||||
av_log(avctx, AV_LOG_ERROR, "Could not create DeckLink iterator\n"); |
||||
return AVERROR(EIO); |
||||
} |
||||
av_log(avctx, AV_LOG_INFO, "Blackmagic DeckLink devices:\n"); |
||||
while (iter->Next(&dl) == S_OK) { |
||||
const char *displayName; |
||||
ff_decklink_get_display_name(dl, &displayName); |
||||
av_log(avctx, AV_LOG_INFO, "\t'%s'\n", displayName); |
||||
av_free((void *) displayName); |
||||
dl->Release(); |
||||
} |
||||
iter->Release(); |
||||
return 0; |
||||
} |
||||
|
||||
int ff_decklink_list_formats(AVFormatContext *avctx, decklink_direction_t direction) |
||||
{ |
||||
struct decklink_cctx *cctx = (struct decklink_cctx *) avctx->priv_data; |
||||
struct decklink_ctx *ctx = (struct decklink_ctx *)cctx->ctx; |
||||
IDeckLinkDisplayModeIterator *itermode; |
||||
IDeckLinkDisplayMode *mode; |
||||
int i=0; |
||||
HRESULT res; |
||||
|
||||
if (direction == DIRECTION_IN) { |
||||
res = ctx->dli->GetDisplayModeIterator (&itermode); |
||||
} else { |
||||
res = ctx->dlo->GetDisplayModeIterator (&itermode); |
||||
} |
||||
|
||||
if (res!= S_OK) { |
||||
av_log(avctx, AV_LOG_ERROR, "Could not get Display Mode Iterator\n"); |
||||
return AVERROR(EIO); |
||||
} |
||||
|
||||
av_log(avctx, AV_LOG_INFO, "Supported formats for '%s':\n", |
||||
avctx->filename); |
||||
while (itermode->Next(&mode) == S_OK) { |
||||
BMDTimeValue tb_num, tb_den; |
||||
mode->GetFrameRate(&tb_num, &tb_den); |
||||
av_log(avctx, AV_LOG_INFO, "\t%d\t%ldx%ld at %d/%d fps", |
||||
++i,mode->GetWidth(), mode->GetHeight(), |
||||
(int) tb_den, (int) tb_num); |
||||
switch (mode->GetFieldDominance()) { |
||||
case bmdLowerFieldFirst: |
||||
av_log(avctx, AV_LOG_INFO, " (interlaced, lower field first)"); break; |
||||
case bmdUpperFieldFirst: |
||||
av_log(avctx, AV_LOG_INFO, " (interlaced, upper field first)"); break; |
||||
} |
||||
av_log(avctx, AV_LOG_INFO, "\n"); |
||||
mode->Release(); |
||||
} |
||||
|
||||
itermode->Release(); |
||||
|
||||
return 0; |
||||
} |
||||
@ -0,0 +1,98 @@ |
||||
/*
|
||||
* Blackmagic DeckLink common code |
||||
* Copyright (c) 2013-2014 Ramiro Polla, Luca Barbato, Deti Fliegl |
||||
* |
||||
* This file is part of FFmpeg. |
||||
* |
||||
* FFmpeg is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* FFmpeg is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with FFmpeg; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#include "decklink_common_c.h" |
||||
|
||||
class decklink_output_callback; |
||||
class decklink_input_callback; |
||||
|
||||
typedef struct AVPacketQueue { |
||||
AVPacketList *first_pkt, *last_pkt; |
||||
int nb_packets; |
||||
unsigned long long size; |
||||
int abort_request; |
||||
pthread_mutex_t mutex; |
||||
pthread_cond_t cond; |
||||
AVFormatContext *avctx; |
||||
} AVPacketQueue; |
||||
|
||||
struct decklink_ctx { |
||||
/* DeckLink SDK interfaces */ |
||||
IDeckLink *dl; |
||||
IDeckLinkOutput *dlo; |
||||
IDeckLinkInput *dli; |
||||
decklink_output_callback *output_callback; |
||||
decklink_input_callback *input_callback; |
||||
|
||||
/* DeckLink mode information */ |
||||
BMDTimeValue bmd_tb_den; |
||||
BMDTimeValue bmd_tb_num; |
||||
BMDDisplayMode bmd_mode; |
||||
int bmd_width; |
||||
int bmd_height; |
||||
int bmd_field_dominance; |
||||
|
||||
/* Capture buffer queue */ |
||||
AVPacketQueue queue; |
||||
|
||||
/* Streams present */ |
||||
int audio; |
||||
int video; |
||||
|
||||
/* Status */ |
||||
int playback_started; |
||||
int capture_started; |
||||
int64_t last_pts; |
||||
unsigned long frameCount; |
||||
unsigned int dropped; |
||||
AVStream *audio_st; |
||||
AVStream *video_st; |
||||
|
||||
/* Options */ |
||||
int list_devices; |
||||
int list_formats; |
||||
double preroll; |
||||
|
||||
int frames_preroll; |
||||
int frames_buffer; |
||||
|
||||
sem_t semaphore; |
||||
|
||||
int channels; |
||||
}; |
||||
|
||||
typedef enum { DIRECTION_IN, DIRECTION_OUT} decklink_direction_t; |
||||
|
||||
#ifdef _WIN32 |
||||
typedef unsigned long buffercount_type; |
||||
IDeckLinkIterator *CreateDeckLinkIteratorInstance(void); |
||||
char *dup_wchar_to_utf8(wchar_t *w); |
||||
#else |
||||
typedef uint32_t buffercount_type; |
||||
#endif |
||||
|
||||
|
||||
HRESULT ff_decklink_get_display_name(IDeckLink *This, const char **displayName); |
||||
int ff_decklink_set_format(AVFormatContext *avctx, int width, int height, int tb_num, int tb_den, decklink_direction_t direction = DIRECTION_OUT, int num = 0); |
||||
int ff_decklink_set_format(AVFormatContext *avctx, decklink_direction_t direction, int num); |
||||
int ff_decklink_list_devices(AVFormatContext *avctx); |
||||
int ff_decklink_list_formats(AVFormatContext *avctx, decklink_direction_t direction = DIRECTION_OUT); |
||||
|
||||
@ -0,0 +1,32 @@ |
||||
/*
|
||||
* Blackmagic DeckLink common code |
||||
* Copyright (c) 2013-2014 Ramiro Polla |
||||
* |
||||
* This file is part of FFmpeg. |
||||
* |
||||
* FFmpeg is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* FFmpeg is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with FFmpeg; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
struct decklink_cctx { |
||||
const AVClass *cclass; |
||||
|
||||
void *ctx; |
||||
|
||||
/* Options */ |
||||
int list_devices; |
||||
int list_formats; |
||||
double preroll; |
||||
}; |
||||
|
||||
Loading…
Reference in new issue