@shaderfrog/core 0.1.1 → 1.0.0

package/dist/util/ast.js

@@ -1,4 +1,328 @@
-import { generate } from '@shaderfrog/glsl-parser';
+var __read = (this && this.__read) || function (o, n) {
+    var m = typeof Symbol === "function" && o[Symbol.iterator];
+    if (!m) return o;
+    var i = m.call(o), r, ar = [], e;
+    try {
+        while ((n === void 0 || n-- > 0) && !(r = i.next()).done) ar.push(r.value);
+    }
+    catch (error) { e = { error: error }; }
+    finally {
+        try {
+            if (r && !r.done && (m = i["return"])) m.call(i);
+        }
+        finally { if (e) throw e.error; }
+    }
+    return ar;
+};
+var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
+    if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
+        if (ar || !(i in from)) {
+            if (!ar) ar = Array.prototype.slice.call(from, 0, i);
+            ar[i] = from[i];
+        }
+    }
+    return to.concat(ar || Array.prototype.slice.call(from));
+};
+import { parser, generate } from '@shaderfrog/glsl-parser';
+import { visit, } from '@shaderfrog/glsl-parser/ast';
+var log = function () {
+    var _a;
+    var args = [];
+    for (var _i = 0; _i < arguments.length; _i++) {
+        args[_i] = arguments[_i];
+    }
+    return (_a = console.log).call.apply(_a, __spreadArray([console, '\x1b[31m(core.manipulate)\x1b[0m'], __read(args), false));
+};
+export var findVec4Constructor = function (ast) {
+    var parent;
+    var visitors = {
+        function_call: {
+            enter: function (path) {
+                var _a, _b;
+                if (((_a = path.node.identifier) === null || _a === void 0 ? void 0 : _a.specifier).token === 'vec4') {
+                    parent = (_b = path.findParent(function (p) { return 'right' in p.node; })) === null || _b === void 0 ? void 0 : _b.node;
+                    path.skip();
+                }
+            },
+        },
+    };
+    visit(ast, visitors);
+    return parent;
+};
+export var findAssignmentTo = function (ast, assignTo) {
+    var assign;
+    var visitors = {
+        expression_statement: {
+            enter: function (path) {
+                var _a, _b;
+                if (((_b = (_a = path.node.expression) === null || _a === void 0 ? void 0 : _a.left) === null || _b === void 0 ? void 0 : _b.identifier) === assignTo) {
+                    assign = path.node;
+                }
+                path.skip();
+            },
+        },
+    };
+    visit(ast, visitors);
+    return assign;
+};
+export var findDeclarationOf = function (ast, declarationOf) {
+    var declaration;
+    var visitors = {
+        declaration_statement: {
+            enter: function (path) {
+                var _a, _b;
+                var foundDecl = (_b = (_a = path.node.declaration) === null || _a === void 0 ? void 0 : _a.declarations) === null || _b === void 0 ? void 0 : _b.find(function (decl) { var _a; return ((_a = decl === null || decl === void 0 ? void 0 : decl.identifier) === null || _a === void 0 ? void 0 : _a.identifier) === declarationOf; });
+                if (foundDecl) {
+                    declaration = foundDecl;
+                }
+                path.skip();
+            },
+        },
+    };
+    visit(ast, visitors);
+    return declaration;
+};
+export var from2To3 = function (ast, stage) {
+    var glOut = 'fragmentColor';
+    // TODO: add this back in when there's only one after the merge
+    // ast.program.unshift({
+    //   type: 'preprocessor',
+    //   line: '#version 300 es',
+    //   _: '\n',
+    // });
+    if (stage === 'fragment') {
+        ast.program.unshift(makeStatement("out vec4 ".concat(glOut)));
+    }
+    visit(ast, {
+        function_call: {
+            enter: function (path) {
+                var identifier = path.node.identifier;
+                if (identifier.type === 'identifier' &&
+                    identifier.identifier === 'texture2D') {
+                    identifier.identifier = 'texture';
+                }
+            },
+        },
+        identifier: {
+            enter: function (path) {
+                if (path.node.identifier === 'gl_FragColor') {
+                    path.node.identifier = glOut;
+                }
+            },
+        },
+        keyword: {
+            enter: function (path) {
+                if ((path.node.token === 'attribute' || path.node.token === 'varying') &&
+                    path.findParent(function (path) { return path.node.type === 'declaration_statement'; })) {
+                    path.node.token =
+                        stage === 'vertex' && path.node.token === 'varying' ? 'out' : 'in';
+                }
+            },
+        },
+    });
+};
+export var outDeclaration = function (name) { return ({
+    type: 'declaration_statement',
+    declaration: {
+        type: 'declarator_list',
+        specified_type: {
+            type: 'fully_specified_type',
+            qualifiers: [{ type: 'keyword', token: 'out', whitespace: ' ' }],
+            specifier: {
+                type: 'type_specifier',
+                specifier: { type: 'keyword', token: 'vec4', whitespace: ' ' },
+                quantifier: null,
+            },
+        },
+        declarations: [
+            {
+                type: 'declaration',
+                identifier: {
+                    type: 'identifier',
+                    identifier: name,
+                    whitespace: undefined,
+                },
+                quantifier: null,
+                operator: undefined,
+                initializer: undefined,
+            },
+        ],
+        commas: [],
+    },
+    semi: { type: 'literal', literal: ';', whitespace: '\n    ' },
+}); };
+export var makeStatement = function (stmt) {
+    // log(`Parsing "${stmt}"`);
+    var ast;
+    try {
+        ast = parser.parse("".concat(stmt, ";\n"), { quiet: true });
+    }
+    catch (error) {
+        console.error({ stmt: stmt, error: error });
+        throw new Error("Error parsing stmt \"".concat(stmt, "\": ").concat(error === null || error === void 0 ? void 0 : error.message));
+    }
+    // log(util.inspect(ast, false, null, true));
+    return ast.program[0];
+};
+export var makeFnStatement = function (fnStmt) {
+    var ast;
+    try {
+        ast = parser.parse("\n  void main() {\n      ".concat(fnStmt, ";\n    }"), { quiet: true });
+    }
+    catch (error) {
+        console.error({ fnStmt: fnStmt, error: error });
+        throw new Error("Error parsing fnStmt \"".concat(fnStmt, "\": ").concat(error === null || error === void 0 ? void 0 : error.message));
+    }
+    // log(util.inspect(ast, false, null, true));
+    return ast.program[0].body.statements[0];
+};
+export var makeExpression = function (expr) {
+    var ast;
+    try {
+        ast = parser.parse("void main() {\n          a = ".concat(expr, ";\n        }"), { quiet: true });
+    }
+    catch (error) {
+        console.error({ expr: expr, error: error });
+        throw new Error("Error parsing expr \"".concat(expr, "\": ").concat(error === null || error === void 0 ? void 0 : error.message));
+    }
+    return ast.program[0].body
+        .statements[0].expression.right;
+};
+export var makeExpressionWithScopes = function (expr) {
+    var ast;
+    try {
+        ast = parser.parse("void main() {\n          ".concat(expr, ";\n        }"), { quiet: true });
+    }
+    catch (error) {
+        console.error({ expr: expr, error: error });
+        throw new Error("Error parsing expr \"".concat(expr, "\": ").concat(error === null || error === void 0 ? void 0 : error.message));
+    }
+    // log(util.inspect(ast, false, null, true));
+    return {
+        scope: ast.scopes[1],
+        expression: ast.program[0].body
+            .statements[0].expression,
+    };
+};
+export var makeFnBodyStatementWithScopes = function (body) {
+    var ast;
+    try {
+        ast = parser.parse("void main() {\n".concat(body, "\n        }"), { quiet: true });
+    }
+    catch (error) {
+        console.error({ body: body, error: error });
+        throw new Error("Error parsing body \"".concat(body, "\": ").concat(error === null || error === void 0 ? void 0 : error.message));
+    }
+    // log(util.inspect(ast, false, null, true));
+    return {
+        scope: ast.scopes[1],
+        statements: ast.program[0].body.statements,
+    };
+};
+export var findFn = function (ast, name) {
+    return ast.program.find(function (stmt) {
+        return stmt.type === 'function' && stmt.prototype.header.name.identifier === name;
+    });
+};
+export var returnGlPosition = function (fnName, ast) {
+    return convertVertexMain(fnName, ast, 'vec4', function (assign) { return assign.expression.right; });
+};
+export var returnGlPositionHardCoded = function (fnName, ast, returnType, hardCodedReturn) {
+    return convertVertexMain(fnName, ast, returnType, function () {
+        return makeExpression(hardCodedReturn);
+    });
+};
+export var returnGlPositionVec3Right = function (fnName, ast) {
+    return convertVertexMain(fnName, ast, 'vec3', function (assign) {
+        var found;
+        visit(assign, {
+            function_call: {
+                enter: function (path) {
+                    var _a, _b, _c, _d, _e;
+                    log('returnGlPositionVec3Right', path.node);
+                    var node = path.node;
+                    if (((_b = (_a = node === null || node === void 0 ? void 0 : node.identifier) === null || _a === void 0 ? void 0 : _a.specifier) === null || _b === void 0 ? void 0 : _b.token) === 'vec4' &&
+                        ((_e = (_d = (_c = node === null || node === void 0 ? void 0 : node.args) === null || _c === void 0 ? void 0 : _c[2]) === null || _d === void 0 ? void 0 : _d.token) === null || _e === void 0 ? void 0 : _e.includes('1.'))) {
+                        found = node.args[0];
+                    }
+                },
+            },
+        });
+        if (!found) {
+            console.error(generate(ast));
+            throw new Error('Could not find position assignment to convert to return!');
+        }
+        return found;
+    });
+};
+var convertVertexMain = function (fnName, ast, returnType, generateRight) {
+    var mainReturnVar = "frogOut";
+    var main = findFn(ast, fnName);
+    if (!main) {
+        throw new Error("No ".concat(fnName, " fn found!"));
+    }
+    // Convert the main function to one that returns
+    main.prototype.header.returnType.specifier.specifier.token =
+        returnType;
+    // Find the gl_position assignment line
+    var assign = main.body.statements.find(function (stmt) {
+        var _a;
+        return stmt.type === 'expression_statement' &&
+            ((_a = stmt.expression.left) === null || _a === void 0 ? void 0 : _a.identifier) === 'gl_Position';
+    });
+    if (!assign) {
+        throw new Error("No gl position assign found in main fn!");
+    }
+    var rtnStmt = makeFnStatement("".concat(returnType, " ").concat(mainReturnVar, " = 1.0"));
+    rtnStmt.declaration.declarations[0].initializer =
+        generateRight(assign);
+    main.body.statements.splice(main.body.statements.indexOf(assign), 1, rtnStmt);
+    main.body.statements.push(makeFnStatement("return ".concat(mainReturnVar)));
+};
+export var convert300MainToReturn = function (suffix, ast) {
+    var mainReturnVar = "frogOut_".concat(suffix);
+    // Find the output variable, as in "pc_fragColor" from  "out highp vec4 pc_fragColor;"
+    var outName;
+    ast.program.find(function (line, index) {
+        var _a, _b;
+        var declaration = line === null || line === void 0 ? void 0 : line.declaration;
+        if (
+        // line.type === 'declaration_statement' &&
+        ((_b = (_a = declaration === null || declaration === void 0 ? void 0 : declaration.specified_type) === null || _a === void 0 ? void 0 : _a.qualifiers) === null || _b === void 0 ? void 0 : _b.find(function (n) { return n.token === 'out'; })) &&
+            declaration.specified_type.specifier.specifier.token ===
+                'vec4') {
+            // Remove the out declaration
+            ast.program.splice(index, 1);
+            outName = declaration.declarations[0].identifier.identifier;
+            return true;
+        }
+    });
+    if (!outName) {
+        console.error(generate(ast));
+        throw new Error('No "out vec4" line found in the fragment shader');
+    }
+    ast.program.unshift(makeStatement("vec4 ".concat(mainReturnVar)));
+    visit(ast, {
+        identifier: {
+            enter: function (path) {
+                if (path.node.identifier === outName) {
+                    path.node.identifier = mainReturnVar;
+                    // @ts-ignore
+                    path.node.doNotDescope = true; // hack because this var is in the scope which gets renamed later
+                }
+            },
+        },
+        function: {
+            enter: function (path) {
+                if (path.node.prototype.header.name.identifier === 'main') {
+                    path.node.prototype.header.returnType.specifier
+                        .specifier.token = 'vec4';
+                    path.node.body.statements.push(makeFnStatement("return ".concat(mainReturnVar)));
+                }
+            },
+        },
+    });
+};
 export var generateFiller = function (filler) {
     if (!filler) {
         throw new Error('Cannot generate void filler!');

package/dist/util/ensure.d.ts

@@ -0,0 +1 @@
+export declare const ensure: <T>(argument: T, message?: string) => T;

package/dist/util/id.d.ts

@@ -0,0 +1 @@
+export declare const makeId: () => string;

package/package.json

@@ -1,8 +1,7 @@
 {
   "name": "@shaderfrog/core",
-  "version": "0.1.1",
+  "version": "1.0.0",
   "description": "Shaderfrog core",
-  "main": "index.js",
   "scripts": {
     "prepare": "npm run build",
     "build": "./build.sh",
@@ -42,8 +41,8 @@
   "peerDependencies": {
     "@shaderfrog/glsl-parser": "^2.0.0-beta.5",
     "babylonjs": ">=4",
-    "three": ">=0.50",
-    "playcanvas": "^1.65.3"
+    "playcanvas": "^1.65.3",
+    "three": ">=0.50"
   },
   "peerDependenciesMeta": {
     "babylonjs": {

package/dist/ast/manipulate.js

@@ -1,328 +0,0 @@
-var __read = (this && this.__read) || function (o, n) {
-    var m = typeof Symbol === "function" && o[Symbol.iterator];
-    if (!m) return o;
-    var i = m.call(o), r, ar = [], e;
-    try {
-        while ((n === void 0 || n-- > 0) && !(r = i.next()).done) ar.push(r.value);
-    }
-    catch (error) { e = { error: error }; }
-    finally {
-        try {
-            if (r && !r.done && (m = i["return"])) m.call(i);
-        }
-        finally { if (e) throw e.error; }
-    }
-    return ar;
-};
-var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
-    if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
-        if (ar || !(i in from)) {
-            if (!ar) ar = Array.prototype.slice.call(from, 0, i);
-            ar[i] = from[i];
-        }
-    }
-    return to.concat(ar || Array.prototype.slice.call(from));
-};
-/**
- * Utility functions to work with ASTs
- */
-import { parser, generate } from '@shaderfrog/glsl-parser';
-import { visit, } from '@shaderfrog/glsl-parser/ast';
-var log = function () {
-    var _a;
-    var args = [];
-    for (var _i = 0; _i < arguments.length; _i++) {
-        args[_i] = arguments[_i];
-    }
-    return (_a = console.log).call.apply(_a, __spreadArray([console, '\x1b[31m(core.manipulate)\x1b[0m'], __read(args), false));
-};
-export var findVec4Constructor = function (ast) {
-    var parent;
-    var visitors = {
-        function_call: {
-            enter: function (path) {
-                var _a, _b;
-                if (((_a = path.node.identifier) === null || _a === void 0 ? void 0 : _a.specifier).token === 'vec4') {
-                    parent = (_b = path.findParent(function (p) { return 'right' in p.node; })) === null || _b === void 0 ? void 0 : _b.node;
-                    path.skip();
-                }
-            },
-        },
-    };
-    visit(ast, visitors);
-    return parent;
-};
-export var findAssignmentTo = function (ast, assignTo) {
-    var assign;
-    var visitors = {
-        expression_statement: {
-            enter: function (path) {
-                var _a, _b;
-                if (((_b = (_a = path.node.expression) === null || _a === void 0 ? void 0 : _a.left) === null || _b === void 0 ? void 0 : _b.identifier) === assignTo) {
-                    assign = path.node;
-                }
-                path.skip();
-            },
-        },
-    };
-    visit(ast, visitors);
-    return assign;
-};
-export var findDeclarationOf = function (ast, declarationOf) {
-    var declaration;
-    var visitors = {
-        declaration_statement: {
-            enter: function (path) {
-                var _a, _b;
-                var foundDecl = (_b = (_a = path.node.declaration) === null || _a === void 0 ? void 0 : _a.declarations) === null || _b === void 0 ? void 0 : _b.find(function (decl) { var _a; return ((_a = decl === null || decl === void 0 ? void 0 : decl.identifier) === null || _a === void 0 ? void 0 : _a.identifier) === declarationOf; });
-                if (foundDecl) {
-                    declaration = foundDecl;
-                }
-                path.skip();
-            },
-        },
-    };
-    visit(ast, visitors);
-    return declaration;
-};
-export var from2To3 = function (ast, stage) {
-    var glOut = 'fragmentColor';
-    // TODO: add this back in when there's only one after the merge
-    // ast.program.unshift({
-    //   type: 'preprocessor',
-    //   line: '#version 300 es',
-    //   _: '\n',
-    // });
-    if (stage === 'fragment') {
-        ast.program.unshift(makeStatement("out vec4 ".concat(glOut)));
-    }
-    visit(ast, {
-        function_call: {
-            enter: function (path) {
-                var identifier = path.node.identifier;
-                if (identifier.type === 'identifier' &&
-                    identifier.identifier === 'texture2D') {
-                    identifier.identifier = 'texture';
-                }
-            },
-        },
-        identifier: {
-            enter: function (path) {
-                if (path.node.identifier === 'gl_FragColor') {
-                    path.node.identifier = glOut;
-                }
-            },
-        },
-        keyword: {
-            enter: function (path) {
-                if ((path.node.token === 'attribute' || path.node.token === 'varying') &&
-                    path.findParent(function (path) { return path.node.type === 'declaration_statement'; })) {
-                    path.node.token =
-                        stage === 'vertex' && path.node.token === 'varying' ? 'out' : 'in';
-                }
-            },
-        },
-    });
-};
-export var outDeclaration = function (name) { return ({
-    type: 'declaration_statement',
-    declaration: {
-        type: 'declarator_list',
-        specified_type: {
-            type: 'fully_specified_type',
-            qualifiers: [{ type: 'keyword', token: 'out', whitespace: ' ' }],
-            specifier: {
-                type: 'type_specifier',
-                specifier: { type: 'keyword', token: 'vec4', whitespace: ' ' },
-                quantifier: null,
-            },
-        },
-        declarations: [
-            {
-                type: 'declaration',
-                identifier: {
-                    type: 'identifier',
-                    identifier: name,
-                    whitespace: undefined,
-                },
-                quantifier: null,
-                operator: undefined,
-                initializer: undefined,
-            },
-        ],
-        commas: [],
-    },
-    semi: { type: 'literal', literal: ';', whitespace: '\n    ' },
-}); };
-export var makeStatement = function (stmt) {
-    // log(`Parsing "${stmt}"`);
-    var ast;
-    try {
-        ast = parser.parse("".concat(stmt, ";\n"), { quiet: true });
-    }
-    catch (error) {
-        console.error({ stmt: stmt, error: error });
-        throw new Error("Error parsing stmt \"".concat(stmt, "\": ").concat(error === null || error === void 0 ? void 0 : error.message));
-    }
-    // log(util.inspect(ast, false, null, true));
-    return ast.program[0];
-};
-export var makeFnStatement = function (fnStmt) {
-    var ast;
-    try {
-        ast = parser.parse("\n  void main() {\n      ".concat(fnStmt, ";\n    }"), { quiet: true });
-    }
-    catch (error) {
-        console.error({ fnStmt: fnStmt, error: error });
-        throw new Error("Error parsing fnStmt \"".concat(fnStmt, "\": ").concat(error === null || error === void 0 ? void 0 : error.message));
-    }
-    // log(util.inspect(ast, false, null, true));
-    return ast.program[0].body.statements[0];
-};
-export var makeExpression = function (expr) {
-    var ast;
-    try {
-        ast = parser.parse("void main() {\n          a = ".concat(expr, ";\n        }"), { quiet: true });
-    }
-    catch (error) {
-        console.error({ expr: expr, error: error });
-        throw new Error("Error parsing expr \"".concat(expr, "\": ").concat(error === null || error === void 0 ? void 0 : error.message));
-    }
-    return ast.program[0].body
-        .statements[0].expression.right;
-};
-export var makeExpressionWithScopes = function (expr) {
-    var ast;
-    try {
-        ast = parser.parse("void main() {\n          ".concat(expr, ";\n        }"), { quiet: true });
-    }
-    catch (error) {
-        console.error({ expr: expr, error: error });
-        throw new Error("Error parsing expr \"".concat(expr, "\": ").concat(error === null || error === void 0 ? void 0 : error.message));
-    }
-    // log(util.inspect(ast, false, null, true));
-    return {
-        scope: ast.scopes[1],
-        expression: ast.program[0].body
-            .statements[0].expression,
-    };
-};
-export var makeFnBodyStatementWithScopes = function (body) {
-    var ast;
-    try {
-        ast = parser.parse("void main() {\n".concat(body, "\n        }"), { quiet: true });
-    }
-    catch (error) {
-        console.error({ body: body, error: error });
-        throw new Error("Error parsing body \"".concat(body, "\": ").concat(error === null || error === void 0 ? void 0 : error.message));
-    }
-    // log(util.inspect(ast, false, null, true));
-    return {
-        scope: ast.scopes[1],
-        statements: ast.program[0].body.statements,
-    };
-};
-export var findFn = function (ast, name) {
-    return ast.program.find(function (stmt) {
-        return stmt.type === 'function' && stmt.prototype.header.name.identifier === name;
-    });
-};
-export var returnGlPosition = function (fnName, ast) {
-    return convertVertexMain(fnName, ast, 'vec4', function (assign) { return assign.expression.right; });
-};
-export var returnGlPositionHardCoded = function (fnName, ast, returnType, hardCodedReturn) {
-    return convertVertexMain(fnName, ast, returnType, function () {
-        return makeExpression(hardCodedReturn);
-    });
-};
-export var returnGlPositionVec3Right = function (fnName, ast) {
-    return convertVertexMain(fnName, ast, 'vec3', function (assign) {
-        var found;
-        visit(assign, {
-            function_call: {
-                enter: function (path) {
-                    var _a, _b, _c, _d, _e;
-                    log('returnGlPositionVec3Right', path.node);
-                    var node = path.node;
-                    if (((_b = (_a = node === null || node === void 0 ? void 0 : node.identifier) === null || _a === void 0 ? void 0 : _a.specifier) === null || _b === void 0 ? void 0 : _b.token) === 'vec4' &&
-                        ((_e = (_d = (_c = node === null || node === void 0 ? void 0 : node.args) === null || _c === void 0 ? void 0 : _c[2]) === null || _d === void 0 ? void 0 : _d.token) === null || _e === void 0 ? void 0 : _e.includes('1.'))) {
-                        found = node.args[0];
-                    }
-                },
-            },
-        });
-        if (!found) {
-            console.error(generate(ast));
-            throw new Error('Could not find position assignment to convert to return!');
-        }
-        return found;
-    });
-};
-var convertVertexMain = function (fnName, ast, returnType, generateRight) {
-    var mainReturnVar = "frogOut";
-    var main = findFn(ast, fnName);
-    if (!main) {
-        throw new Error("No ".concat(fnName, " fn found!"));
-    }
-    // Convert the main function to one that returns
-    main.prototype.header.returnType.specifier.specifier.token =
-        returnType;
-    // Find the gl_position assignment line
-    var assign = main.body.statements.find(function (stmt) {
-        var _a;
-        return stmt.type === 'expression_statement' &&
-            ((_a = stmt.expression.left) === null || _a === void 0 ? void 0 : _a.identifier) === 'gl_Position';
-    });
-    if (!assign) {
-        throw new Error("No gl position assign found in main fn!");
-    }
-    var rtnStmt = makeFnStatement("".concat(returnType, " ").concat(mainReturnVar, " = 1.0"));
-    rtnStmt.declaration.declarations[0].initializer =
-        generateRight(assign);
-    main.body.statements.splice(main.body.statements.indexOf(assign), 1, rtnStmt);
-    main.body.statements.push(makeFnStatement("return ".concat(mainReturnVar)));
-};
-export var convert300MainToReturn = function (suffix, ast) {
-    var mainReturnVar = "frogOut_".concat(suffix);
-    // Find the output variable, as in "pc_fragColor" from  "out highp vec4 pc_fragColor;"
-    var outName;
-    ast.program.find(function (line, index) {
-        var _a, _b;
-        var declaration = line === null || line === void 0 ? void 0 : line.declaration;
-        if (
-        // line.type === 'declaration_statement' &&
-        ((_b = (_a = declaration === null || declaration === void 0 ? void 0 : declaration.specified_type) === null || _a === void 0 ? void 0 : _a.qualifiers) === null || _b === void 0 ? void 0 : _b.find(function (n) { return n.token === 'out'; })) &&
-            declaration.specified_type.specifier.specifier.token ===
-                'vec4') {
-            // Remove the out declaration
-            ast.program.splice(index, 1);
-            outName = declaration.declarations[0].identifier.identifier;
-            return true;
-        }
-    });
-    if (!outName) {
-        console.error(generate(ast));
-        throw new Error('No "out vec4" line found in the fragment shader');
-    }
-    ast.program.unshift(makeStatement("vec4 ".concat(mainReturnVar)));
-    visit(ast, {
-        identifier: {
-            enter: function (path) {
-                if (path.node.identifier === outName) {
-                    path.node.identifier = mainReturnVar;
-                    // @ts-ignore
-                    path.node.doNotDescope = true; // hack because this var is in the scope which gets renamed later
-                }
-            },
-        },
-        function: {
-            enter: function (path) {
-                if (path.node.prototype.header.name.identifier === 'main') {
-                    path.node.prototype.header.returnType.specifier
-                        .specifier.token = 'vec4';
-                    path.node.body.statements.push(makeFnStatement("return ".concat(mainReturnVar)));
-                }
-            },
-        },
-    });
-};