300 lines
9.7 KiB
Python
300 lines
9.7 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
import glob
|
||
|
|
|
||
|
|
from utils import *
|
||
|
|
from utils import _make_path_relative
|
||
|
|
from utils import xml_indent
|
||
|
|
|
||
|
|
import xml.etree.ElementTree as etree
|
||
|
|
from xml.etree.ElementTree import SubElement
|
||
|
|
|
||
|
|
source_pattern = ['*.c', '*.cpp', '*.cxx', '*.s', '*.S', '*.asm']
|
||
|
|
|
||
|
|
def OSPath(path):
|
||
|
|
import platform
|
||
|
|
|
||
|
|
if type(path) == type('str'):
|
||
|
|
if platform.system() == 'Windows':
|
||
|
|
return path.replace('/', '\\')
|
||
|
|
else:
|
||
|
|
return path.replace('\\', '/')
|
||
|
|
else:
|
||
|
|
if platform.system() == 'Windows':
|
||
|
|
return [item.replace('/', '\\') for item in path]
|
||
|
|
else:
|
||
|
|
return [item.replace('\\', '/') for item in path]
|
||
|
|
|
||
|
|
def CollectPaths(paths):
|
||
|
|
all_paths = []
|
||
|
|
|
||
|
|
def ParentPaths(path):
|
||
|
|
ret = os.path.dirname(path)
|
||
|
|
if ret == path or ret == '':
|
||
|
|
return []
|
||
|
|
|
||
|
|
return [ret] + ParentPaths(ret)
|
||
|
|
|
||
|
|
for path in paths:
|
||
|
|
# path = os.path.abspath(path)
|
||
|
|
path = path.replace('\\', '/')
|
||
|
|
all_paths = all_paths + [path] + ParentPaths(path)
|
||
|
|
|
||
|
|
all_paths = list(set(all_paths))
|
||
|
|
return sorted(all_paths)
|
||
|
|
|
||
|
|
'''
|
||
|
|
Collect all of files under paths
|
||
|
|
'''
|
||
|
|
def CollectFiles(paths, pattern):
|
||
|
|
files = []
|
||
|
|
for path in paths:
|
||
|
|
if type(pattern) == type(''):
|
||
|
|
files = files + glob.glob(path + '/' + pattern)
|
||
|
|
else:
|
||
|
|
for item in pattern:
|
||
|
|
# print('--> %s' % (path + '/' + item))
|
||
|
|
files = files + glob.glob(path + '/' + item)
|
||
|
|
|
||
|
|
return sorted(files)
|
||
|
|
|
||
|
|
def CollectAllFilesinPath(path, pattern):
|
||
|
|
files = []
|
||
|
|
|
||
|
|
for item in pattern:
|
||
|
|
files += glob.glob(path + '/' + item)
|
||
|
|
|
||
|
|
list = os.listdir(path)
|
||
|
|
if len(list):
|
||
|
|
for item in list:
|
||
|
|
if item.startswith('.'):
|
||
|
|
continue
|
||
|
|
if item == 'bsp':
|
||
|
|
continue
|
||
|
|
|
||
|
|
if os.path.isdir(os.path.join(path, item)):
|
||
|
|
files = files + CollectAllFilesinPath(os.path.join(path, item), pattern)
|
||
|
|
return files
|
||
|
|
|
||
|
|
'''
|
||
|
|
Exclude files from infiles
|
||
|
|
'''
|
||
|
|
def ExcludeFiles(infiles, files):
|
||
|
|
in_files = set([OSPath(file) for file in infiles])
|
||
|
|
exl_files = set([OSPath(file) for file in files])
|
||
|
|
|
||
|
|
exl_files = in_files - exl_files
|
||
|
|
|
||
|
|
return exl_files
|
||
|
|
|
||
|
|
def ExcludePaths(filepath, paths):
|
||
|
|
ret = []
|
||
|
|
|
||
|
|
files = os.listdir(filepath)
|
||
|
|
for file in files:
|
||
|
|
if file.startswith('.'):
|
||
|
|
continue
|
||
|
|
|
||
|
|
fullname = os.path.join(filepath, file)
|
||
|
|
|
||
|
|
if os.path.isdir(fullname):
|
||
|
|
# print(fullname)
|
||
|
|
if not fullname in paths:
|
||
|
|
ret = ret + [fullname]
|
||
|
|
else:
|
||
|
|
ret = ret + ExcludePaths(fullname, paths)
|
||
|
|
|
||
|
|
return ret
|
||
|
|
|
||
|
|
def HandleToolOption(tools, env):
|
||
|
|
project = ProjectInfo(env)
|
||
|
|
BSP_ROOT = os.path.abspath(env['BSP_ROOT'])
|
||
|
|
|
||
|
|
CPPDEFINES = project['CPPDEFINES']
|
||
|
|
paths = ['${ProjDirPath}/' + _make_path_relative(BSP_ROOT, os.path.normpath(i)).replace('\\', '/') for i in project['CPPPATH']]
|
||
|
|
|
||
|
|
for tool in tools:
|
||
|
|
if tool.get('id').find('c.compile') != 1:
|
||
|
|
options = tool.findall('option')
|
||
|
|
for option in options:
|
||
|
|
if option.get('id').find('c.compiler.include.paths') != -1:
|
||
|
|
# find all of paths in this project
|
||
|
|
include_paths = option.findall('listOptionValue')
|
||
|
|
project_paths = []
|
||
|
|
for item in include_paths:
|
||
|
|
project_paths += [item.get('value')]
|
||
|
|
|
||
|
|
if len(project_paths) > 0:
|
||
|
|
cproject_paths = set(paths) - set(project_paths)
|
||
|
|
else:
|
||
|
|
cproject_paths = paths
|
||
|
|
|
||
|
|
# print('c.compiler.include.paths')
|
||
|
|
cproject_paths = sorted(cproject_paths)
|
||
|
|
for item in cproject_paths:
|
||
|
|
SubElement(option, 'listOptionValue', {'builtIn': 'false', 'value': item})
|
||
|
|
|
||
|
|
if option.get('id').find('c.compiler.defs') != -1:
|
||
|
|
defs = option.findall('listOptionValue')
|
||
|
|
project_defs = []
|
||
|
|
for item in defs:
|
||
|
|
project_defs += [item.get('value')]
|
||
|
|
if len(project_defs) > 0:
|
||
|
|
cproject_defs = set(CPPDEFINES) - set(project_defs)
|
||
|
|
else:
|
||
|
|
cproject_defs = CPPDEFINES
|
||
|
|
|
||
|
|
# print('c.compiler.defs')
|
||
|
|
cproject_defs = sorted(cproject_defs)
|
||
|
|
for item in cproject_defs:
|
||
|
|
SubElement(option, 'listOptionValue', {'builtIn': 'false', 'value': item})
|
||
|
|
|
||
|
|
if tool.get('id').find('c.linker') != -1:
|
||
|
|
options = tool.findall('option')
|
||
|
|
for option in options:
|
||
|
|
if option.get('id').find('c.linker.scriptfile') != -1:
|
||
|
|
linker_script = 'link.lds'
|
||
|
|
items = env['LINKFLAGS'].split(' ')
|
||
|
|
if '-T' in items:
|
||
|
|
linker_script = items[items.index('-T') + 1]
|
||
|
|
linker_script = '${ProjDirPath}/' + linker_script
|
||
|
|
|
||
|
|
# print('c.linker.scriptfile')
|
||
|
|
listOptionValue = option.find('listOptionValue')
|
||
|
|
if listOptionValue != None:
|
||
|
|
listOptionValue.set('value', linker_script)
|
||
|
|
else:
|
||
|
|
SubElement(option, 'listOptionValue', {'builtIn': 'false', 'value': linker_script})
|
||
|
|
|
||
|
|
if option.get('id').find('c.linker.nostart') != -1:
|
||
|
|
if env['LINKFLAGS'].find('-nostartfiles') != -1:
|
||
|
|
option.set('value', 'true')
|
||
|
|
else:
|
||
|
|
option.set('value', 'false')
|
||
|
|
|
||
|
|
return
|
||
|
|
|
||
|
|
def HandleRTTRoot(env):
|
||
|
|
bsp_root = env['BSP_ROOT']
|
||
|
|
rtt_root = env['RTT_ROOT']
|
||
|
|
|
||
|
|
if not rtt_root.startswith(bsp_root):
|
||
|
|
to_SubElement = True
|
||
|
|
# print('handle virtual root')
|
||
|
|
|
||
|
|
# always use '/' path separator
|
||
|
|
rtt_root = rtt_root.replace('\\', '/')
|
||
|
|
|
||
|
|
project = etree.parse('.project')
|
||
|
|
root = project.getroot()
|
||
|
|
|
||
|
|
linkedResources = root.find('linkedResources')
|
||
|
|
if linkedResources == None:
|
||
|
|
# add linkedResources
|
||
|
|
linkedResources = SubElement(root, 'linkedResources')
|
||
|
|
# print('add linkedResources')
|
||
|
|
else:
|
||
|
|
links = linkedResources.findall('link')
|
||
|
|
# search exist 'rt-thread' virtual folder
|
||
|
|
for link in links:
|
||
|
|
if link.find('name').text == 'rt-thread':
|
||
|
|
# handle location
|
||
|
|
to_SubElement = False
|
||
|
|
location = link.find('location')
|
||
|
|
location.text = rtt_root
|
||
|
|
|
||
|
|
if to_SubElement:
|
||
|
|
# print('to subelement for virtual folder')
|
||
|
|
link = SubElement(linkedResources, 'link')
|
||
|
|
name = SubElement(link, 'name')
|
||
|
|
name.text = 'rt-thread'
|
||
|
|
type = SubElement(link, 'type')
|
||
|
|
type.text = '2'
|
||
|
|
location = SubElement(link, 'location')
|
||
|
|
location.text = rtt_root
|
||
|
|
|
||
|
|
out = open('.project', 'w')
|
||
|
|
out.write('<?xml version="1.0" encoding="UTF-8"?>\n')
|
||
|
|
xml_indent(root)
|
||
|
|
out.write(etree.tostring(root, encoding='utf-8'))
|
||
|
|
out.close()
|
||
|
|
|
||
|
|
return
|
||
|
|
|
||
|
|
def TargetEclipse(env):
|
||
|
|
global source_pattern
|
||
|
|
|
||
|
|
print('Update eclipse setting...')
|
||
|
|
|
||
|
|
if not os.path.exists('.cproject'):
|
||
|
|
print('no eclipse CDT project found!')
|
||
|
|
return
|
||
|
|
|
||
|
|
HandleRTTRoot(env)
|
||
|
|
|
||
|
|
project = ProjectInfo(env)
|
||
|
|
|
||
|
|
all_paths = [OSPath(path) for path in CollectPaths(project['DIRS'])]
|
||
|
|
# print(all_paths)
|
||
|
|
bsp_root = os.path.abspath(env['BSP_ROOT'])
|
||
|
|
|
||
|
|
exclude_paths = ExcludePaths(bsp_root, all_paths)
|
||
|
|
paths = exclude_paths
|
||
|
|
exclude_paths = []
|
||
|
|
for path in paths:
|
||
|
|
# add bsp and libcpu folder and not collect source files (too more files)
|
||
|
|
if path.endswith('rt-thread\\bsp') or path.endswith('rt-thread\\libcpu'):
|
||
|
|
exclude_paths += [path]
|
||
|
|
continue
|
||
|
|
|
||
|
|
set = CollectAllFilesinPath(path, source_pattern)
|
||
|
|
if len(set):
|
||
|
|
exclude_paths += [path]
|
||
|
|
|
||
|
|
exclude_paths = [_make_path_relative(bsp_root, path).replace('\\', '/') for path in exclude_paths]
|
||
|
|
env['ExPaths'] = exclude_paths
|
||
|
|
|
||
|
|
all_files = CollectFiles(all_paths, source_pattern)
|
||
|
|
src_files = project['FILES']
|
||
|
|
|
||
|
|
exclude_files = ExcludeFiles(all_files, src_files)
|
||
|
|
exclude_files = [_make_path_relative(bsp_root, file).replace('\\', '/') for file in exclude_files]
|
||
|
|
env['ExFiles'] = exclude_files
|
||
|
|
|
||
|
|
cproject = etree.parse('.cproject')
|
||
|
|
|
||
|
|
root = cproject.getroot()
|
||
|
|
cconfigurations = root.findall('storageModule/cconfiguration')
|
||
|
|
for cconfiguration in cconfigurations:
|
||
|
|
tools = cconfiguration.findall('storageModule/configuration/folderInfo/toolChain/tool')
|
||
|
|
HandleToolOption(tools, env)
|
||
|
|
|
||
|
|
sourceEntries = cconfiguration.find('storageModule/configuration/sourceEntries')
|
||
|
|
entry = sourceEntries.find('entry')
|
||
|
|
if entry != None:
|
||
|
|
sourceEntries.remove(entry)
|
||
|
|
|
||
|
|
excluding = exclude_paths + exclude_files
|
||
|
|
excluding = sorted(excluding)
|
||
|
|
value = ''
|
||
|
|
for item in excluding:
|
||
|
|
if value == '':
|
||
|
|
value = item
|
||
|
|
else:
|
||
|
|
value += '|' + item
|
||
|
|
excluding = value
|
||
|
|
|
||
|
|
SubElement(sourceEntries, 'entry', {'excluding': excluding, 'flags': 'VALUE_WORKSPACE_PATH|RESOLVED', 'kind':'sourcePath', 'name':""})
|
||
|
|
|
||
|
|
# write back to .cproject
|
||
|
|
out = open('.cproject', 'w')
|
||
|
|
out.write('<?xml version="1.0" encoding="UTF-8" standalone="no"?>\n')
|
||
|
|
out.write('<?fileVersion 4.0.0?>')
|
||
|
|
xml_indent(root)
|
||
|
|
out.write(etree.tostring(root, encoding='utf-8'))
|
||
|
|
out.close()
|
||
|
|
|
||
|
|
print('done!')
|
||
|
|
|
||
|
|
return
|