2 * Use with CVS JOCR/GOCR.
4 * You will have to change 'vobsub_id' value if you want another subtitle than number 0.
6 * HINT: you can view the subtitle that is being decoded with "display subtitle-*.pgm"
10 /* Make sure this accesses the CVS version of JOCR/GOCR */
11 #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 const vo_functions_t
*hw_spu
;
78 static spudec_handle_t
*spudec
;
79 static FILE *fsub
= NULL
;
80 static unsigned int sub_idx
= 0;
83 process_gocr_output(const char *const fname
, unsigned int start
, unsigned int end
)
86 int temp
, h
, m
, s
, ms
;
88 file
= fopen(fname
, "r");
90 perror("fopen failed");
102 fprintf(fsub
, "%d\n%02d:%02d:%02d,%03d --> ", ++sub_idx
, h
, m
, s
, ms
);
112 fprintf(fsub
, "%02d:%02d:%02d,%03d\n", h
, m
, s
, ms
);
114 while ((c
= getc(file
)) != EOF
) {
132 output_pgm(FILE *f
, int w
, int h
, unsigned char *src
, unsigned char *srca
, int stride
)
140 for (y
= 0; y
< h
; ++y
) {
141 for (x
= 0; x
< w
; ++x
) {
144 res
= src
[x
] * (256 - srca
[x
]);
147 res
= (65535 - res
) >> 8;
158 draw_alpha(int x0
, int y0
, int w
, int h
, unsigned char *src
, unsigned char *srca
, int stride
)
164 const char *const tmpfname
= tmpnam(NULL
);
165 sprintf(buf
, "subtitle-%d-%d.pgm", spudec
->start_pts
/ 90, spudec
->end_pts
/ 90);
167 output_pgm(f
, w
, h
, src
, srca
, stride
);
169 /* see <URL:http://cvs.sourceforge.net/cgi-bin/viewcvs.cgi/subtitleripper/subtitleripper/src/README.gocr?rev=HEAD&content-type=text/vnd.viewcvs-markup> */
170 sprintf(cmd
, GOCR_PROGRAM
" -v 1 -s 7 -d 0 -m 130 -m 256 -m 32 -i %s -o %s", buf
, tmpfname
);
171 cmdres
= system(cmd
);
173 perror("system failed");
177 fprintf(stderr
, GOCR_PROGRAM
" returned %d\n", cmdres
);
180 process_gocr_output(tmpfname
, spudec
->start_pts
, spudec
->end_pts
);
186 main(int argc
, char **argv
)
188 const char *vobsubname
, *subripname
;
194 if (argc
< 2 || 4 < argc
) {
195 fprintf(stderr
, "Usage: %s <vobsub basename> [<subid> [<output filename>] ]\n", argv
[0]);
198 vobsubname
= argv
[1];
202 vobsub_id
= atoi(argv
[2]);
204 subripname
= argv
[3];
205 fsub
= fopen(subripname
, "w");
208 vobsub
= vobsub_open(vobsubname
, NULL
, 0, &spudec
);
209 while ((packet_len
=vobsub_get_next_packet(vobsub
, &packet
, &pts100
)) >= 0) {
210 spudec_assemble(spudec
, packet
, packet_len
, pts100
);
211 if (spudec
->queue_head
) {
212 spudec_heartbeat(spudec
, spudec
->queue_head
->start_pts
);
213 if (spudec_changed(spudec
))
214 spudec_draw(spudec
, draw_alpha
);
219 vobsub_close(vobsub
);