org.aya.util.terck.Diagonal Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tools-kala Show documentation

The Aya proof assistant

There is a newer version: 0.36.0

// Copyright (c) 2020-2023 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.util.terck;

import kala.collection.immutable.ImmutableSeq;
import kala.collection.mutable.MutableList;
import kala.range.primitive.IntRange;
import org.aya.pretty.doc.Doc;
import org.aya.pretty.doc.Docile;
import org.jetbrains.annotations.NotNull;

public record Diagonal(
  @NotNull CallMatrix matrix,
  @NotNull ImmutableSeq diagonal
) implements Docile {
  public static  @NotNull Diagonal create(@NotNull CallMatrix matrix) {
    assert matrix.rows() == matrix.cols();
    var diag = IntRange.closedOpen(0, matrix.rows())
      .mapToObjTo(MutableList.create(), i -> matrix.matrix()[i][i])
      .toImmutableSeq();
    return new Diagonal<>(matrix, diag);
  }

  public @NotNull Doc toDoc() {
    return Doc.stickySep(diagonal.view().map(Relation::toDoc));
  }
}