auto-editor 27.0.0__py3-none-any.whl → 27.1.1__py3-none-any.whl

{auto_editor-27.0.0.dist-info → auto_editor-27.1.1.dist-info}/RECORD

@@ -1,56 +1,54 @@
-auto_editor/__init__.py,sha256=KViW4EBA7GbEItgvzRFMRG3Xq3ASIbwCSsNE9WwakVc,23
-auto_editor/__main__.py,sha256=iKfQKFOQDv0-uZecLzfb8A-0k5dAxxe94FaIIEayGXc,15575
+auto_editor/__init__.py,sha256=0ioEHfOkA9Eoup9m_rzaYLCKq4b4PYoylPtUx3iCRPE,23
+auto_editor/__main__.py,sha256=WfNtjKwx5fDMPpfSNLarigXD3Jp0My98FpzQqSAxQZ8,15807
 auto_editor/analyze.py,sha256=CeJG0LI9wXZk1R-QPrNGPS4za-_Avd8y7H-D437DqLg,12300
-auto_editor/edit.py,sha256=9rJC6IY-jMdQDolJBxfzng43ddYQpC8zuZQHLzEct8s,20698
-auto_editor/ffwrapper.py,sha256=b2XDsJMmBnYtRAGrIjqhYH909GNo30tMW3FCZUhhZ5k,4781
+auto_editor/edit.py,sha256=qeuStjcBbRVhLweap_pxxAfME3Jh6tHyxywOjrYJ_ts,19125
+auto_editor/ffwrapper.py,sha256=Wet6B5nohgnjpBX7o20Zq0rYr-H9mUuOqHrbQAPPj38,5128
 auto_editor/help.py,sha256=CzfDTsL4GuGu596ySHKj_wKnxGR9h8B0KUdkZpo33oE,8044
 auto_editor/json.py,sha256=8IVhZJSLx2IVqJsbR5YKDvbHOhgIOvdQmYNpMdMG_xA,9332
-auto_editor/make_layers.py,sha256=VKdBj6Kbe5CAoDIDK_MdqHVu2TXZ0sjeEbqyFTA27gI,9617
-auto_editor/output.py,sha256=YgkZw0WyVfcTYH-4j6kYfvB6A1BjxGLmfOQVFi-QK_o,2561
+auto_editor/make_layers.py,sha256=nSEeCHysMot2eze23q05g2HFDuskN_4Jk108xlk2Rw8,10102
 auto_editor/preview.py,sha256=cqQdozM2IB-5qXHNxeqiSrSdEIzlMfjD4SU-NX9sYZ0,3052
-auto_editor/timeline.py,sha256=f1cxAhduoVHBTRgRwGqI3BKBb31aAVrWkEahH5gJ-0o,8379
+auto_editor/timeline.py,sha256=wUduvIB7JRoV_kswSBWJar_aNJH6i-RO917xrON6lIM,9521
 auto_editor/vanparse.py,sha256=Ug5A2QaRqGiw4l55Z_h9T2QU1x0WqRibR7yY5rQ0WTk,10002
-auto_editor/wavfile.py,sha256=afFfje8cK9lFjIkYoBbQHfvQIpsEPxWvspCtFhUKlAw,9499
 auto_editor/cmds/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 auto_editor/cmds/cache.py,sha256=bViYbtVXefTeEIUvSanDfA6cG35ep1N_Jvtz7ZjgIkY,1959
-auto_editor/cmds/desc.py,sha256=GDrKJYiHMaeTrplZAceXl1JwoqD78XsV2_5lc0Xd7po,869
-auto_editor/cmds/info.py,sha256=Af2aa9vW_bNfFGbRgQMqc7jj17skC6koE0XL3aF0jl0,7005
-auto_editor/cmds/levels.py,sha256=EQ2hfRTe6779OtyzPIBZc5g3WhMZRddVkQWSCVV7UlE,5635
+auto_editor/cmds/desc.py,sha256=DSAWPKt8PS9soBTgzpsFIEWoTe4gPTWwjXpNL-p3WsI,866
+auto_editor/cmds/info.py,sha256=VA2WkTBbQPvrVHjDaJyqryoVMtiiN6cguiUMdWgBJfU,7002
+auto_editor/cmds/levels.py,sha256=2Hbvoy6wMbRRoHdZf25PMqq1uvhRKyPcITNPMpZFo7s,5632
 auto_editor/cmds/palet.py,sha256=ONzTqemaQq9YEfIOsDRNnwzfqnEMUMSXIQrETxyroRU,749
-auto_editor/cmds/repl.py,sha256=8DgMw-XyfR5XctSmwtk4_1-zxs3ooMs72BfMRlVqLvY,3412
+auto_editor/cmds/repl.py,sha256=HSUTDaVykPb5Bd-v_jz_8R7HvFmKOcT_ZVmP-d0AbUY,3247
 auto_editor/cmds/subdump.py,sha256=kHg8nfUi6I6VeJjEgMxupPa666qsYUh7ZEUxint7Gqo,2443
-auto_editor/cmds/test.py,sha256=DK7T5BEtMhfrsTJSmXN_tlBKgHdajS-bICiGttq8M0c,27416
+auto_editor/cmds/test.py,sha256=UNN1r6J69-woqA6QU7TeY0WlPxukQzwCRl5DGBTLK8g,28837
 auto_editor/formats/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_editor/formats/fcp11.py,sha256=sqjC36jI47ICPLjZJYiqGwY7foOnWOiNjkPFLdgSnI4,5208
-auto_editor/formats/fcp7.py,sha256=x5cagTzGCAW3i3M6m7TZC1h8gLfSmX1UK-iiDuCpdfs,20289
-auto_editor/formats/json.py,sha256=_dPrNr1ZC8kfSHRinG0rFudv1XFsZf2VlGAl084lMQ8,7663
+auto_editor/formats/fcp11.py,sha256=uaJbLiwiVxqoVy5JCA14wZj-m1wqZIzWh7rS-BsnSQM,5219
+auto_editor/formats/fcp7.py,sha256=Q_raDxOBqo8QcnhaiXrIFPB-Un7wUvQ4SYt6KALOI1s,19212
+auto_editor/formats/json.py,sha256=UUBhFR_79vn4Lxu73B0cVBFgw4qytrmMP-TiCmDFMd0,7666
 auto_editor/formats/shotcut.py,sha256=-ES854LLFCMCBe100JRJedDmuk8zPev17aQMTrzPv-g,4923
 auto_editor/formats/utils.py,sha256=LYXDiqOk9WwUorLGw2D0M7In9BNDkoKikNawuks7hqE,1648
 auto_editor/lang/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 auto_editor/lang/libintrospection.py,sha256=6H1rGp0wqaCud5IPaoEmzULGnYt6ec7_0h32ATcw2oY,261
 auto_editor/lang/libmath.py,sha256=z33A161Oe6vYYK7R6pgYjdZZe63dQkN38Qf36TL3prg,847
 auto_editor/lang/palet.py,sha256=RQjyIZMJSWnzDkHTu-5mt74o9_4zO4VrcH-wLojCF7A,24113
-auto_editor/lang/stdenv.py,sha256=Acf9CVXzglh3KgsJXrO7IyMMjTY5jqpz5u0QADXtDHA,44127
+auto_editor/lang/stdenv.py,sha256=o7kFu7EbaH71XPFGxJUXYGxSeZ8O3i1_C5Hmi9uya4Q,44150
 auto_editor/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 auto_editor/lib/contracts.py,sha256=lExGQymcQUmwG5lC1lO4qm4GY8W0q_yzK_miTaAoPA4,7586
 auto_editor/lib/data_structs.py,sha256=Hnzl5gWvo-geTU0g-lGejj6HQW3VvPv0NBEj2XoGskY,7089
 auto_editor/lib/err.py,sha256=UlszQJdzMZwkbT8x3sY4GkCV_5x9yrd6uVVUzvA8iiI,35
 auto_editor/render/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_editor/render/audio.py,sha256=7iQMtex8hzzLH80pwLD_nwN_ZH3GHjWRmER45ZVEzPk,12646
+auto_editor/render/audio.py,sha256=ejV_NIvrmxMJt6mf7nPNIzlPHOumNhcIO2J-irHO-k8,17427
 auto_editor/render/subtitle.py,sha256=F27T8OsAojUIGTGBWqTdH36h0BnHXSExxIqzOtqyZoE,6129
-auto_editor/render/video.py,sha256=g2TbuCNzhbE8KsS-_3XRLLdmQKFROdwfkymwIbGmtqc,12012
+auto_editor/render/video.py,sha256=uIrYzF4bDZ3vwfX2F6TdR6F73GI4yruGssto9xEQ-AA,11999
 auto_editor/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 auto_editor/utils/bar.py,sha256=Ky9JRf37JTgLyvNuIXDfucaUE8H1vBbCqKLjttmsmmo,4156
 auto_editor/utils/chunks.py,sha256=J-eGKtEz68gFtRrj1kOSgH4Tj_Yz6prNQ7Xr-d9NQJw,52
 auto_editor/utils/cmdkw.py,sha256=aUGBvBel2Ko1o6Rwmr4rEL-BMc5hEnzYLbyZ1GeJdcY,5729
-auto_editor/utils/container.py,sha256=UPqlAos7UIf_ARIAbB3wuaSv1iNLFgy22YFGK8559yM,2487
-auto_editor/utils/func.py,sha256=C8ucgsSEzPyBc-8obhsCXd_uQW0cnCdBn1KVxB7FHjU,2747
+auto_editor/utils/container.py,sha256=CNHChHbhzIrjmDdWw6UzMqscrr9u7A-ZqKWejGjJwYE,2628
+auto_editor/utils/func.py,sha256=ODyjXnzSDatEu08w398K8_xBKYdXMY3IPHiJpGRZDyQ,3250
 auto_editor/utils/log.py,sha256=wPNf6AabV-0cnoS_bPLv1Lh7llQBtNqPKeh07einOuc,3701
 auto_editor/utils/types.py,sha256=j2hd4zMQ9EftDy41Ji2_PFru_7HEZObd9yKA0BJxFaY,7616
-auto_editor-27.0.0.dist-info/licenses/LICENSE,sha256=yiq99pWITHfqS0pbZMp7cy2dnbreTuvBwudsU-njvIM,1210
+auto_editor-27.1.1.dist-info/licenses/LICENSE,sha256=yiq99pWITHfqS0pbZMp7cy2dnbreTuvBwudsU-njvIM,1210
 docs/build.py,sha256=g1uc1H9T_naGaermUiVMMwUpbT0IWElRhjgT0fvCh8w,1914
-auto_editor-27.0.0.dist-info/METADATA,sha256=VhQA-UaBuv5unUT8MUVkQ_oVwGisctF3BS5tTSTkDfA,6165
-auto_editor-27.0.0.dist-info/WHEEL,sha256=tTnHoFhvKQHCh4jz3yCn0WPTYIy7wXx3CJtJ7SJGV7c,91
-auto_editor-27.0.0.dist-info/entry_points.txt,sha256=UAsTc7qJQbnAzHd7KWg-ALo_X9Hj2yDs3M9I2DV3eyI,212
-auto_editor-27.0.0.dist-info/top_level.txt,sha256=jBV5zlbWRbKOa-xaWPvTD45QL7lGExx2BDzv-Ji4dTw,17
-auto_editor-27.0.0.dist-info/RECORD,,
+auto_editor-27.1.1.dist-info/METADATA,sha256=ri7aLym52FDEnnDR7czUR3xqZ025BLmepV0Em6GXjQc,6176
+auto_editor-27.1.1.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
+auto_editor-27.1.1.dist-info/entry_points.txt,sha256=UAsTc7qJQbnAzHd7KWg-ALo_X9Hj2yDs3M9I2DV3eyI,212
+auto_editor-27.1.1.dist-info/top_level.txt,sha256=jBV5zlbWRbKOa-xaWPvTD45QL7lGExx2BDzv-Ji4dTw,17
+auto_editor-27.1.1.dist-info/RECORD,,

{auto_editor-27.0.0.dist-info → auto_editor-27.1.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (77.0.1)
+Generator: setuptools (80.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

auto_editor/output.py

@@ -1,86 +0,0 @@
-from __future__ import annotations
-
-import os.path
-from dataclasses import dataclass, field
-
-import bv
-from bv.audio.resampler import AudioResampler
-
-from auto_editor.ffwrapper import FileInfo
-from auto_editor.utils.bar import Bar
-from auto_editor.utils.log import Log
-from auto_editor.utils.types import split_num_str
-
-
-def parse_bitrate(input_: str, log: Log) -> int:
-    try:
-        val, unit = split_num_str(input_)
-    except Exception as e:
-        log.error(e)
-
-    if unit.lower() == "k":
-        return int(val * 1000)
-    if unit == "M":
-        return int(val * 1_000_000)
-    if unit == "G":
-        return int(val * 1_000_000_000)
-    if unit == "":
-        return int(val)
-
-    log.error(f"Unknown bitrate: {input_}")
-
-
-@dataclass(slots=True)
-class Ensure:
-    _bar: Bar
-    _sr: int
-    log: Log
-    _audios: list[tuple[FileInfo, int]] = field(default_factory=list)
-
-    def audio(self, src: FileInfo, stream: int) -> str:
-        try:
-            label = self._audios.index((src, stream))
-            first_time = False
-        except ValueError:
-            self._audios.append((src, stream))
-            label = len(self._audios) - 1
-            first_time = True
-
-        out_path = os.path.join(self.log.temp, f"{label:x}.wav")
-
-        if first_time:
-            sample_rate = self._sr
-            bar = self._bar
-            self.log.debug(f"Making external audio: {out_path}")
-
-            in_container = bv.open(src.path, "r")
-            out_container = bv.open(
-                out_path, "w", format="wav", options={"rf64": "always"}
-            )
-            astream = in_container.streams.audio[stream]
-
-            if astream.duration is None or astream.time_base is None:
-                dur = 1.0
-            else:
-                dur = float(astream.duration * astream.time_base)
-
-            bar.start(dur, "Extracting audio")
-
-            output_astream = out_container.add_stream(
-                "pcm_s16le", layout="stereo", rate=sample_rate
-            )
-            resampler = AudioResampler(format="s16", layout="stereo", rate=sample_rate)
-            for i, frame in enumerate(in_container.decode(astream)):
-                if i % 1500 == 0 and frame.time is not None:
-                    bar.tick(frame.time)
-
-                for new_frame in resampler.resample(frame):
-                    out_container.mux(output_astream.encode(new_frame))
-
-            out_container.mux(output_astream.encode(None))
-
-            out_container.close()
-            in_container.close()
-            bar.end()
-
-        return out_path

auto_editor/wavfile.py

@@ -1,310 +0,0 @@
-from __future__ import annotations
-
-import io
-import struct
-import sys
-from typing import TYPE_CHECKING, Literal
-
-import numpy as np
-
-PCM = 0x0001
-IEEE_FLOAT = 0x0003
-EXTENSIBLE = 0xFFFE
-
-AudioData = np.memmap | np.ndarray
-Endian = Literal[">", "<"]  # Big Endian, Little Endian
-ByteOrd = Literal["big", "little"]
-
-if TYPE_CHECKING:
-    Reader = io.BufferedReader | io.BytesIO
-    Writer = io.BufferedWriter | io.BytesIO
-
-
-class WavError(Exception):
-    pass
-
-
-def _read_fmt_chunk(
-    fid: Reader, bytes_order: ByteOrd
-) -> tuple[int, int, int, int, int]:
-    size = int.from_bytes(fid.read(4), bytes_order)
-
-    if size < 16:
-        raise WavError("Binary structure of wave file is not compliant")
-
-    format_tag = int.from_bytes(fid.read(2), bytes_order)
-    channels = int.from_bytes(fid.read(2), bytes_order)
-    sr = int.from_bytes(fid.read(4), bytes_order)
-    fid.read(4)  # This is bitrate but we don't need it
-    block_align = int.from_bytes(fid.read(2), bytes_order)
-    bit_depth = int.from_bytes(fid.read(2), bytes_order)
-    bytes_read = 16
-
-    if format_tag == EXTENSIBLE and size >= 18:
-        ext_chunk_size = int.from_bytes(fid.read(2), bytes_order)
-        bytes_read += 2
-        if ext_chunk_size >= 22:
-            extensible_chunk_data = fid.read(22)
-            bytes_read += 22
-            raw_guid = extensible_chunk_data[6:22]
-
-            if bytes_order == "big":
-                tail = b"\x00\x00\x00\x10\x80\x00\x00\xaa\x00\x38\x9b\x71"
-            else:
-                tail = b"\x00\x00\x10\x00\x80\x00\x00\xaa\x00\x38\x9b\x71"
-            if raw_guid.endswith(tail):
-                format_tag = int.from_bytes(raw_guid[:4], bytes_order)
-        else:
-            raise WavError("Binary structure of wave file is not compliant")
-
-    if format_tag not in {PCM, IEEE_FLOAT}:
-        raise WavError(
-            f"Encountered unknown format tag: {format_tag:#06x}, while reading fmt chunk."
-        )
-
-    # move file pointer to next chunk
-    if size > bytes_read:
-        fid.read(size - bytes_read)
-
-    # fmt should always be 16, 18 or 40, but handle it just in case
-    _handle_pad_byte(fid, size)
-
-    return format_tag, channels, sr, block_align, bit_depth
-
-
-def _read_data_chunk(
-    fid: Reader,
-    format_tag: int,
-    channels: int,
-    bit_depth: int,
-    en: Endian,
-    block_align: int,
-    data_size: int | None,
-) -> AudioData:
-    bytes_order: ByteOrd = "big" if en == ">" else "little"
-    size = int.from_bytes(fid.read(4), bytes_order)
-
-    if data_size is not None:
-        # size is only 32-bits here, so get real size from header.
-        size = data_size
-
-    bytes_per_sample = block_align // channels
-
-    if bytes_per_sample in {3, 5, 6, 7}:
-        raise WavError(f"Unsupported bytes per sample: {bytes_per_sample}")
-
-    if format_tag == PCM:
-        if 1 <= bit_depth <= 8:
-            dtype = "u1"  # WAVs of 8-bit integer or less are unsigned
-        elif bit_depth <= 64:
-            dtype = f"{en}i{bytes_per_sample}"
-        else:
-            raise WavError(
-                f"Unsupported bit depth: the WAV file has {bit_depth}-bit integer data."
-            )
-    elif format_tag == IEEE_FLOAT:
-        if bit_depth in {32, 64}:
-            dtype = f"{en}f{bytes_per_sample}"
-        else:
-            raise WavError(
-                f"Unsupported bit depth: the WAV file has {bit_depth}-bit floating-point data."
-            )
-    else:
-        raise WavError(
-            f"Unknown wave file format: {format_tag:#06x}. Supported formats: PCM, IEEE_FLOAT"
-        )
-    if size % 2 == 0:
-        n_samples = size // block_align
-    else:
-        n_samples = (size - 1) // block_align
-
-    if isinstance(fid, io.BufferedReader):
-        data: AudioData = np.memmap(
-            fid, dtype=dtype, mode="c", offset=fid.tell(), shape=(n_samples, channels)
-        )
-        fid.seek(size, 1)
-    else:
-        bytes_per_sample = np.dtype(dtype).itemsize
-        buffer = fid.read(n_samples * channels * bytes_per_sample)
-        data = np.frombuffer(buffer, dtype=dtype).reshape((n_samples, channels))
-
-    _handle_pad_byte(fid, size)
-
-    return data
-
-
-def _skip_unknown_chunk(fid: Reader, en: Endian) -> None:
-    data = fid.read(4)
-
-    if len(data) == 4:
-        size = struct.unpack(f"{en}I", data)[0]
-        if size == 0:
-            raise WavError("Unknown chunk size is 0")
-        fid.seek(size, 1)
-        _handle_pad_byte(fid, size)
-    elif len(data) == 0:
-        pass  # It's okay, we've hit EOF
-    else:
-        raise WavError(
-            f"Unknown chunk size has wrong length, expected 4, got {len(data)}"
-        )
-
-
-def _read_rf64_chunk(fid: Reader) -> tuple[int, int, Endian]:
-    # https://tech.ebu.ch/docs/tech/tech3306v1_0.pdf
-    # https://www.itu.int/dms_pubrec/itu-r/rec/bs/R-REC-BS.2088-1-201910-I!!PDF-E.pdf
-
-    heading = fid.read(12)
-    if heading != b"\xff\xff\xff\xffWAVEds64":
-        raise WavError(f"Invalid heading for rf64 chunk: {heading!r}")
-
-    chunk_size = fid.read(4)
-
-    bw_size_low = fid.read(4)
-    bw_size_high = fid.read(4)
-
-    en: Endian = ">" if (bw_size_high > bw_size_low) else "<"
-    bytes_order: ByteOrd = "big" if en == ">" else "little"
-
-    data_size_low = fid.read(4)
-    data_size_high = fid.read(4)
-
-    file_size = int.from_bytes(bw_size_low + bw_size_high, "little")
-    data_size = int.from_bytes(data_size_low + data_size_high, "little")
-
-    int_chunk_size = int.from_bytes(chunk_size, bytes_order)
-    if type(int_chunk_size) is not int or int_chunk_size > 40:
-        raise WavError("Invalid chunk size in RF64 chunk")
-
-    fid.read(40 - int_chunk_size)
-
-    return data_size, file_size, en
-
-
-def _read_riff_chunk(sig: bytes, fid: Reader) -> tuple[None, int, Endian]:
-    en: Endian = "<" if sig == b"RIFF" else ">"
-    bytes_order: ByteOrd = "big" if en == ">" else "little"
-
-    file_size = int.from_bytes(fid.read(4), bytes_order) + 8
-
-    form = fid.read(4)
-    if form != b"WAVE":
-        raise WavError(f"Not a WAV file. RIFF form type is {form!r}.")
-
-    return None, file_size, en
-
-
-def _handle_pad_byte(fid: Reader, size: int) -> None:
-    if size % 2 == 1:
-        fid.seek(1, 1)
-
-
-def read(fid: Reader) -> tuple[int, AudioData]:
-    file_sig = fid.read(4)
-    if file_sig in {b"RIFF", b"RIFX"}:
-        data_size, file_size, en = _read_riff_chunk(file_sig, fid)
-    elif file_sig == b"RF64":
-        data_size, file_size, en = _read_rf64_chunk(fid)
-    else:
-        raise WavError(f"File format {file_sig!r} not supported.")
-
-    bytes_order: ByteOrd = "big" if en == ">" else "little"
-    fmt_chunk_received = False
-
-    while fid.tell() < file_size:
-        chunk_id = fid.read(4)
-
-        if not chunk_id:
-            raise WavError("Unexpected end of file.")
-        if len(chunk_id) < 4:
-            raise WavError(f"Incomplete chunk ID: {chunk_id!r}")
-
-        if chunk_id == b"fmt ":
-            fmt_chunk_received = True
-            format_tag, channels, sr, block_align, bit_depth = _read_fmt_chunk(
-                fid, bytes_order
-            )
-        elif chunk_id == b"data":
-            if not fmt_chunk_received:
-                raise WavError("No fmt chunk before data")
-
-            data = _read_data_chunk(
-                fid,
-                format_tag,
-                channels,
-                bit_depth,
-                en,
-                block_align,
-                data_size,
-            )
-
-            fid.seek(0)
-            return sr, data
-
-        elif chunk_id == b"\x00\x00\x00\x00":
-            raise WavError("Invalid chunk ID")
-        else:
-            _skip_unknown_chunk(fid, en)
-
-    raise WavError("Found no data")
-
-
-def write(fid: Writer, sr: int, arr: np.ndarray) -> None:
-    channels = 1 if arr.ndim == 1 else arr.shape[1]
-    bit_depth = arr.dtype.itemsize * 8
-    block_align = channels * (bit_depth // 8)
-    data_size = arr.nbytes
-    total_size = 44 + data_size  # Basic WAV header size + data size
-
-    if is_rf64 := total_size > 0xFFFFFFFF:
-        fid.write(b"RF64\xff\xff\xff\xffWAVE")
-        ds64_size = 28
-        ds64_chunk_data = (0).to_bytes(ds64_size, "little")  # placeholder values
-        fid.write(b"ds64" + struct.pack("<I", ds64_size) + ds64_chunk_data)
-    else:
-        fid.write(b"RIFF" + struct.pack("<I", total_size - 8) + b"WAVE")
-
-    dkind = arr.dtype.kind
-    format_tag = IEEE_FLOAT if dkind == "f" else PCM
-
-    fmt_chunk_data = struct.pack(
-        "<HHIIHH", format_tag, channels, sr, 0, block_align, bit_depth
-    )
-    fid.write(b"fmt " + struct.pack("<I", len(fmt_chunk_data)) + fmt_chunk_data)
-
-    # Data chunk
-    fid.write(b"data")
-    fid.write(struct.pack("<I", 0xFFFFFFFF if is_rf64 else data_size))
-
-    if arr.dtype.byteorder == ">" or (
-        arr.dtype.byteorder == "=" and sys.byteorder == "big"
-    ):
-        arr = arr.byteswap()
-    fid.write(arr.ravel().view("b").data)
-
-    if is_rf64:
-        end_position = fid.tell()
-        fid.seek(16)  # Position at the start of 'ds64' chunk size
-
-        file_size = end_position - 20
-        fid.write(struct.pack("<I", ds64_size))
-        fid.write(file_size.to_bytes(8, "little") + data_size.to_bytes(8, "little"))
-
-        fid.seek(end_position)
-
-
-def main() -> None:
-    data = np.random.rand(48000, 2)
-    with open("test.wav", "wb") as file:
-        write(file, 48_000, data)
-
-    with open("test.wav", "rb") as file:
-        read_sr, read_data = read(file)
-
-    assert read_sr == 48_000
-    assert np.array_equal(data, read_data)
-    print("success")
-
-
-if __name__ == "__main__":
-    main()