dvb_filter.c

Go to the documentation of this file.

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/string.h>
#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);