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('\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('\n') out.write('') xml_indent(root) out.write(etree.tostring(root, encoding='utf-8')) out.close() print('done!') return