mpeg2videoes.c

Sun公司Dream项目

/*
 * The contents of this file are subject to the terms
 * of the Common Development and Distribution License
 * (the "License").  You may not use this file except
 * in compliance with the License.
 *
 * You can obtain a copy of the license at
 * http://www.opensource.org/licenses/cddl1.php
 * See the License for the specific language governing
 * permissions and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL
 * HEADER in each file and include the License file at
 * http://www.opensource.org/licenses/cddl1.php.  If 
 * applicable, add the following below this CDDL HEADER, 
 * with the fields enclosed by brackets "[]" replaced 
 * with your own identifying information: 
 * Portions Copyright [yyyy]
 * [name of copyright owner]
 */ 

/*
 * $(@)Mpeg2VideoEs.c $Revision: 1.2 $ $Date: 2006/07/15 00:02:39 $
 * 
 * Copyright 2006 Sun Microsystems, Inc. All Rights Reserved.
 */
/* This file would contain all the parser utilities of a Mpeg2 Video 
Elementary Stream */

#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdio.h>

#include "cobjs/Log.h"
#include "cobjs/Macros.h"
#include "cobjs/Types.h"
#include "cobjs/RetCode.h"

#include "mmp/Mmp.h"
#include "mmp/Context.h"
#include "mmp/Pipe.h"
#include "mmp/Mpeg2VideoEs.h"

//#include "util/utLog.h"
#include <assert.h>
#include <synch.h>
#include "metadata.h"


static size_t  mpeg2VideoEsMaxChunk(void *instp);

static void    mpeg2VideoEsRecover(void *instp);

static RetCode mpeg2VideoEsParse(void *instp, MmpContextObject *cop, Pipe pipe) ;

static u64  adjustIntraQuantMatrix(u64 *, u64);

RetCode mpeg2VideoSetOutputFile(Mpeg2VideoEs mpeg2VideoEs, int filep);

extern metadata *init_metadata(char *meta_name, int speed);

u64
adjustIntraQuantMatrix(u64 *p , u64 initbit){

  u64 nextbit ;
  int i;
  
  for(i = 0 ;  i < 8 ; i++){
    nextbit = *p & 0x0000000000000001 ;
    *p = *p >> 1 ;
    *p = (*p) | ( initbit << 63) ; 
    initbit = nextbit ;
    if(i < 7) 
      p++ ;
  }
  return nextbit ;
}
    
  
  
Mpeg2VideoEsStartCode *
skipToNextStartCode(Pipe pipe){
  Mpeg2VideoEsStartCode *ppp ;

  while(1){
    ppp = PIPE_PEEK(pipe, Mpeg2VideoEsStartCode) ;
    if(IS_MPEG2_VIDEO_ES_START_CODE_PREFIX(ppp))
      break ;
    else 
      PIPE_SKIP(pipe, u8) ;
  }
  return ppp ;
    
}


Mpeg2VideoEsStartCode *
copyTillNextStartCode(Pipe pipe, FILE *file){
  Mpeg2VideoEsStartCode *ppp ;

  while(1){
    ppp = PIPE_PEEK(pipe, Mpeg2VideoEsStartCode) ;
    if(IS_MPEG2_VIDEO_ES_START_CODE_PREFIX(ppp))
      break ;
    else{ 
      assert(fwrite(ppp,1,sizeof(u8),file));
      PIPE_SKIP(pipe, u8) ;
    }
  }
  return ppp ;
    
}

/*



Mpeg2VideoEsStartCode *
skipToNextStartCode(Pipe pipe){
  Mpeg2VideoEsStartCode *ppp ;

  while(1){
    ppp = PIPE_PEEK(pipe, Mpeg2VideoEsStartCode) ;
    if(IS_MPEG2_VIDEO_ES_START_CODE_PREFIX(ppp))
      break ;
    else 
      PIPE_SKIP(pipe, u8) ;
  }
  return ppp ;
    
}

PipePosition 
copyTillNextStartCode(Pipe pipe,Mpeg2VideoEsStartCode **ppp, FILE *file){

 PipePosition pos =0; 
  while(1){
    *ppp = PIPE_PEEK(pipe, Mpeg2VideoEsStartCode) ;
    if(IS_MPEG2_VIDEO_ES_START_CODE_PREFIX(*ppp))
      break ;
    else{ 
      assert(fwrite(*ppp,1,sizeof(u8),file));
      PIPE_SKIP(pipe, u8) ;
      pos++;
    }
  }
  return pos;
    
}


*/

static RetCodeId    retCodeId;
FILE *file;
FILE *metadata_file;
Mpeg2VideoEs
mpeg2VideoEsNew(void){
  
  Mpeg2VideoEs      mpeg2VideoEs ;

/*  if(retCodeId == 0){
     retCodeId = retCodeRegisterWithTable(MPEG2VIDEO_ES_CLASSNAME,
     mpeg2VideoEsErrorTable);
  }
 */
  //UT_LOG_MESG(&logp, (&logp, "Mpeg2VideoEsNew : called \n"));


  mpeg2VideoEs  = NEW_ZEROED(struct _Mpeg2VideoEs, 1);
  mpeg2VideoEs->bufBase = mpeg2VideoEs->bufDataptr = (u8 *)malloc(DEMUX_BUF_LEN * sizeof(u8));
  mpeg2VideoEs->bufLimit = (unsigned long*)malloc(sizeof(u8));
  *(mpeg2VideoEs->bufLimit) = DEMUX_BUF_LEN;
  mpeg2VideoEs->bufDatasize = (unsigned long*)malloc(sizeof(u8));
  *(mpeg2VideoEs->bufDatasize) = 0;

  mpeg2VideoEs->po.maxChunk = mpeg2VideoEsMaxChunk ;
  mpeg2VideoEs->po.recognize = mpeg2VideoEsRecognize ;
  mpeg2VideoEs->po.parse  = mpeg2VideoEsParse ;
  mpeg2VideoEs->po.recover = mpeg2VideoEsRecover ;
  mpeg2VideoEs->po.instp  = mpeg2VideoEs ;
  mpeg2VideoEs->context = contextNew();
  mpeg2VideoEs->cop = contextMmpContextObject(mpeg2VideoEs->context);
  mpeg2VideoEs->resourceName = "";
  mpeg2VideoEs->metaName = "";
 
  return mpeg2VideoEs ;
  
}

MmpParserObject *
mpeg2VideoEsParserObject(Mpeg2VideoEs mpeg2VideoEs){
  return &mpeg2VideoEs->po;
}

RetCode
mpeg2VideoSetOutputFile(Mpeg2VideoEs mpeg2VideoEs, int filep){
  /* This sets filedescriptor for writing Video Elementary Stream */
  /*  mpeg2VideoEs->filep = filep;*/
  return RETCODE_SUCCESS ;
}

void mpeg2VideoEsSetResourceMetaName(Mpeg2VideoEs mpeg2VideoEs, char
*resourceName, char* metaName){
    mpeg2VideoEs->resourceName = resourceName;
    mpeg2VideoEs->metaName = metaName;
}

void
mpeg2VideoEsSetCallBack(Mpeg2VideoEs mpeg2VideoEs, Mpeg2VideoEsCallBack callBack,
			void *token){
  //UT_LOG_MESG(&logp, (&logp, "mpeg2VideoEsSetCallBack : called \n"));
  mpeg2VideoEs->callBack = callBack ;
  mpeg2VideoEs->token = token ;
}

static size_t
mpeg2VideoEsMaxChunk(void *instp){
  return MPEG2_VIDEOES_MAX_CHUNK ;
}

static RetCode
mpeg2VideoEsRecognize(void *instp, Pipe pipe){
  Mpeg2VideoEsStartCode          *ppp =  PIPE_PEEK(pipe, Mpeg2VideoEsStartCode);
  
  
  if(IS_MPEG2_VIDEO_ES_START_CODE_PREFIX(ppp)){
    //UT_LOG_MESG(&logp, (&logp, " mpeg2VideoEsRecognize : Video Es recognized \n"));
    return RETCODE_SUCCESS;
  }else{
     //UT_LOG_MESG(&logp, (&logp, " mpeg2VideoEsRecognize : Video Es not recongized \n"));
     return RETCODE_FAILED;
  }
 
}

void
mpeg2VideoEsFree(Mpeg2VideoEs mpeg2VideoEs){
  free(mpeg2VideoEs);
  //UT_LOG_MESG(&logp, (&logp, " mpeg2VideoEsFree : called \n"));
}

static void
mpeg2VideoEsRecover(void *instp){

  Mpeg2VideoEs        mpeg2video = (Mpeg2VideoEs)instp;
  mpeg2video->isDirty = FALSE;
 // UT_LOG_MESG(&logp, (&logp, " mpeg2VideoEsRecover : Called \n"));
  /* will be implemented later */
}



RetCode
mpeg2VideoEsParse(void *instp, MmpContextObject *cop, Pipe pipe){

  Mpeg2VideoEs               mpeg2VideoEs = (Mpeg2VideoEs)instp ;
  Mpeg2VideoEsInfo           info ;
  Mpeg2VideoEsInfo          *infop = &info ;
  Mpeg2VideoEsStartCode     *ppp ;
  int                        seqHdrflag = 0, len, getlen;
  u8                        *buf ;
  RetCode                    retcode = RETCODE_SUCCESS ;
  unsigned                   pid ;
  Boolean                    exitwhile = FALSE;
  unsigned                   startcode ;
  PipePosition               pos, pos2;
  u32                        curlen = 0;
  u8	     				 *tmp;
  

   mpeg2VideoEs->isDirty = TRUE ;
   
   pid = MMP_CONTEXT_PIPETOPID(cop, pipe);
   
   //	resource_name="../../../repository/temp";
   



   //UT_LOG_MESG(&logp, (&logp, " mpeg2VideoEsParse : video Parser called \n"));
	
	create_1X_trickplay_stream(pipe, mpeg2VideoEs->resourceName,
                                mpeg2VideoEs->metaName );
	
	return RETCODE_SUCCESS;
}


Boolean create_1X_trickplay_stream(Pipe pipe, char *resource_name , char *meta_file_name ){

   FileOffset pos;
   Mpeg2VideoEsStartCode     *ppp;
   u8 startcode;
   Mpeg2VideoPictureHeader *picHeader;
   long frameNumber;
//   char *meta_file_name;
   char *output_stream_name;
   metadata *meta;
   FILE* meta_file;
   FILE* file, *temp_file;
   int err;
   int frametype;
   picture_metadata temp;
   long avg_frame_len;