package org.dromara.easyai.voice;


import java.io.IOException;
import java.io.InputStream;

public class MP3 {

    /**
     * 横坐标为MPEG(V),纵坐标为Layer(L),sample[0][2]为MPEG-1,Layer-3的每帧采样数
     */

    private final static int[][] Mp3_Sample = {{384, 384, 384},
            {1152, 1152, 1152}, {1152, 576, 576}};

    /**
     * 二维数组长度为14,横坐标范围:1~14.MPEG1中,纵坐标分别对应Layer-1,Layer-2,Layer-3.MPEG2中,纵坐标分别对应Layer-1,Layer-2或Layer-3.
     */
    private final static int[][] MPeg1ByteRate = {{32, 32, 32},
            {64, 48, 40}, {96, 56, 48}, {128, 64, 56}, {160, 80, 64},
            {192, 96, 80}, {224, 112, 96}, {256, 128, 112},
            {288, 160, 128}, {320, 192, 160}, {352, 224, 192},
            {384, 256, 224}, {416, 320, 256}, {448, 384, 320}};

    private final static int[][] MPeg2ByteRate = {{32, 8}, {48, 16},
            {56, 24}, {64, 32}, {80, 40}, {96, 48}, {112, 56},
            {128, 64}, {144, 80}, {160, 96}, {176, 112}, {192, 128},
            {224, 144}, {256, 160}};

    /**
     * 采样频率.横坐标为变量,纵坐标为MPEG版本.
     */
    private final static int[][] SampleFrequency = {{44100, 48000, 32000},
            {22050, 24000, 16000}, {11025, 12000, 8000}};

    // ==================================================================================
    private InputStream stream;

    public MP3(InputStream stream) {
        this.stream = stream;
    }

    /**
     * MpeG版本
     */
    private int Mpeg_Version;

    /**
     * Layer版本
     */
    private int Layer_Version;

    /**
     * 采样速率.(kbps)
     */
    private int ByteRate;

    /**
     * 采样频率.khz
     */
    private int Frequency;

    /**
     * 帧长度调整值.0或1
     */
    private int Padding;

    /**
     * 采样位数.
     */
    private int Sample;

    public int parserMp3Header() throws IOException {
        FrameHeader = new int[3];
        byte[] tempBytes = readFull(3);
        if (tempBytes == null) {
            return -1;
        }
        for (int i = 0; i < 3; i++) {
            FrameHeader[i] = tempBytes[i] & 0xFF;
        }
        int TagHeaderSize = 0;
        if (FrameHeader[0] == 'I' && FrameHeader[1] == 'D'
                && FrameHeader[2] == '3') {
            byte[] tagHeader = readFull(7);
            if (tagHeader == null) {
                return -1;
            }
            int tagHeaderSize = ((tagHeader[3] & 0x7F) << 21)
                    + ((tagHeader[4] & 0x7F) << 14)
                    + ((tagHeader[5] & 0x7F) << 7) + (tagHeader[6] & 0x7F);
            // tagHeaderSize不包括前面10个字节.加上10以后,是整个标签头的大小.
            if (!skipBytes(tagHeaderSize)) {
                return -1;
            }
            // IDV3标签头的长度.
            TagHeaderSize = tagHeaderSize + 10;
            tempBytes = readFull(3);
            if (tempBytes == null) {
                return -1;
            }
            for (int i = 0; i < 3; i++) {
                FrameHeader[i] = tempBytes[i] & 0xFF;
            }
        }
        if (FrameHeader[0] != 0xFF || (FrameHeader[1] >> 5) != 7) {
            return -1;
        }
        switch ((FrameHeader[1] & 24) >> 3) {
            case 0:
                Mpeg_Version = 3;// MPEG-2.5
                break;
            case 2:
                Mpeg_Version = 2;
                break;
            case 3:
                Mpeg_Version = 1;
                break;
            default:
                return -1;
        }
        switch ((FrameHeader[1] & 6) >> 1) {
            case 1:
                Layer_Version = 3;
                break;
            case 2:
                Layer_Version = 2;
                break;
            case 3:
                Layer_Version = 1;
                break;
            default:
                return -1;
        }
        int index = FrameHeader[2] >> 4;
        if (index < 1 || index > 14) {
            return -1;
        }
        --index;
        if (Mpeg_Version == 1) {
            switch (Layer_Version) {
                case 1:
                    ByteRate = MPeg1ByteRate[index][0];
                    break;
                case 2:
                    ByteRate = MPeg1ByteRate[index][1];
                    break;
                case 3:
                    ByteRate = MPeg1ByteRate[index][2];
                    break;
            }
        } else {
            switch (Layer_Version) {
                case 1:
                    ByteRate = MPeg2ByteRate[index][0];
                    break;
                case 2:
                case 3:
                    ByteRate = MPeg2ByteRate[index][1];
                    break;
            }
        }
        Frequency = SampleFrequency[Mpeg_Version - 1][(FrameHeader[2] & 12) >> 2];
        Padding = (FrameHeader[2] & 2) >> 1;
        Sample = Mp3_Sample[Layer_Version - 1][Mpeg_Version - 1];
        return TagHeaderSize;
    }

    public long SkipFrame(long num) throws IOException {
        long skipped = 0;
        byte[] temp = null;
        for (int i = 0; i < num; i++) {
            temp = ParserFrame();
            if (temp == null) {
                return skipped;
            }
            skipped += temp.length;
        }
        return skipped;
    }

    /**
     * 第一帧的前三个字节.用来提取除采样速率以外的其它定值. 在前三个字节中,第一二两个字节是不变的.只有第三个字节会左右帧长度的变化.
     */
    private int[] FrameHeader;

    /**
     * 发生网络数据错误,或者没有读到需要的数据,都会抛出异常.因此,外部调用程序要控制数据大小,到了MP3.length -
     * 128的时候就不要再读了,程序不会检查是否已经到了末尾.
     *
     * @return byte[] 读到的一帧完整的可以播放的数据.遇到IDV1时,返回NULL
     * @throws IOException
     * @throws Exception
     */
    public byte[] ParserFrame() throws IOException {
        byte[] byteFrameHeader;
        int index = 0;
        if (FrameHeader == null) {
            byteFrameHeader = readFull(3);
            if (byteFrameHeader == null || byteFrameHeader.length < 3
                    || (byteFrameHeader[0] & 0xFF) == 'T'
                    && (byteFrameHeader[1] & 0xFF) == 'A'
                    && (byteFrameHeader[2] & 0xFF) == 'G') {
                return null;
            }
            if ((byteFrameHeader[0] & 0xFF) != 0xFF
                    || ((byteFrameHeader[1] & 0xFF) >> 5) != 7) {
                System.out.println("该MP3文件非法.");
                return null;// 帧头的第一个字节和第二个字节的前三位不全部为1,该MP3文件非法.
            }
            index = (byteFrameHeader[2] & 0xFF) >> 4;
            Padding = (byteFrameHeader[2] & 2) >> 1;
        } else {
            if (FrameHeader == null || FrameHeader.length < 3
                    || FrameHeader[0] == 'T' && FrameHeader[1] == 'A'
                    && FrameHeader[2] == 'G') {
                return null;
            }
            index = FrameHeader[2] >> 4;
            byteFrameHeader = new byte[3];
            byteFrameHeader[0] = (byte) FrameHeader[0];
            byteFrameHeader[0] = (byte) FrameHeader[1];
            byteFrameHeader[0] = (byte) FrameHeader[2];
            FrameHeader = null;
        }
        if (index < 1 || index > 14) {
            return null;// 获取位速的查找索引非法.
        }
        --index;
        if (Mpeg_Version == 1) {
            switch (Layer_Version) {
                case 1:
                    ByteRate = MPeg1ByteRate[index][0];
                    break;
                case 2:
                    ByteRate = MPeg1ByteRate[index][1];
                    break;
                case 3:
                    ByteRate = MPeg1ByteRate[index][2];
                    break;
            }
        } else {
            switch (Layer_Version) {
                case 1:
                    ByteRate = MPeg2ByteRate[index][0];
                    break;
                case 2:
                case 3:
                    ByteRate = MPeg2ByteRate[index][1];
                    break;
            }
        }
        // 计算帧长.(ByteRate,Frequency,Padding是可变的)
        int frameSize = Sample / 8 * ByteRate * 1000 / Frequency + Padding;
        byte[] temp = readFull(frameSize - 3);
        if (temp == null) {
            return null;
        }
        byte[] data = new byte[frameSize];
        System.arraycopy(byteFrameHeader, 0, data, 0, 3);
        System.arraycopy(temp, 0, data, 3, temp.length);
        return data;
    }

    public byte[] readFull(int size) throws IOException {
        byte[] data = new byte[size];
        int n = 0;
        while (n < size) {
            int k = stream.read(data, n, size - n);
            if (k < 0) {
                break;
            }
            n += k;
        }
        if (n <= 0) {
            return null;
        }
        if (n < size) {
            byte[] temp = new byte[n];
            System.arraycopy(data, 0, temp, 0, n);
            data = null;
            return temp;
        }
        return data;
    }

    public boolean skipBytes(long skipLength) throws IOException {
        long k = 0;
        while (k < skipLength) {
            long n = stream.skip(skipLength - k);
            if (n < 0) {
                return false;
            }
            k += n;
        }
        return true;
    }

}