com.google.gwt.reflect.rebind.generators.ReflectionGeneratorContext Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of gwt-user Show documentation
The newest version!
package com.google.gwt.reflect.rebind.generators;

import com.google.gwt.core.ext.TreeLogger;
import com.google.gwt.core.ext.UnableToCompleteException;
import com.google.gwt.core.ext.typeinfo.TypeOracle;
import com.google.gwt.dev.javac.StandardGeneratorContext;
import com.google.gwt.dev.jjs.UnifyAstView;
import com.google.gwt.dev.jjs.ast.Context;
import com.google.gwt.dev.jjs.ast.JClassLiteral;
import com.google.gwt.dev.jjs.ast.JMethod;
import com.google.gwt.dev.jjs.ast.JMethodCall;
import com.google.gwt.thirdparty.xapi.dev.source.SourceBuilder;

import java.io.PrintWriter;
import java.lang.reflect.Modifier;
import java.util.HashMap;
import java.util.Map;

public class ReflectionGeneratorContext {

  private final UnifyAstView ast;
  private final JClassLiteral clazz;
  private final Context context;
  private final JMethod enclosingMethod;
  private final TreeLogger logger;
  private final JMethodCall methodCall;
  private final Map> builders;
  private ConstPoolGenerator constPool;

  public ReflectionGeneratorContext(final TreeLogger logger, final JClassLiteral clazz,
    final JMethodCall methodCall, final JMethod enclosingMethod, final Context context, final UnifyAstView ast) {
    this.ast = ast;
    this.clazz = clazz;
    this.context = context;
    this.enclosingMethod = enclosingMethod;
    this.logger = logger;
    this.methodCall = methodCall;
    builders = new HashMap>();
  }

  /**
   * @return the ast
   */
  public UnifyAstView getAst() {
    return ast;
  }

  /**
   * @return the clazz
   */
  public JClassLiteral getClazz() {
    return clazz;
  }

  /**
   * @return the context
   */
  public Context getContext() {
    return context;
  }

  /**
   * @return the currentMethod
   */
  public JMethod getCurrentMethod() {
    return enclosingMethod;
  }

  /**
   * @return the logger
   */
  public TreeLogger getLogger() {
    return logger;
  }

  /**
   * @return the methodCall
   */
  public JMethodCall getMethodCall() {
    return methodCall;
  }

  public StandardGeneratorContext getGeneratorContext() {
    return getAst().getGeneratorContext();
  }

  public TypeOracle getTypeOracle() {
    return getGeneratorContext().getTypeOracle();
  }

  public SourceBuilder tryCreate(final int modifier, final String pkg, final String clsName) {
    final String fqcn = pkg + "." + clsName;
    SourceBuilder builder = builders.get(fqcn);
    if (builder == null) {
      final StandardGeneratorContext ctx = getGeneratorContext();
      final PrintWriter writer = ctx.tryCreate(getLogger(), pkg, clsName);
      if (writer == null) {
        return null;
      }
      builder = new SourceBuilder
          (Modifier.toString(modifier)+" class "+clsName);
      builder.setPackage(pkg);
      builder.setPayload(writer);
      builders.put(fqcn, builder);
    }
    return builder;
  }

  public void commit(TreeLogger logger) {
    if (logger == null) {
      logger = this.logger;
    }
    final StandardGeneratorContext ctx = getGeneratorContext();
    for (final SourceBuilder builder : builders.values()) {
      final PrintWriter pw = builder.getPayload();
      pw.print(builder.toString());
      ctx.commit(logger, pw);
    }
    builders.clear();
    if (constPool != null) {
      constPool.commit(logger, ctx);
      constPool = null;
    }
  }

  public ConstPoolGenerator getConstPool() {
    if (constPool == null) {
      constPool = ConstPoolGenerator.getGenerator();
    }
    return constPool;
  }

  public void finish(final TreeLogger logger) throws UnableToCompleteException {
    getAst().getGeneratorContext().finish(logger);
  }


}