############################################
# Copyright (c) 2012 Microsoft Corporation
# 
# Auxiliary scripts for generating Makefiles 
# and Visual Studio project files.
#
# Author: Leonardo de Moura (leonardo)
############################################
import os
import glob

BUILD_DIR='build'
SRC_DIR='src'
MODES=[]
PLATFORMS=[]

class MKException(Exception):
    def __init__(self, value):
        self.value = value
    def __str__(self):
        return repr(self.value)

def set_build_dir(d):
    global BUILD_DIR
    BUILD_DIR = d
    mk_dir(BUILD_DIR)

def set_src_dir(d):
    global SRC_DIR
    SRC_DIR = d

def set_modes(l):
    global MODES
    MODES=l

def set_platforms(l):
    global PLATFORMS
    PLATFORMS=l

VS_COMMON_OPTIONS='WIN32'
VS_DBG_OPTIONS='_DEBUG'
VS_RELEASE_OPTIONS='NDEBUG'

GUI = 0
Name2GUI = {}

def mk_gui_str(id):
    return '4D2F40D8-E5F9-473B-B548-%012d' % id

MODULES = []
HEADERS = []
LIBS = []
EXES = []
DEPS = {}

def set_vs_options(common, dbg, release):
    global VS_COMMON_OPTIONS, VS_DBG_OPTIONS, VS_RELEASE_OPTIONS
    VS_COMMON_OPTIONS = common
    VS_DBG_OPTIONS = dbg
    VS_RELEASE_OPTIONS = release

def is_debug(mode):
    return mode == 'Debug'

def is_x64(platform):
    return platform == 'x64'

def mk_dir(d):
    if not os.path.exists(d):
        os.makedirs(d)

def module_src_dir(name):
    return '%s%s%s' % (SRC_DIR, os.sep, name)

def module_build_dir(name):
    return '%s%s%s' % (BUILD_DIR, os.sep, name)

LIB_KIND = 0
EXE_KIND = 1

def get_extension(kind):
    if kind == LIB_KIND:
        return 'lib'
    elif kind == EXE_KIND:
        return 'exe'
    else:
        raise MKException('unknown kind %s' % kind)

def vs_header(f):
    f.write(
'<?xml version="1.0" encoding="utf-8"?>\n'
'<Project DefaultTargets="Build" ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">\n')

def vs_project_configurations(f, name):
    global GUI, Name2GUI
    f.write('  <ItemGroup Label="ProjectConfigurations">\n')
    for mode in MODES:
        for platform in PLATFORMS:
            f.write('    <ProjectConfiguration Include="%s|%s">\n' % (mode, platform))
            f.write('      <Configuration>%s</Configuration>\n' % mode)
            f.write('      <Platform>%s</Platform>\n' % platform)
            f.write('    </ProjectConfiguration>\n')
    f.write('  </ItemGroup>\n')

    f.write('   <PropertyGroup Label="Globals">\n')
    f.write('    <ProjectGuid>{%s}</ProjectGuid>\n' % mk_gui_str(GUI))
    f.write('    <ProjectName>%s</ProjectName>\n' % name)
    f.write('    <Keyword>Win32Proj</Keyword>\n')
    f.write('  </PropertyGroup>\n')
    f.write('  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />\n')
    Name2GUI[name] = GUI
    GUI = GUI + 1

def vs_configurations(f, name, kind):
    for mode in MODES:
        for platform in PLATFORMS:
            f.write('  <PropertyGroup Condition="\'$(Configuration)|$(Platform)\'==\'%s|%s\'" Label="Configuration">\n' % (mode, platform))
            if kind == LIB_KIND:
                f.write('    <ConfigurationType>StaticLibrary</ConfigurationType>\n')
            elif kind == EXE_KIND:
                f.write('    <ConfigurationType>Application</ConfigurationType>\n')
            else:
                raise MKException("unknown kind %s" % kind)
            f.write('    <CharacterSet>Unicode</CharacterSet>\n')
            f.write('    <UseOfMfc>false</UseOfMfc>\n')
            f.write('  </PropertyGroup>\n')

    f.write('  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />\n')
    f.write('  <ImportGroup Label="ExtensionSettings">\n')
    f.write('   </ImportGroup>\n')
    f.write('   <ImportGroup Label="PropertySheets">\n')
    f.write('    <Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists(\'$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props\')" Label="LocalAppDataPlatform" />  </ImportGroup>\n')
    f.write('  <PropertyGroup Label="UserMacros" />\n')

    f.write('  <PropertyGroup>\n')
    for mode in MODES:
        for platform in PLATFORMS:
            if is_x64(platform):
                f.write('    <OutDir Condition="\'$(Configuration)|$(Platform)\'==\'%s|%s\'">$(SolutionDir)$(Platform)\$(Configuration)\</OutDir>\n' % 
                        (mode, platform))
            else:
                f.write('    <OutDir Condition="\'$(Configuration)|$(Platform)\'==\'%s|%s\'">$(SolutionDir)$(Configuration)\</OutDir>\n' % (mode, platform))
    for mode in MODES:
        for platform in PLATFORMS:
            f.write('    <TargetName Condition="\'$(Configuration)|$(Platform)\'==\'%s|%s\'">%s</TargetName>\n' % (mode, platform, name))
            f.write('    <TargetExt Condition="\'$(Configuration)|$(Platform)\'==\'%s|%s\'">.%s</TargetExt>\n' % (mode, platform, get_extension(kind)))
    f.write('  </PropertyGroup>\n')

def vs_compilation_options(f, name, deps, kind):
    for mode in MODES:
        for platform in PLATFORMS:
            f.write('  <ItemDefinitionGroup Condition="\'$(Configuration)|$(Platform)\'==\'%s|%s\'">\n' % (mode, platform))
            if is_x64(platform):
                f.write('    <Midl>\n')
                f.write('      <TargetEnvironment>X64</TargetEnvironment>\n')
                f.write('    </Midl>\n')
            f.write('    <ClCompile>\n')
            if is_debug(mode):
                f.write('      <Optimization>Disabled</Optimization>\n')
            else:
                f.write('      <Optimization>Full</Optimization>\n')
            options = VS_COMMON_OPTIONS
            if is_debug(mode):
                options = "%s;%s" % (options, VS_DBG_OPTIONS)
            else:
                options = "%s;%s" % (options, VS_RELEASE_OPTIONS)
            if is_x64(platform):
                options = "%s;_AMD64_" % options
            f.write('      <PreprocessorDefinitions>%s;%%(PreprocessorDefinitions)</PreprocessorDefinitions>\n' % options)
            if is_debug(mode):
                f.write('      <MinimalRebuild>true</MinimalRebuild>\n')
                f.write('      <BasicRuntimeChecks>EnableFastChecks</BasicRuntimeChecks>\n')
                f.write('      <WarningLevel>Level3</WarningLevel>\n')
            f.write('      <RuntimeLibrary>MultiThreadedDebugDLL</RuntimeLibrary>\n')
            f.write('      <OpenMPSupport>true</OpenMPSupport>\n')
            f.write('      <DebugInformationFormat>ProgramDatabase</DebugInformationFormat>\n')
            f.write('      <AdditionalIncludeDirectories>')
            f.write('..\..\src\%s' % name)
            for dep in deps:
                f.write(';..\..\src\%s' % dep)
            f.write('</AdditionalIncludeDirectories>\n')
            f.write('    </ClCompile>\n')
            f.write('    <Link>\n')
            f.write('      <OutputFile>$(OutDir)%s.%s</OutputFile>\n' % (name, get_extension(kind)))
            f.write('      <AdditionalLibraryDirectories>%(AdditionalLibraryDirectories)</AdditionalLibraryDirectories>\n')
            if is_x64(platform):
                f.write('      <TargetMachine>MachineX64</TargetMachine>\n')
            else:
                f.write('      <TargetMachine>MachineX86</TargetMachine>\n')
            if kind == EXE_KIND:
                f.write('<AdditionalDependencies>kernel32.lib;user32.lib;gdi32.lib;winspool.lib;comdlg32.lib;advapi32.lib;shell32.lib;ole32.lib;oleaut32.lib;uuid.lib;odbc32.lib;odbccp32.lib')
                for dep in deps:
                    f.write(';$(OutDir)%s.lib' % dep)
                    # if is_x64(platform):
                    #    f.write(';..\%s\%s\%s\%s.lib' % (dep, platform, mode, dep))
                    # else:
                    #    f.write(';..\%s\%s\%s.lib' % (dep, mode, dep))
                f.write(';%(AdditionalDependencies)</AdditionalDependencies>\n')
            f.write('    </Link>\n')
            f.write('  </ItemDefinitionGroup>\n')

def add_vs_cpps(f, name):
    f.write('  <ItemGroup>\n')
    srcs = module_src_dir(name)
    for cppfile in glob.glob(os.path.join(srcs, '*.cpp')):
       f.write('    <ClCompile Include="..%s..%s%s" />\n' % (os.sep, os.sep, cppfile))
    f.write('  </ItemGroup>\n')

def vs_footer(f):
    f.write(
'  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />\n'
'  <ImportGroup Label="ExtensionTargets">\n'
'  </ImportGroup>\n'
'</Project>\n')

def check_new_component(name):
    if (name in HEADERS) or (name in LIBS) or (name in EXES):
        raise MKException("Component '%s' was already defined" % name)

# Add a directory containing only .h files
def add_header(name):
    check_new_component(name)
    HEADERS.append(name)

def find_all_deps(name, deps):
    new_deps = []
    for dep in deps:
        if dep in LIBS:
            if not (dep in new_deps):
                new_deps.append(dep)
            for dep_dep in DEPS[dep]:
                if not (dep_dep in new_deps):
                    new_deps.append(dep_dep)
        elif dep in HEADERS:
            if not (dep in new_deps):
                new_deps.append(dep)
        else:
            raise MKException("Unknown component '%s' at '%s'." % (dep, name))
    return new_deps

def add_component(name, deps, kind):
    check_new_component(name)
    if kind == LIB_KIND:
        LIBS.append(name)
    elif kind == EXE_KIND:
        EXES.append(name)
    else:
        raise MKException("unknown kind %s" % kind)
    MODULES.append(name)
    deps = find_all_deps(name, deps)
    DEPS[name] = deps
    print "Dependencies for '%s': %s" % (name, deps)

    module_dir = module_build_dir(name)
    mk_dir(module_dir)

    vs_proj = open('%s%s%s.vcxproj' % (module_dir, os.sep, name), 'w')
    vs_header(vs_proj)
    vs_project_configurations(vs_proj, name)
    vs_configurations(vs_proj, name, kind)
    vs_compilation_options(vs_proj, name, deps, kind)
    add_vs_cpps(vs_proj, name)
    vs_footer(vs_proj)

def add_lib(name, deps):
    add_component(name, deps, LIB_KIND)

def add_exe(name, deps):
    add_component(name, deps, EXE_KIND)

def is_lib(name):
    # Add DLL dependency
    return name in LIBS

def mk_vs_solution():
    sln = open('%s%sz3.sln' % (BUILD_DIR, os.sep), 'w')
    sln.write('\n')
    sln.write("Microsoft Visual Studio Solution File, Format Version 11.00\n")
    sln.write("# Visual Studio 2010\n")
    for module in MODULES:
        gui = Name2GUI[module]
        deps = DEPS[module]
        sln.write('Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "%s", "%s%s%s.vcxproj", "{%s}"\n' %
                  (module, module, os.sep, module, mk_gui_str(gui)))
        if len(deps) > 0:
            sln.write('    ProjectSection(ProjectDependencies) = postProject\n')
            for dep in deps:
                if is_lib(dep):
                    i = Name2GUI[dep]
                    sln.write('        {%s} = {%s}\n' % (mk_gui_str(i), mk_gui_str(i)))
            sln.write('    EndProjectSection\n')
        sln.write('EndProject\n')
    sln.write('Global\n')
    sln.write('GlobalSection(SolutionConfigurationPlatforms) = preSolution\n')
    for mode in MODES:
        for platform in PLATFORMS:
            sln.write('   %s|%s = %s|%s\n' % (mode, platform, mode, platform))
    sln.write('EndGlobalSection\n')
    sln.write('GlobalSection(ProjectConfigurationPlatforms) = postSolution\n')
    for module in MODULES:
        gui = Name2GUI[module]
        for mode in MODES:
            for platform in PLATFORMS:
                sln.write('    {%s}.%s|%s.ActiveCfg = %s|%s\n' % (mk_gui_str(gui), mode, platform, mode, platform))
                sln.write('    {%s}.%s|%s.Build.0   = %s|%s\n' % (mk_gui_str(gui), mode, platform, mode, platform))
    sln.write('EndGlobalSection\n')
                
    print "Visual Solution was generated."