Use GPU in worker rendering

Author: Aditi-1400

src/display/api.js

@@ -462,6 +462,7 @@ function getDocument(src = {}) {
     pdfBug,
     styleElement,
     enableHWA,
+    enableWebGPU,
     loadingParams: {
       disableAutoFetch,
       enableXfa,
@@ -1663,6 +1664,7 @@ class PDFPageProxy {
       pdfBug: this._pdfBug,
       pageColors,
       enableHWA: this._transport.enableHWA,
+      enableWebGPU: this._transport.enableWebGPU,
       operationsFilter,
       rendererHandler:
         shouldRecordOperations || shouldRecordImages
@@ -2673,6 +2675,7 @@ class WorkerTransport {
       styleElement: params.styleElement,
     });
     this.enableHWA = params.enableHWA;
+    this.enableWebGPU = params.enableWebGPU === true;
     this.rendererHandler = params.rendererHandler || null;
     this.loadingParams = params.loadingParams;
     this._params = params;
@@ -3469,6 +3472,7 @@ class InternalRenderTask {
     pdfBug = false,
     pageColors = null,
     enableHWA = false,
+    enableWebGPU = false,
     operationsFilter = null,
     rendererHandler = null,
   }) {
@@ -3501,6 +3505,7 @@ class InternalRenderTask {
     this._canvas = params.canvas;
     this._canvasContext = params.canvas ? null : params.canvasContext;
     this._enableHWA = enableHWA;
+    this._enableWebGPU = enableWebGPU;
     this._dependencyTracker = params.dependencyTracker;
     this._imagesTracker = params.imagesTracker;
     this._operationsFilter = operationsFilter;
@@ -3633,6 +3638,7 @@ class InternalRenderTask {
           pageIndex: this._pageIndex,
           renderTaskId: this._renderTaskId,
           enableHWA: this._enableHWA,
+          enableWebGPU: this._enableWebGPU,
           optionalContentConfigData,
           optionalContentConfigState,
           optionalContentConfigRenderingIntent:

src/display/renderer_worker.js

@@ -15,6 +15,7 @@
 
 import { CanvasGraphics } from "./canvas.js";
 import { FontLoader } from "./font_loader.js";
+import { initGPU } from "./webgpu.js";
 import { isNodeJS } from "../shared/util.js";
 import { MessageHandler } from "../shared/message_handler.js";
 import { ObjectHandler } from "./object_handler.js";
@@ -164,12 +165,13 @@ class RendererMessageHandler {
       this.#cleanupRenderTask(renderTaskId);
     });
 
-    handler.on("InitializeGraphics", data => {
+    handler.on("InitializeGraphics", async data => {
       const {
         canvas,
         pageIndex,
         renderTaskId = pageIndex,
         enableHWA = false,
+        enableWebGPU = false,
         optionalContentConfigData,
         optionalContentConfigState,
         optionalContentConfigRenderingIntent,
@@ -179,6 +181,9 @@ class RendererMessageHandler {
         transparency,
         background,
       } = data;
+      if (enableWebGPU) {
+        await initGPU();
+      }
       const objs = this.#getPageObjs(pageIndex);
       const optionalContentConfig = new OptionalContentConfig(
         optionalContentConfigData,