test/h264enc_test.cpp

   1 /*******************************************************************************
   2  *    File Name : h264enc_test.c
   3  *
   4  *       Author : Henry He
   5  * Created Time : Sat 31 Oct 2009 04:26:05 PM CST
   6  *  Description :
   7  ******************************************************************************/
   8
   9
  10 /*******************************************************************************
  11  * Desc : Includes Files
  12  ******************************************************************************/
  13
  14 #include  "mfc/common.h"
  15
  16 #include  <sys/time.h>
  17 #include  <sys/mman.h>
  18
  19
  20
  21 /*******************************************************************************
  22  * Desc : Macro Definations
  23  ******************************************************************************/
  24
  25
  26 /*******************************************************************************
  27  * Desc : Type Definations
  28  ******************************************************************************/
  29
  30
  31 /*******************************************************************************
  32  * Desc : Global Variables
  33  ******************************************************************************/
  34
  35
  36 /*******************************************************************************
  37  * Desc : File Variables
  38  ******************************************************************************/
  39
  40
  41
  42
  43
  44
  45
  46 unsigned int measureTime(struct timeval *start, struct timeval *stop)
  47 {
  48         unsigned int sec, usec, time;
  49
  50         sec = stop->tv_sec - start->tv_sec;
  51         if(stop->tv_usec >= start->tv_usec)
  52         {
  53                 usec = stop->tv_usec - start->tv_usec;
  54         }
  55         else
  56         {
  57                 usec = stop->tv_usec + 1000000 - start->tv_usec;
  58                 sec--;
  59         }
  60         time = sec*1000 + usec/1000;
  61         return time;
  62 }
  63
  64
  65
  66
  67
  68
  69
  70
  71
  72
  73
  74
  75
  76
  77 int Test_H264_Encoder(int argc, char **argv)
  78 {
  79         int             in_fd;
  80         int             out_fd;
  81         char           *in_addr;
  82         int             file_size;
  83
  84         int             frame_count;
  85         int             backup_frame_cnt;
  86         int             frame_size;
  87
  88         int             width;
  89         int             height;
  90         int             frame_rate;
  91         int             bitrate;
  92         int             gop_num;
  93
  94         void           *p_outbuf;
  95         UINT32          size;
  96
  97         int             hdr_size;
  98 //      int             para_change[2];
  99         int             frame_cnt        = 0;
 100         struct stat     s;
 101
 102   H264Encoder     zheEncoder;
 103
 104
 105         struct timeval  start;
 106         struct timeval  stop;
 107         unsigned int    time             = 0;
 108
 109   int             nRetVal;
 110
 111
 112         if (argc != 8) {
 113                 printf("Usage : mfc <YUV file name> <output filename> <width> <height> ");
 114                 printf("<frame rate> <bitrate> <GOP number>\n");
 115                 return -1;
 116         }
 117
 118         // in/out file open
 119         in_fd   = open(argv[1], O_RDONLY);
 120         out_fd  = open(argv[2], O_RDWR | O_CREAT | O_TRUNC, 0644);
 121
 122         if( (in_fd < 0) || (out_fd < 0) ) {
 123                 printf("input/output file open error\n");
 124                 return -1;
 125         }
 126
 127         // get input file size
 128         fstat(in_fd, &s);
 129         file_size = s.st_size;
 130
 131         // mapping input file to memory
 132         in_addr = (char *)mmap(0, file_size, PROT_READ, MAP_SHARED, in_fd, 0);
 133         if(in_addr == NULL) {
 134                 printf("input file memory mapping failed\n");
 135                 return -1;
 136         }
 137
 138   width      = atoi(argv[3]);
 139   height     = atoi(argv[4]);
 140   frame_rate = atoi(argv[5]);
 141   bitrate    = atoi(argv[6]);
 142   gop_num    = atoi(argv[7]);
 143
 144         frame_size      = (width * height * 3) >> 1;
 145         frame_count     = file_size / frame_size;
 146
 147         printf("file_size : %d, frame_size : %d, frame count : %d\n", file_size, frame_size, frame_count);
 148
 149
 150         nRetVal = zheEncoder.Create (width, height, frame_rate, bitrate, gop_num);
 151         if (nRetVal < 0) {
 152                 LOG_MSG(LOG_ERROR, "Test_Encoder", "create encoder failed\n");
 153                 return -1;
 154         }
 155
 156         backup_frame_cnt = frame_count;
 157
 158         while (frame_count > 0)
 159         {
 160                 printf ("frame count : %d\n", frame_count);
 161
 162                 // copy YUV data into input buffer
 163                 gettimeofday(&start, NULL);
 164
 165                 p_outbuf = zheEncoder.Encode (in_addr, frame_size, size);
 166                 if (frame_count == backup_frame_cnt) {
 167                         zheEncoder.GetConfig(H264_ENC_GETCONF_HEADER_SIZE, &hdr_size);
 168                         printf("Header Size : %d\n", hdr_size);
 169                 }
 170                 in_addr += frame_size;
 171
 172                 gettimeofday (&stop, NULL);
 173                 time += measureTime (&start, &stop);
 174
 175                 frame_cnt++;
 176
 177                 write(out_fd, p_outbuf, size);
 178
 179                 frame_count--;
 180         }
 181
 182         printf("Decoding Time : %u, Frame Count : %d, FPS : %f\n", time, frame_cnt, (float)frame_cnt*1000/time);
 183
 184         zheEncoder.Destroy ();
 185
 186         munmap(in_addr, file_size);
 187         close(in_fd);
 188         close(out_fd);
 189
 190         return 0;
 191 }
 192
 193
 194
 195
 196 int main ( int argc, char *argv[] )
 197 {
 198         Test_H264_Encoder(argc, argv);
 199         return 0;
 200 }                               /* ----------  end of function main  ---------- */
 201