001: /*
002: $Id: NodePrinter.java 2355 2005-06-18 07:49:15Z cstein $
003:
004: Copyright 2003 (C) James Strachan and Bob Mcwhirter. All Rights Reserved.
005:
006: Redistribution and use of this software and associated documentation
007: ("Software"), with or without modification, are permitted provided
008: that the following conditions are met:
009:
010: 1. Redistributions of source code must retain copyright
011: statements and notices. Redistributions must also contain a
012: copy of this document.
013:
014: 2. Redistributions in binary form must reproduce the
015: above copyright notice, this list of conditions and the
016: following disclaimer in the documentation and/or other
017: materials provided with the distribution.
018:
019: 3. The name "groovy" must not be used to endorse or promote
020: products derived from this Software without prior written
021: permission of The Codehaus. For written permission,
022: please contact info@codehaus.org.
023:
024: 4. Products derived from this Software may not be called "groovy"
025: nor may "groovy" appear in their names without prior written
026: permission of The Codehaus. "groovy" is a registered
027: trademark of The Codehaus.
028:
029: 5. Due credit should be given to The Codehaus -
030: http://groovy.codehaus.org/
031:
032: THIS SOFTWARE IS PROVIDED BY THE CODEHAUS AND CONTRIBUTORS
033: ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT
034: NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
035: FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
036: THE CODEHAUS OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
037: INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
038: (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
039: SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
040: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
041: STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
042: ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
043: OF THE POSSIBILITY OF SUCH DAMAGE.
044:
045: */
046: package groovy.util;
047:
048: import org.codehaus.groovy.runtime.InvokerHelper;
049:
050: import java.io.OutputStreamWriter;
051: import java.io.PrintWriter;
052: import java.util.Iterator;
053: import java.util.List;
054: import java.util.Map;
055:
056: /**
057: * A helper class for creating nested trees of data
058: *
059: * @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
060: * @author Christian Stein
061: * @version $Revision: 2355 $
062: */
063: public class NodePrinter {
064:
065: protected final IndentPrinter out;
066:
067: public NodePrinter() {
068: this (new IndentPrinter(new PrintWriter(new OutputStreamWriter(
069: System.out))));
070: }
071:
072: public NodePrinter(PrintWriter out) {
073: this (new IndentPrinter(out));
074: }
075:
076: public NodePrinter(IndentPrinter out) {
077: if (out == null) {
078: throw new NullPointerException(
079: "IndentPrinter 'out' must not be null!");
080: }
081: this .out = out;
082: }
083:
084: public void print(Node node) {
085: out.printIndent();
086: printName(node);
087: Map attributes = node.attributes();
088: boolean hasAttributes = attributes != null
089: && !attributes.isEmpty();
090: if (hasAttributes) {
091: printAttributes(attributes);
092: }
093: Object value = node.value();
094: if (value instanceof List) {
095: if (!hasAttributes) {
096: out.print("()");
097: }
098: printList((List) value);
099: } else {
100: if (value instanceof String) {
101: out.print("('");
102: out.print((String) value);
103: out.println("')");
104: } else {
105: out.println("()");
106: }
107: }
108: out.flush();
109: }
110:
111: protected void printName(Node node) {
112: Object name = node.name();
113: if (name != null) {
114: out.print(name.toString());
115: } else {
116: out.print("null");
117: }
118: }
119:
120: protected void printList(List list) {
121: if (list.isEmpty()) {
122: out.println("");
123: } else {
124: out.println(" {");
125: out.incrementIndent();
126: for (Iterator iter = list.iterator(); iter.hasNext();) {
127: Object value = iter.next();
128: if (value instanceof Node) {
129: print((Node) value);
130: } else {
131: out.printIndent();
132: out.print("builder.append(");
133: out.print(InvokerHelper.toString(value));
134: out.println(")");
135: }
136: }
137: out.decrementIndent();
138: out.printIndent();
139: out.println("}");
140: }
141: }
142:
143: protected void printAttributes(Map attributes) {
144: out.print("(");
145: boolean first = true;
146: for (Iterator iter = attributes.entrySet().iterator(); iter
147: .hasNext();) {
148: Map.Entry entry = (Map.Entry) iter.next();
149: if (first) {
150: first = false;
151: } else {
152: out.print(", ");
153: }
154: out.print(entry.getKey().toString());
155: out.print(":");
156: if (entry.getValue() instanceof String) {
157: out.print("'" + entry.getValue() + "'");
158: } else {
159: out.print(InvokerHelper.toString(entry.getValue()));
160: }
161: }
162: out.print(")");
163: }
164:
165: }
|