mirror of
https://github.com/justUmen/Bjornulf_custom_nodes.git
synced 2026-03-21 20:52:11 -03:00
0.61
This commit is contained in:
justumen
302
ffmpeg_convert.py
Normal file
302
ffmpeg_convert.py
Normal file
@@ -0,0 +1,302 @@
|
||||
import subprocess
|
||||
from pathlib import Path
|
||||
import os
|
||||
import json
|
||||
|
||||
class ConvertVideo:
|
||||
@classmethod
|
||||
def INPUT_TYPES(cls):
|
||||
return {
|
||||
"required": {
|
||||
"video_path": ("STRING", {"forceInput": True}),
|
||||
"output_filename": ("STRING", {"default": "converted.mp4"}),
|
||||
"use_python_ffmpeg": ("BOOLEAN", {"default": False}),
|
||||
},
|
||||
"optional": {
|
||||
"FFMPEG_CONFIG_JSON": ("STRING", {"forceInput": True}),
|
||||
},
|
||||
}
|
||||
|
||||
RETURN_TYPES = ("STRING", "STRING",)
|
||||
RETURN_NAMES = ("video_path", "ffmpeg_command",)
|
||||
FUNCTION = "convert_video"
|
||||
OUTPUT_NODE = True
|
||||
CATEGORY = "Bjornulf"
|
||||
|
||||
def __init__(self):
|
||||
self.output_dir = Path(os.path.abspath("ffmpeg/converted_videos"))
|
||||
os.makedirs(self.output_dir, exist_ok=True)
|
||||
|
||||
def get_default_config(self):
|
||||
"""Provide basic default configuration."""
|
||||
return {
|
||||
'ffmpeg_path': 'ffmpeg', # Assuming ffmpeg is in PATH
|
||||
'video_codec': 'copy',
|
||||
'video_bitrate': '3045K',
|
||||
'preset': 'medium',
|
||||
'pixel_format': 'yuv420p',
|
||||
'container_format': 'mp4',
|
||||
'crf': 19,
|
||||
'force_fps': 30,
|
||||
'width': None,
|
||||
'height': None,
|
||||
'ignore_audio': False,
|
||||
'audio_codec': 'aac',
|
||||
'audio_bitrate': '128k'
|
||||
}
|
||||
|
||||
def parse_config_json(self, config_json: str) -> dict:
|
||||
"""Parse the JSON configuration string into a dictionary format compatible with the converter"""
|
||||
config = json.loads(config_json)
|
||||
|
||||
return {
|
||||
# 'use_python_ffmpeg': config['ffmpeg']['use_python_ffmpeg'],
|
||||
'ffmpeg_path': config['ffmpeg']['path'],
|
||||
'video_codec': None if config['video']['codec'] == 'None' else config['video']['codec'],
|
||||
'video_bitrate': config['video']['bitrate'],
|
||||
'preset': None if config['video']['preset'] == 'None' else config['video']['preset'],
|
||||
'pixel_format': None if config['video']['pixel_format'] == 'None' else config['video']['pixel_format'],
|
||||
'container_format': None if config['output']['container_format'] == 'None' else config['output']['container_format'],
|
||||
'crf': config['video']['crf'],
|
||||
'force_fps': config['video']['fps']['force_fps'],
|
||||
'width': config['video']['resolution']['width'],
|
||||
'height': config['video']['resolution']['height'],
|
||||
'ignore_audio': not config['audio']['enabled'],
|
||||
'audio_codec': None if config['audio']['codec'] == 'None' else config['audio']['codec'],
|
||||
'audio_bitrate': config['audio']['bitrate']
|
||||
}
|
||||
|
||||
def convert_video_subprocess(self, input_path, output_path, FFMPEG_CONFIG_JSON):
|
||||
"""Use subprocess to run ffmpeg command"""
|
||||
cmd = [
|
||||
FFMPEG_CONFIG_JSON['ffmpeg_path'], '-y',
|
||||
'-i', str(input_path)
|
||||
]
|
||||
|
||||
# Add video codec settings if not None
|
||||
if FFMPEG_CONFIG_JSON['video_codec'] is not None:
|
||||
if FFMPEG_CONFIG_JSON['video_codec'] == 'copy':
|
||||
cmd.extend(['-c:v', 'copy'])
|
||||
else:
|
||||
cmd.extend(['-c:v', FFMPEG_CONFIG_JSON['video_codec']])
|
||||
|
||||
# Add preset if specified
|
||||
if FFMPEG_CONFIG_JSON['preset'] is not None:
|
||||
cmd.extend(['-preset', FFMPEG_CONFIG_JSON['preset']])
|
||||
|
||||
# Add width and height if specified
|
||||
if FFMPEG_CONFIG_JSON['width'] and FFMPEG_CONFIG_JSON['height']:
|
||||
cmd.extend(['-vf', f'scale={FFMPEG_CONFIG_JSON["width"]}:{FFMPEG_CONFIG_JSON["height"]}'])
|
||||
|
||||
# Add video bitrate if specified
|
||||
if FFMPEG_CONFIG_JSON['video_bitrate']:
|
||||
cmd.extend(['-b:v', FFMPEG_CONFIG_JSON['video_bitrate']])
|
||||
|
||||
# Add CRF if video codec isn't copy
|
||||
cmd.extend(['-crf', str(FFMPEG_CONFIG_JSON['crf'])])
|
||||
|
||||
# Add pixel format if specified
|
||||
if FFMPEG_CONFIG_JSON['pixel_format'] is not None:
|
||||
cmd.extend(['-pix_fmt', FFMPEG_CONFIG_JSON['pixel_format']])
|
||||
|
||||
# Add force fps if enabled
|
||||
if FFMPEG_CONFIG_JSON['force_fps'] > 0:
|
||||
cmd.extend(['-r', str(FFMPEG_CONFIG_JSON['force_fps'])])
|
||||
|
||||
# Add audio codec settings
|
||||
if FFMPEG_CONFIG_JSON['ignore_audio'] or FFMPEG_CONFIG_JSON['audio_codec'] is None:
|
||||
cmd.extend(['-an'])
|
||||
elif FFMPEG_CONFIG_JSON['audio_codec'] == 'copy':
|
||||
cmd.extend(['-c:a', 'copy'])
|
||||
else:
|
||||
cmd.extend([
|
||||
'-c:a', FFMPEG_CONFIG_JSON['audio_codec'],
|
||||
'-b:a', FFMPEG_CONFIG_JSON['audio_bitrate']
|
||||
])
|
||||
|
||||
# Add output path
|
||||
cmd.append(str(output_path))
|
||||
|
||||
process = subprocess.run(
|
||||
cmd,
|
||||
check=True,
|
||||
capture_output=True,
|
||||
text=True
|
||||
)
|
||||
|
||||
def convert_video_python_ffmpeg(self, input_path, output_path, FFMPEG_CONFIG_JSON):
|
||||
"""Use ffmpeg-python library"""
|
||||
try:
|
||||
import ffmpeg
|
||||
except ImportError:
|
||||
raise ImportError("ffmpeg-python is not installed. Please install it with: pip install ffmpeg-python")
|
||||
|
||||
# Start building the ffmpeg-python chain
|
||||
stream = ffmpeg.input(str(input_path))
|
||||
|
||||
# Build stream arguments based on config
|
||||
stream_args = {}
|
||||
|
||||
# Video settings if not None
|
||||
if FFMPEG_CONFIG_JSON['video_codec'] is not None:
|
||||
if FFMPEG_CONFIG_JSON['video_codec'] != 'copy':
|
||||
stream_args['vcodec'] = FFMPEG_CONFIG_JSON['video_codec']
|
||||
|
||||
if FFMPEG_CONFIG_JSON['preset'] is not None:
|
||||
stream_args['preset'] = FFMPEG_CONFIG_JSON['preset']
|
||||
|
||||
# Add width and height if specified
|
||||
if FFMPEG_CONFIG_JSON['width'] and FFMPEG_CONFIG_JSON['height']:
|
||||
stream = ffmpeg.filter(stream, 'scale',
|
||||
w=FFMPEG_CONFIG_JSON['width'],
|
||||
h=FFMPEG_CONFIG_JSON['height'])
|
||||
|
||||
if FFMPEG_CONFIG_JSON['video_bitrate']:
|
||||
stream_args['video_bitrate'] = FFMPEG_CONFIG_JSON['video_bitrate']
|
||||
|
||||
if FFMPEG_CONFIG_JSON['force_fps'] > 0:
|
||||
stream_args['crf'] = FFMPEG_CONFIG_JSON['crf']
|
||||
else:
|
||||
stream_args['crf'] = 19
|
||||
|
||||
if FFMPEG_CONFIG_JSON['pixel_format'] is not None:
|
||||
stream_args['pix_fmt'] = FFMPEG_CONFIG_JSON['pixel_format']
|
||||
|
||||
if FFMPEG_CONFIG_JSON['force_fps'] > 0:
|
||||
stream_args['r'] = FFMPEG_CONFIG_JSON['force_fps']
|
||||
else:
|
||||
stream_args['vcodec'] = 'copy'
|
||||
|
||||
# Audio settings
|
||||
if FFMPEG_CONFIG_JSON['ignore_audio'] or FFMPEG_CONFIG_JSON['audio_codec'] is None:
|
||||
stream_args['an'] = None
|
||||
elif FFMPEG_CONFIG_JSON['audio_codec'] == 'copy':
|
||||
stream_args['acodec'] = 'copy'
|
||||
else:
|
||||
stream_args.update({
|
||||
'acodec': FFMPEG_CONFIG_JSON['audio_codec'],
|
||||
'audio_bitrate': FFMPEG_CONFIG_JSON['audio_bitrate']
|
||||
})
|
||||
|
||||
# Run the ffmpeg operation
|
||||
stream = ffmpeg.output(stream, str(output_path), **stream_args, y=None)
|
||||
stream.run()
|
||||
|
||||
def convert_video(self, video_path: str, output_filename: str, FFMPEG_CONFIG_JSON: str = None, use_python_ffmpeg: bool = False):
|
||||
"""
|
||||
Convert a video using either subprocess or python-ffmpeg based on config.
|
||||
If no configuration is provided, uses default configuration.
|
||||
"""
|
||||
# Use default configuration if no JSON is provided
|
||||
if FFMPEG_CONFIG_JSON is None:
|
||||
default_config = self.get_default_config()
|
||||
# Create a JSON-like structure to match the parse_config_json method's expectations
|
||||
FFMPEG_CONFIG_JSON = {
|
||||
'ffmpeg': {
|
||||
'path': default_config['ffmpeg_path']
|
||||
},
|
||||
'video': {
|
||||
'codec': default_config['video_codec'],
|
||||
'bitrate': default_config['video_bitrate'],
|
||||
'preset': default_config['preset'],
|
||||
'pixel_format': default_config['pixel_format'],
|
||||
'crf': default_config['crf'],
|
||||
'fps': {
|
||||
'force_fps': default_config['force_fps']
|
||||
},
|
||||
'resolution': {
|
||||
'width': default_config['width'],
|
||||
'height': default_config['height']
|
||||
}
|
||||
},
|
||||
'output': {
|
||||
'container_format': default_config['container_format']
|
||||
},
|
||||
'audio': {
|
||||
'enabled': not default_config['ignore_audio'],
|
||||
'codec': default_config['audio_codec'],
|
||||
'bitrate': default_config['audio_bitrate']
|
||||
}
|
||||
}
|
||||
# Convert to JSON string
|
||||
FFMPEG_CONFIG_JSON = json.dumps(FFMPEG_CONFIG_JSON)
|
||||
|
||||
# Parse the JSON configuration
|
||||
FFMPEG_CONFIG_JSON = self.parse_config_json(FFMPEG_CONFIG_JSON)
|
||||
|
||||
# Validate input path
|
||||
input_path = Path(os.path.abspath(video_path))
|
||||
if not input_path.exists():
|
||||
raise ValueError(f"Input video path does not exist: {input_path}")
|
||||
|
||||
# Set output path
|
||||
if FFMPEG_CONFIG_JSON['container_format']:
|
||||
output_filename = Path(output_filename).with_suffix(f".{FFMPEG_CONFIG_JSON['container_format']}")
|
||||
output_path = self.output_dir / output_filename
|
||||
output_path = output_path.absolute()
|
||||
|
||||
# Construct FFmpeg command for command string return
|
||||
cmd = [
|
||||
FFMPEG_CONFIG_JSON['ffmpeg_path'], '-y',
|
||||
'-i', str(input_path)
|
||||
]
|
||||
|
||||
# Add video codec settings if not None
|
||||
if FFMPEG_CONFIG_JSON['video_codec'] is not None:
|
||||
if FFMPEG_CONFIG_JSON['video_codec'] == 'copy':
|
||||
cmd.extend(['-c:v', 'copy'])
|
||||
else:
|
||||
cmd.extend(['-c:v', FFMPEG_CONFIG_JSON['video_codec']])
|
||||
|
||||
if FFMPEG_CONFIG_JSON['preset'] is not None:
|
||||
cmd.extend(['-preset', FFMPEG_CONFIG_JSON['preset']])
|
||||
|
||||
if FFMPEG_CONFIG_JSON['width'] and FFMPEG_CONFIG_JSON['height']:
|
||||
cmd.extend(['-vf', f'scale={FFMPEG_CONFIG_JSON["width"]}:{FFMPEG_CONFIG_JSON["height"]}'])
|
||||
|
||||
if FFMPEG_CONFIG_JSON['video_bitrate']:
|
||||
cmd.extend(['-b:v', FFMPEG_CONFIG_JSON['video_bitrate']])
|
||||
|
||||
if FFMPEG_CONFIG_JSON['crf'] > 0:
|
||||
cmd.extend(['-crf', str(FFMPEG_CONFIG_JSON['crf'])])
|
||||
else:
|
||||
cmd.extend(['-crf', '19'])
|
||||
|
||||
if FFMPEG_CONFIG_JSON['pixel_format'] is not None:
|
||||
cmd.extend(['-pix_fmt', FFMPEG_CONFIG_JSON['pixel_format']])
|
||||
|
||||
if FFMPEG_CONFIG_JSON['force_fps'] > 0:
|
||||
cmd.extend(['-r', str(FFMPEG_CONFIG_JSON['force_fps'])])
|
||||
|
||||
# Add audio codec settings
|
||||
if FFMPEG_CONFIG_JSON['ignore_audio'] or FFMPEG_CONFIG_JSON['audio_codec'] is None:
|
||||
cmd.extend(['-an'])
|
||||
elif FFMPEG_CONFIG_JSON['audio_codec'] == 'copy':
|
||||
cmd.extend(['-c:a', 'copy'])
|
||||
else:
|
||||
cmd.extend([
|
||||
'-c:a', FFMPEG_CONFIG_JSON['audio_codec'],
|
||||
'-b:a', FFMPEG_CONFIG_JSON['audio_bitrate']
|
||||
])
|
||||
|
||||
cmd.append(str(output_path))
|
||||
|
||||
# Convert command list to string
|
||||
ffmpeg_command = ' '.join(cmd)
|
||||
|
||||
try:
|
||||
if use_python_ffmpeg:
|
||||
self.convert_video_python_ffmpeg(input_path, output_path, FFMPEG_CONFIG_JSON)
|
||||
else:
|
||||
self.convert_video_subprocess(input_path, output_path, FFMPEG_CONFIG_JSON)
|
||||
|
||||
return (str(output_path), ffmpeg_command)
|
||||
|
||||
except subprocess.CalledProcessError as e:
|
||||
raise RuntimeError(f"FFmpeg error: {e.stderr}")
|
||||
except Exception as e:
|
||||
raise RuntimeError(f"Error during video conversion: {str(e)}")
|
||||
|
||||
@classmethod
|
||||
def IS_CHANGED(cls, **kwargs):
|
||||
return float("NaN")
|
||||
Reference in New Issue
Block a user