576 lines
18 KiB
C
576 lines
18 KiB
C
|
/*
|
||
|
|
* 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 <va/va.h>
|
||
|
|
#include <va/va_enc_jpeg.h>
|
||
|
|
|
||
|
|
#include "libavutil/avassert.h"
|
||
|
|
#include "libavutil/common.h"
|
||
|
|
#include "libavutil/internal.h"
|
||
|
|
#include "libavutil/opt.h"
|
||
|
|
#include "libavutil/pixdesc.h"
|
||
|
|
|
||
|
|
#include "avcodec.h"
|
||
|
|
#include "bytestream.h"
|
||
|
|
#include "cbs.h"
|
||
|
|
#include "cbs_jpeg.h"
|
||
|
|
#include "internal.h"
|
||
|
|
#include "jpegtables.h"
|
||
|
|
#include "mjpeg.h"
|
||
|
|
#include "put_bits.h"
|
||
|
|
#include "vaapi_encode.h"
|
||
|
|
|
||
|
|
|
||
|
|
// Standard JPEG quantisation tables, in zigzag order.
|
||
|
|
static const unsigned char vaapi_encode_mjpeg_quant_luminance[64] = {
|
||
|
|
16, 11, 12, 14, 12, 10, 16, 14,
|
||
|
|
13, 14, 18, 17, 16, 19, 24, 40,
|
||
|
|
26, 24, 22, 22, 24, 49, 35, 37,
|
||
|
|
29, 40, 58, 51, 61, 60, 57, 51,
|
||
|
|
56, 55, 64, 72, 92, 78, 64, 68,
|
||
|
|
87, 69, 55, 56, 80, 109, 81, 87,
|
||
|
|
95, 98, 103, 104, 103, 62, 77, 113,
|
||
|
|
121, 112, 100, 120, 92, 101, 103, 99,
|
||
|
|
};
|
||
|
|
static const unsigned char vaapi_encode_mjpeg_quant_chrominance[64] = {
|
||
|
|
17, 18, 18, 24, 21, 24, 47, 26,
|
||
|
|
26, 47, 99, 66, 56, 66, 99, 99,
|
||
|
|
99, 99, 99, 99, 99, 99, 99, 99,
|
||
|
|
99, 99, 99, 99, 99, 99, 99, 99,
|
||
|
|
99, 99, 99, 99, 99, 99, 99, 99,
|
||
|
|
99, 99, 99, 99, 99, 99, 99, 99,
|
||
|
|
99, 99, 99, 99, 99, 99, 99, 99,
|
||
|
|
99, 99, 99, 99, 99, 99, 99, 99,
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef struct VAAPIEncodeMJPEGContext {
|
||
|
|
VAAPIEncodeContext common;
|
||
|
|
|
||
|
|
// User options.
|
||
|
|
int jfif;
|
||
|
|
int huffman;
|
||
|
|
|
||
|
|
// Derived settings.
|
||
|
|
int quality;
|
||
|
|
uint8_t jfif_data[14];
|
||
|
|
|
||
|
|
// Writer structures.
|
||
|
|
JPEGRawFrameHeader frame_header;
|
||
|
|
JPEGRawScan scan;
|
||
|
|
JPEGRawApplicationData jfif_header;
|
||
|
|
JPEGRawQuantisationTableSpecification quant_tables;
|
||
|
|
JPEGRawHuffmanTableSpecification huffman_tables;
|
||
|
|
|
||
|
|
CodedBitstreamContext *cbc;
|
||
|
|
CodedBitstreamFragment current_fragment;
|
||
|
|
} VAAPIEncodeMJPEGContext;
|
||
|
|
|
||
|
|
static int vaapi_encode_mjpeg_write_image_header(AVCodecContext *avctx,
|
||
|
|
VAAPIEncodePicture *pic,
|
||
|
|
VAAPIEncodeSlice *slice,
|
||
|
|
char *data, size_t *data_len)
|
||
|
|
{
|
||
|
|
VAAPIEncodeMJPEGContext *priv = avctx->priv_data;
|
||
|
|
CodedBitstreamFragment *frag = &priv->current_fragment;
|
||
|
|
int err;
|
||
|
|
|
||
|
|
if (priv->jfif) {
|
||
|
|
err = ff_cbs_insert_unit_content(priv->cbc, frag, -1,
|
||
|
|
JPEG_MARKER_APPN + 0,
|
||
|
|
&priv->jfif_header, NULL);
|
||
|
|
if (err < 0)
|
||
|
|
goto fail;
|
||
|
|
}
|
||
|
|
|
||
|
|
err = ff_cbs_insert_unit_content(priv->cbc, frag, -1,
|
||
|
|
JPEG_MARKER_DQT,
|
||
|
|
&priv->quant_tables, NULL);
|
||
|
|
if (err < 0)
|
||
|
|
goto fail;
|
||
|
|
|
||
|
|
err = ff_cbs_insert_unit_content(priv->cbc, frag, -1,
|
||
|
|
JPEG_MARKER_SOF0,
|
||
|
|
&priv->frame_header, NULL);
|
||
|
|
if (err < 0)
|
||
|
|
goto fail;
|
||
|
|
|
||
|
|
if (priv->huffman) {
|
||
|
|
err = ff_cbs_insert_unit_content(priv->cbc, frag, -1,
|
||
|
|
JPEG_MARKER_DHT,
|
||
|
|
&priv->huffman_tables, NULL);
|
||
|
|
if (err < 0)
|
||
|
|
goto fail;
|
||
|
|
}
|
||
|
|
|
||
|
|
err = ff_cbs_insert_unit_content(priv->cbc, frag, -1,
|
||
|
|
JPEG_MARKER_SOS,
|
||
|
|
&priv->scan, NULL);
|
||
|
|
if (err < 0)
|
||
|
|
goto fail;
|
||
|
|
|
||
|
|
err = ff_cbs_write_fragment_data(priv->cbc, frag);
|
||
|
|
if (err < 0) {
|
||
|
|
av_log(avctx, AV_LOG_ERROR, "Failed to write image header.\n");
|
||
|
|
goto fail;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (*data_len < 8 * frag->data_size) {
|
||
|
|
av_log(avctx, AV_LOG_ERROR, "Image header too large: "
|
||
|
|
"%zu < %zu.\n", *data_len, 8 * frag->data_size);
|
||
|
|
err = AVERROR(ENOSPC);
|
||
|
|
goto fail;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Remove the EOI at the end of the fragment.
|
||
|
|
memcpy(data, frag->data, frag->data_size - 2);
|
||
|
|
*data_len = 8 * (frag->data_size - 2);
|
||
|
|
|
||
|
|
err = 0;
|
||
|
|
fail:
|
||
|
|
ff_cbs_fragment_reset(priv->cbc, frag);
|
||
|
|
return err;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int vaapi_encode_mjpeg_write_extra_buffer(AVCodecContext *avctx,
|
||
|
|
VAAPIEncodePicture *pic,
|
||
|
|
int index, int *type,
|
||
|
|
char *data, size_t *data_len)
|
||
|
|
{
|
||
|
|
VAAPIEncodeMJPEGContext *priv = avctx->priv_data;
|
||
|
|
int t, i, k;
|
||
|
|
|
||
|
|
if (index == 0) {
|
||
|
|
// Write quantisation tables.
|
||
|
|
JPEGRawFrameHeader *fh = &priv->frame_header;
|
||
|
|
JPEGRawQuantisationTableSpecification *dqt = &priv->quant_tables;
|
||
|
|
VAQMatrixBufferJPEG *quant;
|
||
|
|
|
||
|
|
if (*data_len < sizeof(*quant))
|
||
|
|
return AVERROR(ENOSPC);
|
||
|
|
*type = VAQMatrixBufferType;
|
||
|
|
*data_len = sizeof(*quant);
|
||
|
|
|
||
|
|
quant = (VAQMatrixBufferJPEG*)data;
|
||
|
|
memset(quant, 0, sizeof(*quant));
|
||
|
|
|
||
|
|
quant->load_lum_quantiser_matrix = 1;
|
||
|
|
for (i = 0; i < 64; i++)
|
||
|
|
quant->lum_quantiser_matrix[i] = dqt->table[fh->Tq[0]].Q[i];
|
||
|
|
|
||
|
|
if (fh->Nf > 1) {
|
||
|
|
quant->load_chroma_quantiser_matrix = 1;
|
||
|
|
for (i = 0; i < 64; i++)
|
||
|
|
quant->chroma_quantiser_matrix[i] =
|
||
|
|
dqt->table[fh->Tq[1]].Q[i];
|
||
|
|
}
|
||
|
|
|
||
|
|
} else if (index == 1) {
|
||
|
|
// Write huffman tables.
|
||
|
|
JPEGRawScanHeader *sh = &priv->scan.header;
|
||
|
|
JPEGRawHuffmanTableSpecification *dht = &priv->huffman_tables;
|
||
|
|
VAHuffmanTableBufferJPEGBaseline *huff;
|
||
|
|
|
||
|
|
if (*data_len < sizeof(*huff))
|
||
|
|
return AVERROR(ENOSPC);
|
||
|
|
*type = VAHuffmanTableBufferType;
|
||
|
|
*data_len = sizeof(*huff);
|
||
|
|
|
||
|
|
huff = (VAHuffmanTableBufferJPEGBaseline*)data;
|
||
|
|
memset(huff, 0, sizeof(*huff));
|
||
|
|
|
||
|
|
for (t = 0; t < 1 + (sh->Ns > 1); t++) {
|
||
|
|
const JPEGRawHuffmanTable *ht;
|
||
|
|
|
||
|
|
huff->load_huffman_table[t] = 1;
|
||
|
|
|
||
|
|
ht = &dht->table[2 * t];
|
||
|
|
for (i = k = 0; i < 16; i++)
|
||
|
|
k += (huff->huffman_table[t].num_dc_codes[i] = ht->L[i]);
|
||
|
|
av_assert0(k <= sizeof(huff->huffman_table[t].dc_values));
|
||
|
|
for (i = 0; i < k; i++)
|
||
|
|
huff->huffman_table[t].dc_values[i] = ht->V[i];
|
||
|
|
|
||
|
|
ht = &dht->table[2 * t + 1];
|
||
|
|
for (i = k = 0; i < 16; i++)
|
||
|
|
k += (huff->huffman_table[t].num_ac_codes[i] = ht->L[i]);
|
||
|
|
av_assert0(k <= sizeof(huff->huffman_table[t].ac_values));
|
||
|
|
for (i = 0; i < k; i++)
|
||
|
|
huff->huffman_table[t].ac_values[i] = ht->V[i];
|
||
|
|
}
|
||
|
|
|
||
|
|
} else {
|
||
|
|
return AVERROR_EOF;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int vaapi_encode_mjpeg_init_picture_params(AVCodecContext *avctx,
|
||
|
|
VAAPIEncodePicture *pic)
|
||
|
|
{
|
||
|
|
VAAPIEncodeMJPEGContext *priv = avctx->priv_data;
|
||
|
|
JPEGRawFrameHeader *fh = &priv->frame_header;
|
||
|
|
JPEGRawScanHeader *sh = &priv->scan.header;
|
||
|
|
VAEncPictureParameterBufferJPEG *vpic = pic->codec_picture_params;
|
||
|
|
const AVPixFmtDescriptor *desc;
|
||
|
|
const uint8_t components_rgb[3] = { 'R', 'G', 'B' };
|
||
|
|
const uint8_t components_yuv[3] = { 1, 2, 3 };
|
||
|
|
const uint8_t *components;
|
||
|
|
int t, i, quant_scale, len;
|
||
|
|
|
||
|
|
av_assert0(pic->type == PICTURE_TYPE_IDR);
|
||
|
|
|
||
|
|
desc = av_pix_fmt_desc_get(priv->common.input_frames->sw_format);
|
||
|
|
av_assert0(desc);
|
||
|
|
if (desc->flags & AV_PIX_FMT_FLAG_RGB)
|
||
|
|
components = components_rgb;
|
||
|
|
else
|
||
|
|
components = components_yuv;
|
||
|
|
|
||
|
|
// Frame header.
|
||
|
|
|
||
|
|
fh->P = 8;
|
||
|
|
fh->Y = avctx->height;
|
||
|
|
fh->X = avctx->width;
|
||
|
|
fh->Nf = desc->nb_components;
|
||
|
|
|
||
|
|
for (i = 0; i < fh->Nf; i++) {
|
||
|
|
fh->C[i] = components[i];
|
||
|
|
fh->H[i] = 1 + (i == 0 ? desc->log2_chroma_w : 0);
|
||
|
|
fh->V[i] = 1 + (i == 0 ? desc->log2_chroma_h : 0);
|
||
|
|
|
||
|
|
fh->Tq[i] = !!i;
|
||
|
|
}
|
||
|
|
|
||
|
|
fh->Lf = 8 + 3 * fh->Nf;
|
||
|
|
|
||
|
|
// JFIF header.
|
||
|
|
if (priv->jfif) {
|
||
|
|
JPEGRawApplicationData *app = &priv->jfif_header;
|
||
|
|
AVRational sar = pic->input_image->sample_aspect_ratio;
|
||
|
|
int sar_w, sar_h;
|
||
|
|
PutByteContext pbc;
|
||
|
|
|
||
|
|
bytestream2_init_writer(&pbc, priv->jfif_data,
|
||
|
|
sizeof(priv->jfif_data));
|
||
|
|
|
||
|
|
bytestream2_put_buffer(&pbc, "JFIF", 5);
|
||
|
|
bytestream2_put_be16(&pbc, 0x0102);
|
||
|
|
bytestream2_put_byte(&pbc, 0);
|
||
|
|
|
||
|
|
av_reduce(&sar_w, &sar_h, sar.num, sar.den, 65535);
|
||
|
|
if (sar_w && sar_h) {
|
||
|
|
bytestream2_put_be16(&pbc, sar_w);
|
||
|
|
bytestream2_put_be16(&pbc, sar_h);
|
||
|
|
} else {
|
||
|
|
bytestream2_put_be16(&pbc, 1);
|
||
|
|
bytestream2_put_be16(&pbc, 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
bytestream2_put_byte(&pbc, 0);
|
||
|
|
bytestream2_put_byte(&pbc, 0);
|
||
|
|
|
||
|
|
av_assert0(bytestream2_get_bytes_left_p(&pbc) == 0);
|
||
|
|
|
||
|
|
app->Lp = 2 + sizeof(priv->jfif_data);
|
||
|
|
app->Ap = priv->jfif_data;
|
||
|
|
app->Ap_ref = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Quantisation tables.
|
||
|
|
|
||
|
|
if (priv->quality < 50)
|
||
|
|
quant_scale = 5000 / priv->quality;
|
||
|
|
else
|
||
|
|
quant_scale = 200 - 2 * priv->quality;
|
||
|
|
|
||
|
|
len = 2;
|
||
|
|
|
||
|
|
for (t = 0; t < 1 + (fh->Nf > 1); t++) {
|
||
|
|
JPEGRawQuantisationTable *quant = &priv->quant_tables.table[t];
|
||
|
|
const uint8_t *data = t == 0 ?
|
||
|
|
vaapi_encode_mjpeg_quant_luminance :
|
||
|
|
vaapi_encode_mjpeg_quant_chrominance;
|
||
|
|
|
||
|
|
quant->Pq = 0;
|
||
|
|
quant->Tq = t;
|
||
|
|
for (i = 0; i < 64; i++)
|
||
|
|
quant->Q[i] = av_clip(data[i] * quant_scale / 100, 1, 255);
|
||
|
|
|
||
|
|
len += 65;
|
||
|
|
}
|
||
|
|
|
||
|
|
priv->quant_tables.Lq = len;
|
||
|
|
|
||
|
|
// Huffman tables.
|
||
|
|
|
||
|
|
len = 2;
|
||
|
|
|
||
|
|
for (t = 0; t < 2 + 2 * (fh->Nf > 1); t++) {
|
||
|
|
JPEGRawHuffmanTable *huff = &priv->huffman_tables.table[t];
|
||
|
|
const uint8_t *lengths, *values;
|
||
|
|
int k;
|
||
|
|
|
||
|
|
switch (t) {
|
||
|
|
case 0:
|
||
|
|
lengths = avpriv_mjpeg_bits_dc_luminance + 1;
|
||
|
|
values = avpriv_mjpeg_val_dc;
|
||
|
|
break;
|
||
|
|
case 1:
|
||
|
|
lengths = avpriv_mjpeg_bits_ac_luminance + 1;
|
||
|
|
values = avpriv_mjpeg_val_ac_luminance;
|
||
|
|
break;
|
||
|
|
case 2:
|
||
|
|
lengths = avpriv_mjpeg_bits_dc_chrominance + 1;
|
||
|
|
values = avpriv_mjpeg_val_dc;
|
||
|
|
break;
|
||
|
|
case 3:
|
||
|
|
lengths = avpriv_mjpeg_bits_ac_chrominance + 1;
|
||
|
|
values = avpriv_mjpeg_val_ac_chrominance;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
huff->Tc = t % 2;
|
||
|
|
huff->Th = t / 2;
|
||
|
|
|
||
|
|
for (i = k = 0; i < 16; i++)
|
||
|
|
k += (huff->L[i] = lengths[i]);
|
||
|
|
|
||
|
|
for (i = 0; i < k; i++)
|
||
|
|
huff->V[i] = values[i];
|
||
|
|
|
||
|
|
len += 17 + k;
|
||
|
|
}
|
||
|
|
|
||
|
|
priv->huffman_tables.Lh = len;
|
||
|
|
|
||
|
|
// Scan header.
|
||
|
|
|
||
|
|
sh->Ns = fh->Nf;
|
||
|
|
|
||
|
|
for (i = 0; i < fh->Nf; i++) {
|
||
|
|
sh->Cs[i] = fh->C[i];
|
||
|
|
sh->Td[i] = i > 0;
|
||
|
|
sh->Ta[i] = i > 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
sh->Ss = 0;
|
||
|
|
sh->Se = 63;
|
||
|
|
sh->Ah = 0;
|
||
|
|
sh->Al = 0;
|
||
|
|
|
||
|
|
sh->Ls = 6 + 2 * sh->Ns;
|
||
|
|
|
||
|
|
|
||
|
|
*vpic = (VAEncPictureParameterBufferJPEG) {
|
||
|
|
.reconstructed_picture = pic->recon_surface,
|
||
|
|
.coded_buf = pic->output_buffer,
|
||
|
|
|
||
|
|
.picture_width = fh->X,
|
||
|
|
.picture_height = fh->Y,
|
||
|
|
|
||
|
|
.pic_flags.bits = {
|
||
|
|
.profile = 0,
|
||
|
|
.progressive = 0,
|
||
|
|
.huffman = 1,
|
||
|
|
.interleaved = 0,
|
||
|
|
.differential = 0,
|
||
|
|
},
|
||
|
|
|
||
|
|
.sample_bit_depth = fh->P,
|
||
|
|
.num_scan = 1,
|
||
|
|
.num_components = fh->Nf,
|
||
|
|
|
||
|
|
// The driver modifies the provided quantisation tables according
|
||
|
|
// to this quality value; the middle value of 50 makes that the
|
||
|
|
// identity so that they are used unchanged.
|
||
|
|
.quality = 50,
|
||
|
|
};
|
||
|
|
|
||
|
|
for (i = 0; i < fh->Nf; i++) {
|
||
|
|
vpic->component_id[i] = fh->C[i];
|
||
|
|
vpic->quantiser_table_selector[i] = fh->Tq[i];
|
||
|
|
}
|
||
|
|
|
||
|
|
pic->nb_slices = 1;
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int vaapi_encode_mjpeg_init_slice_params(AVCodecContext *avctx,
|
||
|
|
VAAPIEncodePicture *pic,
|
||
|
|
VAAPIEncodeSlice *slice)
|
||
|
|
{
|
||
|
|
VAAPIEncodeMJPEGContext *priv = avctx->priv_data;
|
||
|
|
JPEGRawScanHeader *sh = &priv->scan.header;
|
||
|
|
VAEncSliceParameterBufferJPEG *vslice = slice->codec_slice_params;
|
||
|
|
int i;
|
||
|
|
|
||
|
|
*vslice = (VAEncSliceParameterBufferJPEG) {
|
||
|
|
.restart_interval = 0,
|
||
|
|
.num_components = sh->Ns,
|
||
|
|
};
|
||
|
|
|
||
|
|
for (i = 0; i < sh->Ns; i++) {
|
||
|
|
vslice->components[i].component_selector = sh->Cs[i];
|
||
|
|
vslice->components[i].dc_table_selector = sh->Td[i];
|
||
|
|
vslice->components[i].ac_table_selector = sh->Ta[i];
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static av_cold int vaapi_encode_mjpeg_configure(AVCodecContext *avctx)
|
||
|
|
{
|
||
|
|
VAAPIEncodeContext *ctx = avctx->priv_data;
|
||
|
|
VAAPIEncodeMJPEGContext *priv = avctx->priv_data;
|
||
|
|
int err;
|
||
|
|
|
||
|
|
priv->quality = ctx->rc_quality;
|
||
|
|
if (priv->quality < 1 || priv->quality > 100) {
|
||
|
|
av_log(avctx, AV_LOG_ERROR, "Invalid quality value %d "
|
||
|
|
"(must be 1-100).\n", priv->quality);
|
||
|
|
return AVERROR(EINVAL);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Hack: the implementation calls the JPEG image header (which we
|
||
|
|
// will use in the same way as a slice header) generic "raw data".
|
||
|
|
// Therefore, if after the packed header capability check we have
|
||
|
|
// PACKED_HEADER_RAW_DATA available, rewrite it as
|
||
|
|
// PACKED_HEADER_SLICE so that the header-writing code can do the
|
||
|
|
// right thing.
|
||
|
|
if (ctx->va_packed_headers & VA_ENC_PACKED_HEADER_RAW_DATA) {
|
||
|
|
ctx->va_packed_headers &= ~VA_ENC_PACKED_HEADER_RAW_DATA;
|
||
|
|
ctx->va_packed_headers |= VA_ENC_PACKED_HEADER_SLICE;
|
||
|
|
}
|
||
|
|
|
||
|
|
err = ff_cbs_init(&priv->cbc, AV_CODEC_ID_MJPEG, avctx);
|
||
|
|
if (err < 0)
|
||
|
|
return err;
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static const VAAPIEncodeProfile vaapi_encode_mjpeg_profiles[] = {
|
||
|
|
{ FF_PROFILE_MJPEG_HUFFMAN_BASELINE_DCT,
|
||
|
|
8, 1, 0, 0, VAProfileJPEGBaseline },
|
||
|
|
{ FF_PROFILE_MJPEG_HUFFMAN_BASELINE_DCT,
|
||
|
|
8, 3, 1, 1, VAProfileJPEGBaseline },
|
||
|
|
{ FF_PROFILE_MJPEG_HUFFMAN_BASELINE_DCT,
|
||
|
|
8, 3, 1, 0, VAProfileJPEGBaseline },
|
||
|
|
{ FF_PROFILE_MJPEG_HUFFMAN_BASELINE_DCT,
|
||
|
|
8, 3, 0, 0, VAProfileJPEGBaseline },
|
||
|
|
{ FF_PROFILE_UNKNOWN }
|
||
|
|
};
|
||
|
|
|
||
|
|
static const VAAPIEncodeType vaapi_encode_type_mjpeg = {
|
||
|
|
.profiles = vaapi_encode_mjpeg_profiles,
|
||
|
|
|
||
|
|
.flags = FLAG_CONSTANT_QUALITY_ONLY |
|
||
|
|
FLAG_INTRA_ONLY,
|
||
|
|
|
||
|
|
.configure = &vaapi_encode_mjpeg_configure,
|
||
|
|
|
||
|
|
.default_quality = 80,
|
||
|
|
|
||
|
|
.picture_params_size = sizeof(VAEncPictureParameterBufferJPEG),
|
||
|
|
.init_picture_params = &vaapi_encode_mjpeg_init_picture_params,
|
||
|
|
|
||
|
|
.slice_params_size = sizeof(VAEncSliceParameterBufferJPEG),
|
||
|
|
.init_slice_params = &vaapi_encode_mjpeg_init_slice_params,
|
||
|
|
|
||
|
|
.slice_header_type = VAEncPackedHeaderRawData,
|
||
|
|
.write_slice_header = &vaapi_encode_mjpeg_write_image_header,
|
||
|
|
|
||
|
|
.write_extra_buffer = &vaapi_encode_mjpeg_write_extra_buffer,
|
||
|
|
};
|
||
|
|
|
||
|
|
static av_cold int vaapi_encode_mjpeg_init(AVCodecContext *avctx)
|
||
|
|
{
|
||
|
|
VAAPIEncodeContext *ctx = avctx->priv_data;
|
||
|
|
|
||
|
|
ctx->codec = &vaapi_encode_type_mjpeg;
|
||
|
|
|
||
|
|
// The JPEG image header - see note above.
|
||
|
|
ctx->desired_packed_headers =
|
||
|
|
VA_ENC_PACKED_HEADER_RAW_DATA;
|
||
|
|
|
||
|
|
ctx->surface_width = FFALIGN(avctx->width, 8);
|
||
|
|
ctx->surface_height = FFALIGN(avctx->height, 8);
|
||
|
|
|
||
|
|
return ff_vaapi_encode_init(avctx);
|
||
|
|
}
|
||
|
|
|
||
|
|
static av_cold int vaapi_encode_mjpeg_close(AVCodecContext *avctx)
|
||
|
|
{
|
||
|
|
VAAPIEncodeMJPEGContext *priv = avctx->priv_data;
|
||
|
|
|
||
|
|
ff_cbs_fragment_free(priv->cbc, &priv->current_fragment);
|
||
|
|
ff_cbs_close(&priv->cbc);
|
||
|
|
|
||
|
|
return ff_vaapi_encode_close(avctx);
|
||
|
|
}
|
||
|
|
|
||
|
|
#define OFFSET(x) offsetof(VAAPIEncodeMJPEGContext, x)
|
||
|
|
#define FLAGS (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)
|
||
|
|
static const AVOption vaapi_encode_mjpeg_options[] = {
|
||
|
|
VAAPI_ENCODE_COMMON_OPTIONS,
|
||
|
|
|
||
|
|
{ "jfif", "Include JFIF header",
|
||
|
|
OFFSET(jfif), AV_OPT_TYPE_BOOL,
|
||
|
|
{ .i64 = 0 }, 0, 1, FLAGS },
|
||
|
|
{ "huffman", "Include huffman tables",
|
||
|
|
OFFSET(huffman), AV_OPT_TYPE_BOOL,
|
||
|
|
{ .i64 = 1 }, 0, 1, FLAGS },
|
||
|
|
|
||
|
|
{ NULL },
|
||
|
|
};
|
||
|
|
|
||
|
|
static const AVCodecDefault vaapi_encode_mjpeg_defaults[] = {
|
||
|
|
{ "b", "0" },
|
||
|
|
{ NULL },
|
||
|
|
};
|
||
|
|
|
||
|
|
static const AVClass vaapi_encode_mjpeg_class = {
|
||
|
|
.class_name = "mjpeg_vaapi",
|
||
|
|
.item_name = av_default_item_name,
|
||
|
|
.option = vaapi_encode_mjpeg_options,
|
||
|
|
.version = LIBAVUTIL_VERSION_INT,
|
||
|
|
};
|
||
|
|
|
||
|
|
AVCodec ff_mjpeg_vaapi_encoder = {
|
||
|
|
.name = "mjpeg_vaapi",
|
||
|
|
.long_name = NULL_IF_CONFIG_SMALL("MJPEG (VAAPI)"),
|
||
|
|
.type = AVMEDIA_TYPE_VIDEO,
|
||
|
|
.id = AV_CODEC_ID_MJPEG,
|
||
|
|
.priv_data_size = sizeof(VAAPIEncodeMJPEGContext),
|
||
|
|
.init = &vaapi_encode_mjpeg_init,
|
||
|
|
.send_frame = &ff_vaapi_encode_send_frame,
|
||
|
|
.receive_packet = &ff_vaapi_encode_receive_packet,
|
||
|
|
.close = &vaapi_encode_mjpeg_close,
|
||
|
|
.priv_class = &vaapi_encode_mjpeg_class,
|
||
|
|
.capabilities = AV_CODEC_CAP_HARDWARE,
|
||
|
|
.caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
|
||
|
|
.defaults = vaapi_encode_mjpeg_defaults,
|
||
|
|
.pix_fmts = (const enum AVPixelFormat[]) {
|
||
|
|
AV_PIX_FMT_VAAPI,
|
||
|
|
AV_PIX_FMT_NONE,
|
||
|
|
},
|
||
|
|
.hw_configs = ff_vaapi_encode_hw_configs,
|
||
|
|
.wrapper_name = "vaapi",
|
||
|
|
};
|