跳过无需捕获的帧

Vinlic · Nov 23, 2023 · b412c93 · b412c93
1 parent 574847f
commit b412c93
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/core/CaptureContext.js b/core/CaptureContext.js
@@ -180,6 +180,9 @@ export default class CaptureContext {
                         return ____screencastCompleted();
 
                 }
+                // 跳过无需捕获的帧
+                else
+                    await ____skipFrame();
 
                 // 开始捕获下一帧
                 nextFrame.bind(this)();

diff --git a/core/Page.js b/core/Page.js
@@ -485,6 +485,8 @@ export default class Page extends EventEmitter {
         await this.target.exposeFunction("____screencastCompleted", this.#emitScreencastCompleted.bind(this));
         // 暴露CSS动画控制函数
         await this.target.exposeFunction("____seekCSSAnimations", this.#seekCSSAnimations.bind(this));
+        // 暴露跳帧函数
+        await this.target.exposeFunction("____skipFrame", this.#skipFrame.bind(this));
         // 暴露下一帧函数
         await this.target.exposeFunction("____captureFrame", this.#captureFrame.bind(this));
         // 暴露添加音频函数
@@ -539,6 +541,27 @@ export default class Page extends EventEmitter {
         await Promise.all(seekPromises);
     }
 
+    /**
+     * 跳过帧
+     */
+    async #skipFrame() {
+        if (globalConfig.compatibleRenderingMode)
+            return;
+        let timer;
+        // 帧数据捕获
+        const frameData = await Promise.race([
+            this.#cdpSession.send("HeadlessExperimental.beginFrame"),
+            // 帧渲染超时处理
+            new Promise(resolve => timer = setTimeout(() => resolve(false), this.beginFrameTimeout))
+        ]);
+        clearTimeout(timer);
+        // 帧渲染超时处理
+        if (frameData === false) {
+            this.#setState(Page.STATE.UNAVAILABLED);
+            throw new Error("beginFrame wait timeout");
+        }
+    }
+
     /**
      * 捕获帧
      */