eu.stratosphere.meteor.TraceableSopremoTreeAdaptor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of meteor-meteor Show documentation

The newest version!

package eu.stratosphere.meteor;

import java.util.IdentityHashMap;
import java.util.Map;

public class TraceableSopremoTreeAdaptor extends SopremoTreeAdaptor {
	private final Map instantiationInfos = new IdentityHashMap();

	public void addJavaFragment(final Object result, final StringBuilder builder) {
		final Object[] params = this.instantiationInfos.get(result);

		builder.append("new ").append(result.getClass().getSimpleName()).append("(");
		if (params == null)
			builder.append(result);
		else
			for (int index = 0; index < params.length; index++) {
				if (index > 0) {
					builder.append(",");
					builder.append("\n");
				}
				this.addJavaFragment(params[index], builder);
			}
		builder.append(")");
	}

	@Override
	protected Object instantiate(final Class expressionClass, final Object[] params) {
		final Object result = super.instantiate(expressionClass, params);
		this.instantiationInfos.put(result, params);
		return result;
	}

}