AudioRecorder.java

/**
 * @(#)AudioRecorder.java
 *
 * Copyright (c) 2012 Werner Randelshofer, Goldau, Switzerland. All rights
 * reserved.
 *
 * You may not use, copy or modify this file, except in compliance with the
 * license agreement you entered into with Werner Randelshofer. For details see
 * accompanying license terms.
 */
package org.monte.audiodemo;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.sound.sampled.AudioFormat;
import javax.sound.sampled.AudioSystem;
import javax.sound.sampled.DataLine;
import javax.sound.sampled.LineUnavailableException;
import javax.sound.sampled.TargetDataLine;
import org.monte.media.AudioFormatKeys;
import org.monte.media.Buffer;
import org.monte.media.math.Rational;
import org.monte.media.avi.AVIWriter;

/**
 * {@code AudioRecorder} records audio into an AVI file.
 *
 * @author Werner Randelshofer
 * @version 1.0 2012-12-21 Created.
 */
public class AudioRecorder implements Runnable {

    private volatile Thread worker;
    private File file;

    public AudioRecorder(File file) {
        this.file = file;
    }

    public void start() throws LineUnavailableException {
        stop();
        worker = new Thread(this);
        worker.start();
    }

    public void stop() {
        if (worker != null) {
            Thread stopMe = worker;
            worker = null;
            try {
                stopMe.join();
            } catch (InterruptedException ex) {
                //ignore
            }
        }
    }

    /**
     * This method is called from the worker thread.
     */
    @Override
    public void run() {
        DataLine.Info lineInfo = new DataLine.Info(
                TargetDataLine.class, new AudioFormat(44100, 16, 2, true, true));

        AVIWriter writer = null;
        TargetDataLine line = null;
        try {
            line = (TargetDataLine) AudioSystem.getLine(lineInfo);
            AudioFormat lineFormat = line.getFormat();
            Buffer buf = new Buffer();
            buf.format = AudioFormatKeys.fromAudioFormat(lineFormat);
            buf.sampleDuration = new Rational(1, (long) lineFormat.getSampleRate());
            buf.data = new byte[(int) (lineFormat.getFrameSize() * lineFormat.getSampleRate())];
            writer = new AVIWriter(file);
            writer.addTrack(buf.format);
            line.open();
            line.start();
            
            
            while (worker == Thread.currentThread()) {
                buf.length = line.read((byte[]) buf.data, 0, ((byte[]) buf.data).length);
                buf.sampleCount = buf.length / lineFormat.getFrameSize();
                writer.write(0, buf);
            }
        } catch (IOException ex) {
            ex.printStackTrace();
        } catch (LineUnavailableException ex) {
            ex.printStackTrace();
        } finally {
            if (line != null) {
                line.close();
            }
            if (writer != null) {
                try {
                    writer.close();
                } catch (IOException ex) {
                    ex.printStackTrace();
                }
            }
        }
    }

    /**
     * @param args the command line arguments
     */
    public static void main(String[] args) throws IOException, LineUnavailableException {
        SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd 'at' HH.mm.ss");
        File file = new File(System.getProperty("user.home"), "Movies/AudioRecording " + dateFormat.format(new Date()) + ".avi");
        if (!file.getParentFile().isDirectory()) {
            file.getParentFile().mkdirs();
        }


        System.out.println("Press ENTER to start audio recording.");
        while (System.in.read() != '\n') ;
        AudioRecorder r = new AudioRecorder(file);
        r.start();
        System.out.println("Press ENTER to stop audio recording.");
        while (System.in.read() != '\n') ;
        r.stop();

        System.out.println("You can find the recorded audio in file");
        System.out.println(file);
    }
}