pkg/wasm_builder/lib/src/ir/table.dart

// Copyright (c) 2023, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import '../serialize/printer.dart';
import '../serialize/serialize.dart';
import 'ir.dart';

/// An (imported or defined) table.
class Table with Indexable, Exportable implements Serializable {
  @override
  final FinalizableIndex finalizableIndex;
  final RefType type;
  // Mutable so that a table's size does not need to be known prior to the table
  // being instantiated.
  int minSize;
  final int? maxSize;
  @override
  final Module enclosingModule;

  Table(
    this.enclosingModule,
    this.finalizableIndex,
    this.type,
    this.minSize,
    this.maxSize,
  );

  @override
  void serialize(Serializer s) {
    s.write(type);
    if (maxSize == null) {
      s.writeByte(0x00);
      s.writeUnsigned(minSize);
    } else {
      s.writeByte(0x01);
      s.writeUnsigned(minSize);
      s.writeUnsigned(maxSize!);
    }
  }

  /// Export a table from the module.
  @override
  Export buildExport(String name) => TableExport(name, this);

  void collectUsedTypes(Set<DefType> usedTypes) {
    final defType = type.containedDefType;
    if (defType != null) usedTypes.add(defType);
  }
}

/// A table defined in a module.
class DefinedTable extends Table {
  DefinedTable(
    super.enclosingModule,
    super.finalizableIndex,
    super.type,
    super.minSize,
    super.maxSize,
  );

  void printTo(IrPrinter p) {
    p.write('(table ');
    p.writeTableReference(this, alwaysPrint: true);
    String? exportName;
    for (final e in enclosingModule.exports.exported) {
      if (e is TableExport && e.table == this) {
        exportName = e.name;
        break;
      }
    }
    if (exportName != null) {
      p.write(' ');
      p.writeExport(exportName);
    }

    p.write(' $minSize ');
    p.writeValueType(type);
    p.write(')');
  }
}

/// An imported table.
class ImportedTable extends Table implements Import {
  @override
  final String module;
  @override
  final String name;

  ImportedTable(
    super.enclosingModule,
    this.module,
    this.name,
    super.finalizableIndex,
    super.type,
    super.minSize,
    super.maxSize,
  );

  @override
  void serialize(Serializer s) {
    s.writeName(module);
    s.writeName(name);
    s.writeByte(0x01);
    super.serialize(s);
  }

  void printTo(IrPrinter p) {
    p.write('(table ');
    p.writeTableReference(this, alwaysPrint: true);
    p.write(' ');
    p.writeImport(module, name);
    p.write(' $minSize ');
    p.writeValueType(type);
    p.write(')');
  }
}

class TableExport extends Export {
  final Table table;

  TableExport(super.name, this.table);

  @override
  void serialize(Serializer s) {
    s.writeName(name);
    s.writeByte(0x01);
    s.writeUnsigned(table.index);
  }
}