Chris@159: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@159: 
Chris@159: /*
Chris@159:     Sonic Visualiser
Chris@159:     An audio file viewer and annotation editor.
Chris@159:     Centre for Digital Music, Queen Mary, University of London.
Chris@159:     This file copyright 2006 Chris Cannam.
Chris@159:     
Chris@159:     This program is free software; you can redistribute it and/or
Chris@159:     modify it under the terms of the GNU General Public License as
Chris@159:     published by the Free Software Foundation; either version 2 of the
Chris@159:     License, or (at your option) any later version.  See the file
Chris@159:     COPYING included with this distribution for more information.
Chris@159: */
Chris@159: 
Chris@159: #include "FFTMemoryCache.h"
Chris@159: #include "system/System.h"
Chris@159: 
Chris@159: #include <iostream>
Chris@159: 
Chris@374: //#define DEBUG_FFT_MEMORY_CACHE 1
Chris@374: 
Chris@264: FFTMemoryCache::FFTMemoryCache(StorageType storageType) :
Chris@159:     m_width(0),
Chris@159:     m_height(0),
Chris@159:     m_magnitude(0),
Chris@159:     m_phase(0),
Chris@264:     m_fmagnitude(0),
Chris@264:     m_fphase(0),
Chris@334:     m_freal(0),
Chris@334:     m_fimag(0),
Chris@264:     m_factor(0),
Chris@264:     m_storageType(storageType)
Chris@159: {
Chris@374: #ifdef DEBUG_FFT_MEMORY_CACHE
Chris@264:     std::cerr << "FFTMemoryCache[" << this << "]::FFTMemoryCache (type "
Chris@264:               << m_storageType << ")" << std::endl;
Chris@374: #endif
Chris@159: }
Chris@159: 
Chris@159: FFTMemoryCache::~FFTMemoryCache()
Chris@159: {
Chris@374: #ifdef DEBUG_FFT_MEMORY_CACHE
Chris@374:     std::cerr << "FFTMemoryCache[" << this << "]::~FFTMemoryCache" << std::endl;
Chris@374: #endif
Chris@159: 
Chris@159:     for (size_t i = 0; i < m_width; ++i) {
Chris@159: 	if (m_magnitude && m_magnitude[i]) free(m_magnitude[i]);
Chris@159: 	if (m_phase && m_phase[i]) free(m_phase[i]);
Chris@264: 	if (m_fmagnitude && m_fmagnitude[i]) free(m_fmagnitude[i]);
Chris@264: 	if (m_fphase && m_fphase[i]) free(m_fphase[i]);
Chris@334:         if (m_freal && m_freal[i]) free(m_freal[i]);
Chris@334:         if (m_fimag && m_fimag[i]) free(m_fimag[i]);
Chris@159:     }
Chris@159: 
Chris@159:     if (m_magnitude) free(m_magnitude);
Chris@159:     if (m_phase) free(m_phase);
Chris@264:     if (m_fmagnitude) free(m_fmagnitude);
Chris@264:     if (m_fphase) free(m_fphase);
Chris@334:     if (m_freal) free(m_freal);
Chris@334:     if (m_fimag) free(m_fimag);
Chris@159:     if (m_factor) free(m_factor);
Chris@159: }
Chris@159: 
Chris@159: void
Chris@159: FFTMemoryCache::resize(size_t width, size_t height)
Chris@159: {
Chris@408:     Profiler profiler("FFTMemoryCache::resize");
Chris@408: 
Chris@374: #ifdef DEBUG_FFT_MEMORY_CACHE
Chris@359:     std::cerr << "FFTMemoryCache[" << this << "]::resize(" << width << "x" << height << " = " << width*height << ")" << std::endl;
Chris@374: #endif
Chris@159:     
Chris@159:     if (m_width == width && m_height == height) return;
Chris@159: 
Chris@264:     if (m_storageType == Compact) {
Chris@264:         resize(m_magnitude, width, height);
Chris@264:         resize(m_phase, width, height);
Chris@334:     } else if (m_storageType == Polar) {
Chris@264:         resize(m_fmagnitude, width, height);
Chris@264:         resize(m_fphase, width, height);
Chris@334:     } else {
Chris@334:         resize(m_freal, width, height);
Chris@334:         resize(m_fimag, width, height);
Chris@264:     }
Chris@264: 
Chris@159:     m_colset.resize(width);
Chris@159: 
Chris@159:     m_factor = (float *)realloc(m_factor, width * sizeof(float));
Chris@159: 
Chris@159:     m_width = width;
Chris@159:     m_height = height;
Chris@159: 
Chris@374: #ifdef DEBUG_FFT_MEMORY_CACHE
Chris@374:     std::cerr << "done, width = " << m_width << " height = " << m_height << std::endl;
Chris@374: #endif
Chris@159: }
Chris@159: 
Chris@159: void
Chris@159: FFTMemoryCache::resize(uint16_t **&array, size_t width, size_t height)
Chris@159: {
Chris@159:     for (size_t i = width; i < m_width; ++i) {
Chris@159: 	free(array[i]);
Chris@159:     }
Chris@159: 
Chris@159:     if (width != m_width) {
Chris@159: 	array = (uint16_t **)realloc(array, width * sizeof(uint16_t *));
Chris@159: 	if (!array) throw std::bad_alloc();
Chris@159: 	MUNLOCK(array, width * sizeof(uint16_t *));
Chris@159:     }
Chris@159: 
Chris@159:     for (size_t i = m_width; i < width; ++i) {
Chris@159: 	array[i] = 0;
Chris@159:     }
Chris@159: 
Chris@159:     for (size_t i = 0; i < width; ++i) {
Chris@159: 	array[i] = (uint16_t *)realloc(array[i], height * sizeof(uint16_t));
Chris@159: 	if (!array[i]) throw std::bad_alloc();
Chris@159: 	MUNLOCK(array[i], height * sizeof(uint16_t));
Chris@159:     }
Chris@159: }
Chris@159: 
Chris@159: void
Chris@264: FFTMemoryCache::resize(float **&array, size_t width, size_t height)
Chris@264: {
Chris@264:     for (size_t i = width; i < m_width; ++i) {
Chris@264: 	free(array[i]);
Chris@264:     }
Chris@264: 
Chris@264:     if (width != m_width) {
Chris@264: 	array = (float **)realloc(array, width * sizeof(float *));
Chris@264: 	if (!array) throw std::bad_alloc();
Chris@264: 	MUNLOCK(array, width * sizeof(float *));
Chris@264:     }
Chris@264: 
Chris@264:     for (size_t i = m_width; i < width; ++i) {
Chris@264: 	array[i] = 0;
Chris@264:     }
Chris@264: 
Chris@264:     for (size_t i = 0; i < width; ++i) {
Chris@264: 	array[i] = (float *)realloc(array[i], height * sizeof(float));
Chris@264: 	if (!array[i]) throw std::bad_alloc();
Chris@264: 	MUNLOCK(array[i], height * sizeof(float));
Chris@264:     }
Chris@264: }
Chris@264: 
Chris@264: void
Chris@159: FFTMemoryCache::reset()
Chris@159: {
Chris@264:     switch (m_storageType) {
Chris@264: 
Chris@264:     case Compact:
Chris@264:         for (size_t x = 0; x < m_width; ++x) {
Chris@264:             for (size_t y = 0; y < m_height; ++y) {
Chris@264:                 m_magnitude[x][y] = 0;
Chris@264:                 m_phase[x][y] = 0;
Chris@264:             }
Chris@264:             m_factor[x] = 1.0;
Chris@264:         }
Chris@264:         break;
Chris@264:         
Chris@264:     case Polar:
Chris@264:         for (size_t x = 0; x < m_width; ++x) {
Chris@264:             for (size_t y = 0; y < m_height; ++y) {
Chris@264:                 m_fmagnitude[x][y] = 0;
Chris@264:                 m_fphase[x][y] = 0;
Chris@264:             }
Chris@264:             m_factor[x] = 1.0;
Chris@264:         }
Chris@264:         break;
Chris@334: 
Chris@334:     case Rectangular:
Chris@334:         for (size_t x = 0; x < m_width; ++x) {
Chris@334:             for (size_t y = 0; y < m_height; ++y) {
Chris@334:                 m_freal[x][y] = 0;
Chris@334:                 m_fimag[x][y] = 0;
Chris@334:             }
Chris@334:             m_factor[x] = 1.0;
Chris@334:         }
Chris@334:         break;        
Chris@159:     }
Chris@159: }	    
Chris@159: 
Chris@159: void
Chris@334: FFTMemoryCache::setColumnAt(size_t x, float *mags, float *phases, float factor)
Chris@334: {
Chris@408:     Profiler profiler("FFTMemoryCache::setColumnAt: from polar");
Chris@408: 
Chris@334:     setNormalizationFactor(x, factor);
Chris@334: 
Chris@334:     if (m_storageType == Rectangular) {
Chris@408:         Profiler subprof("FFTMemoryCache::setColumnAt: polar to cart");
Chris@334:         for (size_t y = 0; y < m_height; ++y) {
Chris@334:             m_freal[x][y] = mags[y] * cosf(phases[y]);
Chris@334:             m_fimag[x][y] = mags[y] * sinf(phases[y]);
Chris@334:         }
Chris@334:     } else {
Chris@334:         for (size_t y = 0; y < m_height; ++y) {
Chris@334:             setMagnitudeAt(x, y, mags[y]);
Chris@334:             setPhaseAt(x, y, phases[y]);
Chris@334:         }
Chris@334:     }
Chris@334: 
Chris@334:     m_colset.set(x);
Chris@334: }
Chris@334: 
Chris@334: void
Chris@159: FFTMemoryCache::setColumnAt(size_t x, float *reals, float *imags)
Chris@159: {
Chris@408:     Profiler profiler("FFTMemoryCache::setColumnAt: from cart");
Chris@408: 
Chris@159:     float max = 0.0;
Chris@159: 
Chris@264:     switch (m_storageType) {
Chris@264: 
Chris@334:     case Rectangular:
Chris@334:         for (size_t y = 0; y < m_height; ++y) {
Chris@334:             m_freal[x][y] = reals[y];
Chris@334:             m_fimag[x][y] = imags[y];
Chris@334:             float mag = sqrtf(reals[y] * reals[y] + imags[y] * imags[y]);
Chris@334:             if (mag > max) max = mag;
Chris@334:         }
Chris@334:         break;
Chris@334: 
Chris@264:     case Compact:
Chris@264:     case Polar:
Chris@408:     {
Chris@408:         Profiler subprof("FFTMemoryCache::setColumnAt: cart to polar");
Chris@264:         for (size_t y = 0; y < m_height; ++y) {
Chris@264:             float mag = sqrtf(reals[y] * reals[y] + imags[y] * imags[y]);
Chris@264:             float phase = atan2f(imags[y], reals[y]);
Chris@264:             reals[y] = mag;
Chris@264:             imags[y] = phase;
Chris@264:             if (mag > max) max = mag;
Chris@264:         }
Chris@264:         break;
Chris@408:     }
Chris@264:     };
Chris@159: 
Chris@334:     if (m_storageType == Rectangular) {
Chris@334:         m_factor[x] = max;
Chris@334:         m_colset.set(x);
Chris@334:     } else {
Chris@334:         setColumnAt(x, reals, imags, max);
Chris@334:     }
Chris@159: }
Chris@159: 
Chris@170: size_t
Chris@264: FFTMemoryCache::getCacheSize(size_t width, size_t height, StorageType type)
Chris@170: {
Chris@264:     size_t sz = 0;
Chris@264: 
Chris@264:     switch (type) {
Chris@264: 
Chris@264:     case Compact:
Chris@264:         sz = (height * 2 + 1) * width * sizeof(uint16_t);
Chris@264: 
Chris@264:     case Polar:
Chris@334:     case Rectangular:
Chris@264:         sz = (height * 2 + 1) * width * sizeof(float);
Chris@264:     }
Chris@264: 
Chris@264:     return sz;
Chris@170: }
Chris@170: