Blame doc/examples/gaussfilt2.c

Packit 67cb25
#include <stdio.h>
Packit 67cb25
#include <stdlib.h>
Packit 67cb25
Packit 67cb25
#include <gsl/gsl_math.h>
Packit 67cb25
#include <gsl/gsl_filter.h>
Packit 67cb25
#include <gsl/gsl_rng.h>
Packit 67cb25
#include <gsl/gsl_randist.h>
Packit 67cb25
#include <gsl/gsl_vector.h>
Packit 67cb25
Packit 67cb25
int
Packit 67cb25
main(void)
Packit 67cb25
{
Packit 67cb25
  const size_t N = 1000;                       /* length of time series */
Packit 67cb25
  const size_t K = 61;                         /* window size */
Packit 67cb25
  const double alpha = 3.0;                    /* Gaussian kernel has +/- 3 standard deviations */
Packit 67cb25
  gsl_vector *x = gsl_vector_alloc(N);         /* input vector */
Packit 67cb25
  gsl_vector *y = gsl_vector_alloc(N);         /* filtered output vector */
Packit 67cb25
  gsl_vector *dy = gsl_vector_alloc(N);        /* first derivative filtered vector */
Packit 67cb25
  gsl_vector *d2y = gsl_vector_alloc(N);       /* second derivative filtered vector */
Packit 67cb25
  gsl_rng *r = gsl_rng_alloc(gsl_rng_default);
Packit 67cb25
  gsl_filter_gaussian_workspace *gauss_p = gsl_filter_gaussian_alloc(K);
Packit 67cb25
  size_t i;
Packit 67cb25
Packit 67cb25
  /* generate input signal */
Packit 67cb25
  for (i = 0; i < N; ++i)
Packit 67cb25
    {
Packit 67cb25
      double xi = (i > N / 2) ? 0.5 : 0.0;
Packit 67cb25
      double ei = gsl_ran_gaussian(r, 0.1);
Packit 67cb25
Packit 67cb25
      gsl_vector_set(x, i, xi + ei);
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  /* apply filters */
Packit 67cb25
  gsl_filter_gaussian(GSL_FILTER_END_PADVALUE, alpha, 0, x, y, gauss_p);
Packit 67cb25
  gsl_filter_gaussian(GSL_FILTER_END_PADVALUE, alpha, 1, x, dy, gauss_p);
Packit 67cb25
  gsl_filter_gaussian(GSL_FILTER_END_PADVALUE, alpha, 2, x, d2y, gauss_p);
Packit 67cb25
Packit 67cb25
  /* print results */
Packit 67cb25
  for (i = 0; i < N; ++i)
Packit 67cb25
    {
Packit 67cb25
      double xi = gsl_vector_get(x, i);
Packit 67cb25
      double yi = gsl_vector_get(y, i);
Packit 67cb25
      double dyi = gsl_vector_get(dy, i);
Packit 67cb25
      double d2yi = gsl_vector_get(d2y, i);
Packit 67cb25
      double dxi;
Packit 67cb25
Packit 67cb25
      /* compute finite difference of x vector */
Packit 67cb25
      if (i == 0)
Packit 67cb25
        dxi = gsl_vector_get(x, i + 1) - xi;
Packit 67cb25
      else if (i == N - 1)
Packit 67cb25
        dxi = gsl_vector_get(x, i) - gsl_vector_get(x, i - 1);
Packit 67cb25
      else
Packit 67cb25
        dxi = 0.5 * (gsl_vector_get(x, i + 1) - gsl_vector_get(x, i - 1));
Packit 67cb25
Packit 67cb25
      printf("%.12e %.12e %.12e %.12e %.12e\n",
Packit 67cb25
             xi,
Packit 67cb25
             yi,
Packit 67cb25
             dxi,
Packit 67cb25
             dyi,
Packit 67cb25
             d2yi);
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  gsl_vector_free(x);
Packit 67cb25
  gsl_vector_free(y);
Packit 67cb25
  gsl_vector_free(dy);
Packit 67cb25
  gsl_vector_free(d2y);
Packit 67cb25
  gsl_rng_free(r);
Packit 67cb25
  gsl_filter_gaussian_free(gauss_p);
Packit 67cb25
Packit 67cb25
  return 0;
Packit 67cb25
}