virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-03-03 14:32:16 +02:00
Code Issues Releases Activity

avfilter/vf_colormap: avoid reallocating memory to build map

Browse Source
This commit is contained in:
Paul B Mahol 2022-04-26 09:11:31 +02:00
parent 494139bcc7
commit 08ac8bda38
1 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
libavfilter/vf_colormap.c
View File

@ -62,6 +62,10 @@ typedef struct ColorMapContext {
float (*kernel)(const float *x, const float *y);
FFFrameSync fs;
double A[(MAX_SIZE + 4) * (MAX_SIZE + 4)];
double b[MAX_SIZE + 4];
int pivot[MAX_SIZE + 4];
} ColorMapContext;
#define OFFSET(x) offsetof(ColorMapContext, x)
@ -262,12 +266,9 @@ static void build_map(AVFilterContext *ctx)
{
const int N = s->nb_maps;
const int N4 = N + 4;
double *A = av_calloc(sizeof(*A), N4 * N4);
double *b = av_calloc(sizeof(*b), N4);
int *pivot = NULL;
if (!A || !b)
goto error;
double *A = s->A;
double *b = s->b;
int *pivot = s->pivot;
for (int j = 0; j < N; j++)
for (int i = j; i < N; i++)
@ -286,10 +287,6 @@ static void build_map(AVFilterContext *ctx)
for (int i = N;i < N4; i++)
A[j * N4 + i] = 0.;
pivot = av_calloc(N4, sizeof(*pivot));
if (!pivot)
goto error;
if (gauss_make_triangular(A, pivot, N4)) {
for (int i = 0; i < N; i++)
b[i] = s->target[i][c];
@ -304,10 +301,6 @@ static void build_map(AVFilterContext *ctx)
for (int i = 0; i < 4; i++)
s->icoeff[i][c] = b[N + i];
}
error:
av_free(pivot);
av_free(b);
av_free(A);
}
}
}