Track live instance and allocation counts for classes

runtime/vm/stub_code_arm.cc

 // Copyright (c) 2013, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include "vm/globals.h"
 #if defined(TARGET_ARCH_ARM)
 
 #include "vm/assembler.h"
 #include "vm/code_generator.h"
 #include "vm/compiler.h"
@@ skipping 605 matching lines @@
     // R3: array size.
     // R7: potential next object start.
     // R8: points to new space object.
     __ LoadFromOffset(kWord, IP, R8, Scavenger::end_offset());
     __ cmp(R7, ShifterOperand(IP));
     __ b(&slow_case, CS);  // Branch if unsigned higher or equal.
 
     // Successfully allocated the object(s), now update top to point to
     // next object start and initialize the object.
     // R0: potential new object start.
+    // R3: array size.
     // R7: potential next object start.
     // R8: Points to new space object.
     __ StoreToOffset(kWord, R7, R8, Scavenger::top_offset());
     __ add(R0, R0, ShifterOperand(kHeapObjectTag));
+    __ UpdateAllocationStatsWithSize(kArrayCid, R3, R8);
 
     // R0: new object start as a tagged pointer.
     // R1: array element type.
     // R2: array length as Smi.
     // R3: array size.
     // R7: new object end address.
 
     // Store the type argument field.
     __ StoreIntoObjectNoBarrier(
         R0,
@@ skipping 312 matching lines @@
     }
 
     // Successfully allocated the object, now update top to point to
     // next object start and initialize the object.
     // R0: new object.
     // R1: number of context variables.
     // R2: object size.
     // R3: next object start.
     __ str(R3, Address(R5, 0));
     __ add(R0, R0, ShifterOperand(kHeapObjectTag));
+    __ UpdateAllocationStatsWithSize(context_class.id(), R2, R5);
 
     // Calculate the size tag.
     // R0: new object.
     // R1: number of context variables.
     // R2: object size.
     const intptr_t shift = RawObject::kSizeTagBit - kObjectAlignmentLog2;
     __ CompareImmediate(R2, RawObject::SizeTag::kMaxSizeTag);
     // If no size tag overflow, shift R2 left, else set R2 to zero.
     __ mov(R2, ShifterOperand(R2, LSL, shift), LS);
     __ mov(R2, ShifterOperand(0), HI);
@@ skipping 162 matching lines @@
     __ cmp(R3, ShifterOperand(IP));
     if (FLAG_use_slow_path) {
       __ b(&slow_case);
     } else {
       __ b(&slow_case, CS);  // Branch if unsigned higher or equal.
     }
 
     // Successfully allocated the object(s), now update top to point to
     // next object start and initialize the object.
     __ str(R3, Address(R5, 0));
+    __ UpdateAllocationStats(cls.id(), R5);
 
     if (is_cls_parameterized) {
       // Initialize the type arguments field in the object.
       // R2: new object start.
       // R4: potential new object end and, if R4 != R3, potential new
       // InstantiatedTypeArguments object start.
       // R3: next object start.
       Label type_arguments_ready;
       __ cmp(R4, ShifterOperand(R3));
       __ b(&type_arguments_ready, EQ);
@@ skipping 145 matching lines @@
     if (FLAG_use_slow_path) {
       __ b(&slow_case);
     } else {
       __ b(&slow_case, CS);  // Branch if unsigned higher or equal.
     }
 
     // Successfully allocated the object, now update top to point to
     // next object start and initialize the object.
     __ str(R3, Address(R5, 0));
 
+    if (is_implicit_instance_closure) {
+      // This closure allocates a context, update allocation stats.
+      // R3: context size.
+      __ LoadImmediate(R3, context_size);
+      // R5: Clobbered.
+     __ UpdateAllocationStatsWithSize(kContextCid, R3, R5);
+    }
+    // The closure allocation is attributed to the signature class.
+    // R5: Will be clobbered.
+    __ UpdateAllocationStats(cls.id(), R5);
+
     // R2: new closure object.
     // R4: new context object (only if is_implicit_closure).
     // Set the tags.
     uword tags = 0;
     tags = RawObject::SizeTag::update(closure_size, tags);
     tags = RawObject::ClassIdTag::update(cls.id(), tags);
     __ LoadImmediate(R0, tags);
     __ str(R0, Address(R2, Instance::tags_offset()));
 
     // Initialize the function field in the object.
@@ skipping 801 matching lines @@
   const Register right = R0;
   __ ldr(left, Address(SP, 1 * kWordSize));
   __ ldr(right, Address(SP, 0 * kWordSize));
   GenerateIdenticalWithNumberCheckStub(assembler, left, right, temp);
   __ Ret();
 }
 
 }  // namespace dart
 
 #endif  // defined TARGET_ARCH_ARM