Added complex number support to normalization function.

Author: Unknown

rss_ringoccs/diffrec/src/__window_function_normalization.c

@@ -111,8 +111,8 @@ double Window_Normalization_Long(long *ker, long dim,
 }
 
 double Window_Normalization_Long_Long(long long *ker, long dim,
-                                      double dx, double f_scale){
-
+                                      double dx, double f_scale)
+{
     /*  Declare variable for indexing.                                        */
     long i;
 
@@ -128,4 +128,59 @@ double Window_Normalization_Long_Long(long long *ker, long dim,
     return SQRT_2 * f_scale / T1;
 }
 
+float Window_Normalization_Complex_Float(complex float *ker, long dim,
+                                         float dx, float f_scale)
+{
+    /*  Declare variable for indexing.                                        */
+    long i;
+
+    /*  Compute the Free-Space integral.                                      */
+    complex float T1 = 0.0;
+
+    for (i=0; i<dim; ++i){
+        T1 += ker[i];
+    }
+    T1 = cabsf(T1 * dx);
+
+    /* Retur the normalization factor.                                        */
+    return SQRT_2 * f_scale / T1;
+}
+
+double Window_Normalization_Complex_Double(complex double *ker, long dim,
+                                           double dx, double f_scale)
+{
+    /*  Declare variable for indexing.                                        */
+    long i;
+
+    /*  Compute the Free-Space integral.                                      */
+    complex double T1 = 0.0;
+
+    for (i=0; i<dim; ++i){
+        T1 += ker[i];
+    }
+    T1 = cabs(T1 * dx);
+
+    /* Retur the normalization factor.                                        */
+    return SQRT_2 * f_scale / T1;
+}
+
+long double Window_Normalization_Complex_Long_Double(complex long double *ker,
+                                                     long dim, long double dx,
+                                                     long double f_scale)
+{
+    /*  Declare variable for indexing.                                        */
+    long i;
+
+    /*  Compute the Free-Space integral.                                      */
+    complex long double T1 = 0.0;
+
+    for (i=0; i<dim; ++i){
+        T1 += ker[i];
+    }
+    T1 = cabsl(T1 * dx);
+
+    /* Retur the normalization factor.                                        */
+    return SQRT_2 * f_scale / T1;
+}
+
 #endif

rss_ringoccs/diffrec/src/__window_functions.h

@@ -1,6 +1,7 @@
 #ifndef RSS_RINGOCCS_WINDOW_FUNCTIONS_H
 #define RSS_RINGOCCS_WINDOW_FUNCTIONS_H
 
+#include <complex.h>
 #include "__math_functions.h"
 
 /*  Kaiser-Bessel function with alpha = 2pi                                   */
@@ -86,6 +87,16 @@ extern double Window_Normalization_Long(long *ker, long dim,
 double Window_Normalization_Long_Long(long long *ker, long dim,
                                       double dx, double f_scale);
 
+extern float Window_Normalization_Complex_Float(complex float *ker, long dim,
+                                                float dx, float f_scale);
+
+extern double Window_Normalization_Complex_Double(complex double *ker, long dim,
+                                                  double dx, double f_scale);
+
+extern long double Window_Normalization_Complex_Long_Double(
+    complex long double *ker, long dim, long double dx, long double f_scale
+);
+
 /*  Kaiser-Bessel function with arbitrary alpha.                              */
 extern float Kaiser_Bessel_Al_Float(float x, float W, float alpha);
 

rss_ringoccs/diffrec/src/_window_functions.c

@@ -51,7 +51,28 @@ static PyObject *window_norm(PyObject *self, PyObject *args){
         dim     = PyArray_DIMS(arr)[0];
         data    = PyArray_DATA(arr);
 
-        if (typenum == NPY_FLOAT){
+        if (typenum == NPY_CFLOAT){
+            return PyFloat_FromDouble(
+                Window_Normalization_Complex_Float(
+                    (complex float *)data, dim, dx, f_scale
+                )
+            );
+        }
+        else if (typenum == NPY_CDOUBLE){
+            return PyFloat_FromDouble(
+                Window_Normalization_Complex_Double(
+                    (complex double *)data, dim, dx, f_scale
+                )
+            );
+        }
+        else if (typenum == NPY_CLONGDOUBLE){
+            return PyFloat_FromDouble(
+                Window_Normalization_Complex_Long_Double(
+                    (complex long double *)data, dim, dx, f_scale
+                )
+            );
+        }
+        else if (typenum == NPY_FLOAT){
             return PyFloat_FromDouble(
                 Window_Normalization_Float((float *)data, dim, dx, f_scale)
             );