1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

symbol.c: return the results directly

* symbol.c (lookup_str_id, lookup_str_sym, lookup_id_str): return
  the result ID, Symbol, and string directly instead of returning
  via a pointer.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@46971 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
nobu 2014-07-27 01:46:50 +00:00
parent 22e09d1e22
commit b2aee6f45b

126
symbol.c
View file

@ -97,7 +97,7 @@ Init_sym(void)
} }
static ID attrsetname_to_attr(VALUE name); static ID attrsetname_to_attr(VALUE name);
static int lookup_id_str(ID id, st_data_t *data); static VALUE lookup_id_str(ID id);
ID ID
rb_id_attrset(ID id) rb_id_attrset(ID id)
@ -120,10 +120,10 @@ rb_id_attrset(ID id)
return id; return id;
default: default:
{ {
st_data_t data; VALUE str;
if (lookup_id_str(id, &data)) { if ((str = lookup_id_str(id)) != 0) {
rb_name_error(id, "cannot make unknown type ID %d:%"PRIsVALUE" attrset", rb_name_error(id, "cannot make unknown type ID %d:%"PRIsVALUE" attrset",
scope, (VALUE)data); scope, str);
} }
else { else {
rb_name_error_str(Qnil, "cannot make unknown type anonymous ID %d:%"PRIxVALUE" attrset", rb_name_error_str(Qnil, "cannot make unknown type anonymous ID %d:%"PRIxVALUE" attrset",
@ -325,8 +325,8 @@ register_static_symid_str(ID id, VALUE str)
RUBY_DTRACE_SYMBOL_CREATE(RSTRING_PTR(str), rb_sourcefile(), rb_sourceline()); RUBY_DTRACE_SYMBOL_CREATE(RSTRING_PTR(str), rb_sourcefile(), rb_sourceline());
} }
st_add_direct(global_symbols.str_id, (st_data_t)str, id); st_add_direct(global_symbols.str_id, (st_data_t)str, (st_data_t)id);
st_add_direct(global_symbols.id_str, id, (st_data_t)str); st_add_direct(global_symbols.id_str, (st_data_t)id, (st_data_t)str);
rb_gc_register_mark_object(str); rb_gc_register_mark_object(str);
return id; return id;
@ -357,10 +357,10 @@ must_be_dynamic_symbol(VALUE x)
{ {
if (UNLIKELY(!DYNAMIC_SYM_P(x))) { if (UNLIKELY(!DYNAMIC_SYM_P(x))) {
if (STATIC_SYM_P(x)) { if (STATIC_SYM_P(x)) {
st_data_t str; VALUE str = lookup_id_str(RSHIFT((unsigned long)(x),RUBY_SPECIAL_SHIFT));
if (lookup_id_str(RSHIFT((unsigned long)(x),RUBY_SPECIAL_SHIFT), &str)) { if (str) {
rb_bug("wrong argument: %s (inappropriate Symbol)", RSTRING_PTR((VALUE)str)); rb_bug("wrong argument: %s (inappropriate Symbol)", RSTRING_PTR(str));
} }
else { else {
rb_bug("wrong argument: inappropriate Symbol (%p)", (void *)x); rb_bug("wrong argument: inappropriate Symbol (%p)", (void *)x);
@ -395,7 +395,7 @@ dsymbol_alloc(const VALUE klass, const VALUE str, rb_encoding * const enc)
RSYMBOL(dsym)->type = type; RSYMBOL(dsym)->type = type;
st_add_direct(global_symbols.str_id, (st_data_t)str, (st_data_t)dsym); st_add_direct(global_symbols.str_id, (st_data_t)str, (st_data_t)dsym);
st_add_direct(global_symbols.id_str, (ID)dsym, (st_data_t)str); st_add_direct(global_symbols.id_str, (st_data_t)dsym, (st_data_t)str);
rb_hash_aset(global_symbols.dsymbol_fstr_hash, str, Qtrue); rb_hash_aset(global_symbols.dsymbol_fstr_hash, str, Qtrue);
if (RUBY_DTRACE_SYMBOL_CREATE_ENABLED()) { if (RUBY_DTRACE_SYMBOL_CREATE_ENABLED()) {
@ -445,54 +445,49 @@ dsymbol_pindown(VALUE sym)
} }
static ID static ID
lookup_str_id(st_data_t str, st_data_t *id_datap) lookup_str_id(VALUE str)
{ {
if (st_lookup(global_symbols.str_id, str, id_datap)) { st_data_t id_data;
const ID id = (ID)*id_datap; if (st_lookup(global_symbols.str_id, (st_data_t)str, &id_data)) {
const ID id = (ID)id_data;
if (ID_DYNAMIC_SYM_P(id) && !SYMBOL_PINNED_P(id)) { if (!ID_DYNAMIC_SYM_P(id) || SYMBOL_PINNED_P(id)) {
*id_datap = 0; return id;
return FALSE; }
}
return (ID)0;
}
static VALUE
lookup_str_sym(const VALUE str)
{
st_data_t sym_data;
if (st_lookup(global_symbols.str_id, (st_data_t)str, &sym_data)) {
const ID id = (ID)sym_data;
if (ID_DYNAMIC_SYM_P(id)) {
return dsymbol_check(id);
} }
else { else {
return TRUE; return STATIC_ID2SYM(id);
} }
} }
else { else {
return FALSE; return (VALUE)0;
} }
} }
static VALUE static VALUE
lookup_str_sym(const st_data_t str, st_data_t *sym_datap) lookup_id_str(ID id)
{ {
if (st_lookup(global_symbols.str_id, str, sym_datap)) { st_data_t data;
const ID id = *sym_datap;
if (ID_DYNAMIC_SYM_P(id)) { if (ID_DYNAMIC_SYM_P(id)) {
*sym_datap = dsymbol_check(id); return RSYMBOL(id)->fstr;
} }
else { if (st_lookup(global_symbols.id_str, id, &data)) {
*sym_datap = STATIC_ID2SYM(id); return (VALUE)data;
} }
return TRUE; return 0;
}
else {
return FALSE;
}
}
static int
lookup_id_str(ID id, st_data_t *data)
{
if (ID_DYNAMIC_SYM_P(id)) {
*data = RSYMBOL(id)->fstr;
return TRUE;
}
if (st_lookup(global_symbols.id_str, id, data)) {
return TRUE;
}
return FALSE;
} }
ID ID
@ -668,9 +663,9 @@ rb_intern(const char *name)
ID ID
rb_intern_str(VALUE str) rb_intern_str(VALUE str)
{ {
st_data_t sym; VALUE sym = lookup_str_sym(str);
if (lookup_str_sym(str, &sym)) { if (sym) {
return SYM2ID(sym); return SYM2ID(sym);
} }
@ -719,9 +714,9 @@ rb_str_dynamic_intern(VALUE str)
{ {
#if USE_SYMBOL_GC #if USE_SYMBOL_GC
rb_encoding *enc, *ascii; rb_encoding *enc, *ascii;
VALUE sym; VALUE sym = lookup_str_sym(str);
if (lookup_str_sym(str, &sym)) { if (sym) {
return sym; return sym;
} }
@ -782,7 +777,7 @@ rb_sym2str(VALUE sym)
VALUE VALUE
rb_id2str(ID id) rb_id2str(ID id)
{ {
st_data_t data; VALUE str;
if (id < tLAST_OP_ID) { if (id < tLAST_OP_ID) {
int i = 0; int i = 0;
@ -800,8 +795,7 @@ rb_id2str(ID id)
} }
} }
if (lookup_id_str(id, &data)) { if ((str = lookup_id_str(id)) != 0) {
VALUE str = (VALUE)data;
if (RBASIC(str)->klass == 0) if (RBASIC(str)->klass == 0)
RBASIC_SET_CLASS_RAW(str, rb_cString); RBASIC_SET_CLASS_RAW(str, rb_cString);
return str; return str;
@ -809,7 +803,6 @@ rb_id2str(ID id)
if (is_attrset_id(id)) { if (is_attrset_id(id)) {
ID id_stem = (id & ~ID_SCOPE_MASK) | ID_STATIC_SYM; ID id_stem = (id & ~ID_SCOPE_MASK) | ID_STATIC_SYM;
VALUE str;
do { do {
if (!!(str = rb_id2str(id_stem | ID_LOCAL))) break; if (!!(str = rb_id2str(id_stem | ID_LOCAL))) break;
@ -823,8 +816,7 @@ rb_id2str(ID id)
str = rb_str_dup(str); str = rb_str_dup(str);
rb_str_cat(str, "=", 1); rb_str_cat(str, "=", 1);
register_static_symid_str(id, str); register_static_symid_str(id, str);
if (lookup_id_str(id, &data)) { if ((str = lookup_id_str(id)) != 0) {
VALUE str = (VALUE)data;
if (RBASIC(str)->klass == 0) if (RBASIC(str)->klass == 0)
RBASIC_SET_CLASS_RAW(str, rb_cString); RBASIC_SET_CLASS_RAW(str, rb_cString);
return str; return str;
@ -947,7 +939,7 @@ rb_is_junk_id(ID id)
ID ID
rb_check_id(volatile VALUE *namep) rb_check_id(volatile VALUE *namep)
{ {
st_data_t id; ID id;
VALUE tmp; VALUE tmp;
VALUE name = *namep; VALUE name = *namep;
@ -976,7 +968,7 @@ rb_check_id(volatile VALUE *namep)
sym_check_asciionly(name); sym_check_asciionly(name);
if (lookup_str_id(name, &id)) { if ((id = lookup_str_id(name)) != 0) {
return id; return id;
} }
@ -991,7 +983,7 @@ rb_check_id(volatile VALUE *namep)
VALUE VALUE
rb_check_symbol(volatile VALUE *namep) rb_check_symbol(volatile VALUE *namep)
{ {
st_data_t sym; VALUE sym;
VALUE tmp; VALUE tmp;
VALUE name = *namep; VALUE name = *namep;
@ -1011,7 +1003,7 @@ rb_check_symbol(volatile VALUE *namep)
sym_check_asciionly(name); sym_check_asciionly(name);
if (lookup_str_sym(name, &sym)) { if ((sym = lookup_str_sym(name)) != 0) {
return sym; return sym;
} }
@ -1026,21 +1018,21 @@ rb_check_symbol(volatile VALUE *namep)
ID ID
rb_check_id_cstr(const char *ptr, long len, rb_encoding *enc) rb_check_id_cstr(const char *ptr, long len, rb_encoding *enc)
{ {
st_data_t id; ID id;
struct RString fake_str; struct RString fake_str;
const VALUE name = setup_fake_str(&fake_str, ptr, len); const VALUE name = setup_fake_str(&fake_str, ptr, len);
rb_enc_associate(name, enc); rb_enc_associate(name, enc);
sym_check_asciionly(name); sym_check_asciionly(name);
if (lookup_str_id(name, &id)) { if ((id = lookup_str_id(name)) != 0) {
return id; return id;
} }
if (rb_is_attrset_name(name)) { if (rb_is_attrset_name(name)) {
fake_str.as.heap.len = len - 1; fake_str.as.heap.len = len - 1;
if (lookup_str_id((st_data_t)name, &id)) { if ((id = lookup_str_id(name)) != 0) {
return rb_id_attrset((ID)id); return rb_id_attrset(id);
} }
} }
@ -1050,20 +1042,20 @@ rb_check_id_cstr(const char *ptr, long len, rb_encoding *enc)
VALUE VALUE
rb_check_symbol_cstr(const char *ptr, long len, rb_encoding *enc) rb_check_symbol_cstr(const char *ptr, long len, rb_encoding *enc)
{ {
st_data_t sym; VALUE sym;
struct RString fake_str; struct RString fake_str;
const VALUE name = setup_fake_str(&fake_str, ptr, len); const VALUE name = setup_fake_str(&fake_str, ptr, len);
rb_enc_associate(name, enc); rb_enc_associate(name, enc);
sym_check_asciionly(name); sym_check_asciionly(name);
if (lookup_str_sym(name, &sym)) { if ((sym = lookup_str_sym(name)) != 0) {
return sym; return sym;
} }
if (rb_is_attrset_name(name)) { if (rb_is_attrset_name(name)) {
fake_str.as.heap.len = len - 1; fake_str.as.heap.len = len - 1;
if (lookup_str_sym((st_data_t)name, &sym)) { if ((sym = lookup_str_sym(name)) != 0) {
return ID2SYM(rb_id_attrset(SYM2ID(sym))); return ID2SYM(rb_id_attrset(SYM2ID(sym)));
} }
} }
@ -1075,15 +1067,15 @@ static ID
attrsetname_to_attr(VALUE name) attrsetname_to_attr(VALUE name)
{ {
if (rb_is_attrset_name(name)) { if (rb_is_attrset_name(name)) {
st_data_t id; ID id;
struct RString fake_str; struct RString fake_str;
/* make local name by chopping '=' */ /* make local name by chopping '=' */
const VALUE localname = setup_fake_str(&fake_str, RSTRING_PTR(name), RSTRING_LEN(name) - 1); const VALUE localname = setup_fake_str(&fake_str, RSTRING_PTR(name), RSTRING_LEN(name) - 1);
rb_enc_copy(localname, name); rb_enc_copy(localname, name);
OBJ_FREEZE(localname); OBJ_FREEZE(localname);
if (lookup_str_id((st_data_t)localname, &id)) { if ((id = lookup_str_id(localname)) != 0) {
return (ID)id; return id;
} }
RB_GC_GUARD(name); RB_GC_GUARD(name);
} }