nir: fix deref offset builder
[mesa.git] / src / freedreno / vulkan / tu_image.c
index 0a4bb8712842c46ceaf2ca5de8ba520e07e809ff..0b14a28d44d732241c7d1be60e4c3a902e739c93 100644 (file)
@@ -39,89 +39,109 @@ image_level_linear(struct tu_image *image, int level)
    return w < 16;
 }
 
-/* indexed by cpp: */
-static const struct
+enum a6xx_tile_mode
+tu6_get_image_tile_mode(struct tu_image *image, int level)
 {
+   if (image_level_linear(image, level))
+      return TILE6_LINEAR;
+   else
+      return image->tile_mode;
+}
+
+/* indexed by cpp, including msaa 2x and 4x: */
+static const struct {
    unsigned pitchalign;
    unsigned heightalign;
 } tile_alignment[] = {
-   [1] = { 128, 32 }, [2] = { 128, 16 }, [3] = { 128, 16 }, [4] = { 64, 16 },
-   [8] = { 64, 16 },  [12] = { 64, 16 }, [16] = { 64, 16 },
+   [1]  = { 128, 32 },
+   [2]  = { 128, 16 },
+   [3]  = {  64, 32 },
+   [4]  = {  64, 16 },
+   [6]  = {  64, 16 },
+   [8]  = {  64, 16 },
+   [12] = {  64, 16 },
+   [16] = {  64, 16 },
+   [24] = {  64, 16 },
+   [32] = {  64, 16 },
+   [48] = {  64, 16 },
+   [64] = {  64, 16 },
+
+   /* special case for r8g8: */
+   [0]  = { 64, 32 },
 };
 
 static void
 setup_slices(struct tu_image *image, const VkImageCreateInfo *pCreateInfo)
 {
-   enum vk_format_layout layout =
-      vk_format_description(pCreateInfo->format)->layout;
+   VkFormat format = pCreateInfo->format;
+   enum util_format_layout layout = vk_format_description(format)->layout;
    uint32_t layer_size = 0;
-   uint32_t width = pCreateInfo->extent.width;
-   uint32_t height = pCreateInfo->extent.height;
-   uint32_t depth = pCreateInfo->extent.depth;
-   bool layer_first = pCreateInfo->imageType != VK_IMAGE_TYPE_3D;
-   uint32_t alignment = pCreateInfo->imageType == VK_IMAGE_TYPE_3D ? 4096 : 1;
-   uint32_t cpp = vk_format_get_blocksize(pCreateInfo->format);
+   int ta = image->cpp;
 
-   uint32_t heightalign = tile_alignment[cpp].heightalign;
+   /* The r8g8 format seems to not play by the normal tiling rules: */
+   if (image->cpp == 2 && vk_format_get_nr_components(format) == 2)
+      ta = 0;
 
    for (unsigned level = 0; level < pCreateInfo->mipLevels; level++) {
       struct tu_image_level *slice = &image->levels[level];
-      bool linear_level = image_level_linear(image, level);
+      uint32_t width = u_minify(pCreateInfo->extent.width, level);
+      uint32_t height = u_minify(pCreateInfo->extent.height, level);
+      uint32_t depth = u_minify(pCreateInfo->extent.depth, level);
       uint32_t aligned_height = height;
       uint32_t blocks;
       uint32_t pitchalign;
 
-      if (image->tile_mode && !linear_level) {
-         pitchalign = tile_alignment[cpp].pitchalign;
-         aligned_height = align(aligned_height, heightalign);
+      if (image->tile_mode && !image_level_linear(image, level)) {
+         /* tiled levels of 3D textures are rounded up to PoT dimensions: */
+         if (pCreateInfo->imageType == VK_IMAGE_TYPE_3D) {
+            width = util_next_power_of_two(width);
+            height = aligned_height = util_next_power_of_two(height);
+         }
+         pitchalign = tile_alignment[ta].pitchalign;
+         aligned_height = align(aligned_height, tile_alignment[ta].heightalign);
       } else {
          pitchalign = 64;
-
-         /* The blits used for mem<->gmem work at a granularity of
-          * 32x32, which can cause faults due to over-fetch on the
-          * last level.  The simple solution is to over-allocate a
-          * bit the last level to ensure any over-fetch is harmless.
-          * The pitch is already sufficiently aligned, but height
-          * may not be:
-          */
-         if ((level + 1 == pCreateInfo->mipLevels))
-            aligned_height = align(aligned_height, 32);
       }
 
-      if (layout == VK_FORMAT_LAYOUT_ASTC)
-         slice->pitch = util_align_npot(
-            width,
-            pitchalign * vk_format_get_blockwidth(pCreateInfo->format));
+      /* The blits used for mem<->gmem work at a granularity of
+       * 32x32, which can cause faults due to over-fetch on the
+       * last level.  The simple solution is to over-allocate a
+       * bit the last level to ensure any over-fetch is harmless.
+       * The pitch is already sufficiently aligned, but height
+       * may not be:
+       */
+      if (level + 1 == pCreateInfo->mipLevels)
+         aligned_height = align(aligned_height, 32);
+
+      if (layout == UTIL_FORMAT_LAYOUT_ASTC)
+         slice->pitch =
+            util_align_npot(width, pitchalign * vk_format_get_blockwidth(format));
       else
          slice->pitch = align(width, pitchalign);
 
       slice->offset = layer_size;
-      blocks = vk_format_get_block_count(pCreateInfo->format, slice->pitch,
-                                         aligned_height);
+      blocks = vk_format_get_block_count(format, slice->pitch, aligned_height);
 
       /* 1d array and 2d array textures must all have the same layer size
-       * for each miplevel on a3xx. 3d textures can have different layer
+       * for each miplevel on a6xx. 3d textures can have different layer
        * sizes for high levels, but the hw auto-sizer is buggy (or at least
        * different than what this code does), so as soon as the layer size
        * range gets into range, we stop reducing it.
        */
-      if (pCreateInfo->imageType == VK_IMAGE_TYPE_3D &&
-          (level == 1 ||
-           (level > 1 && image->levels[level - 1].size > 0xf000)))
-         slice->size = align(blocks * cpp, alignment);
-      else if (level == 0 || layer_first || alignment == 1)
-         slice->size = align(blocks * cpp, alignment);
-      else
-         slice->size = image->levels[level - 1].size;
+      if (pCreateInfo->imageType == VK_IMAGE_TYPE_3D) {
+         if (level < 1 || image->levels[level - 1].size > 0xf000) {
+            slice->size = align(blocks * image->cpp, 4096);
+         } else {
+            slice->size = image->levels[level - 1].size;
+         }
+      } else {
+         slice->size = blocks * image->cpp;
+      }
 
       layer_size += slice->size * depth;
-
-      width = u_minify(width, 1);
-      height = u_minify(height, 1);
-      depth = u_minify(depth, 1);
    }
 
-   image->layer_size = layer_size;
+   image->layer_size = align(layer_size, 4096);
 }
 
 VkResult
@@ -156,12 +176,14 @@ tu_image_create(VkDevice _device,
    image->extent = pCreateInfo->extent;
    image->level_count = pCreateInfo->mipLevels;
    image->layer_count = pCreateInfo->arrayLayers;
+   image->samples = pCreateInfo->samples;
+   image->cpp = vk_format_get_blocksize(image->vk_format) * image->samples;
 
    image->exclusive = pCreateInfo->sharingMode == VK_SHARING_MODE_EXCLUSIVE;
    if (pCreateInfo->sharingMode == VK_SHARING_MODE_CONCURRENT) {
       for (uint32_t i = 0; i < pCreateInfo->queueFamilyIndexCount; ++i)
          if (pCreateInfo->pQueueFamilyIndices[i] ==
-             VK_QUEUE_FAMILY_EXTERNAL_KHR)
+             VK_QUEUE_FAMILY_EXTERNAL)
             image->queue_family_mask |= (1u << TU_MAX_QUEUE_FAMILIES) - 1u;
          else
             image->queue_family_mask |=
@@ -170,9 +192,21 @@ tu_image_create(VkDevice _device,
 
    image->shareable =
       vk_find_struct_const(pCreateInfo->pNext,
-                           EXTERNAL_MEMORY_IMAGE_CREATE_INFO_KHR) != NULL;
+                           EXTERNAL_MEMORY_IMAGE_CREATE_INFO) != NULL;
+
+   image->tile_mode = TILE6_3;
+
+   if (pCreateInfo->tiling == VK_IMAGE_TILING_LINEAR ||
+       /* compressed textures can't use tiling? */
+       vk_format_is_compressed(image->vk_format) ||
+       /* scanout needs to be linear (what about tiling modifiers?) */
+       create_info->scanout ||
+       /* image_to_image copy doesn't deal with tiling+swap */
+       tu6_get_native_format(image->vk_format)->swap ||
+       /* r8g8 formats are tiled different and could break image_to_image copy */
+       (image->cpp == 2 && vk_format_get_nr_components(image->vk_format) == 2))
+      image->tile_mode = TILE6_LINEAR;
 
-   image->tile_mode = pCreateInfo->tiling == VK_IMAGE_TILING_OPTIMAL ? 3 : 0;
    setup_slices(image, pCreateInfo);
 
    image->size = image->layer_size * pCreateInfo->arrayLayers;
@@ -181,6 +215,71 @@ tu_image_create(VkDevice _device,
    return VK_SUCCESS;
 }
 
+static enum a6xx_tex_fetchsize
+tu6_fetchsize(VkFormat format)
+{
+   if (vk_format_description(format)->layout == UTIL_FORMAT_LAYOUT_ASTC)
+      return TFETCH6_16_BYTE;
+
+   switch (vk_format_get_blocksize(format) / vk_format_get_blockwidth(format)) {
+   case 1: return TFETCH6_1_BYTE;
+   case 2: return TFETCH6_2_BYTE;
+   case 4: return TFETCH6_4_BYTE;
+   case 8: return TFETCH6_8_BYTE;
+   case 16: return TFETCH6_16_BYTE;
+   default:
+      unreachable("bad block size");
+   }
+}
+
+static uint32_t
+tu6_texswiz(const VkComponentMapping *comps, const unsigned char *fmt_swiz)
+{
+   unsigned char swiz[4] = {comps->r, comps->g, comps->b, comps->a};
+   unsigned char vk_swizzle[] = {
+      [VK_COMPONENT_SWIZZLE_ZERO] = A6XX_TEX_ZERO,
+      [VK_COMPONENT_SWIZZLE_ONE]  = A6XX_TEX_ONE,
+      [VK_COMPONENT_SWIZZLE_R] = A6XX_TEX_X,
+      [VK_COMPONENT_SWIZZLE_G] = A6XX_TEX_Y,
+      [VK_COMPONENT_SWIZZLE_B] = A6XX_TEX_Z,
+      [VK_COMPONENT_SWIZZLE_A] = A6XX_TEX_W,
+   };
+   for (unsigned i = 0; i < 4; i++) {
+      swiz[i] = (swiz[i] == VK_COMPONENT_SWIZZLE_IDENTITY) ? i : vk_swizzle[swiz[i]];
+      /* if format has 0/1 in channel, use that (needed for bc1_rgb) */
+      if (swiz[i] < 4) {
+         switch (fmt_swiz[swiz[i]]) {
+         case PIPE_SWIZZLE_0: swiz[i] = A6XX_TEX_ZERO; break;
+         case PIPE_SWIZZLE_1: swiz[i] = A6XX_TEX_ONE;  break;
+         }
+      }
+   }
+
+   return A6XX_TEX_CONST_0_SWIZ_X(swiz[0]) |
+          A6XX_TEX_CONST_0_SWIZ_Y(swiz[1]) |
+          A6XX_TEX_CONST_0_SWIZ_Z(swiz[2]) |
+          A6XX_TEX_CONST_0_SWIZ_W(swiz[3]);
+}
+
+static enum a6xx_tex_type
+tu6_tex_type(VkImageViewType type)
+{
+   switch (type) {
+   default:
+   case VK_IMAGE_VIEW_TYPE_1D:
+   case VK_IMAGE_VIEW_TYPE_1D_ARRAY:
+      return A6XX_TEX_1D;
+   case VK_IMAGE_VIEW_TYPE_2D:
+   case VK_IMAGE_VIEW_TYPE_2D_ARRAY:
+      return A6XX_TEX_2D;
+   case VK_IMAGE_VIEW_TYPE_3D:
+      return A6XX_TEX_3D;
+   case VK_IMAGE_VIEW_TYPE_CUBE:
+   case VK_IMAGE_VIEW_TYPE_CUBE_ARRAY:
+      return A6XX_TEX_CUBE;
+   }
+}
+
 void
 tu_image_view_init(struct tu_image_view *iview,
                    struct tu_device *device,
@@ -221,6 +320,46 @@ tu_image_view_init(struct tu_image_view *iview,
    iview->layer_count = tu_get_layerCount(image, range);
    iview->base_mip = range->baseMipLevel;
    iview->level_count = tu_get_levelCount(image, range);
+
+   memset(iview->descriptor, 0, sizeof(iview->descriptor));
+
+   const struct tu_native_format *fmt = tu6_get_native_format(iview->vk_format);
+   struct tu_image_level *slice0 = &image->levels[iview->base_mip];
+   uint64_t base_addr = image->bo->iova + iview->base_layer * image->layer_size + slice0->offset;
+   uint32_t pitch = (slice0->pitch / vk_format_get_blockwidth(iview->vk_format)) *
+                        vk_format_get_blocksize(iview->vk_format);
+   enum a6xx_tile_mode tile_mode =
+      image_level_linear(image, iview->base_mip) ? TILE6_LINEAR : image->tile_mode;
+
+   iview->descriptor[0] =
+      A6XX_TEX_CONST_0_TILE_MODE(tile_mode) |
+      COND(vk_format_is_srgb(iview->vk_format), A6XX_TEX_CONST_0_SRGB) |
+      A6XX_TEX_CONST_0_FMT(fmt->tex) |
+      A6XX_TEX_CONST_0_SAMPLES(tu_msaa_samples(image->samples)) |
+      A6XX_TEX_CONST_0_SWAP(image->tile_mode ? WZYX : fmt->swap) |
+      tu6_texswiz(&pCreateInfo->components, vk_format_description(iview->vk_format)->swizzle) |
+      A6XX_TEX_CONST_0_MIPLVLS(iview->level_count - 1);
+   iview->descriptor[1] =
+      A6XX_TEX_CONST_1_WIDTH(u_minify(image->extent.width, iview->base_mip)) |
+      A6XX_TEX_CONST_1_HEIGHT(u_minify(image->extent.height, iview->base_mip));
+   iview->descriptor[2] =
+      A6XX_TEX_CONST_2_FETCHSIZE(tu6_fetchsize(iview->vk_format)) |
+      A6XX_TEX_CONST_2_PITCH(pitch) |
+      A6XX_TEX_CONST_2_TYPE(tu6_tex_type(pCreateInfo->viewType));
+   iview->descriptor[3] = 0;
+   iview->descriptor[4] = base_addr;
+   iview->descriptor[5] = base_addr >> 32;
+
+   if (pCreateInfo->viewType != VK_IMAGE_VIEW_TYPE_3D) {
+      iview->descriptor[3] |= A6XX_TEX_CONST_3_ARRAY_PITCH(image->layer_size);
+      iview->descriptor[5] |= A6XX_TEX_CONST_5_DEPTH(iview->layer_count);
+   } else {
+      iview->descriptor[3] |=
+         A6XX_TEX_CONST_3_MIN_LAYERSZ(image->levels[image->level_count - 1].size) |
+         A6XX_TEX_CONST_3_ARRAY_PITCH(slice0->size);
+      iview->descriptor[5] |=
+         A6XX_TEX_CONST_5_DEPTH(u_minify(image->extent.depth, iview->base_mip));
+   }
 }
 
 unsigned
@@ -230,7 +369,7 @@ tu_image_queue_family_mask(const struct tu_image *image,
 {
    if (!image->exclusive)
       return image->queue_family_mask;
-   if (family == VK_QUEUE_FAMILY_EXTERNAL_KHR)
+   if (family == VK_QUEUE_FAMILY_EXTERNAL)
       return (1u << TU_MAX_QUEUE_FAMILIES) - 1u;
    if (family == VK_QUEUE_FAMILY_IGNORED)
       return 1u << queue_family;
@@ -252,10 +391,14 @@ tu_CreateImage(VkDevice device,
                                    pAllocator, pImage);
 #endif
 
+   const struct wsi_image_create_info *wsi_info =
+      vk_find_struct_const(pCreateInfo->pNext, WSI_IMAGE_CREATE_INFO_MESA);
+   bool scanout = wsi_info && wsi_info->scanout;
+
    return tu_image_create(device,
                           &(struct tu_image_create_info) {
                              .vk_info = pCreateInfo,
-                             .scanout = false,
+                             .scanout = scanout,
                           },
                           pAllocator, pImage);
 }
@@ -283,12 +426,18 @@ tu_GetImageSubresourceLayout(VkDevice _device,
                              const VkImageSubresource *pSubresource,
                              VkSubresourceLayout *pLayout)
 {
-   tu_stub();
+   TU_FROM_HANDLE(tu_image, image, _image);
+
+   const uint32_t layer_offset = image->layer_size * pSubresource->arrayLayer;
+   const struct tu_image_level *level =
+      image->levels + pSubresource->mipLevel;
 
-   /* Even though this is a stub, let's avoid heisenbugs by providing
-    * deterministic behavior.
-    */
-   memset(pLayout, 0, sizeof(*pLayout));
+   pLayout->offset = layer_offset + level->offset;
+   pLayout->size = level->size;
+   pLayout->rowPitch =
+      level->pitch * vk_format_get_blocksize(image->vk_format);
+   pLayout->arrayPitch = image->layer_size;
+   pLayout->depthPitch = level->size;
 }
 
 VkResult