org.aya.concrete.visitor.ExprConsumer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of base Show documentation
The Aya proof assistant
There is a newer version: 0.33.0
// Copyright (c) 2020-2021 Yinsen (Tesla) Zhang.
// Use of this source code is governed by the MIT license that can be found in the LICENSE.md file.
package org.aya.concrete.visitor;

import kala.collection.immutable.ImmutableSeq;
import kala.tuple.Unit;
import org.aya.concrete.Expr;
import org.jetbrains.annotations.NotNull;

public interface ExprConsumer extends Expr.Visitor {
  @Override default Unit visitRef(Expr.@NotNull RefExpr expr, P p) {
    return Unit.unit();
  }

  @Override default Unit visitUnresolved(Expr.@NotNull UnresolvedExpr expr, P p) {
    return Unit.unit();
  }

  @Override default Unit visitError(Expr.@NotNull ErrorExpr error, P p) {
    return Unit.unit();
  }

  @Override default Unit visitHole(Expr.@NotNull HoleExpr holeExpr, P p) {
    var expr = holeExpr.filling();
    if (expr != null) expr.accept(this, p);
    return Unit.unit();
  }

  @Override default Unit visitBinOpSeq(Expr.@NotNull BinOpSeq binOpSeq, P p) {
    binOpSeq.seq().forEach(e -> e.expr().accept(this, p));
    return Unit.unit();
  }

  @Override default Unit visitNew(Expr.@NotNull NewExpr expr, P p) {
    expr.fields().forEach(e -> e.body().accept(this, p));
    return expr.struct().accept(this, p);
  }

  @Override default Unit visitUniv(Expr.@NotNull UnivExpr expr, P p) {
    return Unit.unit();
  }

  @Override default Unit visitRawUniv(Expr.@NotNull RawUnivExpr expr, P p) {
    return Unit.unit();
  }

  @Override default Unit visitRawUnivArgs(Expr.@NotNull RawUnivArgsExpr expr, P p) {
    expr.univArgs().forEach(e -> e.accept(this, p));
    return Unit.unit();
  }

  @Override default Unit visitUnivArgs(Expr.@NotNull UnivArgsExpr expr, P p) {
    return Unit.unit();
  }

  @Override default Unit visitApp(Expr.@NotNull AppExpr expr, P p) {
    expr.argument().expr().accept(this, p);
    return expr.function().accept(this, p);
  }

  default void visitParams(@NotNull ImmutableSeq params, P p) {
    params.forEach(param -> {
      if (param.type() != null) param.type().accept(this, p);
    });
  }

  @Override default Unit visitLam(Expr.@NotNull LamExpr expr, P p) {
    visitParams(ImmutableSeq.of(expr.param()), p);
    return expr.body().accept(this, p);
  }

  @Override default Unit visitPi(Expr.@NotNull PiExpr expr, P p) {
    visitParams(ImmutableSeq.of(expr.param()), p);
    return expr.last().accept(this, p);
  }

  @Override default Unit visitSigma(Expr.@NotNull SigmaExpr expr, P p) {
    visitParams(expr.params(), p);
    return Unit.unit();
  }

  @Override default Unit visitTup(Expr.@NotNull TupExpr expr, P p) {
    expr.items().forEach(item -> item.accept(this, p));
    return Unit.unit();
  }

  @Override default Unit visitProj(Expr.@NotNull ProjExpr expr, P p) {
    return expr.tup().accept(this, p);
  }

  @Override default Unit visitLitInt(Expr.@NotNull LitIntExpr expr, P p) {
    return Unit.unit();
  }

  @Override default Unit visitLmax(Expr.@NotNull LMaxExpr expr, P p) {
    for (var level : expr.levels()) level.accept(this, p);
    return Unit.unit();
  }

  @Override default Unit visitLsuc(Expr.@NotNull LSucExpr expr, P p) {
    return expr.expr().accept(this, p);
  }

  @Override default Unit visitLitString(Expr.@NotNull LitStringExpr expr, P p) {
    return Unit.unit();
  }
}