#ifndef AUDIO_H
#define AUDIO_H

#include <queue>
#include <stdint.h>
#include <vector>
#include <AudioFile.h>
using namespace std;

class AudioFrame {
  private:
    int start;
    int end;
    int len;

  public:
    AudioFrame();
    AudioFrame(int len);

    ~AudioFrame();
    int set_start(int val);
    int set_end(int val, int max_len);
    int get_start();
    int get_len();
    int disp();
};

class Audio {
    private:
        vector<float> speech_data;
        int16_t *speech_buff;
        int speech_len;
        int speech_align_len;
        int16_t sample_rate;
        int offset;
        float align_size;
        int data_type;
        queue<AudioFrame *> frame_queue;

    public:
        vector<float> speech_vec;
        Audio(int data_type);
        Audio(int data_type, int size);
        ~Audio();
        void disp();
        void loadwav(const char *filename);
        void loadwavfrommem(AudioFile<float>audio);
        int fetch_chunck(float *&dout, int len);
        int fetch(vector<float> &dout, int &len, int &flag);
        void padding();
        void split();
};

#endif