source: src/effects/rubberband_utils.c @ b2f41be

feature/timestretch
Last change on this file since b2f41be was b2f41be, checked in by Paul Brossier <piem@piem.org>, 12 months ago

[effects] clean up aubio_split_str, fix issues in aubio_get_rubberband_opts

  • Property mode set to 100644
File size: 6.8 KB
Line 
1
2
3#include "aubio_priv.h"
4
5#ifdef HAVE_RUBBERBAND
6
7#include <rubberband/rubberband-c.h>
8
9// check rubberband is 1.8.1, warn if 1.3
10#if !((RUBBERBAND_API_MAJOR_VERSION >= 2) && \
11    (RUBBERBAND_API_MINOR_VERSION >= 5))
12#warning RubberBandOptionDetectorSoft not available, \
13 please upgrade rubberband to version 1.8.1 or higher
14#define RubberBandOptionDetectorSoft 0x00000000
15#endif
16
17#include <stdio.h>
18#include <stdlib.h>
19#include <string.h>
20#include <assert.h>
21
22char_t** aubio_split_str(const char_t* str, const char_t sep) {
23  char_t** result = 0;
24  uint_t count = 0;
25  char_t input[PATH_MAX];
26  char_t* in_ptr = input;
27  char_t* last_sep = 0;
28  char_t delim[2]; delim[0] = sep; delim[1] = 0;
29
30  strncpy(input, str, PATH_MAX);
31  input[PATH_MAX - 1] = '\0';
32
33  // count number of elements
34  while (*in_ptr) {
35    if (sep == *in_ptr) {
36      count++;
37      last_sep = in_ptr;
38    }
39    in_ptr++;
40  }
41  // add space for trailing token.
42  count += last_sep < (input + strlen(input) - 1);
43  count++;
44
45  result = AUBIO_ARRAY(char_t*, count);
46  if (result) {
47    uint_t idx  = 0;
48    char_t* params = strtok(input, delim);
49    while (params) {
50      // make sure we don't got in the wild
51      assert(idx < count);
52      *(result + idx++) = strdup(params);
53      params = strtok(0, delim);
54    }
55    assert(idx == count - 1);
56    // add null string at the end
57    *(result + idx) = 0;
58  }
59  return result;
60}
61
62RubberBandOptions aubio_get_rubberband_opts(const char_t *mode)
63{
64  RubberBandOptions rboptions = RubberBandOptionProcessRealTime;
65
66  if ( strcmp(mode,"crispness:0") == 0 ) {
67    rboptions |= RubberBandOptionTransientsSmooth;
68    rboptions |= RubberBandOptionWindowLong;
69    rboptions |= RubberBandOptionPhaseIndependent;
70  } else if ( strcmp(mode, "crispness:1") == 0 ) {
71    rboptions |= RubberBandOptionDetectorSoft;
72    rboptions |= RubberBandOptionTransientsSmooth;
73    rboptions |= RubberBandOptionWindowLong;
74    rboptions |= RubberBandOptionPhaseIndependent;
75  } else if ( strcmp(mode, "crispness:2") == 0 ) {
76    rboptions |= RubberBandOptionTransientsSmooth;
77    rboptions |= RubberBandOptionPhaseIndependent;
78  } else if ( strcmp(mode, "crispness:3") == 0 ) {
79    rboptions |= RubberBandOptionTransientsSmooth;
80  } else if ( strcmp(mode, "crispness:4") == 0 ) {
81    // same as "default"
82  } else if ( strcmp(mode, "crispness:5") == 0 ) {
83    rboptions |= RubberBandOptionTransientsCrisp;
84  } else if ( strcmp(mode, "crispness:6") == 0 ) {
85    rboptions |= RubberBandOptionTransientsCrisp;
86    rboptions |= RubberBandOptionWindowShort;
87    rboptions |= RubberBandOptionPhaseIndependent;
88  } else if ( strcmp(mode, "default") == 0 ) {
89    // nothing to do
90  } else {
91    // attempt to parse a list of options, separated with ','
92    char_t **params = aubio_split_str(mode, ':');
93    uint_t i = 0;
94    if (!params || !params[0]) {
95      // memory failure occurred or empty string was passed
96      AUBIO_ERR("rubberband_utils: failed parsing options\n");
97      rboptions = -1;
98    }
99    while (*(params + i) != NULL) {
100      if ( strcmp(params[i], "ProcessOffline" ) == 0 )        {
101             rboptions = RubberBandOptionProcessOffline;
102        // TODO: add wrapper to rb study(smpl_t *input, uint_t write)
103        AUBIO_ERR("rubberband_utils: RubberBandOptionProcessOffline is not available\n");
104        rboptions = -1;
105      }
106      else if ( strcmp(params[i], "ProcessRealTime" ) == 0 )       rboptions |= RubberBandOptionProcessRealTime;
107      else if ( strcmp(params[i], "StretchElastic" ) == 0 )        rboptions |= RubberBandOptionStretchElastic;
108      else if ( strcmp(params[i], "StretchPrecise" ) == 0 )        rboptions |= RubberBandOptionStretchPrecise;
109      else if ( strcmp(params[i], "TransientsCrisp" ) == 0 )       rboptions |= RubberBandOptionTransientsCrisp;
110      else if ( strcmp(params[i], "TransientsMixed" ) == 0 )       rboptions |= RubberBandOptionTransientsMixed;
111      else if ( strcmp(params[i], "TransientsSmooth" ) == 0 )      rboptions |= RubberBandOptionTransientsSmooth;
112      else if ( strcmp(params[i], "DetectorCompound" ) == 0 )      rboptions |= RubberBandOptionDetectorCompound;
113      else if ( strcmp(params[i], "DetectorPercussive" ) == 0 )    rboptions |= RubberBandOptionDetectorPercussive;
114      else if ( strcmp(params[i], "DetectorSoft" ) == 0 )          rboptions |= RubberBandOptionDetectorSoft;
115      else if ( strcmp(params[i], "PhaseLaminar" ) == 0 )          rboptions |= RubberBandOptionPhaseLaminar;
116      else if ( strcmp(params[i], "PhaseIndependent" ) == 0 )      rboptions |= RubberBandOptionPhaseIndependent;
117      else if ( strcmp(params[i], "ThreadingAuto" ) == 0 )         rboptions |= RubberBandOptionThreadingAuto;
118      else if ( strcmp(params[i], "ThreadingNever" ) == 0 )        rboptions |= RubberBandOptionThreadingNever;
119      else if ( strcmp(params[i], "ThreadingAlways" ) == 0 )       rboptions |= RubberBandOptionThreadingAlways;
120      else if ( strcmp(params[i], "WindowStandard" ) == 0 )        rboptions |= RubberBandOptionWindowStandard;
121      else if ( strcmp(params[i], "WindowShort" ) == 0 )           rboptions |= RubberBandOptionWindowShort;
122      else if ( strcmp(params[i], "WindowLong" ) == 0 )            rboptions |= RubberBandOptionWindowLong;
123      else if ( strcmp(params[i], "SmoothingOff" ) == 0 )          rboptions |= RubberBandOptionSmoothingOff;
124      else if ( strcmp(params[i], "SmoothingOn" ) == 0 )           rboptions |= RubberBandOptionSmoothingOn;
125      else if ( strcmp(params[i], "FormantShifted" ) == 0 )        rboptions |= RubberBandOptionFormantShifted;
126      else if ( strcmp(params[i], "FormantPreserved" ) == 0 )      rboptions |= RubberBandOptionFormantPreserved;
127      else if ( strcmp(params[i], "PitchHighSpeed" ) == 0 )        rboptions |= RubberBandOptionPitchHighSpeed;
128      else if ( strcmp(params[i], "PitchHighQuality" ) == 0 )      rboptions |= RubberBandOptionPitchHighQuality;
129      else if ( strcmp(params[i], "PitchHighConsistency" ) == 0 )  rboptions |= RubberBandOptionPitchHighConsistency;
130      else if ( strcmp(params[i], "ChannelsApart" ) == 0 )         rboptions |= RubberBandOptionChannelsApart;
131      else if ( strcmp(params[i], "ChannelsTogether" ) == 0 )      rboptions |= RubberBandOptionChannelsTogether;
132      else {
133        AUBIO_ERR("rubberband_utils: did not understand option '%s', should be one of: "
134          "StretchElastic|StretchPrecise, TransientsCrisp|TransientsMixed|TransientsSmooth, "
135          "DetectorCompound|DetectorPercussive|DetectorSoft, PhaseLaminar|PhaseIndependent, "
136          "ThreadingAuto|ThreadingNever|ThreadingAlways, WindowStandard|WindowLong|WindowShort, "
137          "SmoothingOn|SmoothingOff, FormantShifted|FormantPreserved, "
138          "PitchHighSpeed|PitchHighQuality|PitchHighConsistency, ChannelsApart|ChannelsTogether\n"
139          , params[i]);
140        rboptions = -1;
141      }
142      AUBIO_FREE(params[i]);
143      i++;
144    }
145    AUBIO_FREE(params);
146  }
147  return rboptions;
148}
149
150#endif
Note: See TracBrowser for help on using the repository browser.