Support external arrays in Crankshaft

src/x64/lithium-codegen-x64.cc

 // Copyright 2011 the V8 project authors. All rights reserved.
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions are
 // met:
 //
 //     * Redistributions of source code must retain the above copyright
 //       notice, this list of conditions and the following disclaimer.
 //     * Redistributions in binary form must reproduce the above
 //       copyright notice, this list of conditions and the following
 //       disclaimer in the documentation and/or other materials provided
@@ skipping 2134 matching lines @@
 void LCodeGen::DoLoadElements(LLoadElements* instr) {
   Register result = ToRegister(instr->result());
   Register input = ToRegister(instr->InputAt(0));
   __ movq(result, FieldOperand(input, JSObject::kElementsOffset));
   if (FLAG_debug_code) {
     NearLabel done;
     __ CompareRoot(FieldOperand(result, HeapObject::kMapOffset),
                    Heap::kFixedArrayMapRootIndex);
     __ j(equal, &done);
     __ CompareRoot(FieldOperand(result, HeapObject::kMapOffset),
-                   Heap::kExternalPixelArrayMapRootIndex);
+                   Heap::kFixedCOWArrayMapRootIndex);
     __ j(equal, &done);
-    __ CompareRoot(FieldOperand(result, HeapObject::kMapOffset),
-                   Heap::kFixedCOWArrayMapRootIndex);
-    __ Check(equal, "Check for fast elements failed.");
+    Register temp((result.is(rax)) ? rbx : rax);
+    __ push(temp);
+    __ movq(temp, FieldOperand(result, HeapObject::kMapOffset));
+    __ movzxbq(temp, FieldOperand(temp, Map::kInstanceTypeOffset));
+    __ subq(temp, Immediate(FIRST_EXTERNAL_ARRAY_TYPE));
+    __ cmpq(temp, Immediate(kExternalArrayTypeCount));
+    __ pop(temp);
+    __ Check(below, "Check for fast elements failed.");
     __ bind(&done);
   }
 }
 
 
 void LCodeGen::DoLoadExternalArrayPointer(
     LLoadExternalArrayPointer* instr) {
   Register result = ToRegister(instr->result());
   Register input = ToRegister(instr->InputAt(0));
   __ movq(result, FieldOperand(input,
@@ skipping 30 matching lines @@
                                key,
                                times_pointer_size,
                                FixedArray::kHeaderSize));
 
   // Check for the hole value.
   __ CompareRoot(result, Heap::kTheHoleValueRootIndex);
   DeoptimizeIf(equal, instr->environment());
 }
 
 
-void LCodeGen::DoLoadPixelArrayElement(LLoadPixelArrayElement* instr) {
-  Register external_elements = ToRegister(instr->external_pointer());
+void LCodeGen::DoLoadKeyedSpecializedArrayElement(
+    LLoadKeyedSpecializedArrayElement* instr) {
+  Register external_pointer = ToRegister(instr->external_pointer());
   Register key = ToRegister(instr->key());
-  Register result = ToRegister(instr->result());
-  ASSERT(result.is(external_elements));
-
-  // Load the result.
-  __ movzxbq(result, Operand(external_elements, key, times_1, 0));
+  ExternalArrayType array_type = instr->array_type();
+  switch (array_type) {
+    case kExternalByteArray:
+      __ movsxbq(ToRegister(instr->result()),
+                 Operand(external_pointer, key, times_1, 0));
+      break;
+    case kExternalUnsignedByteArray:
+    case kExternalPixelArray:
+      __ movzxbq(ToRegister(instr->result()),
+                 Operand(external_pointer, key, times_1, 0));
+      break;
+    case kExternalShortArray:
+      __ movsxwq(ToRegister(instr->result()),
+                 Operand(external_pointer, key, times_2, 0));
+      break;
+    case kExternalUnsignedShortArray:
+      __ movzxwq(ToRegister(instr->result()),
+                 Operand(external_pointer, key, times_2, 0));
+      break;
+    case kExternalIntArray:
+      __ movsxlq(ToRegister(instr->result()),
+                 Operand(external_pointer, key, times_4, 0));
+      break;
+    case kExternalUnsignedIntArray: {
+      Register result(ToRegister(instr->result()));
+      __ movl(result, Operand(external_pointer, key, times_4, 0));
+      __ testl(result, result);
+      // TODO(danno): we could be more clever here, perhaps having a special
+      // version of the stub that detects if the overflow case actually happens,
+      // and generate code that returns a double rather than int.
+      DeoptimizeIf(negative, instr->environment());
+      break;
+    }
+    case kExternalFloatArray: {
+      XMMRegister result(ToDoubleRegister(instr->result()));
+      __ movss(result, Operand(external_pointer, key, times_4, 0));
+      __ cvtss2sd(result, result);
+      break;
+    }
+  }
 }
 
 
 void LCodeGen::DoLoadKeyedGeneric(LLoadKeyedGeneric* instr) {
   ASSERT(ToRegister(instr->object()).is(rdx));
   ASSERT(ToRegister(instr->key()).is(rax));
 
   Handle<Code> ic = isolate()->builtins()->KeyedLoadIC_Initialize();
   CallCode(ic, RelocInfo::CODE_TARGET, instr);
 }
@@ skipping 593 matching lines @@
   ASSERT(ToRegister(instr->value()).is(rax));
 
   __ Move(rcx, instr->hydrogen()->name());
   Handle<Code> ic = info_->is_strict()
       ? isolate()->builtins()->StoreIC_Initialize_Strict()
       : isolate()->builtins()->StoreIC_Initialize();
   CallCode(ic, RelocInfo::CODE_TARGET, instr);
 }
 
 
-void LCodeGen::DoStorePixelArrayElement(LStorePixelArrayElement* instr) {
+void LCodeGen::DoStoreKeyedSpecializedArrayElement(
+    LStoreKeyedSpecializedArrayElement* instr) {
   Register external_pointer = ToRegister(instr->external_pointer());
   Register key = ToRegister(instr->key());
-  Register value = ToRegister(instr->value());
 
-  {  // Clamp the value to [0..255].
-    NearLabel done;
-    __ testl(value, Immediate(0xFFFFFF00));
-    __ j(zero, &done);
-    __ setcc(negative, value);  // 1 if negative, 0 if positive.
-    __ decb(value);  // 0 if negative, 255 if positive.
-    __ bind(&done);
+  switch (instr->array_type()) {
+    case kExternalPixelArray:
+      {  // Clamp the value to [0..255].
+        NearLabel done;
+        Register value(ToRegister(instr->value()));
+        __ testl(value, Immediate(0xFFFFFF00));
+        __ j(zero, &done);
+        __ setcc(negative, value);  // 1 if negative, 0 if positive.
+        __ decb(value);  // 0 if negative, 255 if positive.
+        __ bind(&done);
+        __ movb(Operand(external_pointer, key, times_1, 0), value);
+      }
+      break;
+    case kExternalByteArray:
+    case kExternalUnsignedByteArray:
+      __ movb(Operand(external_pointer, key, times_1, 0),
+               ToRegister(instr->value()));
+      break;
+    case kExternalShortArray:
+    case kExternalUnsignedShortArray:
+      __ movw(Operand(external_pointer, key, times_2, 0),
+               ToRegister(instr->value()));
+      break;
+    case kExternalIntArray:
+    case kExternalUnsignedIntArray:
+      __ movl(Operand(external_pointer, key, times_4, 0),
+              ToRegister(instr->value()));
+      break;
+    case kExternalFloatArray: {
+      XMMRegister value(ToDoubleRegister(instr->value()));
+      __ cvtsd2ss(value, value);
+      __ movss(Operand(external_pointer, key, times_4, 0), value);
+      break;
+    }
   }
-
-  __ movb(Operand(external_pointer, key, times_1, 0), value);
 }
 
 
 void LCodeGen::DoBoundsCheck(LBoundsCheck* instr) {
   if (instr->length()->IsRegister()) {
     __ cmpq(ToRegister(instr->index()), ToRegister(instr->length()));
   } else {
     __ cmpq(ToRegister(instr->index()), ToOperand(instr->length()));
   }
   DeoptimizeIf(above_equal, instr->environment());
@@ skipping 959 matching lines @@
   RegisterEnvironmentForDeoptimization(environment);
   ASSERT(osr_pc_offset_ == -1);
   osr_pc_offset_ = masm()->pc_offset();
 }
 
 #undef __
 
 } }  // namespace v8::internal
 
 #endif  // V8_TARGET_ARCH_X64