diff options
| -rw-r--r-- | include/mruby/variable.h | 1 | ||||
| -rw-r--r-- | src/variable.c | 15 |
2 files changed, 16 insertions, 0 deletions
diff --git a/include/mruby/variable.h b/include/mruby/variable.h index 5126315c4..7189ef690 100644 --- a/include/mruby/variable.h +++ b/include/mruby/variable.h @@ -41,6 +41,7 @@ int mrb_const_defined(mrb_state*, mrb_value, mrb_sym); mrb_value mrb_obj_iv_get(mrb_state *mrb, struct RObject *obj, mrb_sym sym); void mrb_obj_iv_set(mrb_state *mrb, struct RObject *obj, mrb_sym sym, mrb_value v); int mrb_obj_iv_defined(mrb_state *mrb, struct RObject *obj, mrb_sym sym); +void mrb_obj_iv_ifnone(mrb_state *mrb, struct RObject *obj, mrb_sym sym, mrb_value v); mrb_value mrb_iv_get(mrb_state *mrb, mrb_value obj, mrb_sym sym); void mrb_iv_set(mrb_state *mrb, mrb_value obj, mrb_sym sym, mrb_value v); int mrb_iv_defined(mrb_state*, mrb_value, mrb_sym); diff --git a/src/variable.c b/src/variable.c index 6907a8e2a..992663408 100644 --- a/src/variable.c +++ b/src/variable.c @@ -461,6 +461,21 @@ mrb_obj_iv_set(mrb_state *mrb, struct RObject *obj, mrb_sym sym, mrb_value v) } void +mrb_obj_iv_ifnone(mrb_state *mrb, struct RObject *obj, mrb_sym sym, mrb_value v) +{ + iv_tbl *t = obj->iv; + + if (!t) { + t = obj->iv = iv_new(mrb); + } + else if (iv_get(mrb, t, sym, &v)) { + return; + } + mrb_write_barrier(mrb, (struct RBasic*)obj); + iv_put(mrb, t, sym, v); +} + +void mrb_iv_set(mrb_state *mrb, mrb_value obj, mrb_sym sym, mrb_value v) { if (obj_iv_p(obj)) { |