Class LTXVSparseTrackEditor

sparse_tracks.py:65–139

Interactive spline editor for drawing sparse motion tracks. Provides a canvas widget where users can draw and edit spline control points on top of a reference image. Outputs interpolated track coordinates compatible with LTXVDrawTracks.

@comfy_node(name="LTXVSparseTrackEditor", description="LTX Sparse Track Editor")
class LTXVSparseTrackEditor(io.ComfyNode):
    """Interactive spline editor for drawing sparse motion tracks.

    Provides a canvas widget where users can draw and edit spline control
    points on top of a reference image. Outputs interpolated track
    coordinates compatible with LTXVDrawTracks.
    """

    @classmethod
    def define_schema(cls):
        return io.Schema(
            node_id="LTXVSparseTrackEditor",
            category="Lightricks/motion_tracking",
            description=(
                "Interactive spline editor for drawing sparse motion tracks "
                "on a reference image."
            ),
            inputs=[
                io.Image.Input(
                    "image",
                    tooltip="Reference image displayed as the editor canvas background.",
                ),
                io.String.Input(
                    "points_store",
                    default="[]",
                    tooltip="JSON array of spline control points managed by the editor widget.",
                ),
                io.String.Input(
                    "coordinates",
                    default="[]",
                    tooltip="JSON array of interpolated track coordinates produced by the editor.",
                ),
                io.Int.Input(
                    "points_to_sample",
                    default=121,
                    min=2,
                    max=10000,
                    tooltip="Number of points sampled along each spline curve.",
                ),
            ],
            outputs=[
                io.String.Output("tracks"),
            ],
            is_output_node=True,
        )

    @classmethod
    def execute(
        cls,
        image,
        points_store: str,
        coordinates: str,
        points_to_sample: int,
    ) -> io.NodeOutput:
        # Re-interpolate from control points so that changes to
        # points_to_sample are always respected, regardless of JS sync.
        try:
            splines = json.loads(points_store) if points_store else []