@thi.ng/shader-ast-js 1.0.26 → 1.0.28

package/target.js

@@ -1,232 +1,220 @@
 import { isBoolean } from "@thi.ng/checks/is-boolean";
 import { isNumber } from "@thi.ng/checks/is-number";
 import { unsupported } from "@thi.ng/errors/unsupported";
-import { isBool, isInt, isMat, isUint, isVec, } from "@thi.ng/shader-ast/ast/checks";
+import {
+  isBool,
+  isInt,
+  isMat,
+  isUint,
+  isVec
+} from "@thi.ng/shader-ast/ast/checks";
 import { defTarget } from "@thi.ng/shader-ast/target";
 import { JS_DEFAULT_ENV } from "./env.js";
 const CMP_OPS = {
-    "!": "not",
-    "<": "lt",
-    "<=": "lte",
-    "==": "eq",
-    "!=": "neq",
-    ">=": "gte",
-    ">": "gt",
+  "!": "not",
+  "<": "lt",
+  "<=": "lte",
+  "==": "eq",
+  "!=": "neq",
+  ">=": "gte",
+  ">": "gt"
 };
 const OP_IDS = {
-    ...CMP_OPS,
-    "+": "add",
-    "-": "sub",
-    "*": "mul",
-    "/": "div",
-    "%": "modi",
-    "++": "inc",
-    "--": "dec",
-    "||": "or",
-    "&&": "and",
-    "|": "bitor",
-    "&": "bitand",
-    "^": "bitxor",
-    "~": "bitnot",
-    "<<": "lshift",
-    ">>": "rshift",
+  ...CMP_OPS,
+  "+": "add",
+  "-": "sub",
+  "*": "mul",
+  "/": "div",
+  "%": "modi",
+  "++": "inc",
+  "--": "dec",
+  "||": "or",
+  "&&": "and",
+  "|": "bitor",
+  "&": "bitand",
+  "^": "bitxor",
+  "~": "bitnot",
+  "<<": "lshift",
+  ">>": "rshift"
 };
 const VEC_TYPES = [
-    "vec2",
-    "vec3",
-    "vec4",
-    "ivec2",
-    "ivec3",
-    "ivec4",
-    "uvec2",
-    "uvec3",
-    "uvec4",
+  "vec2",
+  "vec3",
+  "vec4",
+  "ivec2",
+  "ivec3",
+  "ivec4",
+  "uvec2",
+  "uvec3",
+  "uvec4"
 ];
 const PRELUDE = [
-    "float",
-    "int",
-    "uint",
-    ...VEC_TYPES,
-    "bvec2",
-    "bvec3",
-    "bvec4",
-    "mat2",
-    "mat3",
-    "mat4",
-    "sampler2D",
-    "sampler3D",
-    "samplerCube",
-    "sampler2DShadow",
-    "samplerCubeShadow",
-]
-    .map((x) => `const ${x} = env.${x};`)
-    .join("\n");
-const POOL_PRELUDE = VEC_TYPES.map((x) => `const $${x} = env.pools.${x}.from;`).join("\n");
+  "float",
+  "int",
+  "uint",
+  ...VEC_TYPES,
+  "bvec2",
+  "bvec3",
+  "bvec4",
+  "mat2",
+  "mat3",
+  "mat4",
+  "sampler2D",
+  "sampler3D",
+  "samplerCube",
+  "sampler2DShadow",
+  "samplerCubeShadow"
+].map((x) => `const ${x} = env.${x};`).join("\n");
+const POOL_PRELUDE = VEC_TYPES.map(
+  (x) => `const $${x} = env.pools.${x}.from;`
+).join("\n");
 const COMPS = { x: 0, y: 1, z: 2, w: 3 };
 const RE_SEMI = /[};]$/;
 const RESET = `for(let t in env.pools) env.pools[t].reset();`;
 const isIntOrBool = (l) => isInt(l) || isUint(l) || isBool(l);
 const isVecOrMat = (l) => isVec(l) || isMat(l);
 const swizzle = (id) => [...id].map((x) => COMPS[x]).join(", ");
-const buildComments = (t) => `/**\n${t.args.map((p) => ` * @param ${p.id} ${p.type}`).join("\n")}\n */`;
-const buildExports = (tree) => tree.tag === "scope"
-    ? tree.body
-        .filter((x) => x.tag === "fn")
-        .map((f) => `${f.id}: ${f.id}`)
-        .join(",\n")
-    : tree.tag === "fn"
-        ? `${tree.id}: ${tree.id}`
-        : "";
-export const targetJS = (opts) => {
-    opts = { ...opts };
-    const ff = opts.prec !== undefined
-        ? (x) => (x === (x | 0) ? x : x.toFixed(opts.prec))
-        : String;
-    const $list = (body, sep = ", ") => body.map(emit).join(sep);
-    const $fn = (name, args) => `${name}(${$list(args)})`;
-    const $vecFromPool = ({ val, info, type }) => !info ? `$${type}(${$list(val)})` : `env.${type}${info}(${$list(val)})`;
-    const $vec = ({ val, info, type }) => !info ? `[${$list(val)}]` : `env.${type}${info}(${$list(val)})`;
-    const $num = (v, f) => isNumber(v) ? String(v) : f(v);
-    const $float = (v, f) => (isNumber(v) ? ff(v) : f(v));
-    const emit = defTarget({
-        arg: (t) => t.id,
-        array_init: (t) => `[${$list(t.init)}]`,
-        assign: (t) => {
-            const rhs = emit(t.r);
-            if (t.l.tag === "swizzle") {
-                const s = t.l;
-                const id = swizzle(s.id);
-                const val = emit(s.val);
-                return s.id.length > 1
-                    ? `env.set_swizzle${s.id.length}(${val}, ${rhs}, ${id})`
-                    : `(${val}[${id}] = ${rhs})`;
-            }
-            return `${emit(t.l)} = ${rhs}`;
-        },
-        ctrl: (t) => t.id,
-        call: (t) => $fn(t.id, t.args),
-        call_i: (t) => $fn(`${t.args[0].type}.${t.id}${t.info || ""}`, t.args),
-        decl: ({ type, id }) => {
-            const res = [];
-            res.push(id.opts.const ? "const" : "let", `/*${type}*/`, id.id);
-            id.init
-                ? res.push(`= ${emit(id.init)}`)
-                : id.opts.num !== undefined
-                    ? res.push(`= new Array(${id.opts.num})`)
-                    : undefined;
-            return res.join(" ");
-        },
-        fn: (t) => {
-            let body;
-            if (t.id === "main") {
-                body = `{\n${RESET}\n${emit({
-                    ...t.scope,
-                    global: true,
-                })}}`;
-            }
-            else {
-                body = emit(t.scope);
-            }
-            return `${buildComments(t)}\nfunction ${t.id}(${$list(t.args)}) ${body}`;
-        },
-        for: (t) => `for(${t.init ? emit(t.init) : ""}; ${emit(t.test)}; ${t.iter ? emit(t.iter) : ""}) ${emit(t.scope)}`,
-        idx: (t) => `${emit(t.val)}[${emit(t.id)}]`,
-        idxm: (t) => `${t.val.type}.idx(${emit(t.val)},${emit(t.id)})`,
-        if: (t) => {
-            const res = `if (${emit(t.test)}) ${emit(t.t)}`;
-            return t.f ? `${res} else ${emit(t.f)}` : res;
-        },
-        lit: (t) => {
-            const v = t.val;
-            switch (t.type) {
-                case "bool":
-                    return isBoolean(v) ? String(v) : `!!(${emit(v)})`;
-                case "float":
-                    return $float(v, () => isBool(v) ? `(${emit(v)} & 1)` : emit(v));
-                case "int":
-                    return $num(v, () => `(${emit(v)} | 0)`);
-                case "uint":
-                    return $num(v, () => `(${emit(v)} >>> 0)`);
-                case "vec2":
-                case "vec3":
-                case "vec4":
-                case "ivec2":
-                case "ivec3":
-                case "ivec4":
-                case "uvec2":
-                case "uvec3":
-                case "uvec4":
-                    return $vecFromPool(t);
-                case "bvec2":
-                case "bvec3":
-                case "bvec4":
-                case "mat2":
-                case "mat3":
-                case "mat4":
-                    return $vec(t);
-                default:
-                    return unsupported(`unknown type: ${t.type}`);
-            }
-        },
-        op1: (t) => {
-            const complex = isVecOrMat(t) || isInt(t);
-            const op = t.op;
-            const val = emit(t.val);
-            return complex && t.post
-                ? `${t.val.id} = ${t.type}.${OP_IDS[op]}(${val})`
-                : complex
-                    ? `${t.type}.${OP_IDS[op]}1(${val})`
-                    : t.post
-                        ? `(${val}${op})`
-                        : `${op}${val}`;
-        },
-        op2: (t) => {
-            const { l, r } = t;
-            const vec = isVecOrMat(l)
-                ? l.type
-                : isVecOrMat(r)
-                    ? r.type
-                    : undefined;
-            const int = !vec
-                ? isIntOrBool(l)
-                    ? l.type
-                    : isIntOrBool(r)
-                        ? r.type
-                        : undefined
-                : undefined;
-            const el = emit(l);
-            const er = emit(r);
-            return vec || (int && !CMP_OPS[t.op])
-                ? `${vec || int}.${OP_IDS[t.op]}${t.info || ""}(${el}, ${er})`
-                : `(${el} ${t.op} ${er})`;
-        },
-        ret: (t) => "return" + (t.val ? " " + emit(t.val) : ""),
-        scope: (t) => {
-            let res = $list(t.body, ";\n");
-            res += t.body.length && !RE_SEMI.test(res) ? ";" : "";
-            return t.global ? res : `{\n${res}\n}`;
-        },
-        swizzle: (t) => t.id.length > 1
-            ? `env.swizzle${t.id.length}(${emit(t.val)}, ${swizzle(t.id)})`
-            : `${emit(t.val)}[${swizzle(t.id)}]`,
-        sym: (t) => t.id,
-        ternary: (t) => `(${emit(t.test)} ? ${emit(t.t)} : ${emit(t.f)})`,
-        while: (t) => `while (${emit(t.test)}) ${emit(t.scope)}`,
-    });
-    Object.assign(emit, {
-        compile: (tree, env = JS_DEFAULT_ENV) => {
-            const exports = buildExports(tree);
-            return new Function("env", [
-                PRELUDE,
-                POOL_PRELUDE,
-                emit(tree),
-                "return {",
-                `__reset: () => {${RESET}},`,
-                `__stats: () => Object.entries(env.pools).reduce((acc, [k, v]) => (acc[k] = v.index, acc), {}),`,
-                exports,
-                "};",
-            ].join("\n"))(env);
-        },
-    });
-    return emit;
+const buildComments = (t) => `/**
+${t.args.map((p) => ` * @param ${p.id} ${p.type}`).join("\n")}
+ */`;
+const buildExports = (tree) => tree.tag === "scope" ? tree.body.filter((x) => x.tag === "fn").map((f) => `${f.id}: ${f.id}`).join(",\n") : tree.tag === "fn" ? `${tree.id}: ${tree.id}` : "";
+const targetJS = (opts) => {
+  opts = { ...opts };
+  const ff = opts.prec !== void 0 ? (x) => x === (x | 0) ? x : x.toFixed(opts.prec) : String;
+  const $list = (body, sep = ", ") => body.map(emit).join(sep);
+  const $fn = (name, args) => `${name}(${$list(args)})`;
+  const $vecFromPool = ({ val, info, type }) => !info ? `$${type}(${$list(val)})` : `env.${type}${info}(${$list(val)})`;
+  const $vec = ({ val, info, type }) => !info ? `[${$list(val)}]` : `env.${type}${info}(${$list(val)})`;
+  const $num = (v, f) => isNumber(v) ? String(v) : f(v);
+  const $float = (v, f) => isNumber(v) ? ff(v) : f(v);
+  const emit = defTarget({
+    arg: (t) => t.id,
+    array_init: (t) => `[${$list(t.init)}]`,
+    assign: (t) => {
+      const rhs = emit(t.r);
+      if (t.l.tag === "swizzle") {
+        const s = t.l;
+        const id = swizzle(s.id);
+        const val = emit(s.val);
+        return s.id.length > 1 ? `env.set_swizzle${s.id.length}(${val}, ${rhs}, ${id})` : `(${val}[${id}] = ${rhs})`;
+      }
+      return `${emit(t.l)} = ${rhs}`;
+    },
+    ctrl: (t) => t.id,
+    call: (t) => $fn(t.id, t.args),
+    call_i: (t) => $fn(`${t.args[0].type}.${t.id}${t.info || ""}`, t.args),
+    decl: ({ type, id }) => {
+      const res = [];
+      res.push(id.opts.const ? "const" : "let", `/*${type}*/`, id.id);
+      id.init ? res.push(`= ${emit(id.init)}`) : id.opts.num !== void 0 ? res.push(`= new Array(${id.opts.num})`) : void 0;
+      return res.join(" ");
+    },
+    fn: (t) => {
+      let body;
+      if (t.id === "main") {
+        body = `{
+${RESET}
+${emit({
+          ...t.scope,
+          global: true
+        })}}`;
+      } else {
+        body = emit(t.scope);
+      }
+      return `${buildComments(t)}
+function ${t.id}(${$list(
+        t.args
+      )}) ${body}`;
+    },
+    for: (t) => `for(${t.init ? emit(t.init) : ""}; ${emit(t.test)}; ${t.iter ? emit(t.iter) : ""}) ${emit(t.scope)}`,
+    idx: (t) => `${emit(t.val)}[${emit(t.id)}]`,
+    idxm: (t) => `${t.val.type}.idx(${emit(t.val)},${emit(t.id)})`,
+    if: (t) => {
+      const res = `if (${emit(t.test)}) ${emit(t.t)}`;
+      return t.f ? `${res} else ${emit(t.f)}` : res;
+    },
+    lit: (t) => {
+      const v = t.val;
+      switch (t.type) {
+        case "bool":
+          return isBoolean(v) ? String(v) : `!!(${emit(v)})`;
+        case "float":
+          return $float(
+            v,
+            () => isBool(v) ? `(${emit(v)} & 1)` : emit(v)
+          );
+        case "int":
+          return $num(v, () => `(${emit(v)} | 0)`);
+        case "uint":
+          return $num(v, () => `(${emit(v)} >>> 0)`);
+        case "vec2":
+        case "vec3":
+        case "vec4":
+        case "ivec2":
+        case "ivec3":
+        case "ivec4":
+        case "uvec2":
+        case "uvec3":
+        case "uvec4":
+          return $vecFromPool(t);
+        case "bvec2":
+        case "bvec3":
+        case "bvec4":
+        case "mat2":
+        case "mat3":
+        case "mat4":
+          return $vec(t);
+        default:
+          return unsupported(`unknown type: ${t.type}`);
+      }
+    },
+    op1: (t) => {
+      const complex = isVecOrMat(t) || isInt(t);
+      const op = t.op;
+      const val = emit(t.val);
+      return complex && t.post ? `${t.val.id} = ${t.type}.${OP_IDS[op]}(${val})` : complex ? `${t.type}.${OP_IDS[op]}1(${val})` : t.post ? `(${val}${op})` : `${op}${val}`;
+    },
+    op2: (t) => {
+      const { l, r } = t;
+      const vec = isVecOrMat(l) ? l.type : isVecOrMat(r) ? r.type : void 0;
+      const int = !vec ? isIntOrBool(l) ? l.type : isIntOrBool(r) ? r.type : void 0 : void 0;
+      const el = emit(l);
+      const er = emit(r);
+      return vec || int && !CMP_OPS[t.op] ? `${vec || int}.${OP_IDS[t.op]}${t.info || ""}(${el}, ${er})` : `(${el} ${t.op} ${er})`;
+    },
+    ret: (t) => "return" + (t.val ? " " + emit(t.val) : ""),
+    scope: (t) => {
+      let res = $list(t.body, ";\n");
+      res += t.body.length && !RE_SEMI.test(res) ? ";" : "";
+      return t.global ? res : `{
+${res}
+}`;
+    },
+    swizzle: (t) => t.id.length > 1 ? `env.swizzle${t.id.length}(${emit(t.val)}, ${swizzle(t.id)})` : `${emit(t.val)}[${swizzle(t.id)}]`,
+    sym: (t) => t.id,
+    ternary: (t) => `(${emit(t.test)} ? ${emit(t.t)} : ${emit(t.f)})`,
+    while: (t) => `while (${emit(t.test)}) ${emit(t.scope)}`
+  });
+  Object.assign(emit, {
+    compile: (tree, env = JS_DEFAULT_ENV) => {
+      const exports = buildExports(tree);
+      return new Function(
+        "env",
+        [
+          PRELUDE,
+          POOL_PRELUDE,
+          emit(tree),
+          "return {",
+          `__reset: () => {${RESET}},`,
+          `__stats: () => Object.entries(env.pools).reduce((acc, [k, v]) => (acc[k] = v.index, acc), {}),`,
+          exports,
+          "};"
+        ].join("\n")
+      )(env);
+    }
+  });
+  return emit;
+};
+export {
+  targetJS
 };