Class FunctionCall

UnLuac/src/unluac/decompile/expression/FunctionCall.java:9–104

import unluac.decompile.Walker;

public class FunctionCall extends Expression {

  private final Expression function;
  private final Expression[] arguments;
  private final boolean multiple;
  
  public FunctionCall(Expression function, Expression[] arguments, boolean multiple) {
    super(PRECEDENCE_ATOMIC);
    this.function = function;
    this.arguments = arguments;
    this.multiple = multiple;
  }

  @Override
  public void walk(Walker w) {
    w.visitExpression(this);
    function.walk(w);
    for(Expression expression : arguments) {
      expression.walk(w);
    }
  }
  
  @Override
  public int getConstantIndex() {
    int index = function.getConstantIndex();
    for(Expression argument : arguments) {
      index = Math.max(argument.getConstantIndex(), index);
    }
    return index;
  }
  
  @Override
  public boolean isMultiple() {
    return multiple;
  }
  
  @Override
  public void printMultiple(Decompiler d, Output out) {
    if(!multiple) {
      out.print("(");
    }
    print(d, out);
    if(!multiple) {
      out.print(")");
    }
  }
  
  private boolean isMethodCall() {
    return function.isMemberAccess() && arguments.length > 0 && function.getTable() == arguments[0];
  }
  
  @Override
  public boolean beginsWithParen() {
    if(isMethodCall()) {
      Expression obj = function.getTable();
      return obj.isUngrouped() || obj.beginsWithParen();
    } else {
      return function.isUngrouped() || function.beginsWithParen();