org.aya.tyck.Jdg 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.tyck;

import kala.collection.SeqView;
import kala.value.LazyValue;
import org.aya.syntax.core.term.SortTerm;
import org.aya.syntax.core.term.Term;
import org.aya.syntax.ref.LocalVar;
import org.jetbrains.annotations.NotNull;

import java.util.function.UnaryOperator;

public sealed interface Jdg {
  @NotNull Term wellTyped();
  @NotNull Term type();

  default @NotNull Jdg bindTele(@NotNull SeqView vars) { return map(t -> t.bindTele(vars)); }

  /**
   * @apiNote the mapper {@param f} may not executed immediately, so make sure that {@param f} is pure.
   */
  @NotNull Jdg map(@NotNull UnaryOperator f);
  default Jdg lift(int lift) { return map(t -> t.elevate(lift)); }

  /**
   * {@link Default#type} is the type of {@link Default#wellTyped}.
   */
  record Default(@Override @NotNull Term wellTyped, @Override @NotNull Term type) implements Jdg {
    @Override public @NotNull Default map(@NotNull UnaryOperator f) {
      return new Default(f.apply(wellTyped), f.apply(type));
    }
  }

  record Sort(@Override @NotNull SortTerm wellTyped) implements Jdg {
    @Override public @NotNull SortTerm type() { return wellTyped.succ(); }
    @Override public @NotNull Jdg map(@NotNull UnaryOperator f) {
      return this;
    }
  }

  record Lazy(@Override @NotNull Term wellTyped, @NotNull LazyValue lazyType) implements Jdg {
    @Override public @NotNull Term type() { return lazyType.get(); }
    @Override public @NotNull Lazy map(@NotNull UnaryOperator f) {
      return new Lazy(f.apply(wellTyped), lazyType.map(f));
    }
  }
}