main.py

@@ -1,46 +1,41 @@
 #!/usr/bin/env python3
 
 # ffmpeg wrapper
-import multiprocessing
-from os.path import isdir, isfile
+from os import sched_get_priority_max
 import ffmpy
 
 # argument parsing
 import argparse
 
-# multiprocessing stuff
-from multiprocessing import Pool
-from multiprocessing import cpu_count
-
 # executing some commands
 import subprocess
 
 # parsing json output of loudnorm
 import json
 
-# file/directory handling
-import os
-
-# most recent starttime for program
-import time
-
-from random import randint
-
 """
+parser = argparse.ArgumentParser(description="")
 
+# Input file
+parser.add_argument("-i", "--input-file", required=True, type=str, help="Input file")
+
+args = parser.parse_args()
+
+inputfile = args.input_file
+"""
+"""
+ffmpeg -y -i FalKKonE\ -\ 01\ Aria\ \(From\ \"Berserk:\ The\ Golden\ Age\ Arc\"\).flac -pass 1 -filter:a loudnorm=print_format=json -f flac /dev/null
+
+ffmpeg -i FalKKonE\ -\ 01\ Aria\ \(From\ \"Berserk:\ The\ Golden\ Age\ Arc\"\).flac -pass 2 -filter:a loudnorm=I=-30.0:linear=true:measured_I=-4.52:measured_LRA=1.90:measured_thresh=-14.64 -c:a libopus -b:a 320k test441.opus
+ffmpeg -i $FILE -c:v libx264 -b:v 4000k -pass 2 -filter:a loudnorm=linear=true:measured_I=$input_i:measured_LRA=$input_lra:measured_tp=$input_tp:measured_thresh=$input_thresh -c:a aac -b:a 256k $FILE.mkv
 """
 
 # FIXME
-# inputfile = (
-#    '/home/marc/Downloads/FalKKonE - 01 Aria (From "Berserk: The Golden Age Arc").flac'
-# )
+inputfile = (
+    '/home/marc/Downloads/FalKKonE - 01 Aria (From "Berserk: The Golden Age Arc").flac'
+)
 # inputfile = "/home/marc/Downloads/test441.opus"
-# outputfile = "/home/marc/Downloads/test441_out.opus"
-
-# srcfolder = "/home/marc/Downloads/MusikRaw"
-# destfolder = "/home/marc/Downloads/Musik"
-
-musicfile_extensions = (".flac", ".wav", ".mp3", ".m4a", ".aac", ".opus")
+outputfile = "/home/marc/Downloads/test441_out.opus"
 
 
 def get_format(inputfile) -> str:
@@ -50,7 +45,8 @@ def get_format(inputfile) -> str:
     ff = ffmpy.FFprobe(
         inputs={inputfile: None},
         global_options=(
-            "-v quiet",
+            "-v",
+            "quiet",
             "-select_streams a",
             "-show_entries stream=codec_name",
             "-of default=noprint_wrappers=1:nokey=1",
@@ -70,25 +66,6 @@ def get_format(inputfile) -> str:
     return format
 
 
-def remove_picture(inputfile):
-    """
-    This function makes sure no image is attached to the audio stream.
-    An image might cause problems for the later conversion to opus.
-
-    Parameters:
-        inputfile (str): Path to file
-    """
-    tmpfile = os.path.splitext(inputfile)[0] + ".tmp" + os.path.splitext(inputfile)[1]
-    ff = ffmpy.FFmpeg(
-        inputs={inputfile: None},
-        outputs={tmpfile: "-vn -c:a copy"},
-        global_options=("-v error"),
-    )
-    ff.run()
-    os.remove(inputfile)
-    os.rename(tmpfile, inputfile)
-
-
 def loudness_info(inputfile) -> dict[str, str]:
     # get format from file
     # inputformat = get_format(inputfile)
@@ -114,8 +91,6 @@ def loudness_info(inputfile) -> dict[str, str]:
     # decode json to dict
     loudness: dict[str, str] = json.loads(loudness_json)
     # print(loudness_json)
-    # print(ff.cmd)
-    print("Measuring loudness of ", os.path.basename(inputfile))
     return loudness
 
 
@@ -142,113 +117,11 @@ def convert(inputfile, outputfile, loudness):
                 input_thresh=loudness["input_thresh"],
             )
         },
-        global_options=("-y", "-v error"),
     )
-    # print(ff.cmd)
-    print("Working on ", os.path.basename(inputfile))
+    print(ff.cmd)
     ff.run()
 
 
-def main(inputfile: str):
-    """
-    Main program loop
-
-    Parameters:
-        inputfile (str): Path to input file
-    """
-    # set output folder to parent path + "normalized"
-    outputfolder = os.path.join(os.path.dirname(inputfile), "normalized")
-    # NOTE create output folder
-    # because multiple parallel processes are at work here,
-    # there might be conflicts with one trying to create the directory although it already exists
-    # this while loop makes sure the directory does exist
-    # the try/except block ensures the error is caught and (hopefully) doesn't happen again just after with random sleep
-    # there's very likely a better way to do this, idk
-    while not os.path.isdir(outputfolder):
-        try:
-            os.mkdir(outputfolder)
-        except:
-            time.sleep(randint(0, 4))
-
-    # output file path
-    noext_infile: str = os.path.splitext(os.path.basename(inputfile))[0]
-    outputfile: str = os.path.join(outputfolder, noext_infile + ".opus")
-
-    # print(inputfile)
-    # print(os.path.dirname(inputfile))
-    # print(os.path.basename(inputfile))
-    # print(outputfile)
-
-    remove_picture(inputfile=inputfile)
+if __name__ == "__main__":
     loudness = loudness_info(inputfile=inputfile)
     convert(inputfile=inputfile, outputfile=outputfile, loudness=loudness)
-
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="")
-
-    # Input directory
-    parser.add_argument(
-        "-i", "--input-dir", required=True, type=str, help="Input source directory"
-    )
-
-    # number of cpus/threads to use, defaults to all available
-    parser.add_argument(
-        "-c",
-        "--cpu-count",
-        required=False,
-        type=int,
-        help="Number of cpu cores",
-        default=multiprocessing.cpu_count(),
-    )
-
-    # in case you wanted to rerun the conversion for everything
-    parser.add_argument(
-        "-r",
-        "--reset",
-        required=False,
-        action="store_true",
-        help="Rerun conversion for all files",
-    )
-
-    args = parser.parse_args()
-
-    srcfolder = args.input_dir
-
-    cpu = args.cpu_count
-
-    reset = args.reset
-
-    # file where last run timestamp is stored
-    timefile = os.path.join(srcfolder, "run.time")
-
-    # get time of previous run
-    if reset:
-        timeprev = 0
-    elif os.path.isfile(timefile):
-        with open(timefile, "r") as file:
-            timeprev = file.read()
-    else:
-        timeprev = 0
-
-    # print(timeprev)
-
-    musicfiles = []
-    for root, dirs, files in os.walk(srcfolder):
-        # ignore the "normalized" subfolder
-        dirs[:] = [d for d in dirs if d not in ["normalized"]]
-        for file in files:
-            if file.endswith(musicfile_extensions):
-                filepath = os.path.join(root, file)
-                # only file newer than the last run are added
-                if os.path.getmtime(filepath) >= float(timeprev):
-                    musicfiles.append(os.path.join(root, file))
-
-    # print(musicfiles)
-
-    with Pool(cpu) as p:
-        p.map(main, musicfiles)
-
-    # write this run's time into file
-    with open(timefile, "w") as file:
-        file.write(str(time.time()))