lbajardsilogic@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
lbajardsilogic@0: 
lbajardsilogic@0: /*
lbajardsilogic@0:     Sonic Visualiser
lbajardsilogic@0:     An audio file viewer and annotation editor.
lbajardsilogic@0:     Centre for Digital Music, Queen Mary, University of London.
lbajardsilogic@0:     This file copyright 2006 Chris Cannam.
lbajardsilogic@0:     
lbajardsilogic@0:     This program is free software; you can redistribute it and/or
lbajardsilogic@0:     modify it under the terms of the GNU General Public License as
lbajardsilogic@0:     published by the Free Software Foundation; either version 2 of the
lbajardsilogic@0:     License, or (at your option) any later version.  See the file
lbajardsilogic@0:     COPYING included with this distribution for more information.
lbajardsilogic@0: */
lbajardsilogic@0: 
lbajardsilogic@0: #ifndef _WINDOW_H_
lbajardsilogic@0: #define _WINDOW_H_
lbajardsilogic@0: 
lbajardsilogic@0: #include <cmath>
lbajardsilogic@0: #include <iostream>
lbajardsilogic@0: #include <map>
lbajardsilogic@0: 
lbajardsilogic@0: enum WindowType {
lbajardsilogic@0:     RectangularWindow,
lbajardsilogic@0:     BartlettWindow,
lbajardsilogic@0:     HammingWindow,
lbajardsilogic@0:     HanningWindow,
lbajardsilogic@0:     BlackmanWindow,
lbajardsilogic@0:     GaussianWindow,
lbajardsilogic@0:     ParzenWindow,
lbajardsilogic@0:     NuttallWindow,
lbajardsilogic@0:     BlackmanHarrisWindow
lbajardsilogic@0: };
lbajardsilogic@0: 
lbajardsilogic@0: template <typename T>
lbajardsilogic@0: class Window
lbajardsilogic@0: {
lbajardsilogic@0: public:
lbajardsilogic@0:     /**
lbajardsilogic@0:      * Construct a windower of the given type.
lbajardsilogic@0:      */
lbajardsilogic@0:     Window(WindowType type, size_t size) : m_type(type), m_size(size) { encache(); }
lbajardsilogic@0:     Window(const Window &w) : m_type(w.m_type), m_size(w.m_size) { encache(); }
lbajardsilogic@0:     Window &operator=(const Window &w) {
lbajardsilogic@0: 	if (&w == this) return *this;
lbajardsilogic@0: 	m_type = w.m_type;
lbajardsilogic@0: 	m_size = w.m_size;
lbajardsilogic@0: 	encache();
lbajardsilogic@0: 	return *this;
lbajardsilogic@0:     }
lbajardsilogic@0:     virtual ~Window() { delete[] m_cache; }
lbajardsilogic@0:     
lbajardsilogic@0:     void cut(T *src) const { cut(src, src); }
lbajardsilogic@0:     void cut(T *src, T *dst) const {
lbajardsilogic@0: 	for (size_t i = 0; i < m_size; ++i) dst[i] = src[i] * m_cache[i];
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     T getArea() { return m_area; }
lbajardsilogic@0:     T getValue(size_t i) { return m_cache[i]; }
lbajardsilogic@0: 
lbajardsilogic@0:     WindowType getType() const { return m_type; }
lbajardsilogic@0:     size_t getSize() const { return m_size; }
lbajardsilogic@0: 
lbajardsilogic@0: protected:
lbajardsilogic@0:     WindowType m_type;
lbajardsilogic@0:     size_t m_size;
lbajardsilogic@0:     T *m_cache;
lbajardsilogic@0:     T m_area;
lbajardsilogic@0:     
lbajardsilogic@0:     void encache();
lbajardsilogic@0:     void cosinewin(T *, T, T, T, T);
lbajardsilogic@0: };
lbajardsilogic@0: 
lbajardsilogic@0: template <typename T>
lbajardsilogic@0: void Window<T>::encache()
lbajardsilogic@0: {
lbajardsilogic@0:     int n = int(m_size);
lbajardsilogic@0:     T *mult = new T[n];
lbajardsilogic@0:     int i;
lbajardsilogic@0:     for (i = 0; i < n; ++i) mult[i] = 1.0;
lbajardsilogic@0: 
lbajardsilogic@0:     switch (m_type) {
lbajardsilogic@0: 		
lbajardsilogic@0:     case RectangularWindow:
lbajardsilogic@0: 	for (i = 0; i < n; ++i) {
lbajardsilogic@0: 	    mult[i] *= 0.5;
lbajardsilogic@0: 	}
lbajardsilogic@0: 	break;
lbajardsilogic@0: 	    
lbajardsilogic@0:     case BartlettWindow:
lbajardsilogic@0: 	for (i = 0; i < n/2; ++i) {
lbajardsilogic@0: 	    mult[i] *= (i / T(n/2));
lbajardsilogic@0: 	    mult[i + n/2] *= (1.0 - (i / T(n/2)));
lbajardsilogic@0: 	}
lbajardsilogic@0: 	break;
lbajardsilogic@0: 	    
lbajardsilogic@0:     case HammingWindow:
lbajardsilogic@0:         cosinewin(mult, 0.54, 0.46, 0.0, 0.0);
lbajardsilogic@0: 	break;
lbajardsilogic@0: 	    
lbajardsilogic@0:     case HanningWindow:
lbajardsilogic@0:         cosinewin(mult, 0.50, 0.50, 0.0, 0.0);
lbajardsilogic@0: 	break;
lbajardsilogic@0: 	    
lbajardsilogic@0:     case BlackmanWindow:
lbajardsilogic@0:         cosinewin(mult, 0.42, 0.50, 0.08, 0.0);
lbajardsilogic@0: 	break;
lbajardsilogic@0: 	    
lbajardsilogic@0:     case GaussianWindow:
lbajardsilogic@0: 	for (i = 0; i < n; ++i) {
lbajardsilogic@0:             mult[i] *= pow(2, - pow((i - (n-1)/2.0) / ((n-1)/2.0 / 3), 2));
lbajardsilogic@0: 	}
lbajardsilogic@0: 	break;
lbajardsilogic@0: 	    
lbajardsilogic@0:     case ParzenWindow:
lbajardsilogic@0:     {
lbajardsilogic@0:         int N = n-1;
lbajardsilogic@0:         for (i = 0; i < N/4; ++i) {
lbajardsilogic@0:             T m = 2 * pow(1.0 - (T(N)/2 - i) / (T(N)/2), 3);
lbajardsilogic@0:             mult[i] *= m;
lbajardsilogic@0:             mult[N-i] *= m;
lbajardsilogic@0:         }
lbajardsilogic@0:         for (i = N/4; i <= N/2; ++i) {
lbajardsilogic@0:             int wn = i - N/2;
lbajardsilogic@0:             T m = 1.0 - 6 * pow(wn / (T(N)/2), 2) * (1.0 - abs(wn) / (T(N)/2));
lbajardsilogic@0:             mult[i] *= m;
lbajardsilogic@0:             mult[N-i] *= m;
lbajardsilogic@0:         }            
lbajardsilogic@0:         break;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     case NuttallWindow:
lbajardsilogic@0:         cosinewin(mult, 0.3635819, 0.4891775, 0.1365995, 0.0106411);
lbajardsilogic@0: 	break;
lbajardsilogic@0: 
lbajardsilogic@0:     case BlackmanHarrisWindow:
lbajardsilogic@0:         cosinewin(mult, 0.35875, 0.48829, 0.14128, 0.01168);
lbajardsilogic@0:         break;
lbajardsilogic@0:     }
lbajardsilogic@0: 	
lbajardsilogic@0:     m_cache = mult;
lbajardsilogic@0: 
lbajardsilogic@0:     m_area = 0;
lbajardsilogic@0:     for (int i = 0; i < n; ++i) {
lbajardsilogic@0:         m_area += m_cache[i];
lbajardsilogic@0:     }
lbajardsilogic@0:     m_area /= n;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: template <typename T>
lbajardsilogic@0: void Window<T>::cosinewin(T *mult, T a0, T a1, T a2, T a3)
lbajardsilogic@0: {
lbajardsilogic@0:     int n = int(m_size);
lbajardsilogic@0:     for (int i = 0; i < n; ++i) {
lbajardsilogic@0:         mult[i] *= (a0
lbajardsilogic@0:                     - a1 * cos(2 * M_PI * i / n)
lbajardsilogic@0:                     + a2 * cos(4 * M_PI * i / n)
lbajardsilogic@0:                     - a3 * cos(6 * M_PI * i / n));
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: #endif