Dreamspire's picture
custom_nodes
f2dbf59
class VideoInfoNode:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"video_info": ("VHS_VIDEOINFO",),
}
}
CATEGORY = "tbox/Video"
RETURN_TYPES = ("FLOAT", "INT", "FLOAT", "INT", "INT", "FLOAT","INT", "FLOAT", "INT", "INT")
RETURN_NAMES = (
"source_fps",
"source_frame_count",
"source_duration",
"source_width",
"source_height",
"loaded_fps",
"loaded_frame_count",
"loaded_duration",
"loaded_width",
"loaded_height",
)
FUNCTION = "get_video_info"
def get_video_info(self, video_info):
keys = ["fps", "frame_count", "duration", "width", "height"]
source_info = []
loaded_info = []
for key in keys:
source_info.append(video_info[f"source_{key}"])
loaded_info.append(video_info[f"loaded_{key}"])
return (*source_info, *loaded_info)