Skip to content

sub-pgs

SubFilePGS dataclass

Bases: MuxingFile

Utility class representing a PGS/SUP subtitle file.

Parameters:

Name Type Description Default
file PathLike | list[PathLike] | GlobSearch

Can be a string, Path object or GlobSearch.

 required
container_delay int

Set a container delay used in the muxing process later.

 0
source PathLike | None

The file this sub originates from, will be set by the constructor.

 None
Source code in .venv/lib/python3.11/site-packages/muxtools/subtitle/sub_pgs.py 
13
14
15
16
17
18
19
20
21
22
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
@dataclass
class SubFilePGS(MuxingFile):
    """
    Utility class representing a PGS/SUP subtitle file.

    :param file:            Can be a string, Path object or GlobSearch.
    :param container_delay: Set a container delay used in the muxing process later.
    :param source:          The file this sub originates from, will be set by the constructor.
    """

    def __post_init__(self):
        super().__post_init__()
        parsed = ParsedFile.from_file(self.file, self)
        parsed_track = parsed.find_tracks(type=TrackType.SUB, relative_id=0, error_if_empty=True, caller=self)[0]
        if parsed_track.codec_name.lower() != "hdmv_pgs_subtitle":
            raise error(f"The passed file is not a PGS subtitle file. ({parsed_track.codec_name})", caller=self)

    def shift(self, shift: int, shift_is_ms: bool = False, timesource: TimeSourceT = None, timescale: TimeScaleT = None, quiet: bool = True) -> Self:
        """
        Shifts all lines by any frame number with supmover.

        :param shift:               Number of frames to shift by
        :param shift_is_ms:         If True, the shift is in milliseconds.
        :param timesource:          The source of timestamps/timecodes. For details check the docstring on the type.
        :param timescale:           Unit of time (in seconds) in terms of which frame timestamps are represented.\n
                                    For details check the docstring on the type.
        :param quiet:               If True, suppresses supmover output.
        """
        supmover = get_executable("SupMover")
        out = make_output(self.file, "sup", temp=True)
        fileIn = ensure_path(self.file, self)
        args = [supmover, str(fileIn), str(out), "--delay"]
        if shift_is_ms:
            args.append(str(shift))
        else:
            resolved_ts = resolve_timesource_and_scale(timesource, timescale, fetch_from_setup=True, caller=self)
            ms = resolved_ts.frame_to_time(abs(shift), TimeType.START, 3)
            args.append(str(ms) if shift >= 0 else f"-{ms}")

        if run_commandline(args, quiet):
            clean_temp_files()
            raise error("Failed to shift subtitle file.", caller=self)

        fileIn.unlink(missing_ok=True)
        move(out, fileIn)
        clean_temp_files()
        return self

    @classmethod
    def extract_from(cls: type[Self], fileIn: PathLike, track: int = 0, preserve_delay: bool = False, quiet: bool = True) -> Self:
        """
        Extract a PGS subtitle track from a file using ffmpeg.\n

        :param fileIn:          The input file to extract from.
        :param track:           The track number to extract.
        :param preserve_delay:  If True, the container delay will be preserved.
        :param quiet:           If True, suppresses ffmpeg output.
        :return:                An instance of SubFilePGS containing the extracted subtitle.
        """
        caller = "SubFilePGS.extract_from"
        parsed = ParsedFile.from_file(fileIn, caller)
        parsed_track = parsed.find_tracks(type=TrackType.SUB, relative_id=track, error_if_empty=True, caller=caller)[0]

        if parsed_track.codec_name.lower() != "hdmv_pgs_subtitle":
            raise error(f"The specified track is not a PGS subtitle. ({parsed_track.codec_name})", caller=caller)

        info(f"Extracting PGS subtitle track {track} from '{parsed.source.name}'", caller=caller)
        ffmpeg = get_executable("ffmpeg")

        out = make_output(fileIn, "sup", str(parsed_track.index))
        args = [ffmpeg, "-hide_banner", "-i", str(parsed.source), "-map", f"0:s:{str(track)}", "-c", "copy", str(out)]
        if run_commandline(args, quiet):
            raise error("Failed to extract subtitle track from file.", caller=caller)

        delay = 0 if not preserve_delay else parsed_track.container_delay

        return cls(out, delay, fileIn)

extract_from(fileIn, track=0, preserve_delay=False, quiet=True) classmethod

Extract a PGS subtitle track from a file using ffmpeg.

Parameters:

Name Type Description Default
fileIn PathLike

The input file to extract from.

 required
track int

The track number to extract.

 0
preserve_delay bool

If True, the container delay will be preserved.

 False
quiet bool

If True, suppresses ffmpeg output.

 True

Returns:

Type Description
Self

An instance of SubFilePGS containing the extracted subtitle.
Source code in .venv/lib/python3.11/site-packages/muxtools/subtitle/sub_pgs.py 
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
@classmethod
def extract_from(cls: type[Self], fileIn: PathLike, track: int = 0, preserve_delay: bool = False, quiet: bool = True) -> Self:
    """
    Extract a PGS subtitle track from a file using ffmpeg.\n

    :param fileIn:          The input file to extract from.
    :param track:           The track number to extract.
    :param preserve_delay:  If True, the container delay will be preserved.
    :param quiet:           If True, suppresses ffmpeg output.
    :return:                An instance of SubFilePGS containing the extracted subtitle.
    """
    caller = "SubFilePGS.extract_from"
    parsed = ParsedFile.from_file(fileIn, caller)
    parsed_track = parsed.find_tracks(type=TrackType.SUB, relative_id=track, error_if_empty=True, caller=caller)[0]

    if parsed_track.codec_name.lower() != "hdmv_pgs_subtitle":
        raise error(f"The specified track is not a PGS subtitle. ({parsed_track.codec_name})", caller=caller)

    info(f"Extracting PGS subtitle track {track} from '{parsed.source.name}'", caller=caller)
    ffmpeg = get_executable("ffmpeg")

    out = make_output(fileIn, "sup", str(parsed_track.index))
    args = [ffmpeg, "-hide_banner", "-i", str(parsed.source), "-map", f"0:s:{str(track)}", "-c", "copy", str(out)]
    if run_commandline(args, quiet):
        raise error("Failed to extract subtitle track from file.", caller=caller)

    delay = 0 if not preserve_delay else parsed_track.container_delay

    return cls(out, delay, fileIn)

shift(shift, shift_is_ms=False, timesource=None, timescale=None, quiet=True)

Shifts all lines by any frame number with supmover.

Parameters:

Name Type Description Default
shift int

Number of frames to shift by

 required
shift_is_ms bool

If True, the shift is in milliseconds.

 False
timesource TimeSourceT

The source of timestamps/timecodes. For details check the docstring on the type.

 None
timescale TimeScaleT

Unit of time (in seconds) in terms of which frame timestamps are represented. For details check the docstring on the type.

 None
quiet bool

If True, suppresses supmover output.

 True
Source code in .venv/lib/python3.11/site-packages/muxtools/subtitle/sub_pgs.py 
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
def shift(self, shift: int, shift_is_ms: bool = False, timesource: TimeSourceT = None, timescale: TimeScaleT = None, quiet: bool = True) -> Self:
    """
    Shifts all lines by any frame number with supmover.

    :param shift:               Number of frames to shift by
    :param shift_is_ms:         If True, the shift is in milliseconds.
    :param timesource:          The source of timestamps/timecodes. For details check the docstring on the type.
    :param timescale:           Unit of time (in seconds) in terms of which frame timestamps are represented.\n
                                For details check the docstring on the type.
    :param quiet:               If True, suppresses supmover output.
    """
    supmover = get_executable("SupMover")
    out = make_output(self.file, "sup", temp=True)
    fileIn = ensure_path(self.file, self)
    args = [supmover, str(fileIn), str(out), "--delay"]
    if shift_is_ms:
        args.append(str(shift))
    else:
        resolved_ts = resolve_timesource_and_scale(timesource, timescale, fetch_from_setup=True, caller=self)
        ms = resolved_ts.frame_to_time(abs(shift), TimeType.START, 3)
        args.append(str(ms) if shift >= 0 else f"-{ms}")

    if run_commandline(args, quiet):
        clean_temp_files()
        raise error("Failed to shift subtitle file.", caller=self)

    fileIn.unlink(missing_ok=True)
    move(out, fileIn)
    clean_temp_files()
    return self