package org.stringtemplate.v4.gui;
import org.antlr.runtime.tree.CommonTree;
import org.stringtemplate.v4.InstanceScope;
import org.stringtemplate.v4.Interpreter;
import org.stringtemplate.v4.ST;
import org.stringtemplate.v4.StringRenderer;
import org.stringtemplate.v4.debug.AddAttributeEvent;
import javax.swing.event.TreeModelListener;
import javax.swing.tree.TreeModel;
import javax.swing.tree.TreePath;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
public class JTreeScopeStackModel implements TreeModel {
CommonTree root;
public static class StringTree extends CommonTree {
String text;
public StringTree(String text) {this.text = text;}
@Override
public boolean isNil() {
return text==null;
}
@Override
public String toString() {
if ( !isNil() ) return text;
return "nil";
}
}
public JTreeScopeStackModel(InstanceScope scope) {
root = new StringTree("Scope stack:");
Set<String> names = new HashSet<String>();
List<InstanceScope> stack = Interpreter.getScopeStack(scope, false);
for (InstanceScope s : stack) {
StringTree templateNode = new StringTree(s.st.getName());
root.insertChild(0, templateNode);
addAttributeDescriptions(s.st, templateNode, names);
}
}
public void addAttributeDescriptions(ST st, StringTree node, Set<String> names) {
Map<String, Object> attrs = st.getAttributes();
if ( attrs==null ) return;
for (String a : attrs.keySet()) {
String descr;
if ( st.debugState!=null && st.debugState.addAttrEvents!=null ) {
List<AddAttributeEvent> events = st.debugState.addAttrEvents.get(a);
StringBuilder locations = new StringBuilder();
int i = 0;
if ( events!=null ) {
for (AddAttributeEvent ae : events) {
if ( i>0 ) locations.append(", ");
locations.append(ae.getFileName()+":"+ae.getLine());
i++;
}
}
if ( locations.length()>0 ) {
descr = a+" = "+attrs.get(a)+" @ "+locations.toString();
}
else {
descr = a + " = " +attrs.get(a);
}
}
else {
descr = a + " = " +attrs.get(a);
}
if (!names.add(a)) {
StringBuilder builder = new StringBuilder();
builder.append("<html><font color=\"gray\">");
builder.append(StringRenderer.escapeHTML(descr));
builder.append("</font></html>");
descr = builder.toString();
}
node.addChild( new StringTree(descr) );
}
}
@Override
public Object getRoot() {
return root;
}
@Override
public Object getChild(Object parent, int i) {
StringTree t = (StringTree)parent;
return t.getChild(i);
}
@Override
public int getChildCount(Object parent) {
StringTree t = (StringTree)parent;
return t.getChildCount();
}
@Override
public boolean isLeaf(Object node) {
return getChildCount(node) == 0;
}
@Override
public int getIndexOfChild(Object parent, Object child) {
StringTree c = (StringTree)child;
return c.getChildIndex();
}
@Override
public void valueForPathChanged(TreePath treePath, Object o) {
}
@Override
public void addTreeModelListener(TreeModelListener treeModelListener) {
}
@Override
public void removeTreeModelListener(TreeModelListener treeModelListener) {
}
}