/*
 * Copyright © 2017 Intel Corporation
 * SPDX-License-Identifier: MIT
 */

#include <stdio.h>
#include "pipe/p_defines.h"
#include "pipe/p_state.h"
#include "pipe/p_context.h"
#include "pipe/p_screen.h"
#include "util/format/u_format.h"
#include "util/u_inlines.h"
#include "util/u_surface.h"
#include "util/ralloc.h"
#include "intel/blorp/blorp.h"
#include "iris_context.h"
#include "iris_resource.h"
#include "iris_screen.h"

/**
 * Helper function for handling mirror image blits.
 *
 * If coord0 > coord1, swap them and return "true" (mirrored).
 */
static bool
apply_mirror(float *coord0, float *coord1)
{
   if (*coord0 > *coord1) {
      float tmp = *coord0;
      *coord0 = *coord1;
      *coord1 = tmp;
      return true;
   }
   return false;
}

/**
 * Compute the number of pixels to clip for each side of a rect
 *
 * \param x0 The rect's left coordinate
 * \param y0 The rect's bottom coordinate
 * \param x1 The rect's right coordinate
 * \param y1 The rect's top coordinate
 * \param min_x The clipping region's left coordinate
 * \param min_y The clipping region's bottom coordinate
 * \param max_x The clipping region's right coordinate
 * \param max_y The clipping region's top coordinate
 * \param clipped_x0 The number of pixels to clip from the left side
 * \param clipped_y0 The number of pixels to clip from the bottom side
 * \param clipped_x1 The number of pixels to clip from the right side
 * \param clipped_y1 The number of pixels to clip from the top side
 *
 * \return false if we clip everything away, true otherwise
 */
static inline bool
compute_pixels_clipped(float x0, float y0, float x1, float y1,
                       float min_x, float min_y, float max_x, float max_y,
                       float *clipped_x0, float *clipped_y0,
                       float *clipped_x1, float *clipped_y1)
{
   /* If we are going to clip everything away, stop. */
   if (!(min_x <= max_x &&
         min_y <= max_y &&
         x0 <= max_x &&
         y0 <= max_y &&
         min_x <= x1 &&
         min_y <= y1 &&
         x0 <= x1 &&
         y0 <= y1)) {
      return false;
   }

   if (x0 < min_x)
      *clipped_x0 = min_x - x0;
   else
      *clipped_x0 = 0;
   if (max_x < x1)
      *clipped_x1 = x1 - max_x;
   else
      *clipped_x1 = 0;

   if (y0 < min_y)
      *clipped_y0 = min_y - y0;
   else
      *clipped_y0 = 0;
   if (max_y < y1)
      *clipped_y1 = y1 - max_y;
   else
      *clipped_y1 = 0;

   return true;
}

/**
 * Clips a coordinate (left, right, top or bottom) for the src or dst rect
 * (whichever requires the largest clip) and adjusts the coordinate
 * for the other rect accordingly.
 *
 * \param mirror true if mirroring is required
 * \param src the source rect coordinate (for example src_x0)
 * \param dst0 the dst rect coordinate (for example dst_x0)
 * \param dst1 the opposite dst rect coordinate (for example dst_x1)
 * \param clipped_dst0 number of pixels to clip from the dst coordinate
 * \param clipped_dst1 number of pixels to clip from the opposite dst coordinate
 * \param scale the src vs dst scale involved for that coordinate
 * \param is_left_or_bottom true if we are clipping the left or bottom sides
 *        of the rect.
 */
static void
clip_coordinates(bool mirror,
                 float *src, float *dst0, float *dst1,
                 float clipped_dst0,
                 float clipped_dst1,
                 float scale,
                 bool is_left_or_bottom)
{
   /* When clipping we need to add or subtract pixels from the original
    * coordinates depending on whether we are acting on the left/bottom
    * or right/top sides of the rect respectively. We assume we have to
    * add them in the code below, and multiply by -1 when we should
    * subtract.
    */
   int mult = is_left_or_bottom ? 1 : -1;

   if (!mirror) {
      *dst0 += clipped_dst0 * mult;
      *src += clipped_dst0 * scale * mult;
   } else {
      *dst1 -= clipped_dst1 * mult;
      *src += clipped_dst1 * scale * mult;
   }
}

/**
 * Apply a scissor rectangle to blit coordinates.
 *
 * Returns true if the blit was entirely scissored away.
 */
static bool
apply_blit_scissor(const struct pipe_scissor_state *scissor,
                   float *src_x0, float *src_y0,
                   float *src_x1, float *src_y1,
                   float *dst_x0, float *dst_y0,
                   float *dst_x1, float *dst_y1,
                   bool mirror_x, bool mirror_y)
{
   float clip_dst_x0, clip_dst_x1, clip_dst_y0, clip_dst_y1;

   /* Compute number of pixels to scissor away. */
   if (!compute_pixels_clipped(*dst_x0, *dst_y0, *dst_x1, *dst_y1,
                               scissor->minx, scissor->miny,
                               scissor->maxx, scissor->maxy,
                               &clip_dst_x0, &clip_dst_y0,
                               &clip_dst_x1, &clip_dst_y1))
      return true;

   // XXX: comments assume source clipping, which we don't do

   /* When clipping any of the two rects we need to adjust the coordinates
    * in the other rect considering the scaling factor involved.  To obtain
    * the best precision we want to make sure that we only clip once per
    * side to avoid accumulating errors due to the scaling adjustment.
    *
    * For example, if src_x0 and dst_x0 need both to be clipped we want to
    * avoid the situation where we clip src_x0 first, then adjust dst_x0
    * accordingly but then we realize that the resulting dst_x0 still needs
    * to be clipped, so we clip dst_x0 and adjust src_x0 again.  Because we are
    * applying scaling factors to adjust the coordinates in each clipping
    * pass we lose some precision and that can affect the results of the
    * blorp blit operation slightly.  What we want to do here is detect the
    * rect that we should clip first for each side so that when we adjust
    * the other rect we ensure the resulting coordinate does not need to be
    * clipped again.
    *
    * The code below implements this by comparing the number of pixels that
    * we need to clip for each side of both rects considering the scales
    * involved.  For example, clip_src_x0 represents the number of pixels
    * to be clipped for the src rect's left side, so if clip_src_x0 = 5,
    * clip_dst_x0 = 4 and scale_x = 2 it means that we are clipping more
    * from the dst rect so we should clip dst_x0 only and adjust src_x0.
    * This is because clipping 4 pixels in the dst is equivalent to
    * clipping 4 * 2 = 8 > 5 in the src.
    */

   if (*src_x0 == *src_x1 || *src_y0 == *src_y1
       || *dst_x0 == *dst_x1 || *dst_y0 == *dst_y1)
      return true;

   float scale_x = (float) (*src_x1 - *src_x0) / (*dst_x1 - *dst_x0);
   float scale_y = (float) (*src_y1 - *src_y0) / (*dst_y1 - *dst_y0);

   /* Clip left side */
   clip_coordinates(mirror_x, src_x0, dst_x0, dst_x1,
                    clip_dst_x0, clip_dst_x1, scale_x, true);

   /* Clip right side */
   clip_coordinates(mirror_x, src_x1, dst_x1, dst_x0,
                    clip_dst_x1, clip_dst_x0, scale_x, false);

   /* Clip bottom side */
   clip_coordinates(mirror_y, src_y0, dst_y0, dst_y1,
                    clip_dst_y0, clip_dst_y1, scale_y, true);

   /* Clip top side */
   clip_coordinates(mirror_y, src_y1, dst_y1, dst_y0,
                    clip_dst_y1, clip_dst_y0, scale_y, false);

   /* Check for invalid bounds
    * Can't blit for 0-dimensions
    */
   return *src_x0 == *src_x1 || *src_y0 == *src_y1
      || *dst_x0 == *dst_x1 || *dst_y0 == *dst_y1;
}

void
iris_blorp_surf_for_resource(struct iris_batch *batch,
                             struct blorp_surf *surf,
                             struct pipe_resource *p_res,
                             enum isl_aux_usage aux_usage,
                             unsigned level,
                             bool is_dest)
{
   const struct isl_device *isl_dev = &batch->screen->isl_dev;
   struct iris_resource *res = (void *) p_res;
   const struct intel_device_info *devinfo = isl_dev->info;

   *surf = (struct blorp_surf) {
      .surf = &res->surf,
      .addr = (struct blorp_address) {
         .buffer = res->bo,
         .offset = res->offset,
         .reloc_flags = is_dest ? IRIS_BLORP_RELOC_FLAGS_EXEC_OBJECT_WRITE : 0,
         .mocs = iris_mocs(res->bo, isl_dev,
                           iris_blorp_batch_usage(batch, is_dest)),
         .local_hint = iris_bo_likely_local(res->bo),
      },
      .aux_usage = aux_usage,
   };

   if (aux_usage != ISL_AUX_USAGE_NONE) {
      surf->aux_surf = &res->aux.surf;
      surf->aux_addr = (struct blorp_address) {
         .buffer = res->aux.bo,
         .offset = res->aux.offset,
         .reloc_flags = is_dest ? IRIS_BLORP_RELOC_FLAGS_EXEC_OBJECT_WRITE : 0,
         .mocs = iris_mocs(res->bo, isl_dev, 0),
         .local_hint = devinfo->has_flat_ccs ||
                       iris_bo_likely_local(res->aux.bo),
      };
      surf->clear_color = res->aux.clear_color;
      surf->clear_color_addr = (struct blorp_address) {
         .buffer = res->aux.clear_color_bo,
         .offset = res->aux.clear_color_offset,
         .reloc_flags = 0,
         .mocs = iris_mocs(res->aux.clear_color_bo, isl_dev, 0),
         .local_hint = devinfo->has_flat_ccs ||
                       iris_bo_likely_local(res->aux.clear_color_bo),
      };
   }
}

static bool
is_astc(enum isl_format format)
{
   return format != ISL_FORMAT_UNSUPPORTED &&
          isl_format_get_layout(format)->txc == ISL_TXC_ASTC;
}

static void
tex_cache_flush_hack(struct iris_batch *batch,
                     enum isl_format view_format,
                     enum isl_format surf_format)
{
   const struct intel_device_info *devinfo = batch->screen->devinfo;

   /* The WaSamplerCacheFlushBetweenRedescribedSurfaceReads workaround says:
    *
    *    "Currently Sampler assumes that a surface would not have two
    *     different format associate with it.  It will not properly cache
    *     the different views in the MT cache, causing a data corruption."
    *
    * We may need to handle this for texture views in general someday, but
    * for now we handle it here, as it hurts copies and blits particularly
    * badly because they ofter reinterpret formats.
    *
    * If the BO hasn't been referenced yet this batch, we assume that the
    * texture cache doesn't contain any relevant data nor need flushing.
    *
    * Icelake (Gfx11+) claims to fix this issue, but seems to still have
    * issues with ASTC formats.
    */
   bool need_flush = devinfo->ver >= 11 ?
                     is_astc(surf_format) != is_astc(view_format) :
                     view_format != surf_format;
   if (!need_flush)
      return;

   const char *reason =
      "workaround: WaSamplerCacheFlushBetweenRedescribedSurfaceReads";

   iris_emit_pipe_control_flush(batch, reason, PIPE_CONTROL_CS_STALL);
   iris_emit_pipe_control_flush(batch, reason,
                                PIPE_CONTROL_TEXTURE_CACHE_INVALIDATE);
}

static struct iris_resource *
iris_resource_for_aspect(struct pipe_resource *p_res, unsigned pipe_mask)
{
   if (pipe_mask == PIPE_MASK_S) {
      struct iris_resource *junk, *s_res;
      iris_get_depth_stencil_resources(p_res, &junk, &s_res);
      return s_res;
   } else {
      return (struct iris_resource *)p_res;
   }
}

static enum pipe_format
pipe_format_for_aspect(enum pipe_format format, unsigned pipe_mask)
{
   if (pipe_mask == PIPE_MASK_S) {
      return util_format_stencil_only(format);
   } else if (pipe_mask == PIPE_MASK_Z) {
      return util_format_get_depth_only(format);
   } else {
      return format;
   }
}

static bool
clear_color_is_fully_zero(const struct iris_resource *res)
{
   return !res->aux.clear_color_unknown &&
          res->aux.clear_color.u32[0] == 0 &&
          res->aux.clear_color.u32[1] == 0 &&
          res->aux.clear_color.u32[2] == 0 &&
          res->aux.clear_color.u32[3] == 0;
}

static bool
try_hw_blitter_copy(struct iris_context *ice,
                    const struct intel_device_info *devinfo,
                    const struct pipe_blit_info *info)
{
   struct iris_resource *src_res =
      iris_resource_for_aspect(info->src.resource, PIPE_MASK_RGBA);
   struct iris_resource *dst_res =
      iris_resource_for_aspect(info->dst.resource, PIPE_MASK_RGBA);
   enum pipe_format src_pfmt =
      pipe_format_for_aspect(info->src.format, PIPE_MASK_RGBA);
   enum pipe_format dst_pfmt =
      pipe_format_for_aspect(info->dst.format, PIPE_MASK_RGBA);
   struct iris_format_info src_fmt =
      iris_format_for_usage(devinfo, src_pfmt, ISL_SURF_USAGE_BLITTER_SRC_BIT);
   struct iris_format_info dst_fmt =
      iris_format_for_usage(devinfo, dst_pfmt, ISL_SURF_USAGE_BLITTER_DST_BIT);
   enum isl_aux_usage src_aux_usage =
      iris_resource_render_aux_usage(ice, src_res, src_fmt.fmt,
                                     info->src.level, false);
   enum isl_aux_usage dst_aux_usage =
      iris_resource_render_aux_usage(ice, dst_res, dst_fmt.fmt,
                                     info->dst.level, false);

   if (!blorp_copy_supports_blitter(&ice->blorp,
                                    &src_res->surf, &dst_res->surf,
                                    src_aux_usage, dst_aux_usage)) {
      return false;
   }

   assert(!info->render_condition_enable);
   assert(util_can_blit_via_copy_region(info, false, false));
   iris_copy_region(&ice->blorp, &ice->batches[IRIS_BATCH_BLITTER],
                    info->dst.resource, info->dst.level,
                    info->dst.box.x, info->dst.box.y, info->dst.box.z,
                    info->src.resource, info->src.level,
                    &info->src.box);
   return true;
}

/**
 * The pipe->blit() driver hook.
 *
 * This performs a blit between two surfaces, which copies data but may
 * also perform format conversion, scaling, flipping, and so on.
 */
static void
iris_blit(struct pipe_context *ctx, const struct pipe_blit_info *info)
{
   struct iris_context *ice = (void *) ctx;
   struct iris_screen *screen = (struct iris_screen *)ctx->screen;
   const struct intel_device_info *devinfo = screen->devinfo;
   struct iris_batch *batch = &ice->batches[IRIS_BATCH_RENDER];
   enum blorp_batch_flags blorp_flags = iris_blorp_flags_for_batch(batch);

   /* We don't support color masking. */
   assert((info->mask & PIPE_MASK_RGBA) == PIPE_MASK_RGBA ||
          (info->mask & PIPE_MASK_RGBA) == 0);

   if (info->render_condition_enable) {
      if (ice->state.predicate == IRIS_PREDICATE_STATE_DONT_RENDER)
         return;

      if (ice->state.predicate == IRIS_PREDICATE_STATE_USE_BIT)
         blorp_flags |= BLORP_BATCH_PREDICATE_ENABLE;
   }

   float src_x0 = info->src.box.x;
   float src_x1 = info->src.box.x + info->src.box.width;
   float src_y0 = info->src.box.y;
   float src_y1 = info->src.box.y + info->src.box.height;
   float dst_x0 = info->dst.box.x;
   float dst_x1 = info->dst.box.x + info->dst.box.width;
   float dst_y0 = info->dst.box.y;
   float dst_y1 = info->dst.box.y + info->dst.box.height;
   bool mirror_x = apply_mirror(&src_x0, &src_x1);
   bool mirror_y = apply_mirror(&src_y0, &src_y1);
   enum blorp_filter filter;

   if (info->scissor_enable) {
      bool noop = apply_blit_scissor(&info->scissor,
                                     &src_x0, &src_y0, &src_x1, &src_y1,
                                     &dst_x0, &dst_y0, &dst_x1, &dst_y1,
                                     mirror_x, mirror_y);
      if (noop)
         return;
   }

   /* Do DRI PRIME blits on the hardware blitter on Gfx12+ */
   if (devinfo->ver >= 12 &&
       (info->dst.resource->bind & PIPE_BIND_PRIME_BLIT_DST)) {
      if (try_hw_blitter_copy(ice, devinfo, info))
         return;
   }

   if (abs(info->dst.box.width) == abs(info->src.box.width) &&
       abs(info->dst.box.height) == abs(info->src.box.height)) {
      if (info->src.resource->nr_samples > 1 &&
          info->dst.resource->nr_samples <= 1) {
         /* The OpenGL ES 3.2 specification, section 16.2.1, says:
          *
          *    "If the read framebuffer is multisampled (its effective
          *     value of SAMPLE_BUFFERS is one) and the draw framebuffer
          *     is not (its value of SAMPLE_BUFFERS is zero), the samples
          *     corresponding to each pixel location in the source are
          *     converted to a single sample before being written to the
          *     destination.  The filter parameter is ignored.  If the
          *     source formats are integer types or stencil values, a
          *     single sample’s value is selected for each pixel.  If the
          *     source formats are floating-point or normalized types,
          *     the sample values for each pixel are resolved in an
          *     implementation-dependent manner.  If the source formats
          *     are depth values, sample values are resolved in an
          *     implementation-dependent manner where the result will be
          *     between the minimum and maximum depth values in the pixel."
          *
          * When selecting a single sample, we always choose sample 0.
          */
         if (util_format_is_depth_or_stencil(info->src.format) ||
             util_format_is_pure_integer(info->src.format)) {
            filter = BLORP_FILTER_SAMPLE_0;
         } else {
            filter = BLORP_FILTER_AVERAGE;
         }
      } else {
         /* The OpenGL 4.6 specification, section 18.3.1, says:
          *
          *    "If the source and destination dimensions are identical,
          *     no filtering is applied."
          *
          * Using BLORP_FILTER_NONE will also handle the upsample case by
          * replicating the one value in the source to all values in the
          * destination.
          */
         filter = BLORP_FILTER_NONE;
      }
   } else if (info->filter == PIPE_TEX_FILTER_LINEAR) {
      filter = BLORP_FILTER_BILINEAR;
   } else {
      filter = BLORP_FILTER_NEAREST;
   }

   struct blorp_batch blorp_batch;
   blorp_batch_init(&ice->blorp, &blorp_batch, batch, blorp_flags);

   float src_z_step = (float)info->src.box.depth / (float)info->dst.box.depth;

   /* There is no interpolation to the pixel center during rendering, so
    * add the 0.5 offset ourselves here.
    */
   float depth_center_offset = 0;
   if (info->src.resource->target == PIPE_TEXTURE_3D)
      depth_center_offset = 0.5 / info->dst.box.depth * info->src.box.depth;

   /* Perform a blit for each aspect requested by the caller. PIPE_MASK_R is
    * used to represent the color aspect. */
   unsigned aspect_mask = info->mask & (PIPE_MASK_R | PIPE_MASK_ZS);
   while (aspect_mask) {
      unsigned aspect = 1 << u_bit_scan(&aspect_mask);

      struct iris_resource *src_res =
         iris_resource_for_aspect(info->src.resource, aspect);
      struct iris_resource *dst_res =
         iris_resource_for_aspect(info->dst.resource, aspect);

      enum pipe_format src_pfmt =
         pipe_format_for_aspect(info->src.format, aspect);
      enum pipe_format dst_pfmt =
         pipe_format_for_aspect(info->dst.format, aspect);

      struct iris_format_info src_fmt =
         iris_format_for_usage(devinfo, src_pfmt, ISL_SURF_USAGE_TEXTURE_BIT);
      enum isl_aux_usage src_aux_usage =
         iris_resource_texture_aux_usage(ice, src_res, src_fmt.fmt,
                                         info->src.level, 1);

      iris_resource_prepare_texture(ice, src_res, src_fmt.fmt,
                                    info->src.level, 1, info->src.box.z,
                                    info->src.box.depth);
      iris_emit_buffer_barrier_for(batch, src_res->bo,
                                   IRIS_DOMAIN_SAMPLER_READ);

      struct iris_format_info dst_fmt =
         iris_format_for_usage(devinfo, dst_pfmt,
                               ISL_SURF_USAGE_RENDER_TARGET_BIT);
      enum isl_aux_usage dst_aux_usage =
         iris_resource_render_aux_usage(ice, dst_res, dst_fmt.fmt,
                                        info->dst.level, false);

      iris_resource_prepare_render(ice, dst_res, dst_fmt.fmt, info->dst.level,
                                   info->dst.box.z, info->dst.box.depth,
                                   dst_aux_usage);
      iris_emit_buffer_barrier_for(batch, dst_res->bo,
                                   IRIS_DOMAIN_RENDER_WRITE);

      struct blorp_surf src_surf, dst_surf;
      iris_blorp_surf_for_resource(batch,  &src_surf,
                                   &src_res->base.b, src_aux_usage,
                                   info->src.level, false);
      iris_blorp_surf_for_resource(batch, &dst_surf,
                                   &dst_res->base.b, dst_aux_usage,
                                   info->dst.level, true);

      if (iris_batch_references(batch, src_res->bo))
         tex_cache_flush_hack(batch, src_fmt.fmt, src_res->surf.format);

      if (dst_res->base.b.target == PIPE_BUFFER) {
         util_range_add(&dst_res->base.b, &dst_res->valid_buffer_range,
                        dst_x0, dst_x1);
      }

      for (int slice = 0; slice < info->dst.box.depth; slice++) {
         unsigned dst_z = info->dst.box.z + slice;
         float src_z = info->src.box.z + slice * src_z_step +
                       depth_center_offset;

         iris_batch_maybe_flush(batch, 1500);
         iris_batch_sync_region_start(batch);

         blorp_blit(&blorp_batch,
                    &src_surf, info->src.level, src_z,
                    src_fmt.fmt, src_fmt.swizzle,
                    &dst_surf, info->dst.level, dst_z,
                    dst_fmt.fmt, dst_fmt.swizzle,
                    src_x0, src_y0, src_x1, src_y1,
                    dst_x0, dst_y0, dst_x1, dst_y1,
                    filter, mirror_x, mirror_y);

         iris_batch_sync_region_end(batch);
      }

      tex_cache_flush_hack(batch, src_fmt.fmt, src_res->surf.format);

      iris_resource_finish_render(ice, dst_res, info->dst.level,
                                  info->dst.box.z, info->dst.box.depth,
                                  dst_aux_usage);
   }

   blorp_batch_finish(&blorp_batch);

   iris_dirty_for_history(ice, (struct iris_resource *)info->dst.resource);
}

static enum isl_aux_usage
copy_region_aux_usage(struct iris_context *ice,
                      const struct iris_batch *batch,
                      struct iris_resource *res,
                      enum isl_format view_format,
                      unsigned level,
                      bool is_dest)
{
   struct iris_screen *screen = (void *) ice->ctx.screen;
   const struct intel_device_info *devinfo = screen->devinfo;

   if (batch->name == IRIS_BATCH_RENDER) {
      if (is_dest) {
         return iris_resource_render_aux_usage(ice, res, view_format, level,
                                               false);
      } else {
         return iris_resource_texture_aux_usage(ice, res, view_format, level,
                                                1);
      }
   } else {
      assert(batch->name == IRIS_BATCH_BLITTER);

      if (is_dest) {

         /* We only blit to images created with PIPE_BIND_PRIME_BLIT_DST.
          * These are only created with dri3_alloc_render_buffer. That
          * function also makes them linear, so they lack compression.
          *
          * Note: this code block could be substituted with the one below, but
          * this setup clarifies that no additional handling of FCV is
          * necessary.
          */
         assert(res->base.b.bind & PIPE_BIND_PRIME_BLIT_DST);
         assert(res->surf.tiling == ISL_TILING_LINEAR);
         assert(res->aux.usage == ISL_AUX_USAGE_NONE);

         return ISL_AUX_USAGE_NONE;
      } else {
         return devinfo->verx10 >= 125 ? res->aux.usage : ISL_AUX_USAGE_NONE;
      }
   }
}

static void
prepare_copy_region(struct iris_context *ice,
                    const struct iris_batch *batch,
                    struct iris_resource *res,
                    enum isl_format view_format,
                    uint32_t level,
                    uint32_t start_layer, uint32_t layer_count,
                    enum isl_aux_usage aux_usage,
                    bool is_dest)
{
   if (batch->name == IRIS_BATCH_RENDER) {
      if (is_dest) {
         iris_resource_prepare_render(ice, res, view_format, level,
                                      start_layer, layer_count, aux_usage);
      } else {
         iris_resource_prepare_texture(ice, res, view_format, level, 1,
                                       start_layer, layer_count);
      }
   } else {
      assert(batch->name == IRIS_BATCH_BLITTER);
      iris_resource_prepare_access(ice, res, level, 1,
                                   start_layer, layer_count, aux_usage,
                                   isl_aux_usage_has_fast_clears(aux_usage) &&
                                   clear_color_is_fully_zero(res));
   }
}

/**
 * Perform a GPU-based raw memory copy between compatible view classes.
 *
 * Does not perform any flushing - the new data may still be left in the
 * render cache, and old data may remain in other caches.
 *
 * Wraps blorp_copy() and blorp_buffer_copy().
 */
void
iris_copy_region(struct blorp_context *blorp,
                 struct iris_batch *batch,
                 struct pipe_resource *dst,
                 unsigned dst_level,
                 unsigned dstx, unsigned dsty, unsigned dstz,
                 struct pipe_resource *src,
                 unsigned src_level,
                 const struct pipe_box *src_box)
{
   struct blorp_batch blorp_batch;
   struct iris_context *ice = blorp->driver_ctx;
   struct iris_screen *screen = (void *) ice->ctx.screen;
   struct iris_resource *src_res = (void *) src;
   struct iris_resource *dst_res = (void *) dst;

   enum iris_domain write_domain =
      batch->name == IRIS_BATCH_BLITTER ? IRIS_DOMAIN_OTHER_WRITE
                                        : IRIS_DOMAIN_RENDER_WRITE;

   enum isl_format src_fmt, dst_fmt;
   blorp_copy_get_formats(&screen->isl_dev, &src_res->surf, &dst_res->surf,
                          &src_fmt, &dst_fmt);

   enum isl_aux_usage src_aux_usage =
      copy_region_aux_usage(ice, batch, src_res, src_fmt, src_level, false);
   enum isl_aux_usage dst_aux_usage =
      copy_region_aux_usage(ice, batch, dst_res, dst_fmt, dst_level, true);

   if (iris_batch_references(batch, src_res->bo)) {
      /* blorp_copy_get_formats() is only valid for compressed surfaces. */
      tex_cache_flush_hack(batch,
                           src_aux_usage == ISL_AUX_USAGE_NONE ?
                           ISL_FORMAT_UNSUPPORTED : src_fmt,
                           src_res->surf.format);
   }

   if (dst->target == PIPE_BUFFER)
      util_range_add(&dst_res->base.b, &dst_res->valid_buffer_range, dstx, dstx + src_box->width);

   enum blorp_batch_flags blorp_flags = iris_blorp_flags_for_batch(batch);

   blorp_batch_init(blorp, &blorp_batch, batch, blorp_flags);

   if (dst->target == PIPE_BUFFER && src->target == PIPE_BUFFER) {
      struct blorp_address src_addr = {
         .buffer = src_res->bo, .offset = src_res->offset + src_box->x,
         .mocs = iris_mocs(src_res->bo, &screen->isl_dev,
                           iris_blorp_batch_usage(batch, false /* is_dest */)),
         .local_hint = iris_bo_likely_local(src_res->bo),
      };
      struct blorp_address dst_addr = {
         .buffer = dst_res->bo, .offset = dst_res->offset + dstx,
         .reloc_flags = IRIS_BLORP_RELOC_FLAGS_EXEC_OBJECT_WRITE,
         .mocs = iris_mocs(dst_res->bo, &screen->isl_dev,
                           iris_blorp_batch_usage(batch, true /* is_dest */)),
         .local_hint = iris_bo_likely_local(dst_res->bo),
      };

      iris_emit_buffer_barrier_for(batch, src_res->bo,
                                   IRIS_DOMAIN_SAMPLER_READ);
      iris_emit_buffer_barrier_for(batch, dst_res->bo, write_domain);

      iris_batch_maybe_flush(batch, 1500);

      iris_batch_sync_region_start(batch);
      blorp_buffer_copy(&blorp_batch, src_addr, dst_addr, src_box->width);
      iris_batch_sync_region_end(batch);
   } else {
      // XXX: what about one surface being a buffer and not the other?

      prepare_copy_region(ice, batch, src_res, src_fmt, src_level,
                          src_box->z, src_box->depth, src_aux_usage, false);
      prepare_copy_region(ice, batch, dst_res, dst_fmt, dst_level,
                          dstz, src_box->depth, dst_aux_usage, true);

      iris_emit_buffer_barrier_for(batch, src_res->bo,
                                   IRIS_DOMAIN_SAMPLER_READ);
      iris_emit_buffer_barrier_for(batch, dst_res->bo, write_domain);

      struct blorp_surf src_surf, dst_surf;
      iris_blorp_surf_for_resource(batch, &src_surf, src,
                                   src_aux_usage, src_level, false);
      iris_blorp_surf_for_resource(batch, &dst_surf, dst,
                                   dst_aux_usage, dst_level, true);

      for (int slice = 0; slice < src_box->depth; slice++) {
         iris_batch_maybe_flush(batch, 1500);

         iris_batch_sync_region_start(batch);
         blorp_copy(&blorp_batch, &src_surf, src_level, src_box->z + slice,
                    &dst_surf, dst_level, dstz + slice,
                    src_box->x, src_box->y, dstx, dsty,
                    src_box->width, src_box->height);
         iris_batch_sync_region_end(batch);
      }

      iris_resource_finish_write(ice, dst_res, dst_level, dstz,
                                 src_box->depth, dst_aux_usage);
   }

   blorp_batch_finish(&blorp_batch);

   tex_cache_flush_hack(batch, src_fmt, src_res->surf.format);
}

/**
 * The pipe->resource_copy_region() driver hook.
 *
 * This implements ARB_copy_image semantics - a raw memory copy between
 * compatible view classes.
 */
static void
iris_resource_copy_region(struct pipe_context *ctx,
                          struct pipe_resource *p_dst,
                          unsigned dst_level,
                          unsigned dstx, unsigned dsty, unsigned dstz,
                          struct pipe_resource *p_src,
                          unsigned src_level,
                          const struct pipe_box *src_box)
{
   struct iris_context *ice = (void *) ctx;
   struct iris_batch *batch = &ice->batches[IRIS_BATCH_RENDER];

   iris_copy_region(&ice->blorp, batch, p_dst, dst_level, dstx, dsty, dstz,
                    p_src, src_level, src_box);

   if (util_format_is_depth_and_stencil(p_dst->format) &&
       util_format_has_stencil(util_format_description(p_src->format))) {
      struct iris_resource *junk, *s_src_res, *s_dst_res;
      iris_get_depth_stencil_resources(p_src, &junk, &s_src_res);
      iris_get_depth_stencil_resources(p_dst, &junk, &s_dst_res);

      iris_copy_region(&ice->blorp, batch, &s_dst_res->base.b, dst_level, dstx,
                       dsty, dstz, &s_src_res->base.b, src_level, src_box);
   }

   iris_dirty_for_history(ice, (struct iris_resource *) p_dst);
}

void
iris_init_blit_functions(struct pipe_context *ctx)
{
   ctx->blit = iris_blit;
   ctx->resource_copy_region = iris_resource_copy_region;
}
