gofront 1.7.0 → 1.9.0

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

@@ -4,13 +4,35 @@
 
 import {
 	isArrayType,
+	isComplex,
+	isComplex64,
 	isInterfaceType,
 	isPointerToStruct,
-	isSigned,
 	isStringType,
 	toWasmType,
 } from "./types.js";
 
+function _complexTypeFor(mod, t) {
+	if (!isComplex(t)) return null;
+	const is64 = isComplex64(t);
+	return mod.getComplexType(is64 ? "complex64" : "complex128");
+}
+
+function _resolveStructType(mod, goType) {
+	let t = goType;
+	while (t && (t.kind === "TypeName" || t.kind === "Ident")) {
+		const info = mod.getStructType(t.name);
+		if (info) return info;
+		const resolved = mod.checker?.types?.get(t.name);
+		if (!resolved || resolved === t) break;
+		t = resolved;
+	}
+	if (t?.kind === "named") return mod.getStructType(t.name) ?? null;
+	if ((t?.kind === "struct" || t?.kind === "StructType") && t.name)
+		return mod.getStructType(t.name) ?? null;
+	return null;
+}
+
 export class MapsEmitter {
 	emitMapHelper(fn) {
 		const kind = fn._mapHelperKind;
@@ -20,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);
@@ -38,31 +60,19 @@ 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;
 		}
 	}
 
-	// Struct info for a Go type used as a map key (named struct or alias).
+	// Struct info for a Go type used as a map key (named struct, alias, or complex).
 	_structInfoForGoType(goType) {
-		let t = goType;
-		while (t && (t.kind === "TypeName" || t.kind === "Ident")) {
-			const info = this.mod.getStructType(t.name);
-			if (info) return info;
-			const resolved = this.mod.checker?.types?.get(t.name);
-			if (!resolved || resolved === t) break;
-			t = resolved;
-		}
-		if (t?.kind === "named") return this.mod.getStructType(t.name) ?? null;
-		if ((t?.kind === "struct" || t?.kind === "StructType") && t.name)
-			return this.mod.getStructType(t.name) ?? null;
-		return null;
+		return (
+			_complexTypeFor(this.mod, goType) ?? _resolveStructType(this.mod, goType)
+		);
 	}
 
 	_unsupportedMapKey(_keyGoType) {
@@ -166,6 +176,34 @@ export class MapsEmitter {
 			this.pushInstruction("i32.mul");
 			return;
 		}
+		if (wType === "v128") {
+			// Lane-wise like a [4]float32 key: h = 17; h = h*31 + hash(lane).
+			const isMask = this._isSimdMaskType(keyGoType);
+			const laneType = {
+				kind: "basic",
+				name: isMask ? "int32" : "float32",
+			};
+			const h = this.allocLocal(null, "i32");
+			this.pushInstruction({ op: "i32.const", value: 17 });
+			this.pushInstruction({ op: "local.set", index: h });
+			for (let lane = 0; lane < 4; lane++) {
+				const lt = this.allocLocal(null, isMask ? "i32" : "f32");
+				this.pushInstruction({ op: "local.get", index: keyLocal });
+				this.pushInstruction({
+					op: isMask ? "i32x4.extract_lane" : "f32x4.extract_lane",
+					lane,
+				});
+				this.pushInstruction({ op: "local.set", index: lt });
+				this.pushInstruction({ op: "local.get", index: h });
+				this.pushInstruction({ op: "i32.const", value: 31 });
+				this.pushInstruction("i32.mul");
+				this.emitKeyHash(lt, laneType);
+				this.pushInstruction("i32.add");
+				this.pushInstruction({ op: "local.set", index: h });
+			}
+			this.pushInstruction({ op: "local.get", index: h });
+			return;
+		}
 		// Floats: `x + 0.0` folds -0 onto +0 (they are equal keys in Go) and
 		// NaN elements never match (f32.eq/f64.eq returns false in key comparison).
 		if (wType === "f32") {
@@ -336,6 +374,12 @@ export class MapsEmitter {
 			this.pushInstruction("f64.eq");
 			return;
 		}
+		if (wType === "v128") {
+			this.pushInstruction({ op: "local.get", index: k1Local });
+			this.pushInstruction({ op: "local.get", index: k2Local });
+			this._emitSimdEq(keyGoType);
+			return;
+		}
 		const structInfo = this._structInfoForGoType(keyGoType);
 		if (
 			structInfo &&
@@ -373,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");
@@ -488,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");
@@ -499,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");
 
@@ -585,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");
@@ -602,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" });
 	}
 
@@ -1144,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({
@@ -1684,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",
@@ -1874,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",
@@ -1907,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" });
-	}
 }