src/wavelet.h

Go to the documentation of this file.

/* **************************************************************************
 *   Copyright (C) 2008 by Matthias Ihrke   *
 *   mihrke@uni-goettingen.de   *
 *                                                                         *
 *   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 WAVELET_H
# define WAVELET_H

#include "definitions.h"
#include <gsl/gsl_wavelet.h>

#include <gsl/gsl_rng.h>
#include <gsl/gsl_randist.h>

#include "eeg.h"

#ifdef __cplusplus
extern "C" {
#endif
  /*-----------------------------------------------------------
     - Function Pointers
     ---------------------------------------------------------*/
  typedef double  (*ThresholdSelectionFunction)   (const double*,int);
  typedef double  (*ThresholdFunction)            (double,double);
  typedef double* (*SignalExtensionFunction)      (double*,int,int);

  typedef struct {
     int first_thresholding_level; 
     gsl_wavelet_type *wavelet;         
     int vanishing_moments;        
     ThresholdSelectionFunction threshselfct; 
     ThresholdFunction threshfct;             
     SignalExtensionFunction sigextfct;       
  } WaveletParameters;

  /* ---------------------------------------------------------------------------- 
      -- Wavelet-based Denoising routines                                       -- 
      ---------------------------------------------------------------------------- */
  double soft_thresholding( double d, double lambda );
  double hard_thresholding( double d, double lambda );

  double translation_invariant_thresholding( const double *data, int n );
  double conventional_thresholding         ( const double *data, int n );
  double sureshrink                        ( const double *data, int n );
  double heuristic_sure                    ( const double *data, int n );

  WaveletParameters wavelet_init();
  int  wavelet_denoise           ( double *data, int n, WaveletParameters P );
  int  wavelet_extend_and_denoise( double *data, int n, WaveletParameters P );
  EEG* eeg_wavelet_denoise       ( EEG *eeg, WaveletParameters P, bool alloc );

#ifdef __cplusplus
}
#endif
#endif /* - WAVELET_H - */