From 0d77b327a998c35e53025690415a3d355717eb95 Mon Sep 17 00:00:00 2001
From: "bernard.xiong" <bernard.xiong@gmail.com>
Date: Wed, 7 Nov 2012 23:02:25 +0000
Subject: [PATCH] Add Visual Studio project generation script.

git-svn-id: https://rt-thread.googlecode.com/svn/trunk@2398 bbd45198-f89e-11dd-88c7-29a3b14d5316
---
 tools/building.py |   7 ++-
 tools/vs.py       | 116 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 122 insertions(+), 1 deletion(-)
 create mode 100644 tools/vs.py

diff --git a/tools/building.py b/tools/building.py
index 0307bf74fd..032e9dd8b9 100644
--- a/tools/building.py
+++ b/tools/building.py
@@ -92,7 +92,8 @@ def PrepareBuilding(env, root_directory, has_libcpu=False, remove_components = [
     #{target_name:(CROSS_TOOL, PLATFORM)}
     tgt_dict = {'mdk':('keil', 'armcc'),
                 'mdk4':('keil', 'armcc'),
-                'iar':('iar', 'iar')}
+                'iar':('iar', 'iar'),
+                'vs':('msvc', 'cl')}
     tgt_name = GetOption('target')
     if tgt_name:
         SetOption('no_exec', 1)
@@ -285,6 +286,7 @@ def EndBuilding(target, program = None):
     from keil import MDKProject
     from keil import MDK4Project
     from iar import IARProject
+    from vs import VSProject
 
     Env.AddPostAction(target, rtconfig.POST_ACTION)
 
@@ -305,6 +307,9 @@ def EndBuilding(target, program = None):
     if GetOption('target') == 'iar':
         IARProject('project.ewp', Projects) 
 
+    if GetOption('target') == 'vs':
+        VSProject('project.vcproj', Projects)
+
     if GetOption('copy') and program != None:
         MakeCopy(program)
     if GetOption('copy-header') and program != None:
diff --git a/tools/vs.py b/tools/vs.py
new file mode 100644
index 0000000000..ac01348c0f
--- /dev/null
+++ b/tools/vs.py
@@ -0,0 +1,116 @@
+import os
+import sys
+import string
+import building
+
+import xml.etree.ElementTree as etree
+from xml.etree.ElementTree import SubElement
+from utils import _make_path_relative
+from utils import xml_indent
+fs_encoding = sys.getfilesystemencoding()
+
+def VS_AddGroup(ProjectFiles, parent, name, files, project_path):
+    Filter = SubElement(parent, 'Filter')
+    Filter.set('Name', name) #set group name to group
+
+    for f in files:
+        fn = f.rfile()
+        name = fn.name
+        path = os.path.dirname(fn.abspath)
+
+        path = _make_path_relative(project_path, path)
+        path = os.path.join(path, name)
+
+        File = SubElement(Filter, 'File')
+        File.set('RelativePath', path.decode(fs_encoding))
+
+def VSProject(target, script):
+    project_path = os.path.dirname(os.path.abspath(target))
+    
+    tree = etree.parse('template.vcproj')
+    root = tree.getroot()
+    
+    out = file(target, 'wb')
+    out.write('<?xml version="1.0" encoding="UTF-8"?>\r\n')
+    
+    CPPPATH = []
+    CPPDEFINES = []
+    LINKFLAGS = ''
+    CCFLAGS = ''
+    ProjectFiles = []
+    
+    # add group
+    for elem in tree.iter(tag='Filter'):
+        if elem.attrib['Name'] == 'Source Files':
+            #print elem.tag, elem.attrib
+            break
+
+    for group in script:
+        group_xml = VS_AddGroup(ProjectFiles, elem, group['name'], group['src'], project_path)
+        
+        # get each include path
+        if group.has_key('CPPPATH') and group['CPPPATH']:
+            if CPPPATH:
+                CPPPATH += group['CPPPATH']
+            else:
+                CPPPATH += group['CPPPATH']
+        
+        # get each group's definitions
+        if group.has_key('CPPDEFINES') and group['CPPDEFINES']:
+            if CPPDEFINES:
+                CPPDEFINES += group['CPPDEFINES']
+            else:
+                CPPDEFINES += group['CPPDEFINES']
+        
+        # get each group's link flags
+        if group.has_key('LINKFLAGS') and group['LINKFLAGS']:
+            if LINKFLAGS:
+                LINKFLAGS += ' ' + group['LINKFLAGS']
+            else:
+                LINKFLAGS += group['LINKFLAGS']
+    
+    # remove repeat path
+    paths = set()
+    for path in CPPPATH:
+        inc = _make_path_relative(project_path, os.path.normpath(path))
+        paths.add(inc) #.replace('\\', '/')
+    
+    paths = [i for i in paths]
+    paths.sort()
+    CPPPATH = string.join(paths, ';')
+    
+    # write include path, definitions
+    for elem in tree.iter(tag='Tool'):
+        if elem.attrib['Name'] == 'VCCLCompilerTool':
+            #print elem.tag, elem.attrib
+            break
+    elem.set('AdditionalIncludeDirectories', CPPPATH)
+
+    definitions = ';'.join(building.Env['CPPDEFINES'])
+    elem.set('PreprocessorDefinitions', definitions)
+    # write link flags
+
+    # write lib dependence 
+    for elem in tree.iter(tag='Tool'):
+        if elem.attrib['Name'] == 'VCLinkerTool':
+            break
+    libs_with_extention = [i+'.lib' for i in building.Env['LIBS']]
+    libs = ' '.join(libs_with_extention)
+    elem.set('AdditionalDependencies', libs)
+
+    if building.Env.has_key('LIBPATH'):
+    #if building.Env['LIBPATH']:
+        lib_path = building.Env['LIBPATH']
+        paths  = set()
+        for path in lib_path:
+            inc = _make_path_relative(project_path, os.path.normpath(path))
+            paths.add(inc) #.replace('\\', '/')
+    
+        paths = [i for i in paths]
+        paths.sort()
+        lib_paths = ';'.join(paths)
+        elem.set('AdditionalLibraryDirectories', lib_paths)
+
+    xml_indent(root)
+    out.write(etree.tostring(root, encoding='utf-8'))
+    out.close()