Эхлэл Бүгдийг харах Тусламж
Нэвтрэх
beef
/
beefytech.Beef
-ын хуулбар https://github.com/beefytech/Beef
2
0
Салаа 0
Файлууд Асуудлууд 0 Мэдлэгийн сан
Мод: 22ec4a86b8
Салаанууд Тагууд
beefytech.Beef
/
IDEHelper
/
Compiler
/
BfPrinter.h

BfPrinter.h 8.6 KB
Түүх Анхны өгөгдөл

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207 
  1. #include "BeefySysLib/Common.h"
    2. 

  2. //#include "llvm/Support/raw_ostream.h"
    3. 

  3. #include "BfAst.h"
    4. 

  4. #include "BfElementVisitor.h"
    5. 

  5. 

  6. NS_BF_BEGIN
    7. 

  7. 

  8. 

  9. class BfPrinter : public BfElementVisitor
    10. 

  10. {
    11. 

  11. public:
    12. 

  12. 	struct StateModify
    13. 

  13. 	{
    14. 

  14. 	public:
    15. 

  15. 		bool mExpectingSpace;		
    16. 

  16. 		int mWantNewLineIdx;
    17. 

  17. 		bool mDoingBlockOpen;
    18. 

  18. 		bool mDoingBlockClose;		
    19. 

  19. 		int mWantVirtualIndent;
    20. 

  20. 		BfAstNode* mQueuedNode;
    21. 

  21. 

  22. 	public:
    23. 

  23. 		StateModify()
    24. 

  24. 		{
    25. 

  25. 			Clear();
    26. 

  26. 			mWantVirtualIndent = 0;
    27. 

  27. 			mWantNewLineIdx = 0;
    28. 

  28. 		}
    29. 

  29. 

  30. 		void Clear()
    31. 

  31. 		{
    32. 

  32. 			mExpectingSpace = false;			
    33. 

  33. 			mDoingBlockOpen = false;
    34. 

  34. 			mDoingBlockClose = false;
    35. 

  35. 			mQueuedNode = NULL;
    36. 

  36. 		}
    37. 

  37. 	};
    38. 

  38. 	
    39. 

  39. 	BfSourceData* mSource;
    40. 

  40. 	BfParserData* mParser;
    41. 

  41. 

  42. 	BfBlock::Iterator mSidechannelItr;
    43. 

  43. 	BfAstNode* mSidechannelNextNode;	
    44. 

  44. 

  45. 	BfBlock::Iterator mErrorItr;
    46. 

  46. 	BfAstNode* mErrorNextNode;
    47. 

  47. 	BfAstNode* mCurBlockMember;
    48. 

  48. 	BfTypeDeclaration* mCurTypeDecl;
    49. 

  49. 

  50. 	int mTriviaIdx;
    51. 

  51. 	int mCurSrcIdx;
    52. 

  52. 	Array<StateModify> mChildNodeQueue;
    53. 

  53. 	int mFormatStart;
    54. 

  54. 	int mFormatEnd;	
    55. 

  55. 	StateModify mNextStateModify;
    56. 

  56. 

  57. 	String mOutString;
    58. 

  58. 	bool mReformatting;		
    59. 

  59. 	bool mIgnoreTrivia;
    60. 

  60. 	bool mDocPrep;
    61. 

  61. 	int mCurIndentLevel;
    62. 

  62. 	int mQueuedSpaceCount;
    63. 

  63. 	int mLastSpaceOffset; // Indent difference from original to new	
    64. 

  64. 	bool mExpectingNewLine;
    65. 

  65. 

  66. 	bool mIsFirstStatementInBlock;
    67. 

  67. 	bool mForceUseTrivia;
    68. 

  68. 	bool mInSideChannel;	
    69. 

  69. 

  70. 	int mStateModifyVirtualIndentLevel;
    71. 

  71. 	int mVirtualIndentLevel;
    72. 

  72. 	int mVirtualNewLineIdx;
    73. 

  73. 			
    74. 

  74. 	Array<int>* mCharMapping;
    75. 

  75. 	int mHighestCharId;
    76. 

  76. 

  77. public:	
    78. 

  78. 	BfPrinter(BfRootNode* rootNode, BfRootNode* sidechannelRootNode, BfRootNode* errorRootNode);
    79. 

  79. 		
    80. 

  80. public:	
    81. 

  81. 	bool CheckReplace(BfAstNode* astNode);
    82. 

  82. 	void FlushIndent();
    83. 

  83. 	void Write(const StringImpl& str);
    84. 

  84. 	void Write(BfAstNode* node, int start, int len);
    85. 

  85. 	void WriteSourceString(BfAstNode* node);
    86. 

  86. 	void QueueVisitChild(BfAstNode* astNode);
    87. 

  87. 	void QueueVisitErrorNodes(BfRootNode* astNode);	
    88. 

  88. 	void FlushVisitChild();
    89. 

  89. 	void VisitChildWithPrecedingSpace(BfAstNode* bfAstNode);
    90. 

  90. 	void VisitChildWithProceedingSpace(BfAstNode* bfAstNode);
    91. 

  91. 	void ExpectSpace();
    92. 

  92. 	void ExpectNewLine();
    93. 

  93. 	void ExpectIndent();
    94. 

  94. 	void ExpectUnindent();
    95. 

  95. 	void VisitChildNextLine(BfAstNode* node);
    96. 

  96. 	void DoBlockOpen(BfBlock* block, bool queue, bool* outDoInlineBlock);
    97. 

  97. 	void DoBlockClose(BfBlock* block, bool queue, bool doInlineBlock);
    98. 

  98. 	void QueueMethodDeclaration(BfMethodDeclaration* methodDeclaration);	
    99. 

  99. 	int CalcOrigLineSpacing(BfAstNode* bfAstNode, int* lineStartIdx);
    100. 

  100. 	void WriteIgnoredNode(BfAstNode* node);
    101. 

  101. 

  102. 	virtual void Visit(BfAstNode* bfAstNode) override;
    103. 

  103. 	virtual void Visit(BfErrorNode* bfErrorNode) override;
    104. 

  104. 	virtual void Visit(BfExpression* expr) override;
    105. 

  105. 	virtual void Visit(BfExpressionStatement* exprStmt) override;
    106. 

  106. 	virtual void Visit(BfAttributedExpression* attribExpr) override;
    107. 

  107. 	virtual void Visit(BfStatement* stmt) override;	
    108. 

  108. 	virtual void Visit(BfLabelableStatement* labelableStmt) override;
    109. 

  109. 

  110. 	virtual void Visit(BfCommentNode* commentNode) override;
    111. 

  111. 	virtual void Visit(BfPreprocesorIgnoredSectionNode* preprocesorIgnoredSection) override;
    112. 

  112. 	virtual void Visit(BfPreprocessorNode* preprocessorNode) override;
    113. 

  113. 

  114. 	virtual void Visit(BfAttributeDirective* attributeDirective) override;	
    115. 

  115. 	virtual void Visit(BfGenericParamsDeclaration* genericParams) override;
    116. 

  116. 	virtual void Visit(BfGenericConstraintsDeclaration* genericConstraints) override;
    117. 

  117. 	virtual void Visit(BfGenericArgumentsNode* genericArgumentsNode) override;
    118. 

  118. 

  119. 	virtual void Visit(BfEmptyStatement* emptyStmt) override;	
    120. 

  120. 	virtual void Visit(BfTokenNode* tokenNode) override;
    121. 

  121. 	virtual void Visit(BfLiteralExpression* literalExpr) override;
    122. 

  122. 	virtual void Visit(BfIdentifierNode* identifierNode) override;
    123. 

  123. 	virtual void Visit(BfQualifiedNameNode* nameNode) override;
    124. 

  124. 	virtual void Visit(BfThisExpression* thisExpr) override;
    125. 

  125. 	virtual void Visit(BfBaseExpression* baseExpr) override;
    126. 

  126. 	virtual void Visit(BfMixinExpression* mixinExpr) override;
    127. 

  127. 	virtual void Visit(BfSizedArrayCreateExpression* createExpr) override;
    128. 

  128. 	virtual void Visit(BfCollectionInitializerExpression* initExpr) override;
    129. 

  129. 	virtual void Visit(BfArraySizeSpecifier* arraySizeSpecifier) override;
    130. 

  130. 	virtual void Visit(BfTypeReference* typeRef) override;
    131. 

  131. 	virtual void Visit(BfNamedTypeReference* typeRef) override;
    132. 

  132. 	virtual void Visit(BfQualifiedTypeReference* qualifiedType) override;
    133. 

  133. 	virtual void Visit(BfVarTypeReference* typeRef) override;
    134. 

  134. 	virtual void Visit(BfLetTypeReference * typeRef) override;
    135. 

  135. 	virtual void Visit(BfConstTypeRef* typeRef) override;
    136. 

  136. 	virtual void Visit(BfConstExprTypeRef* typeRef) override;
    137. 

  137. 	virtual void Visit(BfRefTypeRef* typeRef) override;
    138. 

  138. 	virtual void Visit(BfArrayTypeRef* typeRef) override;
    139. 

  139. 	virtual void Visit(BfGenericInstanceTypeRef* typeRef) override;
    140. 

  140. 	virtual void Visit(BfTupleTypeRef * typeRef) override;
    141. 

  141. 	virtual void Visit(BfPointerTypeRef* typeRef) override;
    142. 

  142. 	virtual void Visit(BfNullableTypeRef* typeRef) override;
    143. 

  143. 	virtual void Visit(BfVariableDeclaration* varDecl) override;
    144. 

  144. 	virtual void Visit(BfParameterDeclaration* paramDecl) override;
    145. 

  145. 	virtual void Visit(BfParamsExpression* paramsExpr) override;
    146. 

  146. 	virtual void Visit(BfTypeOfExpression* typeOfExpr) override;
    147. 

  147. 	virtual void Visit(BfSizeOfExpression* sizeOfExpr) override;
    148. 

  148. 	virtual void Visit(BfDefaultExpression* defaultExpr) override;
    149. 

  149. 	virtual void Visit(BfCheckTypeExpression* checkTypeExpr) override;
    150. 

  150. 	virtual void Visit(BfDynamicCastExpression* dynCastExpr) override;
    151. 

  151. 	virtual void Visit(BfCastExpression* castExpr) override;
    152. 

  152. 	virtual void Visit(BfDelegateBindExpression* invocationExpr) override;	
    153. 

  153. 	virtual void Visit(BfLambdaBindExpression* lambdaBindExpr) override;
    154. 

  154. 	virtual void Visit(BfObjectCreateExpression* invocationExpr) override;
    155. 

  155. 	virtual void Visit(BfBoxExpression* boxExpr) override;
    156. 

  156. 	virtual void Visit(BfInvocationExpression* invocationExpr) override;	
    157. 

  157. 	virtual void Visit(BfSwitchCase* switchCase) override;
    158. 

  158. 	virtual void Visit(BfWhenExpression* whenExpr) override;
    159. 

  159. 	virtual void Visit(BfSwitchStatement* switchStmt) override;
    160. 

  160. 	virtual void Visit(BfTryStatement* tryStmt) override;
    161. 

  161. 	virtual void Visit(BfCatchStatement* catchStmt) override;
    162. 

  162. 	virtual void Visit(BfFinallyStatement* finallyStmt) override;
    163. 

  163. 	virtual void Visit(BfCheckedStatement* checkedStmt) override;
    164. 

  164. 	virtual void Visit(BfUncheckedStatement* uncheckedStmt) override;
    165. 

  165. 	virtual void Visit(BfIfStatement* ifStmt) override;
    166. 

  166. 	virtual void Visit(BfThrowStatement* throwStmt) override;
    167. 

  167. 	virtual void Visit(BfDeleteStatement* deleteStmt) override;
    168. 

  168. 	virtual void Visit(BfReturnStatement* returnStmt) override;
    169. 

  169. 	virtual void Visit(BfBreakStatement* breakStmt) override;
    170. 

  170. 	virtual void Visit(BfContinueStatement* continueStmt) override;
    171. 

  171. 	virtual void Visit(BfFallthroughStatement* fallthroughStmt) override;
    172. 

  172. 	virtual void Visit(BfUsingStatement* whileStmt) override;
    173. 

  173. 	virtual void Visit(BfDoStatement* whileStmt) override;
    174. 

  174. 	virtual void Visit(BfRepeatStatement* repeatStmt) override;
    175. 

  175. 	virtual void Visit(BfWhileStatement* whileStmt) override;
    176. 

  176. 	virtual void Visit(BfForStatement* forStmt) override;
    177. 

  177. 	virtual void Visit(BfForEachStatement* forEachStmt) override;
    178. 

  178. 	virtual void Visit(BfDeferStatement* deferStmt) override;
    179. 

  179. 	virtual void Visit(BfEnumCaseBindExpression * caseBindExpr) override;
    180. 

  180. 	virtual void Visit(BfCaseExpression * caseExpr) override;
    181. 

  181. 	virtual void Visit(BfConditionalExpression* condExpr) override;
    182. 

  182. 	virtual void Visit(BfAssignmentExpression* assignExpr) override;
    183. 

  183. 	virtual void Visit(BfParenthesizedExpression* parenExpr) override;
    184. 

  184. 	virtual void Visit(BfTupleExpression * tupleExpr) override;
    185. 

  185. 	virtual void Visit(BfMemberReferenceExpression* memberRefExpr) override;
    186. 

  186. 	virtual void Visit(BfIndexerExpression* indexerExpr) override;
    187. 

  187. 	virtual void Visit(BfUnaryOperatorExpression* binOpExpr) override;
    188. 

  188. 	virtual void Visit(BfBinaryOperatorExpression* binOpExpr) override;
    189. 

  189. 	virtual void Visit(BfConstructorDeclaration* ctorDeclaration) override;
    190. 

  190. 	virtual void Visit(BfDestructorDeclaration* dtorDeclaration) override;
    191. 

  191. 	virtual void Visit(BfMethodDeclaration* methodDeclaration) override;
    192. 

  192. 	virtual void Visit(BfOperatorDeclaration* opreratorDeclaration) override;
    193. 

  193. 	virtual void Visit(BfPropertyMethodDeclaration* propertyDeclaration) override;
    194. 

  194. 	virtual void Visit(BfPropertyDeclaration* propertyDeclaration) override;
    195. 

  195. 	virtual void Visit(BfFieldDeclaration* fieldDeclaration) override;
    196. 

  196. 	virtual void Visit(BfEnumCaseDeclaration* enumCaseDeclaration) override;
    197. 

  197. 	virtual void Visit(BfTypeAliasDeclaration* typeDeclaration) override;
    198. 

  198. 	virtual void Visit(BfTypeDeclaration* typeDeclaration) override;
    199. 

  199. 	virtual void Visit(BfUsingDirective* usingDirective) override;
    200. 

  200. 	virtual void Visit(BfUsingStaticDirective* usingDirective) override;
    201. 

  201. 	virtual void Visit(BfNamespaceDeclaration* namespaceDeclaration) override;
    202. 

  202. 	virtual void Visit(BfBlock* block) override;
    203. 

  203. 	virtual void Visit(BfRootNode* rootNode) override;
    204. 

  204. 	virtual void Visit(BfInlineAsmStatement* asmStmt) override;
    205. 

  205. };
    206. 

  206. 

  207. NS_BF_END
    208.