1 #ifndef STREAM_H_

2 #define STREAM_H_

3

4 #include <pthread.h>

5

6 #define AUDIO_FFT_BINS 16

7 #define AUDIO_FFT_DOWNSAMPLE 1

8

9 #define AUDIO_NUM_BUFFERS 8

10 #define AUDIO_BUFFER_MSEC 32

11 // TODO should the sampling rate be hardcoded?

12 #define AUDIO_BUFFER_SAMPLES (AUDIO_BUFFER_MSEC * 44100 / 1000)

13 // TODO unhardcode the channels number

14 #define AUDIO_BUFFER_BYTES (AUDIO_BUFFER_SAMPLES * 2 * 2)

15

16 #define AUDIO_FFT_SAMPLES (AUDIO_BUFFER_SAMPLES / AUDIO_FFT_DOWNSAMPLE)

17

18 enum AUDIO_PLAYMODE

19 {

20 AUDIO_PLAYMODE_ONCE,

21 AUDIO_PLAYMODE_LOOP

22 };

23

24 struct AudioStreamBuffer {

25 char samples[AUDIO_BUFFER_BYTES];

26

27 int num_samples;

28 int channels;

29 int sample_rate;

30 };

31

32 struct FFTState;

33

34 class AudioStream {

35 private:

36 pthread_t play_thread;

37 pthread_mutex_t mutex;

38

39 float volume;

40 bool done, loop;

41 unsigned int poll_interval;

42 unsigned int alsrc;

43

44 int *freqhist; // this will always point to the fft of the currently playing buffer

45

46 bool use_fft;

47

48 virtual bool more_samples(AudioStreamBuffer *buf) = 0;

49 virtual void calc_freq(AudioStreamBuffer *buf, int *bins, FFTState *fft);

50

51 public:

52 void poll_loop();

53

54 AudioStream();

55 virtual ~AudioStream();

56

57 virtual void enable_fft();

58 virtual void disable_fft();

59 virtual bool is_fft_enabled() const;

60

61 virtual bool open(const char *fname);

62 virtual void close();

63

64 virtual void set_volume(float vol);

65 virtual float get_volume() const;

66

67 virtual void play(AUDIO_PLAYMODE mode);

68 virtual void stop();

69

70 virtual void rewind() = 0;

71

72 virtual int freq_count(int bin) const;

73 virtual float freq_normalized(int bin) const;

74

75 // frequency range in hertz

76 virtual int freq_count(int range_start, int range_end) const;

77 };

78

79 #endif // AUDIO_STREAM_H_