Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
tjh.dev
kernel
os
linux
1#!/usr/bin/env python3
2# ex: set filetype=python:
3
4"""Generate code to handle XDR enum types"""
5
6from generators import SourceGenerator, create_jinja2_environment
7from xdr_ast import _XdrEnum, public_apis, big_endian, get_header_name
8from xdr_parse import get_xdr_enum_validation
9
10
11class XdrEnumGenerator(SourceGenerator):
12 """Generate source code for XDR enum types"""
13
14 def __init__(self, language: str, peer: str):
15 """Initialize an instance of this class"""
16 self.environment = create_jinja2_environment(language, "enum")
17 self.peer = peer
18
19 def emit_declaration(self, node: _XdrEnum) -> None:
20 """Emit one declaration pair for an XDR enum type"""
21 if node.name in public_apis:
22 template = self.environment.get_template("declaration/enum.j2")
23 print(template.render(name=node.name))
24
25 def emit_definition(self, node: _XdrEnum) -> None:
26 """Emit one definition for an XDR enum type"""
27 template = self.environment.get_template("definition/open.j2")
28 print(template.render(name=node.name))
29
30 template = self.environment.get_template("definition/enumerator.j2")
31 for enumerator in node.enumerators:
32 print(template.render(name=enumerator.name, value=enumerator.value))
33
34 if node.name in big_endian:
35 template = self.environment.get_template("definition/close_be.j2")
36 else:
37 template = self.environment.get_template("definition/close.j2")
38 print(template.render(name=node.name))
39
40 def emit_decoder(self, node: _XdrEnum) -> None:
41 """Emit one decoder function for an XDR enum type"""
42 if node.name in big_endian:
43 template = self.environment.get_template("decoder/enum_be.j2")
44 else:
45 template = self.environment.get_template("decoder/enum.j2")
46 print(
47 template.render(
48 name=node.name,
49 enumerators=node.enumerators,
50 validate=get_xdr_enum_validation(),
51 )
52 )
53
54 def emit_encoder(self, node: _XdrEnum) -> None:
55 """Emit one encoder function for an XDR enum type"""
56 if node.name in big_endian:
57 template = self.environment.get_template("encoder/enum_be.j2")
58 else:
59 template = self.environment.get_template("encoder/enum.j2")
60 print(template.render(name=node.name))
61
62 def emit_maxsize(self, node: _XdrEnum) -> None:
63 """Emit one maxsize macro for an XDR enum type"""
64 macro_name = get_header_name().upper() + "_" + node.name + "_sz"
65 template = self.environment.get_template("maxsize/enum.j2")
66 print(
67 template.render(
68 macro=macro_name,
69 width=" + ".join(node.symbolic_width()),
70 )
71 )