package cn.xuan.test;

import javax.sound.sampled.AudioFormat;
import javax.sound.sampled.AudioInputStream;
import javax.sound.sampled.AudioSystem;
import javax.sound.sampled.SourceDataLine;
import javax.swing.*;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.Timer;
import java.util.TimerTask;
/**
 * 繪製波形圖
 *
 * FileName: Wavdemo
 * Author:   xuan zongjun
 * Date:     2018/10/18 17:16
 * Description:
 */


public class Wavdemo {

    public static class WaveformGraph extends JFrame {

        private LinkedList<Short> deque = new LinkedList<Short>();
        private Timer timer;
        private Image buffered;
        private Image showing;

        public WaveformGraph(int width, int height) {
            setSize(width, height);
            timer = new Timer();
            buffered = new BufferedImage(width, height, BufferedImage.TYPE_4BYTE_ABGR);
            timer.schedule(new TimerTask() {
                @Override public void run() {

                    Graphics g = buffered.getGraphics();
                    g.setColor(Color.WHITE);
                    g.fillRect(0, 0, getWidth(), getHeight());
                    g.setColor(Color.BLACK);

                    g.translate(10, getHeight()/2);

                    synchronized (deque) {
                        float heightRate = 1;
                        if(deque.size() > 1) {
                            Iterator<Short> iter = deque.iterator();
                            Short p1 = iter.next();
                            Short p2 = iter.next();
                            int x1 = 0, x2 = 0;
                            while(iter.hasNext()) {
                                g.drawLine(x1, (int)(p1*heightRate), x2, (int)(p2*heightRate));

                                p1 = p2;
                                p2 = iter.next();
                                x1 = x2;
                                x2 += 1;
                            }
                        }
                    }
                    g.dispose();

                    SwingUtilities.invokeLater(new Runnable() {
                        @Override public void run() {
                            showing = buffered;
                            repaint();
                            showing = null;
                        }
                    });
                }
            }, 100, 100);
        }

        @Override
        public void paint(Graphics g) {
            super.paint(g);
            if(buffered!=null) {
                g.drawImage(buffered, 0, 0, null);
            }
        }

        public void put(short v) {
            synchronized (deque) {
                deque.add(v);
                if(deque.size() > 500) {
                    deque.removeFirst();
                }
            }
        }

        public void clear() {
            deque.clear();
        }
    }

    public static void main(String[] args) throws Exception {
//  record();
        WaveformGraph waveformGraph = new WaveformGraph(500, 300);
        waveformGraph.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        waveformGraph.setVisible(true);

        AudioInputStream ais = AudioSystem.getAudioInputStream(new File("D:\\audio\\wav\\pcmpcmpcm.wav"));
        printFormat(ais.getFormat());


        SourceDataLine player = AudioSystem.getSourceDataLine(ais.getFormat());

        player.open();
        player.start();

        byte[] buf = new byte[4];
        int len;
        while((len=ais.read(buf))!=-1) {

            if(ais.getFormat().getChannels() == 2) {
                if(ais.getFormat().getSampleRate() == 16) {
                    waveformGraph.put((short) ((buf[1] << 8) | buf[0]));//左聲道

//     waveformGraph.put((short) ((buf[3] << 8) | buf[2]));//右聲道
                } else {
                    waveformGraph.put(buf[1]);//左聲道
                    waveformGraph.put(buf[3]);//左聲道

//     waveformGraph.put(buf[2]);//右聲道
//     waveformGraph.put(buf[4]);//右聲道
                }
            } else {
                if(ais.getFormat().getSampleRate() == 16) {
                    waveformGraph.put((short) ((buf[1] << 8) | buf[0]));
                    waveformGraph.put((short) ((buf[3] << 8) | buf[2]));
                } else {
                    waveformGraph.put(buf[1]);
                    waveformGraph.put(buf[2]);
                    waveformGraph.put(buf[3]);
                    waveformGraph.put(buf[4]);
                }
            }

            player.write(buf, 0, len);
        }

        player.close();
        ais.close();
    }

    public static void printFormat(AudioFormat format) {
        System.out.println(format.getEncoding() + " => "
                + format.getSampleRate()+" hz, "
                + format.getSampleSizeInBits() + " bit, "
                + format.getChannels() + " channel, "
                + format.getFrameRate() + " frames/second, "
                + format.getFrameSize() + " bytes/frame");
    }


}