Microphone.java

package com.darkprograms.speech.microphone;

import javax.sound.sampled.*;

import java.io.Closeable;
import java.io.File;

/***************************************************************************
 * Microphone class that contains methods to capture audio from microphone
 *
 * @author Luke Kuza, Aaron Gokaslan
 ***************************************************************************/
public class Microphone implements Closeable{

    /**
     * TargetDataLine variable to receive data from microphone
     */
    private TargetDataLine targetDataLine;


    /**
     * Enum for current Microphone state
     */
    public enum CaptureState {
        PROCESSING_AUDIO, STARTING_CAPTURE, CLOSED;
    }

    /**
     * Variable for enum
     */
    CaptureState state;

    /**
     * Variable for the audios saved file type
     */
    private AudioFileFormat.Type fileType;

    /**
     * Variable that holds the saved audio file
     */
    private File audioFile;

    private AudioInputStream audioStream;
    private float sampleRate;

    /**
     * Constructor
     *
     * @param fileType File type to save the audio in<br>
     *                 Example, to save as WAVE use AudioFileFormat.Type.WAVE
     */
    public Microphone(AudioFileFormat.Type fileType) {
        setState(CaptureState.CLOSED);
        setFileType(fileType);
        initTargetDataLine();
    }

    /**
     * Constructor for use with {@link #captureAudioToStream()}
     * @param sampleRate samples per second - 16_000 (recommended) or 8_000
     */
    public Microphone(float sampleRate) {
        setState(CaptureState.CLOSED);
        initTargetDataLine(sampleRate);
    }

    /**
     * Gets the current state of Microphone
     *
     * @return PROCESSING_AUDIO is returned when the Thread is recording Audio and/or saving it to a file<br>
     *         STARTING_CAPTURE is returned if the Thread is setting variables<br>
     *         CLOSED is returned if the Thread is not doing anything/not capturing audio
     */
    public CaptureState getState() {
        return state;
    }

    /**
     * Sets the current state of Microphone
     *
     * @param state State from enum
     */
    private void setState(CaptureState state) {
        this.state = state;
    }

    public File getAudioFile() {
        return audioFile;
    }

    public void setAudioFile(File audioFile) {
        this.audioFile = audioFile;
    }

    public AudioFileFormat.Type getFileType() {
        return fileType;
    }

    public void setFileType(AudioFileFormat.Type fileType) {
        this.fileType = fileType;
    }

    public TargetDataLine getTargetDataLine() {
        return targetDataLine;
    }

    
    public void setTargetDataLine(TargetDataLine targetDataLine) {
        this.targetDataLine = targetDataLine;
    }

    /**
     * Initializes the target data line.
     */
    private TargetDataLine initTargetDataLine() {
        return initTargetDataLine(8_000F);
    }
    private TargetDataLine initTargetDataLine(float sampleRate) {
        this.sampleRate = sampleRate;
        DataLine.Info dataLineInfo = new DataLine.Info(TargetDataLine.class, getAudioFormat());
        try {
            TargetDataLine targetDataLine = (TargetDataLine)AudioSystem.getLine(dataLineInfo);
			setTargetDataLine(targetDataLine);
			return targetDataLine;
		} catch (LineUnavailableException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			return null;
		}
    }

    public AudioInputStream captureAudioToStream() {
        setState(CaptureState.STARTING_CAPTURE);
        if(getTargetDataLine() == null) {
            initTargetDataLine();
        }

        open();
        audioStream = new AudioInputStream(getTargetDataLine());
        return audioStream;
    }

    /**
     * Captures audio from the microphone and saves it a file
     *
     * @param audioFile The File to save the audio to
     * @throws LineUnavailableException 
     */
    public void captureAudioToFile(File audioFile) throws LineUnavailableException {
        setState(CaptureState.STARTING_CAPTURE);
        setAudioFile(audioFile);

        if(getTargetDataLine() == null){
        	initTargetDataLine();
        }

        //Get Audio
        new Thread(new CaptureThread()).start();
    }

    /**
     * Captures audio from the microphone and saves it a file
     *
     * @param audioFile The fully path (String) to a file you want to save the audio in
     * @throws LineUnavailableException 
     */
    public void captureAudioToFile(String audioFile) throws LineUnavailableException {
        File file = new File(audioFile);
        captureAudioToFile(file);
    }

    /**
     * The audio format to save in
     *
     * @return Returns AudioFormat to be used later when capturing audio from microphone
     */
    public AudioFormat getAudioFormat() {
        return getAudioFormat(sampleRate);
    }

    /**
     * @param sampleRate set to 16_000.0F for AWS Lex
     */
    public AudioFormat getAudioFormat(float sampleRate) {
        //8000,11025,16000,22050,44100
        int sampleSizeInBits = 16;
        //8,16
        int channels = 1;
        //1,2
        boolean signed = true;
        //true,false
        boolean bigEndian = false;
        //true,false
        return new AudioFormat(sampleRate, sampleSizeInBits, channels, signed, bigEndian);
    }

    /**
     * Opens the microphone, starting the targetDataLine.
     * If it's already open, it does nothing.
     */
    public void open(){
        if(getTargetDataLine()==null){
        	initTargetDataLine();
        }
        TargetDataLine targetDataLine = getTargetDataLine();
        if(!targetDataLine.isOpen() && !targetDataLine.isRunning() && !targetDataLine.isActive()) {
           	try {
                setState(CaptureState.PROCESSING_AUDIO);

                try {
System.out.println("???????????????????????????????????????????????????????????");
                    System.out.println("???????????????????????????????????????????????????????????");System.out.println("???????????????????????????????????????????????????????????");System.out.println("???????????????????????????????????????????????????????????");
                    System.out.println("???????????????????????????????????????????????????????????");


                    if (targetDataLine.isControlSupported(FloatControl.Type.MASTER_GAIN)) {
System.out.println("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ Setting gain!!!!!!!!!!!!!!");
                        System.out.println("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ Setting gain!!!!!!!!!!!!!!");
                        System.out.println("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ Setting gain!!!!!!!!!!!!!!");
                        System.out.println("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ Setting gain!!!!!!!!!!!!!!");
                        System.out.println("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ Setting gain!!!!!!!!!!!!!!");
                        System.out.println("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ Setting gain!!!!!!!!!!!!!!");
                        System.out.println("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ Setting gain!!!!!!!!!!!!!!");
                        System.out.println("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ Setting gain!!!!!!!!!!!!!!");
                        System.out.println("+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ Setting gain!!!!!!!!!!!!!!");
                        FloatControl gainControl = (FloatControl) getTargetDataLine().getControl(FloatControl.Type.MASTER_GAIN);
                        gainControl.setValue(40);
                    }
                } catch (Exception e) {
                    try {
                        FloatControl gainControl = (FloatControl) getTargetDataLine().getControl(FloatControl.Type.VOLUME);
                        gainControl.setValue(-10);
                    } catch (Exception e1) {
                        e1.printStackTrace();
                    }
                }
        		getTargetDataLine().open(getAudioFormat());

            	getTargetDataLine().start();
			} catch (LineUnavailableException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
				return;
			}
        }
    }

    /**
     * Close the microphone capture, saving all processed audio to the specified file.<br>
     * If already closed, this does nothing
     */
    public void close() {
        if (getState() == CaptureState.CLOSED) {
        } else {
            getTargetDataLine().stop();
            getTargetDataLine().close();
            setState(CaptureState.CLOSED);
        }
    }

    /**
     * Thread to capture the audio from the microphone and save it to a file
     */
    private class CaptureThread implements Runnable {
        /**
         * Run method for thread
         */
        public void run() {
            try {
                AudioFileFormat.Type fileType = getFileType();
                File audioFile = getAudioFile();
                open();
                AudioSystem.write(new AudioInputStream(getTargetDataLine()), fileType, audioFile);
                //Will write to File until it's closed.
            } catch (Exception ex) {
                ex.printStackTrace();
            }
        }
    }

    /*private class ListenThread implements Runnable {
        public void run() {
            try {
                open();
                audioStream = new AudioInputStream(getTargetDataLine());
            } catch (Exception ex) {
                ex.printStackTrace();
            }
        }
    }*/
}