/* * [The "BSD license"] * Copyright (c) 2011 Terence Parr * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. 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. * 3. The name of the author may not be used to endorse or promote products * derived from this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 AUTHOR 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 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; /** From a scope, get stack of enclosing scopes in order from root down * to scope. Then show each scope's (ST's) attributes as children. */ 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); } //System.out.println(root.toStringTree()); } 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) { } }