feat: code opt

Author: Sway007

packages/shader-lab/src/codeGen/GLESVisitor.ts

@@ -45,15 +45,15 @@ export abstract class GLESVisitor extends CodeGenVisitor {
 
   vertexMain(entry: string, data: ShaderData): string {
     const { symbolTable } = data;
-    const fnSymbol = symbolTable.lookupBy(entry, ESymbolType.FN);
+    const fnSymbol = symbolTable.lookup(entry, ESymbolType.FN);
     if (!fnSymbol?.astNode) throw `no entry function found: ${entry}`;
 
     const fnNode = fnSymbol.astNode;
     VisitorContext.context.stage = EShaderStage.VERTEX;
 
     const returnType = fnNode.protoType.returnType;
     if (typeof returnType.type === "string") {
-      const varyStruct = symbolTable.lookupBy(returnType.type, ESymbolType.STRUCT);
+      const varyStruct = symbolTable.lookup(returnType.type, ESymbolType.STRUCT);
       if (!varyStruct) {
         this._reportError(returnType.location, `invalid varying struct: ${returnType.type}`);
       } else {
@@ -67,7 +67,7 @@ export abstract class GLESVisitor extends CodeGenVisitor {
     if (paramList?.length) {
       for (const paramInfo of paramList) {
         if (typeof paramInfo.typeInfo.type === "string") {
-          const structSymbol = symbolTable.lookupBy(paramInfo.typeInfo.type, ESymbolType.STRUCT);
+          const structSymbol = symbolTable.lookup(paramInfo.typeInfo.type, ESymbolType.STRUCT);
           if (!structSymbol) {
             this._reportError(paramInfo.astNode.location, `Not found attribute struct "${paramInfo.typeInfo.type}".`);
             continue;
@@ -103,7 +103,7 @@ export abstract class GLESVisitor extends CodeGenVisitor {
 
   private _fragmentMain(entry: string, data: ShaderData): string {
     const { symbolTable } = data;
-    const fnSymbol = symbolTable.lookupBy(entry, ESymbolType.FN);
+    const fnSymbol = symbolTable.lookup(entry, ESymbolType.FN);
     if (!fnSymbol?.astNode) throw `no entry function found: ${entry}`;
     const fnNode = fnSymbol.astNode;
 
@@ -117,7 +117,7 @@ export abstract class GLESVisitor extends CodeGenVisitor {
 
     const { type: returnDataType, location: returnLocation } = fnNode.protoType.returnType;
     if (typeof returnDataType === "string") {
-      const mrtStruct = symbolTable.lookupBy(returnDataType, ESymbolType.STRUCT);
+      const mrtStruct = symbolTable.lookup(returnDataType, ESymbolType.STRUCT);
       if (!mrtStruct) {
         this._reportError(returnLocation, `invalid mrt struct: ${returnDataType}`);
       } else {

packages/shader-lab/src/codeGen/VisitorContext.ts

@@ -121,7 +121,7 @@ export class VisitorContext {
       }
       return;
     }
-    const sm = this._passSymbolTable.lookupBy(ident, type);
+    const sm = this._passSymbolTable.lookup(ident, type);
     if (sm) {
       this._referencedGlobals[ident] = sm;
     }

packages/shader-lab/src/contentParser/ContentSymbolTable.ts

@@ -21,7 +21,7 @@ export default class ContentSymbolTable extends BaseSymbolTable<ISymbol> {
     this._table.set(sm.ident, entry);
   }
 
-  lookupByType(ident: string, type: TokenType): ISymbol | undefined {
+  lookup(ident: string, type: TokenType): ISymbol | undefined {
     const entry = this._table.get(ident);
     if (entry) {
       for (let length = entry.length, i = 0; i < length; i++) {

packages/shader-lab/src/contentParser/ShaderContentParser.ts

@@ -117,7 +117,7 @@ export class ShaderContentParser {
     const stack = ShaderContentParser._symbolTableStack.stack;
     for (let length = stack.length, i = length - 1; i >= 0; i--) {
       const symbolTable = stack[i];
-      const ret = symbolTable.lookupByType(ident, type);
+      const ret = symbolTable.lookup(ident, type);
       if (ret) return ret;
     }
   }

packages/shader-lab/src/parser/SemanticAnalyzer.ts

@@ -80,7 +80,7 @@ export default class SematicAnalyzer {
     const stack = this.symbolTableStack.stack;
     for (let length = stack.length, i = length - 1; i >= 0; i--) {
       const symbolTable = stack[i];
-      const ret = symbolTable.lookupBy(ident, symbolType, signature, astNode);
+      const ret = symbolTable.lookup(ident, symbolType, signature, astNode);
       if (ret) return ret;
     }
   }

packages/shader-lab/src/parser/symbolTable/SymbolTable.ts

@@ -21,7 +21,7 @@ export class SymbolTable extends BaseSymbolTable<SymbolInfo> {
     this._table.set(sm.ident, entry);
   }
 
-  lookupBy<T extends ESymbolType>(
+  lookup<T extends ESymbolType>(
     ident: string,
     symbolType: T,
     signature?: GalaceanDataType[],