/*
* Copyright 1999-2004 Carnegie Mellon University.
* Portions Copyright 2002-2004 Sun Microsystems, Inc.
* Portions Copyright 2002-2004 Mitsubishi Electric Research Laboratories.
* All Rights Reserved. Use is subject to license terms.
*
* See the file "license.terms" for information on usage and
* redistribution of this file, and for a DISCLAIMER OF ALL
* WARRANTIES.
*
*/
package edu.cmu.sphinx.tools.audio;
import edu.cmu.sphinx.frontend.Data;
import edu.cmu.sphinx.frontend.DoubleData;
import edu.cmu.sphinx.frontend.FrontEnd;
import edu.cmu.sphinx.frontend.util.Microphone;
import edu.cmu.sphinx.frontend.util.StreamDataSource;
import edu.cmu.sphinx.frontend.window.RaisedCosineWindower;
import edu.cmu.sphinx.util.props.ConfigurationManager;
import edu.cmu.sphinx.util.props.PropertySheet;
import javax.sound.sampled.*;
import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.*;
import java.net.URL;
import java.util.Arrays;
import java.util.StringTokenizer;
import java.util.prefs.Preferences;
/**
* Records and displays the waveform and spectrogram of an audio signal. See <a href="doc-files/HowToRunAudioTool.html">How
* to Run AudioTool</a> for information on how to run AudioTool.
*/
public class AudioTool {
static final String CONTEXT = "AudioTool";
static final String PREFS_CONTEXT = "/edu/cmu/sphinx/tools/audio/"
+ CONTEXT;
static final String FILENAME_PREFERENCE = "filename";
// -------------------------------
// Component names
// -----------------------------------
// These names should match the corresponding names in the
// spectrogram.config.xml
static final String MICROPHONE = "microphone";
static final String FRONT_END = "frontEnd";
static final String CESPTRUM_FRONT_END = "cepstrumFrontEnd";
static final String DATA_SOURCE = "streamDataSource";
static final String CEPSTRUM_DATA_SOURCE = "cstreamDataSource";
static final String WINDOWER = "windower";
static AudioData audio;
static JFrame jframe;
static AudioPanel audioPanel;
static SpectrogramPanel spectrogramPanel;
static CepstrumPanel cepstrumPanel;
static JFileChooser fileChooser;
static String filename;
static File file;
static AudioPlayer player;
static Microphone recorder;
static boolean recording;
static Preferences prefs;
static float zoom = 1.0f;
private static JMenuItem saveMenuItem;
private static JButton playButton;
private static JButton recordButton;
private static JButton zoomInButton;
private static JButton zoomOutButton;
private static JButton zoomResetButton;
private static ActionListener recordListener;
/* Dumps the information about a line. */
private static void dumpLineInfo(String indent,
Line.Info[] lineInfo) {
int numDumped = 0;
if (lineInfo != null) {
for (Line.Info info : lineInfo) {
if (info instanceof DataLine.Info) {
AudioFormat[] formats =
((DataLine.Info)info).getFormats();
for (AudioFormat format : formats) {
System.out.println(indent + format);
}
numDumped++;
} else if (info instanceof Port.Info) {
System.out.println(indent + info);
numDumped++;
}
}
}
if (numDumped == 0) {
System.out.println(indent + "none");
}
}
/* Lists all the available audio devices. */
private static void dumpMixers() {
Mixer.Info[] mixerInfo = AudioSystem.getMixerInfo();
for (int i = 0; i < mixerInfo.length; i++) {
Mixer mixer = AudioSystem.getMixer(mixerInfo[i]);
System.out.println("Mixer[" + i + "]: \""
+ mixerInfo[i].getName() + '\"');
System.out.println(" Description: "
+ mixerInfo[i].getDescription());
System.out.println(" SourceLineInfo (e.g., speakers):");
dumpLineInfo(" ", mixer.getSourceLineInfo());
System.out.println(" TargetLineInfo (e.g., microphones):");
dumpLineInfo(" ", mixer.getTargetLineInfo());
}
}
/* Gets a filename. */
static public void getFilename(String title, int type) {
int returnVal;
fileChooser.setDialogTitle(title);
fileChooser.setCurrentDirectory(file);
fileChooser.setDialogType(type);
if (type == JFileChooser.OPEN_DIALOG) {
returnVal = fileChooser.showOpenDialog(jframe);
} else {
returnVal = fileChooser.showSaveDialog(jframe);
}
if (returnVal == JFileChooser.APPROVE_OPTION) {
file = fileChooser.getSelectedFile();
filename = file.getAbsolutePath();
prefs.put(FILENAME_PREFERENCE, filename);
}
}
static public void populateAudio(String filename) {
try {
AudioData newAudio = Utils.readAudioFile(filename);
if (newAudio == null) {
newAudio = Utils.readRawFile(filename);
}
zoomReset();
audio.setAudioData(newAudio.getAudioData());
/*
* Play only if user requests. Auto play is annoying if
* the audio is too long
*
* player.play(audioPanel.getSelectionStart(),
* audioPanel.getSelectionEnd());
*/
} catch (IOException e) {
/* just ignore bad files. */
}
}
static public void getAudioFromFile(String filename) throws IOException {
/* Supports alignment data. The format of the alignment file
* is as follows:
*
* input filename String
* number of (time tag) lines int
* time tag float String
* time tag float String
* time tag float String
* ...
*
* Times are in seconds.
*/
if (filename.endsWith(".align")) {
BufferedReader reader = new BufferedReader(
new InputStreamReader(new FileInputStream(filename)));
populateAudio(reader.readLine());
int numPoints = Integer.parseInt(reader.readLine());
float[] times = new float[numPoints];
String[] labels = new String[numPoints];
for (int i = 0; i < numPoints; i++) {
StringTokenizer tokenizer = new StringTokenizer(
reader.readLine());
while (tokenizer.hasMoreTokens()) {
times[i] = Float.parseFloat(tokenizer.nextToken());
labels[i] = tokenizer.nextToken();
}
}
audioPanel.setLabels(times, labels);
reader.close();
} else {
populateAudio(filename);
}
}
/* Gets the audio that's in the recorder. This should only be called after recorder.stopRecording is called. */
static private short[] getRecordedAudio(Microphone recorder) {
short[] shorts = new short[0];
int sampleRate = 8000;
/* [[[WDW - TODO: this is not the most efficient way
* to do this, but it at least works for now.]]]
*/
while (recorder.hasMoreData()) {
try {
Data data = recorder.getData();
if (data instanceof DoubleData) {
sampleRate =
((DoubleData) data).getSampleRate();
double[] values =
((DoubleData) data).getValues();
short[] newShorts = Arrays.copyOf(shorts, shorts.length + values.length);
for (int i = 0; i < values.length; i++) {
newShorts[shorts.length + i] = (short)values[i];
}
shorts = newShorts;
}
} catch (Exception e) {
e.printStackTrace();
}
}
if (sampleRate > 8000) {
System.out.println("Downsampling from " +
sampleRate + " to 8000.");
shorts = Downsampler.downsample(
shorts,
sampleRate / 1000,
16);
}
return shorts;
}
/* Zoom the panels according to the zoom scale. */
private static void zoomPanels() {
if (audioPanel != null) {
audioPanel.zoomSet(zoom);
}
if (spectrogramPanel != null) {
spectrogramPanel.zoomSet(zoom);
}
if (cepstrumPanel != null) {
cepstrumPanel.zoomSet(zoom);
}
}
private static void zoomIn() {
zoom *= 2.0f;
zoomPanels();
}
private static void zoomOut() {
zoom /= 2.0f;
zoomPanels();
}
private static void zoomReset() {
zoom = 1.0f;
zoomPanels();
}
private static void createMenuBar(JFrame jframe) {
JMenuBar menuBar = new JMenuBar();
jframe.setJMenuBar(menuBar);
JMenu menu = new JMenu("File");
menuBar.add(menu);
JMenuItem menuItem = new JMenuItem("Open...");
menuItem.setAccelerator(KeyStroke.getKeyStroke("control O"));
menuItem.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent evt) {
getFilename("Open...", JFileChooser.OPEN_DIALOG);
if (filename == null || filename.isEmpty()) {
return;
}
try {
getAudioFromFile(filename);
} catch (Exception e) {
e.printStackTrace();
}
}
});
menu.add(menuItem);
saveMenuItem = new JMenuItem("Save");
saveMenuItem.setAccelerator(KeyStroke.getKeyStroke("control S"));
saveMenuItem.setEnabled(false);
saveMenuItem.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent evt) {
if (filename != null && !filename.isEmpty()) {
try {
Utils.writeRawFile(audio, filename);
saveMenuItem.setEnabled(false);
} catch (IOException e) {
e.printStackTrace();
}
}
}
});
menu.add(saveMenuItem);
menuItem = new JMenuItem("Save As...");
menuItem.setAccelerator(KeyStroke.getKeyStroke("control V"));
menuItem.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent evt) {
getFilename("Save As...", JFileChooser.SAVE_DIALOG);
if (filename == null || filename.isEmpty()) {
return;
}
try {
Utils.writeRawFile(audio, filename);
saveMenuItem.setEnabled(false);
} catch (IOException e) {
e.printStackTrace();
}
}
});
menu.add(menuItem);
menuItem = new JMenuItem("Quit");
menuItem.setAccelerator(KeyStroke.getKeyStroke("control Q"));
menuItem.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent evt) {
System.exit(0);
}
});
menu.add(menuItem);
menu = new JMenu("Edit");
menuBar.add(menu);
menuItem = new JMenuItem("Select All");
menuItem.setAccelerator(KeyStroke.getKeyStroke("control A"));
menuItem.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent evt) {
audioPanel.selectAll();
}
});
menu.add(menuItem);
menuItem = new JMenuItem("Crop");
menuItem.setAccelerator(KeyStroke.getKeyStroke("control X"));
menuItem.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent evt) {
audioPanel.crop();
}
});
menu.add(menuItem);
menu = new JMenu("View");
menuBar.add(menu);
menuItem = new JMenuItem("Zoom In");
menuItem.setAccelerator(KeyStroke.getKeyStroke('>'));
menuItem.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent evt) {
zoomIn();
}
});
menu.add(menuItem);
menuItem = new JMenuItem("Zoom Out");
menuItem.setAccelerator(KeyStroke.getKeyStroke('<'));
menuItem.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent evt) {
zoomOut();
}
});
menu.add(menuItem);
menuItem = new JMenuItem("Original Size");
menuItem.setAccelerator(KeyStroke.getKeyStroke('!'));
menuItem.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent evt) {
zoomReset();
}
});
menu.add(menuItem);
menu = new JMenu("Audio");
menuBar.add(menu);
menuItem = new JMenuItem("Play");
menuItem.setAccelerator(KeyStroke.getKeyStroke("control P"));
menuItem.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent evt) {
player.play(audioPanel.getSelectionStart(),
audioPanel.getSelectionEnd());
}
});
menu.add(menuItem);
recordListener = new ActionListener() {
public void actionPerformed(ActionEvent evt) {
if (!recording) {
recording = true;
recorder.startRecording();
recordButton.setText("Stop");
saveMenuItem.setEnabled(true);
} else {
recording = false;
recorder.stopRecording();
audio.setAudioData(getRecordedAudio(recorder));
recordButton.setText("Record");
player.play(audioPanel.getSelectionStart(),
audioPanel.getSelectionEnd());
}
}
};
menuItem = new JMenuItem("Record Start/Stop");
menuItem.setAccelerator(KeyStroke.getKeyStroke("control R"));
menuItem.addActionListener(recordListener);
menu.add(menuItem);
}
/**
* Create the Panel where all the buttons are.
* @return created panel
*/
private static JPanel createButtonPanel() {
JPanel buttonPanel = new JPanel();
FlowLayout layout = new FlowLayout();
layout.setAlignment(FlowLayout.LEFT);
buttonPanel.setLayout(layout);
playButton = new JButton("Play");
playButton.setEnabled(true);
playButton.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
player.play(audioPanel.getSelectionStart(),
audioPanel.getSelectionEnd());
}
});
recordButton = new JButton("Record");
recordButton.setEnabled(true);
recordButton.addActionListener(recordListener);
zoomInButton = new JButton("Zoom In");
zoomInButton.setEnabled(true);
zoomInButton.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
zoomIn();
}
});
zoomOutButton = new JButton("Zoom Out");
zoomOutButton.setEnabled(true);
zoomOutButton.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
zoomOut();
}
});
zoomResetButton = new JButton("Reset Size");
zoomResetButton.setEnabled(true);
zoomResetButton.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
zoomReset();
}
});
JButton exitButton = new JButton("Exit");
exitButton.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
System.exit(0);
}
});
buttonPanel.add(recordButton);
buttonPanel.add(playButton);
buttonPanel.add(zoomInButton);
buttonPanel.add(zoomOutButton);
buttonPanel.add(zoomResetButton);
buttonPanel.add(exitButton);
return buttonPanel;
}
static public void main(String[] args) {
FrontEnd frontEnd;
FrontEnd cepstrumFrontEnd;
StreamDataSource dataSource;
StreamDataSource cepstrumDataSource;
prefs = Preferences.userRoot().node(PREFS_CONTEXT);
filename = prefs.get(FILENAME_PREFERENCE, "untitled.raw");
file = new File(filename);
if ((args.length == 1) && (args[0].equals("-dumpMixers"))) {
dumpMixers();
System.exit(0);
}
try {
URL url;
if (args.length > 0) {
filename = args[0];
}
if (args.length == 2) {
url = new File(args[1]).toURI().toURL();
} else {
url = AudioTool.class.getResource("spectrogram.config.xml");
}
ConfigurationManager cm = new ConfigurationManager(url);
recorder = (Microphone) cm.lookup(MICROPHONE);
recorder.initialize();
audio = new AudioData();
frontEnd = (FrontEnd) cm.lookup(FRONT_END);
dataSource = (StreamDataSource) cm.lookup(DATA_SOURCE);
cepstrumFrontEnd = (FrontEnd) cm.lookup(CESPTRUM_FRONT_END);
cepstrumDataSource = (StreamDataSource) cm.lookup(CEPSTRUM_DATA_SOURCE);
PropertySheet ps = cm.getPropertySheet(WINDOWER);
float windowShiftInMs = ps.getFloat(RaisedCosineWindower.PROP_WINDOW_SHIFT_MS);
final JFrame jframe = new JFrame("AudioTool");
fileChooser = new JFileChooser();
createMenuBar(jframe);
/* Scale the width according to the size of the
* spectrogram.
*/
float windowShiftInSamples = windowShiftInMs
* audio.getAudioFormat().getSampleRate() / 1000.0f;
audioPanel = new AudioPanel(audio,
1.0f / windowShiftInSamples,
0.004f);
spectrogramPanel = new SpectrogramPanel(frontEnd, dataSource, audio);
cepstrumPanel = new CepstrumPanel(cepstrumFrontEnd, cepstrumDataSource, audio);
JPanel panel = new JPanel();
panel.setLayout(new BoxLayout(panel, BoxLayout.PAGE_AXIS));
panel.add(audioPanel);
audioPanel.setAlignmentX(0.0f);
panel.add(spectrogramPanel);
spectrogramPanel.setAlignmentX(0.0f);
panel.add(cepstrumPanel);
cepstrumPanel.setAlignmentX(0.0f);
JScrollPane scroller = new JScrollPane(panel);
JPanel outerPanel = new JPanel(new BorderLayout());
outerPanel.add(createButtonPanel(), BorderLayout.NORTH);
outerPanel.add(scroller);
player = new AudioPlayer(audio);
player.start();
getAudioFromFile(filename);
jframe.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
jframe.setContentPane(outerPanel);
jframe.pack();
jframe.setSize(640, 400);
jframe.setVisible(true);
} catch (Exception e) {
e.printStackTrace();
}
}
}