c@225: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
c@225: 
c@225: /*
c@225:     QM DSP Library
c@225: 
c@225:     Centre for Digital Music, Queen Mary, University of London.
c@309:     This file 2005-2006 Christian Landone.
c@309: 
c@309:     This program is free software; you can redistribute it and/or
c@309:     modify it under the terms of the GNU General Public License as
c@309:     published by the Free Software Foundation; either version 2 of the
c@309:     License, or (at your option) any later version.  See the file
c@309:     COPYING included with this distribution for more information.
c@225: */
c@225: 
c@225: #include "Framer.h"
c@225: 
cannam@502: #include <limits.h>
c@225: 
cannam@502: Framer::Framer() :
cannam@502:     m_sampleLen(0),
cannam@502:     m_framesRead(0),
cannam@502:     m_srcBuffer(0),
cannam@502:     m_dataFrame(0),
cannam@502:     m_strideFrame(0),
cannam@502:     m_frameLength(0),
cannam@502:     m_stepSize(0),
cannam@502:     m_maxFrames(0),
cannam@502:     m_srcIndex(0)
c@225: {
c@225: }
c@225: 
c@225: Framer::~Framer()
c@225: {
cannam@502:     delete[] m_dataFrame;
cannam@502:     delete[] m_strideFrame;
c@225: }
c@225: 
cannam@502: void Framer::configure(int frameLength, int hop)
c@225: {
c@225:     m_frameLength = frameLength;
c@225:     m_stepSize = hop;
c@225: 
c@225:     resetCounters();
c@225: 
cannam@502:     delete[] m_dataFrame;  
c@225:     m_dataFrame = new double[ m_frameLength ];
c@225: 
cannam@502:     delete [] m_strideFrame;        
c@225:     m_strideFrame = new double[ m_stepSize ];
c@225: }
c@225: 
c@225: void Framer::getFrame(double *dst)
c@225: {
cannam@502:     if ((m_srcIndex + int64_t(m_frameLength)) < m_sampleLen) {
c@225: 
cannam@502:         for (int i = 0; i < m_frameLength; i++) {
cannam@502:             dst[i] = m_srcBuffer[m_srcIndex++]; 
cannam@502:         }
cannam@502:         m_srcIndex -= (m_frameLength - m_stepSize);
c@225: 
cannam@502:     } else { // m_srcIndex is within m_frameLength of m_sampleLen
c@225: 
cannam@502:         int rem = int(m_sampleLen - m_srcIndex);
cannam@502:         int zero = m_frameLength - rem;
cannam@483: 
cannam@502:         for (int i = 0; i < rem; i++) {
cannam@502:             dst[i] = m_srcBuffer[m_srcIndex++];
cannam@483:         }
cannam@483:                 
cannam@502:         for (int i = 0; i < zero; i++ ) {
cannam@502:             dst[rem + i] = 0.0;
cannam@483:         }
cannam@483: 
cannam@502:         m_srcIndex -= (rem - m_stepSize);
c@225:     }
c@225: 
c@225:     m_framesRead++;
c@225: }
c@225: 
c@225: void Framer::resetCounters()
c@225: {
c@225:     m_framesRead = 0;
cannam@502:     m_srcIndex = 0;
c@225: }
c@225: 
cannam@502: int Framer::getMaxNoFrames()
c@225: {
c@225:     return m_maxFrames;
c@225: }
c@225: 
cannam@502: void Framer::setSource(double *src, int64_t length)
c@225: {
c@225:     m_srcBuffer = src;
cannam@502:     m_sampleLen = length;
c@225: 
cannam@502:     int64_t maxFrames = length / int64_t(m_stepSize);
cannam@502:     if (maxFrames * int64_t(m_stepSize) < length) {
cannam@502:         ++maxFrames;
cannam@502:     }
cannam@502:     if (maxFrames > INT_MAX) maxFrames = INT_MAX;
cannam@502:     m_maxFrames = maxFrames;
c@225: }