source: src/effects/pitchshift_rubberband.c @ c41637b

feature/cnnfeature/crepefeature/timestretchfix/ffmpeg5
Last change on this file since c41637b was 34d4232, checked in by Paul Brossier <piem@piem.org>, 5 years ago

[pitchshift] make sure rb is set to null first

  • Property mode set to 100644
File size: 4.3 KB
Line 
1/*
2  Copyright (C) 2016 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
23#ifdef HAVE_RUBBERBAND
24
25#include "fvec.h"
26#include "effects/pitchshift.h"
27
28#include "rubberband/rubberband-c.h"
29
30/** generic pitch shifting structure */
31struct _aubio_pitchshift_t
32{
33  uint_t samplerate;              /**< samplerate */
34  uint_t hopsize;                 /**< hop size */
35  smpl_t pitchscale;              /**< pitch scale */
36
37  RubberBandState rb;
38  RubberBandOptions rboptions;
39};
40
41extern RubberBandOptions aubio_get_rubberband_opts(const char_t *mode);
42
43aubio_pitchshift_t *
44new_aubio_pitchshift (const char_t * mode,
45    smpl_t transpose, uint_t hopsize, uint_t samplerate)
46{
47  aubio_pitchshift_t *p = AUBIO_NEW (aubio_pitchshift_t);
48  p->samplerate = samplerate;
49  p->hopsize = hopsize;
50  p->pitchscale = 1.;
51  p->rb = NULL;
52  if ((sint_t)hopsize <= 0) {
53    AUBIO_ERR("pitchshift: hop_size should be >= 0, got %d\n", hopsize);
54    goto beach;
55  }
56  if ((sint_t)samplerate <= 0) {
57    AUBIO_ERR("pitchshift: samplerate should be >= 0, got %d\n", samplerate);
58    goto beach;
59  }
60
61  p->rboptions = aubio_get_rubberband_opts(mode);
62  if (p->rboptions < 0) {
63    AUBIO_ERR("pitchshift: unknown pitch shifting method %s\n", mode);
64    goto beach;
65  }
66
67  //AUBIO_MSG("pitchshift: using pitch shifting method %s\n", mode);
68
69  p->rb = rubberband_new(samplerate, 1, p->rboptions, 1., p->pitchscale);
70  rubberband_set_max_process_size(p->rb, p->hopsize);
71  //rubberband_set_debug_level(p->rb, 10);
72
73  if (aubio_pitchshift_set_transpose(p, transpose)) goto beach;
74
75#if 1
76  // warm up rubber band
77  unsigned int latency = MAX(p->hopsize, rubberband_get_latency(p->rb));
78  int available = rubberband_available(p->rb);
79  fvec_t *zeros = new_fvec(p->hopsize);
80  while (available <= (int)latency) {
81    rubberband_process(p->rb, (const float* const*)&(zeros->data), p->hopsize, 0);
82    available = rubberband_available(p->rb);
83  }
84  del_fvec(zeros);
85#endif
86
87  return p;
88
89beach:
90  del_aubio_pitchshift(p);
91  return NULL;
92}
93
94void
95del_aubio_pitchshift (aubio_pitchshift_t * p)
96{
97  if (p->rb) {
98    rubberband_delete(p->rb);
99  }
100  AUBIO_FREE (p);
101}
102
103uint_t aubio_pitchshift_get_latency (aubio_pitchshift_t * p) {
104  return rubberband_get_latency(p->rb);
105}
106
107uint_t
108aubio_pitchshift_set_pitchscale (aubio_pitchshift_t * p, smpl_t pitchscale)
109{
110  if (pitchscale >= 0.25  && pitchscale <= 4.) {
111    p->pitchscale = pitchscale;
112    rubberband_set_pitch_scale(p->rb, p->pitchscale);
113    return AUBIO_OK;
114  } else {
115    AUBIO_ERR("pitchshift: could not set pitchscale to %.2f\n", pitchscale);
116    return AUBIO_FAIL;
117  }
118}
119
120smpl_t
121aubio_pitchshift_get_pitchscale (aubio_pitchshift_t * p)
122{
123  return p->pitchscale;
124}
125
126uint_t
127aubio_pitchshift_set_transpose(aubio_pitchshift_t * p, smpl_t transpose)
128{
129  if (transpose >= -24. && transpose <= 24.) {
130    smpl_t pitchscale = POW(2., transpose / 12.);
131    return aubio_pitchshift_set_pitchscale(p, pitchscale);
132  } else {
133    AUBIO_ERR("pitchshift: could not set transpose to %.2f\n", transpose);
134    return AUBIO_FAIL;
135  }
136}
137
138smpl_t
139aubio_pitchshift_get_transpose(aubio_pitchshift_t * p)
140{
141  return 12. * LOG(p->pitchscale) / LOG(2.0);
142}
143
144void
145aubio_pitchshift_do (aubio_pitchshift_t * p, const fvec_t * in, fvec_t * out)
146{
147  // third parameter is always 0 since we are never expecting a final frame
148  rubberband_process(p->rb, (const float* const*)&(in->data), p->hopsize, 0);
149  if (rubberband_available(p->rb) >= (int)p->hopsize) {
150    rubberband_retrieve(p->rb, (float* const*)&(out->data), p->hopsize);
151  } else {
152    AUBIO_WRN("pitchshift: catching up with zeros, only %d available, needed: %d, "
153        "current pitchscale: %f\n",
154        rubberband_available(p->rb), p->hopsize, p->pitchscale);
155    fvec_zeros(out);
156  }
157}
158
159#endif
Note: See TracBrowser for help on using the repository browser.