piixishstream 0.0.0-stage → 1.0.2

package/dist/client/processing/SPSVUIRewriter.js

@@ -0,0 +1,297 @@
+import { AnnexBBitstreamReader, AnnexBBitstreamWriter, } from "./AnnexBBitstreamReaderWriter.js";
+export function rewriteSPSVUI(buffer) {
+    const reader = new AnnexBBitstreamReader(buffer.subarray(1));
+    const writer = new AnnexBBitstreamWriter();
+    const readBit = (n = 1) => reader.readBits(n);
+    const writeBit = (v, n = 1) => writer.writeBits(v, n);
+    const readU = (n) => reader.readUnsigned(n);
+    const writeU = (v, n) => writer.writeUnsigned(v, n);
+    const readUE = () => reader.readUnsignedExpGolomb();
+    const writeUE = (v) => writer.writeUnsignedExpGolomb(v);
+    const readSE = () => reader.readSignedExpGolomb();
+    const writeSE = (v) => writer.writeSignedExpGolomb(v);
+    // Rewrite the NAL header
+    writeU(buffer[0], 8);
+    const profile_idc = readU(8);
+    writeU(profile_idc, 8);
+    const constraint_flags = readU(8);
+    writeU(constraint_flags, 8);
+    const level_idc = readU(8);
+    writeU(level_idc, 8);
+    const seq_parameter_set_id = readUE();
+    writeUE(seq_parameter_set_id);
+    // If profile in high profiles, additional fields
+    const highProfiles = new Set([
+        100, 110, 122, 244, 44, 83, 86, 118, 128, 138, 144,
+    ]);
+    if (highProfiles.has(profile_idc)) {
+        const chroma_format_idc = readUE();
+        writeUE(chroma_format_idc);
+        if (chroma_format_idc === 3) {
+            const separate_colour_plane_flag = readBit(1);
+            writeBit(separate_colour_plane_flag, 1);
+        }
+        const bit_depth_luma_minus8 = readUE();
+        writeUE(bit_depth_luma_minus8);
+        const bit_depth_chroma_minus8 = readUE();
+        writeUE(bit_depth_chroma_minus8);
+        const qpprime_y_zero_transform_bypass_flag = readBit(1);
+        writeBit(qpprime_y_zero_transform_bypass_flag, 1);
+        const seq_scaling_matrix_present_flag = readBit(1);
+        writeBit(seq_scaling_matrix_present_flag, 1);
+        if (seq_scaling_matrix_present_flag) {
+            const scalingCount = chroma_format_idc !== 3 ? 8 : 12;
+            for (let i = 0; i < scalingCount; i++) {
+                const seq_scaling_list_present_flag = readBit(1);
+                writeBit(seq_scaling_list_present_flag, 1);
+                if (seq_scaling_list_present_flag) {
+                    const size = i < 6 ? 16 : 64;
+                    // scaling_list(size)
+                    let lastScale = 8;
+                    let nextScale = 8;
+                    for (let j = 0; j < size; j++) {
+                        const delta = readSE();
+                        writeSE(delta);
+                        nextScale = (lastScale + delta + 256) % 256;
+                        if (nextScale !== 0)
+                            lastScale = nextScale;
+                    }
+                }
+            }
+        }
+    }
+    const log2_max_frame_num_minus4 = readUE();
+    writeUE(log2_max_frame_num_minus4);
+    const pic_order_cnt_type = readUE();
+    writeUE(pic_order_cnt_type);
+    if (pic_order_cnt_type === 0) {
+        const log2_max_pic_order_cnt_lsb_minus4 = readUE();
+        writeUE(log2_max_pic_order_cnt_lsb_minus4);
+    }
+    else if (pic_order_cnt_type === 1) {
+        const delta_pic_order_always_zero_flag = readBit(1);
+        writeBit(delta_pic_order_always_zero_flag, 1);
+        const offset_for_non_ref_pic = readSE();
+        writeSE(offset_for_non_ref_pic);
+        const offset_for_top_to_bottom_field = readSE();
+        writeSE(offset_for_top_to_bottom_field);
+        const num_ref_frames_in_pic_order_cnt_cycle = readUE();
+        writeUE(num_ref_frames_in_pic_order_cnt_cycle);
+        for (let i = 0; i < num_ref_frames_in_pic_order_cnt_cycle; i++) {
+            const offset_for_ref_frame = readSE();
+            writeSE(offset_for_ref_frame);
+        }
+    }
+    const max_num_ref_frames = readUE();
+    writeUE(max_num_ref_frames);
+    const gaps_in_frame_num_value_allowed_flag = readBit(1);
+    writeBit(gaps_in_frame_num_value_allowed_flag, 1);
+    const pic_width_in_mbs_minus1 = readUE();
+    writeUE(pic_width_in_mbs_minus1);
+    const pic_height_in_map_units_minus1 = readUE();
+    writeUE(pic_height_in_map_units_minus1);
+    const frame_mbs_only_flag = readBit(1);
+    writeBit(frame_mbs_only_flag, 1);
+    if (frame_mbs_only_flag === 0) {
+        const mb_adaptive_frame_field_flag = readBit(1);
+        writeBit(mb_adaptive_frame_field_flag, 1);
+    }
+    const direct_8x8_inference_flag = readBit(1);
+    writeBit(direct_8x8_inference_flag, 1);
+    const frame_cropping_flag = readBit(1);
+    writeBit(frame_cropping_flag, 1);
+    if (frame_cropping_flag) {
+        const frame_crop_left_offset = readUE();
+        writeUE(frame_crop_left_offset);
+        const frame_crop_right_offset = readUE();
+        writeUE(frame_crop_right_offset);
+        const frame_crop_top_offset = readUE();
+        writeUE(frame_crop_top_offset);
+        const frame_crop_bottom_offset = readUE();
+        writeUE(frame_crop_bottom_offset);
+    }
+    // https://webrtc.googlesource.com/src/+/5f2c9278f35e47ff72eb191669d473b7400c9f3e/common_video/h264/sps_vui_rewriter.cc#283
+    function addBitstreamRestriction() {
+        // motion_vectors_over_pic_boundaries_flag: u(1)
+        // Default is 1 when not present.
+        writeBit(1, 1);
+        // max_bytes_per_pic_denom: ue(v)
+        // Default is 2 when not present.
+        writeUE(2);
+        // max_bits_per_mb_denom: ue(v)
+        // Default is 1 when not present.
+        writeUE(1);
+        // log2_max_mv_length_horizontal: ue(v)
+        // log2_max_mv_length_vertical: ue(v)
+        // Both default to 16 when not present.
+        writeUE(16);
+        writeUE(16);
+        // ********* IMPORTANT! **********
+        // max_num_reorder_frames: ue(v)
+        writeUE(0);
+        // max_dec_frame_buffering: ue(v)
+        writeUE(max_num_ref_frames);
+    }
+    const vui_parameters_present_flag = readBit(1);
+    writeBit(1, 1);
+    // If no VUI exists, write one
+    if (!vui_parameters_present_flag) {
+        // aspect_ratio_info_present_flag, overscan_info_present_flag. Both u(1).
+        writeBit(0, 2);
+        // video_signal_type_present_flag, u(1).
+        // Just write 0 here because I'm not gonna bother myself with color space and whatnot
+        writeBit(0, 1);
+        // chroma_loc_info_present_flag, timing_info_present_flag,
+        // nal_hrd_parameters_present_flag, vcl_hrd_parameters_present_flag,
+        // pic_struct_present_flag, All u(1)
+        writeBit(0, 5);
+        // bitstream_restriction_flag: u(1)
+        writeBit(1, 1);
+        addBitstreamRestriction();
+    }
+    else {
+        // VUI parsing and copying
+        const aspect_ratio_info_present_flag = readBit(1);
+        writeBit(aspect_ratio_info_present_flag, 1);
+        if (aspect_ratio_info_present_flag) {
+            const aspect_ratio_idc = readU(8);
+            writeU(aspect_ratio_idc, 8);
+            if (aspect_ratio_idc === 255) {
+                // Extended_SAR
+                const sar_width = readU(16);
+                writeU(sar_width, 16);
+                const sar_height = readU(16);
+                writeU(sar_height, 16);
+            }
+        }
+        const overscan_info_present_flag = readBit(1);
+        writeBit(overscan_info_present_flag, 1);
+        if (overscan_info_present_flag) {
+            const overscan_appropriate_flag = readBit(1);
+            writeBit(overscan_appropriate_flag, 1);
+        }
+        // Read the video signal type, but don't copy it
+        const video_signal_type_present_flag = readBit(1);
+        writeBit(0, 1);
+        if (video_signal_type_present_flag) {
+            const _video_format = readBit(3);
+            // writeBit(video_format, 3);
+            const _video_full_range_flag = readBit(1);
+            // writeBit(video_full_range_flag, 1);
+            const colour_description_present_flag = readBit(1);
+            // writeBit(colour_description_present_flag, 1);
+            if (colour_description_present_flag) {
+                const _colour_primaries = readU(8);
+                // writeU(colour_primaries, 8);
+                const _transfer_characteristics = readU(8);
+                // writeU(transfer_characteristics, 8);
+                const _matrix_coeffs = readU(8);
+                // writeU(matrix_coeffs, 8);
+            }
+        }
+        const chroma_loc_info_present_flag = readBit(1);
+        writeBit(chroma_loc_info_present_flag, 1);
+        if (chroma_loc_info_present_flag) {
+            const chroma_sample_loc_type_top_field = readUE();
+            writeUE(chroma_sample_loc_type_top_field);
+            const chroma_sample_loc_type_bottom_field = readUE();
+            writeUE(chroma_sample_loc_type_bottom_field);
+        }
+        const timing_info_present_flag = readBit(1);
+        writeBit(timing_info_present_flag, 1);
+        if (timing_info_present_flag) {
+            const num_units_in_tick = readU(32);
+            writeU(num_units_in_tick, 32);
+            const time_scale = readU(32);
+            writeU(time_scale, 32);
+            const fixed_frame_rate_flag = readBit(1);
+            writeBit(fixed_frame_rate_flag, 1);
+        }
+        const nal_hrd_parameters_present_flag = readBit(1);
+        writeBit(nal_hrd_parameters_present_flag, 1);
+        if (nal_hrd_parameters_present_flag) {
+            // hrd_parameters()
+            const cpb_cnt_minus1 = readUE();
+            writeUE(cpb_cnt_minus1);
+            const bit_rate_scale = readBit(4);
+            writeBit(bit_rate_scale, 4);
+            const cpb_size_scale = readBit(4);
+            writeBit(cpb_size_scale, 4);
+            for (let i = 0; i <= cpb_cnt_minus1; i++) {
+                const bit_rate_value_minus1 = readUE();
+                writeUE(bit_rate_value_minus1);
+                const cpb_size_value_minus1 = readUE();
+                writeUE(cpb_size_value_minus1);
+                const cbr_flag = readBit(1);
+                writeBit(cbr_flag, 1);
+            }
+            const initial_cpb_removal_delay_length_minus1 = readBit(5);
+            writeBit(initial_cpb_removal_delay_length_minus1, 5);
+            const cpb_removal_delay_length_minus1 = readBit(5);
+            writeBit(cpb_removal_delay_length_minus1, 5);
+            const dpb_output_delay_length_minus1 = readBit(5);
+            writeBit(dpb_output_delay_length_minus1, 5);
+            const time_offset_length = readBit(5);
+            writeBit(time_offset_length, 5);
+        }
+        const vcl_hrd_parameters_present_flag = readBit(1);
+        writeBit(vcl_hrd_parameters_present_flag, 1);
+        if (vcl_hrd_parameters_present_flag) {
+            // hrd_parameters()
+            const cpb_cnt_minus1 = readUE();
+            writeUE(cpb_cnt_minus1);
+            const bit_rate_scale = readBit(4);
+            writeBit(bit_rate_scale, 4);
+            const cpb_size_scale = readBit(4);
+            writeBit(cpb_size_scale, 4);
+            for (let i = 0; i <= cpb_cnt_minus1; i++) {
+                const bit_rate_value_minus1 = readUE();
+                writeUE(bit_rate_value_minus1);
+                const cpb_size_value_minus1 = readUE();
+                writeUE(cpb_size_value_minus1);
+                const cbr_flag = readBit(1);
+                writeBit(cbr_flag, 1);
+            }
+            const initial_cpb_removal_delay_length_minus1 = readBit(5);
+            writeBit(initial_cpb_removal_delay_length_minus1, 5);
+            const cpb_removal_delay_length_minus1 = readBit(5);
+            writeBit(cpb_removal_delay_length_minus1, 5);
+            const dpb_output_delay_length_minus1 = readBit(5);
+            writeBit(dpb_output_delay_length_minus1, 5);
+            const time_offset_length = readBit(5);
+            writeBit(time_offset_length, 5);
+        }
+        if (nal_hrd_parameters_present_flag || vcl_hrd_parameters_present_flag) {
+            const low_delay_hrd_flag = readBit(1);
+            writeBit(low_delay_hrd_flag, 1);
+        }
+        const pic_struct_present_flag = readBit(1);
+        writeBit(pic_struct_present_flag, 1);
+        const bitstream_restriction_flag = readBit(1);
+        writeBit(1, 1);
+        if (!bitstream_restriction_flag) {
+            addBitstreamRestriction();
+        }
+        else {
+            const motion_vectors_over_pic_boundaries_flag = readBit(1);
+            writeBit(motion_vectors_over_pic_boundaries_flag, 1);
+            const max_bytes_per_pic_denom = readUE();
+            writeUE(max_bytes_per_pic_denom);
+            const max_bits_per_mb_denom = readUE();
+            writeUE(max_bits_per_mb_denom);
+            const log2_max_mv_length_horizontal = readUE();
+            writeUE(log2_max_mv_length_horizontal);
+            const log2_max_mv_length_vertical = readUE();
+            writeUE(log2_max_mv_length_vertical);
+            const _num_reorder_frames = readUE();
+            writeUE(0);
+            const _max_dec_frame_buffering = readUE();
+            writeUE(max_num_ref_frames);
+        }
+    }
+    writeBit(1, 1); // rbsp_stop_one_bit
+    writer.flush();
+    // return the rewritten RBSP as a buffer
+    return writer.toBuffer();
+}
+//# sourceMappingURL=SPSVUIRewriter.js.map

package/dist/client/processing/SPSVUIRewriter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SPSVUIRewriter.js","sourceRoot":"","sources":["../../../src/client/processing/SPSVUIRewriter.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,qBAAqB,GACtB,MAAM,kCAAkC,CAAC;AAE1C,MAAM,UAAU,aAAa,CAAC,MAAc;IAC1C,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7D,MAAM,MAAM,GAAG,IAAI,qBAAqB,EAAE,CAAC;IAE3C,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAC9C,MAAM,QAAQ,GAAG,CAAC,CAAS,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9D,MAAM,KAAK,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACpD,MAAM,MAAM,GAAG,CAAC,CAAS,EAAE,CAAS,EAAE,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACpE,MAAM,MAAM,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC;IACpD,MAAM,OAAO,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IAChE,MAAM,MAAM,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC;IAClD,MAAM,OAAO,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;IAE9D,yBAAyB;IACzB,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAErB,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAC7B,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IAEvB,MAAM,gBAAgB,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAClC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAC3B,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;IAErB,MAAM,oBAAoB,GAAG,MAAM,EAAE,CAAC;IACtC,OAAO,CAAC,oBAAoB,CAAC,CAAC;IAE9B,iDAAiD;IACjD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC;QAC3B,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG;KACnD,CAAC,CAAC;IACH,IAAI,YAAY,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;QAClC,MAAM,iBAAiB,GAAG,MAAM,EAAE,CAAC;QACnC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAE3B,IAAI,iBAAiB,KAAK,CAAC,EAAE,CAAC;YAC5B,MAAM,0BAA0B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAC9C,QAAQ,CAAC,0BAA0B,EAAE,CAAC,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,qBAAqB,GAAG,MAAM,EAAE,CAAC;QACvC,OAAO,CAAC,qBAAqB,CAAC,CAAC;QAC/B,MAAM,uBAAuB,GAAG,MAAM,EAAE,CAAC;QACzC,OAAO,CAAC,uBAAuB,CAAC,CAAC;QAEjC,MAAM,oCAAoC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACxD,QAAQ,CAAC,oCAAoC,EAAE,CAAC,CAAC,CAAC;QAElD,MAAM,+BAA+B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACnD,QAAQ,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;QAC7C,IAAI,+BAA+B,EAAE,CAAC;YACpC,MAAM,YAAY,GAAG,iBAAiB,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,MAAM,6BAA6B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;gBACjD,QAAQ,CAAC,6BAA6B,EAAE,CAAC,CAAC,CAAC;gBAC3C,IAAI,6BAA6B,EAAE,CAAC;oBAClC,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC7B,qBAAqB;oBACrB,IAAI,SAAS,GAAG,CAAC,CAAC;oBAClB,IAAI,SAAS,GAAG,CAAC,CAAC;oBAClB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;wBAC9B,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC;wBACvB,OAAO,CAAC,KAAK,CAAC,CAAC;wBACf,SAAS,GAAG,CAAC,SAAS,GAAG,KAAK,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;wBAC5C,IAAI,SAAS,KAAK,CAAC;4BAAE,SAAS,GAAG,SAAS,CAAC;oBAC7C,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,yBAAyB,GAAG,MAAM,EAAE,CAAC;IAC3C,OAAO,CAAC,yBAAyB,CAAC,CAAC;IAEnC,MAAM,kBAAkB,GAAG,MAAM,EAAE,CAAC;IACpC,OAAO,CAAC,kBAAkB,CAAC,CAAC;IAC5B,IAAI,kBAAkB,KAAK,CAAC,EAAE,CAAC;QAC7B,MAAM,iCAAiC,GAAG,MAAM,EAAE,CAAC;QACnD,OAAO,CAAC,iCAAiC,CAAC,CAAC;IAC7C,CAAC;SAAM,IAAI,kBAAkB,KAAK,CAAC,EAAE,CAAC;QACpC,MAAM,gCAAgC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACpD,QAAQ,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;QAC9C,MAAM,sBAAsB,GAAG,MAAM,EAAE,CAAC;QACxC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAChC,MAAM,8BAA8B,GAAG,MAAM,EAAE,CAAC;QAChD,OAAO,CAAC,8BAA8B,CAAC,CAAC;QACxC,MAAM,qCAAqC,GAAG,MAAM,EAAE,CAAC;QACvD,OAAO,CAAC,qCAAqC,CAAC,CAAC;QAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,qCAAqC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/D,MAAM,oBAAoB,GAAG,MAAM,EAAE,CAAC;YACtC,OAAO,CAAC,oBAAoB,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,MAAM,kBAAkB,GAAG,MAAM,EAAE,CAAC;IACpC,OAAO,CAAC,kBAAkB,CAAC,CAAC;IAE5B,MAAM,oCAAoC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IACxD,QAAQ,CAAC,oCAAoC,EAAE,CAAC,CAAC,CAAC;IAElD,MAAM,uBAAuB,GAAG,MAAM,EAAE,CAAC;IACzC,OAAO,CAAC,uBAAuB,CAAC,CAAC;IAEjC,MAAM,8BAA8B,GAAG,MAAM,EAAE,CAAC;IAChD,OAAO,CAAC,8BAA8B,CAAC,CAAC;IAExC,MAAM,mBAAmB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IACvC,QAAQ,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC;IACjC,IAAI,mBAAmB,KAAK,CAAC,EAAE,CAAC;QAC9B,MAAM,4BAA4B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAChD,QAAQ,CAAC,4BAA4B,EAAE,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,MAAM,yBAAyB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAC7C,QAAQ,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;IAEvC,MAAM,mBAAmB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IACvC,QAAQ,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC;IACjC,IAAI,mBAAmB,EAAE,CAAC;QACxB,MAAM,sBAAsB,GAAG,MAAM,EAAE,CAAC;QACxC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAChC,MAAM,uBAAuB,GAAG,MAAM,EAAE,CAAC;QACzC,OAAO,CAAC,uBAAuB,CAAC,CAAC;QACjC,MAAM,qBAAqB,GAAG,MAAM,EAAE,CAAC;QACvC,OAAO,CAAC,qBAAqB,CAAC,CAAC;QAC/B,MAAM,wBAAwB,GAAG,MAAM,EAAE,CAAC;QAC1C,OAAO,CAAC,wBAAwB,CAAC,CAAC;IACpC,CAAC;IAED,2HAA2H;IAC3H,SAAS,uBAAuB;QAC9B,gDAAgD;QAChD,iCAAiC;QACjC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACf,iCAAiC;QACjC,iCAAiC;QACjC,OAAO,CAAC,CAAC,CAAC,CAAC;QACX,+BAA+B;QAC/B,iCAAiC;QACjC,OAAO,CAAC,CAAC,CAAC,CAAC;QACX,uCAAuC;QACvC,qCAAqC;QACrC,uCAAuC;QACvC,OAAO,CAAC,EAAE,CAAC,CAAC;QACZ,OAAO,CAAC,EAAE,CAAC,CAAC;QACZ,kCAAkC;QAClC,gCAAgC;QAChC,OAAO,CAAC,CAAC,CAAC,CAAC;QACX,iCAAiC;QACjC,OAAO,CAAC,kBAAkB,CAAC,CAAC;IAC9B,CAAC;IAED,MAAM,2BAA2B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAC/C,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACf,8BAA8B;IAC9B,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACjC,yEAAyE;QACzE,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAEf,wCAAwC;QACxC,qFAAqF;QACrF,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAEf,0DAA0D;QAC1D,oEAAoE;QACpE,oCAAoC;QACpC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAEf,mCAAmC;QACnC,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAEf,uBAAuB,EAAE,CAAC;IAC5B,CAAC;SAAM,CAAC;QACN,0BAA0B;QAC1B,MAAM,8BAA8B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAClD,QAAQ,CAAC,8BAA8B,EAAE,CAAC,CAAC,CAAC;QAC5C,IAAI,8BAA8B,EAAE,CAAC;YACnC,MAAM,gBAAgB,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC;YAC5B,IAAI,gBAAgB,KAAK,GAAG,EAAE,CAAC;gBAC7B,eAAe;gBACf,MAAM,SAAS,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC5B,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;gBACtB,MAAM,UAAU,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC7B,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;QAED,MAAM,0BAA0B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC9C,QAAQ,CAAC,0BAA0B,EAAE,CAAC,CAAC,CAAC;QACxC,IAAI,0BAA0B,EAAE,CAAC;YAC/B,MAAM,yBAAyB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAC7C,QAAQ,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;QACzC,CAAC;QAED,gDAAgD;QAChD,MAAM,8BAA8B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAClD,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACf,IAAI,8BAA8B,EAAE,CAAC;YACnC,MAAM,aAAa,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACjC,6BAA6B;YAC7B,MAAM,sBAAsB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAC1C,sCAAsC;YACtC,MAAM,+BAA+B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACnD,gDAAgD;YAChD,IAAI,+BAA+B,EAAE,CAAC;gBACpC,MAAM,iBAAiB,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;gBACnC,+BAA+B;gBAC/B,MAAM,yBAAyB,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;gBAC3C,uCAAuC;gBACvC,MAAM,cAAc,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;gBAChC,4BAA4B;YAC9B,CAAC;QACH,CAAC;QAED,MAAM,4BAA4B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAChD,QAAQ,CAAC,4BAA4B,EAAE,CAAC,CAAC,CAAC;QAC1C,IAAI,4BAA4B,EAAE,CAAC;YACjC,MAAM,gCAAgC,GAAG,MAAM,EAAE,CAAC;YAClD,OAAO,CAAC,gCAAgC,CAAC,CAAC;YAC1C,MAAM,mCAAmC,GAAG,MAAM,EAAE,CAAC;YACrD,OAAO,CAAC,mCAAmC,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,wBAAwB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC5C,QAAQ,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAAC;QACtC,IAAI,wBAAwB,EAAE,CAAC;YAC7B,MAAM,iBAAiB,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC;YACpC,MAAM,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;YAC9B,MAAM,UAAU,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC;YAC7B,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACvB,MAAM,qBAAqB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACzC,QAAQ,CAAC,qBAAqB,EAAE,CAAC,CAAC,CAAC;QACrC,CAAC;QAED,MAAM,+BAA+B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACnD,QAAQ,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;QAC7C,IAAI,+BAA+B,EAAE,CAAC;YACpC,mBAAmB;YACnB,MAAM,cAAc,GAAG,MAAM,EAAE,CAAC;YAChC,OAAO,CAAC,cAAc,CAAC,CAAC;YACxB,MAAM,cAAc,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAClC,QAAQ,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;YAC5B,MAAM,cAAc,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAClC,QAAQ,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;YAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,cAAc,EAAE,CAAC,EAAE,EAAE,CAAC;gBACzC,MAAM,qBAAqB,GAAG,MAAM,EAAE,CAAC;gBACvC,OAAO,CAAC,qBAAqB,CAAC,CAAC;gBAC/B,MAAM,qBAAqB,GAAG,MAAM,EAAE,CAAC;gBACvC,OAAO,CAAC,qBAAqB,CAAC,CAAC;gBAC/B,MAAM,QAAQ,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;gBAC5B,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;YACxB,CAAC;YACD,MAAM,uCAAuC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAC3D,QAAQ,CAAC,uCAAuC,EAAE,CAAC,CAAC,CAAC;YACrD,MAAM,+BAA+B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACnD,QAAQ,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;YAC7C,MAAM,8BAA8B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAClD,QAAQ,CAAC,8BAA8B,EAAE,CAAC,CAAC,CAAC;YAC5C,MAAM,kBAAkB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACtC,QAAQ,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;QAClC,CAAC;QAED,MAAM,+BAA+B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACnD,QAAQ,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;QAC7C,IAAI,+BAA+B,EAAE,CAAC;YACpC,mBAAmB;YACnB,MAAM,cAAc,GAAG,MAAM,EAAE,CAAC;YAChC,OAAO,CAAC,cAAc,CAAC,CAAC;YACxB,MAAM,cAAc,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAClC,QAAQ,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;YAC5B,MAAM,cAAc,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAClC,QAAQ,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;YAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,cAAc,EAAE,CAAC,EAAE,EAAE,CAAC;gBACzC,MAAM,qBAAqB,GAAG,MAAM,EAAE,CAAC;gBACvC,OAAO,CAAC,qBAAqB,CAAC,CAAC;gBAC/B,MAAM,qBAAqB,GAAG,MAAM,EAAE,CAAC;gBACvC,OAAO,CAAC,qBAAqB,CAAC,CAAC;gBAC/B,MAAM,QAAQ,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;gBAC5B,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;YACxB,CAAC;YACD,MAAM,uCAAuC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAC3D,QAAQ,CAAC,uCAAuC,EAAE,CAAC,CAAC,CAAC;YACrD,MAAM,+BAA+B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACnD,QAAQ,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;YAC7C,MAAM,8BAA8B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAClD,QAAQ,CAAC,8BAA8B,EAAE,CAAC,CAAC,CAAC;YAC5C,MAAM,kBAAkB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACtC,QAAQ,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;QAClC,CAAC;QAED,IAAI,+BAA+B,IAAI,+BAA+B,EAAE,CAAC;YACvE,MAAM,kBAAkB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACtC,QAAQ,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;QAClC,CAAC;QAED,MAAM,uBAAuB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC3C,QAAQ,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;QAErC,MAAM,0BAA0B,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC9C,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACf,IAAI,CAAC,0BAA0B,EAAE,CAAC;YAChC,uBAAuB,EAAE,CAAC;QAC5B,CAAC;aAAM,CAAC;YACN,MAAM,uCAAuC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAC3D,QAAQ,CAAC,uCAAuC,EAAE,CAAC,CAAC,CAAC;YACrD,MAAM,uBAAuB,GAAG,MAAM,EAAE,CAAC;YACzC,OAAO,CAAC,uBAAuB,CAAC,CAAC;YACjC,MAAM,qBAAqB,GAAG,MAAM,EAAE,CAAC;YACvC,OAAO,CAAC,qBAAqB,CAAC,CAAC;YAC/B,MAAM,6BAA6B,GAAG,MAAM,EAAE,CAAC;YAC/C,OAAO,CAAC,6BAA6B,CAAC,CAAC;YACvC,MAAM,2BAA2B,GAAG,MAAM,EAAE,CAAC;YAC7C,OAAO,CAAC,2BAA2B,CAAC,CAAC;YACrC,MAAM,mBAAmB,GAAG,MAAM,EAAE,CAAC;YACrC,OAAO,CAAC,CAAC,CAAC,CAAC;YACX,MAAM,wBAAwB,GAAG,MAAM,EAAE,CAAC;YAC1C,OAAO,CAAC,kBAAkB,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,oBAAoB;IACpC,MAAM,CAAC,KAAK,EAAE,CAAC;IACf,wCAAwC;IACxC,OAAO,MAAM,CAAC,QAAQ,EAAE,CAAC;AAC3B,CAAC"}

package/dist/client/voice/BaseMediaConnection.d.ts

@@ -0,0 +1,80 @@
+import Davey from "@snazzah/davey";
+import EventEmitter from "node:events";
+import { WebRtcConnWrapper } from "./WebRtcWrapper.js";
+import { VoiceOpCodesBinary } from "./VoiceOpCodes.js";
+import { type SupportedEncryptionModes } from "../../utils.js";
+import type { Message, GatewayRequest } from "./VoiceMessageTypes.js";
+import type { Streamer } from "../Streamer.js";
+type VoiceConnectionStatus = {
+    hasSession: boolean;
+    hasToken: boolean;
+    started: boolean;
+    resuming: boolean;
+};
+type WebRtcParameters = {
+    address: string;
+    port: number;
+    audioSsrc: number;
+    videoSsrc: number;
+    rtxSsrc: number;
+    supportedEncryptionModes: SupportedEncryptionModes[];
+};
+export type VideoAttributes = {
+    width: number;
+    height: number;
+    fps: number;
+};
+export declare abstract class BaseMediaConnection extends EventEmitter {
+    private interval;
+    guildId: string | null;
+    channelId: string;
+    botId: string;
+    ws: WebSocket | null;
+    status: VoiceConnectionStatus;
+    server: string | null;
+    token: string | null;
+    session_id: string | null;
+    private _webRtcWrapper;
+    private _webRtcParams;
+    private _closed;
+    ready: (conn: WebRtcConnWrapper) => void;
+    private _streamer;
+    private _sequenceNumber;
+    private _daveSession;
+    private _connectedUsers;
+    private _daveProtocolVersion;
+    private _davePendingTransitions;
+    private _daveDowngraded;
+    private _logger;
+    private _loggerDave;
+    constructor(streamer: Streamer, guildId: string | null, botId: string, channelId: string, callback: (conn: WebRtcConnWrapper) => void);
+    abstract get serverId(): string | null;
+    get type(): "guild" | "call";
+    get webRtcConn(): WebRtcConnWrapper;
+    get webRtcParams(): WebRtcParameters | null;
+    get streamer(): Streamer;
+    abstract get daveChannelId(): string;
+    stop(): void;
+    setSession(session_id: string): void;
+    setTokens(server: string, token: string): void;
+    start(): void;
+    handleReady(d: Message.Ready): void;
+    handleProtocolAck(d: Message.SelectProtocolAck): Promise<void>;
+    initDave(): void;
+    processInvalidCommit(transitionId: number): void;
+    executePendingTransition(transitionId: number): void;
+    setupEvents(): void;
+    handleBinaryMessages(msg: Buffer): void;
+    get daveReady(): boolean | 0 | undefined;
+    get daveSession(): Davey.DAVESession | undefined;
+    setupHeartbeat(interval: number): void;
+    sendOpcode<T extends GatewayRequest>(code: T["op"], data: T["d"]): void;
+    sendOpcodeBinary(code: VoiceOpCodesBinary, data: Buffer): void;
+    identify(): void;
+    resume(): void;
+    setProtocols(): Promise<void>;
+    setVideoAttributes(enabled: false): void;
+    setVideoAttributes(enabled: true, attr: VideoAttributes): void;
+    setSpeaking(speaking: boolean): void;
+}
+export {};