mirror of https://github.com/FFmpeg/FFmpeg.git
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>pull/43/merge
Ramiro Polla
11 years ago
committed by
Michael Niedermayer
parent
36a6297953
commit
5965adeb20
8 changed files with 756 additions and 1 deletions
@ -0,0 +1,588 @@ |
|||||||
|
/*
|
||||||
|
* Blackmagic DeckLink output |
||||||
|
* 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 |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <DeckLinkAPI.h> |
||||||
|
#ifdef _WIN32 |
||||||
|
#include <DeckLinkAPI_i.c> |
||||||
|
typedef unsigned long buffercount_type; |
||||||
|
#else |
||||||
|
#include <DeckLinkAPIDispatch.cpp> |
||||||
|
typedef uint32_t buffercount_type; |
||||||
|
#endif |
||||||
|
|
||||||
|
#include <pthread.h> |
||||||
|
#include <semaphore.h> |
||||||
|
|
||||||
|
extern "C" { |
||||||
|
#include "libavformat/avformat.h" |
||||||
|
#include "libavformat/internal.h" |
||||||
|
#include "libavutil/imgutils.h" |
||||||
|
} |
||||||
|
|
||||||
|
#include "decklink_enc.h" |
||||||
|
|
||||||
|
class decklink_callback; |
||||||
|
|
||||||
|
struct decklink_ctx { |
||||||
|
/* DeckLink SDK interfaces */ |
||||||
|
IDeckLink *dl; |
||||||
|
IDeckLinkOutput *dlo; |
||||||
|
decklink_callback *callback; |
||||||
|
|
||||||
|
/* DeckLink mode information */ |
||||||
|
IDeckLinkDisplayModeIterator *itermode; |
||||||
|
BMDTimeValue bmd_tb_den; |
||||||
|
BMDTimeValue bmd_tb_num; |
||||||
|
BMDDisplayMode bmd_mode; |
||||||
|
int bmd_width; |
||||||
|
int bmd_height; |
||||||
|
|
||||||
|
/* Streams present */ |
||||||
|
int audio; |
||||||
|
int video; |
||||||
|
|
||||||
|
/* Status */ |
||||||
|
int playback_started; |
||||||
|
int64_t last_pts; |
||||||
|
|
||||||
|
/* Options */ |
||||||
|
int list_devices; |
||||||
|
int list_formats; |
||||||
|
double preroll; |
||||||
|
|
||||||
|
int frames_preroll; |
||||||
|
int frames_buffer; |
||||||
|
|
||||||
|
sem_t semaphore; |
||||||
|
|
||||||
|
int channels; |
||||||
|
}; |
||||||
|
|
||||||
|
/* DeckLink callback class declaration */ |
||||||
|
class decklink_frame : public IDeckLinkVideoFrame |
||||||
|
{ |
||||||
|
public: |
||||||
|
decklink_frame(struct decklink_ctx *ctx, AVFrame *avframe, long width, |
||||||
|
long height, void *buffer) : |
||||||
|
_ctx(ctx), _avframe(avframe), _width(width), |
||||||
|
_height(height), _buffer(buffer), _refs(0) { } |
||||||
|
|
||||||
|
virtual long STDMETHODCALLTYPE GetWidth (void) { return _width; } |
||||||
|
virtual long STDMETHODCALLTYPE GetHeight (void) { return _height; } |
||||||
|
virtual long STDMETHODCALLTYPE GetRowBytes (void) { return _width<<1; } |
||||||
|
virtual BMDPixelFormat STDMETHODCALLTYPE GetPixelFormat(void) { return bmdFormat8BitYUV; } |
||||||
|
virtual BMDFrameFlags STDMETHODCALLTYPE GetFlags (void) { return bmdVideoOutputFlagDefault; } |
||||||
|
virtual HRESULT STDMETHODCALLTYPE GetBytes (void **buffer) { *buffer = _buffer; return S_OK; } |
||||||
|
|
||||||
|
virtual HRESULT STDMETHODCALLTYPE GetTimecode (BMDTimecodeFormat format, IDeckLinkTimecode **timecode) { return S_FALSE; } |
||||||
|
virtual HRESULT STDMETHODCALLTYPE GetAncillaryData(IDeckLinkVideoFrameAncillary **ancillary) { return S_FALSE; } |
||||||
|
|
||||||
|
virtual HRESULT STDMETHODCALLTYPE QueryInterface(REFIID iid, LPVOID *ppv) { return E_NOINTERFACE; } |
||||||
|
virtual ULONG STDMETHODCALLTYPE AddRef(void) { return ++_refs; } |
||||||
|
virtual ULONG STDMETHODCALLTYPE Release(void) { if (!--_refs) delete this; return _refs; } |
||||||
|
|
||||||
|
struct decklink_ctx *_ctx; |
||||||
|
AVFrame *_avframe; |
||||||
|
|
||||||
|
private: |
||||||
|
long _width; |
||||||
|
long _height; |
||||||
|
void *_buffer; |
||||||
|
int _refs; |
||||||
|
}; |
||||||
|
|
||||||
|
class decklink_callback : public IDeckLinkVideoOutputCallback |
||||||
|
{ |
||||||
|
public: |
||||||
|
virtual HRESULT STDMETHODCALLTYPE ScheduledFrameCompleted(IDeckLinkVideoFrame *_frame, BMDOutputFrameCompletionResult result) |
||||||
|
{ |
||||||
|
decklink_frame *frame = static_cast<decklink_frame *>(_frame); |
||||||
|
struct decklink_ctx *ctx = frame->_ctx; |
||||||
|
AVFrame *avframe = frame->_avframe; |
||||||
|
|
||||||
|
av_frame_free(&avframe); |
||||||
|
|
||||||
|
sem_post(&ctx->semaphore); |
||||||
|
|
||||||
|
return S_OK; |
||||||
|
} |
||||||
|
virtual HRESULT STDMETHODCALLTYPE ScheduledPlaybackHasStopped(void) { return S_OK; } |
||||||
|
virtual HRESULT STDMETHODCALLTYPE QueryInterface(REFIID iid, LPVOID *ppv) { return E_NOINTERFACE; } |
||||||
|
virtual ULONG STDMETHODCALLTYPE AddRef(void) { return 1; } |
||||||
|
virtual ULONG STDMETHODCALLTYPE Release(void) { return 1; } |
||||||
|
}; |
||||||
|
|
||||||
|
#ifdef _WIN32 |
||||||
|
static 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 |
||||||
|
|
||||||
|
/* free() is needed for a string returned by the DeckLink SDL. */ |
||||||
|
#undef free |
||||||
|
|
||||||
|
#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 |
||||||
|
|
||||||
|
static HRESULT IDeckLink_GetDisplayName(IDeckLink *This, const char **displayName) |
||||||
|
{ |
||||||
|
DECKLINK_STR tmpDisplayName; |
||||||
|
HRESULT hr = This->GetDisplayName(&tmpDisplayName); |
||||||
|
if (hr != S_OK) |
||||||
|
return hr; |
||||||
|
*displayName = DECKLINK_STRDUP(tmpDisplayName); |
||||||
|
free((void *) tmpDisplayName); |
||||||
|
return hr; |
||||||
|
} |
||||||
|
|
||||||
|
static int decklink_set_format(struct decklink_ctx *ctx, |
||||||
|
int width, int height, |
||||||
|
int tb_num, int tb_den) |
||||||
|
{ |
||||||
|
BMDDisplayModeSupport support; |
||||||
|
IDeckLinkDisplayMode *mode; |
||||||
|
|
||||||
|
if (tb_num == 1) { |
||||||
|
tb_num *= 1000; |
||||||
|
tb_den *= 1000; |
||||||
|
} |
||||||
|
ctx->bmd_mode = bmdModeUnknown; |
||||||
|
while ((ctx->bmd_mode == bmdModeUnknown) && ctx->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) { |
||||||
|
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; |
||||||
|
} |
||||||
|
|
||||||
|
mode->Release(); |
||||||
|
} |
||||||
|
if (ctx->bmd_mode == bmdModeUnknown) |
||||||
|
return -1; |
||||||
|
if (ctx->dlo->DoesSupportVideoMode(ctx->bmd_mode, bmdFormat8BitYUV, |
||||||
|
bmdVideoOutputFlagDefault, |
||||||
|
&support, NULL) != S_OK) |
||||||
|
return -1; |
||||||
|
if (support == bmdDisplayModeSupported) |
||||||
|
return 0; |
||||||
|
|
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
static int decklink_setup_video(AVFormatContext *avctx, AVStream *st) |
||||||
|
{ |
||||||
|
struct decklink_cctx *cctx = (struct decklink_cctx *) avctx->priv_data; |
||||||
|
struct decklink_ctx *ctx = (struct decklink_ctx *) cctx->ctx; |
||||||
|
AVCodecContext *c = st->codec; |
||||||
|
|
||||||
|
if (ctx->video) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Only one video stream is supported!\n"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
if (c->pix_fmt != AV_PIX_FMT_UYVY422) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Unsupported pixel format!" |
||||||
|
" Only AV_PIX_FMT_UYVY422 is supported.\n"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (decklink_set_format(ctx, c->width, c->height, |
||||||
|
c->time_base.num, c->time_base.den)) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Unsupported video size or framerate!" |
||||||
|
" Check available formats with -list_formats 1.\n"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (ctx->dlo->EnableVideoOutput(ctx->bmd_mode, |
||||||
|
bmdVideoOutputFlagDefault) != S_OK) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not enable video output!\n"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
/* Set callback. */ |
||||||
|
ctx->callback = new decklink_callback(); |
||||||
|
ctx->dlo->SetScheduledFrameCompletionCallback(ctx->callback); |
||||||
|
|
||||||
|
/* Start video semaphore. */ |
||||||
|
ctx->frames_preroll = c->time_base.den * ctx->preroll; |
||||||
|
if (c->time_base.den > 1000) |
||||||
|
ctx->frames_preroll /= 1000; |
||||||
|
|
||||||
|
/* Buffer twice as many frames as the preroll. */ |
||||||
|
ctx->frames_buffer = ctx->frames_preroll * 2; |
||||||
|
ctx->frames_buffer = FFMIN(ctx->frames_buffer, 60); |
||||||
|
sem_init(&ctx->semaphore, 0, ctx->frames_buffer); |
||||||
|
|
||||||
|
/* The device expects the framerate to be fixed. */ |
||||||
|
avpriv_set_pts_info(st, 64, c->time_base.num, c->time_base.den); |
||||||
|
|
||||||
|
ctx->video = 1; |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static int decklink_setup_audio(AVFormatContext *avctx, AVStream *st) |
||||||
|
{ |
||||||
|
struct decklink_cctx *cctx = (struct decklink_cctx *) avctx->priv_data; |
||||||
|
struct decklink_ctx *ctx = (struct decklink_ctx *) cctx->ctx; |
||||||
|
AVCodecContext *c = st->codec; |
||||||
|
|
||||||
|
if (ctx->audio) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Only one audio stream is supported!\n"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (c->sample_rate != 48000) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Unsupported sample rate!" |
||||||
|
" Only 48kHz is supported.\n"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (c->channels != 2 && c->channels != 8) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Unsupported number of channels!" |
||||||
|
" Only stereo and 7.1 are supported.\n"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (ctx->dlo->EnableAudioOutput(bmdAudioSampleRate48kHz, |
||||||
|
bmdAudioSampleType16bitInteger, |
||||||
|
c->channels, |
||||||
|
bmdAudioOutputStreamTimestamped) != S_OK) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not enable audio output!\n"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (ctx->dlo->BeginAudioPreroll() != S_OK) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not begin audio preroll!\n"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
/* The device expects the sample rate to be fixed. */ |
||||||
|
avpriv_set_pts_info(st, 64, 1, c->sample_rate); |
||||||
|
ctx->channels = c->channels; |
||||||
|
|
||||||
|
ctx->audio = 1; |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
av_cold int ff_decklink_write_trailer(AVFormatContext *avctx) |
||||||
|
{ |
||||||
|
struct decklink_cctx *cctx = (struct decklink_cctx *) avctx->priv_data; |
||||||
|
struct decklink_ctx *ctx = (struct decklink_ctx *) cctx->ctx; |
||||||
|
|
||||||
|
if (ctx->playback_started) { |
||||||
|
BMDTimeValue actual; |
||||||
|
ctx->dlo->StopScheduledPlayback(ctx->last_pts * ctx->bmd_tb_num, |
||||||
|
&actual, ctx->bmd_tb_den); |
||||||
|
ctx->dlo->DisableVideoOutput(); |
||||||
|
if (ctx->audio) |
||||||
|
ctx->dlo->DisableAudioOutput(); |
||||||
|
} |
||||||
|
|
||||||
|
if (ctx->dlo) |
||||||
|
ctx->dlo->Release(); |
||||||
|
if (ctx->dl) |
||||||
|
ctx->dl->Release(); |
||||||
|
|
||||||
|
if (ctx->callback) |
||||||
|
delete ctx->callback; |
||||||
|
|
||||||
|
sem_destroy(&ctx->semaphore); |
||||||
|
|
||||||
|
av_freep(&cctx->ctx); |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static int decklink_write_video_packet(AVFormatContext *avctx, AVPacket *pkt) |
||||||
|
{ |
||||||
|
struct decklink_cctx *cctx = (struct decklink_cctx *) avctx->priv_data; |
||||||
|
struct decklink_ctx *ctx = (struct decklink_ctx *) cctx->ctx; |
||||||
|
AVPicture *avpicture = (AVPicture *) pkt->data; |
||||||
|
AVFrame *avframe, *tmp; |
||||||
|
decklink_frame *frame; |
||||||
|
buffercount_type buffered; |
||||||
|
HRESULT hr; |
||||||
|
|
||||||
|
/* HACK while av_uncoded_frame() isn't implemented */ |
||||||
|
int ret; |
||||||
|
|
||||||
|
tmp = av_frame_alloc(); |
||||||
|
if (!tmp) |
||||||
|
return AVERROR(ENOMEM); |
||||||
|
tmp->format = AV_PIX_FMT_UYVY422; |
||||||
|
tmp->width = ctx->bmd_width; |
||||||
|
tmp->height = ctx->bmd_height; |
||||||
|
ret = av_frame_get_buffer(tmp, 32); |
||||||
|
if (ret < 0) { |
||||||
|
av_frame_free(&tmp); |
||||||
|
return ret; |
||||||
|
} |
||||||
|
av_image_copy(tmp->data, tmp->linesize, (const uint8_t **) avpicture->data, |
||||||
|
avpicture->linesize, (AVPixelFormat) tmp->format, tmp->width, |
||||||
|
tmp->height); |
||||||
|
avframe = av_frame_clone(tmp); |
||||||
|
av_frame_free(&tmp); |
||||||
|
if (!avframe) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not clone video frame.\n"); |
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
/* end HACK */ |
||||||
|
|
||||||
|
frame = new decklink_frame(ctx, avframe, ctx->bmd_width, ctx->bmd_height, |
||||||
|
(void *) avframe->data[0]); |
||||||
|
if (!frame) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not create new frame.\n"); |
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
|
||||||
|
/* Always keep at most one second of frames buffered. */ |
||||||
|
sem_wait(&ctx->semaphore); |
||||||
|
|
||||||
|
/* Schedule frame for playback. */ |
||||||
|
hr = ctx->dlo->ScheduleVideoFrame((struct IDeckLinkVideoFrame *) frame, |
||||||
|
pkt->pts * ctx->bmd_tb_num, |
||||||
|
ctx->bmd_tb_num, ctx->bmd_tb_den); |
||||||
|
if (hr != S_OK) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not schedule video frame." |
||||||
|
" error %08x.\n", (uint32_t) hr); |
||||||
|
frame->Release(); |
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
|
||||||
|
ctx->dlo->GetBufferedVideoFrameCount(&buffered); |
||||||
|
av_log(avctx, AV_LOG_DEBUG, "Buffered video frames: %d.\n", (int) buffered); |
||||||
|
if (pkt->pts > 2 && buffered <= 2) |
||||||
|
av_log(avctx, AV_LOG_WARNING, "There are not enough buffered video frames." |
||||||
|
" Video may misbehave!\n"); |
||||||
|
|
||||||
|
/* Preroll video frames. */ |
||||||
|
if (!ctx->playback_started && pkt->pts > ctx->frames_preroll) { |
||||||
|
av_log(avctx, AV_LOG_DEBUG, "Ending audio preroll.\n"); |
||||||
|
if (ctx->audio && ctx->dlo->EndAudioPreroll() != S_OK) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not end audio preroll!\n"); |
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
av_log(avctx, AV_LOG_DEBUG, "Starting scheduled playback.\n"); |
||||||
|
if (ctx->dlo->StartScheduledPlayback(0, ctx->bmd_tb_den, 1.0) != S_OK) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not start scheduled playback!\n"); |
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
ctx->playback_started = 1; |
||||||
|
} |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static int decklink_write_audio_packet(AVFormatContext *avctx, AVPacket *pkt) |
||||||
|
{ |
||||||
|
struct decklink_cctx *cctx = (struct decklink_cctx *) avctx->priv_data; |
||||||
|
struct decklink_ctx *ctx = (struct decklink_ctx *) cctx->ctx; |
||||||
|
int sample_count = pkt->size / (ctx->channels << 1); |
||||||
|
buffercount_type buffered; |
||||||
|
|
||||||
|
ctx->dlo->GetBufferedAudioSampleFrameCount(&buffered); |
||||||
|
if (pkt->pts > 1 && !buffered) |
||||||
|
av_log(avctx, AV_LOG_WARNING, "There's no buffered audio." |
||||||
|
" Audio will misbehave!\n"); |
||||||
|
|
||||||
|
if (ctx->dlo->ScheduleAudioSamples(pkt->data, sample_count, pkt->pts, |
||||||
|
bmdAudioSampleRate48kHz, NULL) != S_OK) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not schedule audio samples.\n"); |
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
extern "C" { |
||||||
|
|
||||||
|
av_cold int ff_decklink_write_header(AVFormatContext *avctx) |
||||||
|
{ |
||||||
|
struct decklink_cctx *cctx = (struct decklink_cctx *) avctx->priv_data; |
||||||
|
struct decklink_ctx *ctx; |
||||||
|
IDeckLinkIterator *iter; |
||||||
|
IDeckLink *dl = NULL; |
||||||
|
unsigned int n; |
||||||
|
|
||||||
|
ctx = (struct decklink_ctx *) av_mallocz(sizeof(struct decklink_ctx)); |
||||||
|
if (!ctx) |
||||||
|
return AVERROR(ENOMEM); |
||||||
|
ctx->list_devices = cctx->list_devices; |
||||||
|
ctx->list_formats = cctx->list_formats; |
||||||
|
ctx->preroll = cctx->preroll; |
||||||
|
cctx->ctx = ctx; |
||||||
|
|
||||||
|
iter = CreateDeckLinkIteratorInstance(); |
||||||
|
if (!iter) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not create DeckLink iterator\n"); |
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
|
||||||
|
/* List available devices. */ |
||||||
|
if (ctx->list_devices) { |
||||||
|
av_log(avctx, AV_LOG_INFO, "Blackmagic DeckLink devices:\n"); |
||||||
|
while (iter->Next(&dl) == S_OK) { |
||||||
|
const char *displayName; |
||||||
|
IDeckLink_GetDisplayName(dl, &displayName); |
||||||
|
av_log(avctx, AV_LOG_INFO, "\t'%s'\n", displayName); |
||||||
|
av_free((void *) displayName); |
||||||
|
dl->Release(); |
||||||
|
} |
||||||
|
iter->Release(); |
||||||
|
return AVERROR_EXIT; |
||||||
|
} |
||||||
|
|
||||||
|
/* Open device. */ |
||||||
|
while (iter->Next(&dl) == S_OK) { |
||||||
|
const char *displayName; |
||||||
|
IDeckLink_GetDisplayName(dl, &displayName); |
||||||
|
if (!strcmp(avctx->filename, displayName)) { |
||||||
|
av_free((void *) displayName); |
||||||
|
ctx->dl = dl; |
||||||
|
break; |
||||||
|
} |
||||||
|
av_free((void *) displayName); |
||||||
|
dl->Release(); |
||||||
|
} |
||||||
|
iter->Release(); |
||||||
|
if (!ctx->dl) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not open '%s'\n", avctx->filename); |
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
|
||||||
|
/* Get output device. */ |
||||||
|
if (ctx->dl->QueryInterface(IID_IDeckLinkOutput, (void **) &ctx->dlo) != S_OK) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not open output device from '%s'\n", |
||||||
|
avctx->filename); |
||||||
|
ctx->dl->Release(); |
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
|
||||||
|
if (ctx->dlo->GetDisplayModeIterator(&ctx->itermode) != S_OK) { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Could not get Display Mode Iterator\n"); |
||||||
|
ctx->dl->Release(); |
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
|
||||||
|
/* List supported formats. */ |
||||||
|
if (ctx->list_formats) { |
||||||
|
IDeckLinkDisplayMode *mode; |
||||||
|
|
||||||
|
av_log(avctx, AV_LOG_INFO, "Supported formats for '%s':\n", |
||||||
|
avctx->filename); |
||||||
|
while (ctx->itermode->Next(&mode) == S_OK) { |
||||||
|
BMDTimeValue tb_num, tb_den; |
||||||
|
mode->GetFrameRate(&tb_num, &tb_den); |
||||||
|
av_log(avctx, AV_LOG_INFO, "\t%ldx%ld at %d/%d fps", |
||||||
|
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(); |
||||||
|
} |
||||||
|
|
||||||
|
ctx->itermode->Release(); |
||||||
|
ctx->dlo->Release(); |
||||||
|
ctx->dl->Release(); |
||||||
|
return AVERROR_EXIT; |
||||||
|
} |
||||||
|
|
||||||
|
/* Setup streams. */ |
||||||
|
for (n = 0; n < avctx->nb_streams; n++) { |
||||||
|
AVStream *st = avctx->streams[n]; |
||||||
|
AVCodecContext *c = st->codec; |
||||||
|
if (c->codec_type == AVMEDIA_TYPE_AUDIO) { |
||||||
|
if (decklink_setup_audio(avctx, st)) |
||||||
|
goto error; |
||||||
|
} else if (c->codec_type == AVMEDIA_TYPE_VIDEO) { |
||||||
|
if (decklink_setup_video(avctx, st)) |
||||||
|
goto error; |
||||||
|
} else { |
||||||
|
av_log(avctx, AV_LOG_ERROR, "Unsupported stream type.\n"); |
||||||
|
goto error; |
||||||
|
} |
||||||
|
} |
||||||
|
ctx->itermode->Release(); |
||||||
|
|
||||||
|
return 0; |
||||||
|
|
||||||
|
error: |
||||||
|
|
||||||
|
ctx->dlo->Release(); |
||||||
|
ctx->dl->Release(); |
||||||
|
|
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
|
||||||
|
int ff_decklink_write_packet(AVFormatContext *avctx, AVPacket *pkt) |
||||||
|
{ |
||||||
|
struct decklink_cctx *cctx = (struct decklink_cctx *) avctx->priv_data; |
||||||
|
struct decklink_ctx *ctx = (struct decklink_ctx *) cctx->ctx; |
||||||
|
AVStream *st = avctx->streams[pkt->stream_index]; |
||||||
|
|
||||||
|
ctx->last_pts = FFMAX(ctx->last_pts, pkt->pts); |
||||||
|
|
||||||
|
if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) |
||||||
|
return decklink_write_video_packet(avctx, pkt); |
||||||
|
else if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) |
||||||
|
return decklink_write_audio_packet(avctx, pkt); |
||||||
|
|
||||||
|
return AVERROR(EIO); |
||||||
|
} |
||||||
|
|
||||||
|
} /* extern "C" */ |
||||||
@ -0,0 +1,43 @@ |
|||||||
|
/*
|
||||||
|
* Blackmagic DeckLink output |
||||||
|
* 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; |
||||||
|
}; |
||||||
|
|
||||||
|
#ifdef __cplusplus |
||||||
|
extern "C" { |
||||||
|
#endif |
||||||
|
|
||||||
|
int ff_decklink_write_header(AVFormatContext *avctx); |
||||||
|
int ff_decklink_write_packet(AVFormatContext *avctx, AVPacket *pkt); |
||||||
|
int ff_decklink_write_trailer(AVFormatContext *avctx); |
||||||
|
|
||||||
|
#ifdef __cplusplus |
||||||
|
} /* extern "C" */ |
||||||
|
#endif |
||||||
@ -0,0 +1,55 @@ |
|||||||
|
/*
|
||||||
|
* Blackmagic DeckLink output |
||||||
|
* 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 |
||||||
|
*/ |
||||||
|
|
||||||
|
#include "libavformat/avformat.h" |
||||||
|
#include "libavutil/opt.h" |
||||||
|
|
||||||
|
#include "decklink_enc.h" |
||||||
|
|
||||||
|
#define OFFSET(x) offsetof(struct decklink_cctx, x) |
||||||
|
#define ENC AV_OPT_FLAG_ENCODING_PARAM |
||||||
|
static const AVOption options[] = { |
||||||
|
{ "list_devices", "list available devices" , OFFSET(list_devices), AV_OPT_TYPE_INT , { .i64 = 0 }, 0, 1, ENC }, |
||||||
|
{ "list_formats", "list supported formats" , OFFSET(list_formats), AV_OPT_TYPE_INT , { .i64 = 0 }, 0, 1, ENC }, |
||||||
|
{ "preroll" , "video preroll in seconds", OFFSET(preroll ), AV_OPT_TYPE_DOUBLE, { .dbl = 0.5 }, 0, 5, ENC }, |
||||||
|
{ NULL }, |
||||||
|
}; |
||||||
|
|
||||||
|
static const AVClass decklink_muxer_class = { |
||||||
|
.class_name = "Blackmagic DeckLink muxer", |
||||||
|
.item_name = av_default_item_name, |
||||||
|
.option = options, |
||||||
|
.version = LIBAVUTIL_VERSION_INT, |
||||||
|
}; |
||||||
|
|
||||||
|
AVOutputFormat ff_decklink_muxer = { |
||||||
|
.name = "decklink", |
||||||
|
.long_name = NULL_IF_CONFIG_SMALL("Blackmagic DeckLink output"), |
||||||
|
.audio_codec = AV_CODEC_ID_PCM_S16LE, |
||||||
|
.video_codec = AV_CODEC_ID_RAWVIDEO, |
||||||
|
.subtitle_codec = AV_CODEC_ID_NONE, |
||||||
|
.flags = AVFMT_NOFILE | AVFMT_RAWPICTURE, |
||||||
|
.priv_class = &decklink_muxer_class, |
||||||
|
.priv_data_size = sizeof(struct decklink_cctx), |
||||||
|
.write_header = ff_decklink_write_header, |
||||||
|
.write_packet = ff_decklink_write_packet, |
||||||
|
.write_trailer = ff_decklink_write_trailer, |
||||||
|
}; |
||||||
Loading…
Reference in new issue