vm/arrays.cpp

   1 #include "master.hpp"
   2
   3 namespace factor
   4 {
   5
   6 /* make a new array with an initial element */
   7 array *allot_array(cell capacity, cell fill_)
   8 {
   9         gc_root<object> fill(fill_);
  10         gc_root<array> new_array(allot_array_internal<array>(capacity));
  11
  12         if(fill.value() == tag_fixnum(0))
  13                 memset(new_array->data(),'\0',capacity * sizeof(cell));
  14         else
  15         {
  16                 /* No need for write barrier here. Either the object is in
  17                 the nursery, or it was allocated directly in tenured space
  18                 and the write barrier is already hit for us in that case. */
  19                 cell i;
  20                 for(i = 0; i < capacity; i++)
  21                         new_array->data()[i] = fill.value();
  22         }
  23         return new_array.untagged();
  24 }
  25
  26 /* push a new array on the stack */
  27 PRIMITIVE(array)
  28 {
  29         cell initial = dpop();
  30         cell size = unbox_array_size();
  31         dpush(tag<array>(allot_array(size,initial)));
  32 }
  33
  34 cell allot_array_1(cell obj_)
  35 {
  36         gc_root<object> obj(obj_);
  37         gc_root<array> a(allot_array_internal<array>(1));
  38         set_array_nth(a.untagged(),0,obj.value());
  39         return a.value();
  40 }
  41
  42 cell allot_array_2(cell v1_, cell v2_)
  43 {
  44         gc_root<object> v1(v1_);
  45         gc_root<object> v2(v2_);
  46         gc_root<array> a(allot_array_internal<array>(2));
  47         set_array_nth(a.untagged(),0,v1.value());
  48         set_array_nth(a.untagged(),1,v2.value());
  49         return a.value();
  50 }
  51
  52 cell allot_array_4(cell v1_, cell v2_, cell v3_, cell v4_)
  53 {
  54         gc_root<object> v1(v1_);
  55         gc_root<object> v2(v2_);
  56         gc_root<object> v3(v3_);
  57         gc_root<object> v4(v4_);
  58         gc_root<array> a(allot_array_internal<array>(4));
  59         set_array_nth(a.untagged(),0,v1.value());
  60         set_array_nth(a.untagged(),1,v2.value());
  61         set_array_nth(a.untagged(),2,v3.value());
  62         set_array_nth(a.untagged(),3,v4.value());
  63         return a.value();
  64 }
  65
  66 PRIMITIVE(resize_array)
  67 {
  68         array* a = untag_check<array>(dpop());
  69         cell capacity = unbox_array_size();
  70         dpush(tag<array>(reallot_array(a,capacity)));
  71 }
  72
  73 void growable_array::add(cell elt_)
  74 {
  75         gc_root<object> elt(elt_);
  76         if(count == array_capacity(elements.untagged()))
  77                 elements = reallot_array(elements.untagged(),count * 2);
  78
  79         set_array_nth(elements.untagged(),count++,elt.value());
  80 }
  81
  82 void growable_array::trim()
  83 {
  84         elements = reallot_array(elements.untagged(),count);
  85 }
  86
  87 }