aubiotempo~.c: update to 0.4.0
[pd-aubio.git] / 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 0.1";
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 threshold;      
24         t_float threshold2;     
25         t_int pos; /*frames%dspblocksize*/
26         t_int bufsize;
27         t_int hopsize;
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_debug(t_aubiopitch_tilde *x)
62 {
63         post("aubiopitch~ bufsize:\t%d", x->bufsize);
64         post("aubiopitch~ hopsize:\t%d", x->hopsize);
65         post("aubiopitch~ threshold:\t%f", x->threshold);
66         post("aubiopitch~ audio in:\t%f", x->vec->data[0]);
67 }
68
69 //static void *aubiopitch_tilde_new (t_floatarg f)
70 static void *aubiopitch_tilde_new (t_symbol * s)
71 {
72         t_aubiopitch_tilde *x = 
73                 (t_aubiopitch_tilde *)pd_new(aubiopitch_tilde_class);
74
75         x->bufsize   = 2048;
76         x->hopsize   = x->bufsize / 2;
77
78         //FIXME: get the real samplerate
79   x->o = new_aubio_pitch(s->s_name, x->bufsize,
80       x->hopsize, 44100. );
81         aubio_pitch_set_tolerance (x->o, 0.7);
82         x->vec = (fvec_t *)new_fvec(x->hopsize);
83         x->pitchvec = (fvec_t *)new_fvec(1);
84
85         //floatinlet_new (&x->x_obj, &x->threshold);
86         x->pitch = outlet_new (&x->x_obj, &s_float);
87
88         post(aubiopitch_version);
89         return (void *)x;
90 }
91
92 static void *aubiopitch_tilde_del(t_aubiopitch_tilde *x)
93 {
94   del_aubio_pitch(x->o);
95         del_fvec(x->vec);
96         del_fvec(x->pitchvec);
97         return 0;
98 }
99
100 void aubiopitch_tilde_setup (void)
101 {
102         aubiopitch_tilde_class = class_new (gensym ("aubiopitch~"),
103                         (t_newmethod)aubiopitch_tilde_new,
104                         (t_method)aubiopitch_tilde_del,
105                         sizeof (t_aubiopitch_tilde),
106                         CLASS_DEFAULT, A_DEFSYMBOL, 0);
107         class_addmethod(aubiopitch_tilde_class, 
108                         (t_method)aubiopitch_tilde_dsp, 
109                         gensym("dsp"), 0);
110         class_addmethod(aubiopitch_tilde_class, 
111                         (t_method)aubiopitch_tilde_debug,
112                         gensym("debug"), 0);
113         CLASS_MAINSIGNALIN(aubiopitch_tilde_class, 
114                         t_aubiopitch_tilde, threshold);
115 }
116