2 * Copyright (C) 2019 Seungha Yang <seungha.yang@navercorp.com>
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Library General Public
6 * License as published by the Free Software Foundation; either
7 * version 2 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Library General Public License for more details.
14 * You should have received a copy of the GNU Library General Public
15 * License along with this library; if not, write to the
16 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
17 * Boston, MA 02110-1301, USA.
20 #include <gst/check/gstcheck.h>
21 #include <gst/check/gstharness.h>
24 /* 64x64 h264 byte-stream format keyframe with sps/pps
25 * encoded using x264enc with videotestsrc */
26 static guint8 h264_frame[] = {
27 0x00, 0x00, 0x00, 0x01, 0x09, 0x10, 0x00, 0x00, 0x00, 0x01, 0x67, 0x64,
28 0x00, 0x14, 0xac, 0xd9, 0x44, 0x26, 0xc0, 0x5a, 0x83, 0x00, 0x83, 0x52,
29 0x80, 0x00, 0x00, 0x03, 0x00, 0x80, 0x00, 0x00, 0x1e, 0x47, 0x8a, 0x14,
30 0xcb, 0x00, 0x00, 0x00, 0x01, 0x68, 0xeb, 0xec, 0xb2, 0x2c, 0x00, 0x00,
31 0x00, 0x01, 0x06, 0x05, 0xff, 0xff, 0xf5, 0xdc, 0x45, 0xe9, 0xbd, 0xe6,
32 0xd9, 0x48, 0xb7, 0x96, 0x2c, 0xd8, 0x20, 0xd9, 0x23, 0xee, 0xef, 0x78,
33 0x32, 0x36, 0x34, 0x20, 0x2d, 0x20, 0x63, 0x6f, 0x72, 0x65, 0x20, 0x31,
34 0x35, 0x37, 0x20, 0x72, 0x32, 0x39, 0x34, 0x35, 0x2b, 0x34, 0x35, 0x20,
35 0x32, 0x37, 0x35, 0x36, 0x35, 0x64, 0x31, 0x20, 0x2d, 0x20, 0x48, 0x2e,
36 0x32, 0x36, 0x34, 0x2f, 0x4d, 0x50, 0x45, 0x47, 0x2d, 0x34, 0x20, 0x41,
37 0x56, 0x43, 0x20, 0x63, 0x6f, 0x64, 0x65, 0x63, 0x20, 0x2d, 0x20, 0x43,
38 0x6f, 0x70, 0x79, 0x6c, 0x65, 0x66, 0x74, 0x20, 0x32, 0x30, 0x30, 0x33,
39 0x2d, 0x32, 0x30, 0x31, 0x38, 0x20, 0x2d, 0x20, 0x68, 0x74, 0x74, 0x70,
40 0x3a, 0x2f, 0x2f, 0x77, 0x77, 0x77, 0x2e, 0x76, 0x69, 0x64, 0x65, 0x6f,
41 0x6c, 0x61, 0x6e, 0x2e, 0x6f, 0x72, 0x67, 0x2f, 0x78, 0x32, 0x36, 0x34,
42 0x2e, 0x68, 0x74, 0x6d, 0x6c, 0x20, 0x2d, 0x20, 0x6f, 0x70, 0x74, 0x69,
43 0x6f, 0x6e, 0x73, 0x3a, 0x20, 0x63, 0x61, 0x62, 0x61, 0x63, 0x3d, 0x31,
44 0x20, 0x72, 0x65, 0x66, 0x3d, 0x33, 0x20, 0x64, 0x65, 0x62, 0x6c, 0x6f,
45 0x63, 0x6b, 0x3d, 0x31, 0x3a, 0x30, 0x3a, 0x30, 0x20, 0x61, 0x6e, 0x61,
46 0x6c, 0x79, 0x73, 0x65, 0x3d, 0x30, 0x78, 0x33, 0x3a, 0x30, 0x78, 0x31,
47 0x31, 0x33, 0x20, 0x6d, 0x65, 0x3d, 0x68, 0x65, 0x78, 0x20, 0x73, 0x75,
48 0x62, 0x6d, 0x65, 0x3d, 0x37, 0x20, 0x70, 0x73, 0x79, 0x3d, 0x31, 0x20,
49 0x70, 0x73, 0x79, 0x5f, 0x72, 0x64, 0x3d, 0x31, 0x2e, 0x30, 0x30, 0x3a,
50 0x30, 0x2e, 0x30, 0x30, 0x20, 0x6d, 0x69, 0x78, 0x65, 0x64, 0x5f, 0x72,
51 0x65, 0x66, 0x3d, 0x31, 0x20, 0x6d, 0x65, 0x5f, 0x72, 0x61, 0x6e, 0x67,
52 0x65, 0x3d, 0x31, 0x36, 0x20, 0x63, 0x68, 0x72, 0x6f, 0x6d, 0x61, 0x5f,
53 0x6d, 0x65, 0x3d, 0x31, 0x20, 0x74, 0x72, 0x65, 0x6c, 0x6c, 0x69, 0x73,
54 0x3d, 0x31, 0x20, 0x38, 0x78, 0x38, 0x64, 0x63, 0x74, 0x3d, 0x31, 0x20,
55 0x63, 0x71, 0x6d, 0x3d, 0x30, 0x20, 0x64, 0x65, 0x61, 0x64, 0x7a, 0x6f,
56 0x6e, 0x65, 0x3d, 0x32, 0x31, 0x2c, 0x31, 0x31, 0x20, 0x66, 0x61, 0x73,
57 0x74, 0x5f, 0x70, 0x73, 0x6b, 0x69, 0x70, 0x3d, 0x31, 0x20, 0x63, 0x68,
58 0x72, 0x6f, 0x6d, 0x61, 0x5f, 0x71, 0x70, 0x5f, 0x6f, 0x66, 0x66, 0x73,
59 0x65, 0x74, 0x3d, 0x2d, 0x32, 0x20, 0x74, 0x68, 0x72, 0x65, 0x61, 0x64,
60 0x73, 0x3d, 0x32, 0x20, 0x6c, 0x6f, 0x6f, 0x6b, 0x61, 0x68, 0x65, 0x61,
61 0x64, 0x5f, 0x74, 0x68, 0x72, 0x65, 0x61, 0x64, 0x73, 0x3d, 0x31, 0x20,
62 0x73, 0x6c, 0x69, 0x63, 0x65, 0x64, 0x5f, 0x74, 0x68, 0x72, 0x65, 0x61,
63 0x64, 0x73, 0x3d, 0x30, 0x20, 0x6e, 0x72, 0x3d, 0x30, 0x20, 0x64, 0x65,
64 0x63, 0x69, 0x6d, 0x61, 0x74, 0x65, 0x3d, 0x31, 0x20, 0x69, 0x6e, 0x74,
65 0x65, 0x72, 0x6c, 0x61, 0x63, 0x65, 0x64, 0x3d, 0x30, 0x20, 0x62, 0x6c,
66 0x75, 0x72, 0x61, 0x79, 0x5f, 0x63, 0x6f, 0x6d, 0x70, 0x61, 0x74, 0x3d,
67 0x30, 0x20, 0x63, 0x6f, 0x6e, 0x73, 0x74, 0x72, 0x61, 0x69, 0x6e, 0x65,
68 0x64, 0x5f, 0x69, 0x6e, 0x74, 0x72, 0x61, 0x3d, 0x30, 0x20, 0x62, 0x66,
69 0x72, 0x61, 0x6d, 0x65, 0x73, 0x3d, 0x33, 0x20, 0x62, 0x5f, 0x70, 0x79,
70 0x72, 0x61, 0x6d, 0x69, 0x64, 0x3d, 0x32, 0x20, 0x62, 0x5f, 0x61, 0x64,
71 0x61, 0x70, 0x74, 0x3d, 0x31, 0x20, 0x62, 0x5f, 0x62, 0x69, 0x61, 0x73,
72 0x3d, 0x30, 0x20, 0x64, 0x69, 0x72, 0x65, 0x63, 0x74, 0x3d, 0x31, 0x20,
73 0x77, 0x65, 0x69, 0x67, 0x68, 0x74, 0x62, 0x3d, 0x31, 0x20, 0x6f, 0x70,
74 0x65, 0x6e, 0x5f, 0x67, 0x6f, 0x70, 0x3d, 0x30, 0x20, 0x77, 0x65, 0x69,
75 0x67, 0x68, 0x74, 0x70, 0x3d, 0x32, 0x20, 0x6b, 0x65, 0x79, 0x69, 0x6e,
76 0x74, 0x3d, 0x33, 0x30, 0x30, 0x20, 0x6b, 0x65, 0x79, 0x69, 0x6e, 0x74,
77 0x5f, 0x6d, 0x69, 0x6e, 0x3d, 0x33, 0x30, 0x20, 0x73, 0x63, 0x65, 0x6e,
78 0x65, 0x63, 0x75, 0x74, 0x3d, 0x34, 0x30, 0x20, 0x69, 0x6e, 0x74, 0x72,
79 0x61, 0x5f, 0x72, 0x65, 0x66, 0x72, 0x65, 0x73, 0x68, 0x3d, 0x30, 0x20,
80 0x72, 0x63, 0x5f, 0x6c, 0x6f, 0x6f, 0x6b, 0x61, 0x68, 0x65, 0x61, 0x64,
81 0x3d, 0x34, 0x30, 0x20, 0x72, 0x63, 0x3d, 0x63, 0x62, 0x72, 0x20, 0x6d,
82 0x62, 0x74, 0x72, 0x65, 0x65, 0x3d, 0x31, 0x20, 0x62, 0x69, 0x74, 0x72,
83 0x61, 0x74, 0x65, 0x3d, 0x32, 0x30, 0x34, 0x38, 0x20, 0x72, 0x61, 0x74,
84 0x65, 0x74, 0x6f, 0x6c, 0x3d, 0x31, 0x2e, 0x30, 0x20, 0x71, 0x63, 0x6f,
85 0x6d, 0x70, 0x3d, 0x30, 0x2e, 0x36, 0x30, 0x20, 0x71, 0x70, 0x6d, 0x69,
86 0x6e, 0x3d, 0x30, 0x20, 0x71, 0x70, 0x6d, 0x61, 0x78, 0x3d, 0x36, 0x39,
87 0x20, 0x71, 0x70, 0x73, 0x74, 0x65, 0x70, 0x3d, 0x34, 0x20, 0x76, 0x62,
88 0x76, 0x5f, 0x6d, 0x61, 0x78, 0x72, 0x61, 0x74, 0x65, 0x3d, 0x32, 0x30,
89 0x34, 0x38, 0x20, 0x76, 0x62, 0x76, 0x5f, 0x62, 0x75, 0x66, 0x73, 0x69,
90 0x7a, 0x65, 0x3d, 0x31, 0x32, 0x32, 0x38, 0x20, 0x6e, 0x61, 0x6c, 0x5f,
91 0x68, 0x72, 0x64, 0x3d, 0x6e, 0x6f, 0x6e, 0x65, 0x20, 0x66, 0x69, 0x6c,
92 0x6c, 0x65, 0x72, 0x3d, 0x30, 0x20, 0x69, 0x70, 0x5f, 0x72, 0x61, 0x74,
93 0x69, 0x6f, 0x3d, 0x31, 0x2e, 0x34, 0x30, 0x20, 0x61, 0x71, 0x3d, 0x31,
94 0x3a, 0x31, 0x2e, 0x30, 0x30, 0x00, 0x80, 0x00, 0x00, 0x00, 0x01, 0x65,
95 0x88, 0x84, 0x00, 0x1a, 0xff, 0xfe, 0xf7, 0xd4, 0xb7, 0xcc, 0xb2, 0xee,
96 0x07, 0x22, 0xb6, 0x0b, 0xa8, 0xf7, 0xa2, 0x9c, 0x8c, 0x81, 0xee, 0x0c,
97 0x19, 0x51, 0xa2, 0x7c, 0x81, 0xbd, 0x7b, 0x46, 0xcd
100 static guint h264_frame_len = 849;
102 #define MAX_PUSH_BUFFER 64
104 GST_START_TEST (test_downstream_reconfigure)
107 GstElement *capsfilter;
109 GstBuffer *in_buf, *out_buf = NULL;
114 h = gst_harness_new_parse ("h264parse ! nvh264dec ! capsfilter");
115 fail_unless (h != NULL);
117 capsfilter = gst_harness_find_element (h, "capsfilter");
119 gst_harness_play (h);
121 gst_harness_set_src_caps_str (h, "video/x-h264,width=64,height=64,"
122 "stream-format=byte-stream,alignment=au");
124 /* enforce system memory */
125 caps = gst_caps_from_string ("video/x-raw");
126 g_object_set (capsfilter, "caps", caps, NULL);
127 gst_caps_unref (caps);
129 in_buf = gst_buffer_new_and_alloc (h264_frame_len);
130 gst_buffer_map (in_buf, &map, GST_MAP_WRITE);
131 memcpy (map.data, h264_frame, h264_frame_len);
132 gst_buffer_unmap (in_buf, &map);
134 GST_BUFFER_DURATION (in_buf) = GST_SECOND;
136 /* Push bufffers until get decoder output */
138 fail_if (i > MAX_PUSH_BUFFER);
140 GST_BUFFER_PTS (in_buf) = GST_BUFFER_DTS (in_buf) = i * GST_SECOND;
142 ret = gst_harness_push (h, gst_buffer_ref (in_buf));
143 fail_unless (ret == GST_FLOW_OK, "GstFlowReturn was %s",
144 gst_flow_get_name (ret));
146 out_buf = gst_harness_try_pull (h);
148 } while (out_buf == NULL);
149 gst_buffer_unref (out_buf);
151 /* Use opengl memory */
152 caps = gst_caps_from_string ("video/x-raw(memory:GLMemory)");
153 g_object_set (capsfilter, "caps", caps, NULL);
154 gst_caps_unref (caps);
157 fail_if (i > 2 * MAX_PUSH_BUFFER);
159 GST_BUFFER_PTS (in_buf) = GST_BUFFER_DTS (in_buf) = i * GST_SECOND;
161 ret = gst_harness_push (h, gst_buffer_ref (in_buf));
162 fail_unless (ret == GST_FLOW_OK, "GstFlowReturn was %s",
163 gst_flow_get_name (ret));
165 out_buf = gst_harness_try_pull (h);
167 } while (out_buf == NULL);
168 gst_buffer_unref (in_buf);
169 gst_buffer_unref (out_buf);
171 gst_object_unref (capsfilter);
172 gst_harness_teardown (h);
178 check_nvcodec_available (void)
183 nvdec = gst_element_factory_make ("nvh264dec", NULL);
185 GST_WARNING ("nvh264dec is not available, possibly driver load failure");
189 /* GST_STATE_READY is meaning that driver could be loaded */
190 if (gst_element_set_state (nvdec,
191 GST_STATE_PAUSED) != GST_STATE_CHANGE_SUCCESS) {
192 GST_WARNING ("cannot open device");
196 gst_element_set_state (nvdec, GST_STATE_NULL);
197 gst_object_unref (nvdec);
208 /* HACK: cuda device init/deinit with fork seems to problematic */
209 g_setenv ("CK_FORK", "no", TRUE);
211 s = suite_create ("nvdec");
212 tc_chain = tcase_create ("general");
214 suite_add_tcase (s, tc_chain);
216 if (!check_nvcodec_available ()) {
217 GST_DEBUG ("Skip nvdec test since cannot open device");
221 tcase_add_test (tc_chain, test_downstream_reconfigure);
227 GST_CHECK_MAIN (nvdec);