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 const ByteData& decoded() const = 0;
280 virtual PictureData::SharedConst encoded(unsigned) const = 0;
282 virtual void repr(std::ostream&) const = 0;
284 friend std::ostream& operator <<(std::ostream& os, const TestPattern& t)
291 template <const unsigned N>
292 class TestPatternData
296 typedef std::map<const unsigned, PictureData::SharedConst> EncodedMap;
297 typedef std::map<const unsigned, const ByteData> ByteDataMap;
300 const ByteData& decoded() const { return getDecoded(); }
302 PictureData::SharedConst encoded(const unsigned fourcc) const
304 const EncodedMap& em = getEncodedMap();
305 const EncodedMap::const_iterator match(em.find(fourcc));
306 if (match == em.end())
307 return PictureData::SharedConst();
308 return match->second;
311 void repr(std::ostream& os) const { os << N; }
313 template <const unsigned W, const unsigned H>
314 static bool initialize(ByteData d, const ByteDataMap& e)
316 getDecoded().swap(d);
318 EncodedMap& em = getEncodedMap();
320 for (auto const &b : e) {
321 auto pd(PictureData::make<W, H>(b.first, b.second));
322 auto pair = std::make_pair(b.first, pd);
323 result &= em.insert(pair).second;
329 static ByteData& getDecoded()
335 static EncodedMap& getEncodedMap()
337 static EncodedMap em;
341 static const bool m_valid;
343 } // namespace Decode
348 typedef VAQMatrixBufferJPEG IQMatrix;
349 typedef VAHuffmanTableBufferJPEGBaseline HuffmanTable;
350 typedef VAEncPictureParameterBufferJPEG PictureParameter;
351 typedef VAEncSliceParameterBufferJPEG SliceParameter;
353 static const VAEntrypoint entrypoint = VAEntrypointEncPicture;
355 static const IQMatrix defaultIQMatrix = { /* Quality 50 */
356 load_lum_quantiser_matrix: 1,
357 load_chroma_quantiser_matrix: 1,
358 lum_quantiser_matrix: {
359 0x10,0x0b,0x0c,0x0e,0x0c,0x0a,0x10,0x0e,
360 0x0d,0x0e,0x12,0x11,0x10,0x13,0x18,0x28,
361 0x1a,0x18,0x16,0x16,0x18,0x31,0x23,0x25,
362 0x1d,0x28,0x3a,0x33,0x3d,0x3c,0x39,0x33,
363 0x38,0x37,0x40,0x48,0x5c,0x4e,0x40,0x44,
364 0x57,0x45,0x37,0x38,0x50,0x6d,0x51,0x57,
365 0x5f,0x62,0x67,0x68,0x67,0x3e,0x4d,0x71,
366 0x79,0x70,0x64,0x78,0x5c,0x65,0x67,0x63,
368 chroma_quantiser_matrix: {
369 0x11,0x12,0x12,0x18,0x15,0x18,0x2f,0x1a,
370 0x1a,0x2f,0x63,0x42,0x38,0x42,0x63,0x63,
371 0x63,0x63,0x63,0x63,0x63,0x63,0x63,0x63,
372 0x63,0x63,0x63,0x63,0x63,0x63,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,
380 static const HuffmanTable defaultHuffmanTable =
381 ::JPEG::Decode::defaultHuffmanTable;
383 static const PictureParameter defaultPictureParameter = {
384 reconstructed_picture: VA_INVALID_ID,
387 coded_buf: VA_INVALID_ID,
388 pic_flags: {value: 0x00100},
392 component_id: {0, 1, 2, 0},
393 quantiser_table_selector: {0, 1, 1, 0},
397 static const SliceParameter defaultSliceParameter = {
400 /* component_selector, dc_table_selector, ac_table_selector */
401 components: {{1,0,0},{2,1,1},{3,1,1}},
405 : public std::enable_shared_from_this<TestInput>
408 typedef std::shared_ptr<TestInput> Shared;
409 typedef std::shared_ptr<const TestInput> SharedConst;
411 static Shared create(const unsigned, const unsigned, const unsigned);
412 const YUVImage::SharedConst toExpectedOutput() const;
414 friend ::std::ostream& operator<<(::std::ostream&, const TestInput&);
415 friend ::std::ostream& operator<<(::std::ostream&, const Shared&);
416 friend ::std::ostream& operator<<(::std::ostream&, const SharedConst&);
418 YUVImage::Shared image;
419 PictureParameter picture;
421 HuffmanTable huffman;
422 SliceParameter slice;
428 class TestInputCreator
431 typedef std::shared_ptr<TestInputCreator> Shared;
432 typedef std::shared_ptr<const TestInputCreator> SharedConst;
434 TestInput::Shared create(const unsigned) const;
436 friend ::std::ostream& operator<<(
437 ::std::ostream&, const TestInputCreator&);
438 friend ::std::ostream& operator<<(
439 ::std::ostream&, const TestInputCreator::Shared&);
440 friend ::std::ostream& operator<<(
441 ::std::ostream&, const TestInputCreator::SharedConst&);
442 virtual std::array<unsigned, 2> getResolution() const = 0;
445 virtual void repr(::std::ostream& os) const = 0;
448 class RandomSizeCreator
449 : public TestInputCreator
452 std::array<unsigned, 2> getResolution() const;
455 void repr(::std::ostream&) const;
458 class FixedSizeCreator
459 : public TestInputCreator
462 FixedSizeCreator(const std::array<unsigned, 2>&);
463 std::array<unsigned, 2> getResolution() const;
466 void repr(::std::ostream& os) const;
469 const std::array<unsigned, 2> res;
472 typedef std::vector<TestInputCreator::SharedConst> InputCreators;
474 } // namespace Encode