2 * Copyright (C) 2016 Intel Corporation. All Rights Reserved.
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the
6 * "Software"), to deal in the Software without restriction, including
7 * without limitation the rights to use, copy, modify, merge, publish,
8 * distribute, sub license, and/or sell copies of the Software, and to
9 * permit persons to whom the Software is furnished to do so, subject to
10 * the following conditions:
12 * The above copyright notice and this permission notice (including the
13 * next paragraph) shall be included in all copies or substantial portions
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
17 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
19 * IN NO EVENT SHALL PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR
20 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
21 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
22 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
25 #ifndef I965_JPEG_TEST_DATA_H
26 #define I965_JPEG_TEST_DATA_H
28 #include "i965_test_image_utils.h"
38 typedef std::vector<uint8_t> ByteData;
40 static const VAProfile profile = VAProfileJPEGBaseline;
42 static inline const ByteData generateSolid(
43 const std::array<uint8_t, 3>& yuv, const std::array<size_t, 2>& dim)
45 size_t count(dim[0] * dim[1]);
46 ByteData data(count, yuv[0]);
47 data.insert(data.end(), count, yuv[1]);
48 data.insert(data.end(), count, yuv[2]);
55 typedef VAIQMatrixBufferJPEGBaseline IQMatrix;
56 typedef VAHuffmanTableBufferJPEGBaseline HuffmanTable;
57 typedef VAPictureParameterBufferJPEGBaseline PictureParameter;
58 typedef VASliceParameterBufferJPEGBaseline SliceParameter;
60 static const VAEntrypoint entrypoint = VAEntrypointVLD;
62 static const HuffmanTable defaultHuffmanTable = {
63 load_huffman_table: { 0x01, 0x01 },
67 0x00,0x01,0x05,0x01,0x01,0x01,0x01,0x01,0x01,0x00,0x00,0x00,
71 0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b
74 0x00,0x02,0x01,0x03,0x03,0x02,0x04,0x03,0x05,0x05,0x04,0x04,
78 0x01,0x02,0x03,0x00,0x04,0x11,0x05,0x12,0x21,0x31,0x41,0x06,
79 0x13,0x51,0x61,0x07,0x22,0x71,0x14,0x32,0x81,0x91,0xa1,0x08,
80 0x23,0x42,0xb1,0xc1,0x15,0x52,0xd1,0xf0,0x24,0x33,0x62,0x72,
81 0x82,0x09,0x0a,0x16,0x17,0x18,0x19,0x1a,0x25,0x26,0x27,0x28,
82 0x29,0x2a,0x34,0x35,0x36,0x37,0x38,0x39,0x3a,0x43,0x44,0x45,
83 0x46,0x47,0x48,0x49,0x4a,0x53,0x54,0x55,0x56,0x57,0x58,0x59,
84 0x5a,0x63,0x64,0x65,0x66,0x67,0x68,0x69,0x6a,0x73,0x74,0x75,
85 0x76,0x77,0x78,0x79,0x7a,0x83,0x84,0x85,0x86,0x87,0x88,0x89,
86 0x8a,0x92,0x93,0x94,0x95,0x96,0x97,0x98,0x99,0x9a,0xa2,0xa3,
87 0xa4,0xa5,0xa6,0xa7,0xa8,0xa9,0xaa,0xb2,0xb3,0xb4,0xb5,0xb6,
88 0xb7,0xb8,0xb9,0xba,0xc2,0xc3,0xc4,0xc5,0xc6,0xc7,0xc8,0xc9,
89 0xca,0xd2,0xd3,0xd4,0xd5,0xd6,0xd7,0xd8,0xd9,0xda,0xe1,0xe2,
90 0xe3,0xe4,0xe5,0xe6,0xe7,0xe8,0xe9,0xea,0xf1,0xf2,0xf3,0xf4,
91 0xf5,0xf6,0xf7,0xf8,0xf9,0xfa
97 0x00,0x03,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x00,
101 0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b
104 0x00,0x02,0x01,0x02,0x04,0x04,0x03,0x04,0x07,0x05,0x04,0x04,
108 0x00,0x01,0x02,0x03,0x11,0x04,0x05,0x21,0x31,0x06,0x12,0x41,
109 0x51,0x07,0x61,0x71,0x13,0x22,0x32,0x81,0x08,0x14,0x42,0x91,
110 0xa1,0xb1,0xc1,0x09,0x23,0x33,0x52,0xf0,0x15,0x62,0x72,0xd1,
111 0x0a,0x16,0x24,0x34,0xe1,0x25,0xf1,0x17,0x18,0x19,0x1a,0x26,
112 0x27,0x28,0x29,0x2a,0x35,0x36,0x37,0x38,0x39,0x3a,0x43,0x44,
113 0x45,0x46,0x47,0x48,0x49,0x4a,0x53,0x54,0x55,0x56,0x57,0x58,
114 0x59,0x5a,0x63,0x64,0x65,0x66,0x67,0x68,0x69,0x6a,0x73,0x74,
115 0x75,0x76,0x77,0x78,0x79,0x7a,0x82,0x83,0x84,0x85,0x86,0x87,
116 0x88,0x89,0x8a,0x92,0x93,0x94,0x95,0x96,0x97,0x98,0x99,0x9a,
117 0xa2,0xa3,0xa4,0xa5,0xa6,0xa7,0xa8,0xa9,0xaa,0xb2,0xb3,0xb4,
118 0xb5,0xb6,0xb7,0xb8,0xb9,0xba,0xc2,0xc3,0xc4,0xc5,0xc6,0xc7,
119 0xc8,0xc9,0xca,0xd2,0xd3,0xd4,0xd5,0xd6,0xd7,0xd8,0xd9,0xda,
120 0xe2,0xe3,0xe4,0xe5,0xe6,0xe7,0xe8,0xe9,0xea,0xf2,0xf3,0xf4,
121 0xf5,0xf6,0xf7,0xf8,0xf9,0xfa,
128 static const IQMatrix defaultIQMatrix = { /* Quality 100 */
129 load_quantiser_table: { 0x1,0x1,0x0,0x0 },
132 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
133 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
134 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
135 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
136 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
140 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
141 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
142 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
143 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
144 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
150 static const PictureParameter defaultPictureParameter = {
153 /* component_id, h_sampling_factor, v_sampling_factor, quantiser_table_selector */
154 components: {{1,1,1,0}, {2,1,1,1}, {3,1,1,1}},
158 static const SliceParameter defaultSliceParameter = {
160 slice_data_offset: 0,
161 slice_data_flag: VA_SLICE_DATA_FLAG_ALL,
162 slice_horizontal_position: 0,
163 slice_vertical_position: 0,
165 /* component_selector, dc_table_selector, ac_table_selector */
166 components: {{1,0,0},{2,1,1},{3,1,1}},
176 typedef std::shared_ptr<PictureData> Shared;
177 typedef std::shared_ptr<const PictureData> SharedConst;
179 template<const unsigned W, const unsigned H>
180 static SharedConst make(
181 const unsigned fourcc,
182 const ByteData& slice,
183 const SliceParameter& sparam = defaultSliceParameter,
184 const PictureParameter& pparam = defaultPictureParameter,
185 const HuffmanTable& huffman = defaultHuffmanTable,
186 const IQMatrix& iqmatrix = defaultIQMatrix)
188 return make(fourcc, slice, W, H, sparam, pparam, huffman, iqmatrix);
191 static SharedConst make(
192 const unsigned fourcc,
193 const ByteData& slice,
194 const unsigned w, const unsigned h,
195 const SliceParameter& sparam = defaultSliceParameter,
196 const PictureParameter& pparam = defaultPictureParameter,
197 const HuffmanTable& huffman = defaultHuffmanTable,
198 const IQMatrix& iqmatrix = defaultIQMatrix)
212 pd->sparam.slice_data_size = slice.size();
213 pd->pparam.picture_width = w;
214 pd->pparam.picture_height = h;
219 pd->format = VA_RT_FORMAT_YUV420;
220 pd->pparam.components[0].h_sampling_factor = 2;
221 pd->pparam.components[0].v_sampling_factor = 2;
224 pd->format = VA_RT_FORMAT_YUV422;
225 pd->pparam.components[0].h_sampling_factor = 2;
226 pd->pparam.components[0].v_sampling_factor = 1;
229 pd->format = VA_RT_FORMAT_YUV422;
230 pd->pparam.components[0].h_sampling_factor = 1;
231 pd->pparam.components[0].v_sampling_factor = 2;
234 pd->format = VA_RT_FORMAT_YUV411;
235 pd->pparam.components[0].h_sampling_factor = 4;
236 pd->pparam.components[0].v_sampling_factor = 1;
239 pd->format = VA_RT_FORMAT_YUV444;
240 pd->pparam.components[0].h_sampling_factor = 1;
241 pd->pparam.components[0].v_sampling_factor = 1;
244 pd->format = VA_RT_FORMAT_YUV400;
245 pd->pparam.components[0].h_sampling_factor = 1;
246 pd->pparam.components[0].h_sampling_factor = 1;
247 pd->pparam.num_components = 1;
248 pd->sparam.num_components = 1;
254 /* Calculate num_mcus */
255 int hfactor = pd->pparam.components[0].h_sampling_factor << 3;
256 int vfactor = pd->pparam.components[0].v_sampling_factor << 3;
257 int wmcu = (w + hfactor - 1) / hfactor;
258 int hmcu = (h + vfactor - 1) / vfactor;
259 pd->sparam.num_mcus = wmcu * hmcu;
264 const ByteData slice;
265 SliceParameter sparam;
266 PictureParameter pparam;
267 HuffmanTable huffman;
276 typedef std::shared_ptr<TestPattern> Shared;
277 typedef std::shared_ptr<const TestPattern> SharedConst;
279 virtual ~TestPattern() { }
281 virtual const ByteData& decoded() const = 0;
282 virtual PictureData::SharedConst encoded(unsigned) const = 0;
284 virtual void repr(std::ostream&) const = 0;
286 friend std::ostream& operator <<(std::ostream& os, const TestPattern& t)
293 template <const unsigned N>
294 class TestPatternData
298 typedef std::map<const unsigned, PictureData::SharedConst> EncodedMap;
299 typedef std::map<const unsigned, const ByteData> ByteDataMap;
302 const ByteData& decoded() const { return getDecoded(); }
304 PictureData::SharedConst encoded(const unsigned fourcc) const
306 const EncodedMap& em = getEncodedMap();
307 const EncodedMap::const_iterator match(em.find(fourcc));
308 if (match == em.end())
309 return PictureData::SharedConst();
310 return match->second;
313 void repr(std::ostream& os) const { os << N; }
315 template <const unsigned W, const unsigned H>
316 static bool initialize(ByteData d, const ByteDataMap& e)
318 getDecoded().swap(d);
320 EncodedMap& em = getEncodedMap();
322 for (auto const &b : e) {
323 auto pd(PictureData::make<W, H>(b.first, b.second));
324 auto pair = std::make_pair(b.first, pd);
325 result &= em.insert(pair).second;
331 static ByteData& getDecoded()
337 static EncodedMap& getEncodedMap()
339 static EncodedMap em;
343 static const bool m_valid;
345 } // namespace Decode
350 typedef VAQMatrixBufferJPEG IQMatrix;
351 typedef VAHuffmanTableBufferJPEGBaseline HuffmanTable;
352 typedef VAEncPictureParameterBufferJPEG PictureParameter;
353 typedef VAEncSliceParameterBufferJPEG SliceParameter;
355 static const VAEntrypoint entrypoint = VAEntrypointEncPicture;
357 static const IQMatrix defaultIQMatrix = { /* Quality 50 */
358 load_lum_quantiser_matrix: 1,
359 load_chroma_quantiser_matrix: 1,
360 lum_quantiser_matrix: {
361 0x10,0x0b,0x0c,0x0e,0x0c,0x0a,0x10,0x0e,
362 0x0d,0x0e,0x12,0x11,0x10,0x13,0x18,0x28,
363 0x1a,0x18,0x16,0x16,0x18,0x31,0x23,0x25,
364 0x1d,0x28,0x3a,0x33,0x3d,0x3c,0x39,0x33,
365 0x38,0x37,0x40,0x48,0x5c,0x4e,0x40,0x44,
366 0x57,0x45,0x37,0x38,0x50,0x6d,0x51,0x57,
367 0x5f,0x62,0x67,0x68,0x67,0x3e,0x4d,0x71,
368 0x79,0x70,0x64,0x78,0x5c,0x65,0x67,0x63,
370 chroma_quantiser_matrix: {
371 0x11,0x12,0x12,0x18,0x15,0x18,0x2f,0x1a,
372 0x1a,0x2f,0x63,0x42,0x38,0x42,0x63,0x63,
373 0x63,0x63,0x63,0x63,0x63,0x63,0x63,0x63,
374 0x63,0x63,0x63,0x63,0x63,0x63,0x63,0x63,
375 0x63,0x63,0x63,0x63,0x63,0x63,0x63,0x63,
376 0x63,0x63,0x63,0x63,0x63,0x63,0x63,0x63,
377 0x63,0x63,0x63,0x63,0x63,0x63,0x63,0x63,
378 0x63,0x63,0x63,0x63,0x63,0x63,0x63,0x63,
382 static const HuffmanTable defaultHuffmanTable =
383 ::JPEG::Decode::defaultHuffmanTable;
385 static const PictureParameter defaultPictureParameter = {
386 reconstructed_picture: VA_INVALID_ID,
389 coded_buf: VA_INVALID_ID,
390 pic_flags: {value: 0x00100},
394 component_id: {0, 1, 2, 0},
395 quantiser_table_selector: {0, 1, 1, 0},
399 static const SliceParameter defaultSliceParameter = {
402 /* component_selector, dc_table_selector, ac_table_selector */
403 components: {{1,0,0},{2,1,1},{3,1,1}},
407 : public std::enable_shared_from_this<TestInput>
410 typedef std::shared_ptr<TestInput> Shared;
411 typedef std::shared_ptr<const TestInput> SharedConst;
413 static Shared create(const unsigned, const unsigned, const unsigned);
414 const YUVImage::SharedConst toExpectedOutput() const;
416 friend ::std::ostream& operator<<(::std::ostream&, const TestInput&);
417 friend ::std::ostream& operator<<(::std::ostream&, const Shared&);
418 friend ::std::ostream& operator<<(::std::ostream&, const SharedConst&);
420 YUVImage::Shared image;
421 PictureParameter picture;
423 HuffmanTable huffman;
424 SliceParameter slice;
430 class TestInputCreator
433 typedef std::shared_ptr<TestInputCreator> Shared;
434 typedef std::shared_ptr<const TestInputCreator> SharedConst;
436 virtual ~TestInputCreator() { }
438 TestInput::Shared create(const unsigned) const;
440 friend ::std::ostream& operator<<(
441 ::std::ostream&, const TestInputCreator&);
442 friend ::std::ostream& operator<<(
443 ::std::ostream&, const TestInputCreator::Shared&);
444 friend ::std::ostream& operator<<(
445 ::std::ostream&, const TestInputCreator::SharedConst&);
446 virtual std::array<unsigned, 2> getResolution() const = 0;
449 virtual void repr(::std::ostream& os) const = 0;
452 class RandomSizeCreator
453 : public TestInputCreator
456 std::array<unsigned, 2> getResolution() const;
459 void repr(::std::ostream&) const;
462 class FixedSizeCreator
463 : public TestInputCreator
466 FixedSizeCreator(const std::array<unsigned, 2>&);
467 std::array<unsigned, 2> getResolution() const;
470 void repr(::std::ostream& os) const;
473 const std::array<unsigned, 2> res;
476 typedef std::vector<TestInputCreator::SharedConst> InputCreators;
478 } // namespace Encode