source: src/phasevoc.c @ c721874

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5pitchshiftsamplertimestretchyinfft+
Last change on this file since c721874 was c721874, checked in by Paul Brossier <piem@piem.org>, 16 years ago

merge changes from banane, more fixes

  • Property mode set to 100644
File size: 4.5 KB
Line 
1/*
2   Copyright (C) 2003 Paul Brossier
3
4   This program is free software; you can redistribute it and/or modify
5   it under the terms of the GNU General Public License as published by
6   the Free Software Foundation; either version 2 of the License, or
7   (at your option) any later version.
8
9   This program is distributed in the hope that it will be useful,
10   but WITHOUT ANY WARRANTY; without even the implied warranty of
11   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12   GNU General Public License for more details.
13
14   You should have received a copy of the GNU General Public License
15   along with this program; if not, write to the Free Software
16   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18*/
19
20#include "aubio_priv.h"
21#include "sample.h"
22#include "fft.h"
23#include "mathutils.h"
24#include "phasevoc.h"
25
26/** phasevocoder internal object */
27struct _aubio_pvoc_t {
28  uint_t win_s;       /** grain length */
29  uint_t hop_s;       /** overlap step */
30  uint_t channels;    /** number of channels */
31  aubio_mfft_t * fft; /** spectral data */
32  fvec_t * synth;     /**cur output grain [win_s] */
33  fvec_t * synthold;  /**last input frame [win_s-hop_s] */
34  fvec_t * data;      /**current input grain [win_s] */
35  fvec_t * dataold;   /**last input frame [win_s-hop_s] */
36  smpl_t * w;          /** grain window [win_s] */
37};
38
39
40/** returns data and dataold slided by hop_s */
41static void aubio_pvoc_swapbuffers(smpl_t * data, smpl_t * dataold, const
42    smpl_t * datanew, uint_t win_s, uint_t hop_s);
43
44/** do additive synthesis from 'old' and 'cur' */
45static void aubio_pvoc_addsynth(const smpl_t * synth, smpl_t * synthold,
46    smpl_t * synthnew, uint_t win_s, uint_t hop_s);
47
48void aubio_pvoc_do(aubio_pvoc_t *pv, fvec_t * datanew, cvec_t *fftgrain) {
49  uint_t i,j;
50  for (i=0; i<pv->channels; i++) {
51    /* slide  */
52    aubio_pvoc_swapbuffers(pv->data->data[i],pv->dataold->data[i],
53        datanew->data[i],pv->win_s,pv->hop_s);
54    /* windowing */
55    for (j=0; j<pv->win_s; j++) pv->data->data[i][j] *= pv->w[j];
56  }
57  /* shift */
58  vec_shift(pv->data);
59  /* calculate fft */
60  aubio_mfft_do (pv->fft,pv->data,fftgrain);
61}
62
63void aubio_pvoc_rdo(aubio_pvoc_t *pv,cvec_t * fftgrain, fvec_t * synthnew) {
64  uint_t i;
65  /* calculate rfft */
66  aubio_mfft_rdo(pv->fft,fftgrain,pv->synth);
67  /* unshift */
68  vec_shift(pv->synth);
69  for (i=0; i<pv->channels; i++) {
70    aubio_pvoc_addsynth(pv->synth->data[i],pv->synthold->data[i],
71        synthnew->data[i],pv->win_s,pv->hop_s);
72  }
73}
74
75aubio_pvoc_t * new_aubio_pvoc (uint_t win_s, uint_t hop_s, uint_t channels) {
76  aubio_pvoc_t * pv = AUBIO_NEW(aubio_pvoc_t);
77
78  if (win_s < 2*hop_s) {
79    AUBIO_ERR("Hop size bigger than half the window size!\n");
80    AUBIO_ERR("Resetting hop size to half the window size.\n");
81    hop_s = win_s / 2;
82  }
83
84  if (hop_s < 1) {
85    AUBIO_ERR("Hop size is smaller than 1!\n");
86    AUBIO_ERR("Resetting hop size to half the window size.\n");
87    hop_s = win_s / 2;
88  }
89
90  pv->fft      = new_aubio_mfft(win_s,channels);
91
92  /* remember old */
93  pv->data     = new_fvec (win_s, channels);
94  pv->synth    = new_fvec (win_s, channels);
95
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        = AUBIO_ARRAY(smpl_t,win_s);
100  aubio_window(pv->w,win_s,aubio_win_hanningz);
101
102  pv->channels = channels;
103  pv->hop_s    = hop_s;
104  pv->win_s    = win_s;
105
106  return pv;
107}
108
109void del_aubio_pvoc(aubio_pvoc_t *pv) {
110  del_fvec(pv->data);
111  del_fvec(pv->synth);
112  del_fvec(pv->dataold);
113  del_fvec(pv->synthold);
114  del_aubio_mfft(pv->fft);
115  AUBIO_FREE(pv->w);
116  AUBIO_FREE(pv);
117}
118
119static void aubio_pvoc_swapbuffers(smpl_t * data, smpl_t * dataold, 
120    const smpl_t * datanew, uint_t win_s, uint_t hop_s)
121{
122  uint_t i;
123  for (i=0;i<win_s-hop_s;i++)
124    data[i] = dataold[i];
125  for (i=0;i<hop_s;i++)
126    data[win_s-hop_s+i] = datanew[i];
127  for (i=0;i<win_s-hop_s;i++)
128    dataold[i] = data[i+hop_s];
129}
130
131static void aubio_pvoc_addsynth(const smpl_t * synth, smpl_t * synthold, 
132                smpl_t * synthnew, uint_t win_s, uint_t hop_s)
133{
134  uint_t i;
135  smpl_t scale = 2*hop_s/(win_s+.0);
136  /* add new synth to old one and put result in synthnew */
137  for (i=0;i<hop_s;i++)
138    synthnew[i] = synthold[i]+synth[i]*scale;
139  /* shift synthold */
140  for (i=0;i<win_s-2*hop_s;i++)
141    synthold[i] = synthold[i+hop_s];
142  /* erase last frame in synthold */
143  for (i=win_s-hop_s;i<win_s;i++)
144    synthold[i-hop_s]=0.;
145  /* additive synth */
146  for (i=0;i<win_s-hop_s;i++)
147    synthold[i] += synth[i+hop_s]*scale;
148}
149
Note: See TracBrowser for help on using the repository browser.