annexb.c

来自「H.264视频编码器(ITU的264编码参考软件)」· C语言 代码 · 共 130 行

/*!
 *************************************************************************************
 * \file annexb.c
 *
 * \brief
 *    Annex B Byte Stream format NAL Unit writing routines
 *
 * \author
 *    Main contributors (see contributors.h for copyright, address and affiliation details)
 *      - Stephan Wenger                  <[e-mail]>
 *************************************************************************************
 */

#include <stdlib.h>
#include <assert.h>

#include "global.h"
#include "nalucommon.h"

static FILE *f = NULL;    // the output file


/*!
 ********************************************************************************************
 * \brief 
 *    Writes a NALU to the Annex B Byte Stream
 *
 * \return
 *    number of bits written
 *
 ********************************************************************************************
*/
int WriteAnnexbNALU (NALU_t *n)
{
  int BitsWritten = 0;

  assert (n != NULL);
  assert (n->forbidden_bit == 0);
//  assert (f != NULL);
  assert (n->startcodeprefix_len == 3 || n->startcodeprefix_len == 4);
// printf("1-the value of xxx is:%d, outlen:%d, encoded_data:0x%x\n", \
   active_pps->slice_group_map_type, *out_len, encoded_data);
 //printf ("WriteAnnexbNALU: writing %d bytes w/ startcode_len %d\n", n->len+1, n->startcodeprefix_len); 
//  if (n->startcodeprefix_len > 3)
//  {
//    putc (0, f);
//    BitsWritten =+ 8;
//  }
//  putc (0, f);
//  putc (0, f);
//  putc (1, f);
  if (n->startcodeprefix_len > 3)
  {
    encoded_data[(*out_len)++]=0;
    BitsWritten =+ 8;
  }
  encoded_data[(*out_len)++]=0;
  encoded_data[(*out_len)++]=0;
  encoded_data[(*out_len)++]=1;
  BitsWritten += 24;
//printf("2-the value of xxx is:%d, outlen:%d, encoded_data:0x%x\n", \
  active_pps->slice_group_map_type, *out_len, encoded_data);
  n->buf[0] =
    n->forbidden_bit << 7      |
    n->nal_reference_idc << 5  |
    n->nal_unit_type;

 //printf ("First Byte %x, nal_ref_idc %x, nal_unit_type %d\n", n->buf[0], n->nal_reference_idc, n->nal_unit_type);
//  if (n->len != fwrite (n->buf, 1, n->len, f))
//  {
//    printf ("Fatal: cannot write %d bytes to bitstream file, exit (-1)\n", n->len);
//    exit (-1);
//  }
  memcpy(encoded_data+(*out_len),n->buf,n->len);
  (*out_len)+=n->len;
  BitsWritten += n->len * 8;
//printf("3-the value of xxx is:%d, outlen:%d, encoded_data:0x%d\n", \
  active_pps->slice_group_map_type, *out_len, encoded_data);
  fflush (f);
#if TRACE
  fprintf (p_trace, "\n\nAnnex B NALU w/ %s startcode, len %d, forbidden_bit %d, nal_reference_idc %d, nal_unit_type %d\n\n",
    n->startcodeprefix_len == 4?"long":"short", n->len, n->forbidden_bit, n->nal_reference_idc, n->nal_unit_type);
  fflush (p_trace);
#endif
  return BitsWritten;
}


/*!
 ********************************************************************************************
 * \brief 
 *    Opens the output file for the bytestream    
 *
 * \param Filename
 *    The filename of the file to be opened
 *
 * \return
 *    none.  Function terminates the program in case of an error
 *
 ********************************************************************************************
*/
void OpenAnnexbFile (char *Filename)
{
//  if ((f = fopen (Filename, "wb")) == NULL)
//  {
//    printf ("Fatal: cannot open Annex B bytestream file '%s', exit (-1)\n", Filename);
//    exit (-1);
//  }
}


/*!
 ********************************************************************************************
 * \brief 
 *    Closes the output bit stream file
 *
 * \return
 *    none.  Funtion trerminates the program in case of an error
 ********************************************************************************************
*/
void CloseAnnexbFile() {
//  if (fclose (f))
//  {
//    printf ("Fatal: cannot close Annex B bytestream file, exit (-1)\n");
//    exit (-1);
//  }
}