source: src/spectral/phasevoc.c @ 7585822

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

src/, examples/: #ifdef HAVE_, not #if HAVE_

  • Property mode set to 100644
File size: 5.5 KB
Line 
1/*
2  Copyright (C) 2003-2014 Paul Brossier <piem@aubio.org>
3
4  This file is part of aubio.
5
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.
10
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.
15
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/>.
18
19*/
20
21#include "aubio_priv.h"
22#include "fvec.h"
23#include "cvec.h"
24#include "mathutils.h"
25#include "spectral/fft.h"
26#include "spectral/phasevoc.h"
27
28/** phasevocoder internal object */
29struct _aubio_pvoc_t {
30  uint_t win_s;       /** grain length */
31  uint_t hop_s;       /** overlap step */
32  aubio_fft_t * fft;  /** fft object */
33  fvec_t * data;      /** current input grain, [win_s] frames */
34  fvec_t * dataold;   /** memory of past grain, [win_s-hop_s] frames */
35  fvec_t * synth;     /** current output grain, [win_s] frames */
36  fvec_t * synthold;  /** memory of past grain, [win_s-hop_s] frames */
37  fvec_t * w;         /** grain window [win_s] */
38  uint_t start;       /** where to start additive synthesis */
39  uint_t end;         /** where to end it */
40  smpl_t scale;       /** scaling factor for synthesis */
41  uint_t end_datasize;  /** size of memory to end */
42  uint_t hop_datasize;  /** size of memory to hop_s */
43};
44
45
46/** returns data and dataold slided by hop_s */
47static void aubio_pvoc_swapbuffers(aubio_pvoc_t *pv, const fvec_t *new);
48
49/** do additive synthesis from 'old' and 'cur' */
50static void aubio_pvoc_addsynth(aubio_pvoc_t *pv, fvec_t * synthnew);
51
52void aubio_pvoc_do(aubio_pvoc_t *pv, const fvec_t * datanew, cvec_t *fftgrain) {
53  /* slide  */
54  aubio_pvoc_swapbuffers(pv, datanew);
55  /* windowing */
56  fvec_weight(pv->data, pv->w);
57  /* shift */
58  fvec_shift(pv->data);
59  /* calculate fft */
60  aubio_fft_do (pv->fft,pv->data,fftgrain);
61}
62
63void aubio_pvoc_rdo(aubio_pvoc_t *pv,cvec_t * fftgrain, fvec_t * synthnew) {
64  /* calculate rfft */
65  aubio_fft_rdo(pv->fft,fftgrain,pv->synth);
66  /* unshift */
67  fvec_ishift(pv->synth);
68  /* additive synthesis */
69  aubio_pvoc_addsynth(pv, synthnew);
70}
71
72aubio_pvoc_t * new_aubio_pvoc (uint_t win_s, uint_t hop_s) {
73  aubio_pvoc_t * pv = AUBIO_NEW(aubio_pvoc_t);
74
75  /* if (win_s < 2*hop_s) {
76    AUBIO_WRN("Hop size bigger than half the window size!\n");
77  } */
78
79  if ((sint_t)hop_s < 1) {
80    AUBIO_ERR("pvoc: got hop_size %d, but can not be < 1\n", hop_s);
81    goto beach;
82  } else if ((sint_t)win_s < 2) {
83    AUBIO_ERR("pvoc: got buffer_size %d, but can not be < 2\n", win_s);
84    goto beach;
85  } else if (win_s < hop_s) {
86    AUBIO_ERR("pvoc: hop size (%d) is larger than win size (%d)\n", win_s, hop_s);
87    goto beach;
88  }
89
90  pv->fft      = new_aubio_fft (win_s);
91
92  /* remember old */
93  pv->data     = new_fvec (win_s);
94  pv->synth    = new_fvec (win_s);
95
96  /* new input output */
97  if (win_s > hop_s) {
98    pv->dataold  = new_fvec  (win_s-hop_s);
99    pv->synthold = new_fvec (win_s-hop_s);
100  } else {
101    pv->dataold  = new_fvec  (1);
102    pv->synthold = new_fvec (1);
103  }
104  pv->w        = new_aubio_window ("hanningz", win_s);
105
106  pv->hop_s    = hop_s;
107  pv->win_s    = win_s;
108
109  /* more than 50% overlap, overlap anyway */
110  if (win_s < 2 * hop_s) pv->start = 0;
111  /* less than 50% overlap, reset latest grain trail */
112  else pv->start = win_s - hop_s - hop_s;
113
114  if (win_s > hop_s) pv->end = win_s - hop_s;
115  else pv->end = 0;
116
117  pv->end_datasize = pv->end * sizeof(smpl_t);
118  pv->hop_datasize = pv->hop_s * sizeof(smpl_t);
119
120  pv->scale = pv->hop_s * 2. / pv->win_s;
121
122  return pv;
123
124beach:
125  AUBIO_FREE (pv);
126  return NULL;
127}
128
129void del_aubio_pvoc(aubio_pvoc_t *pv) {
130  del_fvec(pv->data);
131  del_fvec(pv->synth);
132  del_fvec(pv->dataold);
133  del_fvec(pv->synthold);
134  del_fvec(pv->w);
135  del_aubio_fft(pv->fft);
136  AUBIO_FREE(pv);
137}
138
139static void aubio_pvoc_swapbuffers(aubio_pvoc_t *pv, const fvec_t *new)
140{
141  /* some convenience pointers */
142  smpl_t * data = pv->data->data;
143  smpl_t * dataold = pv->dataold->data;
144  smpl_t * datanew = new->data;
145#ifndef HAVE_MEMCPY_HACKS
146  uint_t i;
147  for (i = 0; i < pv->end; i++)
148    data[i] = dataold[i];
149  for (i = 0; i < pv->hop_s; i++)
150    data[pv->end + i] = datanew[i];
151  for (i = 0; i < pv->end; i++)
152    dataold[i] = data[i + pv->hop_s];
153#else
154  memcpy(data, dataold, pv->end_datasize);
155  data += pv->end;
156  memcpy(data, datanew, pv->hop_datasize);
157  data -= pv->end;
158  data += pv->hop_s;
159  memcpy(dataold, data, pv->end_datasize);
160#endif
161}
162
163static void aubio_pvoc_addsynth(aubio_pvoc_t *pv, fvec_t *synth_new)
164{
165  uint_t i;
166  /* some convenience pointers */
167  smpl_t * synth    = pv->synth->data;
168  smpl_t * synthold = pv->synthold->data;
169  smpl_t * synthnew = synth_new->data;
170
171  /* put new result in synthnew */
172  for (i = 0; i < pv->hop_s; i++)
173    synthnew[i] = synth[i] * pv->scale;
174
175  /* no overlap, nothing else to do */
176  if (pv->end == 0) return;
177
178  /* add new synth to old one */
179  for (i = 0; i < pv->hop_s; i++)
180    synthnew[i] += synthold[i];
181
182  /* shift synthold */
183  for (i = 0; i < pv->start; i++)
184    synthold[i] = synthold[i + pv->hop_s];
185
186  /* erase last frame in synthold */
187  for (i = pv->start; i < pv->end; i++)
188    synthold[i] = 0.;
189
190  /* additive synth */
191  for (i = 0; i < pv->end; i++)
192    synthold[i] += synth[i + pv->hop_s] * pv->scale;
193}
Note: See TracBrowser for help on using the repository browser.