convex.gui.utils.CVXHighlighter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of convex-gui Show documentation
Convex desktop GUI and test applications
The newest version!
package convex.gui.utils;

import java.awt.Color;

import javax.swing.JTextPane;
import javax.swing.text.AttributeSet;
import javax.swing.text.SimpleAttributeSet;
import javax.swing.text.StyleConstants;
import javax.swing.text.StyleContext;
import javax.swing.text.StyledDocument;

import org.antlr.v4.runtime.CharStream;
import org.antlr.v4.runtime.CharStreams;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.Lexer;
import org.antlr.v4.runtime.Token;

import convex.core.lang.reader.AntlrReader;

/**
 * Tools for highlighting Convex Lisp code in Swing
 */
public class CVXHighlighter {
	static StyleContext sc = StyleContext.getDefaultStyleContext();
	static AttributeSet BASE = sc.addAttribute(SimpleAttributeSet.EMPTY, StyleConstants.Italic, false);
	static AttributeSet ITALIC = sc.addAttribute(SimpleAttributeSet.EMPTY, StyleConstants.Italic, true);
	
	static AttributeSet NORMAL = sc.addAttribute(BASE, StyleConstants.Foreground, Color.LIGHT_GRAY);
	static AttributeSet STRING = sc.addAttribute(ITALIC, StyleConstants.Foreground, Color.CYAN);
	static AttributeSet KEYWORD = sc.addAttribute(BASE, StyleConstants.Foreground, Color.PINK);
	static AttributeSet SYMBOL = NORMAL;
	static AttributeSet NUMBER = sc.addAttribute(BASE, StyleConstants.Foreground, Color.ORANGE);

	private static final AttributeSet[] PARENS = new AttributeSet[10];
	
	static {
		// Generator code for Rainbow parens
		for (int i=0; i0) {
					char c=input.charAt(tstart-start);
					
					AttributeSet aset=NORMAL;
					switch (c) {
						case '"': {
							aset=STRING; break;
						}
						case ':': {
							aset=KEYWORD; break;
						}
						case '(':{
							aset=getParen(nest++); break;
						}
						case ')': {
							aset=getParen(--nest); break;
						}
						case '[':{
							nest+=2;
							aset=getParen(nest++); break;
						}
						case ']': {
							aset=getParen(--nest); 
							nest-=2; break;
						}
						case '{':{
							nest+=4;
							aset=getParen(nest++); break;
						}
						case '}': {
							aset=getParen(--nest); 
							nest-=4; break;
						}
						// Whitespace uses paren colour, this colours commas!
						case ' ': case '\t': case ',': {
							aset=getParen(nest); 
							break;
						}

						default: {
							if (Character.isDigit(c)) {
								aset=NUMBER; break;
							}
						}
					}
					
					d.setCharacterAttributes(tstart, tlen, aset, false);
				}
			}
		} catch (Exception t) {
			t.printStackTrace();
		}
	}

	private static AttributeSet getParen(int i) {
		i++;
		if (i<0) i=0;
		return PARENS[i%PARENS.length];
	}

}