| 1 | // | 
|---|
| 2 | // Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo | 
|---|
| 3 | // | 
|---|
| 4 | // The contents of this file are covered under the licence agreement in the | 
|---|
| 5 | // file "LICENCE" distributed with Cforall. | 
|---|
| 6 | // | 
|---|
| 7 | // Eval.cc -- Evaluate parts of the ast at compile time. | 
|---|
| 8 | // | 
|---|
| 9 | // Author           : Richard C. Bilson | 
|---|
| 10 | // Created On       : Mon May 18 07:44:20 2015 | 
|---|
| 11 | // Last Modified By : Peter A. Buhr | 
|---|
| 12 | // Last Modified On : Sat Aug  6 12:11:59 2022 | 
|---|
| 13 | // Update Count     : 119 | 
|---|
| 14 | // | 
|---|
| 15 |  | 
|---|
| 16 | #include "Eval.h" | 
|---|
| 17 |  | 
|---|
| 18 | #include <utility> // for pair | 
|---|
| 19 |  | 
|---|
| 20 | #include "AST/Inspect.hpp" | 
|---|
| 21 | #include "CodeGen/OperatorTable.h"                                              // access: OperatorInfo | 
|---|
| 22 | #include "AST/Pass.hpp" | 
|---|
| 23 | #include "InitTweak/InitTweak.h" | 
|---|
| 24 |  | 
|---|
| 25 | struct EvalNew : public ast::WithShortCircuiting { | 
|---|
| 26 | Evaluation result = { 0, true, true }; | 
|---|
| 27 |  | 
|---|
| 28 | void previsit( const ast::Node * ) { visit_children = false; } | 
|---|
| 29 | void postvisit( const ast::Node * ) { result.isEvaluableInGCC = result.hasKnownValue = false; } | 
|---|
| 30 |  | 
|---|
| 31 | void postvisit( const ast::UntypedExpr * ) { | 
|---|
| 32 | assertf( false, "UntypedExpr in constant expression evaluation" ); // FIX ME, resolve variable | 
|---|
| 33 | } | 
|---|
| 34 |  | 
|---|
| 35 | void postvisit( const ast::ConstantExpr * expr ) {      // only handle int constants | 
|---|
| 36 | result.knownValue = expr->intValue(); | 
|---|
| 37 | result.hasKnownValue = true; | 
|---|
| 38 | result.isEvaluableInGCC = true; | 
|---|
| 39 | } | 
|---|
| 40 |  | 
|---|
| 41 | void postvisit( const ast::SizeofExpr * ) { | 
|---|
| 42 | result.hasKnownValue = false; | 
|---|
| 43 | result.isEvaluableInGCC = true; | 
|---|
| 44 | } | 
|---|
| 45 |  | 
|---|
| 46 | void postvisit( const ast::AlignofExpr * ) { | 
|---|
| 47 | result.hasKnownValue = false; | 
|---|
| 48 | result.isEvaluableInGCC = true; | 
|---|
| 49 | } | 
|---|
| 50 |  | 
|---|
| 51 | void postvisit( const ast::OffsetofExpr * ) { | 
|---|
| 52 | result.hasKnownValue = false; | 
|---|
| 53 | result.isEvaluableInGCC = true; | 
|---|
| 54 | } | 
|---|
| 55 |  | 
|---|
| 56 | void postvisit( const ast::LogicalExpr * expr ) { | 
|---|
| 57 | Evaluation arg1, arg2; | 
|---|
| 58 | arg1 = eval( expr->arg1 ); | 
|---|
| 59 | result.isEvaluableInGCC &= arg1.isEvaluableInGCC; | 
|---|
| 60 | if ( ! result.isEvaluableInGCC ) return; | 
|---|
| 61 | arg2 = eval( expr->arg2 ); | 
|---|
| 62 | result.isEvaluableInGCC &= arg2.isEvaluableInGCC; | 
|---|
| 63 | if ( ! result.isEvaluableInGCC ) return; | 
|---|
| 64 |  | 
|---|
| 65 | result.hasKnownValue &= arg1.hasKnownValue; | 
|---|
| 66 | result.hasKnownValue &= arg2.hasKnownValue; | 
|---|
| 67 | if ( ! result.hasKnownValue ) return; | 
|---|
| 68 |  | 
|---|
| 69 | if ( expr->isAnd ) { | 
|---|
| 70 | result.knownValue = arg1.knownValue && arg2.knownValue; | 
|---|
| 71 | } else { | 
|---|
| 72 | result.knownValue = arg1.knownValue || arg2.knownValue; | 
|---|
| 73 | } // if | 
|---|
| 74 | } | 
|---|
| 75 |  | 
|---|
| 76 | void postvisit( const ast::ConditionalExpr * expr ) { | 
|---|
| 77 | Evaluation arg1, arg2, arg3; | 
|---|
| 78 | arg1 = eval( expr->arg1 ); | 
|---|
| 79 | result.isEvaluableInGCC &= arg1.isEvaluableInGCC; | 
|---|
| 80 | if ( ! result.isEvaluableInGCC ) return; | 
|---|
| 81 | arg2 = eval( expr->arg2 ); | 
|---|
| 82 | result.isEvaluableInGCC &= arg2.isEvaluableInGCC; | 
|---|
| 83 | if ( ! result.isEvaluableInGCC ) return; | 
|---|
| 84 | arg3 = eval( expr->arg3 ); | 
|---|
| 85 | result.isEvaluableInGCC &= arg3.isEvaluableInGCC; | 
|---|
| 86 | if ( ! result.isEvaluableInGCC ) return; | 
|---|
| 87 |  | 
|---|
| 88 | result.hasKnownValue &= arg1.hasKnownValue; | 
|---|
| 89 | result.hasKnownValue &= arg2.hasKnownValue; | 
|---|
| 90 | result.hasKnownValue &= arg3.hasKnownValue; | 
|---|
| 91 | if ( ! result.hasKnownValue ) return; | 
|---|
| 92 |  | 
|---|
| 93 | result.knownValue = arg1.knownValue ? arg2.knownValue : arg3.knownValue; | 
|---|
| 94 | } | 
|---|
| 95 |  | 
|---|
| 96 | void postvisit( const ast::CastExpr * expr ) { | 
|---|
| 97 | // cfa-cc generates a cast before every constant and many other places, e.g., (int)3, | 
|---|
| 98 | // so we must use the value from the cast argument, even though we lack any basis for evaluating wraparound effects, etc | 
|---|
| 99 | result = eval(expr->arg); | 
|---|
| 100 | } | 
|---|
| 101 |  | 
|---|
| 102 | void postvisit( const ast::VariableExpr * expr ) { | 
|---|
| 103 | result.hasKnownValue = false; | 
|---|
| 104 | result.isEvaluableInGCC = false; | 
|---|
| 105 | if ( const ast::EnumInstType * inst = dynamic_cast<const ast::EnumInstType *>(expr->result.get()) ) { | 
|---|
| 106 | if ( const ast::EnumDecl * decl = inst->base ) { | 
|---|
| 107 | result.isEvaluableInGCC = true; | 
|---|
| 108 | result.hasKnownValue = decl->valueOf( expr->var, result.knownValue ); // result.knownValue filled by valueOf | 
|---|
| 109 | } | 
|---|
| 110 | } | 
|---|
| 111 | } | 
|---|
| 112 |  | 
|---|
| 113 | void postvisit( const ast::ApplicationExpr * expr ) { | 
|---|
| 114 | const ast::DeclWithType * function = ast::getFunction(expr); | 
|---|
| 115 | if ( ! function || function->linkage != ast::Linkage::Intrinsic ) { | 
|---|
| 116 | result.isEvaluableInGCC = false; | 
|---|
| 117 | result.hasKnownValue = false; | 
|---|
| 118 | return; | 
|---|
| 119 | } | 
|---|
| 120 | const std::string & fname = function->name; | 
|---|
| 121 | assertf( expr->args.size() == 1 || expr->args.size() == 2, "Intrinsic function with %zd arguments: %s", expr->args.size(), fname.c_str() ); | 
|---|
| 122 |  | 
|---|
| 123 | if ( expr->args.size() == 1 ) { | 
|---|
| 124 | // pre/postfix operators ++ and -- => assignment, which is not constant | 
|---|
| 125 | Evaluation arg1; | 
|---|
| 126 | arg1 = eval(expr->args.front()); | 
|---|
| 127 | result.isEvaluableInGCC &= arg1.isEvaluableInGCC; | 
|---|
| 128 | if ( ! result.isEvaluableInGCC ) return; | 
|---|
| 129 |  | 
|---|
| 130 | result.hasKnownValue &= arg1.hasKnownValue; | 
|---|
| 131 | if ( ! result.hasKnownValue ) return; | 
|---|
| 132 |  | 
|---|
| 133 | if (fname == "+?") { | 
|---|
| 134 | result.knownValue = arg1.knownValue; | 
|---|
| 135 | } else if (fname == "-?") { | 
|---|
| 136 | result.knownValue = -arg1.knownValue; | 
|---|
| 137 | } else if (fname == "~?") { | 
|---|
| 138 | result.knownValue = ~arg1.knownValue; | 
|---|
| 139 | } else if (fname == "!?") { | 
|---|
| 140 | result.knownValue = ! arg1.knownValue; | 
|---|
| 141 | } else { | 
|---|
| 142 | result.isEvaluableInGCC = false; | 
|---|
| 143 | result.hasKnownValue = false; | 
|---|
| 144 | } // if | 
|---|
| 145 | } else { // => expr->args.size() == 2 | 
|---|
| 146 | // infix assignment operators => assignment, which is not constant | 
|---|
| 147 | Evaluation arg1, arg2; | 
|---|
| 148 | arg1 = eval(expr->args.front()); | 
|---|
| 149 | result.isEvaluableInGCC &= arg1.isEvaluableInGCC; | 
|---|
| 150 | if ( ! result.isEvaluableInGCC ) return; | 
|---|
| 151 | arg2 = eval(expr->args.back()); | 
|---|
| 152 | result.isEvaluableInGCC &= arg2.isEvaluableInGCC; | 
|---|
| 153 | if ( ! result.isEvaluableInGCC ) return; | 
|---|
| 154 |  | 
|---|
| 155 | result.hasKnownValue &= arg1.hasKnownValue; | 
|---|
| 156 | result.hasKnownValue &= arg2.hasKnownValue; | 
|---|
| 157 | if ( ! result.hasKnownValue ) return; | 
|---|
| 158 |  | 
|---|
| 159 | if (fname == "?+?") { | 
|---|
| 160 | result.knownValue = arg1.knownValue + arg2.knownValue; | 
|---|
| 161 | } else if (fname == "?-?") { | 
|---|
| 162 | result.knownValue = arg1.knownValue - arg2.knownValue; | 
|---|
| 163 | } else if (fname == "?*?") { | 
|---|
| 164 | result.knownValue = arg1.knownValue * arg2.knownValue; | 
|---|
| 165 | } else if (fname == "?/?") { | 
|---|
| 166 | if ( arg2.knownValue ) result.knownValue = arg1.knownValue / arg2.knownValue; | 
|---|
| 167 | } else if (fname == "?%?") { | 
|---|
| 168 | if ( arg2.knownValue ) result.knownValue = arg1.knownValue % arg2.knownValue; | 
|---|
| 169 | } else if (fname == "?<<?") { | 
|---|
| 170 | result.knownValue = arg1.knownValue << arg2.knownValue; | 
|---|
| 171 | } else if (fname == "?>>?") { | 
|---|
| 172 | result.knownValue = arg1.knownValue >> arg2.knownValue; | 
|---|
| 173 | } else if (fname == "?<?") { | 
|---|
| 174 | result.knownValue = arg1.knownValue < arg2.knownValue; | 
|---|
| 175 | } else if (fname == "?>?") { | 
|---|
| 176 | result.knownValue = arg1.knownValue > arg2.knownValue; | 
|---|
| 177 | } else if (fname == "?<=?") { | 
|---|
| 178 | result.knownValue = arg1.knownValue <= arg2.knownValue; | 
|---|
| 179 | } else if (fname == "?>=?") { | 
|---|
| 180 | result.knownValue = arg1.knownValue >= arg2.knownValue; | 
|---|
| 181 | } else if (fname == "?==?") { | 
|---|
| 182 | result.knownValue = arg1.knownValue == arg2.knownValue; | 
|---|
| 183 | } else if (fname == "?!=?") { | 
|---|
| 184 | result.knownValue = arg1.knownValue != arg2.knownValue; | 
|---|
| 185 | } else if (fname == "?&?") { | 
|---|
| 186 | result.knownValue = arg1.knownValue & arg2.knownValue; | 
|---|
| 187 | } else if (fname == "?^?") { | 
|---|
| 188 | result.knownValue = arg1.knownValue ^ arg2.knownValue; | 
|---|
| 189 | } else if (fname == "?|?") { | 
|---|
| 190 | result.knownValue = arg1.knownValue | arg2.knownValue; | 
|---|
| 191 | } else { | 
|---|
| 192 | result.isEvaluableInGCC = false; | 
|---|
| 193 | result.hasKnownValue = false; | 
|---|
| 194 | } | 
|---|
| 195 | } // if | 
|---|
| 196 | // TODO: implement other intrinsic functions | 
|---|
| 197 | } | 
|---|
| 198 | }; | 
|---|
| 199 |  | 
|---|
| 200 | Evaluation eval( const ast::Expr * expr ) { | 
|---|
| 201 | if ( expr ) { | 
|---|
| 202 |  | 
|---|
| 203 | return ast::Pass<EvalNew>::read(expr); | 
|---|
| 204 | // Evaluation ret = ast::Pass<EvalNew>::read(expr); | 
|---|
| 205 | // ret.knownValue = 777; | 
|---|
| 206 | // return ret; | 
|---|
| 207 |  | 
|---|
| 208 | } else { | 
|---|
| 209 | return { 0, false, false }; | 
|---|
| 210 | } | 
|---|
| 211 | } | 
|---|
| 212 |  | 
|---|
| 213 | // Local Variables: // | 
|---|
| 214 | // tab-width: 4 // | 
|---|
| 215 | // mode: c++ // | 
|---|
| 216 | // compile-command: "make install" // | 
|---|
| 217 | // End: // | 
|---|