visualizer/ast-visualizer.spice
| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | // Copyright (c) 2021-2026 ChilliBits. All rights reserved. | ||
| 2 | |||
| 3 | // Std imports | ||
| 4 | import "std/type/any"; | ||
| 5 | import "std/text/stringstream"; | ||
| 6 | |||
| 7 | // Own imports | ||
| 8 | import "@/ast/abstract-ast-visitor-intf"; | ||
| 9 | import "@/ast/ast-nodes"; | ||
| 10 | import "@/util/save-and-restore"; | ||
| 11 | |||
| 12 | // Generic type defs | ||
| 13 | type T dyn; | ||
| 14 | |||
| 15 | /** | ||
| 16 | * Visitor for debug purposes (is only executed in the compiler debug mode and when explicitly enabling it via cli flag) | ||
| 17 | * | ||
| 18 | * Jobs: | ||
| 19 | * - Visualize AST | ||
| 20 | */ | ||
| 21 | 7864 | public type ASTVisualizer struct : IAbstractAstVisitor { | |
| 22 | String parentNodeId | ||
| 23 | } | ||
| 24 | |||
| 25 | // ─── private helpers ────────────────────────────────────────────────────────── | ||
| 26 | |||
| 27 | 2219346 | f<Any> ASTVisualizer.buildNode<T>(const T* node) { | |
| 28 | 2219346 | StringStream output; | |
| 29 | |||
| 30 | // Prepare strings | ||
| 31 | 2219346 | const dyn typeName = String(typename<T>()); | |
| 32 | 2219346 | String codeLocStr = node.codeLoc.toString(); | |
| 33 | 2219346 | const String nodeName = typeName.getSubstring(typeName.find("AST") + 3l); | |
| 34 | 2219346 | const String nodeId = codeLocStr + "_" + nodeName; | |
| 35 | |||
| 36 | // Build result | ||
| 37 | 2219346 | output << nodeId << " [color=\"lightgreen\",label=\"" << nodeName << "\"];\n"; | |
| 38 | |||
| 39 | // Link parent node with the current one | ||
| 40 | 2219346 | if !this.parentNodeId.isEmpty() { | |
| 41 | 2215414 | output << " " << this.parentNodeId << " -> " << nodeId << ";\n"; | |
| 42 | 2215414 | } | |
| 43 | |||
| 44 | // Set parentNodeId for children | ||
| 45 | 2219346 | SaveAndRestore<String>(this.parentNodeId, nodeId); | |
| 46 | |||
| 47 | // Visit all the children. A plain ASTNode* cannot be dispatched on, so this goes through its IVisitable pointer | ||
| 48 | 4434760 | foreach ASTNode* child : node.children { | |
| 49 | 2215414 | assert child != nil<ASTNode*>; | |
| 50 | 2215414 | assert child.visitable != nil<IVisitable*>; | |
| 51 | 2215414 | Any childDotCode = child.visitable.accept(cast<IAbstractAstVisitor*>(this)); | |
| 52 | 2215414 | output << " " << childDotCode.get<String>(); | |
| 53 | 2215414 | } | |
| 54 | |||
| 55 | 4438692 | return Any(output.str()); | |
| 56 | 2219346 | } | |
| 57 | |||
| 58 | // ─── visitor methods ────────────────────────────────────────────────────────── | ||
| 59 | |||
| 60 | 3932 | public f<Any> ASTVisualizer.visitEntry(ASTEntryNode* node) { return this.buildNode(node); } | |
| 61 | 1024 | public f<Any> ASTVisualizer.visitMainFctDef(ASTMainFctDefNode* node) { return this.buildNode(node); } | |
| 62 | 27984 | public f<Any> ASTVisualizer.visitFctDef(ASTFctDefNode* node) { return this.buildNode(node); } | |
| 63 | 17563 | public f<Any> ASTVisualizer.visitProcDef(ASTProcDefNode* node) { return this.buildNode(node); } | |
| 64 | 45547 | public f<Any> ASTVisualizer.visitFctName(ASTFctNameNode* node) { return this.buildNode(node); } | |
| 65 | 3214 | public f<Any> ASTVisualizer.visitStructDef(ASTStructDefNode* node) { return this.buildNode(node); } | |
| 66 | 323 | public f<Any> ASTVisualizer.visitInterfaceDef(ASTInterfaceDefNode* node) { return this.buildNode(node); } | |
| 67 | 24 | public f<Any> ASTVisualizer.visitUnionDef(ASTUnionDefNode* node) { return this.buildNode(node); } | |
| 68 | 465 | public f<Any> ASTVisualizer.visitEnumDef(ASTEnumDefNode* node) { return this.buildNode(node); } | |
| 69 | 3055 | public f<Any> ASTVisualizer.visitGenericTypeDef(ASTGenericTypeDefNode* node) { return this.buildNode(node); } | |
| 70 | 159 | public f<Any> ASTVisualizer.visitAliasDef(ASTAliasDefNode* node) { return this.buildNode(node); } | |
| 71 | 4263 | public f<Any> ASTVisualizer.visitGlobalVarDef(ASTGlobalVarDefNode* node) { return this.buildNode(node); } | |
| 72 | 5206 | public f<Any> ASTVisualizer.visitExtDecl(ASTExtDeclNode* node) { return this.buildNode(node); } | |
| 73 | 2502 | public f<Any> ASTVisualizer.visitImportDef(ASTImportDefNode* node) { return this.buildNode(node); } | |
| 74 | 12690 | public f<Any> ASTVisualizer.visitUnsafeBlock(ASTUnsafeBlockNode* node) { return this.buildNode(node); } | |
| 75 | 4385 | public f<Any> ASTVisualizer.visitForLoop(ASTForLoopNode* node) { return this.buildNode(node); } | |
| 76 | 434 | public f<Any> ASTVisualizer.visitForeachLoop(ASTForeachLoopNode* node) { return this.buildNode(node); } | |
| 77 | 1852 | public f<Any> ASTVisualizer.visitWhileLoop(ASTWhileLoopNode* node) { return this.buildNode(node); } | |
| 78 | 50 | public f<Any> ASTVisualizer.visitDoWhileLoop(ASTDoWhileLoopNode* node) { return this.buildNode(node); } | |
| 79 | 24991 | public f<Any> ASTVisualizer.visitIfStmt(ASTIfStmtNode* node) { return this.buildNode(node); } | |
| 80 | 1376 | public f<Any> ASTVisualizer.visitElseStmt(ASTElseStmtNode* node) { return this.buildNode(node); } | |
| 81 | 38 | public f<Any> ASTVisualizer.visitSwitchStmt(ASTSwitchStmtNode* node) { return this.buildNode(node); } | |
| 82 | 195 | public f<Any> ASTVisualizer.visitCaseBranch(ASTCaseBranchNode* node) { return this.buildNode(node); } | |
| 83 | 20 | public f<Any> ASTVisualizer.visitDefaultBranch(ASTDefaultBranchNode* node) { return this.buildNode(node); } | |
| 84 | 4077 | public f<Any> ASTVisualizer.visitAssertStmt(ASTAssertStmtNode* node) { return this.buildNode(node); } | |
| 85 | 126 | public f<Any> ASTVisualizer.visitAnonymousBlockStmt(ASTAnonymousBlockStmtNode* node) { return this.buildNode(node); } | |
| 86 | 92282 | public f<Any> ASTVisualizer.visitStmtLst(ASTStmtLstNode* node) { return this.buildNode(node); } | |
| 87 | 30422 | public f<Any> ASTVisualizer.visitTypeLst(ASTTypeLstNode* node) { return this.buildNode(node); } | |
| 88 | 5134 | public f<Any> ASTVisualizer.visitTypeLstWithEllipsis(ASTTypeLstWithEllipsisNode* node) { return this.buildNode(node); } | |
| 89 | 3055 | public f<Any> ASTVisualizer.visitTypeAltsLst(ASTTypeAltsLstNode* node) { return this.buildNode(node); } | |
| 90 | 33023 | public f<Any> ASTVisualizer.visitParamLst(ASTParamLstNode* node) { return this.buildNode(node); } | |
| 91 | 61338 | public f<Any> ASTVisualizer.visitArgLst(ASTArgLstNode* node) { return this.buildNode(node); } | |
| 92 | 465 | public f<Any> ASTVisualizer.visitEnumItemLst(ASTEnumItemLstNode* node) { return this.buildNode(node); } | |
| 93 | 1747 | public f<Any> ASTVisualizer.visitEnumItem(ASTEnumItemNode* node) { return this.buildNode(node); } | |
| 94 | 7808 | public f<Any> ASTVisualizer.visitField(ASTFieldNode* node) { return this.buildNode(node); } | |
| 95 | 742 | public f<Any> ASTVisualizer.visitSignature(ASTSignatureNode* node) { return this.buildNode(node); } | |
| 96 | 83401 | public f<Any> ASTVisualizer.visitDeclStmt(ASTDeclStmtNode* node) { return this.buildNode(node); } | |
| 97 | 63761 | public f<Any> ASTVisualizer.visitExprStmt(ASTExprStmtNode* node) { return this.buildNode(node); } | |
| 98 | 124973 | public f<Any> ASTVisualizer.visitQualifierLst(ASTQualifierLstNode* node) { return this.buildNode(node); } | |
| 99 | 149861 | public f<Any> ASTVisualizer.visitQualifier(ASTQualifierNode* node) { return this.buildNode(node); } | |
| 100 | 1531 | public f<Any> ASTVisualizer.visitModAttr(ASTModAttrNode* node) { return this.buildNode(node); } | |
| 101 | 1567 | public f<Any> ASTVisualizer.visitTopLevelDefinitionAttr(ASTTopLevelDefAttrNode* node) { return this.buildNode(node); } | |
| 102 | 21 | public f<Any> ASTVisualizer.visitLambdaAttr(ASTLambdaAttrNode* node) { return this.buildNode(node); } | |
| 103 | 3119 | public f<Any> ASTVisualizer.visitAttrLst(ASTAttrLstNode* node) { return this.buildNode(node); } | |
| 104 | 3585 | public f<Any> ASTVisualizer.visitAttr(ASTAttrNode* node) { return this.buildNode(node); } | |
| 105 | 200 | public f<Any> ASTVisualizer.visitCaseConstant(ASTCaseConstantNode* node) { return this.buildNode(node); } | |
| 106 | 40013 | public f<Any> ASTVisualizer.visitReturnStmt(ASTReturnStmtNode* node) { return this.buildNode(node); } | |
| 107 | 737 | public f<Any> ASTVisualizer.visitBreakStmt(ASTBreakStmtNode* node) { return this.buildNode(node); } | |
| 108 | 757 | public f<Any> ASTVisualizer.visitContinueStmt(ASTContinueStmtNode* node) { return this.buildNode(node); } | |
| 109 | 16 | public f<Any> ASTVisualizer.visitFallthroughStmt(ASTFallthroughStmtNode* node) { return this.buildNode(node); } | |
| 110 | 30115 | public f<Any> ASTVisualizer.visitAssignExpr(ASTAssignExprNode* node) { return this.buildNode(node); } | |
| 111 | 2586 | public f<Any> ASTVisualizer.visitTernaryExpr(ASTTernaryExprNode* node) { return this.buildNode(node); } | |
| 112 | 4029 | public f<Any> ASTVisualizer.visitLogicalOrExpr(ASTLogicalOrExprNode* node) { return this.buildNode(node); } | |
| 113 | 2112 | public f<Any> ASTVisualizer.visitLogicalAndExpr(ASTLogicalAndExprNode* node) { return this.buildNode(node); } | |
| 114 | 428 | public f<Any> ASTVisualizer.visitBitwiseOrExpr(ASTBitwiseOrExprNode* node) { return this.buildNode(node); } | |
| 115 | 77 | public f<Any> ASTVisualizer.visitBitwiseXorExpr(ASTBitwiseXorExprNode* node) { return this.buildNode(node); } | |
| 116 | 260 | public f<Any> ASTVisualizer.visitBitwiseAndExpr(ASTBitwiseAndExprNode* node) { return this.buildNode(node); } | |
| 117 | 26959 | public f<Any> ASTVisualizer.visitEqualityExpr(ASTEqualityExprNode* node) { return this.buildNode(node); } | |
| 118 | 16330 | public f<Any> ASTVisualizer.visitRelationalExpr(ASTRelationalExprNode* node) { return this.buildNode(node); } | |
| 119 | 607 | public f<Any> ASTVisualizer.visitShiftExpr(ASTShiftExprNode* node) { return this.buildNode(node); } | |
| 120 | 14912 | public f<Any> ASTVisualizer.visitAdditiveExpr(ASTAdditiveExprNode* node) { return this.buildNode(node); } | |
| 121 | 3321 | public f<Any> ASTVisualizer.visitMultiplicativeExpr(ASTMultiplicativeExprNode* node) { return this.buildNode(node); } | |
| 122 | 12381 | public f<Any> ASTVisualizer.visitCastExpr(ASTCastExprNode* node) { return this.buildNode(node); } | |
| 123 | 6979 | public f<Any> ASTVisualizer.visitPrefixUnaryExpr(ASTPrefixUnaryExprNode* node) { return this.buildNode(node); } | |
| 124 | 105077 | public f<Any> ASTVisualizer.visitPostfixUnaryExpr(ASTPostfixUnaryExprNode* node) { return this.buildNode(node); } | |
| 125 | 396433 | public f<Any> ASTVisualizer.visitAtomicExpr(ASTAtomicExprNode* node) { return this.buildNode(node); } | |
| 126 | 90647 | public f<Any> ASTVisualizer.visitValue(ASTValueNode* node) { return this.buildNode(node); } | |
| 127 | 80488 | public f<Any> ASTVisualizer.visitConstant(ASTConstantNode* node) { return this.buildNode(node); } | |
| 128 | 80547 | public f<Any> ASTVisualizer.visitFctCall(ASTFctCallNode* node) { return this.buildNode(node); } | |
| 129 | 180 | public f<Any> ASTVisualizer.visitArrayInitialization(ASTArrayInitializationNode* node) { return this.buildNode(node); } | |
| 130 | 1535 | public f<Any> ASTVisualizer.visitStructInstantiation(ASTStructInstantiationNode* node) { return this.buildNode(node); } | |
| 131 | 67 | public f<Any> ASTVisualizer.visitLambdaFunc(ASTLambdaFuncNode* node) { return this.buildNode(node); } | |
| 132 | 73 | public f<Any> ASTVisualizer.visitLambdaProc(ASTLambdaProcNode* node) { return this.buildNode(node); } | |
| 133 | 4 | public f<Any> ASTVisualizer.visitLambdaExpr(ASTLambdaExprNode* node) { return this.buildNode(node); } | |
| 134 | 197452 | public f<Any> ASTVisualizer.visitDataType(ASTDataTypeNode* node) { return this.buildNode(node); } | |
| 135 | 197452 | public f<Any> ASTVisualizer.visitBaseDataType(ASTBaseDataTypeNode* node) { return this.buildNode(node); } | |
| 136 | 72537 | public f<Any> ASTVisualizer.visitCustomDataType(ASTCustomDataTypeNode* node) { return this.buildNode(node); } | |
| 137 | 705 | public f<Any> ASTVisualizer.visitFunctionDataType(ASTFunctionDataTypeNode* node) { return this.buildNode(node); } | |
| 138 |