org.aya.resolve.StmtResolvers 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.36.0

// Copyright (c) 2020-2024 Tesla (Yinsen) Zhang.
// Use of this source code is governed by the MIT license that can be found in the LICENSE.md file.
package org.aya.resolve;

import kala.collection.immutable.ImmutableSeq;
import org.aya.resolve.module.ModuleLoader;
import org.aya.resolve.salt.Desalt;
import org.aya.resolve.visitor.StmtBinder;
import org.aya.resolve.visitor.StmtPreResolver;
import org.aya.resolve.visitor.StmtResolver;
import org.aya.syntax.concrete.stmt.Stmt;
import org.jetbrains.annotations.NotNull;

public record StmtResolvers(@NotNull ModuleLoader loader, @NotNull ResolveInfo info) {
  private @NotNull ImmutableSeq fillContext(@NotNull ImmutableSeq stmts) {
    return new StmtPreResolver(loader, info).resolveStmt(stmts, info.thisModule());
  }

  private void resolveStmts(@NotNull ImmutableSeq stmts) {
    StmtResolver.resolveStmt(stmts, info);
  }

  private void resolveBind(@NotNull ImmutableSeq stmts) {
    var binder = new StmtBinder(info);
    binder.resolveBind(stmts);
    info.opRename().forEach((var, rename) ->
      binder.bind(rename.bindCtx(), rename.bind(), var));
  }

  public void desugar(@NotNull ImmutableSeq stmts) {
    var salt = new Desalt(info);
    stmts.forEach(stmt -> stmt.descentInPlace(salt, salt.pattern));
  }

  /**
   * Resolve file level {@link Stmt}s
   */
  public void resolve(@NotNull ImmutableSeq stmts) {
    var resolving = fillContext(stmts);
    resolveStmts(resolving); // resolve mutates stmts
    resolveBind(resolving); // mutates bind blocks
    info.opSet().reportIfCyclic();
  }
}