gofront 1.8.0 → 1.9.0

package/src/backend/wasm/emit-maps.js

@@ -8,7 +8,6 @@ import {
 	isComplex64,
 	isInterfaceType,
 	isPointerToStruct,
-	isSigned,
 	isStringType,
 	toWasmType,
 } from "./types.js";
@@ -43,10 +42,10 @@ export class MapsEmitter {
 				this.emitMapMake(mapInfo);
 				break;
 			case "get":
-				this.emitMapGet(mapInfo);
+				this.emitMapLookup(mapInfo, false);
 				break;
 			case "get_ok":
-				this.emitMapGetOk(mapInfo);
+				this.emitMapLookup(mapInfo, true);
 				break;
 			case "set":
 				this.emitMapSet(mapInfo);
@@ -61,13 +60,10 @@ export class MapsEmitter {
 				this.emitMapClear(mapInfo);
 				break;
 			case "keys":
-				this.emitMapKeys(mapInfo);
+				this.emitMapProject(mapInfo, false);
 				break;
 			case "values":
-				this.emitMapValues(mapInfo);
-				break;
-			case "clone":
-				this.emitMapClone(mapInfo);
+				this.emitMapProject(mapInfo, true);
 				break;
 		}
 	}
@@ -421,44 +417,6 @@ export class MapsEmitter {
 		this.pushInstruction("ref.eq");
 	}
 
-	emitElemLt(v1Loc, v2Loc, elemGoType) {
-		if (isStringType(elemGoType, this.mod.checker)) {
-			this.pushInstruction({ op: "local.get", index: v1Loc });
-			this.pushInstruction({ op: "local.get", index: v2Loc });
-			this.pushInstruction({
-				op: "call",
-				funcIndex: this.mod.getStringCmpImportIndex("<"),
-			});
-			return;
-		}
-		const wType = toWasmType(elemGoType, this.mod.checker, this.mod);
-		if (wType === "i64") {
-			this.pushInstruction({ op: "local.get", index: v1Loc });
-			this.pushInstruction({ op: "local.get", index: v2Loc });
-			this.pushInstruction(isSigned(elemGoType) ? "i64.lt_s" : "i64.lt_u");
-			return;
-		}
-		if (wType === "i32") {
-			this.pushInstruction({ op: "local.get", index: v1Loc });
-			this.pushInstruction({ op: "local.get", index: v2Loc });
-			this.pushInstruction(isSigned(elemGoType) ? "i32.lt_s" : "i32.lt_u");
-			return;
-		}
-		if (wType === "f32") {
-			this.pushInstruction({ op: "local.get", index: v1Loc });
-			this.pushInstruction({ op: "local.get", index: v2Loc });
-			this.pushInstruction("f32.lt");
-			return;
-		}
-		if (wType === "f64") {
-			this.pushInstruction({ op: "local.get", index: v1Loc });
-			this.pushInstruction({ op: "local.get", index: v2Loc });
-			this.pushInstruction("f64.lt");
-			return;
-		}
-		this.pushInstruction({ op: "i32.const", value: 0 });
-	}
-
 	emitMapMake(mapInfo) {
 		const nBucketsLoc = this.allocLocal(null, "i32");
 		const cLoc = this.allocLocal(null, "i32");
@@ -536,7 +494,7 @@ export class MapsEmitter {
 		this.pushInstruction({ op: "return" });
 	}
 
-	emitMapGet(mapInfo) {
+	emitMapLookup(mapInfo, withOk) {
 		const hLoc = this.allocLocal(null, "i32");
 		const bLoc = this.allocLocal(null, "i32");
 		const idxLoc = this.allocLocal(null, "i32");
@@ -547,11 +505,12 @@ export class MapsEmitter {
 		});
 		const entryKeyLoc = this.allocLocal(null, mapInfo.keyWType);
 
-		// If m == null -> return zero
+		// If m == null -> return zero (and false when withOk)
 		this.pushInstruction({ op: "local.get", index: 0 });
 		this.pushInstruction("ref.is_null");
 		this.pushInstruction({ op: "if", blockType: "void" });
 		this.emitZeroValue(mapInfo.valGoType, mapInfo.valWType);
+		if (withOk) this.pushInstruction({ op: "i32.const", value: 0 });
 		this.pushInstruction({ op: "return" });
 		this.pushInstruction("end");
 
@@ -633,6 +592,7 @@ export class MapsEmitter {
 			typeIndex: mapInfo.entryTypeIndex,
 			fieldIndex: 1,
 		});
+		if (withOk) this.pushInstruction({ op: "i32.const", value: 1 });
 		this.pushInstruction({ op: "return" });
 		this.pushInstruction("end");
 		this.pushInstruction("end");
@@ -650,121 +610,7 @@ export class MapsEmitter {
 
 		// Not found
 		this.emitZeroValue(mapInfo.valGoType, mapInfo.valWType);
-		this.pushInstruction({ op: "return" });
-	}
-
-	emitMapGetOk(mapInfo) {
-		const hLoc = this.allocLocal(null, "i32");
-		const bLoc = this.allocLocal(null, "i32");
-		const idxLoc = this.allocLocal(null, "i32");
-		const entryLoc = this.allocLocal(null, {
-			kind: "ref",
-			nullable: true,
-			typeIndex: mapInfo.entryTypeIndex,
-		});
-		const entryKeyLoc = this.allocLocal(null, mapInfo.keyWType);
-
-		// If m == null -> return zero, false
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction("ref.is_null");
-		this.pushInstruction({ op: "if", blockType: "void" });
-		this.emitZeroValue(mapInfo.valGoType, mapInfo.valWType);
-		this.pushInstruction({ op: "i32.const", value: 0 });
-		this.pushInstruction({ op: "return" });
-		this.pushInstruction("end");
-
-		this.emitKeyHash(1, mapInfo.keyGoType);
-		this.pushInstruction({ op: "local.set", index: hLoc });
-
-		this.pushInstruction({ op: "local.get", index: hLoc });
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.typeIndex,
-			fieldIndex: 8,
-		});
-		this.pushInstruction({ op: "i32.const", value: 1 });
-		this.pushInstruction("i32.sub");
-		this.pushInstruction("i32.and");
-		this.pushInstruction({ op: "local.set", index: bLoc });
-
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.typeIndex,
-			fieldIndex: 0,
-		});
-		this.pushInstruction({ op: "local.get", index: bLoc });
-		this.pushInstruction({
-			op: "array.get",
-			typeIndex: mapInfo.bucketsTypeIndex,
-		});
-		this.pushInstruction({ op: "local.set", index: idxLoc });
-
-		this.pushInstruction({ op: "block", blockType: "void" });
-		this.pushInstruction({ op: "loop", blockType: "void" });
-		this.pushInstruction({ op: "local.get", index: idxLoc });
-		this.pushInstruction({ op: "i32.const", value: -1 });
-		this.pushInstruction("i32.eq");
-		this.pushInstruction({ op: "br_if", depth: 1 });
-
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.typeIndex,
-			fieldIndex: 1,
-		});
-		this.pushInstruction({ op: "local.get", index: idxLoc });
-		this.pushInstruction({
-			op: "array.get",
-			typeIndex: mapInfo.entriesTypeIndex,
-		});
-		this.pushInstruction({ op: "local.set", index: entryLoc });
-
-		this.pushInstruction({ op: "local.get", index: entryLoc });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.entryTypeIndex,
-			fieldIndex: 5,
-		});
-		this.pushInstruction({ op: "i32.const", value: 1 });
-		this.pushInstruction("i32.eq");
-		this.pushInstruction({ op: "if", blockType: "void" });
-
-		this.pushInstruction({ op: "local.get", index: entryLoc });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.entryTypeIndex,
-			fieldIndex: 0,
-		});
-		this.pushInstruction({ op: "local.set", index: entryKeyLoc });
-
-		this.emitKeyEq(1, entryKeyLoc, mapInfo.keyGoType);
-		this.pushInstruction({ op: "if", blockType: "void" });
-		this.pushInstruction({ op: "local.get", index: entryLoc });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.entryTypeIndex,
-			fieldIndex: 1,
-		});
-		this.pushInstruction({ op: "i32.const", value: 1 });
-		this.pushInstruction({ op: "return" });
-		this.pushInstruction("end");
-		this.pushInstruction("end");
-
-		this.pushInstruction({ op: "local.get", index: entryLoc });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.entryTypeIndex,
-			fieldIndex: 2,
-		});
-		this.pushInstruction({ op: "local.set", index: idxLoc });
-		this.pushInstruction({ op: "br", depth: 0 });
-		this.pushInstruction("end");
-		this.pushInstruction("end");
-
-		this.emitZeroValue(mapInfo.valGoType, mapInfo.valWType);
-		this.pushInstruction({ op: "i32.const", value: 0 });
+		if (withOk) this.pushInstruction({ op: "i32.const", value: 0 });
 		this.pushInstruction({ op: "return" });
 	}
 
@@ -1192,6 +1038,9 @@ export class MapsEmitter {
 
 		// new_entry = struct.new (key, val, next=buckets[b], order_prev=oldTail, order_next=-1, active=1)
 		this.pushInstruction({ op: "local.get", index: 1 });
+		// The stored key is a copy: later writes to the caller's array or
+		// struct must not change the map's key.
+		this._emitCopyMapEntry(mapInfo.keyGoType, mapInfo.keyWType);
 		this.pushInstruction({ op: "local.get", index: 2 });
 		this.pushInstruction({ op: "local.get", index: 0 });
 		this.pushInstruction({
@@ -1732,120 +1581,10 @@ export class MapsEmitter {
 		this.pushInstruction({ op: "return" });
 	}
 
-	emitMapKeys(mapInfo) {
-		const sliceInfo = this.mod.getSliceType(mapInfo.keyGoType);
-		const lenLoc = this.allocLocal(null, "i32");
-		const arrLoc = this.allocLocal(null, {
-			kind: "ref",
-			nullable: true,
-			typeIndex: sliceInfo.arrInfo.typeIndex,
-		});
-		const iLoc = this.allocLocal(null, "i32");
-		const idxLoc = this.allocLocal(null, "i32");
-		const entryLoc = this.allocLocal(null, {
-			kind: "ref",
-			nullable: true,
-			typeIndex: mapInfo.entryTypeIndex,
-		});
-
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction("ref.is_null");
-		this.pushInstruction({ op: "if", blockType: "void" });
-		this.pushInstruction({
-			op: "global.get",
-			index: sliceInfo.emptyGlobalIndex,
-		});
-		this.pushInstruction({ op: "return" });
-		this.pushInstruction("end");
-
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.typeIndex,
-			fieldIndex: 2,
-		});
-		this.pushInstruction({ op: "local.set", index: lenLoc });
-
-		this.pushInstruction({ op: "local.get", index: lenLoc });
-		this.pushInstruction({
-			op: "array.new_default",
-			typeIndex: sliceInfo.arrInfo.typeIndex,
-		});
-		this.pushInstruction({ op: "local.set", index: arrLoc });
-
-		this.pushInstruction({ op: "i32.const", value: 0 });
-		this.pushInstruction({ op: "local.set", index: iLoc });
-
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.typeIndex,
-			fieldIndex: 5,
-		});
-		this.pushInstruction({ op: "local.set", index: idxLoc });
-
-		this.pushInstruction({ op: "block", blockType: "void" });
-		this.pushInstruction({ op: "loop", blockType: "void" });
-		this.pushInstruction({ op: "local.get", index: idxLoc });
-		this.pushInstruction({ op: "i32.const", value: -1 });
-		this.pushInstruction("i32.eq");
-		this.pushInstruction({ op: "br_if", depth: 1 });
-
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.typeIndex,
-			fieldIndex: 1,
-		});
-		this.pushInstruction({ op: "local.get", index: idxLoc });
-		this.pushInstruction({
-			op: "array.get",
-			typeIndex: mapInfo.entriesTypeIndex,
-		});
-		this.pushInstruction({ op: "local.set", index: entryLoc });
-
-		this.pushInstruction({ op: "local.get", index: arrLoc });
-		this.pushInstruction({ op: "local.get", index: iLoc });
-		this.pushInstruction({ op: "local.get", index: entryLoc });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.entryTypeIndex,
-			fieldIndex: 0,
-		});
-		this.pushInstruction({
-			op: "array.set",
-			typeIndex: sliceInfo.arrInfo.typeIndex,
-		});
-
-		this.pushInstruction({ op: "local.get", index: iLoc });
-		this.pushInstruction({ op: "i32.const", value: 1 });
-		this.pushInstruction("i32.add");
-		this.pushInstruction({ op: "local.set", index: iLoc });
-
-		this.pushInstruction({ op: "local.get", index: entryLoc });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.entryTypeIndex,
-			fieldIndex: 4,
-		});
-		this.pushInstruction({ op: "local.set", index: idxLoc });
-		this.pushInstruction({ op: "br", depth: 0 });
-		this.pushInstruction("end");
-		this.pushInstruction("end");
-
-		this.pushInstruction({ op: "local.get", index: arrLoc });
-		this.pushInstruction({ op: "i32.const", value: 0 });
-		this.pushInstruction({ op: "local.get", index: lenLoc });
-		this.pushInstruction({ op: "local.get", index: lenLoc });
-		this.pushInstruction({
-			op: "struct.new",
-			typeIndex: sliceInfo.typeIndex,
-		});
-		this.pushInstruction({ op: "return" });
-	}
-
-	emitMapValues(mapInfo) {
-		const sliceInfo = this.mod.getSliceType(mapInfo.valGoType);
+	emitMapProject(mapInfo, isValues) {
+		const sliceInfo = this.mod.getSliceType(
+			isValues ? mapInfo.valGoType : mapInfo.keyGoType,
+		);
 		const lenLoc = this.allocLocal(null, "i32");
 		const arrLoc = this.allocLocal(null, {
 			kind: "ref",
@@ -1922,7 +1661,7 @@ export class MapsEmitter {
 		this.pushInstruction({
 			op: "struct.get",
 			typeIndex: mapInfo.entryTypeIndex,
-			fieldIndex: 1,
+			fieldIndex: isValues ? 1 : 0,
 		});
 		this.pushInstruction({
 			op: "array.set",
@@ -1955,93 +1694,4 @@ export class MapsEmitter {
 		});
 		this.pushInstruction({ op: "return" });
 	}
-
-	emitMapClone(mapInfo) {
-		const dstLoc = this.allocLocal(null, mapInfo.wType);
-		const idxLoc = this.allocLocal(null, "i32");
-		const entryLoc = this.allocLocal(null, {
-			kind: "ref",
-			nullable: true,
-			typeIndex: mapInfo.entryTypeIndex,
-		});
-
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction("ref.is_null");
-		this.pushInstruction({ op: "if", blockType: "void" });
-		this.pushInstruction({ op: "ref.null", heapType: mapInfo.typeIndex });
-		this.pushInstruction({ op: "return" });
-		this.pushInstruction("end");
-
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.typeIndex,
-			fieldIndex: 2,
-		});
-		this.pushInstruction({
-			op: "call",
-			funcIndex: this.mod.resolveFuncIndex(mapInfo.makeFuncName),
-		});
-		this.pushInstruction({ op: "local.set", index: dstLoc });
-
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.typeIndex,
-			fieldIndex: 5,
-		});
-		this.pushInstruction({ op: "local.set", index: idxLoc });
-
-		this.pushInstruction({ op: "block", blockType: "void" });
-		this.pushInstruction({ op: "loop", blockType: "void" });
-		this.pushInstruction({ op: "local.get", index: idxLoc });
-		this.pushInstruction({ op: "i32.const", value: -1 });
-		this.pushInstruction("i32.eq");
-		this.pushInstruction({ op: "br_if", depth: 1 });
-
-		this.pushInstruction({ op: "local.get", index: 0 });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.typeIndex,
-			fieldIndex: 1,
-		});
-		this.pushInstruction({ op: "local.get", index: idxLoc });
-		this.pushInstruction({
-			op: "array.get",
-			typeIndex: mapInfo.entriesTypeIndex,
-		});
-		this.pushInstruction({ op: "local.set", index: entryLoc });
-
-		this.pushInstruction({ op: "local.get", index: dstLoc });
-		this.pushInstruction({ op: "local.get", index: entryLoc });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.entryTypeIndex,
-			fieldIndex: 0,
-		});
-		this.pushInstruction({ op: "local.get", index: entryLoc });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.entryTypeIndex,
-			fieldIndex: 1,
-		});
-		this.pushInstruction({
-			op: "call",
-			funcIndex: this.mod.resolveFuncIndex(mapInfo.setFuncName),
-		});
-
-		this.pushInstruction({ op: "local.get", index: entryLoc });
-		this.pushInstruction({
-			op: "struct.get",
-			typeIndex: mapInfo.entryTypeIndex,
-			fieldIndex: 4,
-		});
-		this.pushInstruction({ op: "local.set", index: idxLoc });
-		this.pushInstruction({ op: "br", depth: 0 });
-		this.pushInstruction("end");
-		this.pushInstruction("end");
-
-		this.pushInstruction({ op: "local.get", index: dstLoc });
-		this.pushInstruction({ op: "return" });
-	}
 }