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@191: #include <iostream>
lbajardsilogic@191: #include <vector>
lbajardsilogic@191: 
lbajardsilogic@0: #include "system/System.h"
lbajardsilogic@0: 
lbajardsilogic@191: #include "FFTMemoryCache.h"
lbajardsilogic@0: 
lbajardsilogic@0: FFTMemoryCache::FFTMemoryCache() :
lbajardsilogic@0:     m_width(0),
lbajardsilogic@0:     m_height(0),
lbajardsilogic@0:     m_magnitude(0),
lbajardsilogic@0:     m_phase(0),
lbajardsilogic@0:     m_factor(0)
lbajardsilogic@0: {
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: FFTMemoryCache::~FFTMemoryCache()
lbajardsilogic@0: {
lbajardsilogic@0: //    std::cerr << "FFTMemoryCache[" << this << "]::~Cache" << std::endl;
lbajardsilogic@0: 
lbajardsilogic@0:     for (size_t i = 0; i < m_width; ++i) {
lbajardsilogic@0: 	if (m_magnitude && m_magnitude[i]) free(m_magnitude[i]);
lbajardsilogic@0: 	if (m_phase && m_phase[i]) free(m_phase[i]);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_magnitude) free(m_magnitude);
lbajardsilogic@0:     if (m_phase) free(m_phase);
lbajardsilogic@0:     if (m_factor) free(m_factor);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: FFTMemoryCache::resize(size_t width, size_t height)
lbajardsilogic@0: {
lbajardsilogic@0: //    std::cerr << "FFTMemoryCache[" << this << "]::resize(" << width << "x" << height << " = " << width*height << ")" << std::endl;
lbajardsilogic@0:     
lbajardsilogic@0:     if (m_width == width && m_height == height) return;
lbajardsilogic@0: 
lbajardsilogic@0:     resize(m_magnitude, width, height);
lbajardsilogic@0:     resize(m_phase, width, height);
lbajardsilogic@0:     m_colset.resize(width);
lbajardsilogic@0: 
lbajardsilogic@0:     m_factor = (float *)realloc(m_factor, width * sizeof(float));
lbajardsilogic@0: 
lbajardsilogic@0:     m_width = width;
lbajardsilogic@0:     m_height = height;
lbajardsilogic@0: 
lbajardsilogic@0: //    std::cerr << "done, width = " << m_width << " height = " << m_height << std::endl;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: FFTMemoryCache::resize(uint16_t **&array, size_t width, size_t height)
lbajardsilogic@0: {
lbajardsilogic@0:     for (size_t i = width; i < m_width; ++i) {
lbajardsilogic@0: 	free(array[i]);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (width != m_width) {
lbajardsilogic@0: 	array = (uint16_t **)realloc(array, width * sizeof(uint16_t *));
lbajardsilogic@0: 	if (!array) throw std::bad_alloc();
lbajardsilogic@0: 	MUNLOCK(array, width * sizeof(uint16_t *));
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     for (size_t i = m_width; i < width; ++i) {
lbajardsilogic@0: 	array[i] = 0;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     for (size_t i = 0; i < width; ++i) {
lbajardsilogic@0: 	array[i] = (uint16_t *)realloc(array[i], height * sizeof(uint16_t));
lbajardsilogic@0: 	if (!array[i]) throw std::bad_alloc();
lbajardsilogic@0: 	MUNLOCK(array[i], height * sizeof(uint16_t));
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: FFTMemoryCache::reset()
lbajardsilogic@0: {
lbajardsilogic@0:     for (size_t x = 0; x < m_width; ++x) {
lbajardsilogic@0: 	for (size_t y = 0; y < m_height; ++y) {
lbajardsilogic@0: 	    m_magnitude[x][y] = 0;
lbajardsilogic@0: 	    m_phase[x][y] = 0;
lbajardsilogic@0: 	}
lbajardsilogic@0: 	m_factor[x] = 1.0;
lbajardsilogic@0:     }
lbajardsilogic@0: }	    
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: FFTMemoryCache::setColumnAt(size_t x, float *reals, float *imags)
lbajardsilogic@0: {
lbajardsilogic@0:     float max = 0.0;
lbajardsilogic@0: 
lbajardsilogic@0:     for (size_t y = 0; y < m_height; ++y) {
lbajardsilogic@0:         float mag = sqrtf(reals[y] * reals[y] + imags[y] * imags[y]);
lbajardsilogic@0:         float phase = atan2f(imags[y], reals[y]);
lbajardsilogic@0:         phase = princargf(phase);
lbajardsilogic@0:         reals[y] = mag;
lbajardsilogic@0:         imags[y] = phase;
lbajardsilogic@0:         if (mag > max) max = mag;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     setColumnAt(x, reals, imags, max);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: size_t
lbajardsilogic@0: FFTMemoryCache::getCacheSize(size_t width, size_t height)
lbajardsilogic@0: {
lbajardsilogic@0:     return (height * 2 + 1) * width * sizeof(uint16_t);
lbajardsilogic@0: }
lbajardsilogic@0: