changeset 29494:86786d090e11

Introduce and use sws_allocVec().
author ramiro
date Wed, 19 Aug 2009 01:33:17 +0000
parents 335da85c454c
children 735f494d56e0
files libswscale/swscale.c libswscale/swscale.h
diffstat 2 files changed, 24 insertions(+), 19 deletions(-) [+]
line wrap: on
line diff
--- a/libswscale/swscale.c	Wed Aug 19 01:32:06 2009 +0000
+++ b/libswscale/swscale.c	Wed Aug 19 01:33:17 2009 +0000
@@ -3240,20 +3240,28 @@
     return filter;
 }
 
+SwsVector *sws_allocVec(int length)
+{
+    SwsVector *vec = av_malloc(sizeof(SwsVector));
+    if (!vec)
+        return NULL;
+    vec->length = length;
+    vec->coeff  = av_malloc(sizeof(double) * length);
+    if (!vec->coeff)
+        av_freep(&vec);
+    return vec;
+}
+
 SwsVector *sws_getGaussianVec(double variance, double quality)
 {
     const int length= (int)(variance*quality + 0.5) | 1;
     int i;
-    double *coeff= av_malloc(length*sizeof(double));
     double middle= (length-1)*0.5;
-    SwsVector *vec= av_malloc(sizeof(SwsVector));
-
-    vec->coeff= coeff;
-    vec->length= length;
+    SwsVector *vec= sws_allocVec(length);
 
     for (i=0; i<length; i++) {
         double dist= i-middle;
-        coeff[i]= exp(-dist*dist/(2*variance*variance)) / sqrt(2*variance*PI);
+        vec->coeff[i]= exp(-dist*dist/(2*variance*variance)) / sqrt(2*variance*PI);
     }
 
     sws_normalizeVec(vec, 1.0);
@@ -3264,14 +3272,10 @@
 SwsVector *sws_getConstVec(double c, int length)
 {
     int i;
-    double *coeff= av_malloc(length*sizeof(double));
-    SwsVector *vec= av_malloc(sizeof(SwsVector));
-
-    vec->coeff= coeff;
-    vec->length= length;
+    SwsVector *vec= sws_allocVec(length);
 
     for (i=0; i<length; i++)
-        coeff[i]= c;
+        vec->coeff[i]= c;
 
     return vec;
 }
@@ -3397,14 +3401,10 @@
 
 SwsVector *sws_cloneVec(SwsVector *a)
 {
-    double *coeff= av_malloc(a->length*sizeof(double));
     int i;
-    SwsVector *vec= av_malloc(sizeof(SwsVector));
-
-    vec->coeff= coeff;
-    vec->length= a->length;
-
-    for (i=0; i<a->length; i++) coeff[i]= a->coeff[i];
+    SwsVector *vec= sws_allocVec(a->length);
+
+    for (i=0; i<a->length; i++) vec->coeff[i]= a->coeff[i];
 
     return vec;
 }
--- a/libswscale/swscale.h	Wed Aug 19 01:32:06 2009 +0000
+++ b/libswscale/swscale.h	Wed Aug 19 01:33:17 2009 +0000
@@ -182,6 +182,11 @@
                              int *brightness, int *contrast, int *saturation);
 
 /**
+ * Allocates and returns an uninitialized vector with length coefficients.
+ */
+SwsVector *sws_allocVec(int length);
+
+/**
  * Returns a normalized Gaussian curve used to filter stuff
  * quality=3 is high quality, lower is lower quality.
  */