从写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;

}

}