-
Notifications
You must be signed in to change notification settings - Fork 22
/
embossc
executable file
·104 lines (87 loc) · 3.61 KB
/
embossc
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#!/usr/bin/env python3
# Copyright 2019 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Main driver program for the Emboss compiler."""
import argparse
import os
import sys
def _parse_args(argv):
parser = argparse.ArgumentParser(description="Emboss compiler")
parser.add_argument("--color-output",
default="if_tty",
choices=["always", "never", "if_tty", "auto"],
help="Print error messages using color. 'auto' is a "
"synonym for 'if_tty'.")
parser.add_argument("--import-dir", "-I",
dest="import_dirs",
action="append",
default=["."],
help="A directory to use when searching for imported "
"embs. If no import-dirs are specified, the "
"current directory will be used.")
parser.add_argument("--generate",
nargs=1,
choices=["cc"],
default="cc",
help="Which back end to use. Currently only C++ is "
"supported.")
parser.add_argument("--output-path",
nargs=1,
default=".",
help="""Prefix path to use for the generated output file.
Defaults to '.'""")
parser.add_argument("--output-file",
nargs=1,
help="""File name to be used for the generated output
file. Defaults to input_file suffixed by '.h'""")
parser.add_argument("--cc-enum-traits",
action=argparse.BooleanOptionalAction,
default=True,
help="""Controls generation of EnumTraits by the C++
backend""")
parser.add_argument("input_file",
type=str,
nargs=1,
help=".emb file to compile.")
return parser.parse_args(argv[1:])
def main(argv):
flags = _parse_args(argv)
base_path = os.path.dirname(__file__) or "."
sys.path.append(base_path)
from compiler.back_end.cpp import ( # pylint:disable=import-outside-toplevel
emboss_codegen_cpp, header_generator
)
from compiler.front_end import ( # pylint:disable=import-outside-toplevel
emboss_front_end
)
ir, _, errors = emboss_front_end.parse_and_log_errors(
flags.input_file[0], flags.import_dirs, flags.color_output)
if errors:
return 1
config = header_generator.Config(include_enum_traits=flags.cc_enum_traits)
header, errors = emboss_codegen_cpp.generate_headers_and_log_errors(
ir, flags.color_output, config)
if errors:
return 1
if flags.output_file:
output_file = flags.output_file[0]
else:
output_file = flags.input_file[0] + ".h"
output_filepath = os.path.join(flags.output_path[0], output_file)
os.makedirs(os.path.dirname(output_filepath), exist_ok=True)
with open(output_filepath, "w") as output:
output.write(header)
return 0
if __name__ == "__main__":
sys.exit(main(sys.argv))