import sys
import os
import io
from pydub import AudioSegment
from pydub.playback import play

sys.path.append('../src/')

from ammosreader.AmmosAudioReader import AmmosAudioReader

if __name__ == '__main__':

    if len(sys.argv) != 2:
        sys.exit()

    file_name = sys.argv[1]

    print("File name:", file_name)

    dat_file = AmmosAudioReader(file_name)

    dat_file.read_all_frames_left()
    print("Sample rate:", dat_file.container.global_frames[0].global_frame_body.data_header.sample_rate)
    print("Container size:", dat_file.container.size())
    pcm_data = dat_file.pcm_for_channel(0)
    print("PCM data size total:", len(pcm_data))
    data = AudioSegment.from_raw(io.BytesIO(pcm_data),
                                 sample_width=2,
                                 frame_rate=22050,
                                 channels=1)
    play(data)