2 Copyright (C) 2003-2009 Paul Brossier <piem@aubio.org>
4 This file is part of aubio.
6 aubio is free software: you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation, either version 3 of the License, or
9 (at your option) any later version.
11 aubio is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with aubio. If not, see <http://www.gnu.org/licenses/>.
21 #include "aubio_priv.h"
24 #include "mathutils.h"
25 #include "spectral/fft.h"
26 #include "spectral/phasevoc.h"
28 /** phasevocoder internal object */
29 struct _aubio_pvoc_t {
30 uint_t win_s; /** grain length */
31 uint_t hop_s; /** overlap step */
32 uint_t channels; /** number of channels */
33 aubio_fft_t * fft; /** fft object */
34 fvec_t * synth; /** cur output grain [win_s] */
35 fvec_t * synthold; /** last input frame [win_s-hop_s] */
36 fvec_t * data; /** current input grain [win_s] */
37 fvec_t * dataold; /** last input frame [win_s-hop_s] */
38 fvec_t * w; /** grain window [win_s] */
42 /** returns data and dataold slided by hop_s */
43 static void aubio_pvoc_swapbuffers(smpl_t * data, smpl_t * dataold, const
44 smpl_t * datanew, uint_t win_s, uint_t hop_s);
46 /** do additive synthesis from 'old' and 'cur' */
47 static void aubio_pvoc_addsynth(const smpl_t * synth, smpl_t * synthold,
48 smpl_t * synthnew, uint_t win_s, uint_t hop_s);
50 void aubio_pvoc_do(aubio_pvoc_t *pv, fvec_t * datanew, cvec_t *fftgrain) {
52 for (i=0; i<pv->channels; i++) {
54 aubio_pvoc_swapbuffers(pv->data->data[i],pv->dataold->data[i],
55 datanew->data[i],pv->win_s,pv->hop_s);
58 fvec_weight(pv->data, pv->w);
62 aubio_fft_do (pv->fft,pv->data,fftgrain);
65 void aubio_pvoc_rdo(aubio_pvoc_t *pv,cvec_t * fftgrain, fvec_t * synthnew) {
68 aubio_fft_rdo(pv->fft,fftgrain,pv->synth);
70 fvec_shift(pv->synth);
71 for (i=0; i<pv->channels; i++) {
72 aubio_pvoc_addsynth(pv->synth->data[i],pv->synthold->data[i],
73 synthnew->data[i],pv->win_s,pv->hop_s);
77 aubio_pvoc_t * new_aubio_pvoc (uint_t win_s, uint_t hop_s, uint_t channels) {
78 aubio_pvoc_t * pv = AUBIO_NEW(aubio_pvoc_t);
80 /* if (win_s < 2*hop_s) {
81 AUBIO_WRN("Hop size bigger than half the window size!\n");
85 AUBIO_ERR("Hop size is smaller than 1!\n");
86 AUBIO_ERR("Resetting hop size to half the window size.\n");
90 pv->fft = new_aubio_fft(win_s,channels);
93 pv->data = new_fvec (win_s, channels);
94 pv->synth = new_fvec (win_s, channels);
96 /* new input output */
97 pv->dataold = new_fvec (win_s-hop_s, channels);
98 pv->synthold = new_fvec (win_s-hop_s, channels);
99 pv->w = new_aubio_window ("hanningz", win_s);
101 pv->channels = channels;
108 void del_aubio_pvoc(aubio_pvoc_t *pv) {
111 del_fvec(pv->dataold);
112 del_fvec(pv->synthold);
114 del_aubio_fft(pv->fft);
118 static void aubio_pvoc_swapbuffers(smpl_t * data, smpl_t * dataold,
119 const smpl_t * datanew, uint_t win_s, uint_t hop_s)
122 for (i=0;i<win_s-hop_s;i++)
123 data[i] = dataold[i];
124 for (i=0;i<hop_s;i++)
125 data[win_s-hop_s+i] = datanew[i];
126 for (i=0;i<win_s-hop_s;i++)
127 dataold[i] = data[i+hop_s];
130 static void aubio_pvoc_addsynth(const smpl_t * synth, smpl_t * synthold,
131 smpl_t * synthnew, uint_t win_s, uint_t hop_s)
134 smpl_t scale = 2*hop_s/(win_s+.0);
135 /* add new synth to old one and put result in synthnew */
136 for (i=0;i<hop_s;i++)
137 synthnew[i] = synthold[i]+synth[i]*scale;
139 for (i=0;i<win_s-2*hop_s;i++)
140 synthold[i] = synthold[i+hop_s];
141 /* erase last frame in synthold */
142 for (i=win_s-hop_s;i<win_s;i++)
143 synthold[i-hop_s]=0.;
145 for (i=0;i<win_s-hop_s;i++)
146 synthold[i] += synth[i+hop_s]*scale;