gstlal-calibration: new element lal_transferfunction to compute transfer...

gstlal-calibration:  new element lal_transferfunction to compute transfer functions between 2 or more channels in real time.

gstlal-calibration/gst/lal/Makefile.am

@@ -13,7 +13,8 @@ libgstlalcalibration_la_SOURCES = \
 	gstlal_logicalundersample.c gstlal_logicalundersample.h \
 	gstlal_demodulate.c gstlal_demodulate.h \
 	gstlal_insertgap.c gstlal_insertgap.h \
-	gstlal_fccupdate.c gstlal_fccupdate.h
+	gstlal_fccupdate.c gstlal_fccupdate.h \
+	gstlal_transferfunction.c gstlal_transferfunction.h
 libgstlalcalibration_la_CPPFLAGS = $(AM_CPPFLAGS) $(PYTHON_CPPFLAGS)
 libgstlalcalibration_la_CFLAGS = $(AM_CFLAGS) $(LAL_CFLAGS) $(GSTLAL_CFLAGS) $(gstreamer_CFLAGS) $(gstreamer_audio_CFLAGS)
 libgstlalcalibration_la_LDFLAGS = $(AM_LDFLAGS) $(LAL_LIBS) $(GSTLAL_LIBS) $(PYTHON_LIBS) $(gstreamer_LIBS) $(gstreamer_audio_LIBS) $(GSTLAL_PLUGIN_LDFLAGS)

gstlal-calibration/gst/lal/gstlal_transferfunction.h

+/*
+ * Copyright (C) 2017  Aaron Viets <aaron.viets@ligo.org>
+ *
+ *  This program is free software; you can redistribute it and/or modify it
+ *  under the terms of the GNU General Public License as published by the
+ *  Free Software Foundation; either version 2 of the License, or (at your
+ *  option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful, but
+ *  WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ *  General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License along
+ *  with this program; if not, write to the Free Software Foundation, Inc.,
+ *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+
+#ifndef __GSTLAL_TRANSFERFUNCTION_H__
+#define __GSTLAL_TRANSFERFUNCTION_H__
+
+
+#include <complex.h>
+
+#include <glib.h>
+#include <gst/gst.h>
+#include <gst/base/gstbasesink.h>
+
+#include <fftw3.h>
+#include <gsl/gsl_matrix.h>
+#include <gsl/gsl_permutation.h>
+
+
+G_BEGIN_DECLS
+
+
+#define GSTLAL_TRANSFERFUNCTION_TYPE \
+	(gstlal_transferfunction_get_type())
+#define GSTLAL_TRANSFERFUNCTION(obj) \
+	(G_TYPE_CHECK_INSTANCE_CAST((obj), GSTLAL_TRANSFERFUNCTION_TYPE, GSTLALTransferFunction))
+#define GSTLAL_TRANSFERFUNCTION_CLASS(klass) \
+	(G_TYPE_CHECK_CLASS_CAST((klass), GSTLAL_TRANSFERFUNCTION_TYPE, GSTLALTransferFunctionClass))
+#define GST_IS_GSTLAL_TRANSFERFUNCTION(obj) \
+	(G_TYPE_CHECK_INSTANCE_TYPE((obj), GSTLAL_TRANSFERFUNCTION_TYPE))
+#define GST_IS_GSTLAL_TRANSFERFUNCTION_CLASS(klass) \
+	(G_TYPE_CHECK_CLASS_TYPE((klass), GSTLAL_TRANSFERFUNCTION_TYPE))
+
+
+typedef struct _GSTLALTransferFunction GSTLALTransferFunction;
+typedef struct _GSTLALTransferFunctionClass GSTLALTransferFunctionClass;
+
+
+/**
+ * GSTLALTransferFunction:
+ */
+
+
+struct _GSTLALTransferFunction {
+	GstBaseSink basesink;
+
+	/* stream info */
+	gint rate;
+	gint unit_size;
+	gint channels;
+	gint64 sample_count;
+	enum gstlal_transferfunction_data_type {
+		GSTLAL_TRANSFERFUNCTION_F32 = 0,
+		GSTLAL_TRANSFERFUNCTION_F64,
+	} data_type;
+
+	/* timestamp bookkeeping */
+	GstClockTime t0;
+	guint64 offset0;
+	guint64 next_in_offset;
+
+	/* transfer function work space */
+	union {
+		struct {
+			float *fft_window;
+			float *fir_window;
+			double *tukey;
+			float *leftover_data;
+			gint64 num_leftover;
+			complex float *ffts;
+			gint64 num_ffts_in_avg;
+			complex float *autocorrelation_matrix;
+
+			/* gsl stuff */
+			gsl_vector_complex *transfer_functions_at_f;
+			gsl_vector_complex *transfer_functions_solved_at_f;
+			gsl_matrix_complex *autocorrelation_matrix_at_f;
+			gsl_permutation *permutation;
+
+			/* fftwf stuff */
+			complex float *fft;
+			fftwf_plan plan;
+			complex float *fir_filter;
+			fftwf_plan fir_plan;
+		} wspf;  /* workspace single-precision float */
+		struct {
+			double *fft_window;
+			double *fir_window;
+			double *tukey;
+			double *leftover_data;
+			gint64 num_leftover;
+			complex double *ffts;
+			gint64 num_ffts_in_avg;
+			complex double *autocorrelation_matrix;
+
+			/* gsl stuff */
+			gsl_vector_complex *transfer_functions_at_f;
+			gsl_vector_complex *transfer_functions_solved_at_f;
+			gsl_matrix_complex *autocorrelation_matrix_at_f;
+			gsl_permutation *permutation;
+
+			/* fftw stuff */
+			complex double *fft;
+			fftw_plan plan;
+			complex double *fir_filter;
+			fftw_plan fir_plan;
+		} wdpf;  /* workspace double-precision float */
+	} workspace;
+
+	/* properties */
+	gint64 fft_length;
+	gint64 fft_overlap;
+	gint64 num_ffts;
+	gint64 update_samples;
+	gboolean update_after_gap;
+	gboolean write_to_screen;
+	char *filename;
+	gboolean make_fir_filters;
+	int high_pass;
+	int low_pass;
+	complex double *transfer_functions;
+	double *fir_filters;
+};
+
+
+/**
+ * GSTLALTransferFunctionClass:
+ * @parent_class:  the parent class
+ */
+
+
+struct _GSTLALTransferFunctionClass {
+	GstBaseSinkClass parent_class;
+};
+
+
+GType gstlal_transferfunction_get_type(void);
+
+
+G_END_DECLS
+
+
+#endif	/* __GSTLAL_TRANSFERFUNCTION_H__ */

gstlal-calibration/gst/lal/gstlalcalibration.c

@@ -62,6 +62,7 @@
 #include <gstlal_demodulate.h>
 #include <gstlal_insertgap.h>
 #include <gstlal_fccupdate.h>
+#include <gstlal_transferfunction.h>
 
 
 /*
@@ -91,6 +92,7 @@ static gboolean plugin_init(GstPlugin *plugin)
 		{"lal_demodulate", GSTLAL_DEMODULATE_TYPE},
 		{"lal_insertgap", GSTLAL_INSERTGAP_TYPE},
 		{"lal_fcc_update", GSTLAL_FCC_UPDATE_TYPE},
+		{"lal_transferfunction", GSTLAL_TRANSFERFUNCTION_TYPE},
 		{NULL, 0},
 	};