source: svn/trunk/newcon3bcm2_21bu/BSEAV/lib/bcmplayer/utils/mkbtpfile.c

/***************************************************************************
 *     Copyright (c) 2005-2009, Broadcom Corporation
 *     All Rights Reserved
 *     Confidential Property of Broadcom Corporation
 *
 *  THIS SOFTWARE MAY ONLY BE USED SUBJECT TO AN EXECUTED SOFTWARE LICENSE
 *  AGREEMENT  BETWEEN THE USER AND BROADCOM.  YOU HAVE NO RIGHT TO USE OR
 *  EXPLOIT THIS MATERIAL EXCEPT SUBJECT TO THE TERMS OF SUCH AN AGREEMENT.
 *
 * $brcm_Workfile: mkbtpfile.c $
 * $brcm_Revision: 3 $
 * $brcm_Date: 7/13/09 10:26a $
 *
 * Module Description: Transport Stream Index Player
 *
 * Revision History:
 *
 * $brcm_Log: /BSEAV/lib/bcmplayer/utils/mkbtpfile.c $
 * 
 * 3   7/13/09 10:26a erickson
 * PR56423: clarify eBpPlayDecoderGOPTrick, remove unused
 * eBpPlaySTCFastForward
 * 
 * 2   4/30/07 1:10p erickson
 * PR30310: expanded supported playmodes
 * 
 * Irvine_BSEAVSW_Devel/7   1/6/06 10:16a erickson
 * PR17108: updated for magnum basemodules and 64 bit cpus
 * 
 * Irvine_BSEAVSW_Devel/6   8/9/05 2:33p erickson
 * PR16208: uclibc gets doesn't overwrite n with 0
 * 
 * Irvine_BSEAVSW_Devel/5   6/14/05 7:50a erickson
 * PR15861: added host trick mode support
 * 
 * Irvine_BSEAVSW_Devel/4   3/17/05 9:52a erickson
 * PR13202: added speed and start location prompts
 * 
 * Irvine_BSEAVSW_Devel/3   3/9/05 12:28p erickson
 * PR13202: added zeroByteCountEnd logic
 * 
 ***********************************************************/
#include "bstd.h"
#include "bkni.h"
#include "bcmplayer.h"
#include <stdio.h>
#include <assert.h>
#include <string.h>
#include <stdlib.h>

static struct {
        eBpPlayModeParam playMode;
        const char *name;
} g_playModes[] = {
        {eBpPlayNormal, "normal"},
        {eBpPlayNormalByFrames, "normalbyframes"},
        {eBpPlayI, "i"},
        {eBpPlaySkipB, "skipb"},
        {eBpPlayIP, "ip"},
        {eBpPlayBrcm, "brcm"},
        {eBpPlayDecoderGOPTrick, "gop"},
};

static eBpPlayModeParam g_convert_playModeStr(const char *name) 
{
        unsigned i = 0;
        for (i=0;i<sizeof(g_playModes)/sizeof(g_playModes[0]);i++) {
                if (!strcasecmp(name, g_playModes[i].name)) {
                        return g_playModes[i].playMode;
                }
        }
        printf("invalid mode: %s\n", name);
        return eBpPlayNormal;
}

static const char *g_convert_playMode(eBpPlayModeParam playMode) 
{
        unsigned i = 0;
        for (i=0;i<sizeof(g_playModes)/sizeof(g_playModes[0]);i++) {
                if (playMode == g_playModes[i].playMode) {
                        return g_playModes[i].name;
                }
        }
        return "unknown";
}

int main(int argc, char **argv)
{
        BNAV_Player_Handle player;
        BNAV_Player_Settings cfg;
        BNAV_Player_PlayMode playMode;
        BNAV_Player_PlayEntry bcmEntry;
        FILE *mpegfile, *indexfile, *outfile;
        int rc;
        unsigned char packetBuffer[188];
        unsigned short pid;
        char buf[256];
        int speed = -1, startindex = 200;
        
        BKNI_Init();
        BDBG_Init();
        BDBG_SetModuleLevel("bcmplayer", BDBG_eMsg);

        if (argc < 5) {
                printf("usage: mkbtpfile datafile indexfile pid outputfile\n");
                return 1;
        }
                 
        mpegfile = fopen(argv[1], "r");
        indexfile = fopen(argv[2], "r");
        assert(mpegfile);
        assert(indexfile);
        
        pid = strtoul(argv[3], NULL, 0);
        outfile = fopen(argv[4], "w+");
        
        printf("Select play mode ([Brcm], I, IP, SkipB, GOP, Normal, NormalByFrames):\n");
        fgets(buf, 256, stdin);
        playMode.playMode = eBpPlayBrcm;
        if (*buf && *buf != '\n') {
                buf[strlen(buf)-1] = 0;
                playMode.playMode = g_convert_playModeStr(buf);
        }
                
        printf("Enter mode modifier (default: -1):\n");
        fgets(buf, 256, stdin);
        if (*buf && *buf != '\n')
                speed = atoi(buf);
        
        printf("Enter start index (default: 200):\n");
        fgets(buf, 256, stdin);
        if (*buf && *buf != '\n')
                startindex = atoi(buf);
                
        printf("Trick mode: %s, mode %d, start %d\n",
                g_convert_playMode(playMode.playMode), speed, startindex);
        
        BNAV_Player_GetDefaultSettings(&cfg);
        cfg.videoPid = pid;
        cfg.filePointer = indexfile;
        cfg.readCb = (BP_READ_CB)fread;
        cfg.tellCb = (BP_TELL_CB)ftell;
        cfg.seekCb = (BP_SEEK_CB)fseek;
        
        rc = BNAV_Player_Open(&player, &cfg);
        assert(!rc);

        playMode.playModeModifier = speed;
        playMode.loopMode = eBpSinglePlay;
        playMode.disableExtraBOptimization = 0;
        BNAV_Player_SetPlayMode(player, &playMode);

        BNAV_Player_SetCurrentIndex(player, startindex);
                
        while (!BNAV_Player_GetNextPlayEntry(player, &bcmEntry, packetBuffer)) {
                if (bcmEntry.isInsertedPacket) {
                        fwrite(packetBuffer, 1, 188, outfile);
                }
                else {
#define BUFSIZE (64*1024)
                        char buf[BUFSIZE];
                        int read = 0;
                        int zero_at = bcmEntry.byteCount - bcmEntry.zeroByteCountEnd; /* index
                                at which we start zeroing */
                                
                        if (bcmEntry.zeroByteCountEnd)
                                printf("zero_at %d\n", zero_at);
                        
                        fseek(mpegfile, bcmEntry.startOffset, SEEK_SET);
                        while ((unsigned long)read < bcmEntry.byteCount) {
                                int n = bcmEntry.byteCount - read; /* amount we can/should read */
                                if (n > BUFSIZE)
                                        n = BUFSIZE;
                                if (fread(buf, 1, n, mpegfile) != (size_t)n)
                                        return -1;
                                        
                                /* zero out unwanted data */
                                if (read+n >= (int)zero_at && bcmEntry.zeroByteCountEnd) {
                                        if (read > zero_at) {
                                                printf("zeroing %d, %d\n", read, n);
                                                memset(&buf[read], 0, n); 
                                        }
                                        else {
                                                printf("zeroing %d, %d\n", zero_at-read, n-(zero_at-read));
                                                memset(&buf[zero_at-read], 0, n - (zero_at-read)); 
                                        }
                                }
                                
                                fwrite(buf, 1, n, outfile);
                                read += n;
                        }
                }
        }

        fclose(mpegfile);
        fclose(indexfile);
        fclose(outfile);
        return 0;
}