@mediabunny/server 1.49.0 → 1.50.0

package/dist/modules/src/video-encoder.js

@@ -48,6 +48,14 @@ const video_sample_1 = require("./video-sample");
 const codec_data_1 = require("../../../src/codec-data");
 const codec_1 = require("../../../src/codec");
 const misc_2 = require("../../../src/misc");
+const PRORES_FOURCC_TO_PROFILE = {
+    apco: NodeAv.AV_PROFILE_PRORES_PROXY,
+    apcs: NodeAv.AV_PROFILE_PRORES_LT,
+    apcn: NodeAv.AV_PROFILE_PRORES_STANDARD,
+    apch: NodeAv.AV_PROFILE_PRORES_HQ,
+    ap4h: NodeAv.AV_PROFILE_PRORES_4444,
+    ap4x: NodeAv.AV_PROFILE_PRORES_XQ,
+};
 class NodeAvVideoEncoder extends mediabunny_1.CustomVideoEncoder {
     constructor() {
         super(...arguments);
@@ -61,8 +69,8 @@ class NodeAvVideoEncoder extends mediabunny_1.CustomVideoEncoder {
         this.preciseTimings = [];
     }
     static supports(codec, config) {
-        return (codec === 'avc' || codec === 'hevc' || codec === 'vp8' || codec === 'vp9' || codec === 'av1')
-            && config.bitrateMode !== 'quantizer';
+        return (codec === 'avc' || codec === 'hevc' || codec === 'vp8' || codec === 'vp9' || codec === 'av1'
+            || codec === 'prores') && config.bitrateMode !== 'quantizer';
     }
     async init() {
         this.frame = new NodeAv.Frame();
@@ -72,15 +80,25 @@ class NodeAvVideoEncoder extends mediabunny_1.CustomVideoEncoder {
         this.packet.alloc();
         const codecId = misc_1.CODEC_TO_CODEC_ID[this.codec];
         (0, misc_2.assert)(codecId !== undefined);
+        const getSoftwareCodec = () => {
+            if (this.codec === 'prores') {
+                // Prefer prores_ks for ProRes
+                const proresKs = NodeAv.Codec.findEncoderByName(NodeAv.FF_ENCODER_PRORES_KS);
+                if (proresKs) {
+                    return proresKs;
+                }
+            }
+            return NodeAv.Codec.findEncoder(codecId);
+        };
         let codec = null;
         if (this.codec === 'vp9' && this.config.alpha === 'keep') {
             codec = NodeAv.Codec.findEncoderByName(NodeAv.FF_ENCODER_LIBVPX_VP9) ?? NodeAv.Codec.findEncoder(codecId);
         }
         else if (this.config.hardwareAcceleration === 'prefer-software') {
-            codec = NodeAv.Codec.findEncoder(codecId);
+            codec = getSoftwareCodec();
         }
         else {
-            codec = (await (0, misc_1.getHardwareEncoderCodec)(codecId)) ?? NodeAv.Codec.findEncoder(codecId);
+            codec = (await (0, misc_1.getHardwareEncoderCodec)(codecId)) ?? getSoftwareCodec();
         }
         if (!codec) {
             throw new Error(`Unable to obtain libav codec for '${this.codec}'.`);
@@ -97,8 +115,15 @@ class NodeAvVideoEncoder extends mediabunny_1.CustomVideoEncoder {
             if (!this.avCodec.pixelFormats.includes(NodeAv.AV_PIX_FMT_YUV420P)) {
                 pixelFormat = this.avCodec.pixelFormats[0];
             }
-            if (this.config.alpha === 'keep' && this.avCodec.pixelFormats.includes(NodeAv.AV_PIX_FMT_YUVA420P)) {
-                pixelFormat = NodeAv.AV_PIX_FMT_YUVA420P;
+            if (this.config.alpha === 'keep') {
+                if (this.avCodec.pixelFormats.includes(NodeAv.AV_PIX_FMT_YUVA420P)) {
+                    pixelFormat = NodeAv.AV_PIX_FMT_YUVA420P;
+                }
+                else {
+                    // Let FFmpeg pick the best alpha-capable format it supports, minimizing data loss versus a
+                    // high-quality YUVA source
+                    pixelFormat = NodeAv.avcodecFindBestPixFmtOfList(this.avCodec.pixelFormats, NodeAv.AV_PIX_FMT_YUVA444P12LE);
+                }
             }
         }
         const pixelAspectRatio = (0, misc_2.simplifyRational)({
@@ -157,6 +182,12 @@ class NodeAvVideoEncoder extends mediabunny_1.CustomVideoEncoder {
                 codecContext.setOption('preset', '12');
             }
         }
+        if (this.codec === 'prores') {
+            // Pick the encoder profile from the requested ProRes four-character code
+            const profile = PRORES_FOURCC_TO_PROFILE[this.config.codec];
+            (0, misc_2.assert)(profile !== undefined);
+            codecContext.setOption('profile', String(profile));
+        }
         const ret = await codecContext.open2();
         NodeAv.FFmpegError.throwIfError(ret, 'Open codec context');
         this.codecContext = codecContext;
@@ -318,6 +349,7 @@ class NodeAvVideoEncoder extends mediabunny_1.CustomVideoEncoder {
                     hevcCodecInfo: null,
                     vp9CodecInfo: null,
                     av1CodecInfo: null,
+                    proresFormat: null,
                 });
                 if (!expectsAnnexB) {
                     decoderConfigDescription = serializedRecord;
@@ -377,6 +409,7 @@ class NodeAvVideoEncoder extends mediabunny_1.CustomVideoEncoder {
                     hevcCodecInfo: null,
                     vp9CodecInfo: null,
                     av1CodecInfo: null,
+                    proresFormat: null,
                 });
             }
         }
@@ -394,6 +427,7 @@ class NodeAvVideoEncoder extends mediabunny_1.CustomVideoEncoder {
                     hevcCodecInfo: null,
                     vp9CodecInfo,
                     av1CodecInfo: null,
+                    proresFormat: null,
                 });
             }
         }
@@ -411,6 +445,24 @@ class NodeAvVideoEncoder extends mediabunny_1.CustomVideoEncoder {
                     hevcCodecInfo: null,
                     vp9CodecInfo: null,
                     av1CodecInfo,
+                    proresFormat: null,
+                });
+            }
+        }
+        else if (this.codec === 'prores') {
+            if (!this.packetEmitted) {
+                decoderConfigCodecString = (0, codec_1.extractVideoCodecString)({
+                    width: this.config.width,
+                    height: this.config.height,
+                    codec: 'prores',
+                    codecDescription: null,
+                    colorSpace: null,
+                    avcType: null,
+                    avcCodecInfo: null,
+                    hevcCodecInfo: null,
+                    vp9CodecInfo: null,
+                    av1CodecInfo: null,
+                    proresFormat: this.config.codec,
                 });
             }
         }