Skip to content

Latest commit

 

History

History
412 lines (287 loc) · 12.7 KB

slot.md

File metadata and controls

412 lines (287 loc) · 12.7 KB
Raw

slotimage title

contents

related file

  • cpython/Objects/typeobject.c
  • cpython/Objects/clinic/typeobject.c.h
  • cpython/Objects/object.c
  • cpython/Include/cpython/object.h
  • cpython/Objects/descrobject.c
  • cpython/Include/descrobject.h
  • cpython/Python/structmember.c
  • cpython/Include/structmember.h

slot

prerequisites

  • python's attribute accessing behaviour (described in descr)
  • python's descriptor protocol (mentioned in descr)
  • python MRO (described in type)

example

class A(object):
    __slots__ = ["wing", "leg"]

    x = 3

what's the difference of accessing attribute wing and x of instance a ?

what's the difference of accessing attribute wing and x of type A ?

instance access

instance access wing

before set a value

>>> a = A()
>>> a.wing
Traceback (most recent call last):
  File "<input>", line 1, in <module>
AttributeError: wing

according to the attribute accessing procedure described in descr

we can draw the procedure of accessing a.wing in a brief view

instance_desc

the descriptor protocol help us find an object named descr, type is member_descriptor, if you call

repr(descr)
descr: <member 'wing' of 'A' objects>

the descr has a member named d_member, which stores information about the attribute name, type and offset

with the informations stored in d_member, you are able to access the address of attribute in a very fast way

in the current example, if the begin address of the instance a is 0x00, address after adding the offset is 0x18, you can cast the address in 0x18 to a proper object according to type stored in d_member

/* Include/structmember.h */
#define T_OBJECT_EX 16  /* Like T_OBJECT, but raises AttributeError
                           when the value is NULL, instead of
                           converting to None. */

/* Python/structmember.c */
addr += l->offset;
switch (l->type) {
	/* ... */
    case T_OBJECT_EX:
        v = *(PyObject **)addr;
        /* because wing attribute of instance a is never set,
        it raise an attribute error */
        if (v == NULL)
            PyErr_SetString(PyExc_AttributeError, l->name);
        Py_XINCREF(v);
        break;
    /* ... */
}

offset

after set a value

>>> a = A()
>>> a.wing = "wingA"
>>> a.wing
'wingA'

because wing attribute of instance a is set before, AttributeError won't be raised

instance access x

>>> a.x
>>> 3

type of descr is int, it's not a data descriptor(does not have __get__ or __set__ attribute), so the descr will be returned directly

instance_normal

you are not able to create any other attribute in instance a if A defined __slots__, we will figure out why later

>>> a.not_exist = 100
Traceback (most recent call last):
File "<input>", line 1, in <module>
AttributeError: 'A' object has no attribute 'not_exist'

access type

type access wing

>>> A.wing
<member 'wing' of 'A' objects>
>>> type(A.wing)
<class 'member_descriptor'>

the procedure of accessing A.wing is nearly the same as a.wing

type_desc

type access x

>>> A.x
3

the procedure of accessing A.x is nearly the same as a.x

type_normal

difference

with slots

how does attributes initialize in the creation of class A ?

we can learn about the creation procedure in type->creation of class

type object has many fields in it's C structure, the following picture only shows what's necessary for this topic

what's in the __slots__ is sorted and stored as a tuple in ht_slots

the two attributes in __slots__ is preallocated in the tail of the newly created type object A and two PyMemberDef pointers are stored in the order of ht_slots

while attribute x stays in the tp_dict field

the tp_dict field does not have a key named __dict__

type_create

how does attributes initialize in the creation of instance a ?

the memory location of the attributes in __slots__ are preallocated

instance_create

lookup procedure in MRO ?

it just iter through every type object in MRO, and if the name in tp_dict field, retuen tp_dict[name]

/* cpython/Objects/typeobject.c */
/* for the instance a, if we access a.wing
   type: <class '__main__.A'>
   mro: (<class '__main__.A'>, <class 'object'>)
   name: 'wing'
*/
mro = type->tp_mro;
n = PyTuple_GET_SIZE(mro);
for (i = 0; i < n; i++) {
    base = PyTuple_GET_ITEM(mro, i);
    dict = ((PyTypeObject *)base)->tp_dict;
    // in python representation: res = dict[name]
    res = _PyDict_GetItem_KnownHash(dict, name, hash);
    if (res != NULL)
        break;
    if (PyErr_Occurred()) {
        *error = -1;
        goto done;
    }
}

if we try to access attribute wing

>>> type(A.wing)
<class 'member_descriptor'>
>>> type(a).__mro__
(<class '__main__.A'>, <class 'object'>)
>>> print(a.wing)
wingA

the following pseudo shows what's going on

res = None
for each_type in type(a).__mro__:
	if "wing" in each_type.__dict__:
    	res = each_type.__dict__["wing"]
        break
# do the attribute accessing procedure described in the other article
...
if res is a data_descriptor:
    # res is A.wing, it's type is member_descriptor
    # it stores the offset and some other information of the real object stored in instance
    # member_descriptor.__get__ will find the address in a + offset, and cast it to a PyObject and return that object
    return res.__get__(a, type(a))
...

access_slot_attribute

access_slot_attribute2

if we try to access or set a not exist attribute

>>> a.not_exist = 33
Traceback (most recent call last):
File "<input>", line 1, in <module>
AttributeError: 'A' object has no attribute 'not_exist'

follow the descriptor protocol mentioned in descr, we found

res = None
for each_type in type(a).__mro__:
	if "not_exist" in each_type.__dict__:
    	res = each_type.__dict__["not_exist"]
    	break
if res is None:
    # try to find "not_exist" in a.__dict__
    if not hasattr(a, "__dict__") or "not_exist" not in a.__dict__:
    	# go to here
    	raise AttributeError
    return a.__dict__["not_exist"]

when the __slots__ attribute set, tp_dictoffset of type(a) will be 0, it means instance a does not have __dict__ attribute to stores any other attribute name

so AttributeError will be raised

access_slot_not_exist_attribute

without slots

class A(object):

    x = 3
    wing = "wingA"
    leg = "legA"

how does attributes initialize in the creation of class A ?

the tp_dict field has a key named __dict__

type_create_no_slot

how does attributes initialize in the creation of instance a ?

instance_create_no_slot

lookup procedure in MRO ?

the lookup procedure is the same as the procedure in with slots

access_no_slot_attribute

if we try to access or set a not exist attribute

>>> a.not_exist = 33
>>> print(a.not_exist)

follow the descriptor protocol mentioned in descr, we found

res = None
for each_type in type(a).__mro__:
	if "not_exist" in each_type.__dict__:
    	res = each_type.__dict__["not_exist"]
    	break
if res is None:
    # try to find "not_exist" in a.__dict__
    if not hasattr(a, "__dict__") or "not_exist" not in a.__dict__:
    	raise AttributeError
    # go to here
    return a.__dict__["not_exist"]

the __slots__ attribute is not set, tp_dictoffset of type(a) is 16, it means instance a does have __dict__ attribute to stores other attribute name, the location is in (char *)a + 16

so attribute name can be stored in a.__dict__

access_no_slot_not_exist_attribute

access_no_slot_not_exist_attribute2

memory saving measurement

with __slots__

./ipython
>>> import ipython_memory_usage.ipython_memory_usage as imu
>>> imu.start_watching_memory()
In [2] used 0.1367 MiB RAM in 3.59s, peaked 0.00 MiB above current, total RAM usage 41.16 MiB
class MyClass(object):
    __slots__ = ['name', 'identifier']
    def __init__(self, name, identifier):
            self.name = name
            self.identifier = identifier
num = 1024*256
x = [MyClass(1,1) for i in range(num)]

used 27.5508 MiB RAM in 0.28s, peaked 0.00 MiB above current, total RAM usage 69.18 MiB

without __slots__

./ipython
>>> import ipython_memory_usage.ipython_memory_usage as imu
>>> imu.start_watching_memory()
In [2] used 0.1367 MiB RAM in 3.59s, peaked 0.00 MiB above current, total RAM usage 41.16 MiB

class MyClass(object):
        def __init__(self, name, identifier):
                self.name = name
                self.identifier = identifier
num = 1024*256
x = [MyClass(1,1) for i in range(num)]

used 56.0234 MiB RAM in 0.34s, peaked 0.00 MiB above current, total RAM usage 97.63 MiB

the class without __slots__ consumes nearly twice memory than the class with __slots__, it mainly because attributes in the class with __slots__ is preallocated, which occupies 8 bytes per python object pointer, and the class without __slots__ has the overhead of the dict object, which occupies more than a dozen 8 bytes pointers even without storing any elements

read more