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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
|
/*
* Use with CVS JOCR/GOCR.
*
* You will have to change 'vobsub_id' value if you want another subtitle than number 0.
*
* HINT: you can view the subtitle that is being decoded with "display subtitle-*.pgm"
*
* This program 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.
*
* This program 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 this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
/* Make sure this accesses the CVS version of JOCR/GOCR */
#define GOCR_PROGRAM "gocr"
#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include "libvo/video_out.h"
#include "vobsub.h"
#include "spudec.h"
/* XXX Kludge ahead, this MUST be the same as the definitions found in ../spudec.c */
typedef struct packet_t packet_t;
struct packet_t {
unsigned char *packet;
unsigned int palette[4];
unsigned int alpha[4];
unsigned int control_start; /* index of start of control data */
unsigned int current_nibble[2]; /* next data nibble (4 bits) to be
processed (for RLE decoding) for
even and odd lines */
int deinterlace_oddness; /* 0 or 1, index into current_nibble */
unsigned int start_col, end_col;
unsigned int start_row, end_row;
unsigned int width, height, stride;
unsigned int start_pts, end_pts;
packet_t *next;
};
typedef struct {
packet_t *queue_head;
packet_t *queue_tail;
unsigned int global_palette[16];
unsigned int orig_frame_width, orig_frame_height;
unsigned char* packet;
size_t packet_reserve; /* size of the memory pointed to by packet */
unsigned int packet_offset; /* end of the currently assembled fragment */
unsigned int packet_size; /* size of the packet once all fragments are assembled */
unsigned int packet_pts; /* PTS for this packet */
unsigned int palette[4];
unsigned int alpha[4];
unsigned int cuspal[4];
unsigned int custom;
unsigned int now_pts;
unsigned int start_pts, end_pts;
unsigned int start_col, end_col;
unsigned int start_row, end_row;
unsigned int width, height, stride;
size_t image_size; /* Size of the image buffer */
unsigned char *image; /* Grayscale value */
unsigned char *aimage; /* Alpha value */
unsigned int scaled_frame_width, scaled_frame_height;
unsigned int scaled_start_col, scaled_start_row;
unsigned int scaled_width, scaled_height, scaled_stride;
size_t scaled_image_size;
unsigned char *scaled_image;
unsigned char *scaled_aimage;
int auto_palette; /* 1 if we lack a palette and must use an heuristic. */
int font_start_level; /* Darkest value used for the computed font */
const vo_functions_t *hw_spu;
int spu_changed;
} spudec_handle_t;
int vobsub_id=0;
int sub_pos=0;
static spudec_handle_t *spudec;
static FILE *fsub = NULL;
static unsigned int sub_idx = 0;
static void
process_gocr_output(const char *const fname, unsigned int start, unsigned int end)
{
FILE *file;
int temp, h, m, s, ms;
int c, bol;
file = fopen(fname, "r");
if (file == NULL) {
perror("fopen failed");
return;
}
temp = start;
temp /= 90;
h = temp / 3600000;
temp %= 3600000;
m = temp / 60000;
temp %= 60000;
s = temp / 1000;
temp %= 1000;
ms = temp;
fprintf(fsub, "%d\n%02d:%02d:%02d,%03d --> ", ++sub_idx, h, m, s, ms);
temp = end;
temp /= 90;
h = temp / 3600000;
temp %= 3600000;
m = temp / 60000;
temp %= 60000;
s = temp / 1000;
temp %= 1000;
ms = temp;
fprintf(fsub, "%02d:%02d:%02d,%03d\n", h, m, s, ms);
bol = 1;
while ((c = getc(file)) != EOF) {
if (bol) {
if (!isspace(c)) {
putc(c, fsub);
bol=0;
}
}
else if (!bol) {
putc(c, fsub);
bol = c == '\n';
}
}
putc('\n', fsub);
fflush(fsub);
fclose(file);
}
static void
output_pgm(FILE *f, int w, int h, unsigned char *src, unsigned char *srca, int stride)
{
int x, y;
fprintf(f,
"P5\n"
"%d %d\n"
"255\n",
w, h);
for (y = 0; y < h; ++y) {
for (x = 0; x < w; ++x) {
int res;
if (srca[x])
res = src[x] * (256 - srca[x]);
else
res = 0;
res = (65535 - res) >> 8;
putc(res&0xff, f);
}
src += stride;
srca += stride;
}
putc('\n', f);
}
static void
draw_alpha(int x0, int y0, int w, int h, unsigned char *src, unsigned char *srca, int stride)
{
FILE *f;
char buf[128];
char cmd[512];
int cmdres;
const char *const tmpfname = tmpnam(NULL);
sprintf(buf, "subtitle-%d-%d.pgm", spudec->start_pts / 90, spudec->end_pts / 90);
f = fopen(buf, "w");
output_pgm(f, w, h, src, srca, stride);
fclose(f);
/* see <URL:http://cvs.sourceforge.net/cgi-bin/viewcvs.cgi/subtitleripper/subtitleripper/src/README.gocr?rev=HEAD&content-type=text/vnd.viewcvs-markup> */
sprintf(cmd, GOCR_PROGRAM" -v 1 -s 7 -d 0 -m 130 -m 256 -m 32 -i %s -o %s", buf, tmpfname);
cmdres = system(cmd);
if (cmdres < 0) {
perror("system failed");
exit(EXIT_FAILURE);
}
else if (cmdres) {
fprintf(stderr, GOCR_PROGRAM" returned %d\n", cmdres);
exit(cmdres);
}
process_gocr_output(tmpfname, spudec->start_pts, spudec->end_pts);
unlink(buf);
unlink(tmpfname);
}
int
main(int argc, char **argv)
{
const char *vobsubname, *subripname;
void *vobsub;
void *packet;
int packet_len;
unsigned int pts100;
if (argc < 2 || 4 < argc) {
fprintf(stderr, "Usage: %s <vobsub basename> [<subid> [<output filename>] ]\n", argv[0]);
exit(EXIT_FAILURE);
}
vobsubname = argv[1];
subripname = NULL;
fsub = stdout;
if (argc >= 3)
vobsub_id = atoi(argv[2]);
if (argc >= 4) {
subripname = argv[3];
fsub = fopen(subripname, "w");
}
vobsub = vobsub_open(vobsubname, NULL, 0, &spudec);
while ((packet_len=vobsub_get_next_packet(vobsub, &packet, &pts100)) >= 0) {
spudec_assemble(spudec, packet, packet_len, pts100);
if (spudec->queue_head) {
spudec_heartbeat(spudec, spudec->queue_head->start_pts);
if (spudec_changed(spudec))
spudec_draw(spudec, draw_alpha);
}
}
if (vobsub)
vobsub_close(vobsub);
exit(EXIT_SUCCESS);
}
|
