src/binaryen/src/ir/localize.h

   1 /*
   2  * Copyright 2016 WebAssembly Community Group participants
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #ifndef wasm_ir_localizer_h
  18 #define wasm_ir_localizer_h
  19
  20 #include <wasm-builder.h>
  21
  22 namespace wasm {
  23
  24 // Make an expression available in a local. If already in one, just
  25 // use that local, otherwise use a new local
  26
  27 struct Localizer {
  28   Index index;
  29   Expression* expr;
  30
  31   Localizer(Expression* input, Function* func, Module* wasm) {
  32     expr = input;
  33     if (auto* get = expr->dynCast<GetLocal>()) {
  34       index = get->index;
  35     } else if (auto* set = expr->dynCast<SetLocal>()) {
  36       index = set->index;
  37     } else {
  38       index = Builder::addVar(func, expr->type);
  39       expr = Builder(*wasm).makeTeeLocal(index, expr);
  40     }
  41   }
  42 };
  43
  44 } // namespace wasm
  45
  46 #endif // wasm_ir_localizer_h
  47