Make isac_fix_test use gtest (in a hacky way)

webrtc/modules/audio_coding/codecs/isac/fix/test/kenny.cc

 /*
  *  Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <time.h>
 #include <ctype.h>
 
 #include "webrtc/modules/audio_coding/codecs/isac/fix/include/isacfix.h"
+#include "webrtc/test/gtest.h"
 #include "webrtc/test/testsupport/perf_test.h"
 
 // TODO(kma): Clean up the code and change benchmarking the whole codec to
 // separate encoder and decoder.
 
 /* Defines */
 #define SEED_FILE "randseed.txt"  /* Used when running decoder on garbage data */
 #define MAX_FRAMESAMPLES    960   /* max number of samples per frame (= 60 ms frame) */
 #define FRAMESAMPLES_10ms 160   /* number of samples per 10ms frame */
 #define FS           16000 /* sampling frequency (Hz) */
 
 /* Function for reading audio data from PCM file */
 int readframe(int16_t *data, FILE *inp, int length) {
 
   short k, rlen, status = 0;
 
   rlen = fread(data, sizeof(int16_t), length, inp);
   if (rlen < length) {
     for (k = rlen; k < length; k++)
       data[k] = 0;
     status = 1;
   }
 
   return status;
 }
 
+// Globals needed because gtest does not provide access to argv.
+// This should be reworked to use flags.
+static int global_argc;
+static char **global_argv;
+
 /* Struct for bottleneck model */
 typedef struct {
   uint32_t send_time;            /* samples */
   uint32_t arrival_time;         /* samples */
   uint32_t sample_count;         /* samples */
   uint16_t rtp_number;
 } BottleNeckModel;
 
 void get_arrival_time(int current_framesamples,   /* samples */
                       size_t packet_size,         /* bytes */
@@ skipping 31 matching lines @@
   {
     //
     BN_data->arrival_time += (current_framesamples + ((FS/1000) * current_delay));
   }
   //else
   //current packet has same timestamp as previous packet
 
   BN_data->rtp_number++;
 }
 
-int main(int argc, char* argv[])
-{
+TEST(IsacFixTest, Kenny) {
+  int argc = global_argc;
+  char **argv = global_argv;
 
   char inname[100], outname[100],  outbitsname[100], bottleneck_file[100];
   FILE *inp, *outp, *f_bn, *outbits;
   int endfile;
 
   size_t i;
   int errtype, h = 0, k, packetLossPercent = 0;
   int16_t CodingMode;
   int16_t bottleneck;
   int framesize = 30;           /* ms */
@@ skipping 109 matching lines @@
            " getNewBitstream \n");
     printf("                  CE 2 - transcode, getBWE \n");
     printf("                  CE 3 - getSendBWE, setSendBWE.  \n\n");
     printf("[-RTP_INIT num] : if -RTP_INIT option is specified num will be"
            " the initial\n");
     printf("                  value of the rtp sequence number.\n\n");
     printf("infile          : Normal speech input file\n\n");
     printf("outfile         : Speech output file\n\n");
     printf("Example usage   : \n\n");
     printf("%s -I bottleneck.txt speechIn.pcm speechOut.pcm\n\n", argv[0]);
-    exit(0);
+    exit(1);
 
   }
 
   /* Print version number */
   WebRtcIsacfix_version(version_number);
   printf("iSAC version %s \n\n", version_number);
 
   /* Loop over all command line arguments */
   CodingMode = 0;
   testNum = 0;
   testCE = 0;
   for (i = 1; i + 2 < static_cast<size_t>(argc); i++) {
     /* Instantaneous mode */
     if (!strcmp ("-I", argv[i])) {
       printf("\nInstantaneous BottleNeck\n");
       CodingMode = 1;
       i++;
     }
 
     /* Set (initial) bottleneck value */
     if (!strcmp ("-INITRATE", argv[i])) {
       rateBPS = atoi(argv[i + 1]);
       setControlBWE = 1;
       if ((rateBPS < 10000) || (rateBPS > 32000)) {
         printf("\n%d is not a initial rate. "
                "Valid values are in the range 10000 to 32000.\n", rateBPS);
-        exit(0);
+        exit(1);
       }
       printf("\nNew initial rate: %d\n", rateBPS);
       i++;
     }
 
     /* Set (initial) framelength */
     if (!strcmp ("-FL", argv[i])) {
       framesize = atoi(argv[i + 1]);
       if ((framesize != 30) && (framesize != 60)) {
         printf("\n%d is not a valid frame length. "
                "Valid length are 30 and 60 msec.\n", framesize);
-        exit(0);
+        exit(1);
       }
       printf("\nFrame Length: %d\n", framesize);
       i++;
     }
 
     /* Fixed frame length */
     if (!strcmp ("-FIXED_FL", argv[i])) {
       fixedFL = 1;
       setControlBWE = 1;
     }
@@ skipping 12 matching lines @@
       i++;
     }
 
     /* Test of fault scenarious */
     if (!strcmp ("-F", argv[i])) {
       testNum = atoi(argv[i + 1]);
       printf("\nFault test: %d\n", testNum);
       if (testNum < 1 || testNum > 10) {
         printf("\n%d is not a valid Fault Scenario number."
                " Valid Fault Scenarios are numbered 1-10.\n", testNum);
-        exit(0);
+        exit(1);
       }
       i++;
     }
 
     /* Packet loss test */
     if (!strcmp ("-PL", argv[i])) {
       if( isdigit( *argv[i+1] ) ) {
         packetLossPercent = atoi( argv[i+1] );
         if( (packetLossPercent < 0) | (packetLossPercent > 100) ) {
           printf( "\nInvalid packet loss perentage \n" );
-          exit( 0 );
+          exit( 1 );
         }
         if( packetLossPercent > 0 ) {
           printf( "\nSimulating %d %% of independent packet loss\n",
                   packetLossPercent );
         } else {
           printf( "\nNo Packet Loss Is Simulated \n" );
         }
         readLoss = 0;
       } else {
         readLoss = 1;
         plFile = fopen( argv[i+1], "rb" );
         if( plFile == NULL ) {
-          printf( "\n couldn't open the frameloss file: %s\n", argv[i+1] );
-          exit( 0 );
+          FAIL() << "Couldn't open the frameloss file: " << argv[i+1];
         }
         printf( "\nSimulating packet loss through the given "
                 "channel file: %s\n", argv[i+1] );
       }
       i++;
     }
 
     /* Random packetlosses */
     if (!strcmp ("-rnd", argv[i])) {
       srand(time(NULL) );
       printf( "\n Random pattern in lossed packets \n" );
     }
 
     /* Use gns file */
     if (!strcmp ("-G", argv[i])) {
       sscanf(argv[i + 1], "%s", gns_file);
       fp_gns = fopen(gns_file, "rb");
       if (fp_gns  == NULL) {
-        printf("Cannot read file %s.\n", gns_file);
-        exit(0);
+        FAIL() << "Cannot read file " << gns_file << ".";
       }
       gns = 1;
       i++;
     }
 
     /* Run Narrowband interfaces (either encoder or decoder) */
     if (!strcmp ("-NB", argv[i])) {
       nbTest = atoi(argv[i + 1]);
       i++;
     }
 
     /* Run Conference Engine APIs */
     if (!strcmp ("-CE", argv[i])) {
       testCE = atoi(argv[i + 1]);
       if (testCE==1 || testCE==2) {
         i++;
         scale = (float)atof( argv[i+1] );
       } else if (testCE < 1 || testCE > 3) {
         printf("\n%d is not a valid CE-test number, valid Fault "
                "Scenarios are numbered 1-3\n", testCE);
-        exit(0);
+        exit(1);
       }
       i++;
     }
 
     /* Set initial RTP number */
     if (!strcmp ("-RTP_INIT", argv[i])) {
       i++;
     }
   }
 
   /* Get Bottleneck value                                                   */
   /* Gns files and bottleneck should not and can not be used simultaneously */
   bottleneck = atoi(argv[CodingMode+1]);
   if (bottleneck == 0 && gns == 0) {
     sscanf(argv[CodingMode+1], "%s", bottleneck_file);
     f_bn = fopen(bottleneck_file, "rb");
     if (f_bn  == NULL) {
       printf("No value provided for BottleNeck and cannot read file %s\n",
              bottleneck_file);
-      exit(0);
+      exit(0);  // TODO(oprypin): don't silence this error
+      // FAIL() << "Cannot read file " << bottleneck_file;
     } else {
       int aux_var;
       printf("reading bottleneck rates from file %s\n\n",bottleneck_file);
       if (fscanf(f_bn, "%d", &aux_var) == EOF) {
         /* Set pointer to beginning of file */
         fseek(f_bn, 0L, SEEK_SET);
         if (fscanf(f_bn, "%d", &aux_var) == EOF) {
-          exit(0);
+          FAIL();
         }
       }
       bottleneck = (int16_t)aux_var;
       /* Bottleneck is a cosine function
        * Matlab code for writing the bottleneck file:
        * BottleNeck_10ms = 20e3 + 10e3 * cos((0:5999)/5999*2*pi);
        * fid = fopen('bottleneck.txt', 'wb');
        * fprintf(fid, '%d\n', BottleNeck_10ms); fclose(fid);
        */
     }
@@ skipping 13 matching lines @@
   /* Add '.bit' to output bitstream file */
   while ((int)outname[h] != 0) {
     outbitsname[h] = outname[h];
     h++;
   }
   for (k=0; k<5; k++) {
     outbitsname[h] = tmpBit[k];
     h++;
   }
   if ((inp = fopen(inname,"rb")) == NULL) {
-    printf("  iSAC: Cannot read file %s\n", inname);
-    exit(1);
+    FAIL() << "  iSAC: Cannot read file " << inname;
   }
   if ((outp = fopen(outname,"wb")) == NULL) {
-    printf("  iSAC: Cannot write file %s\n", outname);
-    exit(1);
+    FAIL() << "  iSAC: Cannot write file " << outname;
   }
 
   if ((outbits = fopen(outbitsname,"wb")) == NULL) {
-    printf("  iSAC: Cannot write file %s\n", outbitsname);
-    exit(1);
+    FAIL() << "  iSAC: Cannot write file " << outbitsname;
   }
   printf("\nInput:%s\nOutput:%s\n\n", inname, outname);
 
   /* Error test number 10, garbage data */
   if (testNum == 10) {
     /* Test to run decoder with garbage data */
     srand(random_seed);
 
     if ( (seedfile = fopen(SEED_FILE, "a+t") ) == NULL ) {
       printf("Error: Could not open file %s\n", SEED_FILE);
@@ skipping 58 matching lines @@
     }
   } else if(setControlBWE == 1) {
     err = WebRtcIsacfix_ControlBwe(ISAC_main_inst, rateBPS, framesize, fixedFL);
   }
 
   if (payloadSize != 0) {
     err = WebRtcIsacfix_SetMaxPayloadSize(ISAC_main_inst, payloadSize);
     if (err < 0) {
       /* exit if returned with error */
       errtype=WebRtcIsacfix_GetErrorCode(ISAC_main_inst);
-      printf("\n\n Error in SetMaxPayloadSize: %d.\n\n", errtype);
-      exit(EXIT_FAILURE);
+      FAIL() << "Error in SetMaxPayloadSize: " << errtype;
     }
   }
   if (payloadRate != 0) {
     err = WebRtcIsacfix_SetMaxRate(ISAC_main_inst, payloadRate);
     if (err < 0) {
       /* exit if returned with error */
       errtype=WebRtcIsacfix_GetErrorCode(ISAC_main_inst);
-      printf("\n\n Error in SetMaxRateInBytes: %d.\n\n", errtype);
-      exit(EXIT_FAILURE);
+      FAIL() << "Error in SetMaxRateInBytes: " << errtype;
     }
   }
 
   *speechType = 1;
 
 
   while (endfile == 0) {
 
     if(testNum == 7 && (rand()%2 == 0)) {
       err = WebRtcIsacfix_EncoderInit(ISAC_main_inst, CodingMode);
@@ skipping 81 matching lines @@
       }
 
       if (stream_len_int < 0 || err < 0) {
         /* exit if returned with error */
         errtype=WebRtcIsacfix_GetErrorCode(ISAC_main_inst);
         printf("\nError in encoder: %d.\n", errtype);
       } else {
         stream_len = static_cast<size_t>(stream_len_int);
         if (fwrite(streamdata, sizeof(char), stream_len, outbits) !=
             stream_len) {
-          return -1;
+          FAIL();
         }
       }
 
       cur_framesmpls += FRAMESAMPLES_10ms;
 
       /* read next bottleneck rate */
       if (f_bn != NULL) {
         int aux_var;
         if (fscanf(f_bn, "%d", &aux_var) == EOF) {
           /* Set pointer to beginning of file */
           fseek(f_bn, 0L, SEEK_SET);
           if (fscanf(f_bn, "%d", &aux_var) == EOF) {
-            exit(0);
+            FAIL();
           }
         }
         bottleneck = (int16_t)aux_var;
         if (CodingMode == 1) {
           WebRtcIsacfix_Control(ISAC_main_inst, bottleneck, framesize);
         }
       }
 
       /* exit encoder loop if the encoder returned a bitstream */
       if (stream_len != 0) break;
@@ skipping 16 matching lines @@
       for (i = 0; i < stream_len; i++ ) {
         streamdata[i] = rand();
       }
     }
 
     /* set pointer to beginning of file */
     if (fp_gns != NULL) {
       if (fscanf(fp_gns, "%d", &cur_delay) == EOF) {
         fseek(fp_gns, 0L, SEEK_SET);
         if (fscanf(fp_gns, "%d", &cur_delay) == EOF) {
-          exit(0);
+          FAIL();
         }
       }
     }
 
     /* simulate packet handling through NetEq and the modem */
     if (!(testNum == 3 && framecnt == 0)) {
       if (gns == 0) {
         get_arrival_time(cur_framesmpls, stream_len, bottleneck,
                          &BN_data);
       } else {
@@ skipping 88 matching lines @@
 
       if (declen <= 0) {
         /* exit if returned with error */
         errtype=WebRtcIsacfix_GetErrorCode(ISAC_main_inst);
         printf("\nError in decoder: %d.\n", errtype);
       }
 
       /* Write decoded speech frame to file */
       if (fwrite(decoded, sizeof(int16_t),
                  declen, outp) != (size_t)declen) {
-        return -1;
+        FAIL();
       }
       //   fprintf( ratefile, "%f \n", stream_len / ( ((double)declen)/
       // ((double)FS) ) * 8 );
     } else {
       lostPackets++;
     }
     framecnt++;
 
     totalsmpls += declen;
     totalbits += static_cast<int>(8 * stream_len);
@@ skipping 31 matching lines @@
                             (runtime * 10000) / length_file, "us", false);
 
   fclose(inp);
   fclose(outp);
   fclose(outbits);
 
   if ( testCE == 1) {
     WebRtcIsacfix_FreeInternal(ISAC_main_inst);
   }
   WebRtcIsacfix_Free(ISAC_main_inst);
-  return 0;
 }
+
+int main(int argc, char* argv[]) {
+  ::testing::InitGoogleTest(&argc, argv);
+  global_argc = argc;
+  global_argv = argv;
+
+  return RUN_ALL_TESTS();
+}