Ejemplo n.º 1
0
  public MethodStmtCompiler(ClassStmtCompiler clazz, MethodStmtToken statement) {
    super(clazz.getCompiler());
    this.clazz = clazz;
    this.statement = statement;
    this.node = new MethodNodeImpl();

    this.localVariables = new LinkedHashMap<>();

    entity = new MethodEntity(getCompiler().getContext());
    entity.setClazz(clazz.entity);
    if (statement != null) entity.setName(statement.getName().getName());

    realName = entity.getName();
    methodId = compiler.getScope().nextMethodIndex();
  }
Ejemplo n.º 2
0
  @Override
  public MethodEntity compile() {
    if (statement != null) {
      if (external) statement.setDynamicLocal(true);

      if (statement.getDocComment() != null)
        entity.setDocComment(new DocumentComment(statement.getDocComment().getComment()));

      entity.setAbstract(statement.isAbstract());
      entity.setAbstractable(statement.getBody() == null);
      entity.setFinal(statement.isFinal());
      entity.setStatic(statement.isStatic());
      entity.setModifier(statement.getModifier());
      entity.setReturnReference(statement.isReturnReference());
      entity.setTrace(statement.toTraceInfo(compiler.getContext()));
      entity.setImmutable(statement.getArguments().isEmpty());
      entity.setGeneratorEntity(generatorEntity);

      if (clazz.isSystem()) entity.setInternalName(entity.getName());
      else entity.setInternalName(entity.getName() + "$" + clazz.entity.nextMethodIndex());

      ParameterEntity[] parameters = new ParameterEntity[statement.getArguments().size()];
      int i = 0;
      for (ArgumentStmtToken argument : statement.getArguments()) {
        parameters[i] = new ParameterEntity(compiler.getContext());
        ParameterEntity parameter = parameters[i];

        parameter.setReference(argument.isReference());
        parameter.setName(argument.getName().getName());
        parameter.setTrace(argument.toTraceInfo(compiler.getContext()));

        parameter.setMutable(
            statement.isDynamicLocal() || statement.variable(argument.getName()).isMutable());

        parameter.setUsed(!statement.isUnusedVariable(argument.getName()));

        parameter.setVariadic(argument.isVariadic());

        parameter.setType(argument.getHintType());

        if (argument.getHintTypeClass() != null) {
          parameter.setTypeClass(argument.getHintTypeClass().getName());
        }

        ExpressionStmtCompiler expressionStmtCompiler = new ExpressionStmtCompiler(compiler);
        ExprStmtToken value = argument.getValue();

        if (value != null) {
          Memory defaultValue = expressionStmtCompiler.writeExpression(value, true, true, false);

          // try detect constant
          if (value.isSingle()) {
            if (value.getSingle() instanceof NameToken) {
              parameter.setDefaultValueConstName(((NameToken) value.getSingle()).getName());
              if (defaultValue == null) {
                defaultValue = (new ConstantMemory(((NameToken) value.getSingle()).getName()));
                parameter.setMutable(true);
              }
            } else if (value.getSingle() instanceof StaticAccessExprToken) {
              StaticAccessExprToken access = (StaticAccessExprToken) value.getSingle();

              if (access.getClazz() instanceof NameToken
                  && access.getField() instanceof NameToken) {
                if (defaultValue == null)
                  defaultValue =
                      (new ClassConstantMemory(
                          ((NameToken) access.getClazz()).getName(),
                          ((NameToken) access.getField()).getName()));

                parameter.setDefaultValueConstName(
                    ((NameToken) access.getClazz()).getName()
                        + "::"
                        + ((NameToken) access.getField()).getName());
                parameter.setMutable(true);
              }
            }
          }

          if (defaultValue == null)
            compiler
                .getEnvironment()
                .error(
                    argument.toTraceInfo(compiler.getContext()),
                    ErrorType.E_COMPILE_ERROR,
                    Messages.ERR_EXPECTED_CONST_VALUE,
                    "$" + argument.getName().getName());

          parameter.setDefaultValue(defaultValue);
        }
        i++;
      }

      entity.setParameters(parameters);
    }

    if (statement != null && clazz.statement.isInterface()) {
      if (!statement.isInterfacable()) {
        compiler
            .getEnvironment()
            .error(
                entity.getTrace(),
                Messages.ERR_INTERFACE_FUNCTION_CANNOT_CONTAIN_BODY.fetch(
                    entity.getSignatureString(false)));
      }
      if (statement.isAbstract() || statement.isFinal()) {
        compiler
            .getEnvironment()
            .error(
                entity.getTrace(),
                Messages.ERR_ACCESS_TYPE_FOR_INTERFACE_METHOD.fetch(
                    entity.getSignatureString(false)));
      }
    } else {
      writeHeader();

      if (statement.isGenerator()) {
        entity.setEmpty(false);
        entity.setImmutable(false);
        GeneratorStmtCompiler generatorStmtCompiler =
            new GeneratorStmtCompiler(compiler, statement);
        entity.setGeneratorEntity(generatorStmtCompiler.compile());

        ExpressionStmtCompiler expr = new ExpressionStmtCompiler(this, null);

        expr.makeUnknown(new TypeInsnNode(NEW, entity.getGeneratorEntity().getInternalName()));
        expr.stackPush(Memory.Type.REFERENCE);
        expr.writePushDup();

        // env
        expr.writePushEnv();

        // classEntity
        expr.writePushDup();
        expr.writePushConstString(compiler.getModule().getInternalName());
        expr.writePushConstInt((int) entity.getGeneratorEntity().getId());
        expr.writeSysDynamicCall(
            Environment.class, "__getGenerator", ClassEntity.class, String.class, Integer.TYPE);

        // self
        expr.writePushThis();

        // uses
        expr.writeVarLoad("~args");

        expr.writeSysCall(
            entity.getGeneratorEntity().getInternalName(),
            INVOKESPECIAL,
            Constants.INIT_METHOD,
            void.class,
            Environment.class,
            ClassEntity.class,
            Memory.class,
            Memory[].class);
        expr.writeSysStaticCall(ObjectMemory.class, "valueOf", Memory.class, IObject.class);

        expr.makeUnknown(new InsnNode(Opcodes.ARETURN));
        expr.stackPop();
      } else {
        ExpressionStmtCompiler expr = new ExpressionStmtCompiler(this, null);

        entity.setEmpty(true);
        if (statement != null && statement.getBody() != null) {
          expr.writeDefineVariables(statement.getLocal());
          expr.write(statement.getBody());

          if (!statement.getBody().getInstructions().isEmpty()) {
            entity.setEmpty(false);
          }
        }

        if (generatorEntity != null) {
          expr.writeVarLoad("~this");
          expr.writePushConstBoolean(false);
          expr.writeSysDynamicCall(null, "_setValid", void.class, Boolean.TYPE);
        }

        ReturnStmtToken token = new ReturnStmtToken(new TokenMeta("", 0, 0, 0, 0));
        token.setValue(null);
        expr.getCompiler(ReturnStmtToken.class).write(token);
      }

      writeFooter();
    }
    return entity;
  }
Ejemplo n.º 3
0
  void writeHeader() {
    int access = 0;
    if (statement != null) {
      if (compiler.getScope().isDebugMode()) {
        statement.setDynamicLocal(true);
      }

      switch (statement.getModifier()) {
        case PRIVATE:
          access += Opcodes.ACC_PRIVATE;
          break;
        case PROTECTED:
          access += Opcodes.ACC_PROTECTED;
          break;
        case PUBLIC:
          access += Opcodes.ACC_PUBLIC;
          break;
      }

      if (statement.isStatic()) access += Opcodes.ACC_STATIC;
      // if (statement.isAbstract()) access += Opcodes.ACC_ABSTRACT;
      if (statement.isFinal()) access += Opcodes.ACC_FINAL;

      node.access = access;
      node.name =
          clazz.isSystem() || entity == null
              ? statement.getName().getName()
              : entity.getInternalName();
      node.desc =
          Type.getMethodDescriptor(
              Type.getType(Memory.class),
              Type.getType(Environment.class),
              Type.getType(Memory[].class));

      if (external) {
        node.desc =
            Type.getMethodDescriptor(
                Type.getType(Memory.class),
                Type.getType(Environment.class),
                Type.getType(Memory[].class),
                Type.getType(ArrayMemory.class));
      }
    }

    if (statement != null) {
      LabelNode label = labelStart = writeLabel(node, statement.getMeta().getStartLine());

      if (!statement.isStatic()) addLocalVariable("~this", label, Object.class);

      ExpressionStmtCompiler expressionCompiler = new ExpressionStmtCompiler(this, null);

      addLocalVariable("~env", label, Environment.class); // Environment env
      LocalVariable args = addLocalVariable("~args", label, Memory[].class); // Memory[] arguments

      if (statement.isDynamicLocal()) {
        if (external) addLocalVariable("~passedLocal", label, ArrayMemory.class);

        LocalVariable local = addLocalVariable("~local", label, ArrayMemory.class);

        if (external) {
          expressionCompiler.writeVarLoad("~passedLocal");
          expressionCompiler.writeSysStaticCall(
              ArrayMemory.class, "valueOfRef", ArrayMemory.class, ArrayMemory.class);
          expressionCompiler.setStackPeekAsImmutable();
          expressionCompiler.writeVarStore(local, false, true);
        } else {
          expressionCompiler.writePushConstNull();
          expressionCompiler.writeSysStaticCall(
              ArrayMemory.class, "valueOfRef", ArrayMemory.class, ArrayMemory.class);
          expressionCompiler.setStackPeekAsImmutable();
          expressionCompiler.writeVarStore(local, false, true);
        }
      }

      if (statement.getUses() != null && !statement.getUses().isEmpty()) {
        int i = 0;
        expressionCompiler.writeVarLoad("~this");
        expressionCompiler.writeGetDynamic("uses", Memory[].class);

        for (ArgumentStmtToken argument : statement.getUses()) {
          LocalVariable local;
          if (statement.isDynamicLocal()) {
            expressionCompiler.writeDefineVariable(argument.getName());
            local = getLocalVariable(argument.getName().getName());
          } else {
            local = addLocalVariable(argument.getName().getName(), label, Memory.class);
          }

          if (argument.isReference()) {
            local.setReference(true);
            statement.variable(argument.getName()).setUnstable(true);
          }

          expressionCompiler.writePushDup();
          expressionCompiler.writePushGetFromArray(i, Memory.class);

          if (statement.isDynamicLocal()) {
            expressionCompiler.writeVarAssign(local, argument.getName(), false, false);
          } else {
            expressionCompiler.writeVarStore(local, false, false);
          }

          local.pushLevel();
          i++;
        }
        expressionCompiler.writePopAll(1);
      }

      int i = 0;

      for (ArgumentStmtToken argument : statement.getArguments()) {
        if (argument.isReference()) {
          statement.variable(argument.getName()).setReference(true).setUnstable(true);
        }

        LabelNode next = new LabelNode();

        expressionCompiler.writeDefineVariable(argument.getName());
        LocalVariable local = getLocalVariable(argument.getName().getName());

        if (local != null) {
          expressionCompiler.writeVarLoad(args);
          expressionCompiler.writePushGetFromArray(i, Memory.class);
          expressionCompiler.writeVarAssign(local, argument.getName(), true, false);

          // if length <= i then undefined
          node.instructions.add(new JumpInsnNode(Opcodes.IFNONNULL, next));
          expressionCompiler.stackPop();

          if (argument.getValue() == null) expressionCompiler.writePushNull();
          else expressionCompiler.writeExpression(argument.getValue(), true, false);

          expressionCompiler.writeVarAssign(local, argument.getName(), false, false);
          node.instructions.add(next);

          local.pushLevel();
        }

        i++;
      }
    } else {
      LabelNode label = labelStart = writeLabel(node, clazz.statement.getMeta().getStartLine());
    }
  }