test/i965_jpeg_test_data.h

   1 /*
   2  * Copyright (C) 2016 Intel Corporation. All Rights Reserved.
   3  *
   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:
  11  *
  12  * The above copyright notice and this permission notice (including the
  13  * next paragraph) shall be included in all copies or substantial portions
  14  * of the Software.
  15  *
  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.
  23  */
  24
  25 #ifndef I965_JPEG_TEST_DATA_H
  26 #define I965_JPEG_TEST_DATA_H
  27
  28 #include <array>
  29 #include <iostream>
  30 #include <map>
  31 #include <memory>
  32 #include <va/va.h>
  33 #include <vector>
  34
  35 namespace JPEG {
  36     typedef std::vector<uint8_t> ByteData;
  37
  38     static const VAProfile profile = VAProfileJPEGBaseline;
  39
  40     static inline const ByteData generateSolid(
  41         const std::array<uint8_t, 3>& yuv, const std::array<size_t, 2>& dim)
  42     {
  43         size_t count(dim[0] * dim[1]);
  44         ByteData data(count, yuv[0]);
  45         data.insert(data.end(), count, yuv[1]);
  46         data.insert(data.end(), count, yuv[2]);
  47         return data;
  48     }
  49 } // namespace JPEG
  50
  51 namespace JPEG {
  52 namespace Decode {
  53     typedef VAIQMatrixBufferJPEGBaseline                IQMatrix;
  54     typedef VAHuffmanTableBufferJPEGBaseline            HuffmanTable;
  55     typedef VAPictureParameterBufferJPEGBaseline        PictureParameter;
  56     typedef VASliceParameterBufferJPEGBaseline          SliceParameter;
  57
  58     static const VAEntrypoint entrypoint = VAEntrypointVLD;
  59
  60     static const HuffmanTable defaultHuffmanTable = {
  61         load_huffman_table: { 0x01, 0x01 },
  62         huffman_table: {
  63             { // luminance
  64                 num_dc_codes: {
  65                     0x00,0x01,0x05,0x01,0x01,0x01,0x01,0x01,0x01,0x00,0x00,0x00,
  66                     0x00,0x00,0x00,0x00
  67                 },
  68                 dc_values:    {
  69                     0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b
  70                 },
  71                 num_ac_codes: {
  72                     0x00,0x02,0x01,0x03,0x03,0x02,0x04,0x03,0x05,0x05,0x04,0x04,
  73                     0x00,0x00,0x01,0x7d
  74                 },
  75                 ac_values:    {
  76                     0x01,0x02,0x03,0x00,0x04,0x11,0x05,0x12,0x21,0x31,0x41,0x06,
  77                     0x13,0x51,0x61,0x07,0x22,0x71,0x14,0x32,0x81,0x91,0xa1,0x08,
  78                     0x23,0x42,0xb1,0xc1,0x15,0x52,0xd1,0xf0,0x24,0x33,0x62,0x72,
  79                     0x82,0x09,0x0a,0x16,0x17,0x18,0x19,0x1a,0x25,0x26,0x27,0x28,
  80                     0x29,0x2a,0x34,0x35,0x36,0x37,0x38,0x39,0x3a,0x43,0x44,0x45,
  81                     0x46,0x47,0x48,0x49,0x4a,0x53,0x54,0x55,0x56,0x57,0x58,0x59,
  82                     0x5a,0x63,0x64,0x65,0x66,0x67,0x68,0x69,0x6a,0x73,0x74,0x75,
  83                     0x76,0x77,0x78,0x79,0x7a,0x83,0x84,0x85,0x86,0x87,0x88,0x89,
  84                     0x8a,0x92,0x93,0x94,0x95,0x96,0x97,0x98,0x99,0x9a,0xa2,0xa3,
  85                     0xa4,0xa5,0xa6,0xa7,0xa8,0xa9,0xaa,0xb2,0xb3,0xb4,0xb5,0xb6,
  86                     0xb7,0xb8,0xb9,0xba,0xc2,0xc3,0xc4,0xc5,0xc6,0xc7,0xc8,0xc9,
  87                     0xca,0xd2,0xd3,0xd4,0xd5,0xd6,0xd7,0xd8,0xd9,0xda,0xe1,0xe2,
  88                     0xe3,0xe4,0xe5,0xe6,0xe7,0xe8,0xe9,0xea,0xf1,0xf2,0xf3,0xf4,
  89                     0xf5,0xf6,0xf7,0xf8,0xf9,0xfa
  90                 },
  91                 pad: { 0x00, 0x00 }
  92             },
  93             { // chrominance
  94                 num_dc_codes: {
  95                     0x00,0x03,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x00,
  96                     0x00,0x00,0x00,0x00
  97                 },
  98                 dc_values:    {
  99                     0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b
 100                 },
 101                 num_ac_codes: {
 102                     0x00,0x02,0x01,0x02,0x04,0x04,0x03,0x04,0x07,0x05,0x04,0x04,
 103                     0x00,0x01,0x02,0x77
 104                 },
 105                 ac_values:    {
 106                     0x00,0x01,0x02,0x03,0x11,0x04,0x05,0x21,0x31,0x06,0x12,0x41,
 107                     0x51,0x07,0x61,0x71,0x13,0x22,0x32,0x81,0x08,0x14,0x42,0x91,
 108                     0xa1,0xb1,0xc1,0x09,0x23,0x33,0x52,0xf0,0x15,0x62,0x72,0xd1,
 109                     0x0a,0x16,0x24,0x34,0xe1,0x25,0xf1,0x17,0x18,0x19,0x1a,0x26,
 110                     0x27,0x28,0x29,0x2a,0x35,0x36,0x37,0x38,0x39,0x3a,0x43,0x44,
 111                     0x45,0x46,0x47,0x48,0x49,0x4a,0x53,0x54,0x55,0x56,0x57,0x58,
 112                     0x59,0x5a,0x63,0x64,0x65,0x66,0x67,0x68,0x69,0x6a,0x73,0x74,
 113                     0x75,0x76,0x77,0x78,0x79,0x7a,0x82,0x83,0x84,0x85,0x86,0x87,
 114                     0x88,0x89,0x8a,0x92,0x93,0x94,0x95,0x96,0x97,0x98,0x99,0x9a,
 115                     0xa2,0xa3,0xa4,0xa5,0xa6,0xa7,0xa8,0xa9,0xaa,0xb2,0xb3,0xb4,
 116                     0xb5,0xb6,0xb7,0xb8,0xb9,0xba,0xc2,0xc3,0xc4,0xc5,0xc6,0xc7,
 117                     0xc8,0xc9,0xca,0xd2,0xd3,0xd4,0xd5,0xd6,0xd7,0xd8,0xd9,0xda,
 118                     0xe2,0xe3,0xe4,0xe5,0xe6,0xe7,0xe8,0xe9,0xea,0xf2,0xf3,0xf4,
 119                     0xf5,0xf6,0xf7,0xf8,0xf9,0xfa,
 120                 },
 121                 pad: { 0x00, 0x00 }
 122             }
 123         }
 124     };
 125
 126     static const IQMatrix defaultIQMatrix = { /* Quality 100 */
 127         load_quantiser_table: { 0x1,0x1,0x0,0x0 },
 128         quantiser_table: {
 129             {
 130                 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
 131                 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
 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,
 136             },
 137             {
 138                 0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,0x01,
 139                 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,
 144             },
 145         },
 146     };
 147
 148     static const PictureParameter defaultPictureParameter = {
 149         picture_width:  10,
 150         picture_height: 10,
 151         /* component_id, h_sampling_factor, v_sampling_factor, quantiser_table_selector */
 152         components:     {{1,1,1,0}, {2,1,1,1}, {3,1,1,1}},
 153         num_components: 3,
 154     };
 155
 156     static const SliceParameter defaultSliceParameter = {
 157         slice_data_size:                        0,
 158         slice_data_offset:                      0,
 159         slice_data_flag:                        VA_SLICE_DATA_FLAG_ALL,
 160         slice_horizontal_position:              0,
 161         slice_vertical_position:                0,
 162
 163         /* component_selector, dc_table_selector, ac_table_selector */
 164         components: {{1,0,0},{2,1,1},{3,1,1}},
 165
 166         num_components:                         3,
 167         restart_interval:                       0,
 168         num_mcus:                               4,
 169     };
 170
 171     class PictureData
 172     {
 173     public:
 174         typedef std::shared_ptr<PictureData> Shared;
 175         typedef std::shared_ptr<const PictureData> SharedConst;
 176
 177         template<const unsigned W, const unsigned H>
 178         static SharedConst make(
 179             const unsigned fourcc,
 180             const ByteData& slice,
 181             const SliceParameter& sparam = defaultSliceParameter,
 182             const PictureParameter& pparam = defaultPictureParameter,
 183             const HuffmanTable& huffman = defaultHuffmanTable,
 184             const IQMatrix& iqmatrix = defaultIQMatrix)
 185         {
 186             Shared pd(
 187                 new PictureData {
 188                     slice: slice,
 189                     sparam: sparam,
 190                     pparam: pparam,
 191                     huffman: huffman,
 192                     iqmatrix: iqmatrix,
 193                     format: 0,
 194                     fourcc: fourcc,
 195                 }
 196             );
 197
 198             pd->sparam.slice_data_size = slice.size();
 199             pd->pparam.picture_width = W;
 200             pd->pparam.picture_height = H;
 201
 202             switch(fourcc)
 203             {
 204             case VA_FOURCC_IMC3:
 205                 pd->format = VA_RT_FORMAT_YUV420;
 206                 pd->pparam.components[0].h_sampling_factor = 2;
 207                 pd->pparam.components[0].v_sampling_factor = 2;
 208                 break;
 209             case VA_FOURCC_422H:
 210                 pd->format = VA_RT_FORMAT_YUV422;
 211                 pd->pparam.components[0].h_sampling_factor = 2;
 212                 pd->pparam.components[0].v_sampling_factor = 1;
 213                 break;
 214             case VA_FOURCC_422V:
 215                 pd->format = VA_RT_FORMAT_YUV422;
 216                 pd->pparam.components[0].h_sampling_factor = 1;
 217                 pd->pparam.components[0].v_sampling_factor = 2;
 218                 break;
 219             case VA_FOURCC_411P:
 220                 pd->format = VA_RT_FORMAT_YUV411;
 221                 pd->pparam.components[0].h_sampling_factor = 4;
 222                 pd->pparam.components[0].v_sampling_factor = 1;
 223                 break;
 224             case VA_FOURCC_444P:
 225                 pd->format = VA_RT_FORMAT_YUV444;
 226                 pd->pparam.components[0].h_sampling_factor = 1;
 227                 pd->pparam.components[0].v_sampling_factor = 1;
 228             default:
 229                 break;
 230             }
 231
 232             /* Calculate num_mcus */
 233             int hfactor = pd->pparam.components[0].h_sampling_factor << 3;
 234             int vfactor = pd->pparam.components[0].v_sampling_factor << 3;
 235             int wmcu = (W + hfactor - 1) / hfactor;
 236             int hmcu = (H + vfactor - 1) / vfactor;
 237             pd->sparam.num_mcus = wmcu * hmcu;
 238
 239             return pd;
 240         }
 241
 242         const ByteData          slice;
 243         SliceParameter          sparam;
 244         PictureParameter        pparam;
 245         HuffmanTable            huffman;
 246         IQMatrix                iqmatrix;
 247         unsigned                format;
 248         unsigned                fourcc;
 249     };
 250
 251     class TestPattern
 252     {
 253     public:
 254         typedef std::shared_ptr<TestPattern> Shared;
 255         typedef std::shared_ptr<const TestPattern> SharedConst;
 256
 257         virtual const ByteData& decoded() const = 0;
 258         virtual PictureData::SharedConst encoded(unsigned) const = 0;
 259
 260         virtual void repr(std::ostream&) const = 0;
 261
 262         friend std::ostream& operator <<(std::ostream& os, const TestPattern& t)
 263         {
 264             t.repr(os);
 265             return os;
 266         }
 267     };
 268
 269     template <const unsigned N>
 270     class TestPatternData
 271         : public TestPattern
 272     {
 273     private:
 274         typedef std::map<const unsigned, PictureData::SharedConst> EncodedMap;
 275         typedef std::map<const unsigned, const ByteData> ByteDataMap;
 276
 277     public:
 278         const ByteData& decoded() const { return getDecoded(); }
 279
 280         PictureData::SharedConst encoded(const unsigned fourcc) const
 281         {
 282             const EncodedMap& em = getEncodedMap();
 283             const EncodedMap::const_iterator match(em.find(fourcc));
 284             if (match == em.end())
 285                 return PictureData::SharedConst();
 286             return match->second;
 287         }
 288
 289         void repr(std::ostream& os) const { os << N; }
 290
 291         template <const unsigned W, const unsigned H>
 292         static bool initialize(ByteData d, const ByteDataMap& e)
 293         {
 294             getDecoded().swap(d);
 295
 296             EncodedMap& em = getEncodedMap();
 297             bool result = true;
 298             for (auto const &b : e) {
 299                 auto pd(PictureData::make<W, H>(b.first, b.second));
 300                 auto pair = std::make_pair(b.first, pd);
 301                 result &= em.insert(pair).second;
 302             }
 303             return result;
 304         }
 305
 306     private:
 307         static ByteData& getDecoded()
 308         {
 309             static ByteData d;
 310             return d;
 311         }
 312
 313         static EncodedMap& getEncodedMap()
 314         {
 315             static EncodedMap em;
 316             return em;
 317         }
 318
 319         static const bool m_valid;
 320     };
 321 } // namespace Decode
 322 } // namespace JPEG
 323
 324 #endif