generic-library/webrtc
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

git-svn-id: http://webrtc.googlecode.com/svn/trunk@4 4adac7df-926f-26a2-2b94-8c16560cd09d

Browse Source
This commit is contained in:
niklase@google.com
2011-05-30 11:22:19 +00:00
parent 01813fe945
commit 77ae29bc81
1153 changed files with 404089 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

635
modules/audio_coding/codecs/iSAC/main/test/simpleKenny.c Normal file
View File

@@ -0,0 +1,635 @@
/*
* Copyright (c) 2011 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.
*/
/* kenny.c - Main function for the iSAC coder */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#ifdef WIN32
#include "windows.h"
#define CLOCKS_PER_SEC 1000
#endif
#include <ctype.h>
#include <math.h>
/* include API */
#include "isac.h"
#include "utility.h"
//#include "commonDefs.h"
/* max number of samples per frame (= 60 ms frame) */
#define MAX_FRAMESAMPLES_SWB 1920
/* number of samples per 10ms frame */
#define FRAMESAMPLES_SWB_10ms 320
#define FRAMESAMPLES_WB_10ms 160
/* sampling frequency (Hz) */
#define FS_SWB 32000
#define FS_WB 16000
//#define CHANGE_OUTPUT_NAME
#ifdef HAVE_DEBUG_INFO
#include "debugUtility.h"
debugStruct debugInfo;
#endif
unsigned long framecnt = 0;
int main(int argc, char* argv[])
{
//--- File IO ----
FILE* inp;
FILE* outp;
char inname[500];
char outname[500];
char usageFileName[500] = "usage.txt";
/* Runtime statistics */
double starttime;
double runtime;
double length_file;
double rate;
double rateRCU;
double rateLB;
double rateUB;
unsigned long totalbits = 0;
unsigned long totalBitsRCU = 0;
unsigned long totalsmpls =0;
WebRtc_Word32 bottleneck = 39;
WebRtc_Word16 frameSize = 30; /* ms */
WebRtc_Word16 codingMode = 1;
WebRtc_Word16 shortdata[FRAMESAMPLES_SWB_10ms];
WebRtc_Word16 decoded[MAX_FRAMESAMPLES_SWB];
//WebRtc_UWord16 streamdata[1000];
WebRtc_Word16 speechType[1];
WebRtc_Word16 payloadLimit;
WebRtc_Word32 rateLimit;
ISACStruct* ISAC_main_inst;
WebRtc_Word16 stream_len = 0;
WebRtc_Word16 declen;
WebRtc_Word16 err;
WebRtc_Word16 cur_framesmpls;
int endfile;
char outDrive[10];
char outPath[500];
char outPrefix[500];
char outSuffix[500];
char bitrateFileName[500];
FILE* bitrateFile;
FILE* histFile;
FILE* averageFile;
int sampFreqKHz;
int samplesIn10Ms;
WebRtc_Word16 maxStreamLen = 0;
char histFileName[500];
char averageFileName[500];
unsigned int hist[600];
unsigned int tmpSumStreamLen = 0;
unsigned int packetCntr = 0;
unsigned int lostPacketCntr = 0;
WebRtc_UWord16 payload[600];
WebRtc_UWord16 payloadRCU[600];
WebRtc_UWord16 packetLossPercent = 0;
WebRtc_Word16 rcuStreamLen;
int onlyEncode;
int onlyDecode;
BottleNeckModel packetData;
packetData.arrival_time = 0;
packetData.sample_count = 0;
packetData.rtp_number = 0;
memset(hist, 0, sizeof(hist));
/* handling wrong input arguments in the command line */
if(argc < 5)
{
int size;
WebRtcIsac_AssignSize(&size);
printf("\n\nWrong number of arguments or flag values.\n\n");
printf("Usage:\n\n");
printf("%s infile outfile -bn bottelneck [options] \n\n", argv[0]);
printf("with:\n");
printf("-I................... indicates encoding in instantaneous mode.\n");
printf("-bn bottleneck....... the value of the bottleneck in bit/sec, e.g. 39742,\n");
printf(" in instantaneous (channel-independent) mode.\n\n");
printf("infile............... Normal speech input file\n\n");
printf("outfile.............. Speech output file\n\n");
printf("OPTIONS\n");
printf("-------\n");
printf("-fs sampFreq......... sampling frequency of codec 16 or 32 (default) kHz.\n");
printf("-plim payloadLim..... payload limit in bytes,\n");
printf(" default is the maximum possible.\n");
printf("-rlim rateLim........ rate limit in bits/sec, \n");
printf(" default is the maimum possible.\n");
printf("-h file.............. record histogram and *append* to 'file'.\n");
printf("-ave file............ record average rate of 3 sec intervales and *append* to 'file'.\n");
printf("-ploss............... packet-loss percentage.\n");
printf("-enc................. do only encoding and store the bit-stream\n");
printf("-dec................. the input file is a bit-stream, decode it.\n");
printf("\n");
printf("Example usage:\n\n");
printf("%s speechIn.pcm speechOut.pcm -B 40000 -fs 32 \n\n", argv[0]);
printf("structure size %d bytes\n", size);
exit(0);
}
/* Get Bottleneck value */
bottleneck = readParamInt(argc, argv, "-bn", 50000);
fprintf(stderr,"\nfixed bottleneck rate of %d bits/s\n\n", bottleneck);
/* Get Input and Output files */
sscanf(argv[1], "%s", inname);
sscanf(argv[2], "%s", outname);
codingMode = readSwitch(argc, argv, "-I");
sampFreqKHz = (WebRtc_Word16)readParamInt(argc, argv, "-fs", 32);
if(readParamString(argc, argv, "-h", histFileName, 500) > 0)
{
histFile = fopen(histFileName, "a");
if(histFile == NULL)
{
printf("cannot open hist file %s", histFileName);
exit(0);
}
}
else
{
// NO recording of hitstogram
histFile = NULL;
}
packetLossPercent = readParamInt(argc, argv, "-ploss", 0);
if(readParamString(argc, argv, "-ave", averageFileName, 500) > 0)
{
averageFile = fopen(averageFileName, "a");
if(averageFile == NULL)
{
printf("cannot open file to write rate %s", averageFileName);
exit(0);
}
}
else
{
averageFile = NULL;
}
onlyEncode = readSwitch(argc, argv, "-enc");
onlyDecode = readSwitch(argc, argv, "-dec");
switch(sampFreqKHz)
{
case 16:
{
samplesIn10Ms = 160;
break;
}
case 32:
{
samplesIn10Ms = 320;
break;
}
default:
printf("A sampling frequency of %d kHz is not supported,\
valid values are 8 and 16.\n", sampFreqKHz);
exit(-1);
}
payloadLimit = (WebRtc_Word16)readParamInt(argc, argv, "-plim", 400);
rateLimit = readParamInt(argc, argv, "-rlim", 106800);
if ((inp = fopen(inname,"rb")) == NULL) {
printf(" iSAC: Cannot read file %s.\n", inname);
exit(1);
}
if ((outp = fopen(outname,"wb")) == NULL) {
printf(" iSAC: Cannot write file %s.\n", outname);
exit(1);
}
#ifdef WIN32
_splitpath(outname, outDrive, outPath, outPrefix, outSuffix);
_makepath(bitrateFileName, outDrive, outPath, "bitrate", ".txt");
bitrateFile = fopen(bitrateFileName, "a");
fprintf(bitrateFile, "% %%s \n", inname);
#endif
printf("\n");
printf("Input.................... %s\n", inname);
printf("Output................... %s\n", outname);
printf("Encoding Mode............ %s\n",
(codingMode == 1)? "Channel-Independent":"Channel-Adaptive");
printf("Bottleneck............... %d bits/sec\n", bottleneck);
printf("Packet-loss Percentage... %d\n", packetLossPercent);
printf("\n");
starttime = clock()/(double)CLOCKS_PER_SEC; /* Runtime statistics */
/* Initialize the ISAC and BN structs */
err = WebRtcIsac_Create(&ISAC_main_inst);
WebRtcIsac_SetEncSampRate(ISAC_main_inst, (sampFreqKHz == 16)? kIsacWideband: kIsacSuperWideband);
WebRtcIsac_SetDecSampRate(ISAC_main_inst, (sampFreqKHz == 16)? kIsacWideband: kIsacSuperWideband);
/* Error check */
if (err < 0) {
fprintf(stderr,"\n\n Error in create.\n\n");
exit(EXIT_FAILURE);
}
framecnt = 0;
endfile = 0;
/* Initialize encoder and decoder */
if(WebRtcIsac_EncoderInit(ISAC_main_inst, codingMode) < 0)
{
printf("cannot initialize encoder\n");
return -1;
}
if(WebRtcIsac_DecoderInit(ISAC_main_inst) < 0)
{
printf("cannot initialize decoder\n");
return -1;
}
//{
// WebRtc_Word32 b1, b2;
// FILE* fileID = fopen("GetBNTest.txt", "w");
// b2 = 32100;
// while(b2 <= 52000)
// {
// WebRtcIsac_Control(ISAC_main_inst, b2, frameSize);
// WebRtcIsac_GetUplinkBw(ISAC_main_inst, &b1);
// fprintf(fileID, "%5d %5d\n", b2, b1);
// b2 += 10;
// }
//}
if(codingMode == 1)
{
if(WebRtcIsac_Control(ISAC_main_inst, bottleneck, frameSize) < 0)
{
printf("cannot set bottleneck\n");
return -1;
}
}
else
{
if(WebRtcIsac_ControlBwe(ISAC_main_inst, 15000, 30, 1) < 0)
{
printf("cannot configure BWE\n");
return -1;
}
}
if(WebRtcIsac_SetMaxPayloadSize(ISAC_main_inst, payloadLimit) < 0)
{
printf("cannot set maximum payload size %d.\n", payloadLimit);
return -1;
}
if (rateLimit < 106800) {
if(WebRtcIsac_SetMaxRate(ISAC_main_inst, rateLimit) < 0)
{
printf("cannot set the maximum rate %d.\n", rateLimit);
return -1;
}
}
//=====================================
//#ifdef HAVE_DEBUG_INFO
// if(setupDebugStruct(&debugInfo) < 0)
// {
// exit(1);
// }
//#endif
while (endfile == 0)
{
fprintf(stderr," \rframe = %7li", framecnt);
//============== Readind from the file and encoding =================
cur_framesmpls = 0;
stream_len = 0;
if(onlyDecode)
{
WebRtc_UWord8 auxUW8;
if(fread(&auxUW8, sizeof(WebRtc_UWord8), 1, inp) < 1)
{
break;
}
stream_len = ((WebRtc_UWord8)auxUW8) << 8;
if(fread(&auxUW8, sizeof(WebRtc_UWord8), 1, inp) < 1)
{
break;
}
stream_len |= (WebRtc_UWord16)auxUW8;
if(fread(payload, 1, stream_len, inp) < stream_len)
{
printf("last payload is corrupted\n");
break;
}
}
else
{
while(stream_len == 0)
{
// Read 10 ms speech block
endfile = readframe(shortdata, inp, samplesIn10Ms);
if(endfile)
{
break;
}
cur_framesmpls += samplesIn10Ms;
//-------- iSAC encoding ---------
if(framecnt == 11)
{
framecnt = framecnt;
}
stream_len = WebRtcIsac_Encode(ISAC_main_inst, shortdata,
(WebRtc_Word16*)payload);
if(stream_len < 0)
{
// exit if returned with error
//errType=WebRtcIsac_GetErrorCode(ISAC_main_inst);
fprintf(stderr,"\nError in encoder\n");
getchar();
exit(EXIT_FAILURE);
}
}
//===================================================================
if(endfile)
{
break;
}
rcuStreamLen = WebRtcIsac_GetRedPayload(ISAC_main_inst, (WebRtc_Word16*)payloadRCU);
get_arrival_time(cur_framesmpls, stream_len, bottleneck, &packetData,
sampFreqKHz * 1000, sampFreqKHz * 1000);
if(WebRtcIsac_UpdateBwEstimate(ISAC_main_inst,
payload, stream_len, packetData.rtp_number,
packetData.sample_count,
packetData.arrival_time) < 0)
{
printf(" BWE Error at client\n");
return -1;
}
}
if(endfile)
{
break;
}
maxStreamLen = (stream_len > maxStreamLen)? stream_len:maxStreamLen;
packetCntr++;
hist[stream_len]++;
if(averageFile != NULL)
{
tmpSumStreamLen += stream_len;
if(packetCntr == 100)
{
// kbps
fprintf(averageFile, "%8.3f ", (double)tmpSumStreamLen * 8.0 / (30.0 * packetCntr));
packetCntr = 0;
tmpSumStreamLen = 0;
}
}
if(onlyEncode)
{
WebRtc_UWord8 auxUW8;
auxUW8 = (WebRtc_UWord8)(((stream_len & 0x7F00) >> 8) & 0xFF);
fwrite(&auxUW8, sizeof(WebRtc_UWord8), 1, outp);
auxUW8 = (WebRtc_UWord8)(stream_len & 0xFF);
fwrite(&auxUW8, sizeof(WebRtc_UWord8), 1, outp);
fwrite(payload, 1, stream_len, outp);
}
else
{
//======================= iSAC decoding ===========================
if((rand() % 100) < packetLossPercent)
{
declen = WebRtcIsac_DecodeRcu(ISAC_main_inst, payloadRCU,
rcuStreamLen, decoded, speechType);
lostPacketCntr++;
}
else
{
declen = WebRtcIsac_Decode(ISAC_main_inst, payload,
stream_len, decoded, speechType);
}
if(declen <= 0)
{
//errType=WebRtcIsac_GetErrorCode(ISAC_main_inst);
fprintf(stderr,"\nError in decoder.\n");
getchar();
exit(1);
}
// Write decoded speech frame to file
fwrite(decoded, sizeof(WebRtc_Word16), declen, outp);
cur_framesmpls = declen;
}
// Update Statistics
framecnt++;
totalsmpls += cur_framesmpls;
if(stream_len > 0)
{
totalbits += 8 * stream_len;
}
if(rcuStreamLen > 0)
{
totalBitsRCU += 8 * rcuStreamLen;
}
}
rate = ((double)totalbits * (sampFreqKHz)) / (double)totalsmpls;
rateRCU = ((double)totalBitsRCU * (sampFreqKHz)) / (double)totalsmpls;
rateLB = 0;
rateUB = 0;
printf("\n\n");
printf("Sampling Rate......................... %d kHz\n", sampFreqKHz);
printf("Payload Limit......................... %d bytes \n", payloadLimit);
printf("Rate Limit............................ %d bits/sec \n", rateLimit);
#ifdef HAVE_DEBUG_INFO
rateLB = ((double)debugInfo.lbBytes * 8. * (sampFreqKHz)) / (double)totalsmpls;
rateUB = ((double)debugInfo.ubBytes * 8. * (sampFreqKHz)) / (double)totalsmpls;
#endif
#ifdef WIN32
fprintf(bitrateFile, "%d %10u %d %6.3f %6.3f %6.3f\n",
sampFreqKHz,
framecnt,
bottleneck,
rateLB,
rateUB,
rate);
fclose(bitrateFile);
#endif
printf("\n");
printf("Measured bit-rate..................... %0.3f kbps\n", rate);
printf("Measured RCU bit-ratre................ %0.3f kbps\n", rateRCU);
printf("Maximum bit-rate/payloadsize.......... %0.3f / %d\n",
maxStreamLen * 8 / 0.03, maxStreamLen);
printf("Measured packet-loss.................. %0.1f%% \n",
100.0f * (float)lostPacketCntr / (float)packetCntr);
//#ifdef HAVE_DEBUG_INFO
// printf("Measured lower-band bit-rate.......... %0.3f kbps (%.0f%%)\n",
// rateLB, (double)(rateLB) * 100. /(double)(rate));
// printf("Measured upper-band bit-rate.......... %0.3f kbps (%.0f%%)\n",
// rateUB, (double)(rateUB) * 100. /(double)(rate));
//
// printf("Maximum payload lower-band............ %d bytes (%0.3f kbps)\n",
// debugInfo.maxPayloadLB, debugInfo.maxPayloadLB * 8.0 / 0.03);
// printf("Maximum payload upper-band............ %d bytes (%0.3f kbps)\n",
// debugInfo.maxPayloadUB, debugInfo.maxPayloadUB * 8.0 / 0.03);
//#endif
printf("\n");
/* Runtime statistics */
#ifdef WIN32
runtime = (double)(clock()/(double)CLOCKS_PER_SEC-starttime);
length_file = ((double)framecnt*(double)declen/(sampFreqKHz*1000));
printf("Length of speech file................ %.1f s\n", length_file);
printf("Time to run iSAC..................... %.2f s (%.2f %% of realtime)\n\n",
runtime, (100*runtime/length_file));
#endif
printf("\n\n_______________________________________________\n");
if(histFile != NULL)
{
int n;
for(n = 0; n < 600; n++)
{
fprintf(histFile, "%6d ", hist[n]);
}
fprintf(histFile, "\n");
fclose(histFile);
}
if(averageFile != NULL)
{
if(packetCntr > 0)
{
fprintf(averageFile, "%8.3f ", (double)tmpSumStreamLen * 8.0 / (30.0 * packetCntr));
}
fprintf(averageFile, "\n");
fclose(averageFile);
}
fclose(inp);
fclose(outp);
WebRtcIsac_Free(ISAC_main_inst);
#ifdef CHANGE_OUTPUT_NAME
{
char* p;
char myExt[50];
char bitRateStr[10];
char newOutName[500];
strcpy(newOutName, outname);
myExt[0] = '\0';
p = strchr(newOutName, '.');
if(p != NULL)
{
strcpy(myExt, p);
*p = '_';
p++;
*p = '\0';
}
else
{
strcat(newOutName, "_");
}
sprintf(bitRateStr, "%0.0fkbps", rate);
strcat(newOutName, bitRateStr);
strcat(newOutName, myExt);
rename(outname, newOutName);
}
#endif
exit(0);
}
#ifdef HAVE_DEBUG_INFO
int setupDebugStruct(debugStruct* str)
{
str->prevPacketLost = 0;
str->currPacketLost = 0;
OPEN_FILE_WB(str->res0to4FilePtr, "Res0to4.dat");
OPEN_FILE_WB(str->res4to8FilePtr, "Res4to8.dat");
OPEN_FILE_WB(str->res8to12FilePtr, "Res8to12.dat");
OPEN_FILE_WB(str->res8to16FilePtr, "Res8to16.dat");
OPEN_FILE_WB(str->res0to4DecFilePtr, "Res0to4Dec.dat");
OPEN_FILE_WB(str->res4to8DecFilePtr, "Res4to8Dec.dat");
OPEN_FILE_WB(str->res8to12DecFilePtr, "Res8to12Dec.dat");
OPEN_FILE_WB(str->res8to16DecFilePtr, "Res8to16Dec.dat");
OPEN_FILE_WB(str->in0to4FilePtr, "in0to4.dat");
OPEN_FILE_WB(str->in4to8FilePtr, "in4to8.dat");
OPEN_FILE_WB(str->in8to12FilePtr, "in8to12.dat");
OPEN_FILE_WB(str->in8to16FilePtr, "in8to16.dat");
OPEN_FILE_WB(str->out0to4FilePtr, "out0to4.dat");
OPEN_FILE_WB(str->out4to8FilePtr, "out4to8.dat");
OPEN_FILE_WB(str->out8to12FilePtr, "out8to12.dat");
OPEN_FILE_WB(str->out8to16FilePtr, "out8to16.dat");
OPEN_FILE_WB(str->fftFilePtr, "riFFT.dat");
OPEN_FILE_WB(str->fftDecFilePtr, "riFFTDec.dat");
OPEN_FILE_WB(str->arrivalTime, NULL/*"ArivalTime.dat"*/);
str->lastArrivalTime = 0;
str->maxPayloadLB = 0;
str->maxPayloadUB = 0;
str->lbBytes = 0;
str->ubBytes = 0;
return 0;
};
#endif