diff options
Diffstat (limited to 'include/printer.h')
| -rw-r--r-- | include/printer.h | 202 |
1 files changed, 202 insertions, 0 deletions
diff --git a/include/printer.h b/include/printer.h new file mode 100644 index 0000000..5930768 --- /dev/null +++ b/include/printer.h @@ -0,0 +1,202 @@ +// +// Created by ztuowen on 9/24/16. +// + +#ifndef CHILL_PRINTER_H_H +#define CHILL_PRINTER_H_H + +#include "chillAST.h" +#include "chillAST/chillAST_node.hh" +#include <string> +#include <sstream> + +/*! + * \file + * \brief this is a generic AST printer that prints the code out to a C-family like syntax + */ +namespace chill { + namespace printer { + class GenericPrinter { + private: + std::string indentSpace; + public: + GenericPrinter() { indentSpace = " "; } + + void setIndentSpace(int numspaces) { + indentSpace = ""; + for (int i = 0; i < numspaces; ++i) + indentSpace += " "; + } + //! return the Precedence of the corresponding AST node + /*! + * @param n the chillAST_Node + * @return a int representing the subnodes's precedence, 0 being the highest, INT8_MAX being the default + */ + virtual int getPrec(chillAST_Node *n) { return INT8_MAX; } + virtual void print(std::string ident, chillAST_ArraySubscriptExpr *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_BinaryOperator *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_CallExpr *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_CompoundStmt *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_CStyleAddressOf *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_CStyleCastExpr *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_CudaFree *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_CudaKernelCall *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_CudaMalloc *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_CudaMemcpy *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_CudaSyncthreads *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_DeclRefExpr *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_FloatingLiteral *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_ForStmt *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_Free *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_FunctionDecl *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_IfStmt *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_IntegerLiteral *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_ImplicitCastExpr *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_MacroDefinition *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_Malloc *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_MemberExpr *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_NULL *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_NoOp *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_ParenExpr *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_Preprocessing *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_RecordDecl *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_ReturnStmt *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_Sizeof *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_SourceFile *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_TypedefDecl *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_TernaryOperator *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_UnaryOperator *n, std::ostringstream &o)=0; + virtual void print(std::string ident, chillAST_VarDecl *n, std::ostringstream &o)=0; + //! Print the AST to string stream + /*! + * @param ident indentation of the node + * @param n the chillAST_Node + * @param o the string stream + */ + virtual void print(std::string ident, chillAST_Node *n, std::ostringstream &o){ + switch (n->getType()) { + case CHILLAST_NODE_ARRAYSUBSCRIPTEXPR: + print(ident, dynamic_cast<chillAST_ArraySubscriptExpr*>(n),o); + break; + case CHILLAST_NODE_BINARYOPERATOR: + print(ident, dynamic_cast<chillAST_BinaryOperator*>(n),o); + break; + case CHILLAST_NODE_CALLEXPR: + print(ident, dynamic_cast<chillAST_CallExpr*>(n),o); + break; + case CHILLAST_NODE_COMPOUNDSTMT: + print(ident, dynamic_cast<chillAST_CompoundStmt*>(n),o); + break; + case CHILLAST_NODE_CSTYLEADDRESSOF: + print(ident, dynamic_cast<chillAST_CStyleAddressOf*>(n),o); + break; + case CHILLAST_NODE_CSTYLECASTEXPR: + print(ident, dynamic_cast<chillAST_CStyleCastExpr*>(n),o); + break; + case CHILLAST_NODE_CUDAFREE: + print(ident, dynamic_cast<chillAST_CudaFree*>(n),o); + break; + case CHILLAST_NODE_CUDAKERNELCALL: + print(ident, dynamic_cast<chillAST_CudaKernelCall*>(n),o); + break; + case CHILLAST_NODE_CUDAMALLOC: + print(ident, dynamic_cast<chillAST_CudaMalloc*>(n),o); + break; + case CHILLAST_NODE_CUDAMEMCPY: + print(ident, dynamic_cast<chillAST_CudaMemcpy*>(n),o); + break; + case CHILLAST_NODE_CUDASYNCTHREADS: + print(ident, dynamic_cast<chillAST_CudaSyncthreads*>(n),o); + break; + case CHILLAST_NODE_DECLREFEXPR: + print(ident, dynamic_cast<chillAST_DeclRefExpr*>(n),o); + break; + case CHILLAST_NODE_FLOATINGLITERAL: + print(ident, dynamic_cast<chillAST_FloatingLiteral*>(n),o); + break; + case CHILLAST_NODE_LOOP: + case CHILLAST_NODE_FORSTMT: + print(ident, dynamic_cast<chillAST_ForStmt*>(n),o); + break; + case CHILLAST_NODE_FREE: + print(ident, dynamic_cast<chillAST_Free*>(n),o); + break; + case CHILLAST_NODE_FUNCTIONDECL: + print(ident, dynamic_cast<chillAST_FunctionDecl*>(n),o); + break; + case CHILLAST_NODE_IFSTMT: + print(ident, dynamic_cast<chillAST_IfStmt*>(n),o); + break; + case CHILLAST_NODE_IMPLICITCASTEXPR: + print(ident, dynamic_cast<chillAST_ImplicitCastExpr*>(n),o); + break; + case CHILLAST_NODE_INTEGERLITERAL: + print(ident, dynamic_cast<chillAST_IntegerLiteral*>(n),o); + break; + case CHILLAST_NODE_MACRODEFINITION: + print(ident, dynamic_cast<chillAST_MacroDefinition*>(n),o); + break; + case CHILLAST_NODE_MALLOC: + print(ident, dynamic_cast<chillAST_Malloc*>(n),o); + break; + case CHILLAST_NODE_MEMBEREXPR: + print(ident, dynamic_cast<chillAST_MemberExpr*>(n),o); + break; + case CHILLAST_NODE_NOOP: + print(ident, dynamic_cast<chillAST_NoOp*>(n),o); + break; + case CHILLAST_NODE_NULL: + print(ident, dynamic_cast<chillAST_NULL*>(n),o); + break; + case CHILLAST_NODE_PARENEXPR: + print(ident, dynamic_cast<chillAST_ParenExpr*>(n),o); + break; + case CHILLAST_NODE_PREPROCESSING: + print(ident, dynamic_cast<chillAST_Preprocessing*>(n),o); + break; + case CHILLAST_NODE_RECORDDECL: + print(ident, dynamic_cast<chillAST_RecordDecl*>(n),o); + break; + case CHILLAST_NODE_RETURNSTMT: + print(ident, dynamic_cast<chillAST_ReturnStmt*>(n),o); + break; + case CHILLAST_NODE_SIZEOF: + print(ident, dynamic_cast<chillAST_Sizeof*>(n),o); + break; + case CHILLAST_NODE_TRANSLATIONUNIT: + case CHILLAST_NODE_SOURCEFILE: + print(ident, dynamic_cast<chillAST_SourceFile*>(n),o); + break; + case CHILLAST_NODE_TERNARYOPERATOR: + print(ident, dynamic_cast<chillAST_TernaryOperator*>(n),o); + break; + case CHILLAST_NODE_TYPEDEFDECL: + print(ident, dynamic_cast<chillAST_TypedefDecl*>(n),o); + break; + case CHILLAST_NODE_UNARYOPERATOR: + print(ident, dynamic_cast<chillAST_UnaryOperator*>(n),o); + break; + case CHILLAST_NODE_VARDECL: + print(ident, dynamic_cast<chillAST_VarDecl*>(n),o); + break; + case CHILLAST_NODE_UNKNOWN: + default: + CHILL_ERROR("Printing an unknown type of Node: %s\n", n->getTypeString()); + } + } + //! Print the AST to string, overload the print function + /*! + * @param ident indentation of the node + * @param n the chillAST_Node + * @return a string of the corresponding code + */ + virtual std::string print(std::string ident, chillAST_Node *n) { + std::ostringstream os; + print(ident, n, os); + return os.str(); + } + }; + } +} + +#endif //CHILL_PRINTER_H_H |