Add File

src/main/java/org/dromara/easyai/voice/MP3.java

@@ -0,0 +1,299 @@
+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;
+    }
+
+}