2022-10-03 18:14:32 +03:00
|
|
|
#ifndef RUBY_SHAPE_H
|
|
|
|
#define RUBY_SHAPE_H
|
2022-12-15 21:54:07 +03:00
|
|
|
|
|
|
|
#include "internal/gc.h"
|
|
|
|
|
2022-10-03 18:14:32 +03:00
|
|
|
#if (SIZEOF_UINT64_T == SIZEOF_VALUE)
|
|
|
|
#define SIZEOF_SHAPE_T 4
|
|
|
|
#define SHAPE_IN_BASIC_FLAGS 1
|
|
|
|
typedef uint32_t attr_index_t;
|
|
|
|
#else
|
|
|
|
#define SIZEOF_SHAPE_T 2
|
|
|
|
#define SHAPE_IN_BASIC_FLAGS 0
|
|
|
|
typedef uint16_t attr_index_t;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#define MAX_IVARS (attr_index_t)(-1)
|
|
|
|
|
2022-11-18 18:39:09 +03:00
|
|
|
#if SIZEOF_SHAPE_T == 4
|
2022-10-03 18:14:32 +03:00
|
|
|
typedef uint32_t shape_id_t;
|
2022-11-18 21:29:41 +03:00
|
|
|
# define SHAPE_ID_NUM_BITS 32
|
2022-11-18 18:39:09 +03:00
|
|
|
#else
|
2022-10-03 18:14:32 +03:00
|
|
|
typedef uint16_t shape_id_t;
|
2022-11-18 21:29:41 +03:00
|
|
|
# define SHAPE_ID_NUM_BITS 16
|
2022-10-03 18:14:32 +03:00
|
|
|
#endif
|
|
|
|
|
2022-11-18 21:29:41 +03:00
|
|
|
# define SHAPE_MASK (((uintptr_t)1 << SHAPE_ID_NUM_BITS) - 1)
|
|
|
|
# define SHAPE_FLAG_MASK (((VALUE)-1) >> SHAPE_ID_NUM_BITS)
|
2022-10-03 18:14:32 +03:00
|
|
|
|
2022-11-18 21:29:41 +03:00
|
|
|
# define SHAPE_FLAG_SHIFT ((SIZEOF_VALUE * 8) - SHAPE_ID_NUM_BITS)
|
2022-10-03 18:14:32 +03:00
|
|
|
|
|
|
|
# define SHAPE_BITMAP_SIZE 16384
|
|
|
|
|
2022-12-09 01:16:52 +03:00
|
|
|
# define SHAPE_MAX_VARIATIONS 8
|
2023-01-26 19:42:17 +03:00
|
|
|
# define SHAPE_MAX_NUM_IVS 50
|
2022-12-09 01:16:52 +03:00
|
|
|
|
2022-10-03 18:14:32 +03:00
|
|
|
# define MAX_SHAPE_ID (SHAPE_MASK - 1)
|
|
|
|
# define INVALID_SHAPE_ID SHAPE_MASK
|
|
|
|
# define ROOT_SHAPE_ID 0x0
|
2022-12-09 01:16:52 +03:00
|
|
|
|
2022-11-18 02:57:11 +03:00
|
|
|
# define SPECIAL_CONST_SHAPE_ID (SIZE_POOL_COUNT * 2)
|
2022-12-09 01:16:52 +03:00
|
|
|
# define OBJ_TOO_COMPLEX_SHAPE_ID (SPECIAL_CONST_SHAPE_ID + 1)
|
2022-10-03 18:14:32 +03:00
|
|
|
|
|
|
|
struct rb_shape {
|
|
|
|
struct rb_id_table * edges; // id_table from ID (ivar) to next shape
|
|
|
|
ID edge_name; // ID (ivar) for transition from parent to rb_shape
|
2022-10-21 23:24:29 +03:00
|
|
|
attr_index_t next_iv_index;
|
2022-11-08 23:35:31 +03:00
|
|
|
uint32_t capacity; // Total capacity of the object with this shape
|
2022-10-03 18:14:32 +03:00
|
|
|
uint8_t type;
|
2022-11-08 23:35:31 +03:00
|
|
|
uint8_t size_pool_index;
|
2022-10-03 20:52:40 +03:00
|
|
|
shape_id_t parent_id;
|
2022-10-03 18:14:32 +03:00
|
|
|
};
|
|
|
|
|
|
|
|
typedef struct rb_shape rb_shape_t;
|
|
|
|
|
|
|
|
enum shape_type {
|
|
|
|
SHAPE_ROOT,
|
|
|
|
SHAPE_IVAR,
|
|
|
|
SHAPE_FROZEN,
|
2022-11-08 23:35:31 +03:00
|
|
|
SHAPE_CAPACITY_CHANGE,
|
|
|
|
SHAPE_INITIAL_CAPACITY,
|
2022-11-18 02:57:11 +03:00
|
|
|
SHAPE_T_OBJECT,
|
2022-12-09 01:16:52 +03:00
|
|
|
SHAPE_OBJ_TOO_COMPLEX,
|
2022-10-03 18:14:32 +03:00
|
|
|
};
|
|
|
|
|
|
|
|
#if SHAPE_IN_BASIC_FLAGS
|
|
|
|
static inline shape_id_t
|
|
|
|
RBASIC_SHAPE_ID(VALUE obj)
|
|
|
|
{
|
|
|
|
RUBY_ASSERT(!RB_SPECIAL_CONST_P(obj));
|
|
|
|
return (shape_id_t)(SHAPE_MASK & ((RBASIC(obj)->flags) >> SHAPE_FLAG_SHIFT));
|
|
|
|
}
|
|
|
|
|
|
|
|
static inline void
|
|
|
|
RBASIC_SET_SHAPE_ID(VALUE obj, shape_id_t shape_id)
|
|
|
|
{
|
|
|
|
// Ractors are occupying the upper 32 bits of flags, but only in debug mode
|
|
|
|
// Object shapes are occupying top bits
|
|
|
|
RBASIC(obj)->flags &= SHAPE_FLAG_MASK;
|
|
|
|
RBASIC(obj)->flags |= ((VALUE)(shape_id) << SHAPE_FLAG_SHIFT);
|
|
|
|
}
|
|
|
|
|
|
|
|
static inline shape_id_t
|
|
|
|
ROBJECT_SHAPE_ID(VALUE obj)
|
|
|
|
{
|
|
|
|
RBIMPL_ASSERT_TYPE(obj, RUBY_T_OBJECT);
|
|
|
|
return RBASIC_SHAPE_ID(obj);
|
|
|
|
}
|
|
|
|
|
|
|
|
static inline void
|
|
|
|
ROBJECT_SET_SHAPE_ID(VALUE obj, shape_id_t shape_id)
|
|
|
|
{
|
|
|
|
RBIMPL_ASSERT_TYPE(obj, RUBY_T_OBJECT);
|
|
|
|
RBASIC_SET_SHAPE_ID(obj, shape_id);
|
|
|
|
}
|
|
|
|
|
2022-11-01 00:05:37 +03:00
|
|
|
static inline shape_id_t
|
|
|
|
RCLASS_SHAPE_ID(VALUE obj)
|
|
|
|
{
|
|
|
|
RUBY_ASSERT(RB_TYPE_P(obj, T_CLASS) || RB_TYPE_P(obj, T_MODULE));
|
|
|
|
return RBASIC_SHAPE_ID(obj);
|
|
|
|
}
|
|
|
|
|
2022-10-03 18:14:32 +03:00
|
|
|
#else
|
|
|
|
|
|
|
|
static inline shape_id_t
|
|
|
|
ROBJECT_SHAPE_ID(VALUE obj)
|
|
|
|
{
|
|
|
|
RBIMPL_ASSERT_TYPE(obj, RUBY_T_OBJECT);
|
|
|
|
return (shape_id_t)(SHAPE_MASK & (RBASIC(obj)->flags >> SHAPE_FLAG_SHIFT));
|
|
|
|
}
|
|
|
|
|
|
|
|
static inline void
|
|
|
|
ROBJECT_SET_SHAPE_ID(VALUE obj, shape_id_t shape_id)
|
|
|
|
{
|
|
|
|
RBASIC(obj)->flags &= SHAPE_FLAG_MASK;
|
|
|
|
RBASIC(obj)->flags |= ((VALUE)(shape_id) << SHAPE_FLAG_SHIFT);
|
|
|
|
}
|
2022-11-01 00:05:37 +03:00
|
|
|
|
|
|
|
MJIT_SYMBOL_EXPORT_BEGIN
|
|
|
|
shape_id_t rb_rclass_shape_id(VALUE obj);
|
|
|
|
MJIT_SYMBOL_EXPORT_END
|
|
|
|
|
2022-12-02 17:43:53 +03:00
|
|
|
static inline shape_id_t RCLASS_SHAPE_ID(VALUE obj)
|
|
|
|
{
|
2022-11-01 00:05:37 +03:00
|
|
|
return rb_rclass_shape_id(obj);
|
|
|
|
}
|
|
|
|
|
2022-10-03 18:14:32 +03:00
|
|
|
#endif
|
|
|
|
|
2022-11-08 23:35:31 +03:00
|
|
|
rb_shape_t * rb_shape_get_root_shape(void);
|
2022-12-02 20:33:20 +03:00
|
|
|
int32_t rb_shape_id_offset(void);
|
2022-10-03 18:14:32 +03:00
|
|
|
|
2022-11-10 19:36:24 +03:00
|
|
|
rb_shape_t * rb_shape_get_parent(rb_shape_t * shape);
|
2022-10-03 18:14:32 +03:00
|
|
|
|
|
|
|
MJIT_SYMBOL_EXPORT_BEGIN
|
|
|
|
rb_shape_t* rb_shape_get_shape_by_id(shape_id_t shape_id);
|
|
|
|
shape_id_t rb_shape_get_shape_id(VALUE obj);
|
2022-12-23 21:18:57 +03:00
|
|
|
rb_shape_t * rb_shape_get_next_iv_shape(rb_shape_t * shape, ID id);
|
|
|
|
bool rb_shape_get_iv_index(rb_shape_t * shape, ID id, attr_index_t * value);
|
|
|
|
bool rb_shape_obj_too_complex(VALUE obj);
|
|
|
|
MJIT_SYMBOL_EXPORT_END
|
|
|
|
|
|
|
|
void rb_shape_set_shape(VALUE obj, rb_shape_t* shape);
|
2022-10-03 18:14:32 +03:00
|
|
|
rb_shape_t* rb_shape_get_shape(VALUE obj);
|
|
|
|
int rb_shape_frozen_shape_p(rb_shape_t* shape);
|
|
|
|
void rb_shape_transition_shape_frozen(VALUE obj);
|
2022-12-06 03:48:47 +03:00
|
|
|
void rb_shape_transition_shape_remove_ivar(VALUE obj, ID id, rb_shape_t *shape, VALUE * removed);
|
2022-11-08 23:35:31 +03:00
|
|
|
rb_shape_t * rb_shape_transition_shape_capa(rb_shape_t * shape, uint32_t new_capacity);
|
2022-10-03 18:14:32 +03:00
|
|
|
rb_shape_t* rb_shape_get_next(rb_shape_t* shape, VALUE obj, ID id);
|
|
|
|
|
2022-11-08 23:35:31 +03:00
|
|
|
rb_shape_t * rb_shape_rebuild_shape(rb_shape_t * initial_shape, rb_shape_t * dest_shape);
|
|
|
|
|
|
|
|
static inline uint32_t
|
|
|
|
ROBJECT_IV_CAPACITY(VALUE obj)
|
|
|
|
{
|
|
|
|
RBIMPL_ASSERT_TYPE(obj, RUBY_T_OBJECT);
|
2022-12-09 01:16:52 +03:00
|
|
|
// Asking for capacity doesn't make sense when the object is using
|
|
|
|
// a hash table for storing instance variables
|
|
|
|
RUBY_ASSERT(ROBJECT_SHAPE_ID(obj) != OBJ_TOO_COMPLEX_SHAPE_ID);
|
2022-11-08 23:35:31 +03:00
|
|
|
return rb_shape_get_shape_by_id(ROBJECT_SHAPE_ID(obj))->capacity;
|
|
|
|
}
|
|
|
|
|
2022-12-09 01:16:52 +03:00
|
|
|
static inline struct rb_id_table *
|
|
|
|
ROBJECT_IV_HASH(VALUE obj)
|
|
|
|
{
|
|
|
|
RBIMPL_ASSERT_TYPE(obj, RUBY_T_OBJECT);
|
|
|
|
RUBY_ASSERT(ROBJECT_SHAPE_ID(obj) == OBJ_TOO_COMPLEX_SHAPE_ID);
|
|
|
|
return (struct rb_id_table *)ROBJECT(obj)->as.heap.ivptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
static inline void
|
|
|
|
ROBJECT_SET_IV_HASH(VALUE obj, const struct rb_id_table *tbl)
|
|
|
|
{
|
|
|
|
RBIMPL_ASSERT_TYPE(obj, RUBY_T_OBJECT);
|
|
|
|
RUBY_ASSERT(ROBJECT_SHAPE_ID(obj) == OBJ_TOO_COMPLEX_SHAPE_ID);
|
|
|
|
ROBJECT(obj)->as.heap.ivptr = (VALUE *)tbl;
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t rb_id_table_size(const struct rb_id_table *tbl);
|
|
|
|
|
2022-10-15 19:37:44 +03:00
|
|
|
static inline uint32_t
|
|
|
|
ROBJECT_IV_COUNT(VALUE obj)
|
|
|
|
{
|
2022-12-09 01:16:52 +03:00
|
|
|
if (ROBJECT_SHAPE_ID(obj) == OBJ_TOO_COMPLEX_SHAPE_ID) {
|
|
|
|
return (uint32_t)rb_id_table_size(ROBJECT_IV_HASH(obj));
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
RBIMPL_ASSERT_TYPE(obj, RUBY_T_OBJECT);
|
|
|
|
RUBY_ASSERT(ROBJECT_SHAPE_ID(obj) != OBJ_TOO_COMPLEX_SHAPE_ID);
|
|
|
|
return rb_shape_get_shape_by_id(ROBJECT_SHAPE_ID(obj))->next_iv_index;
|
|
|
|
}
|
2022-10-15 19:37:44 +03:00
|
|
|
}
|
|
|
|
|
2022-11-08 23:35:31 +03:00
|
|
|
static inline uint32_t
|
|
|
|
RBASIC_IV_COUNT(VALUE obj)
|
|
|
|
{
|
|
|
|
return rb_shape_get_shape_by_id(rb_shape_get_shape_id(obj))->next_iv_index;
|
|
|
|
}
|
|
|
|
|
2022-11-01 00:05:37 +03:00
|
|
|
static inline uint32_t
|
|
|
|
RCLASS_IV_COUNT(VALUE obj)
|
|
|
|
{
|
|
|
|
RUBY_ASSERT(RB_TYPE_P(obj, RUBY_T_CLASS) || RB_TYPE_P(obj, RUBY_T_MODULE));
|
|
|
|
uint32_t ivc = rb_shape_get_shape_by_id(RCLASS_SHAPE_ID(obj))->next_iv_index;
|
|
|
|
return ivc;
|
|
|
|
}
|
|
|
|
|
2022-10-03 18:14:32 +03:00
|
|
|
rb_shape_t * rb_shape_alloc(ID edge_name, rb_shape_t * parent);
|
2022-11-08 23:35:31 +03:00
|
|
|
rb_shape_t * rb_shape_alloc_with_size_pool_index(ID edge_name, rb_shape_t * parent, uint8_t size_pool_index);
|
2022-10-03 20:52:40 +03:00
|
|
|
rb_shape_t * rb_shape_alloc_with_parent_id(ID edge_name, shape_id_t parent_id);
|
2022-10-03 18:14:32 +03:00
|
|
|
|
2022-12-13 18:11:57 +03:00
|
|
|
rb_shape_t *rb_shape_traverse_from_new_root(rb_shape_t *initial_shape, rb_shape_t *orig_shape);
|
|
|
|
|
2022-10-03 18:14:32 +03:00
|
|
|
bool rb_shape_set_shape_id(VALUE obj, shape_id_t shape_id);
|
|
|
|
|
|
|
|
VALUE rb_obj_debug_shape(VALUE self, VALUE obj);
|
2022-12-09 01:16:52 +03:00
|
|
|
void rb_shape_set_too_complex(VALUE obj);
|
2022-10-03 18:14:32 +03:00
|
|
|
|
2022-12-23 21:18:57 +03:00
|
|
|
// For ext/objspace
|
2022-12-06 14:56:51 +03:00
|
|
|
RUBY_SYMBOL_EXPORT_BEGIN
|
|
|
|
typedef void each_shape_callback(rb_shape_t * shape, void *data);
|
|
|
|
void rb_shape_each_shape(each_shape_callback callback, void *data);
|
|
|
|
size_t rb_shape_memsize(rb_shape_t *shape);
|
|
|
|
size_t rb_shape_edges_count(rb_shape_t *shape);
|
|
|
|
size_t rb_shape_depth(rb_shape_t *shape);
|
2022-12-23 21:18:57 +03:00
|
|
|
shape_id_t rb_shape_id(rb_shape_t * shape);
|
2022-12-06 14:56:51 +03:00
|
|
|
RUBY_SYMBOL_EXPORT_END
|
|
|
|
|
2022-10-03 18:14:32 +03:00
|
|
|
#endif
|