com.sri.ai.praise.sgsolver.demo.editor.HOGMCodeArea Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aic-praise Show documentation
SRI International's AIC PRAiSE (Probabilistic Reasoning As Symbolic Evaluation) Library (for Java 1.8+)
There is a newer version: 1.3.2
/*
 * Copyright (c) 2015, SRI International
 * All rights reserved.
 * Licensed under the The BSD 3-Clause License;
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 * 
 * http://opensource.org/licenses/BSD-3-Clause
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 
 * Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 * 
 * Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * 
 * Neither the name of the aic-praise nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, 
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 * OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package com.sri.ai.praise.sgsolver.demo.editor;

import org.antlr.v4.runtime.ANTLRInputStream;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.Token;
import org.fxmisc.richtext.CodeArea;
import org.fxmisc.richtext.LineNumberFactory;
import org.fxmisc.richtext.StyleSpans;
import org.fxmisc.richtext.StyleSpansBuilder;
import org.fxmisc.undo.UndoManager;

import com.google.common.annotations.Beta;
import com.sri.ai.praise.model.v1.hogm.antlr.HOGMTerminalSymbols;
import com.sri.ai.praise.sgsolver.demo.FXUtil;
import com.sri.ai.praise.model.v1.hogm.antlr.HOGMLexer;

import javafx.scene.layout.AnchorPane;

import java.util.Collection;
import java.util.Collections;

@Beta
public class HOGMCodeArea extends AnchorPane {
	private CodeArea codeArea;
	
	public HOGMCodeArea() {
		this(true);
	}
	
	public HOGMCodeArea(boolean showLineNumbers) {
		super();
		initialize(showLineNumbers);
	}
	
	public void setText(String text) {
		if (getText() != null && !getText().equals(text)) {
			codeArea.replaceText(text);
			codeArea.selectRange(0, 0);
			codeArea.getUndoManager().forgetHistory();
		}
	}
	
	public String getText() {
		return codeArea.getText();
	}
	
	public UndoManager getUndoManager() {
		return codeArea.getUndoManager();
	}
	
	public void undo() {
		codeArea.undo();
	}
	
	public void redo() {
		codeArea.redo();
	}
	
	public void setEditable(boolean value) {
		codeArea.setEditable(value);
	}
	
	public void highlight(int startIdx, int endIdx) {
		int s = startIdx;// > 0 ? startIdx -1 : startIdx;
		int e = endIdx;//  < (getText().length()-1) ? endIdx+1 : endIdx;
		if (s <= e && s >= 0 && e < getText().length()) {
			codeArea.selectRange(s, e);
		}
	}
	
	public void setFocus() {		
		codeArea.requestFocus();
	}
	
	//
	// PRIVATE
	//
	private void initialize(boolean showLineNumbers) {		
		codeArea = new CodeArea();
		if (showLineNumbers) {
			codeArea.setParagraphGraphicFactory(LineNumberFactory.get(codeArea, digits -> "%"+ (digits < 2 ? 2 : digits) + "d"));
		}
		
		codeArea.textProperty().addListener((obs, oldText, newText) -> {
			codeArea.setStyleSpans(0, computeHighlighting(newText));
		});	
		
		FXUtil.anchor(codeArea);		
		getChildren().add(codeArea);
	}
		 
	private static StyleSpans> computeHighlighting(String text) {
		StyleSpansBuilder> spansBuilder = new StyleSpansBuilder<>();
		int lastTokenEnd = 0;
		ANTLRInputStream input = new ANTLRInputStream(text);
		HOGMLexer lexer = new HOGMLexer(input);
		CommonTokenStream tokens = new CommonTokenStream(lexer);
		tokens.fill();	
		for (int i = 0; i < tokens.size(); i++) {
			Token t = tokens.get(i);
			if (t.getType() == Token.EOF) {
				break;
			}			
			String styleClass;
			if (t.getType() == HOGMLexer.COMMENT || t.getType() == HOGMLexer.LINE_COMMENT) {
				styleClass = "hogmCodeComment";
			}
			else if (HOGMTerminalSymbols.isTerminalSymbol(t.getText())) {
				styleClass = "hogmCodeKeyword";
			}
			else {
				styleClass = "hogmCodeOther";
			}
			int spacing = t.getStartIndex() - lastTokenEnd;
			if (spacing > 0) {			
				spansBuilder.add(Collections.emptyList(), spacing);
			}
			int stylesize = (t.getStopIndex() - t.getStartIndex()) +1;			
			spansBuilder.add(Collections.singleton(styleClass), stylesize);
			lastTokenEnd = t.getStopIndex()+1;
		}
		
		return spansBuilder.create();
	}
}