• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1#!/usr/bin/env python
2# -*- coding: utf-8 -*-
3
4#
5# Copyright (c) 2022 Huawei Device Co., Ltd.
6# Licensed under the Apache License, Version 2.0 (the "License");
7# you may not use this file except in compliance with the License.
8# You may obtain a copy of the License at
9#
10#     http://www.apache.org/licenses/LICENSE-2.0
11#
12# Unless required by applicable law or agreed to in writing, software
13# distributed under the License is distributed on an "AS IS" BASIS,
14# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15# See the License for the specific language governing permissions and
16# limitations under the License.
17#
18
19import argparse
20import os
21import sys
22
23
24def decode_cfg_line(data):
25    data.replace('\n', '').replace('\r', '')
26    data = data.strip()
27    if (len(data) == 0 or data[0] == '#'):
28        return "", ""
29    strs = data.split('=')
30    if len(strs) <= 1:
31        return "", ""
32    return strs[0].strip(), strs[1].strip()
33
34
35def get_param_from_cfg(cfg_name):
36    data_dict = {}
37    with open(cfg_name) as afile:
38        data = afile.readline()
39        while data:
40            name, value = decode_cfg_line(data)
41            if len(name) != 0 and len(value) != 0:
42                data_dict[name] = value
43                print("sample file name={%s %s}" % (name, value))
44            data = afile.readline()
45    return data_dict
46
47
48def decode_code_line(data):
49    data.replace('\n', '').replace('\r', '')
50    data = data.strip()
51    if (not data.startswith("PARAM_MAP")):
52        return "", ""
53    data_len = len(data)
54    data = data[len("PARAM_MAP") + 1 :  data_len - 1]
55    data = data.strip()
56    strs = data.split(',')
57    if len(strs) <= 1:
58        return "", ""
59    return strs[0].strip(), data[len(strs[0]) + 1:].strip()
60
61
62def get_param_from_c_code(code_name):
63    data_dict = {}
64    with open(code_name, "r+") as afile:
65        data = afile.readline()
66        while data:
67            name, value = decode_code_line(data)
68            if len(name) != 0 and len(value) != 0:
69                data_dict[name] = value
70            data = afile.readline()
71        afile.truncate(0)
72    return data_dict
73
74
75def write_map_to_code(code_name, data_dict):
76    try:
77        with open(code_name, "w") as f:
78            # start with 0
79            f.seek(0)
80            # write file header
81            f.write('#ifndef PARAM_LITE_DEF_CFG_' + os.linesep)
82            f.write('#define PARAM_LITE_DEF_CFG_' + os.linesep)
83            f.write('#include <stdint.h>' + os.linesep + os.linesep)
84            f.write('#ifdef __cplusplus' + os.linesep)
85            f.write('#if __cplusplus' + os.linesep)
86            f.write('extern "C" {' + os.linesep)
87            f.write('#endif' + os.linesep)
88            f.write('#endif' + os.linesep + os.linesep)
89
90            #define struct
91            f.write('typedef struct Node_ {' + os.linesep)
92            f.write('    const char *name;' + os.linesep)
93            f.write('    const char *value;' + os.linesep)
94            f.write('} Node;' + os.linesep  + os.linesep)
95            f.write('#define PARAM_MAP(name, value) {(const char *)#name, (const char *)#value},')
96            f.write(os.linesep  + os.linesep)
97            # write data
98            f.write('static Node g_paramDefCfgNodes[] = {' + os.linesep)
99            for name, value in  data_dict.items():
100                if (value.startswith("\"")):
101                    tmp_str = "    PARAM_MAP({0}, {1})".format(name, value)
102                    f.write(tmp_str + os.linesep)
103                else:
104                    tmp_str = "    PARAM_MAP({0}, \"{1}\")".format(name, value)
105                    f.write(tmp_str + os.linesep)
106            f.write('};' + os.linesep + os.linesep)
107
108            #end
109            f.write('#ifdef __cplusplus' + os.linesep)
110            f.write('#if __cplusplus' + os.linesep)
111            f.write('}' + os.linesep)
112            f.write('#endif' + os.linesep)
113            f.write('#endif' + os.linesep)
114            f.write('#endif // PARAM_LITE_DEF_CFG_' + os.linesep)
115            f.write(os.linesep)
116            f.truncate()
117    except IOError:
118        print("Error: open or write file %s fail" % {code_name})
119    return 0
120
121
122def add_to_code_dict(code_dict, cfg_dict, high=True):
123    for name, value in  cfg_dict.items():
124        # check if name exit
125        has_key = name in code_dict
126        if has_key and high:
127            code_dict[name] = value
128        elif not has_key:
129            code_dict[name] = value
130    return code_dict
131
132
133def main():
134    parser = argparse.ArgumentParser(
135    description='A common change param.para file to h.')
136    parser.add_argument(
137        '--source',
138        action='append',
139        help='The source to change.',
140        required=True)
141    parser.add_argument(
142        '--dest_dir',
143        help='Path that the source should be changed to.',
144        required=True)
145    args = parser.parse_args()
146
147    out_dir = args.dest_dir
148    if not os.path.exists(out_dir):
149        os.makedirs(out_dir, exist_ok=True)
150    print("out_dir {}".format(out_dir))
151
152    for source in args.source:
153        print("source {}".format(out_dir))
154        if not os.path.exists(source):
155            continue
156
157        src_dict = get_param_from_cfg(source)
158        dst = "".join([out_dir, "param_cfg.h"])
159
160        if os.path.exists(dst):
161            dst_dict = get_param_from_c_code(dst)
162        else:
163            dst_dict = {}
164
165        dst_dict = add_to_code_dict(dst_dict, src_dict, False)
166        write_map_to_code(dst, dst_dict)
167    return 0
168
169
170if __name__ == '__main__':
171    sys.exit(main())
172