Scripts/hq2cd-py/hq2cd.py

79 lines
2.1 KiB
Python

# Recursivly scan from the currenct directory and add every flac file to an array
import os
import json
import subprocess
import ffmpeg
# TODO: Rewrite using ffmpeg-python probe functionality
def _ffprobe(file_path):
command = (
'ffprobe -v quiet -print_format json -show_streams -select_streams a "'f'{file_path}"'
)
result = subprocess.run(command, capture_output=True)
if result.returncode == 0:
return json.loads(result.stdout)
raise RuntimeError(
f'FFProbe failed for {file_path}, output: {result.stderr}'
)
def scan_dir():
flac_files = []
for root, dirs, files in os.walk(os.getcwd()):
for file in files:
if file.endswith(".flac"):
flac_files.append(os.path.join(root, file))
return flac_files
# Filter the array of flac files to only include files with a bit depth higher than 16, and a sample rate higher than 44.1kHz
def filter_files(flac_files):
filtered_files = []
counter = 0
max_counter = len(flac_files)
for file in flac_files:
counter += 1
print("Progress: " + str(counter) + "/" + str(max_counter))
metadata = _ffprobe(file)
if int(metadata['streams'][0]['bits_per_raw_sample']) > 16 and int(
metadata['streams'][0]['sample_rate']) > 44100:
filtered_files.append(file)
return filtered_files
# Print the filtered files to the console
def print_files(filtered_files):
for file in filtered_files:
print(file)
def get_relative_path(file):
return os.path.relpath(file, os.getcwd())
def get_root_path(file):
return os.path.dirname(file)
def convert_files(filtered_files):
for file in filtered_files:
output = ".\\Output\\" + get_relative_path(file)
output_no_basename = os.path.dirname(output)
os.makedirs(output_no_basename, exist_ok=True)
stream = ffmpeg.input(file)
stream = ffmpeg.output(stream, output, sample_fmt="s16", ar='44100')
stream = ffmpeg.overwrite_output(stream)
ffmpeg.run(stream)
# Run the functions
flac_files = scan_dir()
print_files(flac_files)
filtered_files = filter_files(flac_files)
print_files(filtered_files)
convert_files(filtered_files)