org.aya.tyck.order.SigRefFinder Maven / Gradle / Ivy
// 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.tyck.order;
import kala.collection.immutable.ImmutableSeq;
import kala.collection.mutable.DynamicSeq;
import kala.tuple.Unit;
import org.aya.api.ref.DefVar;
import org.aya.concrete.Expr;
import org.aya.concrete.remark.Remark;
import org.aya.concrete.stmt.Command;
import org.aya.concrete.stmt.Decl;
import org.aya.concrete.stmt.Generalize;
import org.aya.concrete.stmt.Sample;
import org.aya.concrete.visitor.ExprConsumer;
import org.jetbrains.annotations.NotNull;
/**
* Concrete version of RefFinder but only header is searched.
*
* @author kiva
* @see org.aya.core.visitor.RefFinder
*/
public class SigRefFinder implements ExprConsumer<@NotNull DynamicSeq> {
public static final @NotNull SigRefFinder HEADER_ONLY = new SigRefFinder();
private void decl(@NotNull DynamicSeq stmts, @NotNull Decl decl) {
tele(stmts, decl.telescope);
decl.result.accept(this, stmts);
}
public void visit(@NotNull TyckUnit sn, @NotNull DynamicSeq stmts) {
switch (sn) {
case Decl decl -> decl(stmts, decl);
case Decl.DataCtor ctor -> tele(stmts, ctor.telescope);
case Decl.StructField field -> {
tele(stmts, field.telescope);
field.result.accept(this, stmts);
}
case Command.Module module -> {}
case Command cmd -> {}
case Remark remark -> {
assert remark.literate != null;
remark.literate.visit(this, stmts);
}
case Sample sample -> visit(sample.delegate(), stmts);
case Generalize.Variables variables -> variables.type.accept(this, stmts);
case Generalize.Levels levels -> {}
}
}
private void tele(@NotNull DynamicSeq stmts, @NotNull ImmutableSeq telescope) {
telescope.mapNotNull(Expr.Param::type).forEach(type -> type.accept(this, stmts));
}
@Override public Unit visitRef(@NotNull Expr.RefExpr expr, @NotNull DynamicSeq stmts) {
if (expr.resolvedVar() instanceof DefVar defVar && defVar.concrete instanceof Decl decl)
stmts.append(decl);
return Unit.unit();
}
}