nalu.rar_SVC video coding_nalu_svc

#include <stdio.h> #include <stdlib.h> #include <string.h> #include "nalu_tool.h" int g_i_num; // how many nalus will be cut out int DUMPNALU=0; __inline void PrintHelp() { printf("nalu_tool.exe -i in_file [options] -o [out_file]\n"); printf("options: \n"); printf("\t-n N : fetch N NALUs from in_file to out_file\n"); printf("\t if need NALU details,usage:add -v \n"); } ////////////////////////////////////////////////////////////////////////// // init nalu ////////////////////////////////////////////////////////////////////////// int InitNalu(Nalu **nalu) { *nalu = (Nalu*)calloc(1, sizeof(Nalu)); if (NULL == (*nalu)) { printf("Mem allocation error in %s, %s\n", __FILE__, __LINE__); exit(1); } (*nalu)->buffer = (byte*)calloc(MAX_NALU_SIZE, sizeof(byte)); if (NULL == ((*nalu)->buffer)) { printf("Mem allocation error in %s, %s\n", __FILE__, __LINE__); exit(1); } return 1; } ////////////////////////////////////////////////////////////////////////// // free nalu ////////////////////////////////////////////////////////////////////////// int FreeNalu(Nalu **nalu) { if (NULL != ((*nalu)->buffer)) { free((*nalu)->buffer); (*nalu)->buffer = NULL; } if (NULL != (*nalu)) { free(*nalu); *nalu = NULL; } return 0; } ////////////////////////////////////////////////////////////////////////// // main function ////////////////////////////////////////////////////////////////////////// int main (int argc, char **argv) { FILE *pf_in = NULL; FILE *pf_out = NULL; Nalu *p_nalu = NULL; InitNalu(&p_nalu); if (0 == ParseParam(&pf_in,&pf_out,argc, argv)) exit(1); while (g_i_num-- && !feof(pf_in)) { GetOneNalu(pf_in, p_nalu); NALU_index++; if(DUMPNALU==1) DumpOneNalu(p_nalu); PutOneNalu(pf_out, p_nalu); } FreeNalu(&p_nalu); if (pf_in != NULL) fclose(pf_in); if (pf_out != NULL) fclose(pf_out); return 0; } ////////////////////////////////////////////////////////////////////////// // Set the parameters. ////////////////////////////////////////////////////////////////////////// int ParseParam(FILE** f_in,FILE** f_out,int argc, char **argv) { // set default parameters // printf("Setting Default Parameters...\n"); int i; //int err; char *in_filename = DEFAULT_IN_FILENAME; char *out_filename = DEFAULT_OUT_FILENAME; g_i_num = DEFAULT_NALU_NUM; // parse command line parameters // printf("Parsing Command Line Parameters...\n"); if (argc < 3 || argc > 8) { PrintHelp(); return 0; } for(i=1;i<argc;i++) { if(0 == strncmp (argv[i], "-i", 2)) { if(fopen(argv[i+1],"rb")!=0) { printf( "Error: file %s not found\n",argv[i+1]); } i++; } if(0 == strncmp (argv[i], "-o", 2)) { out_filename=argv[i+1]; i++; } if(0 == strncmp (argv[i], "-n", 2)) { g_i_num = atoi(argv[i+1]); i++; } if(0 == strncmp (argv[i], "-v", 2)) { DUMPNALU=1; } } if(fopen(out_filename,"wb")!=0) printf( "Error: file can not open\n"); if(i=argc) return 1; PrintHelp(); return 0; }