Initial import.

[mirror_edk2.git] / Tools / Source / Cpptasks / net / sf / antcontrib / cpptasks / userdefine / CommandLineUserDefine.java

package net.sf.antcontrib.cpptasks.userdefine;\r
\r
import java.io.File;\r
import java.util.Iterator;\r
import java.util.StringTokenizer;\r
import java.util.Vector;\r
\r
import net.sf.antcontrib.cpptasks.CCTask;\r
import net.sf.antcontrib.cpptasks.CUtil;\r
import net.sf.antcontrib.cpptasks.types.CommandLineArgument;\r
import net.sf.antcontrib.cpptasks.types.ConditionalFileSet;\r
import net.sf.antcontrib.cpptasks.types.LibrarySet;\r
\r
import org.apache.tools.ant.BuildException;\r
import org.apache.tools.ant.DirectoryScanner;\r
import org.apache.tools.ant.Project;\r
/*\r
 * \r
 * Copyright 2001-2004 The Ant-Contrib project\r
 *\r
 *  Licensed under the Apache License, Version 2.0 (the "License");\r
 *  you may not use this file except in compliance with the License.\r
 *  You may obtain a copy of the License at\r
 *\r
 *      http://www.apache.org/licenses/LICENSE-2.0\r
 *\r
 *  Unless required by applicable law or agreed to in writing, software\r
 *  distributed under the License is distributed on an "AS IS" BASIS,\r
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
 *  See the License for the specific language governing permissions and\r
 *  limitations under the License.\r
 */\r
public class CommandLineUserDefine {\r
\r
    String command;\r
    \r
    /*\r
     * The follows variable set at child class.\r
     */\r
    String includeFileFlag = null;\r
    String entryPointFlag = null;\r
    String subSystemFlag = null;\r
    String mapFlag = null;\r
    String pdbFlag = null;\r
    String outputFileFlag = null;\r
    String includePathDelimiter = null;\r
    \r
    /*\r
     * get lib string if Vendor = "gcc", it should respectively aadd "-(" and ")-" \r
     * at library set before and end. This value set at userDefineCompiler class.\r
     */\r
    Vector<String> libSetList = new Vector<String>();\r
    Vector<String> fileList = new Vector<String>();\r
    public void command(CCTask cctask, UserDefineDef userdefine){\r
        File workdir;\r
        File outdir;\r
        Project project = cctask.getProject();\r
        if(userdefine.getWorkdir() == null) {\r
            workdir = new File(".");\r
        }\r
        else {\r
            workdir = userdefine.getWorkdir();\r
        } \r
        \r
        /*\r
         * generate cmdline= command + args + includepath + endargs + outfile\r
         */ \r
        Vector args = new Vector();\r
        Vector argsWithoutSpace = new Vector();\r
        Vector endargs = new Vector();\r
        Vector endargsWithoutSpace = new Vector();\r
        Vector includePath = new Vector();\r
        \r
        /*\r
         * Generate cmdline = command        + \r
         *                    general args   + \r
         *                    outputflag     + outputfile\r
         *                    subsystemFlag  + subsystemValue  +\r
         *                    includeFlag    + includeFile     +\r
         *                    includeFileincludpath            + \r
         *                    entryPointFlag + entryPointValue +\r
         *                    mapFlag        + mapValue        +\r
         *                    pdbFlag        + pdbValue        +\r
         *                    endargs                          + \r
         *                   \r
         * \r
         */\r
        /*\r
         * get Args.\r
         */\r
        CommandLineArgument[] argument = userdefine.getActiveProcessorArgs();\r
        for (int j = 0; j < argument.length; j++) {\r
            if (argument[j].getLocation() == 0) {\r
                args.addElement(argument[j].getValue());\r
            } else {\r
                endargs.addElement(argument[j].getValue());\r
            }\r
        }\r
        /*\r
         * get include path.\r
         */\r
        String[] incPath = userdefine.getActiveIncludePaths();\r
        for (int j = 0; j < incPath.length; j++) {\r
            if(incPath[j].indexOf(' ') >= 0) {\r
                includePath.addElement( includePathDelimiter + incPath[j]);\r
                //includePath.addElement( includePathDelimiter + "\"" + incPath[j] + "\"");\r
            }\r
            else {\r
                includePath.addElement( includePathDelimiter + incPath[j]);\r
            }\r
        }\r
        /*\r
         * Remove space in args and endargs.\r
         */\r
        for ( int i=0; i < args.size(); i++) {\r
            String str = (String)args.get(i);\r
            StringTokenizer st = new StringTokenizer(str);\r
            while(st.hasMoreTokens()) {\r
                argsWithoutSpace.addElement(st.nextToken());\r
            }\r
        }\r
        for ( int i=0; i < endargs.size(); i++) {\r
            String str = (String)endargs.get(i);\r
            StringTokenizer st = new StringTokenizer(str);\r
            while(st.hasMoreTokens()) {\r
                endargsWithoutSpace.addElement(st.nextToken());\r
            }\r
        }\r
        \r
        int cmdLen = 0;\r
        if(userdefine.getOutdir() == null) {\r
            outdir = new File(".");\r
            /*\r
             * command + args + endargs + includepath + sourcefile\r
             */\r
            cmdLen = 1 + argsWithoutSpace.size() + endargsWithoutSpace.size() + includePath.size() + 1;\r
        }\r
        else {\r
            outdir = userdefine.getOutdir();\r
            /*\r
             * command + args + endargs + includepath + sourcefile + outfile\r
             */\r
            cmdLen = 1 + argsWithoutSpace.size() + endargsWithoutSpace.size() + includePath.size() + 2;\r
        }\r
        if (includeFileFlag != null && includeFileFlag.trim().length() > 0){\r
            cmdLen++;\r
        }\r
        if (entryPointFlag != null && entryPointFlag.trim().length() > 0){\r
            cmdLen++;\r
        }\r
        if (subSystemFlag != null && subSystemFlag.trim().length() > 0){\r
            cmdLen++;\r
        }\r
        if (mapFlag != null && mapFlag.trim().length() > 0){\r
            cmdLen++;\r
        }\r
        if (pdbFlag != null && pdbFlag.trim().length() > 0){\r
            cmdLen++;\r
        }\r
        if (libSetList != null && libSetList.size() > 0){\r
            cmdLen = cmdLen + libSetList.size();\r
        }\r
        if (fileList != null){\r
            cmdLen = cmdLen + fileList.size();\r
        }\r
        /*\r
         * In gcc the "cr" flag should follow space then add outputfile name, otherwise\r
         * it will pop error. \r
         */\r
        if (outputFileFlag != null && outputFileFlag.trim().length() > 0){\r
            if (outputFileFlag.trim().equalsIgnoreCase("-cr")){\r
                cmdLen = cmdLen + 2;\r
            }else {\r
                cmdLen++;\r
            }\r
            \r
        }\r
        /*\r
         *  for every source file\r
         *  if file is header file, just skip it (add later)\r
         */\r
        Vector srcSets = userdefine.getSrcSets();\r
        if (srcSets.size() == 0) {\r
            String[] cmd = new String[cmdLen - 1];\r
            int index = 0;\r
            cmd[index++] = this.command;\r
            \r
           \r
            \r
            Iterator iter = argsWithoutSpace.iterator();\r
            while (iter.hasNext()) {\r
                cmd[index++] = project.replaceProperties((String)iter.next());\r
                //cmd[index++] = (String)iter.next();\r
            }\r
            \r
            iter = endargsWithoutSpace.iterator();\r
            while (iter.hasNext()) {\r
                cmd[index++] = (String)iter.next();\r
            }\r
            \r
            /*\r
             * "OutputFlag + outputFile" as first option follow command.exe.\r
             */\r
            if (outputFileFlag != null && outputFileFlag.trim().length() > 0){\r
                if (outputFileFlag.trim().equalsIgnoreCase("-cr")){\r
                    cmd[index++] = outputFileFlag;\r
                    cmd[index++] = userdefine.getOutputFile();\r
                }else {\r
                    cmd[index++] =  outputFileFlag + userdefine.getOutputFile();\r
                }\r
            }\r
            \r
            /*\r
             * Add fileList to cmd \r
             */\r
            if (fileList != null && fileList.size()> 0){\r
                for (int i = 0; i < fileList.size(); i++){\r
                    cmd[index++] = fileList.get(i);\r
                }\r
            }\r
            \r
            if (subSystemFlag != null && subSystemFlag.trim().length() > 0){\r
                cmd[index++] = subSystemFlag + userdefine.getSubSystemvalue();\r
            }\r
            if (includeFileFlag != null && includeFileFlag.trim().length() > 0){\r
                cmd[index++] = includeFileFlag + userdefine.getIncludeFile();\r
            }\r
         \r
            iter = includePath.iterator();\r
            while (iter.hasNext()) {\r
                cmd[index++] = (String)iter.next();\r
            }\r
            \r
            if (entryPointFlag != null && entryPointFlag.trim().length() > 0){\r
                //\r
                // If GCC link use __ModuleEntrypoint instead of _ModuleEntryPoint;\r
                //\r
                if (entryPointFlag.equalsIgnoreCase("-e")){\r
                    cmd[index++] = entryPointFlag + "_" + userdefine.getEntryPointvalue();\r
                } else {\r
                    cmd[index++] = entryPointFlag + userdefine.getEntryPointvalue();\r
                }\r
                \r
            }\r
            if (mapFlag != null && mapFlag.trim().length() > 0){\r
                cmd[index++] = mapFlag + userdefine.getMapvalue();\r
            }\r
            if (pdbFlag != null && pdbFlag.trim().length() > 0){\r
                cmd[index++] = pdbFlag + userdefine.getPdbvalue();\r
            }\r
\r
            if (userdefine.getOutdir() != null){\r
                // will add code to generate outfile name and flag\r
                cmd[index++] = "/nologo";\r
            }\r
            \r
            if (libSetList != null && libSetList.size() > 0){\r
                for (int i = 0; i < libSetList.size(); i++){\r
                    cmd[index++] = libSetList.get(i);\r
                }\r
            }\r
\r
            // execute the command\r
            int retval = runCommand(cctask, workdir, cmd);\r
            // if with monitor, add more code\r
            if (retval != 0) {\r
               throw new BuildException(this.command\r
                       + " failed with return code " + retval,\r
                       cctask.getLocation());\r
            }\r
        }\r
        \r
        //\r
        // if have source file append source file in command land.\r
        //\r
        for (int i = 0; i < srcSets.size(); i++) {\r
            ConditionalFileSet srcSet = (ConditionalFileSet) srcSets\r
                    .elementAt(i);\r
            if (srcSet.isActive()) {\r
                // Find matching source files\r
                DirectoryScanner scanner = srcSet.getDirectoryScanner(project);\r
                // Check each source file - see if it needs compilation\r
                String[] fileNames = scanner.getIncludedFiles();\r
                for (int j = 0; j < fileNames.length; j++){\r
                    String[] cmd = new String[cmdLen];\r
                    int index = 0;\r
                    cmd[index++] = this.command;\r
                    \r
                    \r
                    \r
                    Iterator iter = argsWithoutSpace.iterator();\r
                    while (iter.hasNext()) {\r
                        cmd[index++] = (String)iter.next();\r
                    }\r
                    \r
                    iter = endargsWithoutSpace.iterator();\r
                    while (iter.hasNext()) {\r
                        cmd[index++] = (String)iter.next();\r
                    }\r
                    \r
                    /*\r
                     * Add outputFileFlag and output file to cmd\r
                     */\r
                    if (outputFileFlag != null && outputFileFlag.length()> 0){\r
                        if (outputFileFlag.trim().equalsIgnoreCase("-cr")){\r
                            cmd[index++] = outputFileFlag;\r
                            cmd[index++] = userdefine.getOutputFile();\r
                        }else {\r
                            cmd[index++] =  outputFileFlag + userdefine.getOutputFile();\r
                        }\r
                    }\r
                    \r
                    /*\r
                     * Add fileList to cmd \r
                     */\r
                    if (fileList != null && fileList.size()> 0){\r
                        for (int s = 0; s < fileList.size(); s++){\r
                            cmd[index++] = fileList.get(s);\r
                        }\r
                    }\r
                    if (subSystemFlag != null && subSystemFlag.length()> 0){\r
                        cmd[index++] = subSystemFlag + userdefine.getSubSystemvalue();\r
                    }\r
                    if (includeFileFlag != null && includeFileFlag.length()> 0){\r
                        cmd[index++] = includeFileFlag + userdefine.getIncludeFile();\r
                    }\r
                    \r
                    iter = includePath.iterator();\r
                    while (iter.hasNext()) {\r
                        cmd[index++] = (String)iter.next();\r
                    }\r
                    if (userdefine.getOutdir() != null){\r
                        // will add code to generate outfile name and flag\r
                        cmd[index++] = "/nologo";\r
                    }\r
                    \r
                    if (entryPointFlag != null && entryPointFlag.length()> 0){\r
                        cmd[index++] = entryPointFlag + userdefine.getEntryPointvalue();\r
                    }\r
                    if (mapFlag != null && mapFlag.length() > 0){\r
                        cmd[index++] = mapFlag + userdefine.getMapvalue();\r
                    }\r
                    if (pdbFlag != null && pdbFlag.length() > 0){\r
                        cmd[index++] = pdbFlag + userdefine.getPdbvalue();\r
                    }\r
                    \r
                    if (libSetList != null && libSetList.size() > 0){\r
                        for (int k = 0; k < libSetList.size(); k++){\r
                            cmd[index++] = libSetList.get(k);\r
                        }\r
                    }\r
                    \r
                    // execute the command\r
                    cmd[index++] = scanner.getBasedir() + "/" + fileNames[j];\r
                    for (int k = 0; k < cmd.length; k++){\r
                    }\r
                    int retval = runCommand(cctask, workdir, cmd);\r
                    // if with monitor, add more code\r
                    if (retval != 0) {\r
                       throw new BuildException(this.command\r
                               + " failed with return code " + retval,\r
                               cctask.getLocation());\r
                    }\r
                }\r
            }\r
        }\r
    }\r
    \r
    protected int runCommand(CCTask task, File workingDir, String[] cmdline)\r
                    throws BuildException {\r
        return CUtil.runCommand(task, workingDir, cmdline, false, null);\r
\r
    }\r
\r
    protected String getInputFileArgument(File outputDir, String filename,\r
                    int index) {\r
        //\r
        // if there is an embedded space,\r
        // must enclose in quotes\r
        if (filename.indexOf(' ') >= 0) {\r
            StringBuffer buf = new StringBuffer("\"");\r
            buf.append(filename);\r
            buf.append("\"");\r
            return buf.toString();\r
        }\r
        return filename;\r
    }\r
\r
}\r