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
171
ffmpeg_configuration.py
Normal file
171
ffmpeg_configuration.py
Normal file
@@ -0,0 +1,171 @@
|
||||
import json
|
||||
import subprocess
|
||||
import ffmpeg # Assuming the Python FFmpeg bindings (ffmpeg-python) are installed
|
||||
|
||||
class FFmpegConfig:
|
||||
@classmethod
|
||||
def INPUT_TYPES(cls):
|
||||
return {
|
||||
"required": {
|
||||
"use_python_ffmpeg": ("BOOLEAN", {"default": False}),
|
||||
"ffmpeg_path": ("STRING", {"default": "ffmpeg"}),
|
||||
"video_codec": ([
|
||||
"None",
|
||||
"copy",
|
||||
"libx264 (H.264)",
|
||||
"h264_nvenc (H.264 / NVIDIA GPU)",
|
||||
"libx265 (H.265)",
|
||||
"hevc_nvenc (H.265 / NVIDIA GPU)",
|
||||
"libvpx-vp9 (WebM)",
|
||||
"libaom-av1"
|
||||
], {"default": "None"}),
|
||||
|
||||
"video_bitrate": ("STRING", {"default": "3045k"}),
|
||||
|
||||
"preset": ([
|
||||
"None",
|
||||
"ultrafast",
|
||||
"superfast",
|
||||
"veryfast",
|
||||
"faster",
|
||||
"fast",
|
||||
"medium",
|
||||
"slow",
|
||||
"slower",
|
||||
"veryslow"
|
||||
], {"default": "medium"}),
|
||||
|
||||
"pixel_format": ([
|
||||
"None",
|
||||
"yuv420p",
|
||||
"yuv444p",
|
||||
"yuv420p10le",
|
||||
"yuv444p10le",
|
||||
"rgb24",
|
||||
"rgba",
|
||||
"yuva420p"
|
||||
], {"default": "yuv420p"}),
|
||||
|
||||
"container_format": ([
|
||||
"None",
|
||||
"mp4",
|
||||
"mkv",
|
||||
"webm",
|
||||
"mov",
|
||||
"avi"
|
||||
], {"default": "mp4"}),
|
||||
|
||||
"crf": ("INT", {"default": 19, "min": 1, "max": 63}),
|
||||
|
||||
"force_fps": ("FLOAT", {
|
||||
"default": 0.0,
|
||||
"min": 0.0,
|
||||
"max": 240.0,
|
||||
"step": 0.01,
|
||||
"description": "Force output FPS (0 = use source FPS)"
|
||||
}),
|
||||
|
||||
"width": ("INT", {"default": 1152, "min": 1, "max": 10000}),
|
||||
"height": ("INT", {"default": 768, "min": 1, "max": 10000}),
|
||||
|
||||
"ignore_audio": ("BOOLEAN", {"default": False}),
|
||||
"audio_codec": ([
|
||||
"None",
|
||||
"copy",
|
||||
"aac",
|
||||
"libmp3lame",
|
||||
"libvorbis",
|
||||
"libopus",
|
||||
"none"
|
||||
], {"default": "aac"}),
|
||||
"audio_bitrate": ("STRING", {"default": "192k"}),
|
||||
}
|
||||
}
|
||||
|
||||
RETURN_TYPES = ("STRING",)
|
||||
RETURN_NAMES = ("FFMPEG_CONFIG_JSON",)
|
||||
FUNCTION = "create_config"
|
||||
CATEGORY = "Bjornulf"
|
||||
|
||||
def get_ffmpeg_version(self, ffmpeg_path, use_python_ffmpeg):
|
||||
if use_python_ffmpeg:
|
||||
try:
|
||||
# Retrieve Python ffmpeg-python version
|
||||
return f"Python FFmpeg binding (ffmpeg-python) version: {ffmpeg.__version__}"
|
||||
except AttributeError:
|
||||
return "Python FFmpeg binding (ffmpeg-python) version: Unknown (no __version__ attribute)"
|
||||
else:
|
||||
try:
|
||||
# Retrieve system FFmpeg version
|
||||
result = subprocess.run(
|
||||
[ffmpeg_path, "-version"],
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
text=True
|
||||
)
|
||||
version_line = result.stdout.splitlines()[0]
|
||||
return version_line
|
||||
except Exception as e:
|
||||
return f"Error fetching FFmpeg version: {e}"
|
||||
|
||||
def create_json_output(self, config, use_python_ffmpeg):
|
||||
"""Create a JSON string containing all FFmpeg configuration."""
|
||||
ffmpeg_version = self.get_ffmpeg_version(config["ffmpeg_path"], use_python_ffmpeg)
|
||||
config_info = {
|
||||
"ffmpeg": {
|
||||
"path": config["ffmpeg_path"],
|
||||
"version": ffmpeg_version
|
||||
},
|
||||
"video": {
|
||||
"codec": config["video_codec"] or "None",
|
||||
"bitrate": config["video_bitrate"],
|
||||
"preset": config["preset"] or "None",
|
||||
"pixel_format": config["pixel_format"] or "None",
|
||||
"crf": config["crf"],
|
||||
"resolution": {
|
||||
"width": config["width"],
|
||||
"height": config["height"]
|
||||
},
|
||||
"fps": {
|
||||
"force_fps": config["force_fps"],
|
||||
"enabled": config["force_fps"] > 0
|
||||
}
|
||||
},
|
||||
"audio": {
|
||||
"enabled": not config["ignore_audio"],
|
||||
"codec": config["audio_codec"] or "None",
|
||||
"bitrate": config["audio_bitrate"]
|
||||
},
|
||||
"output": {
|
||||
"container_format": config["container_format"] or "None"
|
||||
}
|
||||
}
|
||||
return json.dumps(config_info, indent=2)
|
||||
|
||||
def create_config(self, ffmpeg_path, use_python_ffmpeg, ignore_audio, video_codec, audio_codec,
|
||||
video_bitrate, audio_bitrate, preset, pixel_format,
|
||||
container_format, crf, force_fps, width, height):
|
||||
|
||||
config = {
|
||||
"ffmpeg_path": ffmpeg_path,
|
||||
"video_bitrate": video_bitrate,
|
||||
"preset": None if preset == "None" else preset,
|
||||
"crf": crf,
|
||||
"force_fps": force_fps,
|
||||
"ignore_audio": ignore_audio,
|
||||
"audio_bitrate": audio_bitrate,
|
||||
"width": width,
|
||||
"height": height,
|
||||
"video_codec": video_codec.split(" ")[0] if video_codec != "None" else None,
|
||||
"pixel_format": None if pixel_format == "None" else pixel_format,
|
||||
"container_format": None if container_format == "None" else container_format,
|
||||
"audio_codec": None if audio_codec == "None" or ignore_audio else audio_codec,
|
||||
}
|
||||
|
||||
return (self.create_json_output(config, use_python_ffmpeg),)
|
||||
|
||||
@classmethod
|
||||
def IS_CHANGED(cls, ffmpeg_path, use_python_ffmpeg, ignore_audio, video_codec, audio_codec,
|
||||
video_bitrate, audio_bitrate, preset, pixel_format,
|
||||
container_format, crf, force_fps, width, height) -> float:
|
||||
return 0.0
|
||||
Reference in New Issue
Block a user