static GstFlowReturn gst_libde265_dec_handle_frame (VIDEO_DECODER_BASE * parse,
VIDEO_FRAME * frame)
{
GstFlowReturn ret = GST_FLOW_OK;
GstLibde265Dec *dec = GST_LIBDE265_DEC (parse);
const struct de265_image *img = de265_get_next_picture(dec->ctx);
if (img == NULL) {
// need more data
return GST_FLOW_OK;
}
ret = ALLOC_OUTPUT_FRAME(parse, frame);
if (ret != GST_FLOW_OK) {
return ret;
}
uint8_t *dest;
#if GST_CHECK_VERSION(1,0,0)
GstMapInfo info;
if (!gst_buffer_map(frame->output_buffer, &info, GST_MAP_WRITE)) {
return GST_FLOW_ERROR;
}
dest = info.data;
#else
dest = GST_BUFFER_DATA(frame->src_buffer);
#endif
int plane;
for (plane=0; plane<3; plane++) {
int stride;
int width = de265_get_image_width(img, plane);
int height = de265_get_image_height(img, plane);
const uint8_t *src = de265_get_image_plane(img, plane, &stride);
if (stride == width) {
memcpy(dest, src, height * stride);
dest += (height * stride);
} else {
while (height--) {
memcpy(dest, src, width);
src += stride;
dest += width;
}
}
}
#if GST_CHECK_VERSION(1,0,0)
gst_buffer_unmap(frame->output_buffer, &info);
frame->pts = (GstClockTime) de265_get_image_PTS(img);
#endif
return FINISH_FRAME (parse, frame);
}
static int ff_libde265dec_decode(AVCodecContext *avctx,
void *data, int *got_frame, AVPacket *avpkt)
{
DE265Context *ctx = (DE265Context *) avctx->priv_data;
AVFrame *picture = (AVFrame *) data;
const struct de265_image *img;
de265_error err;
int ret;
int64_t pts;
#if LIBDE265_NUMERIC_VERSION >= 0x00050000
int more = 0;
#endif
const uint8_t* src[4];
int stride[4];
if (avpkt->size > 0) {
if (avpkt->pts != AV_NOPTS_VALUE) {
pts = avpkt->pts;
} else {
pts = avctx->reordered_opaque;
}
// replace 4-byte length fields with NAL start codes
uint8_t* avpkt_data = avpkt->data;
uint8_t* avpkt_end = avpkt->data + avpkt->size;
while (avpkt_data + 4 <= avpkt_end) {
int nal_size = AV_RB32(avpkt_data);
#if LIBDE265_NUMERIC_VERSION < 0x00050000
AV_WB32(avpkt_data, 0x00000001);
#else
err = de265_push_NAL(ctx->decoder, avpkt_data + 4, nal_size, pts, NULL);
if (err != DE265_OK) {
const char *error = de265_get_error_text(err);
av_log(avctx, AV_LOG_ERROR, "Failed to push data: %s\n", error);
return AVERROR_INVALIDDATA;
}
#endif
avpkt_data += 4 + nal_size;
}
#if LIBDE265_NUMERIC_VERSION >= 0x00050000
} else {
de265_flush_data(ctx->decoder);
#endif
}
#if LIBDE265_NUMERIC_VERSION < 0x00050000
// insert input packet PTS into sorted queue
if (ctx->pts_queue_len < DE265_MAX_PTS_QUEUE) {
int pos=0;
while (ctx->pts_queue[pos] < pts &&
pos<ctx->pts_queue_len) {
pos++;
}
if (pos < ctx->pts_queue_len) {
memmove(&ctx->pts_queue[pos+1], &ctx->pts_queue[pos],
sizeof(int64_t) * (ctx->pts_queue_len - pos));
}
ctx->pts_queue[pos] = pts;
ctx->pts_queue_len++;
if (ctx->pts_queue_len > ctx->pts_min_queue_len) {
ctx->pts_min_queue_len = ctx->pts_queue_len;
}
}
err = de265_decode_data(ctx->decoder, avpkt->data, avpkt->size);
#else
// decode as much as possible
do {
err = de265_decode(ctx->decoder, &more);
} while (more && err == DE265_OK);
#endif
switch (err) {
case DE265_OK:
case DE265_ERROR_IMAGE_BUFFER_FULL:
#if LIBDE265_NUMERIC_VERSION >= 0x00050000
case DE265_ERROR_WAITING_FOR_INPUT_DATA:
#endif
break;
default:
{
const char *error = de265_get_error_text(err);
av_log(avctx, AV_LOG_ERROR, "Failed to decode frame: %s\n", error);
return AVERROR_INVALIDDATA;
}
}
if ((img = de265_get_next_picture(ctx->decoder)) != NULL) {
int width;
int height;
if (de265_get_chroma_format(img) != de265_chroma_420) {
av_log(avctx, AV_LOG_ERROR, "Unsupported output colorspace (%d)\n",
de265_get_chroma_format(img));
return AVERROR_INVALIDDATA;
}
width = de265_get_image_width(img,0);
height = de265_get_image_height(img,0);
if (width != avctx->width || height != avctx->height) {
if (avctx->width != 0)
av_log(avctx, AV_LOG_INFO, "dimension change! %dx%d -> %dx%d\n",
avctx->width, avctx->height, width, height);
if (av_image_check_size(width, height, 0, avctx)) {
return AVERROR_INVALIDDATA;
}
avcodec_set_dimensions(avctx, width, height);
}
#if LIBDE265_NUMERIC_VERSION < 0x00050000
if (ctx->pts_queue_len < ctx->pts_min_queue_len) {
// fill pts queue to ensure reordering works
return avpkt->size;
}
#endif
picture->width = avctx->width;
picture->height = avctx->height;
picture->format = avctx->pix_fmt;
if ((ret = av_frame_get_buffer(picture, 32)) < 0) {
return ret;
}
for (int i=0;i<4;i++) {
src[i] = de265_get_image_plane(img,i, &stride[i]);
}
av_image_copy(picture->data, picture->linesize, src, stride,
avctx->pix_fmt, width, height);
*got_frame = 1;
#if LIBDE265_NUMERIC_VERSION < 0x00050000
// assign next PTS from queue
if (ctx->pts_queue_len > 0) {
picture->reordered_opaque = ctx->pts_queue[0];
picture->pkt_pts = ctx->pts_queue[0];
if (ctx->pts_queue_len>1) {
memmove(&ctx->pts_queue[0], &ctx->pts_queue[1],
sizeof(int64_t) * (ctx->pts_queue_len-1));
}
ctx->pts_queue_len--;
}
#else
picture->reordered_opaque = de265_get_image_PTS(img);
picture->pkt_pts = de265_get_image_PTS(img);
#endif
}
return avpkt->size;
}
