From f8cdf675608e18762363bb0aa2a38496be9b12bc Mon Sep 17 00:00:00 2001
From: ange <ange@yw5n.com>
Date: Mon, 1 Jul 2024 18:45:15 +0200
Subject: [PATCH] add amr compression

---
 .gitignore          |  1 +
 compression/main.py | 35 ++++++++++++++++++++++++-----------
 2 files changed, 25 insertions(+), 11 deletions(-)

diff --git a/.gitignore b/.gitignore
index bceb365..b40a346 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,6 +3,7 @@
 
 ### Audio ###
 *.aif
+*.amr
 *.flac
 *.iff
 *.m3u
diff --git a/compression/main.py b/compression/main.py
index 66c95c5..e5b28ec 100755
--- a/compression/main.py
+++ b/compression/main.py
@@ -7,31 +7,44 @@ class FFmpeg:
     input: str
     output: str
     format: str
-    bitrate: str | None
-    freq: int | None
+    bitrate: int | None
+    fpass: tuple[int, int] | None
 
     def __init__(
         self,
         input="pipe:0",
         output="pipe:1",
-        format="mp3",
+        format="amr",
         bitrate=None,
-        freq=None,
+        fpass=None,
     ):
         self.input = input
         self.output = output
         self.format = format
         self.bitrate = bitrate
-        self.freq = freq
+        self.fpass = fpass
 
     def __get_cmd(self) -> list[str]:
-        cmd = ["ffmpeg", "-i", self.input, "-f", self.format, self.output]
+        cmd = [
+            "ffmpeg",
+            "-i",
+            self.input,
+            "-f",
+            self.format,
+            "-ac",
+            "1",
+            "-ar",
+            "8000",
+        ]
 
         if self.bitrate is not None:
-            cmd += ["-b:a", self.bitrate]
-        if self.freq is not None:
-            cmd += ["-ar", str(self.freq)]
-        return cmd
+            cmd += ["-b:a", str(self.bitrate)]
+        if self.fpass:
+            cmd += [
+                "-filter:a",
+                f"highpass=f={self.fpass[0]},lowpass=f={self.fpass[1]}",
+            ]
+        return cmd + [self.output]
 
     def run(self):
         cmd = self.__get_cmd()
@@ -40,7 +53,7 @@ class FFmpeg:
 
 
 def main():
-    ff = FFmpeg(bitrate="128k", freq=8000)
+    ff = FFmpeg(bitrate=12200, fpass=[200, 3400])
     ff.run()