# HG changeset patch # User mru # Date 1277491885 0 # Node ID d94cbfa7a17058cd2fcd650bbbc6d0c5e162997b # Parent 4aae15516d2cba601af9e3ee732c8d05a64270d8 elbg: remove VLAs diff -r 4aae15516d2c -r d94cbfa7a170 elbg.c --- a/elbg.c Fri Jun 25 18:48:45 2010 +0000 +++ b/elbg.c Fri Jun 25 18:51:25 2010 +0000 @@ -53,6 +53,7 @@ int *nearest_cb; int *points; AVLFG *rand_state; + int *scratchbuf; } elbg_data; static inline int distance_limited(int *a, int *b, int dim, int limit) @@ -117,7 +118,8 @@ /** * Implementation of the simple LBG algorithm for just two codebooks */ -static int simple_lbg(int dim, +static int simple_lbg(elbg_data *elbg, + int dim, int *centroid[3], int newutility[3], int *points, @@ -125,10 +127,13 @@ { int i, idx; int numpoints[2] = {0,0}; - int newcentroid[2][dim]; + int *newcentroid[2] = { + elbg->scratchbuf + 3*dim, + elbg->scratchbuf + 4*dim + }; cell *tempcell; - memset(newcentroid, 0, sizeof(newcentroid)); + memset(newcentroid[0], 0, 2 * dim * sizeof(*newcentroid[0])); newutility[0] = newutility[1] = 0; @@ -158,8 +163,8 @@ int *newcentroid_p) { cell *tempcell; - int min[elbg->dim]; - int max[elbg->dim]; + int *min = newcentroid_i; + int *max = newcentroid_p; int i; for (i=0; i< elbg->dim; i++) { @@ -174,8 +179,10 @@ } for (i=0; idim; i++) { - newcentroid_i[i] = min[i] + (max[i] - min[i])/3; - newcentroid_p[i] = min[i] + (2*(max[i] - min[i]))/3; + int ni = min[i] + (max[i] - min[i])/3; + int np = min[i] + (2*(max[i] - min[i]))/3; + newcentroid_i[i] = ni; + newcentroid_p[i] = np; } } @@ -248,14 +255,13 @@ { int j, k, olderror=0, newerror, cont=0; int newutility[3]; - int newcentroid[3][elbg->dim]; - int *newcentroid_ptrs[3]; + int *newcentroid[3] = { + elbg->scratchbuf, + elbg->scratchbuf + elbg->dim, + elbg->scratchbuf + 2*elbg->dim + }; cell *tempcell; - newcentroid_ptrs[0] = newcentroid[0]; - newcentroid_ptrs[1] = newcentroid[1]; - newcentroid_ptrs[2] = newcentroid[2]; - for (j=0; j<3; j++) olderror += elbg->utility[idx[j]]; @@ -277,11 +283,11 @@ newerror = newutility[2]; - newerror += simple_lbg(elbg->dim, newcentroid_ptrs, newutility, elbg->points, + newerror += simple_lbg(elbg, elbg->dim, newcentroid, newutility, elbg->points, elbg->cells[idx[1]]); if (olderror > newerror) { - shift_codebook(elbg, idx, newcentroid_ptrs); + shift_codebook(elbg, idx, newcentroid); elbg->error += newerror - olderror; @@ -366,6 +372,7 @@ elbg->nearest_cb = closest_cb; elbg->points = points; elbg->utility_inc = av_malloc(numCB*sizeof(int)); + elbg->scratchbuf = av_malloc(5*dim*sizeof(int)); elbg->rand_state = rand_state; @@ -425,4 +432,5 @@ av_free(list_buffer); av_free(elbg->cells); av_free(elbg->utility_inc); + av_free(elbg->scratchbuf); }