4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenkpackage java.beans;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenkimport java.io.*;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenkimport java.util.*;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenkimport java.lang.reflect.*;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenkimport java.nio.charset.Charset;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenkimport java.nio.charset.CharsetEncoder;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenkimport java.nio.charset.IllegalCharsetNameException;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenkimport java.nio.charset.UnsupportedCharsetException;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenkpublic class XMLEncoder extends Encoder implements AutoCloseable {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private final CharsetEncoder encoder;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private final String charset;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private final boolean declaration;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private OutputStreamWriter out;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private Object owner;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private int indentation = 0;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private boolean internal = false;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private Map<Object, ValueData> valueToExpression;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private Map<Object, List<Statement>> targetToStatementList;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private boolean preambleWritten = false;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private NameGenerator nameGenerator;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private class ValueData {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk public int refs = 0;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk public boolean marked = false; // Marked -> refs > 0 unless ref was a target.

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk public String name = null;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk public Expression exp = null;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk /**

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk public XMLEncoder(OutputStream out) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk this(out, "UTF-8", true, 0);

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk /**

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk public XMLEncoder(OutputStream out, String charset, boolean declaration, int indentation) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk if (out == null) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk throw new IllegalArgumentException("the output stream cannot be null");

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk if (indentation < 0) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk throw new IllegalArgumentException("the indentation must be >= 0");

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk Charset cs = Charset.forName(charset);

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk this.encoder = cs.newEncoder();

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk this.charset = charset;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk this.declaration = declaration;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk this.indentation = indentation;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk this.out = new OutputStreamWriter(out, cs.newEncoder());

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk valueToExpression = new IdentityHashMap<Object, ValueData>();

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk targetToStatementList = new IdentityHashMap<Object, List<Statement>>();

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk nameGenerator = new NameGenerator();

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk /**

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk public void setOwner(Object owner) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk this.owner = owner;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk writeExpression(new Expression(this, "getOwner", new Object[0]));

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk /**

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk public Object getOwner() {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk return owner;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk /**

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk public void writeObject(Object o) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk if (internal) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk super.writeObject(o);

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk else {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk writeStatement(new Statement(this, "writeObject", new Object[]{o}));

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private List<Statement> statementList(Object target) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk List<Statement> list = targetToStatementList.get(target);

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk if (list == null) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk list = new ArrayList<Statement>();

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk targetToStatementList.put(target, list);

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk return list;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private void mark(Object o, boolean isArgument) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk if (o == null || o == this) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk return;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk ValueData d = getValueData(o);

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk Expression exp = d.exp;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk // Do not mark liternal strings. Other strings, which might,

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk // for example, come from resource bundles should still be marked.

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk if (o.getClass() == String.class && exp == null) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk return;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk // Bump the reference counts of all arguments

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk if (isArgument) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk d.refs++;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk if (d.marked) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk return;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk d.marked = true;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk Object target = exp.getTarget();

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk mark(exp);

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk if (!(target instanceof Class)) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk statementList(target).add(exp);

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk // Pending: Why does the reference count need to

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk // be incremented here?

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk d.refs++;

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk private void mark(Statement stm) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk Object[] args = stm.getArguments();

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk for (int i = 0; i < args.length; i++) {

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk Object arg = args[i];

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk mark(arg, true);

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk mark(stm.getTarget(), false);

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk }

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk /**

4b8d88eb610aa1e0bb6ec632f792744b3d6b5f22jeff.schenk public void writeStatement(Statement oldStm) {

// System.out.println("XMLEncoder::writeStatement: " + oldStm);

boolean internal = this.internal;

this.internal = true;

try {

super.writeStatement(oldStm);

/*

mark(oldStm);

Object target = oldStm.getTarget();

if (target instanceof Field) {

String method = oldStm.getMethodName();

Object[] args = oldStm.getArguments();

if ((method == null) || (args == null)) {

}

else if (method.equals("get") && (args.length == 1)) {

target = args[0];

}

else if (method.equals("set") && (args.length == 2)) {

target = args[0];

}

}

statementList(target).add(oldStm);

}

catch (Exception e) {

getExceptionListener().exceptionThrown(new Exception("XMLEncoder: discarding statement " + oldStm, e));

}

this.internal = internal;

}

/**

public void writeExpression(Expression oldExp) {

boolean internal = this.internal;

this.internal = true;

Object oldValue = getValue(oldExp);

if (get(oldValue) == null || (oldValue instanceof String && !internal)) {

getValueData(oldValue).exp = oldExp;

super.writeExpression(oldExp);

}

this.internal = internal;

}

/**

public void flush() {

if (!preambleWritten) { // Don't do this in constructor - it throws ... pending.

if (this.declaration) {

writeln("<?xml version=" + quote("1.0") +

" encoding=" + quote(this.charset) + "?>");

}

writeln("<java version=" + quote(System.getProperty("java.version")) +

" class=" + quote(XMLDecoder.class.getName()) + ">");

preambleWritten = true;

}

indentation++;

List<Statement> statements = statementList(this);

while (!statements.isEmpty()) {

Statement s = statements.remove(0);

if ("writeObject".equals(s.getMethodName())) {

outputValue(s.getArguments()[0], this, true);

}

else {

outputStatement(s, this, false);

}

}

indentation--;

try {

out.flush();

}

catch (IOException e) {

getExceptionListener().exceptionThrown(e);

}

clear();

}

void clear() {

super.clear();

nameGenerator.clear();

valueToExpression.clear();

targetToStatementList.clear();

}

/**

public void close() {

flush();

writeln("</java>");

try {

out.close();

}

catch (IOException e) {

getExceptionListener().exceptionThrown(e);

}

}

private String quote(String s) {

return "\"" + s + "\"";

}

private ValueData getValueData(Object o) {

ValueData d = valueToExpression.get(o);

if (d == null) {

d = new ValueData();

valueToExpression.put(o, d);

}

return d;

}

/**

private static boolean isValidCharCode(int code) {

return (0x0020 <= code && code <= 0xD7FF)

|| (0x000A == code)

|| (0x0009 == code)

|| (0x000D == code)

|| (0xE000 <= code && code <= 0xFFFD)

|| (0x10000 <= code && code <= 0x10ffff);

}

private void writeln(String exp) {

try {

StringBuilder sb = new StringBuilder();

for(int i = 0; i < indentation; i++) {

sb.append(' ');

}

sb.append(exp);

sb.append('\n');

this.out.write(sb.toString());

}

catch (IOException e) {

getExceptionListener().exceptionThrown(e);

}

}

private void outputValue(Object value, Object outer, boolean isArgument) {

if (value == null) {

writeln("<null/>");

return;

}

if (value instanceof Class) {

writeln("<class>" + ((Class)value).getName() + "</class>");

return;

}

ValueData d = getValueData(value);

if (d.exp != null) {

Object target = d.exp.getTarget();

String methodName = d.exp.getMethodName();

if (target == null || methodName == null) {

throw new NullPointerException((target == null ? "target" :

"methodName") + " should not be null");

}

if (target instanceof Field && methodName.equals("get")) {

Field f = (Field)target;

writeln("<object class=" + quote(f.getDeclaringClass().getName()) +

" field=" + quote(f.getName()) + "/>");

return;

}

Class primitiveType = primitiveTypeFor(value.getClass());

if (primitiveType != null && target == value.getClass() &&

methodName.equals("new")) {

String primitiveTypeName = primitiveType.getName();

// Make sure that character types are quoted correctly.

if (primitiveType == Character.TYPE) {

char code = ((Character) value).charValue();

if (!isValidCharCode(code)) {

writeln(createString(code));

return;

}

value = quoteCharCode(code);

if (value == null) {

value = Character.valueOf(code);

}

}

writeln("<" + primitiveTypeName + ">" + value + "</" +

primitiveTypeName + ">");

return;

}

} else if (value instanceof String) {

writeln(createString((String) value));

return;

}

if (d.name != null) {

if (isArgument) {

writeln("<object idref=" + quote(d.name) + "/>");

}

else {

outputXML("void", " idref=" + quote(d.name), value);

}

}

else if (d.exp != null) {

outputStatement(d.exp, outer, isArgument);

}

}

private static String quoteCharCode(int code) {

switch(code) {

case '&': return "&amp;";

case '<': return "&lt;";

case '>': return "&gt;";

case '"': return "&quot;";

case '\'': return "&apos;";

case '\r': return "&#13;";

default: return null;

}

}

private static String createString(int code) {

return "<char code=\"#" + Integer.toString(code, 16) + "\"/>";

}

private String createString(String string) {

StringBuilder sb = new StringBuilder();

sb.append("<string>");

int index = 0;

while (index < string.length()) {

int point = string.codePointAt(index);

int count = Character.charCount(point);

if (isValidCharCode(point) && this.encoder.canEncode(string.substring(index, index + count))) {

String value = quoteCharCode(point);

if (value != null) {

sb.append(value);

} else {

sb.appendCodePoint(point);

}

index += count;

} else {

sb.append(createString(string.charAt(index)));

index++;

}

}

sb.append("</string>");

return sb.toString();

}

private void outputStatement(Statement exp, Object outer, boolean isArgument) {

Object target = exp.getTarget();

String methodName = exp.getMethodName();

if (target == null || methodName == null) {

throw new NullPointerException((target == null ? "target" :

"methodName") + " should not be null");

}

Object[] args = exp.getArguments();

boolean expression = exp.getClass() == Expression.class;

Object value = (expression) ? getValue((Expression)exp) : null;

String tag = (expression && isArgument) ? "object" : "void";

String attributes = "";

ValueData d = getValueData(value);

// Special cases for targets.

if (target == outer) {

}

else if (target == Array.class && methodName.equals("newInstance")) {

tag = "array";

attributes = attributes + " class=" + quote(((Class)args[0]).getName());

attributes = attributes + " length=" + quote(args[1].toString());

args = new Object[]{};

}

else if (target.getClass() == Class.class) {

attributes = attributes + " class=" + quote(((Class)target).getName());

}

else {

d.refs = 2;

if (d.name == null) {

getValueData(target).refs++;

List<Statement> statements = statementList(target);

if (!statements.contains(exp)) {

statements.add(exp);

}

outputValue(target, outer, false);

}

if (expression) {

outputValue(value, outer, isArgument);

}

return;

}

if (expression && (d.refs > 1)) {

String instanceName = nameGenerator.instanceName(value);

d.name = instanceName;

attributes = attributes + " id=" + quote(instanceName);

}

// Special cases for methods.

if ((!expression && methodName.equals("set") && args.length == 2 &&

args[0] instanceof Integer) ||

(expression && methodName.equals("get") && args.length == 1 &&

args[0] instanceof Integer)) {

attributes = attributes + " index=" + quote(args[0].toString());

args = (args.length == 1) ? new Object[]{} : new Object[]{args[1]};

}

else if ((!expression && methodName.startsWith("set") && args.length == 1) ||

(expression && methodName.startsWith("get") && args.length == 0)) {

if (3 < methodName.length()) {

attributes = attributes + " property=" +

quote(Introspector.decapitalize(methodName.substring(3)));

}

}

else if (!methodName.equals("new") && !methodName.equals("newInstance")) {

attributes = attributes + " method=" + quote(methodName);

}

outputXML(tag, attributes, value, args);

}

private void outputXML(String tag, String attributes, Object value, Object... args) {

List<Statement> statements = statementList(value);

// Use XML's short form when there is no body.

if (args.length == 0 && statements.size() == 0) {

writeln("<" + tag + attributes + "/>");

return;

}

writeln("<" + tag + attributes + ">");

indentation++;

for(int i = 0; i < args.length; i++) {

outputValue(args[i], null, true);

}

while (!statements.isEmpty()) {

Statement s = statements.remove(0);

outputStatement(s, value, false);

}

indentation--;

writeln("</" + tag + ">");

}

@SuppressWarnings("rawtypes")

static Class primitiveTypeFor(Class wrapper) {

if (wrapper == Boolean.class) return Boolean.TYPE;

if (wrapper == Byte.class) return Byte.TYPE;

if (wrapper == Character.class) return Character.TYPE;

if (wrapper == Short.class) return Short.TYPE;

if (wrapper == Integer.class) return Integer.TYPE;

if (wrapper == Long.class) return Long.TYPE;

if (wrapper == Float.class) return Float.TYPE;

if (wrapper == Double.class) return Double.TYPE;

if (wrapper == Void.class) return Void.TYPE;

return null;

}

}