asahi: implement blit-based resource_copy_region

doesn't handle complex cases, but faster than hitting the cpu!

Signed-off-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/26963>

src/gallium/drivers/asahi/agx_blit.c

@@ -387,3 +387,71 @@ agx_blit(struct pipe_context *pipe, const struct pipe_blit_info *info)
    agx_blitter_save(ctx, ctx->blitter, info->render_condition_enable);
    util_blitter_blit(ctx->blitter, info);
 }
+
+static bool
+try_copy_via_blit(struct pipe_context *pctx, 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 agx_context *ctx = agx_context(pctx);
+
+   if (dst->target == PIPE_BUFFER)
+      return false;
+
+   struct pipe_blit_info info = {
+      .dst =
+         {
+            .resource = dst,
+            .level = dst_level,
+            .box.x = dstx,
+            .box.y = dsty,
+            .box.z = dstz,
+            .box.width = src_box->width,
+            .box.height = src_box->height,
+            .box.depth = src_box->depth,
+            .format = dst->format,
+         },
+      .src =
+         {
+            .resource = src,
+            .level = src_level,
+            .box = *src_box,
+            .format = src->format,
+         },
+      .mask = util_format_get_mask(src->format),
+      .filter = PIPE_TEX_FILTER_NEAREST,
+      .scissor_enable = 0,
+   };
+
+   /* snorm formats don't round trip, so don't use them for copies */
+   if (util_format_is_snorm(info.dst.format))
+      info.dst.format = util_format_snorm_to_sint(info.dst.format);
+
+   if (util_format_is_snorm(info.src.format))
+      info.src.format = util_format_snorm_to_sint(info.src.format);
+
+   if (util_blitter_is_blit_supported(ctx->blitter, &info) &&
+       info.dst.format == info.src.format) {
+
+      agx_blit(pctx, &info);
+      return true;
+   } else {
+      return false;
+   }
+}
+
+void
+agx_resource_copy_region(struct pipe_context *pctx, 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)
+{
+   if (try_copy_via_blit(pctx, dst, dst_level, dstx, dsty, dstz, src, src_level,
+                         src_box))
+      return;
+
+   /* CPU fallback */
+   util_resource_copy_region(pctx, dst, dst_level, dstx, dsty, dstz, src,
+                             src_level, src_box);
+}

src/gallium/drivers/asahi/agx_pipe.c

@@ -1470,7 +1470,7 @@ agx_create_context(struct pipe_screen *screen, void *priv, unsigned flags)
    pctx->destroy = agx_destroy_context;
    pctx->flush = agx_flush;
    pctx->clear = agx_clear;
-   pctx->resource_copy_region = util_resource_copy_region;
+   pctx->resource_copy_region = agx_resource_copy_region;
    pctx->blit = agx_blit;
    pctx->flush_resource = agx_flush_resource;
 

src/gallium/drivers/asahi/agx_state.h

@@ -967,6 +967,12 @@ void agx_blitter_save(struct agx_context *ctx, struct blitter_context *blitter,
 
 void agx_blit(struct pipe_context *pipe, const struct pipe_blit_info *info);
 
+void agx_resource_copy_region(struct pipe_context *pctx,
+                              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);
+
 /* Batch logic */
 
 struct agx_encoder agx_encoder_allocate(struct agx_batch *batch,