o o h˔@sRddlmZddlZedddgidZ erddlmZmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZmZm Z m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'dd l(m)Z)m*Z*m+Z+m,Z,dd l-m.Z.dd l/m0Z0dd l(m1Z1m2Z2ddl3Z3ddl4Z4GdddZ5Gddde5Z6nGdddZ6ddZ7dS)) import_moduleNz clang.cindexfromlistcindex) import_kwargs)VariableIntegerFloatFunctionPrototypeFunctionDefinition FunctionCallnoneReturn Assignmentintcint8int16int64uint8uint16uint32uint64float32float64float80 aug_assignbool_While CodeBlock) PreDecrement PostDecrement PreIncrement PostIncrement)AddModMulPowRel)And as_BooleanNotOr)Symbol)sympify)falsetruec@s eZdZdZddZddZdS) BaseParserzBase Class for the C parsercCstj|_dS)z6Initializes the Base parser creating a Clang AST indexN)cinIndexcreateindexselfr6l/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sympy/parsing/c/c_parser.py__init__?szBaseParser.__init__c CsL|jjD]}tddddddd|j|jj|jj|jj|jf|dqd S) z%Diagostics function for the Clang ASTz%s %s (line %s, col %s) %sFATALERRORWARNINGNOTEIGNORED)rfileN) tu diagnosticsprintseveritylocationrClinecolumnspelling)r5outdiagr6r6r7rECs$  zBaseParser.diagnosticsN)__name__ __module__ __qualname____doc__r8rEr6r6r6r7r/<s r/cseZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9ZZ S):CCodeConverterzThe Code Convereter for Clang AST The converter object takes the C source code or file as input and converts them to SymPy Expressions. cstg|_tjjtitjjtitjj t tjj t tjj ttjjttjjttjjttjjttjjtitjjttjjttjjtid|_dS)zInitializes the code converter)voidboolintfloatN) superr8 _py_nodesr0TypeKindVOIDr BOOLrSCHARrSHORTrINTrLONGrUCHARrUSHORTrUINTrULONGrFLOATrDOUBLEr LONGDOUBLEr _data_typesr4 __class__r6r7r8\s(   CCodeConverter.__init__cCsjtj|}|jj||tjjd|_|jj D]}|j tj j ks(|j tj jkr1|j||q|jS)aFunction to parse a file with C source code It takes the filename as an attribute and creates a Clang AST Translation Unit parsing the file. Then the transformation function is called on the translation unit, whose reults are collected into a list which is returned by the function. Parameters ========== filename : string Path to the C file to be parsed flags: list Arguments to be passed to Clang while parsing the C code Returns ======= py_nodes: list A list of SymPy AST nodes argsoptions)ospathabspathr3parser0TranslationUnit PARSE_DETAILED_PROCESSING_RECORDrDcursor get_childrenkind CursorKindVAR_DECL FUNCTION_DECLrXappend transform)r5filenameflagsfilepathchildr6r6r7rqxs zCCodeConverter.parsecCstjddd}|||d|jj|j|tjj d|_ | |j j D]}|jtjjks8|jtjjkrA|j||q(|jS)aFunction to parse a string with C source code It takes the source code as an attribute, stores it in a temporary file and creates a Clang AST Translation Unit parsing the file. Then the transformation function is called on the translation unit, whose reults are collected into a list which is returned by the function. Parameters ========== source : string A string containing the C source code to be parsed flags: list Arguments to be passed to Clang while parsing the C code Returns ======= py_nodes: list A list of SymPy AST nodes zw+z.cpp)modesuffixrrk)tempfileNamedTemporaryFilewriteseekr3rqnamer0rrrsrDclosertrurvrwrxryrXrzr{)r5sourcer}rCrr6r6r7 parse_strs  zCCodeConverter.parse_strcCsTt|d|jjd}|dur&td|jddd|Dftjd||S)aFTransformation Function for Clang AST nodes It determines the kind of node and calls the respective transformation function for that node. Raises ====== NotImplementedError : if the transformation for the provided node is not implemented z transform_%sNzIgnoring node of type %s (%s) cs|]}|jVqdSNrK.0tr6r6r7  z+CCodeConverter.transform..rB) getattrrvrlowerrFjoin get_tokenssysstderr)r5nodehandlerr6r6r7r{s   zCCodeConverter.transformcCs|jj|jdvr|jd|jj}n*|jj|jdvr&|jd|jj}n|jj|jdvr9|jd|jj}ntdz|}t|}|jtjjksT|jtjj krft|}|jtjjksT|jtjj ksT| |}tjj tjj tjj tjjtjjtjjtjjg}|j|vrt|trt|}nxt|tr|jj|jdvr|dkrtdntd}n]|jj|jdvr|dkrtdntd }nG|jj|jdvrt|}n9t|ttttfr|jj|jdvrt|}n!|jj|jdvrt|}n|jj|jdvrtt|}n|}t|jj||d WS|jtjjkr't|jj|d WStd d dd|!Dt"yJt|jj|dYSw)aTransformation Function for Variable Declaration Used to create nodes for variable declarations and assignments with values or function call for the respective nodes in the clang AST Returns ======= A variable node as Declaration, with the initial value if given Raises ====== NotImplementedError : if called for data types not currently implemented Notes ===== The function currently supports following data types: Boolean: bool, _Bool Integer: 8-bit: signed char and unsigned char 16-bit: short, short int, signed short, signed short int, unsigned short, unsigned short int 32-bit: int, signed int, unsigned int 64-bit: long, long int, signed long, signed long int, unsigned long, unsigned long int Floating point: Single Precision: float Double Precision: double Extended Precision: long double rUrVrT&Only bool, int and float are supportedFrrAgg?typevalue)rz=Given variable declaration "{}" is not possible to parse yet!rcsrrrrr6r6r7rHrz4CCodeConverter.transform_var_decl..r)#rrvrgNotImplementedErrorrunextr0rw NAMESPACE_REFTYPE_REFr{INTEGER_LITERALFLOATING_LITERALUNEXPOSED_EXPRBINARY_OPERATOR PAREN_EXPRUNARY_OPERATORCXX_BOOL_LITERAL_EXPR isinstancestrr+rTrrr,rUrVrrKas_Declaration CALL_EXPRformatrr StopIteration)r5rrchildrenrval supported_rhsrr6r6r7transform_var_decls'         z!CCodeConverter.transform_var_declc Cs:|jj|jdvr|jd|jj}n=|jj|jdvr&|jd|jj}n*|jj|jdvr9|jd|jj}n|jj|jdvrL|jd|jj}ntdg}g}|D]+}||}|jtjjkrl| |qX|jtjj kr~|D]}| |quqX| |qX|gkrt ||j |d}|St ||j ||d}|S)aTransformation Function For Function Declaration Used to create nodes for function declarations and definitions for the respective nodes in the clang AST Returns ======= function : Codegen AST node - FunctionPrototype node if function body is not present - FunctionDefinition node if the function body is present rUrVrTrSz,Only void, bool, int and float are supported) return_typer parameters)rrrbody) result_typervrgrrur{r0rw PARM_DECLrz COMPOUND_STMTr rKr ) r5rret_typerparamrdeclrfunctionr6r6r7transform_function_declTsD      z&CCodeConverter.transform_function_declcCs|jj|jdvr|jd|jj}n*|jj|jdvr&|jd|jj}n|jj|jdvr9|jd|jj}ntdzi|}t|}|jtjjtjj tjj fvrft|}|jtjjtjj tjj fvsT| |}|jj|jdvryt |}n"|jj|jdvrt |}n|jj|jdvrtt|}ntdt|jj||d}Wntyt|jj|d}Ynwz | t|tdtyY|Sw)aTransformation function for Parameter Declaration Used to create parameter nodes for the required functions for the respective nodes in the clang AST Returns ======= param : Codegen AST Node Variable node with the value and type of the variable Raises ====== ValueError if multiple children encountered in the parameter node rUrVrTrrrz+Can't handle multiple children on parameter)rrvrgrrurr0rwrr TEMPLATE_REFr{rrr,rTrrKrr ValueError)r5rrrrlitrrr6r6r7transform_parm_declsb        z"CCodeConverter.transform_parm_declcCs8z t|j}Wt|Sty|j}Yt|Sw)aTransformation function for integer literal Used to get the value and type of the given integer literal. Returns ======= val : list List with two arguments type and Value type contains the type of the integer value contains the value stored in the variable Notes ===== Only Base Integer type supported for now )rrrKrliteralrUr5rrr6r6r7transform_integer_literals z(CCodeConverter.transform_integer_literalc Cs<z t|j}Wt|Sttfy|j}Yt|Sw)aTransformation function for floating literal Used to get the value and type of the given floating literal. Returns ======= val : list List with two arguments type and Value type contains the type of float value contains the value stored in the variable Notes ===== Only Base Float type supported for now )rrrKrrrrVrr6r6r7transform_floating_literalsz)CCodeConverter.transform_floating_literalcCsdSrr6r5rr6r6r7transform_string_literals z'CCodeConverter.transform_string_literalc Cs@z t|j}Wnttfy|j}Ynwtt|dS)aTransformation function for character literal Used to get the value of the given character literal. Returns ======= val : int val contains the ascii value of the character literal Notes ===== Only for cases where character is assigned to a integer value, since character literal is not in SymPy AST rA)rrrKrrrordrrr6r6r7transform_character_literals  z*CCodeConverter.transform_character_literalc Cs@z t|j}Wnttfy|j}Ynw|dkrdSdS)zTransformation function for boolean literal Used to get the value of the given boolean literal. Returns ======= value : bool value contains the boolean value of the variable r.TF)rrrKrrrrr6r6r7transform_cxx_bool_literal_expr+s   z.CCodeConverter.transform_cxx_bool_literal_exprcCsdS)z2Transformation function for unexposed declarationsNr6rr6r6r7transform_unexposed_decl=z'CCodeConverter.transform_unexposed_declcCsVz |}|t|}Wn tyYdSwzt|tdty*Y|Sw)aTransformation function for unexposed expression Unexposed expressions are used to wrap float, double literals and expressions Returns ======= expr : Codegen AST Node the result from the wrapped expression None : NoneType No childs are found for the node Raises ====== ValueError if the expression contains multiple children Nz&Unexposed expression has > 1 children.rur{rrr)r5rrexprr6r6r7transform_unexposed_exprAs  z'CCodeConverter.transform_unexposed_exprcCs|jS)z-Returns the name of the declaration referencerrr6r6r7transform_decl_ref_exprfsz&CCodeConverter.transform_decl_ref_exprcCsg}|}t|}|jtjjkrt|}|jtjjks|jtjjkr.t|}|jtjjks#||}z3|D]*}||}|jtjjkrL| t |q6|jtjj kr[| t |q6| |q6t ||WStyst |YSw)aJTransformation function for a call expression Used to create function call nodes for the function calls present in the C code Returns ======= FunctionCall : Codegen AST Node FunctionCall node with parameters if any parameters are present )rurrvr0rwrrr{rrzrrrr r)r5rrrr first_childargr6r6r7transform_call_exprjs,       z"CCodeConverter.transform_call_exprcCstt|jS)z.Returns the Return Node for a return statement)r rrurKrr6r6r7transform_return_stmtsz$CCodeConverter.transform_return_stmtcCs*g}|}|D] }|||q|S)aTransformation function for compond statemets Returns ======= expr : list list of Nodes for the expressions present in the statement None : NoneType if the compound statement is empty )rurzr{)r5rrrrr6r6r7transform_compound_stmts  z&CCodeConverter.transform_compound_stmtcCsZz |}|t|}Wn tyYnwz |t|tdty,Y|Sw)a@Transformation function for declaration statements These statements are used to wrap different kinds of declararions like variable or function declaration The function calls the transformer function for the child of the given node Returns ======= statement : Codegen AST Node contains the node returned by the children node for the type of declaration Raises ====== ValueError if multiple children present z,Don't know how to handle multiple statementsr)r5rr statementr6r6r7transform_decl_stmts  z"CCodeConverter.transform_decl_stmtcCs|t|S)z{Transformation function for Parenthesized expressions Returns the result from its children nodes )r{rrurr6r6r7transform_paren_exprsz#CCodeConverter.transform_paren_exprcCs ||S)aTransformation function for handling shorthand operators Returns ======= augmented_assignment_expression: Codegen AST node shorthand assignment expression represented as Codegen AST Raises ====== NotImplementedError If the shorthand operator for bitwise operators (~=, ^=, &=, |=, <<=, >>=) is encountered )transform_binary_operatorrr6r6r7&transform_compound_assignment_operators z5CCodeConverter.transform_compound_assignment_operatorcCsngd}t|}|dj|vr|t|}t|tra|djdkr*t|S|djdkr8t t|dS|djdkrEt t|S|djdkrRt t|S|djdkr_t t|Sd S|djdkrj|S|djdkrvt |dS|djdkrt t t|Sd S|d jd vr|t|}|d jdkrtt|S|d jdkrtt|Sd Std ) aTransformation function for handling unary operators Returns ======= unary_expression: Codegen AST node simplified unary expression represented as Codegen AST Raises ====== NotImplementedError If dereferencing operator(*), address operator(&) or bitwise NOT operator(~) is encountered )+-++--!rrrrrrrA)rrz`Dereferencing operator, Address operator and bitwise NOT operator have not been implemented yet!N)listrrKr{rrurrr+r$r rr)r,rTr!rr)r5roperators_listtokensrr6r6r7transform_unary_operators>        z'CCodeConverter.transform_unary_operatorc CsXt|}gd}g}g}|D]}|jtjjkr|jdkr$|dq|jdkr^|rY|ddkrYt|dkr;t d| }| }| } || ||| |rY|ddks1| q|j|vr|r| |j| |dkrt|dkr|t d| }| }| } || ||| |r| |j| |dksr||jq|jdvrt d|jd vrt d t d |j|jtjjkr||jd gq|jtjjkr||jd gq|jtjjkr|jdvr||jdgqt d |j|r&t|dkr t d| }| }| } || ||| |s|ddS)aTransformation function for handling binary operators Returns ======= binary_expression: Codegen AST node simplified binary expression represented as Codegen AST Raises ====== NotImplementedError If a bitwise operator or unary operator(which is a child of any binary operator in Clang AST) is encountered )rr*/%=>>=<<===!=&&||+=-=*=/=%=()rr@zCUnary operators as a part of binary operators is not supported yet!)&|^z<>z.Bitwise operator has not been implemented yet!)z&=z|=z^=z<<=z>>=z8Shorthand bitwise operator has not been implemented yet!z&Given token {} is not implemented yet! identifierr)r.r-booleanr)rrrvr0 TokenKind PUNCTUATIONrKrzlenrpopperform_operation priority_ofr IDENTIFIERLITERALKEYWORD) r5rrrcombined_variables_stackoperators_stacktokenrhslhsoperatorr6r6r7rs                 z(CCodeConverter.transform_binary_operatorcCs@|dvrdS|dvr dS|dvrdS|dvrdS|d vrd Sd S) z%To get the priority of given operator)rrrrrrrA)rrr@rrrrrrr?)rrr>)rrrrr6)r5opr6r6r7rszCCodeConverter.priority_ofcCs ||}||}|dkrt||dgS|dkr!t|| dgS|dkr,t||dgS|dkr $'v8F %%#: !rRc@seZdZddZdS)rRcOstd)NzModule not Installed) ImportError)r5rlkwargsr6r6r7r8 srjN)rNrOrPr8r6r6r6r7rR s cCs6t}tj|r|j|gd}|S|j|gd}|S)zFunction for converting a C source code The function reads the source code present in the given file and parses it to give out SymPy Expressions Returns ======= src : list List of Python expression strings )r})rRrnroexistsrqr)r convertersrcr6r6r7parse_cs  r!)8sympy.externalrrnr0sympy.codegen.astrrrr r r r r rrrrrrrrrrrrrrrrsympy.codegen.cnodesrrr r! sympy.corer"r#r$r%r&sympy.logic.boolalgr'r(r)r*sympy.core.symbolr+sympy.core.sympifyr,r-r.rrr/rRr!r6r6r6r7s2 &h  =