从写JTextPane组件 实现关键字高亮显示

package my2;

import java.util.StringTokenizer;
import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import javax.swing.text.*;
import javax.swing.text.rtf.RTFEditorKit;
/**
* 关键字特殊处理面板
* @author Administrator
*
*/
public class MyTextPane extends JTextPane {
/**
*
*/
private static final long serialVersionUID = -66377652770879651L;
protected StyleContext m_context;
protected DefaultStyledDocument m_doc;
private MutableAttributeSet keyAttr, normalAttr;
private MutableAttributeSet bracketAttr;
private MutableAttributeSet inputAttributes = new RTFEditorKit()
.getInputAttributes();
/**
* 所有关键字
*/
private final static String[] _keys = new String[] { "select", "from",
"where", "like", "and", "or", "order", "group", "sum", "avg",
"not", "in", "create", "grand", "null", "count", "max", "min",
"start", "with", "connect", "update", "delete", "set", "values",
"view", "table", "as", "distinct", "into", "drop", "is", "on",
"exists", "by", "tree", "table", "cust", "union", "dual",
"trigger", "function", "procedure", "begin", "end", "for", "loop",
"while", "insert", "count", "if", "else", "then", "commit",
"rollback", "return", "declare", "when", "elsif", "open", "fetch",
"close", "exit", "exception", "execute" };
/**
* 所与排除字符集
*/
private final static char[] _character = new char[] { '(', ')', ',', ';',
':', '\t', '\n', '+', '-', '*', '/' };
/**
* 初始化，包括关键字颜色，和非关键字颜色
*/
public MyTextPane() {
super();
m_context = new StyleContext();
m_doc = new DefaultStyledDocument(m_context);
this.setDocument(m_doc);
this.addKeyListener(new KeyAdapter() {
public void keyReleased(KeyEvent ke) {
dealSingleRow();
}
});
// 义关键字显示属性
keyAttr = new SimpleAttributeSet();
StyleConstants.setForeground(keyAttr, Color.RED);
// 义一般文本显示属性
normalAttr = new SimpleAttributeSet();
// StyleConstants.setFontFamily(normalAttr, "serif");
StyleConstants.setBold(normalAttr, false);
StyleConstants.setForeground(normalAttr, Color.black);
bracketAttr = new SimpleAttributeSet();
StyleConstants.setForeground(bracketAttr, Color.RED);
// StyleConstants.setFontFamily(bracketAttr, "serif");
StyleConstants.setBold(bracketAttr, true);
}
/**
* 设置大括号颜色
* @param _text
*/
private void setBracketColor(String _text) {
int len = _text.length();
for (int i = 0; i < len; i++) {
char ch = _text.charAt(i);
if (ch == '{' || ch == '}') {
m_doc.setCharacterAttributes(i, 1, bracketAttr, false);
}
}
}
/**
* 判断字符是不是在排除字符行列
* @param _ch
* @return
*/
private boolean isCharacter(char _ch) {
for (int i = 0; i < _character.length; i++) {
if (_ch == _character[i]) {
return true;
}
}
return false;
}
/**
* 设置关键字颜色
* @param _key
* @param _start
* @param _length
* @return
*/
private int setKeyColor(String _key, int _start, int _length) {
for (int i = 0; i < _keys.length; i++) {
int li_index = _key.indexOf(_keys[i]);
if (li_index < 0) {
continue;
}
int li_legnth = li_index + _keys[i].length();
if (li_legnth == _key.length()) {
if (li_index == 0) {//处理单独一个关键字的情况，例如：if else 等
m_doc.setCharacterAttributes(_start, _keys[i].length(),
keyAttr, false);
} else {//处理关键字前面还有字符的情况，例如：)if ;else 等
char ch_temp = _key.charAt(li_index - 1);
if (isCharacter(ch_temp)) {
m_doc.setCharacterAttributes(_start + li_index,
_keys[i].length(), keyAttr, false);
}
}
} else {
if (li_index == 0) {//处理关键字后面还有字符的情况，例如：if( end;等
char ch_temp = _key.charAt(_keys[i].length());
if (isCharacter(ch_temp)) {
m_doc.setCharacterAttributes(_start, _keys[i].length(),
keyAttr, false);
}
} else {//处理关键字前面和后面都有字符的情况，例如：)if( 等
char ch_temp = _key.charAt(li_index - 1);
char ch_temp_2 = _key.charAt(li_legnth);
if (isCharacter(ch_temp) && isCharacter(ch_temp_2)) {
m_doc.setCharacterAttributes(_start + li_index,
_keys[i].length(), keyAttr, false);
}
}
}
}
return _length + 1;
}
/**
* 处理一行的数据
* @param _start
* @param _end
*/
private void dealText(int _start, int _end) {
String text = "";
try {
text = m_doc.getText(_start, _end - _start).toUpperCase();
} catch (BadLocationException e) {
e.printStackTrace();
}
if (text == null || text.equals("")) {
return;
}
int xStart = 0;
// 析关键字---
m_doc.setCharacterAttributes(_start, text.length(), normalAttr, false);
MyStringTokenizer st = new MyStringTokenizer(text);
while (st.hasMoreTokens()) {
String s = st.nextToken();
if (s == null)
return;
xStart = st.getCurrPosition();
setKeyColor(s.toLowerCase(), _start + xStart, s.length());
}
setBracketColor(text);
inputAttributes.addAttributes(normalAttr);
}
/**
* 在进行文本修改的时候
* 获得光标所在行，只对该行进行处理
*/
private void dealSingleRow() {
Element root = m_doc.getDefaultRootElement();
// 光标当前行
int cursorPos = this.getCaretPosition(); // 前光标的位置
int line = root.getElementIndex(cursorPos);// 当前行
Element para = root.getElement(line);
int start = para.getStartOffset();
int end = para.getEndOffset() - 1;// 除\r字符
dealText(start, end);
}
/**
* 在初始化面板的时候调用该方法，
* 查找整个篇幅的关键字
*/
public void syntaxParse() {
Element root = m_doc.getDefaultRootElement();
int li_count = root.getElementCount();
for (int i = 0; i < li_count; i++) {
Element para = root.getElement(i);
int start = para.getStartOffset();
int end = para.getEndOffset() - 1;// 除\r字符
dealText(start, end);
}
}
}
/**
* 在分析字符串的同时，记录每个token所在的位置
*
*/
class MyStringTokenizer extends StringTokenizer {
String sval = " ";
String oldStr, str;
int m_currPosition = 0, m_beginPosition = 0;
MyStringTokenizer(String str) {
super(str, " ");
this.oldStr = str;
this.str = str;
}
public String nextToken() {
try {
String s = super.nextToken();
int pos = -1;
if (oldStr.equals(s)) {
return s;
}
pos = str.indexOf(s + sval);
if (pos == -1) {
pos = str.indexOf(sval + s);
if (pos == -1)
return null;
else
pos += 1;
}
int xBegin = pos + s.length();
str = str.substring(xBegin);
m_currPosition = m_beginPosition + pos;
m_beginPosition = m_beginPosition + xBegin;
return s;
} catch (java.util.NoSuchElementException ex) {
ex.printStackTrace();
return null;
}
}
// 返回token在字符串中的位置
public int getCurrPosition() {
return m_currPosition;
}
}