File: typeslots.py

package info (click to toggle)
python3.5 3.5.3-1%2Bdeb9u1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 88,448 kB
  • sloc: python: 491,123; ansic: 410,863; sh: 17,674; asm: 14,322; cpp: 4,123; makefile: 2,255; objc: 761; lisp: 502; exp: 499; pascal: 85; xml: 74; csh: 21
file content (43 lines) | stat: -rwxr-xr-x 1,246 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/usr/bin/python
# Usage: typeslots.py < Include/typeslots.h typeslots.inc

import sys, re

def generate_typeslots(out=sys.stdout):
    out.write("/* Generated by typeslots.py */\n")
    res = {}
    for line in sys.stdin:
        m = re.match("#define Py_([a-z_]+) ([0-9]+)", line)
        if not m:
            continue
        member = m.group(1)
        if member.startswith("tp_"):
            member = "ht_type."+member
        elif member.startswith("am_"):
            member = "as_async."+member
        elif member.startswith("nb_"):
            member = "as_number."+member
        elif member.startswith("mp_"):
            member = "as_mapping."+member
        elif member.startswith("sq_"):
            member = "as_sequence."+member
        elif member.startswith("bf_"):
            member = "as_buffer."+member
        res[int(m.group(2))] = member

    M = max(res.keys())+1
    for i in range(1,M):
        if i in res:
            out.write("offsetof(PyHeapTypeObject, %s),\n" % res[i])
        else:
            out.write("0,\n")

def main():
    if len(sys.argv) == 2:
        with open(sys.argv[1], "w") as f:
            generate_typeslots(f)
    else:
        generate_typeslots()

if __name__ == "__main__":
    main()