/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.codehaus.groovy.antlr;
import antlr.CharScanner;
import antlr.Token;
import org.codehaus.groovy.antlr.java.JavaLexer;
import org.codehaus.groovy.antlr.java.JavaTokenTypes;
import org.codehaus.groovy.antlr.parser.GroovyLexer;
import org.codehaus.groovy.antlr.parser.GroovyTokenTypes;
import org.codehaus.groovy.runtime.IOGroovyMethods;
import org.codehaus.groovy.runtime.ResourceGroovyMethods;
import javax.swing.*;
import javax.swing.border.Border;
import javax.swing.text.BadLocationException;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.io.File;
import java.io.FileReader;
import java.io.Reader;
import java.io.StringReader;
import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.util.Hashtable;
/**
* Swing application to graphically display the tokens produced by the lexer.
*/
public class LexerFrame extends JFrame implements ActionListener {
private final JSplitPane jSplitPane1 = new JSplitPane();
private final JScrollPane jScrollPane1 = new JScrollPane();
private final JScrollPane jScrollPane2 = new JScrollPane();
private final JTextPane tokenPane = new HScrollableTextPane();
private final JButton jbutton = new JButton("open");
private final JPanel mainPanel = new JPanel(new BorderLayout());
private final JTextArea scriptPane = new JTextArea();
private final Class lexerClass;
private final Hashtable tokens = new Hashtable();
/**
* Constructor used when invoking as a standalone application
*
* @param lexerClass the lexer class to use
* @param tokenTypesClass the lexer token types class
*/
public LexerFrame(Class lexerClass, Class tokenTypesClass) {
this(lexerClass, tokenTypesClass, null);
}
/**
* Constructor used when invoking for a specific file
*
* @param lexerClass the lexer class to use
* @param tokenTypesClass the lexer token types class
*/
public LexerFrame(Class lexerClass, Class tokenTypesClass, Reader reader) {
super("Token Steam Viewer");
this.lexerClass = lexerClass;
try {
jbInit(reader);
setSize(500, 500);
listTokens(tokenTypesClass);
if (reader == null) {
final JPopupMenu popup = new JPopupMenu();
popup.add(loadFileAction);
jbutton.setSize(30, 30);
jbutton.addMouseListener(new MouseAdapter() {
public void mouseReleased(MouseEvent e) {
//if(e.isPopupTrigger())
popup.show(scriptPane, e.getX(), e.getY());
}
});
} else {
safeScanScript(reader);
}
setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* Creates a Groovy language LexerFrame for the given script text
*
* @param scriptText the Groovy source file to parse/render
* @return the new frame rending the parsed tokens
*/
public static LexerFrame groovyScriptFactory(String scriptText) {
return new LexerFrame(GroovyLexer.class, GroovyTokenTypes.class, new StringReader(scriptText));
}
private void listTokens(Class tokenTypes) throws Exception {
for (Field field : tokenTypes.getDeclaredFields()) {
tokens.put(field.get(null), field.getName());
}
}
public void actionPerformed(ActionEvent ae) {
Token token = (Token) ((JComponent) ae.getSource()).getClientProperty("token");
if (token.getType() == Token.EOF_TYPE) {
scriptPane.select(0, 0);
return;
}
try {
int start = scriptPane.getLineStartOffset(token.getLine() - 1) + token.getColumn() - 1;
scriptPane.select(start, start + token.getText().length());
scriptPane.requestFocus();
} catch (BadLocationException ex) {
// IGNORE
}
}
private final Action loadFileAction = new AbstractAction("Open File...") {
public void actionPerformed(ActionEvent ae) {
final JFileChooser jfc = new JFileChooser();
final int response = jfc.showOpenDialog(LexerFrame.this);
if (response != JFileChooser.APPROVE_OPTION) {
return;
}
safeScanScript(jfc.getSelectedFile());
}
};
private void safeScanScript(File file) {
try {
scanScript(new StringReader(ResourceGroovyMethods.getText(file)));
} catch (final Exception ex) {
ex.printStackTrace();
}
}
private void safeScanScript(Reader reader) {
try {
scanScript(reader instanceof StringReader ? (StringReader) reader : new StringReader(IOGroovyMethods.getText(reader)));
} catch (final Exception ex) {
ex.printStackTrace();
}
}
private void scanScript(final StringReader reader) throws Exception {
scriptPane.read(reader, null);
reader.reset();
// create lexer
final Constructor constructor = lexerClass.getConstructor(Reader.class);
final CharScanner lexer = (CharScanner) constructor.newInstance(reader);
tokenPane.setEditable(true);
tokenPane.setText("");
int line = 1;
final ButtonGroup bg = new ButtonGroup();
Token token;
while (true) {
token = lexer.nextToken();
JToggleButton tokenButton = new JToggleButton((String) tokens.get(Integer.valueOf(token.getType())));
bg.add(tokenButton);
tokenButton.addActionListener(this);
tokenButton.setToolTipText(token.getText());
tokenButton.putClientProperty("token", token);
tokenButton.setMargin(new Insets(0, 1, 0, 1));
tokenButton.setFocusPainted(false);
if (token.getLine() > line) {
tokenPane.getDocument().insertString(tokenPane.getDocument().getLength(), "\n", null);
line = token.getLine();
}
insertComponent(tokenButton);
if (token.getType() == Token.EOF_TYPE) {
break;
}
}
tokenPane.setEditable(false);
tokenPane.setCaretPosition(0);
reader.close();
}
private void insertComponent(JComponent comp) {
try {
tokenPane.getDocument().insertString(tokenPane.getDocument().getLength(), " ", null);
} catch (BadLocationException ex1) {
// Ignore
}
try {
tokenPane.setCaretPosition(tokenPane.getDocument().getLength() - 1);
} catch (Exception ex) {
tokenPane.setCaretPosition(0);
}
tokenPane.insertComponent(comp);
}
private void jbInit(Reader reader) throws Exception {
final Border border = BorderFactory.createEmptyBorder();
jSplitPane1.setOrientation(JSplitPane.VERTICAL_SPLIT);
tokenPane.setEditable(false);
tokenPane.setText("");
scriptPane.setFont(new java.awt.Font("DialogInput", 0, 12));
scriptPane.setEditable(false);
scriptPane.setMargin(new Insets(5, 5, 5, 5));
scriptPane.setText("");
jScrollPane1.setBorder(border);
jScrollPane2.setBorder(border);
jSplitPane1.setMinimumSize(new Dimension(800, 600));
mainPanel.add(jSplitPane1, BorderLayout.CENTER);
if (reader == null) {
mainPanel.add(jbutton, BorderLayout.NORTH);
}
this.getContentPane().add(mainPanel);
jSplitPane1.add(jScrollPane1, JSplitPane.LEFT);
jScrollPane1.getViewport().add(tokenPane, null);
jSplitPane1.add(jScrollPane2, JSplitPane.RIGHT);
jScrollPane2.getViewport().add(scriptPane, null);
jScrollPane1.setColumnHeaderView(new JLabel(" Token Stream:"));
jScrollPane2.setColumnHeaderView(new JLabel(" Input Script:"));
jSplitPane1.setResizeWeight(0.5);
}
public static void main(String[] args) throws Exception {
try {
UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
} catch (Exception ignore) {
// Ignore
}
LexerFrame lexerFrame = null;
if (args.length == 0) {
lexerFrame = new LexerFrame(GroovyLexer.class, GroovyTokenTypes.class);
} else if (args.length > 1) {
System.err.println("usage: java LexerFrame [filename.ext]");
System.exit(1);
} else {
String filename = args[0];
if (filename.endsWith(".java")) {
lexerFrame = new LexerFrame(JavaLexer.class, JavaTokenTypes.class, new FileReader(filename));
} else {
lexerFrame = new LexerFrame(GroovyLexer.class, GroovyTokenTypes.class, new FileReader(filename));
}
}
lexerFrame.setVisible(true);
}
private static class HScrollableTextPane extends JTextPane {
@Override
public boolean getScrollableTracksViewportWidth() {
return (getSize().width < getParent().getSize().width);
}
@Override
public void setSize(final Dimension d) {
if (d.width < getParent().getSize().width) {
d.width = getParent().getSize().width;
}
super.setSize(d);
}
}
}