source: src/effects/pitchshift_rubberband.c @ 24dc867

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

[pitchscale] wrap long lines

  • 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,
82        (const float* const*)&(zeros->data), p->hopsize, 0);
83    available = rubberband_available(p->rb);
84  }
85  del_fvec(zeros);
86#endif
87
88  return p;
89
90beach:
91  del_aubio_pitchshift(p);
92  return NULL;
93}
94
95void
96del_aubio_pitchshift (aubio_pitchshift_t * p)
97{
98  if (p->rb) {
99    rubberband_delete(p->rb);
100  }
101  AUBIO_FREE (p);
102}
103
104uint_t aubio_pitchshift_get_latency (aubio_pitchshift_t * p) {
105  return rubberband_get_latency(p->rb);
106}
107
108uint_t
109aubio_pitchshift_set_pitchscale (aubio_pitchshift_t * p, smpl_t pitchscale)
110{
111  if (pitchscale >= 0.25  && pitchscale <= 4.) {
112    p->pitchscale = pitchscale;
113    rubberband_set_pitch_scale(p->rb, p->pitchscale);
114    return AUBIO_OK;
115  } else {
116    AUBIO_ERR("pitchshift: could not set pitchscale to %.2f\n", pitchscale);
117    return AUBIO_FAIL;
118  }
119}
120
121smpl_t
122aubio_pitchshift_get_pitchscale (aubio_pitchshift_t * p)
123{
124  return p->pitchscale;
125}
126
127uint_t
128aubio_pitchshift_set_transpose(aubio_pitchshift_t * p, smpl_t transpose)
129{
130  if (transpose >= -24. && transpose <= 24.) {
131    smpl_t pitchscale = POW(2., transpose / 12.);
132    return aubio_pitchshift_set_pitchscale(p, pitchscale);
133  } else {
134    AUBIO_ERR("pitchshift: could not set transpose to %.2f\n", transpose);
135    return AUBIO_FAIL;
136  }
137}
138
139smpl_t
140aubio_pitchshift_get_transpose(aubio_pitchshift_t * p)
141{
142  return 12. * LOG(p->pitchscale) / LOG(2.0);
143}
144
145void
146aubio_pitchshift_do (aubio_pitchshift_t * p, const fvec_t * in, fvec_t * out)
147{
148  // third parameter is always 0 since we are never expecting a final frame
149  rubberband_process(p->rb, (const float* const*)&(in->data), p->hopsize, 0);
150  if (rubberband_available(p->rb) >= (int)p->hopsize) {
151    rubberband_retrieve(p->rb, (float* const*)&(out->data), p->hopsize);
152  } else {
153    AUBIO_WRN("pitchshift: catching up with zeros"
154        ", only %d available, needed: %d, current pitchscale: %f\n",
155        rubberband_available(p->rb), p->hopsize, p->pitchscale);
156    fvec_zeros(out);
157  }
158}
159
160#endif
Note: See TracBrowser for help on using the repository browser.