306 lines
11 KiB
Python
306 lines
11 KiB
Python
|
# coding: UTF-8
|
||
|
|
"""
|
||
|
|
将项目描述文件解析为任务队列并执行
|
||
|
|
"""
|
||
|
|
import json
|
||
|
|
import glob
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import argparse
|
||
|
|
import time
|
||
|
|
from multiprocessing import Pool
|
||
|
|
|
||
|
|
|
||
|
|
class ArmccParser(object):
|
||
|
|
'''
|
||
|
|
gcc工具链项目文件解析
|
||
|
|
'''
|
||
|
|
VERSION = "1.0.0"
|
||
|
|
NAME = ""
|
||
|
|
AS = ""
|
||
|
|
CC = ""
|
||
|
|
LD = ""
|
||
|
|
OBJCOPY = ""
|
||
|
|
AR = ""
|
||
|
|
Type = ""
|
||
|
|
OutPath = ""
|
||
|
|
IsCreateBin = False
|
||
|
|
IsCreateHex = False
|
||
|
|
IsCreateMap = False
|
||
|
|
ASFlag = []
|
||
|
|
CCFlag = []
|
||
|
|
LDFlag = []
|
||
|
|
LDFile = ""
|
||
|
|
Includes = []
|
||
|
|
LibPaths = []
|
||
|
|
LibFiles = []
|
||
|
|
SrcFiles = []
|
||
|
|
|
||
|
|
def __init__(self, filename="project.json"):
|
||
|
|
# jsonfile = filename if filename else "project.json"
|
||
|
|
if not os.path.isfile(filename):
|
||
|
|
print("未找到项目描述文件%s" % (filename))
|
||
|
|
exit()
|
||
|
|
else:
|
||
|
|
data = json.load(open(filename, encoding='utf8'))
|
||
|
|
if not data.get('version', "") == self.VERSION:
|
||
|
|
print("项目描述文件版本不兼容")
|
||
|
|
exit()
|
||
|
|
# 项目名称
|
||
|
|
self.NAME = data.get('name', "")
|
||
|
|
# 工具链
|
||
|
|
if data.get("toolchain", ""):
|
||
|
|
path = data["toolchain"].get("path", "")
|
||
|
|
prefix = data["toolchain"].get("prefix", "")
|
||
|
|
self.AS = "%s%s%s" % (
|
||
|
|
path, prefix, data["toolchain"]["as"]) if data["toolchain"].get("as", "") else ""
|
||
|
|
self.CC = "%s%s%s" % (
|
||
|
|
path, prefix, data["toolchain"]["cc"]) if data["toolchain"].get("cc", "") else ""
|
||
|
|
self.LD = "%s%s%s" % (
|
||
|
|
path, prefix, data["toolchain"]["ld"]) if data["toolchain"].get("ld", "") else ""
|
||
|
|
self.OBJCOPY = "%s%s%s" % (
|
||
|
|
path, prefix, data["toolchain"]["objcopy"]) if data["toolchain"].get("objcopy", "") else ""
|
||
|
|
self.AR = "%s%s%s" % (
|
||
|
|
path, prefix, data["toolchain"]["ar"]) if data["toolchain"].get("as", "") else ""
|
||
|
|
# 编译参数
|
||
|
|
if data.get("flag", ""):
|
||
|
|
self.ASFlag = " ".join(data["flag"].get("as", []))
|
||
|
|
self.CCFlag = " ".join(data["flag"].get("cc", []))
|
||
|
|
self.LDFlag = " ".join(data["flag"].get("ld", []))
|
||
|
|
|
||
|
|
# 输出设置
|
||
|
|
if data.get("output", ""):
|
||
|
|
self.Type = data['output'].get("type", "elf")
|
||
|
|
self.OutPath = data['output'].get("path", "")
|
||
|
|
self.IsCreateBin = data['output'].get("bin", False)
|
||
|
|
self.IsCreateHex = data['output'].get("hex", False)
|
||
|
|
# 头文件路径
|
||
|
|
if data.get("include", []):
|
||
|
|
self.Includes = " ".join(map(lambda x: "-I " + x, data['include']))
|
||
|
|
# 源文件
|
||
|
|
if data.get("src", []):
|
||
|
|
for item in data['src']:
|
||
|
|
self.SrcFiles.extend(glob.glob(item))
|
||
|
|
def GetTargetFile(self, Ext):
|
||
|
|
return self.OutPath + self.NAME + Ext
|
||
|
|
|
||
|
|
def GetOutFile(self, infile):
|
||
|
|
return self.OutPath + os.path.basename(infile).split(".")[0] + ".o"
|
||
|
|
|
||
|
|
def CheckUpdate(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def GetCompileCMD(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def GetArCMD(self):
|
||
|
|
return "@%s -r %s %s" % (project.AR, self.OutPath + "lib" + self.NAME + ".a", " ".join(
|
||
|
|
glob.glob(self.OutPath + "*.o")))
|
||
|
|
|
||
|
|
def GetLinkCMD(self):
|
||
|
|
return "@%s %s %s -o %s" % (project.LD, " ".join(
|
||
|
|
glob.glob(self.OutPath + "*.o")), self.LDFlag, project.GetTargetFile(".axf"))
|
||
|
|
|
||
|
|
def GetCreateBinCMD(self):
|
||
|
|
return "@%s --bin -o %s %s" % (project.OBJCOPY, project.GetTargetFile(".bin"), project.GetTargetFile(".axf"))
|
||
|
|
|
||
|
|
def GetCreateHexCMD(self):
|
||
|
|
return "@%s --i32combined -o %s %s" % (project.OBJCOPY, project.GetTargetFile(".hex"), project.GetTargetFile(".axf"))
|
||
|
|
|
||
|
|
def GetCompileTasks(self):
|
||
|
|
_taskList = []
|
||
|
|
for infile in self.SrcFiles:
|
||
|
|
outfile = self.GetOutFile(infile)
|
||
|
|
if os.path.exists(outfile): # 跳过没有变化文件
|
||
|
|
if os.path.getmtime(outfile) > os.path.getmtime(infile):
|
||
|
|
continue
|
||
|
|
if os.path.splitext(infile)[1] == ".c":
|
||
|
|
_taskList.append("@%s -c %s -o %s %s %s" %
|
||
|
|
(self.CC, infile, outfile, self.Includes, self.CCFlag))
|
||
|
|
if os.path.splitext(infile)[1] == ".s":
|
||
|
|
_taskList.append("@%s %s -o %s %s %s" %
|
||
|
|
(self.AS, infile, outfile, self.Includes, self.ASFlag))
|
||
|
|
return _taskList
|
||
|
|
|
||
|
|
class GccParser(object):
|
||
|
|
'''
|
||
|
|
gcc工具链项目文件解析
|
||
|
|
'''
|
||
|
|
VERSION = "1.0.0"
|
||
|
|
NAME = ""
|
||
|
|
AS = ""
|
||
|
|
CC = ""
|
||
|
|
LD = ""
|
||
|
|
OBJCOPY = ""
|
||
|
|
AR = ""
|
||
|
|
Type = ""
|
||
|
|
OutPath = ""
|
||
|
|
IsCreateBin = False
|
||
|
|
IsCreateHex = False
|
||
|
|
IsCreateMap = False
|
||
|
|
ASFlag = []
|
||
|
|
CCFlag = []
|
||
|
|
LDFlag = []
|
||
|
|
LDFile = ""
|
||
|
|
Includes = []
|
||
|
|
LibPaths = []
|
||
|
|
LibFiles = []
|
||
|
|
SrcFiles = []
|
||
|
|
|
||
|
|
def __init__(self, filename="project.json"):
|
||
|
|
# jsonfile = filename if filename else "project.json"
|
||
|
|
if not os.path.isfile(filename):
|
||
|
|
print("未找到项目描述文件%s" % (filename))
|
||
|
|
exit()
|
||
|
|
else:
|
||
|
|
data = json.load(open(filename, encoding='utf8'))
|
||
|
|
if not data.get('version', "") == self.VERSION:
|
||
|
|
print("项目描述文件版本不兼容")
|
||
|
|
exit()
|
||
|
|
# 项目名称
|
||
|
|
self.NAME = data.get('name', "")
|
||
|
|
# 工具链
|
||
|
|
if data.get("toolchain", ""):
|
||
|
|
path = data["toolchain"].get("path", "")
|
||
|
|
prefix = data["toolchain"].get("prefix", "")
|
||
|
|
self.AS = "%s%s%s" % (
|
||
|
|
path, prefix, data["toolchain"]["as"]) if data["toolchain"].get("as", "") else ""
|
||
|
|
self.CC = "%s%s%s" % (
|
||
|
|
path, prefix, data["toolchain"]["cc"]) if data["toolchain"].get("cc", "") else ""
|
||
|
|
self.LD = "%s%s%s" % (
|
||
|
|
path, prefix, data["toolchain"]["ld"]) if data["toolchain"].get("ld", "") else ""
|
||
|
|
self.OBJCOPY = "%s%s%s" % (
|
||
|
|
path, prefix, data["toolchain"]["objcopy"]) if data["toolchain"].get("objcopy", "") else ""
|
||
|
|
self.AR = "%s%s%s" % (
|
||
|
|
path, prefix, data["toolchain"]["ar"]) if data["toolchain"].get("as", "") else ""
|
||
|
|
# 编译参数
|
||
|
|
if data.get("flag", ""):
|
||
|
|
self.ASFlag = " ".join(data["flag"].get("as", []))
|
||
|
|
self.CCFlag = " ".join(data["flag"].get("cc", []))
|
||
|
|
self.LDFlag = " ".join(data["flag"].get("ld", []))
|
||
|
|
# 单独配置的一些参数
|
||
|
|
if data.get('ldfile', ""):
|
||
|
|
if not "-T" in self.LDFlag:
|
||
|
|
self.LDFlag += " -T%s" % (data['ldfile'])
|
||
|
|
else: # 如果重复定义-T参数的做法
|
||
|
|
print('ERROR: already define -T in LDFlag')
|
||
|
|
if data.get('lib', ""):
|
||
|
|
if data['lib'].get('path', ""):
|
||
|
|
self.LDFlag += " %s" % (" ".join(map(lambda x: "-L" +
|
||
|
|
x, data['lib']['path'])))
|
||
|
|
if data['lib'].get('file', ""):
|
||
|
|
self.LDFlag += " %s" % (" ".join(map(lambda x: "-l" +
|
||
|
|
x, data['lib']['file'])))
|
||
|
|
self.LDFile = data.get('ldfile', "")
|
||
|
|
# 输出设置
|
||
|
|
if data.get("output", ""):
|
||
|
|
self.Type = data['output'].get("type", "elf")
|
||
|
|
self.OutPath = data['output'].get("path", "")
|
||
|
|
self.IsCreateBin = data['output'].get("bin", False)
|
||
|
|
self.IsCreateHex = data['output'].get("hex", False)
|
||
|
|
self.IsCreateMap = data['output'].get("map", False)
|
||
|
|
# 头文件路径
|
||
|
|
if data.get("include", []):
|
||
|
|
self.Includes = " ".join(map(lambda x: "-I " + x, data['include']))
|
||
|
|
# 源文件
|
||
|
|
if data.get("src", []):
|
||
|
|
for item in data['src']:
|
||
|
|
self.SrcFiles.extend(glob.glob(item))
|
||
|
|
def GetTargetFile(self, Ext):
|
||
|
|
return self.OutPath + self.NAME + Ext
|
||
|
|
|
||
|
|
def GetOutFile(self, infile):
|
||
|
|
return self.OutPath + os.path.basename(infile).split(".")[0] + ".o"
|
||
|
|
|
||
|
|
def CheckUpdate(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def GetCompileCMD(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def GetArCMD(self):
|
||
|
|
return "@%s -r %s %s" % (project.AR, self.OutPath + "lib" + self.NAME + ".a", " ".join(
|
||
|
|
glob.glob(self.OutPath + "*.o")))
|
||
|
|
|
||
|
|
def GetLinkCMD(self):
|
||
|
|
return "@%s %s -o %s %s %s" % (project.LD, " ".join(
|
||
|
|
glob.glob(self.OutPath + "*.o")), project.GetTargetFile(".elf"), self.LDFlag, "-Wl,-Map=" + self.GetTargetFile(".map") if self.IsCreateMap else "")
|
||
|
|
|
||
|
|
def GetCreateBinCMD(self):
|
||
|
|
return "@%s %s %s -Obinary" % (project.OBJCOPY, project.GetTargetFile(".elf"), project.GetTargetFile(".bin"))
|
||
|
|
|
||
|
|
def GetCreateHexCMD(self):
|
||
|
|
return "@%s %s %s -Oihex" % (project.OBJCOPY, project.GetTargetFile(".elf"), project.GetTargetFile(".hex"))
|
||
|
|
|
||
|
|
def GetCompileTasks(self):
|
||
|
|
_taskList = []
|
||
|
|
for infile in self.SrcFiles:
|
||
|
|
outfile = self.GetOutFile(infile)
|
||
|
|
if os.path.exists(outfile): # 跳过没有变化文件
|
||
|
|
if os.path.getmtime(outfile) > os.path.getmtime(infile):
|
||
|
|
continue
|
||
|
|
if os.path.splitext(infile)[1] == ".c":
|
||
|
|
_taskList.append("@%s -c %s -o %s %s %s" %
|
||
|
|
(self.CC, infile, outfile, self.Includes, self.CCFlag))
|
||
|
|
if os.path.splitext(infile)[1] == ".s":
|
||
|
|
_taskList.append("@%s -c %s -o %s %s %s" %
|
||
|
|
(self.AS, infile, outfile, self.Includes, self.ASFlag))
|
||
|
|
return _taskList
|
||
|
|
|
||
|
|
def excute_shell(cmd):
|
||
|
|
"""执行shell命令"""
|
||
|
|
os.system(cmd)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
# 解析命令行参数
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument(
|
||
|
|
"-f", "--file", help="project file")
|
||
|
|
parser.add_argument(
|
||
|
|
"-c", "--clean", help="clean output file", action="store_true")
|
||
|
|
parser.add_argument(
|
||
|
|
"-d", "--download", help="download bin file", action="store_true")
|
||
|
|
args = parser.parse_args()
|
||
|
|
# 解析json文件
|
||
|
|
if args.file:
|
||
|
|
project = ArmccParser(args.file)
|
||
|
|
else:
|
||
|
|
project = ArmccParser()
|
||
|
|
# clean指令
|
||
|
|
if args.clean:
|
||
|
|
for delfile in glob.glob(project.OutPath + "*.*"):
|
||
|
|
os.remove(delfile)
|
||
|
|
exit()
|
||
|
|
# flash指令
|
||
|
|
if args.download:
|
||
|
|
os.system("%s/mqjlink -o 0x08000000 -d STM32F405RG -f %s" %
|
||
|
|
(sys.path[0], project.GetTargetFile(".bin")))
|
||
|
|
exit()
|
||
|
|
# make指令
|
||
|
|
# 开始编译
|
||
|
|
if not os.path.exists(project.OutPath): # 检查输出目录
|
||
|
|
os.makedirs(project.OutPath)
|
||
|
|
start = time.clock()
|
||
|
|
print("正在编译。。。")
|
||
|
|
# 多线程并发
|
||
|
|
pool = Pool()
|
||
|
|
pool.map(excute_shell, project.GetCompileTasks())
|
||
|
|
pool.close()
|
||
|
|
pool.join()
|
||
|
|
# 多线程并发
|
||
|
|
# exit()
|
||
|
|
# 链接
|
||
|
|
if project.Type == "lib": # 生成库
|
||
|
|
os.system(project.GetArCMD())
|
||
|
|
else: # 连接
|
||
|
|
os.system(project.GetLinkCMD())
|
||
|
|
if project.IsCreateBin:
|
||
|
|
os.system(project.GetCreateBinCMD())
|
||
|
|
if project.IsCreateHex:
|
||
|
|
os.system(project.GetCreateHexCMD())
|
||
|
|
# 完成
|
||
|
|
end = time.clock()
|
||
|
|
print("编译完成，耗时%s" % (end - start))
|
||
|
|
exit()
|