added specific PP definition for EBC because of previous BuildMacro.xml changes

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

package org.tianocore.build;\r
\r
import java.io.BufferedReader;\r
import java.io.File;\r
import java.io.InputStreamReader;\r
import java.util.Iterator;\r
import java.util.LinkedHashSet;\r
import java.util.Map;\r
import java.util.Set;\r
\r
import org.apache.tools.ant.BuildException;\r
import org.apache.tools.ant.Task;\r
import org.tianocore.build.fpd.FpdParserTask;\r
import org.tianocore.build.global.GlobalData;\r
import org.tianocore.build.toolchain.ToolChainInfo;\r
\r
public class FrameworkBuildTask extends Task{\r
\r
    private Set<File> buildFiles = new LinkedHashSet<File>();\r
    \r
    private Set<File> fpdFiles = new LinkedHashSet<File>();\r
    \r
    private Set<File> msaFiles = new LinkedHashSet<File>();\r
    \r
    ///\r
    /// there are three type: all (build), clean and cleanall\r
    ///\r
    private String type = "all";\r
    \r
    public void execute() throws BuildException {\r
        //\r
        // Seach build.xml -> .FPD -> .MSA file\r
        //\r
        try {\r
            //\r
            // Gen Current Working Directory\r
            //\r
            File dummyFile = new File(".");\r
            File cwd = dummyFile.getCanonicalFile();\r
            File[] files = cwd.listFiles();\r
            for (int i = 0; i < files.length; i++) {\r
                if (files[i].isFile()) {\r
                    if (files[i].getName().equalsIgnoreCase("build.xml")) {\r
                        //\r
                        // First, search build.xml, if found, ANT call it\r
                        //\r
                        buildFiles.add(files[i]);\r
\r
                    } else if (files[i].getName().endsWith(".fpd")) {\r
                        //\r
                        // Second, search FPD file, if found, build it\r
                        //\r
                        fpdFiles.add(files[i]);\r
                    } else if (files[i].getName().endsWith(".msa")) {\r
                        //\r
                        // Third, search MSA file, if found, build it\r
                        //\r
                        msaFiles.add(files[i]);\r
                    }\r
                }\r
            }\r
        } catch (Exception e) {\r
            e.printStackTrace();\r
            throw new BuildException(e.getMessage());\r
        }\r
        \r
        //\r
        // If there is no build files or FPD files or MSA files, stop build\r
        //\r
        if (fpdFiles.size() == 0 && msaFiles.size() == 0) {\r
            throw new BuildException("Can't find any build.xml file or FPD files or MSA files in current directory. ");\r
        }\r
        \r
        File buildFile = intercommuniteWithUser();\r
        System.out.println("Start to build file [" + buildFile.getPath() + "] ..>> ");\r
        \r
        //\r
        // Deal with all environment variable (Add them to properties)\r
        //\r
        backupSystemProperties();\r
        \r
        //\r
        // Get ToolChain Info from environment\r
        //\r
        ToolChainInfo envToolChainInfo = new ToolChainInfo(); \r
        envToolChainInfo.addTargets(getProject().getProperty("TARGET")); \r
        envToolChainInfo.addTagnames(getProject().getProperty("TAGNAME")); \r
        envToolChainInfo.addArchs(getProject().getProperty("ARCH")); \r
        GlobalData.setToolChainEnvInfo(envToolChainInfo);\r
        \r
        //\r
        // Global Data initialization\r
        //\r
        String toolsDefFilename = "tools_def.txt";\r
        if (getProject().getProperty("TOOLS_DEF") != null) {\r
            toolsDefFilename = getProject().getProperty("TOOLS_DEF");\r
        }\r
        \r
        GlobalData.initInfo("Tools" + File.separatorChar + "Conf" + File.separatorChar + "FrameworkDatabase.db",\r
                            getProject().getProperty("WORKSPACE_DIR"), toolsDefFilename);\r
        \r
        //\r
        // Build every FPD files (PLATFORM build)\r
        //\r
        if (buildFile.getName().endsWith(".fpd")) {\r
            FpdParserTask fpdParserTask = new FpdParserTask();\r
            fpdParserTask.setType(type);\r
            fpdParserTask.setProject(getProject());\r
            fpdParserTask.setFpdFile(buildFile);\r
            fpdParserTask.execute();\r
        }\r
        \r
        //\r
        // Build every MSA files (SINGLE MODULE BUILD)\r
        //\r
        else if (buildFile.getName().endsWith(".msa")) {\r
            GenBuildTask genBuildTask = new GenBuildTask();\r
            genBuildTask.setType(type);\r
            genBuildTask.setProject(getProject());\r
            genBuildTask.setMsaFile(buildFile);\r
            genBuildTask.execute();\r
        }\r
    }\r
    \r
    /**\r
      Transfer system environment variables to ANT properties. If system variable \r
      already exiests in ANT properties, skip it.\r
      \r
    **/\r
    private void backupSystemProperties() {\r
        Map<String, String> sysProperties = System.getenv();\r
        Set<String> keys = sysProperties.keySet();\r
        Iterator<String> iter = keys.iterator();\r
        while (iter.hasNext()) {\r
            String name = iter.next();\r
            \r
            //\r
            // If system environment variable is not in ANT properties, add it\r
            //\r
            if (getProject().getProperty(name) == null) {\r
                getProject().setProperty(name, sysProperties.get(name));\r
            }\r
        }\r
    }\r
\r
    private File intercommuniteWithUser(){\r
        File file = null;\r
        if (fpdFiles.size() + msaFiles.size() > 1) {\r
            File[] allFiles = new File[fpdFiles.size() + msaFiles.size()];\r
            int index = 0;\r
            Iterator<File> iter = fpdFiles.iterator();\r
            while (iter.hasNext()) {\r
                allFiles[index] = iter.next();\r
                index++;\r
            }\r
            iter = msaFiles.iterator();\r
            while (iter.hasNext()) {\r
                allFiles[index] = iter.next();\r
                index++;\r
            }\r
            System.out.println("Find " + allFiles.length + " FPD and MSA files: ");\r
            for (int i = 0; i < allFiles.length; i++) {\r
                System.out.println("[" + (i + 1) + "]: " + allFiles[i].getName());\r
            }\r
            \r
            boolean flag = true;\r
            System.out.print("Please select one file to build:[1] ");\r
            do{\r
                BufferedReader br = new BufferedReader(new InputStreamReader(System.in));\r
                try {\r
                     String str = br.readLine();\r
                     if (str.trim().length() == 0) {\r
                         file = allFiles[0];\r
                         flag = false;\r
                         continue ;\r
                     }\r
                     int indexSelect = Integer.parseInt(str);\r
                     if (indexSelect <=0 || indexSelect > allFiles.length) {\r
                         System.out.print("Please enter a number between [1.." + allFiles.length + "]:[1] ");\r
                         continue ;\r
                     } else {\r
                         file = allFiles[indexSelect - 1];\r
                         flag = false;\r
                         continue ;\r
                     }\r
                } catch (Exception e) {\r
                    System.out.print("Please enter a valid number:[1] ");\r
                    flag = true;\r
                }\r
            } while (flag);\r
        }\r
        else if (fpdFiles.size() == 1) {\r
            file = fpdFiles.toArray(new File[1])[0];\r
        }\r
        else if (msaFiles.size() == 1) {\r
            file = msaFiles.toArray(new File[1])[0];\r
        }\r
        return file;\r
    }\r
    \r
    \r
    public void setType(String type) {\r
        if (type.equalsIgnoreCase("clean") || type.equalsIgnoreCase("cleanall")) {\r
            this.type = type.toLowerCase();\r
        }\r
        else {\r
            this.type = "all";\r
        }\r
    }\r
}\r