1 /* -*- compile-command: "gcc -g -Wall -I.. -o subrip subrip.c ../vobsub.o ../spudec.o ../mp_msg.o ../unrarlib.o ../postproc/swscale.o ../postproc/rgb2rgb.o ../postproc/yuv2rgb.o ../libmpcodecs/img_format.o -lm" -*- */
3 * Use with CVS JOCR/GOCR.
5 * You will have to change 'vobsub_id' value if you want another subtitle than number 0.
7 * HINT: you can view the subtitle that is being decoded with "display subtitle-*.pgm"
11 /* Make sure this accesses the CVS version of JOCR/GOCR */
12 #define GOCR_PROGRAM "gocr"
18 #include <sys/types.h>
20 #include "libvo/video_out.h"
24 /* XXX Kludge ahead, this MUST be the same as the definitions found in ../spudec.c */
25 typedef struct packet_t packet_t
;
27 unsigned char *packet
;
28 unsigned int palette
[4];
29 unsigned int alpha
[4];
30 unsigned int control_start
; /* index of start of control data */
31 unsigned int current_nibble
[2]; /* next data nibble (4 bits) to be
32 processed (for RLE decoding) for
34 int deinterlace_oddness
; /* 0 or 1, index into current_nibble */
35 unsigned int start_col
, end_col
;
36 unsigned int start_row
, end_row
;
37 unsigned int width
, height
, stride
;
38 unsigned int start_pts
, end_pts
;
44 unsigned int global_palette
[16];
45 unsigned int orig_frame_width
, orig_frame_height
;
46 unsigned char* packet
;
47 size_t packet_reserve
; /* size of the memory pointed to by packet */
48 unsigned int packet_offset
; /* end of the currently assembled fragment */
49 unsigned int packet_size
; /* size of the packet once all fragments are assembled */
50 unsigned int packet_pts
; /* PTS for this packet */
51 unsigned int palette
[4];
52 unsigned int alpha
[4];
53 unsigned int cuspal
[4];
56 unsigned int start_pts
, end_pts
;
57 unsigned int start_col
, end_col
;
58 unsigned int start_row
, end_row
;
59 unsigned int width
, height
, stride
;
60 size_t image_size
; /* Size of the image buffer */
61 unsigned char *image
; /* Grayscale value */
62 unsigned char *aimage
; /* Alpha value */
63 unsigned int scaled_frame_width
, scaled_frame_height
;
64 unsigned int scaled_start_col
, scaled_start_row
;
65 unsigned int scaled_width
, scaled_height
, scaled_stride
;
66 size_t scaled_image_size
;
67 unsigned char *scaled_image
;
68 unsigned char *scaled_aimage
;
69 int auto_palette
; /* 1 if we lack a palette and must use an heuristic. */
70 int font_start_level
; /* Darkest value used for the computed font */
71 vo_functions_t
*hw_spu
;
82 static spudec_handle_t
*spudec
;
83 static FILE *fsub
= NULL
;
84 static unsigned int sub_idx
= 0;
87 process_gocr_output(const char *const fname
, unsigned int start
, unsigned int end
)
90 int temp
, h
, m
, s
, ms
;
92 file
= fopen(fname
, "r");
94 perror("fopen failed");
106 fprintf(fsub
, "%d\n%02d:%02d:%02d,%03d --> ", ++sub_idx
, h
, m
, s
, ms
);
116 fprintf(fsub
, "%02d:%02d:%02d,%03d\n", h
, m
, s
, ms
);
118 while ((c
= getc(file
)) != EOF
) {
136 output_pgm(FILE *f
, int w
, int h
, unsigned char *src
, unsigned char *srca
, int stride
)
144 for (y
= 0; y
< h
; ++y
) {
145 for (x
= 0; x
< w
; ++x
) {
148 res
= src
[x
] * (256 - srca
[x
]);
151 res
= (65535 - res
) >> 8;
162 draw_alpha(int x0
, int y0
, int w
, int h
, unsigned char *src
, unsigned char *srca
, int stride
)
168 const char *const tmpfname
= tmpnam(NULL
);
169 sprintf(buf
, "subtitle-%d-%d.pgm", spudec
->start_pts
/ 90, spudec
->end_pts
/ 90);
171 output_pgm(f
, w
, h
, src
, srca
, stride
);
173 /* see <URL:http://cvs.sourceforge.net/cgi-bin/viewcvs.cgi/subtitleripper/subtitleripper/src/README.gocr?rev=HEAD&content-type=text/vnd.viewcvs-markup> */
174 sprintf(cmd
, GOCR_PROGRAM
" -v 1 -s 7 -d 0 -m 130 -m 256 -m 32 -i %s -o %s", buf
, tmpfname
);
175 cmdres
= system(cmd
);
177 perror("system failed");
181 fprintf(stderr
, GOCR_PROGRAM
" returned %d\n", cmdres
);
184 process_gocr_output(tmpfname
, spudec
->start_pts
, spudec
->end_pts
);
190 fast_memcpy(void *a
, void *b
, int s
)
196 main(int argc
, char **argv
)
198 const char *vobsubname
, *subripname
;
203 unsigned int prev_pts
;
207 if (argc
< 2 || 4 < argc
) {
208 fprintf(stderr
, "Usage: %s <vobsub basename> [<subid> [<output filename>] ]\n", argv
[0]);
211 vobsubname
= argv
[1];
215 vobsub_id
= atoi(argv
[2]);
217 subripname
= argv
[3];
218 fsub
= fopen(subripname
, "w");
221 vobsub
= vobsub_open(vobsubname
, NULL
, 0, &spudec
);
222 while ((packet_len
=vobsub_get_next_packet(vobsub
, &packet
, &pts100
)) >= 0) {
223 spudec_assemble(spudec
, packet
, packet_len
, pts100
);
224 if (spudec
->queue_head
) {
225 spudec_heartbeat(spudec
, spudec
->queue_head
->start_pts
);
226 if (spudec_changed(spudec
))
227 spudec_draw(spudec
, draw_alpha
);
232 vobsub_close(vobsub
);