#include #include #include #include "dvb_filter.h" #if 0 static unsigned int bitrates[3][16] = {{0,32,64,96,128,160,192,224,256,288,320,352,384,416,448,0}, {0,32,48,56,64,80,96,112,128,160,192,224,256,320,384,0}, {0,32,40,48,56,64,80,96,112,128,160,192,224,256,320,0}}; #endif static u32 freq[4] = {480, 441, 320, 0}; static unsigned int ac3_bitrates[32] = {32,40,48,56,64,80,96,112,128,160,192,224,256,320,384,448,512,576,640, 0,0,0,0,0,0,0,0,0,0,0,0,0}; static u32 ac3_frames[3][32] = {{64,80,96,112,128,160,192,224,256,320,384,448,512,640,768,896,1024, 1152,1280,0,0,0,0,0,0,0,0,0,0,0,0,0}, {69,87,104,121,139,174,208,243,278,348,417,487,557,696,835,975,1114, 1253,1393,0,0,0,0,0,0,0,0,0,0,0,0,0}, {96,120,144,168,192,240,288,336,384,480,576,672,768,960,1152,1344, 1536,1728,1920,0,0,0,0,0,0,0,0,0,0,0,0,0}}; #if 0 static void setup_ts2pes(ipack *pa, ipack *pv, u16 *pida, u16 *pidv, void (*pes_write)(u8 *buf, int count, void *data), void *priv) { dvb_filter_ipack_init(pa, IPACKS, pes_write); dvb_filter_ipack_init(pv, IPACKS, pes_write); pa->pid = pida; pv->pid = pidv; pa->data = priv; pv->data = priv; } #endif #if 0 static void ts_to_pes(ipack *p, u8 *buf) // don't need count (=188) { u8 off = 0; if (!buf || !p ){ printk("NULL POINTER IDIOT\n"); return; } if (buf[1]&PAY_START) { if (p->plength == MMAX_PLENGTH-6 && p->found>6){ p->plength = p->found-6; p->found = 0; send_ipack(p); dvb_filter_ipack_reset(p); } } if (buf[3] & ADAPT_FIELD) { // adaptation field? off = buf[4] + 1; if (off+4 > 187) return; } dvb_filter_instant_repack(buf+4+off, TS_SIZE-4-off, p); } #endif #if 0 /* needs 5 byte input, returns picture coding type*/ static int read_picture_header(u8 *headr, struct mpg_picture *pic, int field, int pr) { u8 pct; if (pr) printk( "Pic header: "); pic->temporal_reference[field] = (( headr[0] << 2 ) | (headr[1] & 0x03) )& 0x03ff; if (pr) printk( " temp ref: 0x%04x", pic->temporal_reference[field]); pct = ( headr[1] >> 2 ) & 0x07; pic->picture_coding_type[field] = pct; if (pr) { switch(pct){ case I_FRAME: printk( " I-FRAME"); break; case B_FRAME: printk( " B-FRAME"); break; case P_FRAME: printk( " P-FRAME"); break; } } pic->vinfo.vbv_delay = (( headr[1] >> 5 ) | ( headr[2] << 3) | ( (headr[3] & 0x1F) << 11) ) & 0xffff; if (pr) printk( " vbv delay: 0x%04x", pic->vinfo.vbv_delay); pic->picture_header_parameter = ( headr[3] & 0xe0 ) | ((headr[4] & 0x80) >> 3); if ( pct == B_FRAME ){ pic->picture_header_parameter |= ( headr[4] >> 3 ) & 0x0f; } if (pr) printk( " pic head param: 0x%x", pic->picture_header_parameter); return pct; } #endif #if 0 /* needs 4 byte input */ static int read_gop_header(u8 *headr, struct mpg_picture *pic, int pr) { if (pr) printk("GOP header: "); pic->time_code = (( headr[0] << 17 ) | ( headr[1] << 9) | ( headr[2] << 1 ) | (headr[3] &0x01)) & 0x1ffffff; if (pr) printk(" time: %d:%d.%d ", (headr[0]>>2)& 0x1F, ((headr[0]<<4)& 0x30)| ((headr[1]>>4)& 0x0F), ((headr[1]<<3)& 0x38)| ((headr[2]>>5)& 0x0F)); if ( ( headr[3] & 0x40 ) != 0 ){ pic->closed_gop = 1; } else { pic->closed_gop = 0; } if (pr) printk("closed: %d", pic->closed_gop); if ( ( headr[3] & 0x20 ) != 0 ){ pic->broken_link = 1; } else { pic->broken_link = 0; } if (pr) printk(" broken: %d\n", pic->broken_link); return 0; } #endif #if 0 /* needs 8 byte input */ static int read_sequence_header(u8 *headr, struct dvb_video_info *vi, int pr) { int sw; int form = -1; if (pr) printk("Reading sequence header\n"); vi->horizontal_size = ((headr[1] &0xF0) >> 4) | (headr[0] << 4); vi->vertical_size = ((headr[1] &0x0F) << 8) | (headr[2]); sw = (int)((headr[3]&0xF0) >> 4) ; switch( sw ){ case 1: if (pr) printk("Videostream: ASPECT: 1:1"); vi->aspect_ratio = 100; break; case 2: if (pr) printk("Videostream: ASPECT: 4:3"); vi->aspect_ratio = 133; break; case 3: if (pr) printk("Videostream: ASPECT: 16:9"); vi->aspect_ratio = 177; break; case 4: if (pr) printk("Videostream: ASPECT: 2.21:1"); vi->aspect_ratio = 221; break; case 5 ... 15: if (pr) printk("Videostream: ASPECT: reserved"); vi->aspect_ratio = 0; break; default: vi->aspect_ratio = 0; return -1; } if (pr) printk(" Size = %dx%d",vi->horizontal_size,vi->vertical_size); sw = (int)(headr[3]&0x0F); switch ( sw ) { case 1: if (pr) printk(" FRate: 23.976 fps"); vi->framerate = 23976; form = -1; break; case 2: if (pr) printk(" FRate: 24 fps"); vi->framerate = 24000; form = -1; break; case 3: if (pr) printk(" FRate: 25 fps"); vi->framerate = 25000; form = VIDEO_MODE_PAL; break; case 4: if (pr) printk(" FRate: 29.97 fps"); vi->framerate = 29970; form = VIDEO_MODE_NTSC; break; case 5: if (pr) printk(" FRate: 30 fps"); vi->framerate = 30000; form = VIDEO_MODE_NTSC; break; case 6: if (pr) printk(" FRate: 50 fps"); vi->framerate = 50000; form = VIDEO_MODE_PAL; break; case 7: if (pr) printk(" FRate: 60 fps"); vi->framerate = 60000; form = VIDEO_MODE_NTSC; break; } vi->bit_rate = (headr[4] << 10) | (headr[5] << 2) | (headr[6] & 0x03); vi->vbv_buffer_size = (( headr[6] & 0xF8) >> 3 ) | (( headr[7] & 0x1F )<< 5); if (pr){ printk(" BRate: %d Mbit/s",4*(vi->bit_rate)/10000); printk(" vbvbuffer %d",16*1024*(vi->vbv_buffer_size)); printk("\n"); } vi->video_format = form; return 0; } #endif #if 0 static int get_vinfo(u8 *mbuf, int count, struct dvb_video_info *vi, int pr) { u8 *headr; int found = 0; int c = 0; while (found < 4 && c+4 < count){ u8 *b; b = mbuf+c; if ( b[0] == 0x00 && b[1] == 0x00 && b[2] == 0x01 && b[3] == 0xb3) found = 4; else { c++; } } if (! found) return -1; c += 4; if (c+12 >= count) return -1; headr = mbuf+c; if (read_sequence_header(headr, vi, pr) < 0) return -1; vi->off = c-4; return 0; } #endif #if 0 static int get_ainfo(u8 *mbuf, int count, struct dvb_audio_info *ai, int pr) { u8 *headr; int found = 0; int c = 0; int fr = 0; while (found < 2 && c < count){ u8 b[2]; memcpy( b, mbuf+c, 2); if ( b[0] == 0xff && (b[1] & 0xf8) == 0xf8) found = 2; else { c++; } } if (!found) return -1; if (c+3 >= count) return -1; headr = mbuf+c; ai->layer = (headr[1] & 0x06) >> 1; if (pr) printk("Audiostream: Layer: %d", 4-ai->layer); ai->bit_rate = bitrates[(3-ai->layer)][(headr[2] >> 4 )]*1000; if (pr){ if (ai->bit_rate == 0) printk(" Bit rate: free"); else if (ai->bit_rate == 0xf) printk(" BRate: reserved"); else printk(" BRate: %d kb/s", ai->bit_rate/1000); } fr = (headr[2] & 0x0c ) >> 2; ai->frequency = freq[fr]*100; if (pr){ if (ai->frequency == 3) printk(" Freq: reserved\n"); else printk(" Freq: %d kHz\n",ai->frequency); } ai->off = c; return 0; } #endif int dvb_filter_get_ac3info(u8 *mbuf, int count, struct dvb_audio_info *ai, int pr) { u8 *headr; int found = 0; int c = 0; u8 frame = 0; int fr = 0; while ( !found && c < count){ u8 *b = mbuf+c; if ( b[0] == 0x0b && b[1] == 0x77 ) found = 1; else { c++; } } if (!found) return -1; if (pr) printk("Audiostream: AC3"); ai->off = c; if (c+5 >= count) return -1; ai->layer = 0; // 0 for AC3 headr = mbuf+c+2; frame = (headr[2]&0x3f); ai->bit_rate = ac3_bitrates[frame >> 1]*1000; if (pr) printk(" BRate: %d kb/s", (int) ai->bit_rate/1000); ai->frequency = (headr[2] & 0xc0 ) >> 6; fr = (headr[2] & 0xc0 ) >> 6; ai->frequency = freq[fr]*100; if (pr) printk (" Freq: %d Hz\n", (int) ai->frequency); ai->framesize = ac3_frames[fr][frame >> 1]; if ((frame & 1) && (fr == 1)) ai->framesize++; ai->framesize = ai->framesize << 1; if (pr) printk (" Framesize %d\n",(int) ai->framesize); return 0; } EXPORT_SYMBOL(dvb_filter_get_ac3info); #if 0 static u8 *skip_pes_header(u8 **bufp) { u8 *inbuf = *bufp; u8 *buf = inbuf; u8 *pts = NULL; int skip = 0; static const int mpeg1_skip_table[16] = { 1, 0xffff, 5, 10, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff }; if ((inbuf[6] & 0xc0) == 0x80){ /* mpeg2 */ if (buf[7] & PTS_ONLY) pts = buf+9; else pts = NULL; buf = inbuf + 9 + inbuf[8]; } else { /* mpeg1 */ for (buf = inbuf + 6; *buf == 0xff; buf++) if (buf == inbuf + 6 + 16) { break; } if ((*buf & 0xc0) == 0x40) buf += 2; skip = mpeg1_skip_table [*buf >> 4]; if (skip == 5 || skip == 10) pts = buf; else pts = NULL; buf += mpeg1_skip_table [*buf >> 4]; } *bufp = buf; return pts; } #endif #if 0 static void initialize_quant_matrix( u32 *matrix ) { int i; matrix[0] = 0x08101013; matrix[1] = 0x10131616; matrix[2] = 0x16161616; matrix[3] = 0x1a181a1b; matrix[4] = 0x1b1b1a1a; matrix[5] = 0x1a1a1b1b; matrix[6] = 0x1b1d1d1d; matrix[7] = 0x2222221d; matrix[8] = 0x1d1d1b1b; matrix[9] = 0x1d1d2020; matrix[10] = 0x22222526; matrix[11] = 0x25232322; matrix[12] = 0x23262628; matrix[13] = 0x28283030; matrix[14] = 0x2e2e3838; matrix[15] = 0x3a454553; for ( i = 16 ; i < 32 ; i++ ) matrix[i] = 0x10101010; } #endif #if 0 static void initialize_mpg_picture(struct mpg_picture *pic) { int i; /* set MPEG1 */ pic->mpeg1_flag = 1; pic->profile_and_level = 0x4A ; /* MP@LL */ pic->progressive_sequence = 1; pic->low_delay = 0; pic->sequence_display_extension_flag = 0; for ( i = 0 ; i < 4 ; i++ ){ pic->frame_centre_horizontal_offset[i] = 0; pic->frame_centre_vertical_offset[i] = 0; } pic->last_frame_centre_horizontal_offset = 0; pic->last_frame_centre_vertical_offset = 0; pic->picture_display_extension_flag[0] = 0; pic->picture_display_extension_flag[1] = 0; pic->sequence_header_flag = 0; pic->gop_flag = 0; pic->sequence_end_flag = 0; } #endif #if 0 static void mpg_set_picture_parameter( int32_t field_type, struct mpg_picture *pic ) { int16_t last_h_offset; int16_t last_v_offset; int16_t *p_h_offset; int16_t *p_v_offset; if ( pic->mpeg1_flag ){ pic->picture_structure[field_type] = VIDEO_FRAME_PICTURE; pic->top_field_first = 0; pic->repeat_first_field = 0; pic->progressive_frame = 1; pic->picture_coding_parameter = 0x000010; } /* Reset flag */ pic->picture_display_extension_flag[field_type] = 0; last_h_offset = pic->last_frame_centre_horizontal_offset; last_v_offset = pic->last_frame_centre_vertical_offset; if ( field_type == FIRST_FIELD ){ p_h_offset = pic->frame_centre_horizontal_offset; p_v_offset = pic->frame_centre_vertical_offset; *p_h_offset = last_h_offset; *(p_h_offset + 1) = last_h_offset; *(p_h_offset + 2) = last_h_offset; *p_v_offset = last_v_offset; *(p_v_offset + 1) = last_v_offset; *(p_v_offset + 2) = last_v_offset; } else { pic->frame_centre_horizontal_offset[3] = last_h_offset; pic->frame_centre_vertical_offset[3] = last_v_offset; } } #endif #if 0 static void init_mpg_picture( struct mpg_picture *pic, int chan, int32_t field_type) { pic->picture_header = 0; pic->sequence_header_data = ( INIT_HORIZONTAL_SIZE << 20 ) | ( INIT_VERTICAL_SIZE << 8 ) | ( INIT_ASPECT_RATIO << 4 ) | ( INIT_FRAME_RATE ); pic->mpeg1_flag = 0; pic->vinfo.horizontal_size = INIT_DISP_HORIZONTAL_SIZE; pic->vinfo.vertical_size = INIT_DISP_VERTICAL_SIZE; pic->picture_display_extension_flag[field_type] = 0; pic->pts_flag[field_type] = 0; pic->sequence_gop_header = 0; pic->picture_header = 0; pic->sequence_header_flag = 0; pic->gop_flag = 0; pic->sequence_end_flag = 0; pic->sequence_display_extension_flag = 0; pic->last_frame_centre_horizontal_offset = 0; pic->last_frame_centre_vertical_offset = 0; pic->channel = chan; } #endif void dvb_filter_pes2ts_init(struct dvb_filter_pes2ts *p2ts, unsigned short pid, dvb_filter_pes2ts_cb_t *cb, void *priv) { unsigned char *buf=p2ts->buf; buf[0]=0x47; buf[1]=(pid>>8); buf[2]=pid&0xff; p2ts->cc=0; p2ts->cb=cb; p2ts->priv=priv; } EXPORT_SYMBOL(dvb_filter_pes2ts_init); int dvb_filter_pes2ts(struct dvb_filter_pes2ts *p2ts, unsigned char *pes, int len, int payload_start) { unsigned char *buf=p2ts->buf; int ret=0, rest; //len=6+((pes[4]<<8)|pes[5]); if (payload_start) buf[1]|=0x40; else buf[1]&=~0x40; while (len>=184) { buf[3]=0x10|((p2ts->cc++)&0x0f); memcpy(buf+4, pes, 184); if ((ret=p2ts->cb(p2ts->priv, buf))) return ret; len-=184; pes+=184; buf[1]&=~0x40; } if (!len) return 0; buf[3]=0x30|((p2ts->cc++)&0x0f); rest=183-len; if (rest) { buf[5]=0x00; if (rest-1) memset(buf+6, 0xff, rest-1); } buf[4]=rest; memcpy(buf+5+rest, pes, len); return p2ts->cb(p2ts->priv, buf); } EXPORT_SYMBOL(dvb_filter_pes2ts);