[Interpreter] Implement ForIn in bytecode graph builder.

src/interpreter/bytecode-generator.cc

 // Copyright 2015 the V8 project authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "src/interpreter/bytecode-generator.h"
 
 #include "src/ast/scopes.h"
 #include "src/compiler.h"
 #include "src/interpreter/control-flow-builders.h"
 #include "src/objects.h"
@@ skipping 831 matching lines @@
                                     language_mode());
       break;
     }
     case NAMED_SUPER_PROPERTY:
     case KEYED_SUPER_PROPERTY:
       UNIMPLEMENTED();
   }
 }
 
 
+void BytecodeGenerator::GetForInPreparedItem(Register destination, int index,
+                                             Register prepare_result,
+                                             Register scratch) {
+  builder()->LoadLiteral(Smi::FromInt(index));
+  builder()->StoreAccumulatorInRegister(scratch);
+  builder()->CallRuntime(Runtime::kFixedArrayGet, prepare_result, 2);
+  builder()->StoreAccumulatorInRegister(destination);
+}
+
+
 void BytecodeGenerator::VisitForInStatement(ForInStatement* stmt) {
   EffectResultScope statement_result_scope(this);
 
   if (stmt->subject()->IsNullLiteral() ||
       stmt->subject()->IsUndefinedLiteral(isolate())) {
     // ForIn generates lots of code, skip if it wouldn't produce any effects.
     return;
   }
 
   LoopBuilder loop_builder(builder());
   ControlScopeForIteration control_scope(this, stmt, &loop_builder);
 
   // Prepare the state for executing ForIn.
   VisitForAccumulatorValue(stmt->subject());
   loop_builder.BreakIfUndefined();
   loop_builder.BreakIfNull();
 
   Register receiver = execution_result()->NewRegister();
   builder()->CastAccumulatorToJSObject();
   builder()->StoreAccumulatorInRegister(receiver);
   builder()->CallRuntime(Runtime::kGetPropertyNamesFast, receiver, 1);
   builder()->ForInPrepare(receiver);
   loop_builder.BreakIfUndefined();
 
-  Register for_in_state = execution_result()->NewRegister();
-  builder()->StoreAccumulatorInRegister(for_in_state);
+  // Ensure prepare_result and prepare_index are consecutive for CallRuntime.
+  execution_result()->PrepareForConsecutiveAllocations(2);
+  Register prepare_result = execution_result()->NewRegister();
+  Register prepare_index = execution_result()->NewRegister();
+  builder()->StoreAccumulatorInRegister(prepare_result);
+
+  // Spill the for_in_state to registers so we can de-opt.
+  Register cache_array = execution_result()->NewRegister();
+  GetForInPreparedItem(cache_array, 0, prepare_result, prepare_index);
+  Register cache_type = execution_result()->NewRegister();
+  GetForInPreparedItem(cache_type, 1, prepare_result, prepare_index);
+  Register cache_length = execution_result()->NewRegister();
+  GetForInPreparedItem(cache_length, 2, prepare_result, prepare_index);
 
   // Check loop termination (accumulator holds index).
-  Register index = receiver;  // Re-using register as receiver no longer used.
+  Register index = prepare_result;
   builder()->LoadLiteral(Smi::FromInt(0));
+  builder()->StoreAccumulatorInRegister(index);
   loop_builder.LoopHeader();
   loop_builder.Condition();
-  builder()->StoreAccumulatorInRegister(index).ForInDone(for_in_state);
+  builder()->ForInDone(index, cache_length);
   loop_builder.BreakIfTrue();
-  builder()->ForInNext(for_in_state, index);
+  builder()->ForInNext(receiver, index, cache_type, cache_array);
   loop_builder.ContinueIfUndefined();
-
   VisitForInAssignment(stmt->each(), stmt->EachFeedbackSlot());
   Visit(stmt->body());
-
-  // TODO(oth): replace CountOperation here with ForInStep.
   loop_builder.Next();
-  builder()->LoadAccumulatorWithRegister(index).CountOperation(
-      Token::Value::ADD, language_mode_strength());
+  builder()->ForInStep(index);
   loop_builder.JumpToHeader();
   loop_builder.LoopEnd();
 }
 
 
 void BytecodeGenerator::VisitForOfStatement(ForOfStatement* stmt) {
   UNIMPLEMENTED();
 }
 
 
@@ skipping 1292 matching lines @@
 }
 
 
 int BytecodeGenerator::feedback_index(FeedbackVectorSlot slot) const {
   return info()->feedback_vector()->GetIndex(slot);
 }
 
 }  // namespace interpreter
 }  // namespace internal
 }  // namespace v8