import numpy as np
import wave
import os

import struct

from pydub import AudioSegment

BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
file_DIR = BASE_DIR + "/sound/"

aiz = "00"
hl_file = file_DIR + "L0e0" + aiz + "a.wav"
hr_file = file_DIR + "R0e0" + aiz + "a.wav"
sound_file = "sound/Fe_sp_1.wav"
noise_file = "sound/W2.wav"


def stereo2mono(src: np.array):
    d = (src[:, 0] + src[:, 1]) / 2
    return d


def mono2stereo(dest: wave.Wave_write, src_l: str, src_r: str):
    duration = min(len(src_l), len(src_r))
    for i in range(duration):
        l = src_l[i]
        r = src_r[i]
        packed_value = struct.pack('<hh', np.int16(l), np.int16(r))
        dest.writeframes(packed_value)


def copyWav(dest: wave.Wave_write, src):
    for i in src:
        packed_value = struct.pack('<h', np.int16(i))
        dest.writeframes(packed_value)


def copyWav2(dest: wave.Wave_write, src):
    for i in src:
        packed_value = struct.pack('<hh', np.int16(i[0]), np.int16(i[1]))
        dest.writeframes(packed_value)


def getWavStr(file_dir, mode):
    try:
        source = wave.open(file_dir, mode)
        source_raw = source.readframes(-1)
        source_raw = np.fromstring(source_raw, 'Int16')

        print(file_dir, source.getnchannels())

        return source_raw
    except FileExistsError as e:
        print(e)
    except FileNotFoundError as e:
        print(e)


def wav2str(input_sound: wave.Wave_read):
    source_signal = input_sound.readframes(-1)

    if input_sound.getnchannels() == 1:
        source_signal = np.fromstring(source_signal, 'Int16')
    else:
        source_signal = np.fromstring(source_signal, 'Int16')
        source_signal = np.array(np.reshape(source_signal, (input_sound.getnframes(), 2)))
    return source_signal


def stereo2str(input: wave.Wave_read):
    source_signal = input.readframes(-1)
    source_signal = np.fromstring(source_signal, 'Int16')

    source_signal = np.array(np.reshape(source_signal, (input.getnframes(), 2)))

    return source_signal


def incVolume(wav_path):
    src = AudioSegment.from_wav(file_DIR + wav_path + '.wav')
    src = src - 10
    src.export(file_DIR + wav_path + '_dec' + '.wav', format='wav')


def decVolume(wav_path):
    src = AudioSegment.from_wav(file_DIR + wav_path + '.wav')
    src = src - 10
    src.export(file_DIR + wav_path + '_dec' + 'wav', format='wav')


def synthesize(src1: np.array, src2: np.array):
    return src1 + src2