Small changes for correctness

be-csharp/src/commonMain/resources/lang/temper/be/csharp/temper-core/Float64.cs

@@ -73,7 +73,7 @@ public static bool Near(
             var rel = relTol ?? 1e-9;
             var abs = absTol ?? 0.0;
             var margin = Math.Max(Math.Max(Math.Abs(x), Math.Abs(y)) * rel, abs);
-            return Math.Abs(x - y) < margin;
+            return Math.Abs(x - y) <= margin;
         }
 
         public static double Sign(this double x)
@@ -109,7 +109,7 @@ public static double ToFloat64(this string s)
                 ? double.PositiveInfinity
                 : trimmed == "-Infinity"
                     ? double.NegativeInfinity
-                    : double.Parse(s);
+                    : double.Parse(trimmed);
         }
 
         public static int ToInt(double value)

be-csharp/src/commonMain/resources/lang/temper/be/csharp/temper-core/StringUtil.cs

@@ -62,7 +62,7 @@ public static int CountBetween(string s, int left, int right)
             for (int i = Math.Max(0, left); i < limit; ++i)
             {
                 count += 1;
-                if (i + 1 < right)
+                if (i + 1 < limit)
                 {
                     char c = s[i];
                     if ('\uD800' <= c && c <= '\uDBFF')

be-java/src/commonMain/resources/lang/temper/be/java/temper-core/src/main/java/temper/core/Core.java

@@ -318,7 +318,7 @@ public static boolean float64Near(double x, double y, Double relTol, Double absT
         double rel = relTol == null ? 1e-9 : relTol;
         double abs = absTol == null ? 0.0 : absTol;
         double margin = Math.max(Math.max(Math.abs(x), Math.abs(y)) * rel, abs);
-        return Math.abs(x - y) < margin;
+        return Math.abs(x - y) <= margin;
     }
 
     /**

be-js/src/commonMain/resources/lang/temper/be/js/temper-core/float.js

@@ -19,7 +19,7 @@ export const float64Near = (x, y, relTol, absTol) => {
     absTol = 0;
   }
   const margin = Math.max(Math.max(Math.abs(x), Math.abs(y)) * relTol, absTol);
-  return Math.abs(x - y) < margin;
+  return Math.abs(x - y) <= margin;
 }
 
 /**

be-js/src/commonMain/resources/lang/temper/be/js/temper-core/listed.js

@@ -238,7 +238,7 @@ export const listBuilderReverse = (ls) => {
  * @param {T} newValue
  */
 export const listBuilderSet = (ls, i, newValue) => {
-  if (0 <= i && i <= ls.length) {
+  if (0 <= i && i < ls.length) {
     ls[i] = newValue;
   }
 }

be-js/src/commonMain/resources/lang/temper/be/js/temper-core/string.js

@@ -159,7 +159,7 @@ export const stringHasAtLeast = (s, begin, end, minCount) => {
  * @returns {number}
  */
 export const stringNext = (s, i) => {
-    let iNext = Math.min(s.length, i);
+    let iNext = Math.min(s.length, Math.max(0, i));
     let cp = s.codePointAt(i);
     if (cp !== undefined) {
         iNext += 1 + !!(cp >>> 16);
@@ -173,7 +173,7 @@ export const stringNext = (s, i) => {
  * @returns {number}
  */
 export const stringPrev = (s, i) => {
-    let iPrev = Math.min(s.length, i);
+    let iPrev = Math.min(s.length, Math.max(0, i));
     if (iPrev) {
         iPrev -= 1;
         if (iPrev && s.codePointAt(iPrev - 1) >>> 16) {

be-lua/src/commonMain/kotlin/lang/temper/be/lua/LuaLocalRemover.kt

@@ -3,25 +3,32 @@ package lang.temper.be.lua
 // Could go up to 200 in theory, but smaller numbers allow for larger closures.
 const val MAX_ALLOWABLE_LOCALS = 128
 
+// When wrapping is needed, keep first N locals as real locals for performance.
+// The rest overflow into the env table. Budget: LOCALS_TO_KEEP + 1 (env table) < MAX_ALLOWABLE_LOCALS.
+private const val LOCALS_TO_KEEP = 100
+
 class LuaLocalRemover {
     private var locals = mutableMapOf<LuaName, LuaName>()
     private var luaName = LuaName("_G")
     private var numLocalTables = 0
-    private var shouldRewrite = false
+    private var needsEnvTable = false
+    private var localsKept = 0
     private var localsToDecl = mutableListOf<LuaName>()
 
     private fun allocLocalTableName(): LuaName = LuaName("env_t${++numLocalTables}")
 
     private fun scoped(cb: () -> Lua.Chunk): Lua.Chunk {
         val lastLocals = locals.toMutableMap()
         val lastLuaName = luaName
-        val lastShouldRewrite = shouldRewrite
+        val lastNeedsEnvTable = needsEnvTable
+        val lastLocalsKept = localsKept
         val lastLocalsToDecl = localsToDecl
         localsToDecl = mutableListOf()
         val ret = cb()
         locals = lastLocals
         luaName = lastLuaName
-        shouldRewrite = lastShouldRewrite
+        needsEnvTable = lastNeedsEnvTable
+        localsKept = lastLocalsKept
         val localsToWrap = localsToDecl
         localsToDecl = lastLocalsToDecl
         return luaChunk(
@@ -50,12 +57,12 @@ class LuaLocalRemover {
         )
     }
 
-    private fun define(target: Lua.SetTarget) {
+    private fun define(target: Lua.SetTarget, asEnvField: Boolean) {
         when (target) {
             is Lua.DotSetTarget -> TODO()
             is Lua.IndexSetTarget -> TODO()
             is Lua.NameSetTarget -> {
-                if (shouldRewrite) {
+                if (asEnvField) {
                     locals[target.target.id] = luaName
                 } else {
                     locals.remove(target.target.id)
@@ -64,6 +71,10 @@ class LuaLocalRemover {
         }
     }
 
+    // Decide whether a batch of N new locals should overflow to the env table.
+    private fun shouldOverflow(count: Int): Boolean =
+        needsEnvTable && localsKept + count > LOCALS_TO_KEEP
+
     private fun set(target: Lua.SetTarget): Lua.SetTarget = when (target) {
         is Lua.DotSetTarget -> when (val obj = target.obj) {
             is Lua.SetTarget -> Lua.DotSetTarget(
@@ -230,8 +241,9 @@ class LuaLocalRemover {
     private fun wrapLocals(chunk: Lua.Chunk): List<Lua.Stmt> {
         luaName = allocLocalTableName()
         val numLocals = countLocalsDirectlyIn(chunk)
-        shouldRewrite = numLocals >= MAX_ALLOWABLE_LOCALS
-        if (shouldRewrite) {
+        needsEnvTable = numLocals >= MAX_ALLOWABLE_LOCALS
+        if (needsEnvTable) {
+            localsKept = 0
             return listOf(
                 Lua.LocalStmt(
                     chunk.pos,
@@ -355,17 +367,17 @@ class LuaLocalRemover {
             Lua.Name(stmt.name.pos, stmt.name.id),
         )
         is Lua.LocalDeclStmt -> {
-            stmt.targets.targets.forEach(::define)
-            if (shouldRewrite) {
-                null
-            } else {
+            val overflow = shouldOverflow(stmt.targets.targets.size)
+            stmt.targets.targets.forEach { define(it, asEnvField = overflow) }
+            if (!overflow) {
+                if (needsEnvTable) localsKept += stmt.targets.targets.size
                 stmt.targets.targets.forEach {
                     localsToDecl.add((it as Lua.NameSetTarget).target.id)
                 }
-                null
             }
+            null
         }
-        is Lua.LocalFunctionStmt -> if (shouldRewrite) {
+        is Lua.LocalFunctionStmt -> if (shouldOverflow(1)) {
             locals[stmt.name.id] = luaName
             Lua.SetStmt(
                 stmt.pos,
@@ -405,6 +417,7 @@ class LuaLocalRemover {
                 ),
             )
         } else {
+            if (needsEnvTable) localsKept++
             localsToDecl.add(stmt.name.id)
             Lua.SetStmt(
                 stmt.pos,
@@ -440,7 +453,7 @@ class LuaLocalRemover {
                 ),
             )
         }
-        is Lua.LocalStmt -> if (shouldRewrite) {
+        is Lua.LocalStmt -> if (shouldOverflow(stmt.targets.targets.size)) {
             val ret = Lua.SetStmt(
                 stmt.pos,
                 Lua.SetTargets(
@@ -461,9 +474,10 @@ class LuaLocalRemover {
                     stmt.exprs.exprs.map(::scan),
                 ),
             )
-            stmt.targets.targets.forEach(::define)
+            stmt.targets.targets.forEach { define(it, asEnvField = true) }
             ret
         } else {
+            if (needsEnvTable) localsKept += stmt.targets.targets.size
             val ret = Lua.SetStmt(
                 stmt.pos,
                 Lua.SetTargets(
@@ -479,7 +493,7 @@ class LuaLocalRemover {
                     stmt.exprs.exprs.map(::scan),
                 ),
             )
-            stmt.targets.targets.forEach(::define)
+            stmt.targets.targets.forEach { define(it, asEnvField = false) }
             ret
         }
         is Lua.SetStmt -> {