FilesExpand file tree

 main

/

make_root_pointers.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

55 lines (37 loc) · 1.43 KB

 main

/

make_root_pointers.py

Copy path

Top

File metadata and controls

• Code
• Blame

55 lines (37 loc) · 1.43 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

44

45

46

47

48

49

50

51

52

53

54

55

"""

This pre-processor parses a single file containing a list of

MP_REGISTER_ROOT_POINTER(variable declaration) items.

These are used to generate a header with the required entries for

"struct _mp_state_vm_t" in py/mpstate.h

"""

from __future__ import print_function

import argparse

import io

import re

PATTERN = re.compile(r"MP_REGISTER_ROOT_POINTER\((.*?)\);")

def find_root_pointer_registrations(filename):

"""Find any MP_REGISTER_ROOT_POINTER definitions in the provided file.

:param str filename: path to file to check

:return: List[variable_declaration]

"""

with io.open(filename, encoding="utf-8") as c_file_obj:

return set(re.findall(PATTERN, c_file_obj.read()))

def generate_root_pointer_header(root_pointers):

"""Generate header with root pointer entries.

:param List[variable_declaration] root_pointers: root pointer declarations

:return: None

"""

# Print header file for all external modules.

print("// Automatically generated by make_root_pointers.py.")

print()

for item in root_pointers:

print(item, end=";")

print()

def main():

parser = argparse.ArgumentParser()

parser.add_argument("file", nargs=1, help="file with MP_REGISTER_ROOT_POINTER definitions")

args = parser.parse_args()

root_pointers = find_root_pointer_registrations(args.file[0])

generate_root_pointer_header(sorted(root_pointers))

if __name__ == "__main__":

main()