1
2
3
4
5
6
7
8
9
10
11
12
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
/*
Copyright (C) 2019 Carl Hetherington <cth@carlh.net>
This file is part of DCP-o-matic.
DCP-o-matic is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
DCP-o-matic is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with DCP-o-matic. If not, see <http://www.gnu.org/licenses/>.
*/
#include "examine_ffmpeg_subtitles_job.h"
#include "ffmpeg_content.h"
#include "ffmpeg_subtitle_stream.h"
#include "text_content.h"
#include "cross.h"
extern "C" {
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
}
#include <iostream>
#include "i18n.h"
using std::string;
using std::cout;
using std::shared_ptr;
ExamineFFmpegSubtitlesJob::ExamineFFmpegSubtitlesJob (shared_ptr<const Film> film, shared_ptr<FFmpegContent> c)
: Job (film)
, FFmpeg (c)
, _content (c)
{
}
ExamineFFmpegSubtitlesJob::~ExamineFFmpegSubtitlesJob ()
{
stop_thread ();
}
string
ExamineFFmpegSubtitlesJob::name () const
{
return _("Examining subtitles");
}
string
ExamineFFmpegSubtitlesJob::json_name () const
{
return N_("examine_subtitles");
}
void
ExamineFFmpegSubtitlesJob::run ()
{
int64_t const len = _file_group.length ();
while (true) {
auto packet = av_packet_alloc ();
DCPOMATIC_ASSERT (packet);
int r = av_read_frame (_format_context, packet);
if (r < 0) {
av_packet_free (&packet);
break;
}
if (len > 0) {
set_progress (float(_format_context->pb->pos) / len);
} else {
set_progress_unknown ();
}
if (_content->subtitle_stream() && _content->subtitle_stream()->uses_index(_format_context, packet->stream_index) && _content->only_text()->use()) {
int got_subtitle;
AVSubtitle sub;
if (avcodec_decode_subtitle2(subtitle_codec_context(), &sub, &got_subtitle, packet) >= 0 && got_subtitle) {
for (unsigned int i = 0; i < sub.num_rects; ++i) {
AVSubtitleRect const * rect = sub.rects[i];
if (rect->type == SUBTITLE_BITMAP) {
#ifdef DCPOMATIC_HAVE_AVSUBTITLERECT_PICT
/* sub_p looks up into a BGRA palette which is here
(i.e. first byte B, second G, third R, fourth A)
*/
uint8_t const * palette = rect->pict.data[1];
#else
/* sub_p looks up into a BGRA palette which is here
(i.e. first byte B, second G, third R, fourth A)
*/
uint8_t const * palette = rect->data[1];
#endif
for (int j = 0; j < rect->nb_colors; ++j) {
RGBA c (palette[2], palette[1], palette[0], palette[3]);
_content->subtitle_stream()->set_colour (c, c);
palette += 4;
}
}
}
}
}
av_packet_free (&packet);
}
set_progress (1);
set_state (FINISHED_OK);
}
|
