guillevent / video.machine Goto Github PK

def make_brigther(IN, OUT): # https://video.stackexchange.com/questions/20962/ffmpeg-color-correction-gamma-brightness-and-saturation
	subprocess.check_output(['ffmpeg','-i', IN, '-vf',"eq=gamma=1.5:saturation=1.0",'-c:a','copy',OUT])

# Use VideoToolbox for hardware acceleration on macOS and iOS
    if options['codec'] == "h264":
        cmd.extend(["-c:v", "h264_videotoolbox", "-global_quality", str(your_quality_value)])
    else:
        cmd.extend(["-c:v", "hevc_videotoolbox", "-global_quality", str(your_quality_value)])

parser.add_argument("--audio_codec", choices=["aac", "mp3", "opus"], default="aac", help="Audio codec to use.")
parser.add_argument("--channels", choices=["1", "2"], default="2", help="Number of audio channels.")
parser.add_argument("--audio_only", action="store_true", help="Extract audio track without converting video.")

import os
import argparse
import subprocess
import sys
import json
import logging


class VideoConverter:
    VALID_VIDEO_EXTENSIONS = ['mp4', 'mkv', 'avi', 'mov']
    BITRATES = {
        'low': {'360': '700k', '480': '1200k', '720': '2500k', '1080': '5000k', '2160': '20000k'},
        'mid': {'360': '1000k', '480': '1500k', '720': '4000k', '1080': '8000k', '2160': '35000k'},
        'high': {'360': '1500k', '480': '1800k', '720': '6000k', '1080': '12000k', '2160': '50000k'}
    }
    
    def __init__(self, input_path, options):
        self.input_path = input_path
        self.options = options
        logging.basicConfig(level=logging.INFO)

    def get_video_info(self, input_file):
        #... (same as before)

    def should_process_video(self, input_file):
        #... (same as before)

    def prepare_strings_outfiles(self, in_file):
        #... (same as before, but use self.options)

    def run_ffmpeg_conversion(self, input_file, output_file):
        #... (same as before, but use self.options)

    def run_file(self, in_file):
        #... (same as before, but use self.options)

    def check_command_availability(self, command):
        #... (same as before)

    @classmethod
    def cli(cls):
        parser = argparse.ArgumentParser(description="Video Conversion Tool")
        #... parser arguments as before
        args = parser.parse_args()
        options = {
            'resolution': args.resolution,
            'bitrate': args.bitrate,
            #... other options
        }
        converter = cls(args.input_path, options)
        
        if args.cut and len(args.cut) > 1 and args.length:
            raise ValueError("Error: When providing a list for --cut, you cannot also provide --length. Length is to be used with a start-timestamp in --cut without end-stimestamp.")
        
        converter.check_command_availability("ffmpeg")
        converter.check_command_availability("ffprobe")

        if os.path.isfile(converter.input_path):
            converter.run_file(converter.input_path)
        elif os.path.isdir(converter.input_path):
            # If the remove option is set, ask for confirmation
            if args.remove:
                confirm = input(f"You are about to remove all files in the directory '{converter.input_path}' and its subdirectories. Are you sure? (then type yes or y): ")
                if confirm.lower() not in ['yes','y']:
                    sys.exit("Operation aborted.")
            for dirpath, dirnames, filenames in os.walk(converter.input_path):
                for file in filenames:
                    full_path = os.path.join(dirpath, file)
                    converter.run_file(full_path)
        else:
            print("Invalid path. Please provide a valid video file or directory.")

if __name__ == "__main__":
    VideoConverter.cli()