src/aubiopitch~.c: refix aubiopitch_tilde_del
[pd-aubio.git] / src / aubiopitch~.c
1 /**
2  *
3  * a puredata wrapper for aubio pitch detection functions
4  *
5  * Thanks to Johannes M Zmolnig for writing the excellent HOWTO:
6  *       http://iem.kug.ac.at/pd/externals-HOWTO/
7  *
8  * */
9
10 #include <m_pd.h>
11 #include <aubio/aubio.h>
12 #include <string.h>
13
14 char aubiopitch_version[] = "aubiopitch~ version " PACKAGE_VERSION;
15
16 static t_class *aubiopitch_tilde_class;
17
18 void aubiopitch_tilde_setup (void);
19
20 typedef struct _aubiopitch_tilde
21 {
22   t_object x_obj;
23   t_float tolerance;
24   t_int pos; /*frames%dspblocksize*/
25   t_int bufsize;
26   t_int hopsize;
27   char_t *method;
28   aubio_pitch_t *o;
29   fvec_t *vec;
30   fvec_t *pitchvec;
31   t_outlet *pitch;
32 } t_aubiopitch_tilde;
33
34 static t_int *aubiopitch_tilde_perform(t_int *w)
35 {
36   t_aubiopitch_tilde *x = (t_aubiopitch_tilde *)(w[1]);
37   t_sample *in          = (t_sample *)(w[2]);
38   int n                 = (int)(w[3]);
39   int j;
40   for (j=0;j<n;j++) {
41     /* write input to datanew */
42     fvec_set_sample(x->vec, in[j], x->pos);
43     /*time for fft*/
44     if (x->pos == x->hopsize-1) {
45       /* block loop */
46       aubio_pitch_do(x->o, x->vec, x->pitchvec);
47       outlet_float(x->pitch, x->pitchvec->data[0]);
48       /* end of block loop */
49       x->pos = -1; /* so it will be zero next j loop */
50     }
51     x->pos++;
52   }
53   return (w+4);
54 }
55
56 static void aubiopitch_tilde_dsp(t_aubiopitch_tilde *x, t_signal **sp)
57 {
58   dsp_add(aubiopitch_tilde_perform, 3, x, sp[0]->s_vec, sp[0]->s_n);
59 }
60
61 static void aubiopitch_tilde_tolerance(t_aubiopitch_tilde *x, t_floatarg a)
62 {
63   if (a > 0) {
64     x->tolerance = a;
65     aubio_pitch_set_tolerance(x->o, x->tolerance);
66   } else {
67     post("aubiopitch~ tolerance set to %.2f",
68         aubio_pitch_get_tolerance(x->o));
69   }
70 }
71
72 static void aubiopitch_tilde_debug(t_aubiopitch_tilde *x)
73 {
74   post(aubiopitch_version);
75   post("aubiopitch~ method:\t%s", x->method);
76   post("aubiopitch~ bufsize:\t%d", x->bufsize);
77   post("aubiopitch~ hopsize:\t%d", x->hopsize);
78   post("aubiopitch~ tolerance:\t%f", aubio_pitch_get_tolerance(x->o));
79 }
80
81 static void *aubiopitch_tilde_new (t_symbol * s, int argc, t_atom *argv)
82 {
83
84   t_aubiopitch_tilde *x = (t_aubiopitch_tilde *)pd_new(aubiopitch_tilde_class);
85
86   x->method = "default";
87   x->bufsize = 2048;
88   x->hopsize = x->bufsize / 2;
89
90   if (argc >= 2) {
91     if (argv[1].a_type == A_FLOAT) {
92       x->bufsize = (uint_t)(argv[1].a_w.w_float);
93     }
94     argc--;
95   }
96
97   x->hopsize = x->bufsize / 2;
98
99   if (argc == 2) {
100     if (argv[2].a_type == A_FLOAT) {
101       x->hopsize = (uint_t)(argv[2].a_w.w_float);
102     }
103     argc--;
104   }
105
106   if (argc == 1) {
107     x->method = argv[0].a_w.w_symbol->s_name;
108   }
109
110   x->o = new_aubio_pitch(x->method, x->bufsize, x->hopsize,
111       (uint_t)sys_getsr());
112
113   if (x->o == NULL) return NULL;
114
115   x->vec = (fvec_t *)new_fvec(x->hopsize);
116   x->pitchvec = (fvec_t *)new_fvec(1);
117
118   x->pitch = outlet_new (&x->x_obj, &s_float);
119
120   return (void *)x;
121 }
122
123 void aubiopitch_tilde_del(t_aubiopitch_tilde *x)
124 {
125   outlet_free(x->pitch);
126   del_aubio_pitch(x->o);
127   del_fvec(x->vec);
128   del_fvec(x->pitchvec);
129 }
130
131 void aubiopitch_tilde_setup (void)
132 {
133   aubiopitch_tilde_class = class_new (gensym ("aubiopitch~"),
134       (t_newmethod)aubiopitch_tilde_new,
135       (t_method)aubiopitch_tilde_del,
136       sizeof (t_aubiopitch_tilde),
137       CLASS_DEFAULT, A_GIMME, 0);
138   class_addmethod(aubiopitch_tilde_class,
139       (t_method)aubiopitch_tilde_dsp,
140       gensym("dsp"), 0);
141   class_addmethod(aubiopitch_tilde_class,
142       (t_method)aubiopitch_tilde_tolerance,
143       gensym("tolerance"), A_DEFFLOAT, 0);
144   class_addmethod(aubiopitch_tilde_class,
145       (t_method)aubiopitch_tilde_tolerance,
146       gensym("tol"), A_DEFFLOAT, 0);
147   class_addmethod(aubiopitch_tilde_class,
148       (t_method)aubiopitch_tilde_debug,
149       gensym("debug"), 0);
150   CLASS_MAINSIGNALIN(aubiopitch_tilde_class,
151       t_aubiopitch_tilde, tolerance);
152 }