Skip to content

y5gfunc.source.wobbly.processors.sections

sections

Section processor implementation.

Classes:

Name Description
SectionProcessor

Section processor implementation

Attributes:

Name Type Description
core

core module-attribute 

core = core

SectionProcessor 

SectionProcessor()

Bases: BaseProcessor

Section processor implementation

Initialize section processor

Methods:

Name Description
process

Process sections

Attributes:

Name Type Description
keys
Source code in y5gfunc/source/wobbly/processors/sections.py 
19
20
21
def __init__(self):
    """Initialize section processor"""
    self.keys = WobblyKeys()

keys instance-attribute 

keys = WobblyKeys()

process 

process(clip: VideoNode, project: ProjectData, frame_props: FramePropertyMap, frame_mapping: FrameMap, presets: PresetDict) -> tuple[VideoNode, FramePropertyMap, FrameMap]

Process sections

Parameters:

Name Type Description Default

clip

VideoNode

Input video clip

 required

project

ProjectData

Project data

 required

frame_props

FramePropertyMap

Frame property mapping

 required

frame_mapping

FrameMap

Frame number mapping

 required

presets

PresetDict

Preset function dictionary

 required

Returns:

Type Description
tuple[VideoNode, FramePropertyMap, FrameMap]

Processed clip, updated frame properties and frame mapping
Source code in y5gfunc/source/wobbly/processors/sections.py 
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
def process(
    self,
    clip: vs.VideoNode,
    project: ProjectData,
    frame_props: FramePropertyMap,
    frame_mapping: FrameMap,
    presets: PresetDict,
) -> tuple[vs.VideoNode, FramePropertyMap, FrameMap]:
    """
    Process sections

    Args:
        clip: Input video clip
        project: Project data
        frame_props: Frame property mapping
        frame_mapping: Frame number mapping
        presets: Preset function dictionary

    Returns:
        Processed clip, updated frame properties and frame mapping
    """
    with safe_processing("sections_processing"):
        Keys = self.keys
        sections_list = project.get(Keys.project.sections, [])

        if not sections_list:
            return clip, frame_props, frame_mapping

        # Sort sections by start frame
        sorted_sections = sorted(
            sections_list, key=lambda s: s.get(Keys.sections.start, 0)
        )

        # Mark each frame with its section
        self._mark_section_frames(
            clip, sorted_sections, frame_props, frame_mapping, Keys
        )

        # Apply presets and splice
        sections = []
        new_frame_props: dict[int, dict[str, Any]] = {}
        new_frame_idx = 0

        for i, section_info in enumerate(sorted_sections):
            start = section_info.get(Keys.sections.start, 0)
            next_start = (
                sorted_sections[i + 1].get(Keys.sections.start, clip.num_frames)
                if i + 1 < len(sorted_sections)
                else clip.num_frames
            )

            # Apply presets
            section_clip = clip[start:next_start]
            for preset_name in section_info.get(Keys.sections.presets, []):
                if preset_name in presets:
                    section_clip = presets[preset_name](section_clip)

            # Update frame mapping and properties
            for j in range(section_clip.num_frames):
                src_idx = start + j
                if src_idx < len(frame_mapping):
                    orig_frame = frame_mapping[src_idx]
                    # Copy original frame properties
                    if src_idx in frame_props:
                        new_frame_props[new_frame_idx] = frame_props[src_idx].copy()  # type: ignore[index]
                        # Update mapping
                        frame_mapping[new_frame_idx] = orig_frame
                        new_frame_idx += 1

            sections.append(section_clip)

        # Merge all sections
        if sections:
            clip = core.std.Splice(clips=sections, mismatch=True)
            frame_props = new_frame_props  # type: ignore[assignment] # Update frame properties

    return clip, frame_props, frame_mapping