/*
 * Copyright © 2020 Google LLC
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice (including the next
 * paragraph) shall be included in all copies or substantial portions of the
 * Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
 * IN THE SOFTWARE.
 */

#include "freedreno_layout.h"
#include "fd_layout_test.h"
#include "adreno_common.xml.h"
#include "adreno_pm4.xml.h"
#include "a6xx.xml.h"

#include <stdio.h>

static const struct testcase
   testcases[] =
      {
         /* A straightforward first testcase, linear, with an obvious format. */
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 32,
                  .height0 = 32,
                  .slices =
                     {
                        {.offset = 0, .pitch = 256},
                        {.offset = 8192, .pitch = 256},
                        {.offset = 12288, .pitch = 256},
                        {.offset = 14336, .pitch = 256},
                        {.offset = 15360, .pitch = 256},
                        {.offset = 15872, .pitch = 256},
                     },
               },
         },

         /* A tiled/ubwc layout from the blob driver, at a size where the a630
          * blob driver does something interesting for linear.
          */
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .tile_mode = TILE6_3,
                  .ubwc = true,
                  .width0 = 1024,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 4096},
                        {.offset = 65536, .pitch = 2048},
                        {.offset = 98304, .pitch = 1024},
                        {.offset = 114688, .pitch = 512},
                        {.offset = 122880, .pitch = 256},
                        {.offset = 126976, .pitch = 256},
                        {.offset = 131072, .pitch = 256},
                        {.offset = 135168, .pitch = 256},
                        {.offset = 139264, .pitch = 256},
                        {.offset = 143360, .pitch = 256},
                        {.offset = 147456, .pitch = 256},
                     },
                  .ubwc_slices =
                     {
                        {.offset = 0, .pitch = 64},
                        {.offset = 4096, .pitch = 64},
                        {.offset = 8192, .pitch = 64},
                        {.offset = 12288, .pitch = 64},
                        {.offset = 16384, .pitch = 64},
                        {.offset = 20480, .pitch = 64},
                        {.offset = 24576, .pitch = 64},
                        {.offset = 28672, .pitch = 64},
                        {.offset = 32768, .pitch = 64},
                        {.offset = 36864, .pitch = 64},
                        {.offset = 40960, .pitch = 64},
                     },
               },
         },

/* An interesting layout from the blob driver on a630, showing that
 * per-level pitch must be derived from level 0's pitch, not width0.  We
 * don't do this level 0 pitch disalignment (we pick 4096), so disabled
 * this test for now.
 */
#if 0
	{
		.format = PIPE_FORMAT_R8G8B8A8_UNORM,
		.layout = {
			.width0 = 1024, .height0 = 1,
			.slices = {
				{ .offset = 0, .pitch = 5120 },
				{ .offset = 5120, .pitch = 2560 },
				{ .offset = 7680, .pitch = 1280 },
				{ .offset = 8960, .pitch = 768 },
				{ .offset = 9728, .pitch = 512 },
				{ .offset = 10240, .pitch = 256 },
				{ .offset = 10496, .pitch = 256 },
				{ .offset = 10752, .pitch = 256 },
				{ .offset = 11008, .pitch = 256 },
				{ .offset = 11264, .pitch = 256 },
				{ .offset = 11520, .pitch = 256 },
			},
		},
	},
#endif

         /* A layout that we failed on (129 wide has a surprise level 1 pitch
          * increase), and the sizes bracketing it.
          */
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 128,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 512},
                        {.offset = 512, .pitch = 256},
                        {.offset = 768, .pitch = 256},
                        {.offset = 1024, .pitch = 256},
                        {.offset = 1280, .pitch = 256},
                        {.offset = 1536, .pitch = 256},
                        {.offset = 1792, .pitch = 256},
                        {.offset = 2048, .pitch = 256},
                     },
               },
         },
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 129,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 768},
                        {.offset = 768, .pitch = 512},
                        {.offset = 1280, .pitch = 256},
                        {.offset = 1536, .pitch = 256},
                        {.offset = 1792, .pitch = 256},
                        {.offset = 2048, .pitch = 256},
                        {.offset = 2304, .pitch = 256},
                        {.offset = 2560, .pitch = 256},
                     },
               },
         },
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 130,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 768},
                        {.offset = 768, .pitch = 512},
                        {.offset = 1280, .pitch = 256},
                        {.offset = 1536, .pitch = 256},
                        {.offset = 1792, .pitch = 256},
                        {.offset = 2048, .pitch = 256},
                        {.offset = 2304, .pitch = 256},
                        {.offset = 2560, .pitch = 256},
                     },
               },
         },

         /* The 129 failure seems to be across formats, let's test some cpps */
         {
            .format = PIPE_FORMAT_R8_UNORM,
            .layout =
               {
                  .width0 = 129,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 192},
                        {.offset = 192, .pitch = 128},
                        {.offset = 320, .pitch = 64},
                        {.offset = 384, .pitch = 64},
                        {.offset = 448, .pitch = 64},
                        {.offset = 512, .pitch = 64},
                        {.offset = 576, .pitch = 64},
                        {.offset = 640, .pitch = 64},
                     },
               },
         },
         {
            .format = PIPE_FORMAT_R16_UINT,
            .layout =
               {
                  .width0 = 129,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 384},
                        {.offset = 384, .pitch = 256},
                        {.offset = 640, .pitch = 128},
                        {.offset = 768, .pitch = 128},
                        {.offset = 896, .pitch = 128},
                        {.offset = 1024, .pitch = 128},
                        {.offset = 1152, .pitch = 128},
                        {.offset = 1280, .pitch = 128},
                     },
               },
         },
         {
            .format = PIPE_FORMAT_R32G32B32A32_FLOAT,
            .layout =
               {
                  .width0 = 129,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 3072},
                        {.offset = 3072, .pitch = 2048},
                        {.offset = 5120, .pitch = 1024},
                        {.offset = 6144, .pitch = 1024},
                        {.offset = 7168, .pitch = 1024},
                        {.offset = 8192, .pitch = 1024},
                        {.offset = 9216, .pitch = 1024},
                        {.offset = 10240, .pitch = 1024},
                     },
               },
         },

         /* The 129 failure replicated at every +256 pixels wide.  Pick one of
          * them, and this time increase the height as a new variable as well.
          */
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 385,
                  .height0 = 128,
                  .slices =
                     {
                        {.offset = 0, .pitch = 1792},
                        {.offset = 229376, .pitch = 1024},
                        {.offset = 294912, .pitch = 512},
                        {.offset = 311296, .pitch = 256},
                        {.offset = 315392, .pitch = 256},
                        {.offset = 317440, .pitch = 256},
                        {.offset = 318464, .pitch = 256},
                        {.offset = 318976, .pitch = 256},
                        {.offset = 319232, .pitch = 256},
                     },
               },
         },

         /* At 257-259 (and replicated every +256 pixels) we had another
            failure. */
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 257,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 1280},
                        {.offset = 1280, .pitch = 768},
                        {.offset = 2048, .pitch = 512},
                        {.offset = 2560, .pitch = 256},
                        {.offset = 2816, .pitch = 256},
                        {.offset = 3072, .pitch = 256},
                        {.offset = 3328, .pitch = 256},
                        {.offset = 3584, .pitch = 256},
                        {.offset = 3840, .pitch = 256},
                     },
               },
         },
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 258,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 1280},
                        {.offset = 1280, .pitch = 768},
                        {.offset = 2048, .pitch = 512},
                        {.offset = 2560, .pitch = 256},
                        {.offset = 2816, .pitch = 256},
                        {.offset = 3072, .pitch = 256},
                        {.offset = 3328, .pitch = 256},
                        {.offset = 3584, .pitch = 256},
                        {.offset = 3840, .pitch = 256},
                     },
               },
         },
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 259,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 1280},
                        {.offset = 1280, .pitch = 768},
                        {.offset = 2048, .pitch = 512},
                        {.offset = 2560, .pitch = 256},
                        {.offset = 2816, .pitch = 256},
                        {.offset = 3072, .pitch = 256},
                        {.offset = 3328, .pitch = 256},
                        {.offset = 3584, .pitch = 256},
                        {.offset = 3840, .pitch = 256},
                     },
               },
         },
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 260,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 1280},
                        {.offset = 1280, .pitch = 768},
                        {.offset = 2048, .pitch = 512},
                        {.offset = 2560, .pitch = 256},
                        {.offset = 2816, .pitch = 256},
                        {.offset = 3072, .pitch = 256},
                        {.offset = 3328, .pitch = 256},
                        {.offset = 3584, .pitch = 256},
                        {.offset = 3840, .pitch = 256},
                     },
               },
         },

         /* And, again for the 257-9 failure, test a replica with a larger size*/
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 513,
                  .height0 = 32,
                  .slices =
                     {
                        {.offset = 0, .pitch = 2304},
                        {.offset = 73728, .pitch = 1280},
                        {.offset = 94208, .pitch = 768},
                        {.offset = 100352, .pitch = 512},
                        {.offset = 102400, .pitch = 256},
                        {.offset = 102912, .pitch = 256},
                        {.offset = 103168, .pitch = 256},
                        {.offset = 103424, .pitch = 256},
                        {.offset = 103680, .pitch = 256},
                        {.offset = 103936, .pitch = 256},
                     },
               },
         },

         /* Oh, look.  The 513-517 failure extends up to 518 at the next texture
          * level!
          */
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .width0 = 518,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 2304},
                        {.offset = 2304, .pitch = 1280},
                        {.offset = 3584, .pitch = 768},
                        {.offset = 4352, .pitch = 512},
                        {.offset = 4864, .pitch = 256},
                        {.offset = 5120, .pitch = 256},
                        {.offset = 5376, .pitch = 256},
                        {.offset = 5632, .pitch = 256},
                        {.offset = 5888, .pitch = 256},
                        {.offset = 6144, .pitch = 256},
                     },
               },
         },

         /* Tiled mode testing of the unusual 1/2-bytes-per-pixel pitch
            alignment */
         {
            .format = PIPE_FORMAT_R8_UNORM,
            .layout =
               {
                  .tile_mode = TILE6_3,
                  .width0 = 129,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 256},
                        {.offset = 8192, .pitch = 128},
                        {.offset = 12288, .pitch = 128},
                        {.offset = 16384, .pitch = 128},
                        {.offset = 20480, .pitch = 128},
                        {.offset = 20608, .pitch = 128},
                        {.offset = 20736, .pitch = 128},
                        {.offset = 20864, .pitch = 128},
                     },
               },
         },

         /* Single-level RGBA8888 UBWC following UBWC alignment rules laid out
          * in msm_media_info.h to verify that we don't break buffer sharing.
          */
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .tile_mode = TILE6_3,
                  .ubwc = true,
                  .width0 = 16384,
                  .height0 = 129,
                  .slices =
                     {
                        {.offset = 1024 * 48, .pitch = 16384 * 4},
                     },
                  .ubwc_slices =
                     {
                        {.offset = 0, .pitch = 1024},
                     },
               },
         },

         /* UBWC: Pitch comes from POT-aligned level 0. */
         /* Pitch fixed in this commit, but offsets broken.  Will be fixed in
          * following commits.
          */
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .tile_mode = TILE6_3,
                  .ubwc = true,
                  .width0 = 2049,
                  .height0 = 128,
                  .slices =
                     {
                        {.offset = 0, .pitch = 8448},
                        {.offset = 1081344, .pitch = 4352},
                        {.offset = 1359872, .pitch = 2304},
                        {.offset = 1433600, .pitch = 1280},
                        {.offset = 1454080, .pitch = 768},
                        {.offset = 1466368, .pitch = 512},
                        {.offset = 1474560, .pitch = 256},
                        {.offset = 1478656, .pitch = 256},
                        {.offset = 1482752, .pitch = 256},
                        {.offset = 1486848, .pitch = 256},
                        {.offset = 1490944, .pitch = 256},
                        {.offset = 1495040, .pitch = 256},
                     },
                  .ubwc_slices =
                     {
                        {.offset = 0, .pitch = 256},
                        {.offset = 16384, .pitch = 128},
                        {.offset = 24576, .pitch = 64},
                        {.offset = 28672, .pitch = 64},
                        {.offset = 32768, .pitch = 64},
                        {.offset = 36864, .pitch = 64},
                        {.offset = 40960, .pitch = 64},
                        {.offset = 45056, .pitch = 64},
                        {.offset = 49152, .pitch = 64},
                        {.offset = 53248, .pitch = 64},
                        {.offset = 57344, .pitch = 64},
                        {.offset = 61440, .pitch = 64},
                     },
               },
         },
         /* UBWC: Height comes from POT-aligned level 0. */
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .tile_mode = TILE6_3,
                  .ubwc = true,
                  .width0 = 1024,
                  .height0 = 1025,
                  .slices =
                     {
                        {.offset = 0, .pitch = 4096},
                        {.offset = 4259840, .pitch = 2048},
                        {.offset = 5308416, .pitch = 1024},
                        {.offset = 5570560, .pitch = 512},
                        {.offset = 5636096, .pitch = 256},
                        {.offset = 5652480, .pitch = 256},
                        {.offset = 5660672, .pitch = 256},
                        {.offset = 5664768, .pitch = 256},
                        {.offset = 5668864, .pitch = 256},
                        {.offset = 5672960, .pitch = 256},
                        {.offset = 5677056, .pitch = 256},
                     },
                  .ubwc_slices =
                     {
                        {.offset = 0, .pitch = 64},
                        {.offset = 32768, .pitch = 64},
                        {.offset = 49152, .pitch = 64},
                        {.offset = 57344, .pitch = 64},
                        {.offset = 61440, .pitch = 64},
                        {.offset = 65536, .pitch = 64},
                        {.offset = 69632, .pitch = 64},
                        {.offset = 73728, .pitch = 64},
                        {.offset = 77824, .pitch = 64},
                        {.offset = 81920, .pitch = 64},
                        {.offset = 86016, .pitch = 64},
                     },
               },
         },

         /* UBWC: Get at minimum height of a level across cpps */
         {
            .format = PIPE_FORMAT_R16_UINT,
            .layout =
               {
                  .tile_mode = TILE6_3,
                  .ubwc = true,
                  .width0 = 16384,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 32768},
                        {.offset = 524288, .pitch = 16384},
                        {.offset = 786432, .pitch = 8192},
                        {.offset = 917504, .pitch = 4096},
                        {.offset = 983040, .pitch = 2048},
                        {.offset = 1015808, .pitch = 1024},
                        {.offset = 1032192, .pitch = 512},
                        {.offset = 1040384, .pitch = 256},
                        {.offset = 1044480, .pitch = 256},
                        {.offset = 1048576, .pitch = 256},
                        {.offset = 1052672, .pitch = 256},
                        {.offset = 1056768, .pitch = 256},
                        {.offset = 1060864, .pitch = 256},
                        {.offset = 1064960, .pitch = 256},
                        {.offset = 1069056, .pitch = 256},
                     },
                  .ubwc_slices =
                     {
                        {.offset = 0, .pitch = 1024},
                        {.offset = 65536, .pitch = 512},
                        {.offset = 98304, .pitch = 256},
                        {.offset = 114688, .pitch = 128},
                        {.offset = 122880, .pitch = 64},
                        {.offset = 126976, .pitch = 64},
                        {.offset = 131072, .pitch = 64},
                        {.offset = 135168, .pitch = 64},
                        {.offset = 139264, .pitch = 64},
                        {.offset = 143360, .pitch = 64},
                        {.offset = 147456, .pitch = 64},
                        {.offset = 151552, .pitch = 64},
                        {.offset = 155648, .pitch = 64},
                        {.offset = 159744, .pitch = 64},
                        {.offset = 163840, .pitch = 64},
                     },
               },
         },
         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .tile_mode = TILE6_3,
                  .ubwc = true,
                  .width0 = 16384,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 65536},
                        {.offset = 1048576, .pitch = 32768},
                        {.offset = 1572864, .pitch = 16384},
                        {.offset = 1835008, .pitch = 8192},
                        {.offset = 1966080, .pitch = 4096},
                        {.offset = 2031616, .pitch = 2048},
                        {.offset = 2064384, .pitch = 1024},
                        {.offset = 2080768, .pitch = 512},
                        {.offset = 2088960, .pitch = 256},
                        {.offset = 2093056, .pitch = 256},
                        {.offset = 2097152, .pitch = 256},
                        {.offset = 2101248, .pitch = 256},
                        {.offset = 2105344, .pitch = 256},
                        {.offset = 2109440, .pitch = 256},
                        {.offset = 2113536, .pitch = 256},
                     },
                  .ubwc_slices =
                     {
                        {.offset = 0, .pitch = 1024},
                        {.offset = 65536, .pitch = 512},
                        {.offset = 98304, .pitch = 256},
                        {.offset = 114688, .pitch = 128},
                        {.offset = 122880, .pitch = 64},
                        {.offset = 126976, .pitch = 64},
                        {.offset = 131072, .pitch = 64},
                        {.offset = 135168, .pitch = 64},
                        {.offset = 139264, .pitch = 64},
                        {.offset = 143360, .pitch = 64},
                        {.offset = 147456, .pitch = 64},
                        {.offset = 151552, .pitch = 64},
                        {.offset = 155648, .pitch = 64},
                        {.offset = 159744, .pitch = 64},
                        {.offset = 163840, .pitch = 64},
                     },
               },
         },
         {
            .format = PIPE_FORMAT_R32G32B32A32_FLOAT,
            .layout =
               {
                  .tile_mode = TILE6_3,
                  .ubwc = true,
                  .width0 = 16384,
                  .height0 = 1,
                  .slices =
                     {
                        {.offset = 0, .pitch = 262144},
                        {.offset = 4194304, .pitch = 131072},
                        {.offset = 6291456, .pitch = 65536},
                        {.offset = 7340032, .pitch = 32768},
                        {.offset = 7864320, .pitch = 16384},
                        {.offset = 8126464, .pitch = 8192},
                        {.offset = 8257536, .pitch = 4096},
                        {.offset = 8323072, .pitch = 2048},
                        {.offset = 8355840, .pitch = 1024},
                        {.offset = 8372224, .pitch = 1024},
                        {.offset = 8388608, .pitch = 1024},
                        {.offset = 8404992, .pitch = 1024},
                        {.offset = 8421376, .pitch = 1024},
                        {.offset = 8437760, .pitch = 1024},
                        {.offset = 8454144, .pitch = 1024},
                     },
                  .ubwc_slices =
                     {
                        {.offset = 0, .pitch = 4096},
                        {.offset = 262144, .pitch = 2048},
                        {.offset = 393216, .pitch = 1024},
                        {.offset = 458752, .pitch = 512},
                        {.offset = 491520, .pitch = 256},
                        {.offset = 507904, .pitch = 128},
                        {.offset = 516096, .pitch = 64},
                        {.offset = 520192, .pitch = 64},
                        {.offset = 524288, .pitch = 64},
                        {.offset = 528384, .pitch = 64},
                        {.offset = 532480, .pitch = 64},
                        {.offset = 536576, .pitch = 64},
                        {.offset = 540672, .pitch = 64},
                        {.offset = 544768, .pitch = 64},
                        {.offset = 548864, .pitch = 64},
                     },
               },
         },

         {
            .format = PIPE_FORMAT_R8G8B8A8_UNORM,
            .layout =
               {
                  .tile_mode = TILE6_3,
                  .ubwc = true,
                  .nr_samples = 4,
                  .width0 = 412,
                  .height0 = 732,
                  .slices =
                     {
                        {.offset = 0, .pitch = 7168},
                     },
                  .ubwc_slices =
                     {
                        {.offset = 0, .pitch = 128},
                     },
               },
         },
};

int
main(int argc, char **argv)
{
   int ret = 0;

   for (int i = 0; i < ARRAY_SIZE(testcases); i++) {
      if (!fdl_test_layout(&testcases[i], 630))
         ret = 1;
   }

   return ret;
}
