gstreamer/subprojects/gst-plugins-bad/tests/check/elements/fdkaac.c

272 lines
7.6 KiB
C
Raw Normal View History

/* GStreamer unit test for fdk-aac elements
*
* Copyright (C) 2022 Tim-Philipp Müller <tim centricular com>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library 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
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <gst/check/check.h>
#include <gst/app/app.h>
#include <gst/pbutils/pbutils.h>
typedef struct
{
const gchar *base_profile;
const gchar *profile;
guint8 buf_hdr[16];
guint8 codec_data[16];
gsize codec_data_len;
} AacSample;
static AacSample
encode_with_caps (const gchar * caps_str)
{
GstElement *pipe, *sink;
gchar *pipeline_str;
GError *err = NULL;
GstSample *sample;
AacSample aac_sample = { NULL, };
GST_INFO ("Encoding to %s", caps_str);
pipeline_str =
g_strdup_printf ("audiotestsrc ! fdkaacenc ! appsink name=sink caps=%s",
caps_str);
pipe = gst_parse_launch (pipeline_str, &err);
g_free (pipeline_str);
fail_if (err != NULL, "Error creating pipeline: %s", err->message);
sink = gst_bin_get_by_name (GST_BIN (pipe), "sink");
gst_element_set_state (pipe, GST_STATE_PLAYING);
sample = gst_app_sink_pull_preroll (GST_APP_SINK (sink));
fail_unless (sample != NULL);
{
GstCaps *caps = gst_sample_get_caps (sample);
GstStructure *s = gst_caps_get_structure (caps, 0);
const GValue *val;
GstMapInfo map;
GstBuffer *buf;
GST_INFO ("Got caps %" GST_PTR_FORMAT, caps);
aac_sample.base_profile =
g_intern_string (gst_structure_get_string (s, "base-profile"));
aac_sample.profile =
g_intern_string (gst_structure_get_string (s, "profile"));
val = gst_structure_get_value (s, "codec_data");
if (val != NULL) {
buf = gst_value_get_buffer (val);
gst_buffer_map (buf, &map, GST_MAP_READ);
g_assert (map.size <= 16);
memcpy (aac_sample.codec_data, map.data, map.size);
aac_sample.codec_data_len = map.size;
gst_buffer_unmap (buf, &map);
} else {
aac_sample.codec_data_len = 0;
}
buf = gst_sample_get_buffer (sample);
gst_buffer_map (buf, &map, GST_MAP_READ);
g_assert (map.size >= sizeof (aac_sample.buf_hdr));
memcpy (aac_sample.buf_hdr, map.data, sizeof (aac_sample.buf_hdr));
gst_buffer_unmap (buf, &map);
}
gst_sample_unref (sample);
gst_element_set_state (pipe, GST_STATE_NULL);
gst_object_unref (sink);
gst_object_unref (pipe);
return aac_sample;
}
typedef enum
{
AAC_AOT_UNKNOWN = -1,
AAC_AOT_MAIN = 1,
AAC_AOT_LC = 2,
AAC_AOT_SSR = 3,
AAC_AOT_LTP = 4,
AAC_AOT_SBR = 5,
AAC_AOT_LD = 23,
AAC_AOT_PS = 29,
} AacAOT;
static AacAOT
aac_codec_data_get_aot (const guint8 * codec_data, gsize codec_data_len)
{
AacAOT aot = AAC_AOT_UNKNOWN;
if (codec_data_len >= 2) {
guint8 hdr_aot = codec_data[0] >> 3; // 5 bits
switch (hdr_aot) {
case AAC_AOT_MAIN:
case AAC_AOT_LC:
case AAC_AOT_SSR:
case AAC_AOT_LTP:
case AAC_AOT_SBR:
case AAC_AOT_LD:
case AAC_AOT_PS:
aot = hdr_aot;
break;
default:
break;
}
}
return aot;
}
GST_START_TEST (test_fdkaacenc_adts)
{
// LC profile
{
AacSample aac;
aac = encode_with_caps ("audio/mpeg,profile=lc,stream-format=adts");
fail_unless_equals_string (aac.base_profile, "lc");
fail_unless_equals_string (aac.profile, "lc");
fail_unless (aac.codec_data_len == 0);
// ADTS sync
fail_unless_equals_int (GST_READ_UINT16_BE (aac.buf_hdr) & 0xfff0, 0xfff0);
// MPEG version (0 = MPEG-4)
fail_unless_equals_int (aac.buf_hdr[1] & 0x08, 0);
// AOT minus 1 (2=LC)
fail_unless_equals_int ((aac.buf_hdr[2] & 0xC0) >> 6, 2 - 1);
}
// HE-AACv1 profile
{
AacSample aac;
aac = encode_with_caps ("audio/mpeg,profile=he-aac-v1,stream-format=adts");
fail_unless_equals_string (aac.base_profile, "lc");
fail_unless_equals_string (aac.profile, "he-aac-v1");
fail_unless (aac.codec_data_len == 0);
// ADTS sync
fail_unless_equals_int (GST_READ_UINT16_BE (aac.buf_hdr) & 0xfff0, 0xfff0);
// MPEG version (0 = MPEG-4)
fail_unless_equals_int (aac.buf_hdr[1] & 0x08, 0);
// AOT minus 1 (2=LC)
fail_unless_equals_int ((aac.buf_hdr[2] & 0xC0) >> 6, 2 - 1);
}
// HE-AACv2 profile
{
AacSample aac;
aac = encode_with_caps ("audio/mpeg,profile=he-aac-v2,stream-format=adts");
fail_unless_equals_string (aac.base_profile, "lc");
fail_unless_equals_string (aac.profile, "he-aac-v2");
fail_unless (aac.codec_data_len == 0);
// ADTS sync
fail_unless_equals_int (GST_READ_UINT16_BE (aac.buf_hdr) & 0xfff0, 0xfff0);
// MPEG version (0 = MPEG-4)
fail_unless_equals_int (aac.buf_hdr[1] & 0x08, 0);
// AOT minus 1 (2=LC)
fail_unless_equals_int ((aac.buf_hdr[2] & 0xC0) >> 6, 2 - 1);
}
}
GST_END_TEST;
GST_START_TEST (test_fdkaacenc_raw)
{
const gchar *profile;
AacAOT aot;
// LC profile
{
AacSample aac;
aac = encode_with_caps ("audio/mpeg,profile=lc,stream-format=raw");
fail_unless_equals_string (aac.base_profile, "lc");
fail_unless_equals_string (aac.profile, "lc");
fail_unless (aac.codec_data_len == 2);
profile =
gst_codec_utils_aac_get_profile (aac.codec_data, aac.codec_data_len);
fail_unless_equals_string (profile, "lc");
aot = aac_codec_data_get_aot (aac.codec_data, aac.codec_data_len);
fail_unless_equals_int (aot, AAC_AOT_LC);
}
// HE-AACv1 profile
{
AacSample aac;
aac = encode_with_caps ("audio/mpeg,profile=he-aac-v1,stream-format=raw");
// For raw we use non-backwards compatible hierarchical signaling, so
// caps should not contain a base-profile field
fail_if (aac.base_profile != NULL);
fail_unless (aac.profile != NULL);
fail_unless (aac.codec_data_len > 2);
profile =
gst_codec_utils_aac_get_profile (aac.codec_data, aac.codec_data_len);
fail_unless_equals_string (profile, "lc"); // FIXME: is this right for non-BC hierarchical signaling?
aot = aac_codec_data_get_aot (aac.codec_data, aac.codec_data_len);
fail_unless_equals_int (aot, AAC_AOT_SBR);
}
// HE-AACv2 profile
{
AacSample aac;
aac = encode_with_caps ("audio/mpeg,profile=he-aac-v2,stream-format=raw");
// For raw we use non-backwards compatible hierarchical signaling, so
// caps should not contain a base-profile field
fail_if (aac.base_profile != NULL);
fail_unless (aac.profile != NULL);
fail_unless (aac.codec_data_len > 2);
profile =
gst_codec_utils_aac_get_profile (aac.codec_data, aac.codec_data_len);
fail_unless_equals_string (profile, "lc"); // FIXME: is this right for non-BC hierarchical signaling?
aot = aac_codec_data_get_aot (aac.codec_data, aac.codec_data_len);
fail_unless_equals_int (aot, AAC_AOT_PS);
}
}
GST_END_TEST;
static Suite *
fdkaac_suite (void)
{
Suite *s = suite_create ("fdk-aac");
TCase *tc_chain_enc = tcase_create ("fdkaacenc");
suite_add_tcase (s, tc_chain_enc);
tcase_add_test (tc_chain_enc, test_fdkaacenc_adts);
tcase_add_test (tc_chain_enc, test_fdkaacenc_raw);
return s;
}
GST_CHECK_MAIN (fdkaac);