#
# File : codeblocks.py
# This file is part of RT-Thread RTOS
# COPYRIGHT (C) 2006 - 2015, RT-Thread Development Team
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# Change Logs:
# Date Author Notes
# 2015年01月20日 Bernard Add copyright information
#

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

import utils

fs_encoding = sys.getfilesystemencoding()

def CB_AddHeadFiles(program, elem, project_path):
 utils.source_ext = []
 utils.source_ext = ["h"]
 for item in program:
 utils.walk_children(item) 
 utils.source_list.sort()
 # print utils.source_list
 
 for f in utils.source_list:
 path = _make_path_relative(project_path, f)
 Unit = SubElement(elem, 'Unit')
 Unit.set('filename', path.decode(fs_encoding))

def CB_AddCFiles(ProjectFiles, parent, gname, files, project_path):
 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)

 Unit = SubElement(parent, 'Unit')
 Unit.set('filename', path.decode(fs_encoding))
 Option = SubElement(Unit, 'Option')
 Option.set('compilerVar', "CC")

def CBProject(target, script, program):
 project_path = os.path.dirname(os.path.abspath(target))

 if os.path.isfile('template.cbp'):
 tree = etree.parse('template.cbp')
 else:
 tree = etree.parse(os.path.join(os.path.dirname(__file__), 'template.cbp'))
 
 root = tree.getroot()
 
 out = open(target, 'w')
 out.write('<?xml version="1.0" encoding="UTF-8" standalone="yes" ?>\n')
 
 ProjectFiles = []
 
 # SECTION 1. add "*.c|*.h" files group
 for elem in tree.iter(tag='Project'):
 # print elem.tag, elem.attrib
 break
 # add c files
 for group in script:
 group_xml = CB_AddCFiles(ProjectFiles, elem, group['name'], group['src'], project_path)
 # add h files
 CB_AddHeadFiles(program, elem, project_path)

 # SECTION 2. 
 # write head include path
 if 'CPPPATH' in building.Env:
 cpp_path = building.Env['CPPPATH']
 paths = set()
 for path in cpp_path:
 inc = _make_path_relative(project_path, os.path.normpath(path))
 paths.add(inc) #.replace('\\', '/')
 
 paths = [i for i in paths]
 paths.sort()
 # write include path, definitions
 for elem in tree.iter(tag='Compiler'):
 break
 for path in paths:
 Add = SubElement(elem, 'Add')
 Add.set('directory', path)

 for macro in building.Env.get('CPPDEFINES', []):
 Add = SubElement(elem, 'Add')
 for d in macro:
 Add.set('option', "-D"+d)
 
 # write link flags
 '''
 # write lib dependence 
 if 'LIBS' in building.Env:
 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)

 # write lib include path
 if 'LIBPATH' in building.Env:
 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()