One GenFvImage can handle all archs now.

[mirror_edk2.git] / Tools / Source / FrameworkTasks / org / tianocore / framework / tasks / FwImageTask.java

/** @file\r
 FwImageTask class.\r
\r
 FwImageTask is used to call FwImage.ext to generate the FwImage.\r
  \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
 **/\r
package org.tianocore.framework.tasks;\r
\r
import org.apache.tools.ant.Task;\r
import org.apache.tools.ant.Project;\r
import org.apache.tools.ant.BuildException;\r
import org.apache.tools.ant.taskdefs.Execute;\r
import org.apache.tools.ant.taskdefs.LogStreamHandler;\r
import org.apache.tools.ant.types.Commandline;\r
import org.tianocore.build.*;\r
\r
/**\r
  FwImageTask class.\r
\r
  FwImageTask is used to call FwImage.ext to generate the FwImage.\r
**/\r
public class FwImageTask extends Task implements EfiDefine{\r
    ///\r
    /// time&data \r
    ///\r
    private String time = "";\r
    ///\r
    /// input PE image\r
    /// \r
    private String peImage = "";\r
    ///\r
    /// output EFI image\r
    /// \r
    private String outImage = "";\r
    ///\r
    /// component type\r
    ///\r
    private String componentType = "";\r
\r
    /**\r
     * assemble tool command line & execute tool command line\r
     * \r
     * @throws BuildException\r
     */\r
    /**\r
      execute \r
      \r
      FwimageTask execute function is to assemble tool command line & execute\r
      tool command line\r
      \r
      @throws BuidException\r
    **/\r
    public void execute() throws BuildException {\r
\r
        Project project = this.getOwningTarget().getProject();\r
        //\r
        // absolute path of efi tools\r
        //\r
        String path = project.getProperty("env.FRAMEWORK_TOOLS_PATH");\r
        String command;\r
        if (path == null) {\r
            command = "fwimage";\r
        } else {\r
            command = path + "/" + "fwimage";\r
        }\r
        //\r
        // argument of tools\r
        //\r
        String argument = time + componentType + peImage + outImage;\r
        //\r
        // return value of fwimage execution\r
        //\r
        int revl = -1;\r
\r
        try {\r
            Commandline cmdline = new Commandline();\r
            cmdline.setExecutable(command);\r
            cmdline.createArgument().setLine(argument);\r
\r
            LogStreamHandler streamHandler = new LogStreamHandler(this,\r
                    Project.MSG_INFO, Project.MSG_WARN);\r
            Execute runner = new Execute(streamHandler, null);\r
\r
            runner.setAntRun(project);\r
            runner.setCommandline(cmdline.getCommandline());\r
            //System.out.println(Commandline.toString(cmdline.getCommandline()));\r
            //GenBuildTask.myLogger.log(Commandline.toString(cmdline.getCommandline()),0);\r
            //getProject().log(Commandline.toString(cmdline.getCommandline()));\r
            revl = runner.execute();\r
            if (EFI_SUCCESS == revl) {\r
                //\r
                // command execution success\r
                //\r
                System.out.println("fwimage successed!");\r
            } else {\r
                //\r
                // command execution fail\r
                //\r
                System.out.println("fwimage failed. (error="\r
                        + Integer.toHexString(revl) + ")");\r
                throw new BuildException("fwimage failed. (error="\r
                        + Integer.toHexString(revl) + ")");\r
\r
            }\r
        } catch (Exception e) {\r
            throw new BuildException(e.getMessage());\r
        }\r
    }\r
\r
    /**\r
      setTime\r
      \r
      This function is to set operation of class member "time".\r
      \r
      @param time            string of time\r
    **/\r
    public void setTime(String time) {\r
        this.time = " -t " + time;\r
    }\r
\r
    /**\r
      getTime\r
      \r
      This function is to get class member "time"\r
      @return time          string of time\r
    **/\r
    public String getTime() {\r
        return this.time;\r
    }\r
\r
    /**\r
      getPeImage\r
      \r
      This function is to get class member "peImage".\r
      @return                name of PE image\r
    **/\r
    public String getPeImage() {\r
        return this.peImage;\r
    }\r
\r
    /**\r
      setPeImage\r
      \r
      This function is to set class member "peImage"\r
      @param  peImage        name of PE image\r
    **/\r
    public void setPeImage(String peImage) {\r
        this.peImage = " " + peImage;\r
    }\r
\r
    /**\r
      getOutImage\r
      \r
      This function is to get class member "outImage".\r
      @return                 name of output EFI image\r
    **/\r
    public String getOutImage() {\r
        return this.outImage;\r
    }\r
\r
    /**\r
      setOutImage\r
      \r
      This function is to set class member "outImage".\r
      @param outImage         name of output EFI image\r
    **/\r
    public void setOutImage(String outImage) {\r
        this.outImage = " " + outImage;\r
    }\r
\r
    /**\r
      getComponentType\r
      \r
      This function is to get class member "componentType".\r
      \r
      @return                 string of componentType\r
    **/\r
    public String getComponentType() {\r
        return this.componentType;\r
    }\r
\r
    /**\r
      setComponentType\r
      \r
      This function is to set class member "componentType".\r
      @param  componentType   string of component type\r
    **/\r
    public void setComponentType(String componentType) {\r
        this.componentType = " " + componentType;\r
    }\r
}\r