--- /dev/null
+#include "../libzmpeg3.h"
+
+zsubtitle_t::
+subtitle_t()
+{
+//zmsgs("subtitle=%p\n",this);
+}
+
+// closed caption or display
+zsubtitle_t::
+subtitle_t(int nid, int ww, int hh)
+{
+//zmsgs("subtitle=%p %dx%d\n", this, ww, hh);
+ id = nid;
+ done = -1;
+ active = -1;
+ draw = -1;
+ set_image_size(ww, hh);
+}
+
+zsubtitle_t::
+~subtitle_t()
+{
+//zmsgs("subtitle=%p/%d subtitle->data=%p %d %d\n",
+// this, id, data, data_allocated, data_used);
+ delete [] data;
+ delete [] image_y;
+ delete [] image_u;
+ delete [] image_v;
+ delete [] image_a;
+}
+
+void zsubtitle_t::
+set_image_size(int isz)
+{
+//zmsgs("subtitle=%p/%d\n", this, isz);
+ sz = isz;
+ /* Allocate image buffers */
+ delete [] image_y;
+ delete [] image_u;
+ delete [] image_v;
+ delete [] image_a;
+ image_y = znew(uint8_t,isz);
+ image_u = znew(uint8_t,isz);
+ image_v = znew(uint8_t,isz);
+ image_a = znew(uint8_t,isz);
+}
+
+void zsubtitle_t::
+set_image_size(int ww, int hh)
+{
+ w = ww; h = hh;
+ int isz = ww*hh + ww;
+ if( isz > sz ) set_image_size(isz);
+}
+
+/* Returns 1 if failure */
+/* is_8bit mode has not been tested (no test media) */
+int zsubtitle_t::
+decompress_subtitle(zmpeg3_t *zsrc)
+{
+ //int got_alpha = 0;
+ int is_8bit = 0;
+ uint8_t *yuv_palette = 0;
+ uint8_t *yuv_alpha = 0;
+ int ofsz = 16;
+ bits_t st(zsrc,0);
+ st.use_ptr(data);
+ uint8_t *end = data + data_used;
+ if( st.eos(end,16) ) return 1;
+ int pkt_sz = st.get_bits(16);
+ if( pkt_sz == 0 ) ofsz = 32;
+ int field_offset[3];
+ field_offset[0] = 0; // begining of even field
+ field_offset[1] = 0; // end of even field, begining odd field
+ if( st.eos(end,ofsz) ) return 1;
+ /* end of odd field, beginning of ctl seq */
+ field_offset[2] = st.get_bits(ofsz);
+//zmsgs("%d 0x%02x%02x data_used=%d data_size=%d\n",
+// __LINE__, data[0],data[1],data_used,field_offset[2]);
+ if( data + field_offset[2] > end ) return 1;
+ palette[0] = 0x00; palette[1] = 0x01;
+ palette[2] = 0x02; palette[3] = 0x03;
+ alpha[0] = 0xff; alpha[1] = 0xff;
+ alpha[2] = 0xff; alpha[3] = 0xff;
+ int ww = 0, hh = 0;
+ x1 = 0; x2 = 0;
+ y1 = 0; y2 = 0;
+ force = 0;
+ start_time = stop_time = 0;
+
+ /* Advance to control sequences */
+ uint8_t *ptr = data + field_offset[2];
+ uint8_t *last_control_start = 0;
+ uint8_t *next_control_start = ptr;
+
+ while( last_control_start != next_control_start ) {
+ st.use_ptr(next_control_start);
+ if( st.eos(end,16) ) break; /* date */
+ int date = st.get_bits(16) * 10;
+ /* Offset of next control sequence */
+ if( st.eos(end,ofsz) ) break;
+ int next = st.get_bits(ofsz);
+ last_control_start = next_control_start;
+ next_control_start = data + next;
+ if( next_control_start > end ) return 1;
+
+ int done = 0;
+ while( !done && !st.eos(end,8) ) {
+ int type = st.get_bits(8);
+ switch( type ) {
+ case 0x00:
+ force = 1;
+ break;
+ case 0x01:
+ start_time = date;
+//zmsgs("start_time %d\n", start_time);
+ break;
+ case 0x02:
+ stop_time = date;
+//zmsgs("stop_time %d\n", stop_time);
+ break;
+ case 0x83:
+ if( st.eos(end,768*8) ) return 1;
+ yuv_palette = st.input_ptr;
+ st.input_ptr += 768;
+ break;
+ case 0x03:
+ /* Entry in palette of each color */
+ if( st.eos(end,16) ) return 1;
+ for( int i=4; --i>=0; ) palette[i] = st.get_bits(4);
+//zmsgs("palette %d %d %d %d\n", palette[0], palette[1], palette[2], palette[3]);
+ break;
+ case 0x84:
+ if( st.eos(end,256*8) ) return 1;
+ yuv_alpha = st.input_ptr;
+ st.input_ptr += 256;
+ break;
+ case 0x04:
+ /* Alpha corresponding to each color */
+ if( st.eos(end,16) ) return 1;
+ for( int i=4; --i>=0; ) alpha[i] = st.get_bits(4)*255 / 15;
+ //got_alpha = 1;
+//zmsgs("alphas %d %d %d %d\n", alpha[0], alpha[1], alpha[2], alpha[3]);
+#if 0
+ alpha[3] = 0xff; alpha[2] = 0x80;
+ alpha[1] = 0x40; alpha[0] = 0x00;
+#endif
+ break;
+ case 0x85:
+ is_8bit = 1; // fall thru
+ case 0x05:
+ /* Extent of image on screen */
+ if( st.eos(end,48) ) return 1;
+ x1 = st.get_bits(12) & 0x7ff;
+ x2 = st.get_bits(12) + 1;
+ y1 = st.get_bits(12) & 0x7ff;
+ y2 = st.get_bits(12) + 1;
+ ww = x2 - x1;
+ hh = y2 - y1;
+//zmsgs("20 x1=%d x2=%d y1=%d y2=%d, %dx%d\n",
+// x1, x2, y1, y2, ww, hh);
+ ww = clip(ww, 1, 2048); hh = clip(hh, 1, 2048);
+ x1 = clip(x1, 0, 2048); x2 = clip(x2, 0, 2048);
+ y1 = clip(y1, 0, 2048); y2 = clip(y2, 0, 2048);
+ break;
+ case 0x86:
+ if( st.eos(end,64) ) return 1;
+ field_offset[0] = st.get_bits(32);
+ field_offset[1] = st.get_bits(32);
+//zmsgs("offsets even=0x%x odd=0x%x\n", field_offset[0], field_offset[1]);
+ break;
+ case 0x06:
+/* offsets of even and odd field in compressed data */
+ if( st.eos(end,32) ) return 1;
+ field_offset[0] = st.get_bits(16);
+ field_offset[1] = st.get_bits(16);
+//zmsgs("offsets even=0x%x odd=0x%x\n", field_offset[0], field_offset[1]);
+ break;
+ case 0xff:
+ done = 1;
+ break;
+ default:
+//zmsgs("unknown type %02x\n", type);
+ return 1;
+ }
+ }
+ }
+
+ if( ww <= 0 || hh <= 0 || !field_offset[0] ) return 1;
+ if( is_8bit && !yuv_palette ) return 1;
+ set_image_size(ww, hh);
+
+ /* Decode image */
+ int x = 0, y = 0;
+ int field = 0;
+
+ while( field < 2 ) {
+ y = field;
+ ptr = data + field_offset[field++];
+ end = data + field_offset[field];
+ st.use_ptr(ptr);
+ x = 0;
+ while( !st.eos(end) ) {
+ int len, y_color, u_color, v_color, a_color;
+ if( !is_8bit ) {
+ if( st.eos(end,4) ) break;
+ uint32_t zcode = st.get_bits(4);
+ if( zcode < 0x4 ) {
+ if( st.eos(end,4) ) break;
+ zcode = (zcode << 4) | st.get_bits(4);
+ if( zcode < 0x10 ) {
+ if( st.eos(end,4) ) break;
+ zcode = (zcode << 4) | st.get_bits(4);
+ if( zcode < 0x40 ) {
+ if( st.eos(end,4) ) break;
+ zcode = (zcode << 4) | st.get_bits(4);
+ if( zcode < 0x4 ) /* carriage return */
+ zcode |= (w - x) << 2;
+ }
+ }
+ }
+ int color = (zcode & 0x3);
+ len = zcode >> 2;
+ int n = w - x;
+ if( n < len ) len = n;
+ int k = 4*palette[color];
+ y_color = zsrc->palette[k+0];
+ u_color = zsrc->palette[k+1];
+ v_color = zsrc->palette[k+2];
+ a_color = alpha[color];
+ }
+ else {
+ if( st.eos(end,1) ) break;
+ int has_run = st.get_bits(1);
+ if( st.eos(end,1) ) break;
+ int sz = st.get_bits(1) ? 8 : 2;
+ if( st.eos(end,sz) ) break;
+ int color = st.get_bits(sz);
+ if( has_run ) {
+ if( st.eos(end,1) ) break;
+ if( st.get_bits(1) ) {
+ if( st.eos(end,7) ) break;
+ int n = st.get_bits(7);
+ len = n == 0 ? w - x : n + 9;
+ }
+ else {
+ if( st.eos(end,3) ) break;
+ len = st.get_bits(3) + 2;
+ }
+ }
+ else
+ len = 1;
+ int k = 3*color;
+ y_color = yuv_palette[k+0];
+ u_color = yuv_palette[k+1];
+ v_color = yuv_palette[k+2];
+ a_color = yuv_alpha ? yuv_alpha[color] : 0xff;
+ }
+//zmsgs("%d, 0x%02x 0x%02x 0x%02x %02x\n",len,y_color,u_color,v_color,a_color);
+ int ofs = y * w + x;
+ memset(image_y+ofs,y_color,len);
+ memset(image_u+ofs,u_color,len);
+ memset(image_v+ofs,v_color,len);
+ memset(image_a+ofs,a_color,len);
+ if( (x+=len) >= w ) {
+ st.next_byte_align();
+ if( (y+=2) >= h ) break;
+ x = 0;
+ }
+ }
+ }
+
+#if 0
+/* this code computes a threshold (pass 0),
+ * and forces pixels below the threshold (pass 1)
+ * to black and the pixels above to white, as 2x2 yuv macropixels */
+// Normalize image colors
+ float min_h = 360;
+ float max_h = 0;
+ float threshold = 0;
+#define HISTOGRAM_SIZE 1000
+// Decompression coefficients straight out of jpeglib
+#define V_TO_R 1.40200
+#define V_TO_G -0.71414
+
+#define U_TO_G -0.34414
+#define U_TO_B 1.77200
+ uint8_t histogram[HISTOGRAM_SIZE];
+ bzero(histogram, HISTOGRAM_SIZE);
+ for( int pass=0; pass<2; ++pass ) {
+ for( y=0; y<h; ++y ) {
+ for( x=0; x<w; ++x ) {
+ int ofs = y*w + x;
+ if( image_a[ofs]) {
+ uint8_t *y_color = image_y + ofs;
+ uint8_t *a_color = image_a + ofs;
+ uint8_t *u_color = image_u + ofs;
+ uint8_t *v_color = image_v + ofs;
+
+ /* Convert to RGB */
+ float r = (*y_color + *v_color * V_TO_R);
+ float g = (*y_color + *u_color * U_TO_G + *v_color * V_TO_G);
+ float b = (*y_color + *u_color * U_TO_B);
+
+ /* Multiply alpha */
+ /* r = r * *a_color / 0xff; */
+ /* g = g * *a_color / 0xff; */
+ /* b = b * *a_color / 0xff; */
+ /* Convert to HSV */
+ float h, s, v;
+ float min, max, delta;
+ min = ((r < g) ? r : g) < b ? ((r < g) ? r : g) : b;
+ max = ((r > g) ? r : g) > b ? ((r > g) ? r : g) : b;
+ v = max;
+ delta = max - min;
+ if( max != 0 && delta != 0 ) {
+ s = delta / max; /* s */
+ if( r == max )
+ h = (g - b) / delta; /* between yellow & magenta */
+ else if( g == max )
+ h = 2 + (b - r) / delta; /* between cyan & yellow */
+ else
+ h = 4 + (r - g) / delta; /* between magenta & cyan */
+
+ h *= 60; /* degrees */
+ if( h < 0 ) h += 360;
+ }
+ else {
+ /* r = g = b = 0; */ /* s = 0, v is undefined */
+ s = 0;
+ h = -1;
+ }
+ /* int magnitude = (int)(*y_color * *y_color + */
+ /* *u_color * *u_color + */
+ /* *v_color * *v_color); */
+
+ /* Multiply alpha */
+ h = (h * *a_color) / 0xff;
+ if( pass == 0 ) {
+ ++histogram[(int)h];
+ if( h < min_h ) min_h = h;
+ if( h > max_h ) max_h = h;
+ }
+ else {
+// Set new color in a 2x2 pixel block
+ if( h > threshold ) {
+ *y_color = 0xff;
+ }
+ else {
+ *y_color = 0;
+ }
+
+ *u_color = 0x80;
+ *v_color = 0x80;
+ *a_color = 0xff;
+ }
+ }
+ }
+ }
+
+ if( pass == 0 ) {
+ /* int hist_total = 0; */
+ /* for( i=0; i<HISTOGRAM_SIZE; ++i ) */
+ /* hist_total += histogram[i]; */
+ /* int hist_count = 0; */
+ /* for( i=0; i<HISTOGRAM_SIZE; ++i ) { */
+ /* hist_count += histogram[i]; */
+ /* if( hist_count > hist_total/3 ) { */
+ /* threshold = i; */
+ /* break; */
+ /* } */
+ /* } */
+ threshold = (min_h + max_h) / 2;
+// threshold = 324;
+//zmsgs("min_h=%f max_h=%f threshold=%f\n", min_h, max_h, threshold);
+ }
+ }
+/* end b/w threshold code */
+#endif
+
+//zmsgs("coords: %d,%d - %d,%d size: %d,%d start_time=%d end_time=%d\n",
+// x1, y1, x2, y2, w, h, start_time, stop_time);
+ return 0;
+}
+
+void zvideo_t::
+overlay_subtitle(zsubtitle_t *subtitle)
+{
+ uint8_t *img_y = subtitle->image_y;
+ uint8_t *img_u = subtitle->image_u;
+ uint8_t *img_v = subtitle->image_v;
+ uint8_t *img_a = subtitle->image_a;
+
+ if( !img_y || !img_u || !img_v || !img_a ) return;
+ int iofs = 0;
+ int x1 = subtitle->x1;
+ if( x1 < 0 ) { iofs += -x1; x1 = 0; }
+ int x2 = subtitle->x2;
+ if( x2 > coded_picture_width ) x2 = coded_picture_width;
+ int y1 = subtitle->y1;
+ if( y1 < 0 ) { iofs += -y1*subtitle->w; y1 = 0; }
+ int y2 = subtitle->y2;
+ if( y2 > coded_picture_height ) y2 = coded_picture_height;
+
+ for( int y=y1; y<y2; ++y ) {
+ int oofs = y * coded_picture_width + x1;
+ uint8_t *output_y = subtitle_frame[0] + oofs;
+ oofs = y/2 * chrom_width + x1/2;
+ uint8_t *output_u = subtitle_frame[1] + oofs;
+ uint8_t *output_v = subtitle_frame[2] + oofs;
+ int ofs = iofs + (y-y1) * subtitle->w;
+ uint8_t *input_y = img_y + ofs;
+ uint8_t *input_u = img_u + ofs;
+ uint8_t *input_v = img_v + ofs;
+ uint8_t *input_a = img_a + ofs;
+
+ for( int x=x1; x<x2; ++x ) {
+ int opacity = *input_a;
+ int transparency = 0xff - opacity;
+ *output_y = (*input_y*opacity + *output_y*transparency) / 0xff;
+ ++output_y;
+ if( !(y & 1) && !(x & 1) ) {
+ *output_u = (*input_u*opacity + *output_u*transparency) / 0xff;
+ *output_v = (*input_v*opacity + *output_v*transparency) / 0xff;
+ ++output_u; ++output_v;
+ }
+ ++input_y; ++input_a;
+ ++input_u; ++input_v;
+ }
+ }
+}
+
+void zvideo_t::
+new_output()
+{
+ int sz = coded_picture_width * coded_picture_height;
+ int isz = chrom_width * chrom_height;
+ if( !subtitle_frame[0] ) {
+ subtitle_frame[0] = new uint8_t[sz+8];
+ subtitle_frame[1] = new uint8_t[isz+8];
+ subtitle_frame[2] = new uint8_t[isz+8];
+ }
+
+ memcpy(subtitle_frame[0], output_src[0], sz);
+ memcpy(subtitle_frame[1], output_src[1], isz);
+ memcpy(subtitle_frame[2], output_src[2], isz);
+
+ output_src[0] = subtitle_frame[0];
+ output_src[1] = subtitle_frame[1];
+ output_src[2] = subtitle_frame[2];
+}
+
+int zsubtitle_t::
+decode(video_t *video)
+{
+//zmsgs("id=%d strk=%d i=%d start=%f stop=%f active=%d draw=%d framenow=%f\n",
+// subtitle->id,strk,i,subtitle->start_frametime,subtitle->stop_frametime,
+// subtitle->active,subtitle->draw,frame_now);
+ double framerate = video->frame_rate;
+ if( !framerate ) framerate = 30000./1001;
+ if( !active ) {
+ active = 1;
+ if( decompress_subtitle(video->src) ) return -1;
+ int start_frms = framerate * start_time/1000.;
+ int stop_frms = framerate * stop_time/1000.;
+ start_frame = video->framenum + start_frms;
+ stop_frame = video->framenum + stop_frms;
+ frame_time = video->frame_time;
+ }
+ int life_time = video->frame_time - frame_time;
+ if( life_time > 20*framerate ) return -1; // doesn't live forever
+ if( draw < 0 ) return 0; // never draw
+ if( draw || force > 0 ) return 1; // always draw
+ if( video->seek_time > 2 ) {
+ /* at least 2 consecutive frames shown since last seek */
+ int hold_frames = framerate * SUBTITLE_HOLD_TIME;
+ int early_frame = start_frame-hold_frames;
+ if( video->framenum < early_frame ) return -1; // way too early
+ int late_frame = stop_frame+hold_frames;
+ if( video->framenum > late_frame ) return -1; // way too late
+ }
+ if( video->framenum < start_frame ) return 0; // too early
+ if( video->framenum < stop_frame ) return 1; // not too late
+ return 0;
+}
+
+void zvideo_t::
+decode_subtitle()
+{
+ if( !output_src[0] ) return;
+ int copied = 0;
+ for( int strk=0; strk<src->subtitle_tracks(); ++strk ) {
+ strack_t *strack = src->strack[strk];
+ if( !strack ) continue;
+ if( strack->video && strack->video != this ) continue;
+ strack->rwlock.enter();
+ int i = 0;
+ while( i < strack->total_subtitles ) {
+ subtitle_t *subtitle = strack->subtitles[i];
+ int ret = -1;
+ if( strack->id == subtitle_track || subtitle->force ) {
+ if( (ret=subtitle->decode(this)) > 0 ) {
+ if( !copied ) { copied = 1; new_output(); }
+ overlay_subtitle(subtitle);
+ }
+ }
+ if( ret < 0 ) {
+ strack->del_subtitle(subtitle, -1);
+ if( subtitle->force < 0 ) delete subtitle;
+ continue;
+ }
+ ++i;
+ }
+ strack->rwlock.leave();
+ }
+}
+
+int zmpeg3_t::
+display_subtitle(int stream, int sid, int id,
+ uint8_t *yp, uint8_t *up, uint8_t *vp, uint8_t *ap,
+ int x, int y, int w, int h, double start_msecs, double stop_msecs)
+{
+ int ret = 1;
+ if( stream >= 0 && stream < total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ video_t *video = vtrk->video;
+ strack_t *strack = create_strack(sid, video);
+ strack->rwlock.write_enter();
+ subtitle_t *subtitle = 0;
+ for( int i=strack->total_subtitles; --i>=0 && !subtitle; ) {
+ subtitle_t *sp = strack->subtitles[i];
+ if( sp->id == id ) subtitle = sp;
+ }
+ if( !subtitle ) {
+ subtitle = new subtitle_t(id, w, h);
+ if( strack->append_subtitle(subtitle,0) ) {
+ delete subtitle; subtitle = 0;
+ }
+ }
+ else
+ subtitle->set_image_size(w, h);
+ if( subtitle ) {
+ subtitle->x1 = x; subtitle->x2 = x + w;
+ subtitle->y1 = y; subtitle->y2 = y + h;
+ int sz = w * h;
+ if( yp ) memcpy(subtitle->image_y, yp, sz);
+ else memset(subtitle->image_y, 0xff, sz);
+ if( up ) memcpy(subtitle->image_u, up, sz);
+ else memset(subtitle->image_u, 0x80, sz);
+ if( vp ) memcpy(subtitle->image_v, vp, sz);
+ else memset(subtitle->image_v, 0x80, sz);
+ if( ap ) memcpy(subtitle->image_a, ap, sz);
+ else memset(subtitle->image_a, 0xff, sz);
+ subtitle->start_time = start_msecs;
+ subtitle->stop_time = stop_msecs;
+ double framerate = vtrk->frame_rate;
+ if( !framerate ) framerate = 30000./1001;
+ int start_frms = framerate * start_msecs/1000.;
+ int stop_frms = framerate * stop_msecs/1000.;
+ subtitle->start_frame = video->framenum + start_frms;
+ if( stop_msecs ) {
+ subtitle->stop_frame = video->framenum + stop_frms;
+ subtitle->frame_time = video->frame_time;
+ }
+ else // lives forever
+ subtitle->frame_time = subtitle->stop_frame = INT_MAX;
+ subtitle->force = -1;
+ subtitle->draw = 0;
+ ret = 0;
+ }
+ strack->rwlock.write_leave();
+ }
+ return ret;
+}
+
+int zmpeg3_t::
+delete_subtitle(int stream, int sid, int id)
+{
+ int result = 1;
+ if( stream >= 0 && stream < total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ video_t *vid = vtrk->video;
+ strack_t *strack = get_strack_id(sid, vid);
+ if( strack != 0 ) {
+ strack->rwlock.write_enter();
+ for( int i=strack->total_subtitles; --i>=0; ) {
+ subtitle_t *sp = strack->subtitles[i];
+ if( sp->id == id ) {
+ strack->del_subtitle(i);
+ if( sp->force < 0 ) delete sp;
+ result = 0;
+ break;
+ }
+ }
+ strack->rwlock.write_leave();
+ }
+ }
+ return result;
+}
+