[mirror_edk2.git] / Tools / Source / GenBuild / org / tianocore / build / ModuleBuildFileGenerator.java

/** @file\r
 \r
 Copyright (c) 2006, Intel Corporation\r
 All rights reserved. This program and the accompanying materials\r
 are licensed and made available under the terms and conditions of the BSD License\r
 which accompanies this distribution.  The full text of the license may be found at\r
 http://opensource.org/licenses/bsd-license.php\r
\r
 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
 **/\r
package org.tianocore.build;\r
\r
import java.io.File;\r
import java.util.LinkedHashMap;\r
import java.util.LinkedHashSet;\r
import java.util.Map;\r
import java.util.Set;\r
import java.io.FileOutputStream;\r
import java.io.OutputStreamWriter;\r
\r
import javax.xml.parsers.DocumentBuilder;\r
import javax.xml.parsers.DocumentBuilderFactory;\r
import javax.xml.transform.OutputKeys;\r
import javax.xml.transform.Result;\r
import javax.xml.transform.Source;\r
import javax.xml.transform.Transformer;\r
import javax.xml.transform.TransformerFactory;\r
import javax.xml.transform.dom.DOMSource;\r
import javax.xml.transform.stream.StreamResult;\r
\r
import org.apache.tools.ant.BuildException;\r
import org.apache.tools.ant.Project;\r
import org.tianocore.build.fpd.FpdParserTask;\r
import org.tianocore.build.global.GlobalData;\r
import org.tianocore.build.global.SurfaceAreaQuery;\r
import org.tianocore.build.id.FpdModuleIdentification;\r
import org.tianocore.build.id.ModuleIdentification;\r
import org.tianocore.build.id.PackageIdentification;\r
import org.w3c.dom.Comment;\r
import org.w3c.dom.Document;\r
import org.w3c.dom.Element;\r
import org.w3c.dom.Node;\r
\r
public class ModuleBuildFileGenerator {\r
\r
    ///\r
    /// Pass: TARGET, TOOLCHAIN, ARCH\r
    /// PACKAGE, PACKAGE_GUID, PACKAGE_VERSION\r
    ///\r
    String[] inheritProperties = {"ARCH", "MODULE_GUID", "MODULE_VERSION", "PLATFORM_FILE", "PACKAGE_GUID", "PACKAGE_VERSION"};\r
\r
    ///\r
    /// The information at the header of <em>build.xml</em>.\r
    ///\r
    private String info = "DO NOT EDIT \n"\r
                        + "File auto-generated by build utility\n"\r
                        + "\n"\r
                        + "Abstract:\n"\r
                        + "Auto-generated ANT build file for building of EFI Modules/Platforms\n";\r
\r
    private FpdModuleIdentification fpdModuleId;\r
    \r
    private Project project;\r
    \r
    private String ffsKeyword;\r
    \r
    public ModuleBuildFileGenerator(Project project, String ffsKeyword, FpdModuleIdentification fpdModuleId) {\r
        this.project = project;\r
        this.fpdModuleId = fpdModuleId;\r
        this.ffsKeyword = ffsKeyword;\r
    }\r
    \r
    /**\r
      The whole BaseName_build.xml is composed of seven part. \r
      <ul>\r
      <li> ANT properties; </li>\r
      <li> Dependent module (dependent library instances in most case); </li>\r
      <li> Source files; </li>\r
      <li> Sections if module is not library; </li>\r
      <li> Output (different for library module and driver module); </li>\r
      <li> Clean; </li>\r
      <li> Clean all. </li>\r
      </ul>\r
      \r
      @throws BuildException\r
              Error throws during BaseName_build.xml generating. \r
    **/\r
    public void genBuildFile(String buildFilename) throws BuildException {\r
        FfsProcess fp = new FfsProcess();\r
        DocumentBuilderFactory domfac = DocumentBuilderFactory.newInstance();\r
        try {\r
            DocumentBuilder dombuilder = domfac.newDocumentBuilder();\r
            Document document = dombuilder.newDocument();\r
            Comment rootComment = document.createComment(info);\r
            \r
            //\r
            // create root element and its attributes\r
            //\r
            Element root = document.createElement("project");\r
            root.setAttribute("name", fpdModuleId.getModule().getName());\r
            root.setAttribute("default", "all");\r
            root.setAttribute("basedir", ".");\r
            \r
            //\r
            // element for External ANT tasks\r
            //\r
            root.appendChild(document.createComment("Apply external ANT tasks"));\r
            Element ele = document.createElement("taskdef");\r
            ele.setAttribute("resource", "frameworktasks.tasks");\r
            root.appendChild(ele);\r
            ele = document.createElement("taskdef");\r
            ele.setAttribute("resource", "cpptasks.tasks");\r
            root.appendChild(ele);\r
            ele = document.createElement("typedef");\r
            ele.setAttribute("resource", "cpptasks.types");\r
            root.appendChild(ele);\r
            ele = document.createElement("taskdef");\r
            ele.setAttribute("resource", "net/sf/antcontrib/antlib.xml");\r
            root.appendChild(ele);\r
\r
            //\r
            // Generate the default target,\r
            // which depends on init, sections and output target\r
            //\r
            root.appendChild(document.createComment("Default target"));\r
            ele = document.createElement("target");\r
            ele.setAttribute("name", "all");\r
            ele.setAttribute("depends", "libraries, sourcefiles, sections, output");\r
            root.appendChild(ele);\r
            \r
            //\r
            // compile all source files\r
            //\r
            root.appendChild(document.createComment("Compile all dependency Library instances."));\r
            ele = document.createElement("target");\r
            ele.setAttribute("name", "libraries");\r
\r
            //\r
            // Parse all sourfiles but files specified in sections\r
            //\r
            applyLibraryInstance(document, ele);\r
            root.appendChild(ele);\r
\r
            //\r
            // compile all source files\r
            //\r
            root.appendChild(document.createComment("sourcefiles target"));\r
            ele = document.createElement("target");\r
            ele.setAttribute("name", "sourcefiles");\r
            \r
            //\r
            // Parse all sourfiles but files specified in sections\r
            //\r
            applyCompileElement(document, ele);\r
            root.appendChild(ele);\r
\r
            //\r
            // generate the init target\r
            // main purpose is create all nessary pathes\r
            // generate the sections target\r
            //\r
            root.appendChild(document.createComment("sections target"));\r
            ele = document.createElement("target");\r
            ele.setAttribute("name", "sections");\r
            applySectionsElement(document, ele, fp);\r
            root.appendChild(ele);\r
\r
            //\r
            // generate the output target\r
            //\r
            root.appendChild(document.createComment("output target"));\r
            ele = document.createElement("target");\r
            ele.setAttribute("name", "output");\r
            applyOutputElement(document, ele, fp);\r
            root.appendChild(ele);\r
\r
            \r
            //\r
            // generate the clean target\r
            //\r
            root.appendChild(document.createComment("clean target"));\r
            ele = document.createElement("target");\r
            ele.setAttribute("name", "clean");\r
            applyCleanElement(document, ele);\r
            root.appendChild(ele);\r
            \r
            //\r
            // generate the Clean All target\r
            //\r
            root.appendChild(document.createComment("Clean All target"));\r
            ele = document.createElement("target");\r
            ele.setAttribute("name", "cleanall");\r
            applyDeepcleanElement(document, ele);\r
            root.appendChild(ele);\r
            \r
            //\r
            // add the root element to the document\r
            //\r
            document.appendChild(rootComment);\r
            document.appendChild(root);\r
            //\r
            // Prepare the DOM document for writing\r
            //\r
            Source source = new DOMSource(document);\r
\r
            //\r
            // Prepare the output file\r
            //\r
            File file = new File(buildFilename);\r
\r
            //\r
            // generate all directory path\r
            //\r
            (new File(file.getParent())).mkdirs();\r
            FileOutputStream outputStream = new FileOutputStream(file);\r
            Result result = new StreamResult(new OutputStreamWriter(outputStream));\r
            \r
            //\r
            // Write the DOM document to the file\r
            //\r
            Transformer xformer = TransformerFactory.newInstance().newTransformer();\r
            xformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "2");\r
            xformer.setOutputProperty(OutputKeys.INDENT, "yes");\r
            xformer.transform(source, result);\r
        } catch (Exception ex) {\r
            throw new BuildException("Module [" + fpdModuleId.getModule().getName() + "] generating build file failed.\n" + ex.getMessage());\r
        }\r
    }\r
\r
    /**\r
      Generate the clean elements for BaseName_build.xml. \r
      \r
      @param document current BaseName_build.xml XML document\r
      @param root Root element for current\r
    **/\r
    private void applyCleanElement(Document document, Node root) {\r
        ModuleIdentification[] libinstances = SurfaceAreaQuery.getLibraryInstance(fpdModuleId.getArch());\r
        for (int i = 0; i < libinstances.length; i++) {\r
            //\r
            // Put package file path to module identification\r
            //\r
            PackageIdentification packageId = libinstances[i].getPackage();\r
            \r
            //\r
            // Generate ANT script to clean\r
            //\r
            Element ele = document.createElement("GenBuild");\r
            ele.setAttribute("type", "clean");\r
            \r
            //\r
            // Prepare pass down information\r
            //\r
            Map<String, String> passDownMap = new LinkedHashMap<String, String>();\r
            for (int j = 0; j < inheritProperties.length; j ++){\r
                passDownMap.put(inheritProperties[j], "${" + inheritProperties[j] + "}");\r
            }\r
            passDownMap.put("MODULE_GUID", libinstances[i].getGuid());\r
            passDownMap.put("MODULE_VERSION", libinstances[i].getVersion());\r
            \r
            passDownMap.put("PACKAGE_GUID", packageId.getGuid());\r
            passDownMap.put("PACKAGE_VERSION", packageId.getVersion());\r
            \r
            for (int j = 0; j < inheritProperties.length; j ++){\r
                Element property = document.createElement("property");\r
                property.setAttribute("name", inheritProperties[j]);\r
                property.setAttribute("value", passDownMap.get(inheritProperties[j]));\r
                ele.appendChild(property);\r
            }\r
\r
            root.appendChild(ele);\r
        }\r
        //\r
        // <delete includeemptydirs="true">\r
        //   <fileset dir="${DEST_DIR_OUTPUT}" includes="" excludes="" />\r
        // </delete>\r
        //\r
        Element deleteEle = document.createElement("delete");\r
        deleteEle.setAttribute("includeemptydirs", "true");\r
        Element filesetEle = document.createElement("fileset");\r
        filesetEle.setAttribute("dir", "${DEST_DIR_OUTPUT}");\r
        filesetEle.setAttribute("includes", "**/*");\r
        filesetEle.setAttribute("excludes", "*.xml");\r
        deleteEle.appendChild(filesetEle);\r
        root.appendChild(deleteEle);\r
    }\r
\r
    /**\r
      Generate the cleanall elements for BaseName_build.xml. \r
      \r
      @param document current BaseName_build.xml XML document\r
      @param root Root element for current\r
    **/\r
    private void applyDeepcleanElement(Document document, Node root) {\r
        ModuleIdentification[] libinstances = SurfaceAreaQuery.getLibraryInstance(fpdModuleId.getArch());\r
        for (int i = 0; i < libinstances.length; i++) {\r
            //\r
            // Put package file path to module identification\r
            //\r
            PackageIdentification packageId = libinstances[i].getPackage();\r
            \r
            //\r
            // Generate ANT script to clean\r
            //\r
            Element ele = document.createElement("GenBuild");\r
            ele.setAttribute("type", "cleanall");\r
            \r
            //\r
            // Prepare pass down information\r
            //\r
            Map<String, String> passDownMap = new LinkedHashMap<String, String>();\r
            for (int j = 0; j < inheritProperties.length; j ++){\r
                passDownMap.put(inheritProperties[j], "${" + inheritProperties[j] + "}");\r
            }\r
            \r
            passDownMap.put("MODULE_GUID", libinstances[i].getGuid());\r
            passDownMap.put("MODULE_VERSION", libinstances[i].getVersion());\r
            \r
            passDownMap.put("PACKAGE_GUID", packageId.getGuid());\r
            passDownMap.put("PACKAGE_VERSION", packageId.getVersion());\r
            \r
            for (int j = 0; j < inheritProperties.length; j ++){\r
                Element property = document.createElement("property");\r
                property.setAttribute("name", inheritProperties[j]);\r
                property.setAttribute("value", passDownMap.get(inheritProperties[j]));\r
                ele.appendChild(property);\r
            }\r
\r
            root.appendChild(ele);\r
        }\r
        //\r
        // <delete includeemptydirs="true">\r
        //   <fileset dir="${DEST_DIR_OUTPUT}" includes="" excludes="" />\r
        // </delete>\r
        //\r
        Element deleteEle = document.createElement("delete");\r
        deleteEle.setAttribute("includeemptydirs", "true");\r
        Element filesetEle = document.createElement("fileset");\r
        filesetEle.setAttribute("dir", "${DEST_DIR_OUTPUT}");\r
        filesetEle.setAttribute("includes", "**/*");\r
        filesetEle.setAttribute("excludes", "*.xml");\r
        deleteEle.appendChild(filesetEle);\r
        root.appendChild(deleteEle);\r
        \r
        //\r
        // <delete includeemptydirs="true">\r
        //   <fileset dir="${DEST_DIR_DEBUG}" includes="" />\r
        // </delete>\r
        //\r
        deleteEle = document.createElement("delete");\r
        deleteEle.setAttribute("includeemptydirs", "true");\r
        filesetEle = document.createElement("fileset");\r
        filesetEle.setAttribute("dir", "${DEST_DIR_DEBUG}");\r
        filesetEle.setAttribute("includes", "**/*");\r
        deleteEle.appendChild(filesetEle);\r
        root.appendChild(deleteEle);\r
    }\r
\r
    /**\r
      Generate the dependent library instances elements for BaseName_build.xml. \r
      \r
      @param document current BaseName_build.xml XML document\r
      @param root Root element for current\r
    **/\r
    private void applyLibraryInstance(Document document, Node root) {\r
        ModuleIdentification[] libinstances = SurfaceAreaQuery.getLibraryInstance(fpdModuleId.getArch());\r
//        String propertyLibs = "";\r
        for (int i = 0; i < libinstances.length; i++) {\r
            //\r
            // Put package file path to module identification\r
            //\r
            PackageIdentification packageId = libinstances[i].getPackage();\r
            \r
            //\r
            // Generate ANT script to build library instances\r
            //\r
            Element ele = document.createElement("GenBuild");\r
            ele.setAttribute("type", "build");\r
//            ele.setAttribute("inheritAll", "false");\r
            \r
            //\r
            // Prepare pass down information\r
            //\r
            Map<String, String> passDownMap = new LinkedHashMap<String, String>();\r
            for (int j = 0; j < inheritProperties.length; j ++){\r
                passDownMap.put(inheritProperties[j], "${" + inheritProperties[j] + "}");\r
            }\r
            \r
            passDownMap.put("MODULE_GUID", libinstances[i].getGuid());\r
            passDownMap.put("MODULE_VERSION", libinstances[i].getVersion());\r
            \r
            passDownMap.put("PACKAGE_GUID", packageId.getGuid());\r
            passDownMap.put("PACKAGE_VERSION", packageId.getVersion());\r
            \r
            for (int j = 0; j < inheritProperties.length; j ++){\r
                Element property = document.createElement("property");\r
                property.setAttribute("name", inheritProperties[j]);\r
                property.setAttribute("value", passDownMap.get(inheritProperties[j]));\r
                ele.appendChild(property);\r
            }\r
            \r
            root.appendChild(ele);\r
//            propertyLibs += " " + project.getProperty("BIN_DIR") + File.separatorChar + libinstances[i].getName() + ".lib";\r
        }\r
//        project.setProperty("LIBS", propertyLibs.replaceAll("(\\\\)", "/"));\r
    }\r
    \r
     /**\r
      Return the name of the directory that corresponds to the architecture.\r
      This is a translation from the XML Schema tag to a directory that\r
      corresponds to our directory name coding convention.\r
     \r
      **/\r
    private String archDir(String arch) {\r
        return arch.replaceFirst("X64", "x64")\r
                   .replaceFirst("IPF", "Ipf")\r
                   .replaceFirst("IA32", "Ia32")\r
                   .replaceFirst("ARM", "Arm")\r
                   .replaceFirst("EBC", "Ebc");\r
    }\r
\r
    /**\r
      Generate the build source files elements for BaseName_build.xml. \r
      \r
      @param document current BaseName_build.xml XML document\r
      @param root Root element for current\r
    **/\r
    private void applyCompileElement(Document document, Node root) {\r
        //\r
        // Prepare the includes: PackageDependencies and Output debug direactory\r
        //\r
        Set<String> includes = new LinkedHashSet<String>();\r
        String arch = project.getProperty("ARCH");\r
        \r
        //\r
        // WORKSPACE\r
        //\r
        includes.add("${WORKSPACE_DIR}");\r
        \r
        //\r
        // Module iteself\r
        //\r
        includes.add("${MODULE_DIR}");\r
        includes.add("${MODULE_DIR}" + File.separatorChar + archDir(arch));\r
        \r
        //\r
        // Packages in PackageDenpendencies\r
        //\r
        PackageIdentification[] packageDependencies = SurfaceAreaQuery.getDependencePkg(fpdModuleId.getArch());\r
        for (int i = 0; i < packageDependencies.length; i++) {\r
            GlobalData.refreshPackageIdentification(packageDependencies[i]);\r
            File packageFile = packageDependencies[i].getSpdFile();\r
            includes.add(packageFile.getParent() + File.separatorChar + "Include");\r
            includes.add(packageFile.getParent() + File.separatorChar + "Include" + File.separatorChar + archDir(arch));\r
        }\r
\r
        //\r
        // All Dependency Library Instance's PackageDependencies\r
        //\r
        ModuleIdentification[] libinstances = SurfaceAreaQuery.getLibraryInstance(fpdModuleId.getArch());\r
        for (int i = 0; i < libinstances.length; i++) {\r
            SurfaceAreaQuery.push(GlobalData.getDoc(libinstances[i], fpdModuleId.getArch()));\r
            PackageIdentification[] libraryPackageDependencies = SurfaceAreaQuery.getDependencePkg(fpdModuleId.getArch());\r
            for (int j = 0; j < libraryPackageDependencies.length; j++) {\r
                GlobalData.refreshPackageIdentification(libraryPackageDependencies[j]);\r
                File packageFile = libraryPackageDependencies[j].getSpdFile();\r
                includes.add(packageFile.getParent() + File.separatorChar + "Include");\r
                includes.add(packageFile.getParent() + File.separatorChar + "Include" + File.separatorChar + archDir(arch));\r
            }\r
            SurfaceAreaQuery.pop();\r
        }\r
        \r
        \r
        //\r
        // The package which the module belongs to\r
        // TBD\r
        includes.add(fpdModuleId.getModule().getPackage().getPackageDir() + File.separatorChar + "Include");\r
        includes.add(fpdModuleId.getModule().getPackage().getPackageDir() + File.separatorChar + "Include" + File.separatorChar + archDir(arch));\r
\r
        //\r
        // Debug files output directory\r
        //\r
        includes.add("${DEST_DIR_DEBUG}");\r
        \r
        //\r
        // sourceFiles[][0] is FileType, [][1] is File name relative to Module_Dir\r
        //\r
        String[][] sourceFiles = SurfaceAreaQuery.getSourceFiles(fpdModuleId.getArch());\r
\r
        FileProcess fileProcess = new FileProcess();\r
        fileProcess.init(project, includes, document);\r
\r
        String moduleDir = project.getProperty("MODULE_DIR");\r
        //\r
        // Parse all Unicode files\r
        //\r
        for (int i = 0; i < sourceFiles.length; i++) {\r
            //\r
            // Go through all source files. Add MODULE_DIR to preffix\r
            //\r
            File sourceFile =  new File(moduleDir + File.separatorChar + sourceFiles[i][1]);\r
            sourceFiles[i][1] = sourceFile.getPath();\r
            String filetype = sourceFiles[i][0];\r
            if (filetype != null) {\r
                fileProcess.parseFile(sourceFiles[i][1], filetype, root, true);\r
            } else {\r
                fileProcess.parseFile(sourceFiles[i][1], root, true);\r
            }\r
        }\r
        \r
        //\r
        // If exist Unicode file\r
        //\r
        if (fileProcess.isUnicodeExist()) {\r
            Element ele = document.createElement("Build_Unicode_Database");\r
            ele.setAttribute("FILEPATH", ".");\r
            ele.setAttribute("FILENAME", "${BASE_NAME}");\r
            String[] includePaths = includes.toArray(new String[includes.size()]);\r
            Element includesEle = document.createElement("EXTRA.INC");\r
            for (int i = 0; i < includePaths.length; i++) {\r
                Element includeEle = document.createElement("includepath");\r
                includeEle.setAttribute("path", includePaths[i]);\r
                includesEle.appendChild(includeEle);\r
            }\r
            ele.appendChild(includesEle);\r
            root.appendChild(ele);\r
        }\r
\r
        //\r
        // Parse AutoGen.c & AutoGen.h\r
        //\r
        if ( ! fpdModuleId.getModule().getName().equalsIgnoreCase("Shell")) {\r
            fileProcess.parseFile(project.getProperty("DEST_DIR_DEBUG") + File.separatorChar + "AutoGen.c", root, false);\r
        }\r
        \r
        //\r
        // Parse all source files but Unicode files\r
        //\r
        for (int i = 0; i < sourceFiles.length; i++) {\r
            String filetype = sourceFiles[i][0];\r
            if (filetype != null) {\r
                fileProcess.parseFile(sourceFiles[i][1], filetype, root, false);\r
            } else {\r
                fileProcess.parseFile(sourceFiles[i][1], root, false);\r
            }\r
        }\r
        \r
        //\r
        // Initialize SOURCE_FILES for dependcy check use\r
        //\r
        String str = "";\r
        for (int i = 0; i < sourceFiles.length; i++) {\r
            str += " " + sourceFiles[i][1];\r
        }\r
        project.setProperty("SOURCE_FILES", str.replaceAll("(\\\\)", "/"));\r
    }\r
\r
    /**\r
      Generate the section elements for BaseName_build.xml. Library module will\r
      skip this process.  \r
      \r
      @param document current BaseName_build.xml XML document\r
      @param root Root element for current\r
    **/\r
    private void applySectionsElement(Document document, Node root, FfsProcess fp) {\r
        if (fpdModuleId.getModule().isLibrary()) {\r
            return ;\r
        }\r
        if (fp.initSections(ffsKeyword, project, fpdModuleId)) {\r
            String targetFilename = fpdModuleId.getModule().getGuid() + "-" + "${BASE_NAME}" + FpdParserTask.getSuffix(fpdModuleId.getModule().getModuleType());\r
            String[] list = fp.getGenSectionElements(document, "${BASE_NAME}", fpdModuleId.getModule().getGuid(), targetFilename);\r
\r
            for (int i = 0; i < list.length; i++) {\r
                Element ele = document.createElement(list[i]);\r
                ele.setAttribute("FILEPATH", ".");\r
                ele.setAttribute("FILENAME", "${BASE_NAME}");\r
                root.appendChild(ele);\r
            }\r
        }\r
    }\r
\r
    /**\r
      Generate the output elements for BaseName_build.xml. If module is library,\r
      call the <em>LIB</em> command, else call the <em>GenFfs</em> command. \r
      \r
      @param document current BaseName_build.xml XML document\r
      @param root Root element for current\r
    **/\r
    private void applyOutputElement(Document document, Node root, FfsProcess fp) {\r
        if (fpdModuleId.getModule().isLibrary()) {\r
            //\r
            // call Lib command\r
            //\r
            Element cc = document.createElement("Build_Library");\r
            cc.setAttribute("FILENAME", fpdModuleId.getModule().getName());\r
            root.appendChild(cc);\r
        }\r
        //\r
        // if it is a module but library\r
        //\r
        else {\r
            if (fp.getFfsNode() != null) {\r
                root.appendChild(fp.getFfsNode());\r
            }\r
        }\r
    }\r
\r
}\r