org.aya.compiler.ModuleSerializer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jit-compiler 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.compiler;

import kala.collection.immutable.ImmutableSeq;
import org.aya.primitive.ShapeFactory;
import org.aya.syntax.core.def.*;
import org.aya.syntax.ref.QPath;
import org.aya.util.IterableUtil;
import org.jetbrains.annotations.NotNull;

import static org.aya.compiler.NameSerializer.javifyClassName;

/**
 * Serializing a module, note that it may not a file module, so we need not to make importing.
 */
public final class ModuleSerializer extends AbstractSerializer {
  public record ModuleResult(
    @NotNull QPath name,
    @NotNull ImmutableSeq defs
  ) { }

  private final @NotNull ShapeFactory shapeFactory;

  public ModuleSerializer(@NotNull SourceBuilder builder, @NotNull ShapeFactory shapeFactory) {
    super(builder);
    this.shapeFactory = shapeFactory;
  }

  private void serializeCons(@NotNull DataDef dataDef, @NotNull SourceBuilder serializer) {
    var ser = new ConSerializer(serializer);
    IterableUtil.forEach(dataDef.body, ser::appendLine, ser::serialize);
  }

  private void serializeMems(@NotNull ClassDef classDef, @NotNull SourceBuilder serializer) {
    var ser = new MemberSerializer(serializer);
    IterableUtil.forEach(classDef.members(), ser::appendLine, ser::serialize);
  }

  private void doSerialize(@NotNull TyckDef unit) {
    switch (unit) {
      case FnDef teleDef -> new FnSerializer(this, shapeFactory)
        .serialize(teleDef);
      case DataDef dataDef -> {
        new DataSerializer(this, shapeFactory).serialize(dataDef);
        serializeCons(dataDef, this);
      }
      case ConDef conDef -> new ConSerializer(this)
        .serialize(conDef);
      case PrimDef primDef -> new PrimSerializer(this)
        .serialize(primDef);
      case ClassDef classDef -> {
        new ClassSerializer(this)
          .serialize(classDef);
        serializeMems(classDef, this);
      }
      case MemberDef memberDef -> new MemberSerializer(this)
        .serialize(memberDef);
    }
  }

  private void doSerialize(ModuleResult unit) {
    buildClass(javifyClassName(unit.name, null), null, false, () ->
      IterableUtil.forEach(unit.defs, this::appendLine, this::doSerialize));
  }

  @Override public ModuleSerializer serialize(ModuleResult unit) {
    doSerialize(unit);

    return this;
  }
}