vm/contexts.cpp

   1 #include "master.hpp"
   2
   3 namespace factor
   4 {
   5
   6 context::context(cell datastack_size, cell retainstack_size, cell callstack_size) :
   7         callstack_top(NULL),
   8         callstack_bottom(NULL),
   9         datastack(0),
  10         retainstack(0),
  11         callstack_save(0),
  12         datastack_seg(new segment(datastack_size,false)),
  13         retainstack_seg(new segment(retainstack_size,false)),
  14         callstack_seg(new segment(callstack_size,false))
  15 {
  16         reset();
  17 }
  18
  19 void context::reset_datastack()
  20 {
  21         datastack = datastack_seg->start - sizeof(cell);
  22 }
  23
  24 void context::reset_retainstack()
  25 {
  26         retainstack = retainstack_seg->start - sizeof(cell);
  27 }
  28
  29 void context::reset_callstack()
  30 {
  31         callstack_top = callstack_bottom = CALLSTACK_BOTTOM(this);
  32 }
  33
  34 void context::reset_context_objects()
  35 {
  36         memset_cell(context_objects,false_object,context_object_count * sizeof(cell));
  37 }
  38
  39 void context::reset()
  40 {
  41         reset_datastack();
  42         reset_retainstack();
  43         reset_callstack();
  44         reset_context_objects();
  45 }
  46
  47 void context::fix_stacks()
  48 {
  49         if(datastack + sizeof(cell) < datastack_seg->start
  50                 || datastack + stack_reserved >= datastack_seg->end)
  51                 reset_datastack();
  52
  53         if(retainstack + sizeof(cell) < retainstack_seg->start
  54                 || retainstack + stack_reserved >= retainstack_seg->end)
  55                 reset_retainstack();
  56 }
  57
  58 void context::scrub_stacks(gc_info *info, cell index)
  59 {
  60         u8 *bitmap = info->gc_info_bitmap();
  61
  62         {
  63                 cell base = info->scrub_d_base(index);
  64
  65                 for(int loc = 0; loc < info->scrub_d_count; loc++)
  66                 {
  67                         if(bitmap_p(bitmap,base + loc))
  68                         {
  69 #ifdef DEBUG_GC_MAPS
  70                                 std::cout << "scrubbing datastack location " << loc << std::endl;
  71 #endif
  72                                 *((cell *)datastack - loc) = 0;
  73                         }
  74                 }
  75         }
  76
  77         {
  78                 cell base = info->scrub_r_base(index);
  79
  80                 for(int loc = 0; loc < info->scrub_r_count; loc++)
  81                 {
  82                         if(bitmap_p(bitmap,base + loc))
  83                         {
  84 #ifdef DEBUG_GC_MAPS
  85                                 std::cout << "scrubbing retainstack location " << loc << std::endl;
  86 #endif
  87                                 *((cell *)retainstack - loc) = 0;
  88                         }
  89                 }
  90         }
  91 }
  92
  93 context::~context()
  94 {
  95         delete datastack_seg;
  96         delete retainstack_seg;
  97         delete callstack_seg;
  98 }
  99
 100 /* called on startup */
 101 void factor_vm::init_contexts(cell datastack_size_, cell retainstack_size_, cell callstack_size_)
 102 {
 103         datastack_size = datastack_size_;
 104         retainstack_size = retainstack_size_;
 105         callstack_size = callstack_size_;
 106
 107         ctx = NULL;
 108         spare_ctx = new_context();
 109 }
 110
 111 void factor_vm::delete_contexts()
 112 {
 113         assert(!ctx);
 114         std::vector<context *>::const_iterator iter = unused_contexts.begin();
 115         std::vector<context *>::const_iterator end = unused_contexts.end();
 116         while(iter != end)
 117         {
 118                 delete *iter;
 119                 iter++;
 120         }
 121 }
 122
 123 context *factor_vm::new_context()
 124 {
 125         context *new_context;
 126
 127         if(unused_contexts.empty())
 128         {
 129                 new_context = new context(datastack_size,
 130                         retainstack_size,
 131                         callstack_size);
 132         }
 133         else
 134         {
 135                 new_context = unused_contexts.back();
 136                 unused_contexts.pop_back();
 137         }
 138
 139         new_context->reset();
 140
 141         active_contexts.insert(new_context);
 142
 143         return new_context;
 144 }
 145
 146 void factor_vm::init_context(context *ctx)
 147 {
 148         ctx->context_objects[OBJ_CONTEXT] = allot_alien(ctx);
 149 }
 150
 151 context *new_context(factor_vm *parent)
 152 {
 153         context *new_context = parent->new_context();
 154         parent->init_context(new_context);
 155         return new_context;
 156 }
 157
 158 void factor_vm::delete_context(context *old_context)
 159 {
 160         unused_contexts.push_back(old_context);
 161         active_contexts.erase(old_context);
 162 }
 163
 164 VM_C_API void delete_context(factor_vm *parent, context *old_context)
 165 {
 166         parent->delete_context(old_context);
 167 }
 168
 169 cell factor_vm::begin_callback(cell quot_)
 170 {
 171         data_root<object> quot(quot_,this);
 172
 173         ctx->reset();
 174         spare_ctx = new_context();
 175         callback_ids.push_back(callback_id++);
 176
 177         init_context(ctx);
 178
 179         return quot.value();
 180 }
 181
 182 cell begin_callback(factor_vm *parent, cell quot)
 183 {
 184         return parent->begin_callback(quot);
 185 }
 186
 187 void factor_vm::end_callback()
 188 {
 189         callback_ids.pop_back();
 190         delete_context(ctx);
 191 }
 192
 193 void end_callback(factor_vm *parent)
 194 {
 195         parent->end_callback();
 196 }
 197
 198 void factor_vm::primitive_current_callback()
 199 {
 200         ctx->push(tag_fixnum(callback_ids.back()));
 201 }
 202
 203 void factor_vm::primitive_context_object()
 204 {
 205         fixnum n = untag_fixnum(ctx->peek());
 206         ctx->replace(ctx->context_objects[n]);
 207 }
 208
 209 void factor_vm::primitive_set_context_object()
 210 {
 211         fixnum n = untag_fixnum(ctx->pop());
 212         cell value = ctx->pop();
 213         ctx->context_objects[n] = value;
 214 }
 215
 216 void factor_vm::primitive_context_object_for()
 217 {
 218         context *other_ctx = (context *)pinned_alien_offset(ctx->pop());
 219         fixnum n = untag_fixnum(ctx->pop());
 220         ctx->push(other_ctx->context_objects[n]);
 221 }
 222
 223 cell factor_vm::stack_to_array(cell bottom, cell top)
 224 {
 225         fixnum depth = (fixnum)(top - bottom + sizeof(cell));
 226
 227         if(depth < 0)
 228                 return false_object;
 229         else
 230         {
 231                 array *a = allot_uninitialized_array<array>(depth / sizeof(cell));
 232                 memcpy(a + 1,(void*)bottom,depth);
 233                 return tag<array>(a);
 234         }
 235 }
 236
 237 cell factor_vm::datastack_to_array(context *ctx)
 238 {
 239         cell array = stack_to_array(ctx->datastack_seg->start,ctx->datastack);
 240         if(array == false_object)
 241         {
 242                 general_error(ERROR_DATASTACK_UNDERFLOW,false_object,false_object);
 243                 return false_object;
 244         }
 245         else
 246                 return array;
 247 }
 248
 249 void factor_vm::primitive_datastack()
 250 {
 251         ctx->push(datastack_to_array(ctx));
 252 }
 253
 254 void factor_vm::primitive_datastack_for()
 255 {
 256         context *other_ctx = (context *)pinned_alien_offset(ctx->pop());
 257         ctx->push(datastack_to_array(other_ctx));
 258 }
 259
 260 cell factor_vm::retainstack_to_array(context *ctx)
 261 {
 262         cell array = stack_to_array(ctx->retainstack_seg->start,ctx->retainstack);
 263         if(array == false_object)
 264         {
 265                 general_error(ERROR_RETAINSTACK_UNDERFLOW,false_object,false_object);
 266                 return false_object;
 267         }
 268         else
 269                 return array;
 270 }
 271
 272 void factor_vm::primitive_retainstack()
 273 {
 274         ctx->push(retainstack_to_array(ctx));
 275 }
 276
 277 void factor_vm::primitive_retainstack_for()
 278 {
 279         context *other_ctx = (context *)pinned_alien_offset(ctx->pop());
 280         ctx->push(retainstack_to_array(other_ctx));
 281 }
 282
 283 /* returns pointer to top of stack */
 284 cell factor_vm::array_to_stack(array *array, cell bottom)
 285 {
 286         cell depth = array_capacity(array) * sizeof(cell);
 287         memcpy((void*)bottom,array + 1,depth);
 288         return bottom + depth - sizeof(cell);
 289 }
 290
 291 void factor_vm::set_datastack(context *ctx, array *array)
 292 {
 293         ctx->datastack = array_to_stack(array,ctx->datastack_seg->start);
 294 }
 295
 296 void factor_vm::primitive_set_datastack()
 297 {
 298         set_datastack(ctx,untag_check<array>(ctx->pop()));
 299 }
 300
 301 void factor_vm::set_retainstack(context *ctx, array *array)
 302 {
 303         ctx->retainstack = array_to_stack(array,ctx->retainstack_seg->start);
 304 }
 305
 306 void factor_vm::primitive_set_retainstack()
 307 {
 308         set_retainstack(ctx,untag_check<array>(ctx->pop()));
 309 }
 310
 311 /* Used to implement call( */
 312 void factor_vm::primitive_check_datastack()
 313 {
 314         fixnum out = to_fixnum(ctx->pop());
 315         fixnum in = to_fixnum(ctx->pop());
 316         fixnum height = out - in;
 317         array *saved_datastack = untag_check<array>(ctx->pop());
 318         fixnum saved_height = array_capacity(saved_datastack);
 319         fixnum current_height = (ctx->datastack - ctx->datastack_seg->start + sizeof(cell)) / sizeof(cell);
 320         if(current_height - height != saved_height)
 321                 ctx->push(false_object);
 322         else
 323         {
 324                 cell *ds_bot = (cell *)ctx->datastack_seg->start;
 325                 for(fixnum i = 0; i < saved_height - in; i++)
 326                 {
 327                         if(ds_bot[i] != array_nth(saved_datastack,i))
 328                         {
 329                                 ctx->push(false_object);
 330                                 return;
 331                         }
 332                 }
 333                 ctx->push(true_object);
 334         }
 335 }
 336
 337 void factor_vm::primitive_load_locals()
 338 {
 339         fixnum count = untag_fixnum(ctx->pop());
 340         memcpy((cell *)(ctx->retainstack + sizeof(cell)),
 341                 (cell *)(ctx->datastack - sizeof(cell) * (count - 1)),
 342                 sizeof(cell) * count);
 343         ctx->datastack -= sizeof(cell) * count;
 344         ctx->retainstack += sizeof(cell) * count;
 345 }
 346
 347 }