typed-ffmpeg-compatible 3.5.1__py3-none-any.whl → 3.6__py3-none-any.whl

typed_ffmpeg/compile/validate.py

@@ -21,7 +21,7 @@ from .context import DAGContext
 
 
 def remove_split(
-    current_stream: Stream, mapping: dict[Stream, Stream] = None
+    current_stream: Stream, mapping: dict[Stream, Stream] | None = None
 ) -> tuple[Stream, dict[Stream, Stream]]:
     """
     Remove all split nodes from the graph to prepare for reconstruction.
@@ -43,8 +43,8 @@ def remove_split(
         A tuple containing:
         - The new stream corresponding to the input stream but with splits removed
         - A mapping dictionary relating original streams to their new versions
-    """
 
+    """
     # remove all split nodes
     # add split nodes to the graph
     if mapping is None:
@@ -91,10 +91,10 @@ def remove_split(
 
 def add_split(
     current_stream: Stream,
-    down_node: Node = None,
-    down_index: int = None,
-    context: DAGContext = None,
-    mapping: dict[tuple[Stream, Node | None, int | None], Stream] = None,
+    down_node: Node | None = None,
+    down_index: int | None = None,
+    context: DAGContext | None = None,
+    mapping: dict[tuple[Stream, Node | None, int | None], Stream] | None = None,
 ) -> tuple[Stream, dict[tuple[Stream, Node | None, int | None], Stream]]:
     """
     Add split nodes to the graph where streams are reused.
@@ -122,8 +122,8 @@ def add_split(
 
     Raises:
         FFMpegValueError: If an unsupported stream type is encountered
-    """
 
+    """
     if not context:
         context = DAGContext.build(current_stream.node)
 
@@ -202,8 +202,8 @@ def fix_graph(stream: Stream) -> Stream:
     Note:
         This function creates a new graph structure rather than modifying the
         existing one, preserving the original graph.
-    """
 
+    """
     stream, _ = remove_split(stream)
     stream, _ = add_split(stream)
     return stream
@@ -240,6 +240,7 @@ def validate(stream: Stream, auto_fix: bool = True) -> Stream:
         # Validate will automatically insert a split filter
         valid_stream = ffmpeg.dag.validate(scaled.output("output.mp4"))
         ```
+
     """
     if auto_fix:
         stream = fix_graph(stream)

typed_ffmpeg/dag/factory.py

@@ -28,6 +28,7 @@ def eval_formula(formula: str, **kwargs: Any) -> list[StreamType]:
 
     Returns:
         The result of the formula evaluation
+
     """
     # Convert formula to Python code
     return eval(
@@ -57,6 +58,7 @@ def filter_node_factory(
     Note:
         This function is primarily used internally by the filter generation system
         to create filter nodes from the FFmpeg filter definitions.
+
     """
     for k, v in kwargs.items():
         if isinstance(v, Auto):

typed_ffmpeg/dag/global_runnable/__init__.py

@@ -0,0 +1 @@
+"""Global runnable utilities for FFmpeg DAG operations."""

typed_ffmpeg/dag/global_runnable/global_args.py

@@ -1,5 +1,5 @@
 # NOTE: this file is auto-generated, do not modify
-
+"""Global arguments."""
 
 from __future__ import annotations
 
@@ -136,8 +136,8 @@ class GlobalArgs(ABC):
 
         Returns:
             GlobalStream: GlobalStream instance
-        """
 
+        """
         return self._global_node(
             **merge(
                 {

typed_ffmpeg/dag/global_runnable/runnable.py

@@ -56,6 +56,7 @@ class GlobalRunable(GlobalArgs):
             merged = output1.merge_outputs(output2)
             merged.run()  # Creates both output files with one FFmpeg command
             ```
+
         """
         return self._global_node(*streams).stream()
 
@@ -75,6 +76,7 @@ class GlobalRunable(GlobalArgs):
             # Overwrite output file if it already exists
             ffmpeg.input("input.mp4").output("output.mp4").overwrite_output().run()
             ```
+
         """
         return self._global_node(y=True).stream()
 
@@ -83,6 +85,7 @@ class GlobalRunable(GlobalArgs):
         cmd: str | list[str] = "ffmpeg",
         overwrite_output: bool | None = None,
         auto_fix: bool = True,
+        use_filter_complex_script: bool = False,
     ) -> list[str]:
         """
         Build command-line arguments for invoking FFmpeg.
@@ -100,6 +103,8 @@ class GlobalRunable(GlobalArgs):
                              If None (default), use the current settings
             auto_fix: Whether to automatically fix issues in the filter graph,
                      such as adding split filters for reused streams
+            use_filter_complex_script: If True, use -filter_complex_script with a
+                                      temporary file instead of -filter_complex
 
         Returns:
             A list of strings representing the complete FFmpeg command
@@ -110,6 +115,7 @@ class GlobalRunable(GlobalArgs):
             args = ffmpeg.input("input.mp4").output("output.mp4").compile()
             # Result: ['ffmpeg', '-i', 'input.mp4', 'output.mp4']
             ```
+
         """
         from ...compile.compile_cli import compile_as_list
 
@@ -117,17 +123,30 @@ class GlobalRunable(GlobalArgs):
             cmd = [cmd]
 
         if overwrite_output is True:
-            return self.global_args(y=True).compile(cmd, auto_fix=auto_fix)
+            return self.global_args(y=True).compile(
+                cmd,
+                auto_fix=auto_fix,
+                use_filter_complex_script=use_filter_complex_script,
+            )
         elif overwrite_output is False:
-            return self.global_args(n=True).compile(cmd, auto_fix=auto_fix)
+            return self.global_args(n=True).compile(
+                cmd,
+                auto_fix=auto_fix,
+                use_filter_complex_script=use_filter_complex_script,
+            )
 
-        return cmd + compile_as_list(self._global_node().stream(), auto_fix=auto_fix)
+        return cmd + compile_as_list(
+            self._global_node().stream(),
+            auto_fix=auto_fix,
+            use_filter_complex_script=use_filter_complex_script,
+        )
 
     def compile_line(
         self,
         cmd: str | list[str] = "ffmpeg",
         overwrite_output: bool | None = None,
         auto_fix: bool = True,
+        use_filter_complex_script: bool = False,
     ) -> str:
         """
         Build a command-line string for invoking FFmpeg.
@@ -143,6 +162,8 @@ class GlobalRunable(GlobalArgs):
                              If False, add the -n option to never overwrite
                              If None (default), use the current settings
             auto_fix: Whether to automatically fix issues in the filter graph
+            use_filter_complex_script: If True, use -filter_complex_script with a
+                                      temporary file instead of -filter_complex
 
         Returns:
             A string representing the complete FFmpeg command with proper escaping
@@ -153,9 +174,15 @@ class GlobalRunable(GlobalArgs):
             cmd_str = ffmpeg.input("input.mp4").output("output.mp4").compile_line()
             # Result: 'ffmpeg -i input.mp4 output.mp4'
             ```
+
         """
         return command_line(
-            self.compile(cmd, overwrite_output=overwrite_output, auto_fix=auto_fix)
+            self.compile(
+                cmd,
+                overwrite_output=overwrite_output,
+                auto_fix=auto_fix,
+                use_filter_complex_script=use_filter_complex_script,
+            )
         )
 
     def run_async(
@@ -167,6 +194,7 @@ class GlobalRunable(GlobalArgs):
         quiet: bool = False,
         overwrite_output: bool | None = None,
         auto_fix: bool = True,
+        use_filter_complex_script: bool = False,
     ) -> subprocess.Popen[bytes]:
         """
         Run FFmpeg asynchronously as a subprocess.
@@ -186,6 +214,8 @@ class GlobalRunable(GlobalArgs):
                              If False, add the -n option to never overwrite
                              If None (default), use the current settings
             auto_fix: Whether to automatically fix issues in the filter graph
+            use_filter_complex_script: If True, use -filter_complex_script with a
+                                      temporary file instead of -filter_complex
 
         Returns:
             A subprocess.Popen object representing the running FFmpeg process
@@ -197,16 +227,19 @@ class GlobalRunable(GlobalArgs):
             # Do something while FFmpeg is running
             process.wait()  # Wait for completion
             ```
-        """
 
-        args = self.compile(cmd, overwrite_output=overwrite_output, auto_fix=auto_fix)
+        """
+        args = self.compile(
+            cmd,
+            overwrite_output=overwrite_output,
+            auto_fix=auto_fix,
+            use_filter_complex_script=use_filter_complex_script,
+        )
         stdin_stream = subprocess.PIPE if pipe_stdin else None
         stdout_stream = subprocess.PIPE if pipe_stdout or quiet else None
         stderr_stream = subprocess.PIPE if pipe_stderr or quiet else None
 
-        logger.info(
-            f"Running command: {self.compile_line(cmd, overwrite_output=overwrite_output, auto_fix=auto_fix)}"
-        )
+        logger.info(f"Running command: {' '.join(args)}")
 
         return subprocess.Popen(
             args,
@@ -224,6 +257,7 @@ class GlobalRunable(GlobalArgs):
         quiet: bool = False,
         overwrite_output: bool | None = None,
         auto_fix: bool = True,
+        use_filter_complex_script: bool = False,
     ) -> tuple[bytes, bytes]:
         """
         Run FFmpeg synchronously and wait for completion.
@@ -243,6 +277,8 @@ class GlobalRunable(GlobalArgs):
                              If False, add the -n option to never overwrite
                              If None (default), use the current settings
             auto_fix: Whether to automatically fix issues in the filter graph
+            use_filter_complex_script: If True, use -filter_complex_script with a
+                                      temporary file instead of -filter_complex
 
         Returns:
             A tuple of (stdout_bytes, stderr_bytes), which will be empty bytes
@@ -262,8 +298,8 @@ class GlobalRunable(GlobalArgs):
             )
             print(stderr.decode())  # Print FFmpeg's progress information
             ```
-        """
 
+        """
         process = self.run_async(
             cmd,
             pipe_stdin=input is not None,
@@ -272,6 +308,7 @@ class GlobalRunable(GlobalArgs):
             quiet=quiet,
             overwrite_output=overwrite_output,
             auto_fix=auto_fix,
+            use_filter_complex_script=use_filter_complex_script,
         )
         stdout, stderr = process.communicate(input)
         retcode = process.poll()
@@ -280,7 +317,10 @@ class GlobalRunable(GlobalArgs):
             raise FFMpegExecuteError(
                 retcode=retcode,
                 cmd=self.compile_line(
-                    cmd, overwrite_output=overwrite_output, auto_fix=auto_fix
+                    cmd,
+                    overwrite_output=overwrite_output,
+                    auto_fix=auto_fix,
+                    use_filter_complex_script=use_filter_complex_script,
                 ),
                 stdout=stdout,
                 stderr=stderr,

typed_ffmpeg/dag/io/__init__.py

@@ -0,0 +1 @@
+"""Input/output utilities for FFmpeg DAG operations."""

typed_ffmpeg/dag/io/_input.py

@@ -1,11 +1,17 @@
 # NOTE: this file is auto-generated, do not modify
-
+"""Input node."""
 
 from pathlib import Path
 from typing import Any
 
 from ...codecs.schema import FFMpegDecoderOption
 from ...formats.schema import FFMpegDemuxerOption
+from ...options.codec import (
+    FFMpegAVCodecContextDecoderOption,
+)
+from ...options.format import (
+    FFMpegAVFormatContextDecoderOption,
+)
 from ...streams.av import AVStream
 from ...types import (
     Boolean,
@@ -16,7 +22,9 @@ from ...types import (
     Time,
 )
 from ...utils.frozendict import merge
-from ..nodes import InputNode
+from ..nodes import (
+    InputNode,
+)
 
 
 def input(
@@ -76,10 +84,12 @@ def input(
     top: Int = None,
     decoder_options: FFMpegDecoderOption | None = None,
     demuxer_options: FFMpegDemuxerOption | None = None,
+    format_options: FFMpegAVFormatContextDecoderOption | None = None,
+    codec_options: FFMpegAVCodecContextDecoderOption | None = None,
     extra_options: dict[str, Any] | None = None,
 ) -> AVStream:
     """
-    Input file URL (ffmpeg ``-i`` option)
+    Input file URL (ffmpeg ``-i`` option).
 
     Args:
         filename: Input file URL
@@ -137,6 +147,8 @@ def input(
         top: deprecated, use the setfield video filter
         decoder_options: ffmpeg's decoder options
         demuxer_options: ffmpeg's demuxer options
+        format_options: ffmpeg's AVFormatContext options
+        codec_options: ffmpeg's AVCodecContext options
         extra_options: ffmpeg's input file options
 
     Returns:
@@ -147,6 +159,7 @@ def input(
     >>> input('input.mp4')
     <AVStream:input.mp4:0>
     ```
+
     """
     return InputNode(
         filename=str(filename),
@@ -205,8 +218,10 @@ def input(
                 "dn": dn,
                 "top": top,
             },
-            decoder_options.kwargs if decoder_options else {},
-            demuxer_options.kwargs if demuxer_options else {},
+            decoder_options,
+            demuxer_options,
+            format_options,
+            codec_options,
             extra_options,
         ),
     ).stream()

typed_ffmpeg/dag/io/_output.py

@@ -1,11 +1,17 @@
 # NOTE: this file is auto-generated, do not modify
-
+"""Output node."""
 
 from pathlib import Path
 from typing import Any
 
 from ...codecs.schema import FFMpegEncoderOption
 from ...formats.schema import FFMpegMuxerOption
+from ...options.codec import (
+    FFMpegAVCodecContextEncoderOption,
+)
+from ...options.format import (
+    FFMpegAVFormatContextEncoderOption,
+)
 from ...types import (
     Boolean,
     Float,
@@ -16,7 +22,11 @@ from ...types import (
     Time,
 )
 from ...utils.frozendict import merge
-from ..nodes import FilterableStream, OutputNode, OutputStream
+from ..nodes import (
+    FilterableStream,
+    OutputNode,
+    OutputStream,
+)
 
 
 def output(
@@ -118,10 +128,12 @@ def output(
     top: Int = None,
     encoder_options: FFMpegEncoderOption | None = None,
     muxer_options: FFMpegMuxerOption | None = None,
+    format_options: FFMpegAVFormatContextEncoderOption | None = None,
+    codec_options: FFMpegAVCodecContextEncoderOption | None = None,
     extra_options: dict[str, Any] | None = None,
 ) -> OutputStream:
-    """
-    Output file URL
+    r"""
+    Output file URL.
 
     Args:
         *streams: the streams to output
@@ -142,8 +154,7 @@ def output(
         program: add program with specified streams
         stream_group: add stream group with specified streams and group type-specific arguments
         dframes: set the number of data frames to output
-        target: specify target file type (\"vcd\", \"svcd\", \"dvd\", \"dv\" or \"dv50\
-        "with optional prefixes \"pal-\", \"ntsc-\" or \"film-\")
+        target: specify target file type (\"vcd\", \"svcd\", \"dvd\", \"dv\" or \"dv50\ "with optional prefixes \"pal-\", \"ntsc-\" or \"film-\")
         shortest: finish encoding within shortest input
         shortest_buf_duration: maximum buffering duration (in seconds) for the -shortest option
         bitexact: bitexact mode
@@ -223,12 +234,14 @@ def output(
         top: deprecated, use the setfield video filter
         encoder_options: ffmpeg's encoder options
         muxer_options: ffmpeg's muxer options
+        format_options: ffmpeg's AVFormatContext options
+        codec_options: ffmpeg's AVCodecContext options
         extra_options: the arguments for the output
 
     Returns:
         the output stream
-    """
 
+    """
     return OutputNode(
         inputs=streams,
         filename=str(filename),
@@ -329,8 +342,10 @@ def output(
                 "dn": dn,
                 "top": top,
             },
-            encoder_options.kwargs if encoder_options else {},
-            muxer_options.kwargs if muxer_options else {},
+            encoder_options,
+            muxer_options,
+            format_options,
+            codec_options,
             extra_options,
         ),
     ).stream()

typed_ffmpeg/dag/io/output_args.py

@@ -1,5 +1,5 @@
 # NOTE: this file is auto-generated, do not modify
-
+"""Output arguments."""
 
 from __future__ import annotations
 
@@ -9,6 +9,12 @@ from typing import TYPE_CHECKING, Any
 
 from ...codecs.schema import FFMpegEncoderOption
 from ...formats.schema import FFMpegMuxerOption
+from ...options.codec import (
+    FFMpegAVCodecContextEncoderOption,
+)
+from ...options.format import (
+    FFMpegAVFormatContextEncoderOption,
+)
 from ...types import (
     Boolean,
     Float,
@@ -25,6 +31,8 @@ if TYPE_CHECKING:
 
 
 class OutputArgs(ABC):
+    """Output arguments interface."""
+
     @abstractmethod
     def _output_node(
         self, *streams: FilterableStream, filename: str | Path, **kwargs: Any
@@ -130,10 +138,12 @@ class OutputArgs(ABC):
         top: Int = None,
         encoder_options: FFMpegEncoderOption | None = None,
         muxer_options: FFMpegMuxerOption | None = None,
+        format_options: FFMpegAVFormatContextEncoderOption | None = None,
+        codec_options: FFMpegAVCodecContextEncoderOption | None = None,
         extra_options: dict[str, Any] | None = None,
     ) -> OutputStream:
-        """
-        Output file URL
+        r"""
+        Output file URL.
 
         Args:
             *streams: the streams to output
@@ -234,13 +244,15 @@ class OutputArgs(ABC):
             dn: disable data
             top: deprecated, use the setfield video filter
             encoder_options: ffmpeg's encoder options
-            muxer_options: FFMpegMuxerOption | None = None,
+            muxer_options: FFMpegMuxerOption
+            format_options: FFMpegAVFormatContextEncoderOption
+            codec_options: FFMpegAVCodecContextEncoderOption
             extra_options: the arguments for the output
 
         Returns:
             the output stream
-        """
 
+        """
         return self._output_node(
             *streams,
             filename=filename,
@@ -341,8 +353,10 @@ class OutputArgs(ABC):
                     "dn": dn,
                     "top": top,
                 },
-                encoder_options.kwargs if encoder_options else {},
-                muxer_options.kwargs if muxer_options else {},
+                encoder_options,
+                muxer_options,
+                format_options,
+                codec_options,
                 extra_options,
             ),
         ).stream()

typed_ffmpeg/dag/nodes.py

@@ -1,3 +1,5 @@
+"""DAG node definitions for FFmpeg filter graphs."""
+
 from __future__ import annotations
 
 import logging
@@ -60,6 +62,7 @@ class FilterNode(Node):
 
         Returns:
             The name of the filter
+
         """
         return self.name
 
@@ -81,6 +84,7 @@ class FilterNode(Node):
 
         Raises:
             FFMpegValueError: If the specified index is out of range
+
         """
         from ..streams.video import VideoStream
 
@@ -111,6 +115,7 @@ class FilterNode(Node):
 
         Raises:
             FFMpegValueError: If the specified index is out of range
+
         """
         from ..streams.audio import AudioStream
 
@@ -136,6 +141,7 @@ class FilterNode(Node):
         Raises:
             FFMpegValueError: If the number of inputs doesn't match input_typings
             FFMpegTypeError: If an input stream doesn't match its expected type
+
         """
         from ..streams.audio import AudioStream
         from ..streams.video import VideoStream
@@ -207,6 +213,7 @@ class FilterableStream(Stream, OutputArgs):
                 filename="output.mp4", c="libx264", crf=23
             )
             ```
+
         """
         return OutputNode(
             inputs=(self, *streams),
@@ -244,6 +251,7 @@ class FilterableStream(Stream, OutputArgs):
             # Apply a blur filter to a video stream
             blurred = stream.vfilter(name="boxblur", luma_radius=2)
             ```
+
         """
         return self.filter_multi_output(
             *streams,
@@ -283,6 +291,7 @@ class FilterableStream(Stream, OutputArgs):
             # Apply a volume filter to an audio stream
             louder = stream.afilter(name="volume", volume=2.0)
             ```
+
         """
         return self.filter_multi_output(
             *streams,
@@ -327,6 +336,7 @@ class FilterableStream(Stream, OutputArgs):
             stream1 = split_node.video(0)
             stream2 = split_node.video(1)
             ```
+
         """
         return FilterNode(
             name=name,
@@ -337,6 +347,7 @@ class FilterableStream(Stream, OutputArgs):
         )
 
     def __post_init__(self) -> None:
+        """Validate that filter streams have an index."""
         if not isinstance(self.node, InputNode):
             assert self.index is not None, "Filter streams must have an index"
 
@@ -368,6 +379,7 @@ class InputNode(Node):
 
         Returns:
             The basename of the input file
+
         """
         return os.path.basename(self.filename)
 
@@ -390,6 +402,7 @@ class InputNode(Node):
             # Apply a filter to the video stream
             scaled = video.scale(width=1280, height=720)
             ```
+
         """
         from ..streams.video import VideoStream
 
@@ -414,6 +427,7 @@ class InputNode(Node):
             # Apply a filter to the audio stream
             volume_adjusted = audio.volume(volume=2.0)
             ```
+
         """
         from ..streams.audio import AudioStream
 
@@ -438,6 +452,7 @@ class InputNode(Node):
             # Output both audio and video to a new file
             output = av_stream.output("output.mp4")
             ```
+
         """
         from ..streams.av import AVStream
 
@@ -471,6 +486,7 @@ class OutputNode(Node):
 
         Returns:
             The basename of the output file
+
         """
         return os.path.basename(self.filename)
 
@@ -492,6 +508,7 @@ class OutputNode(Node):
             output_stream = output_node.stream()
             with_global_opts = output_stream.global_args(y=True)
             ```
+
         """
         return OutputStream(node=self)
 
@@ -531,6 +548,7 @@ class OutputStream(Stream, GlobalRunable):
             # Add global options to an output stream
             global_node = output_stream._global_node(y=True, loglevel="quiet")
             ```
+
         """
         return GlobalNode(inputs=(self, *streams), kwargs=FrozenDict(kwargs))
 
@@ -567,6 +585,7 @@ class GlobalNode(Node):
             # Execute the command
             global_stream.run()
             ```
+
         """
         return GlobalStream(node=self)
 
@@ -607,6 +626,7 @@ class GlobalStream(Stream, GlobalRunable):
             global_stream = ffmpeg.output("output.mp4").global_args(y=True)
             enhanced = global_stream._global_node(loglevel="quiet")
             ```
+
         """
         inputs = (*self.node.inputs, *streams)
         kwargs = dict(self.node.kwargs) | kwargs