MechNetSemanticHighlightingCalculator.java

package com.robotoworks.mechanoid.net.ui.editor.syntaxcoloring;

import org.eclipse.xtext.nodemodel.INode;
import org.eclipse.xtext.nodemodel.impl.CompositeNodeWithSemanticElement;
import org.eclipse.xtext.resource.XtextResource;
import org.eclipse.xtext.ui.editor.syntaxcoloring.DefaultHighlightingConfiguration;
import org.eclipse.xtext.ui.editor.syntaxcoloring.IHighlightedPositionAcceptor;
import org.eclipse.xtext.ui.editor.syntaxcoloring.ISemanticHighlightingCalculator;

import com.robotoworks.mechanoid.net.netModel.Path;

public class MechNetSemanticHighlightingCalculator implements
		ISemanticHighlightingCalculator {

	public void provideHighlightingFor(XtextResource resource,
			IHighlightedPositionAcceptor acceptor) {
		if (resource == null || resource.getParseResult() == null)
		    return;
		    
		  INode root = resource.getParseResult().getRootNode();
		  for (INode node : root.getAsTreeIterable()) {
		    if (node instanceof CompositeNodeWithSemanticElement) {
		    	
		    	CompositeNodeWithSemanticElement composite = (CompositeNodeWithSemanticElement) node;
		    	if(composite.getSemanticElement() instanceof Path) {
		    	
			      acceptor.addPosition(node.getOffset(), node.getLength(), 
			    		  DefaultHighlightingConfiguration.STRING_ID);	
		    	}
		    }
		  }
	}
}