ENH: Adding new EXECUTE_PROCESS command that interfaces to KWSys Process Execution.

This commit is contained in:
Brad King 2006-02-03 16:51:46 -05:00
parent 131c4a3a17
commit 2683c5bd0d
3 changed files with 450 additions and 0 deletions

View File

@ -18,6 +18,7 @@
#if defined(CMAKE_BUILD_WITH_CMAKE) #if defined(CMAKE_BUILD_WITH_CMAKE)
#include "cmAuxSourceDirectoryCommand.cxx" #include "cmAuxSourceDirectoryCommand.cxx"
#include "cmEndWhileCommand.cxx" #include "cmEndWhileCommand.cxx"
#include "cmExecuteProcessCommand.cxx"
#include "cmExportLibraryDependencies.cxx" #include "cmExportLibraryDependencies.cxx"
#include "cmEnableLanguageCommand.cxx" #include "cmEnableLanguageCommand.cxx"
#include "cmFLTKWrapUICommand.cxx" #include "cmFLTKWrapUICommand.cxx"
@ -60,6 +61,7 @@ void GetPredefinedCommands(std::list<cmCommand*>&
commands.push_back(new cmAuxSourceDirectoryCommand); commands.push_back(new cmAuxSourceDirectoryCommand);
commands.push_back(new cmEnableLanguageCommand); commands.push_back(new cmEnableLanguageCommand);
commands.push_back(new cmEndWhileCommand); commands.push_back(new cmEndWhileCommand);
commands.push_back(new cmExecuteProcessCommand);
commands.push_back(new cmExportLibraryDependenciesCommand); commands.push_back(new cmExportLibraryDependenciesCommand);
commands.push_back(new cmFLTKWrapUICommand); commands.push_back(new cmFLTKWrapUICommand);
commands.push_back(new cmGetCMakePropertyCommand); commands.push_back(new cmGetCMakePropertyCommand);

View File

@ -0,0 +1,335 @@
/*=========================================================================
Program: CMake - Cross-Platform Makefile Generator
Module: $RCSfile$
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) 2002 Kitware, Inc., Insight Consortium. All rights reserved.
See Copyright.txt or http://www.cmake.org/HTML/Copyright.html for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notices for more information.
=========================================================================*/
#include "cmExecuteProcessCommand.h"
#include "cmSystemTools.h"
#include <cmsys/Process.h>
// cmExecuteProcessCommand
bool cmExecuteProcessCommand::InitialPass(std::vector<std::string> const& args)
{
if(args.size() < 1 )
{
this->SetError("called with incorrect number of arguments");
return false;
}
std::vector< std::vector<const char*> > cmds;
std::string arguments;
bool doing_command = false;
unsigned int command_index = 0;
bool output_quiet = false;
bool error_quiet = false;
std::string timeout_string;
std::string input_file;
std::string output_file;
std::string error_file;
std::string output_variable;
std::string error_variable;
std::string result_variable;
std::string working_directory;
for(size_t i=0; i < args.size(); ++i)
{
if(args[i] == "COMMAND")
{
doing_command = true;
command_index = cmds.size();
cmds.push_back(std::vector<const char*>());
}
else if(args[i] == "OUTPUT_VARIABLE")
{
doing_command = false;
if(++i < args.size())
{
output_variable = args[i];
}
else
{
this->SetError(" called with no value for OUTPUT_VARIABLE.");
return false;
}
}
else if(args[i] == "ERROR_VARIABLE")
{
doing_command = false;
if(++i < args.size())
{
error_variable = args[i];
}
else
{
this->SetError(" called with no value for ERROR_VARIABLE.");
return false;
}
}
else if(args[i] == "RESULT_VARIABLE")
{
doing_command = false;
if(++i < args.size())
{
result_variable = args[i];
}
else
{
this->SetError(" called with no value for RESULT_VARIABLE.");
return false;
}
}
else if(args[i] == "WORKING_DIRECTORY")
{
doing_command = false;
if(++i < args.size())
{
working_directory = args[i];
}
else
{
this->SetError(" called with no value for WORKING_DIRECTORY.");
return false;
}
}
else if(args[i] == "INPUT_FILE")
{
doing_command = false;
if(++i < args.size())
{
input_file = args[i];
}
else
{
this->SetError(" called with no value for INPUT_FILE.");
return false;
}
}
else if(args[i] == "OUTPUT_FILE")
{
doing_command = false;
if(++i < args.size())
{
output_file = args[i];
}
else
{
this->SetError(" called with no value for OUTPUT_FILE.");
return false;
}
}
else if(args[i] == "ERROR_FILE")
{
doing_command = false;
if(++i < args.size())
{
error_file = args[i];
}
else
{
this->SetError(" called with no value for ERROR_FILE.");
return false;
}
}
else if(args[i] == "TIMEOUT")
{
doing_command = false;
if(++i < args.size())
{
timeout_string = args[i];
}
else
{
this->SetError(" called with no value for TIMEOUT.");
return false;
}
}
else if(args[i] == "OUTPUT_QUIET")
{
doing_command = false;
output_quiet = true;
}
else if(args[i] == "ERROR_QUIET")
{
doing_command = false;
error_quiet = true;
}
else if(doing_command)
{
cmds[command_index].push_back(args[i].c_str());
}
}
// Check for commands given.
if(cmds.empty())
{
this->SetError(" called with no COMMAND argument.");
return false;
}
for(unsigned int i=0; i < cmds.size(); ++i)
{
if(cmds[i].empty())
{
this->SetError(" given COMMAND argument with no value.");
return false;
}
else
{
// Add the null terminating pointer to the command argument list.
cmds[i].push_back(0);
}
}
// Parse the timeout string.
double timeout = -1;
if(!timeout_string.empty())
{
if(sscanf(timeout_string.c_str(), "%lg", &timeout) != 1)
{
this->SetError(" called with TIMEOUT value that could not be parsed.");
return false;
}
}
// Create a process instance.
cmsysProcess* cp = cmsysProcess_New();
// Set the command sequence.
for(unsigned int i=0; i < cmds.size(); ++i)
{
cmsysProcess_AddCommand(cp, &*cmds[i].begin());
}
// Set the process working directory.
if(!working_directory.empty())
{
cmsysProcess_SetWorkingDirectory(cp, working_directory.c_str());
}
// Always hide the process window.
cmsysProcess_SetOption(cp, cmsysProcess_Option_HideWindow, 1);
// Check the output variables.
bool merge_output = (output_variable == error_variable);
if(error_variable.empty() && !error_quiet)
{
cmsysProcess_SetPipeShared(cp, cmsysProcess_Pipe_STDERR, 1);
}
if(!input_file.empty())
{
cmsysProcess_SetPipeFile(cp, cmsysProcess_Pipe_STDIN, input_file.c_str());
}
if(!output_file.empty())
{
cmsysProcess_SetPipeFile(cp, cmsysProcess_Pipe_STDOUT, output_file.c_str());
}
if(!error_file.empty())
{
cmsysProcess_SetPipeFile(cp, cmsysProcess_Pipe_STDERR, error_file.c_str());
}
// Set the timeout if any.
if(timeout >= 0)
{
cmsysProcess_SetTimeout(cp, timeout);
}
// Start the process.
cmsysProcess_Execute(cp);
// Read the process output.
std::vector<char> tempOutput;
std::vector<char> tempError;
int length;
char* data;
int p;
while((p = cmsysProcess_WaitForData(cp, &data, &length, 0), p))
{
// Translate NULL characters in the output into valid text.
for(int i=0; i < length; ++i)
{
if(data[i] == '\0')
{
data[i] = ' ';
}
}
// Put the output in the right place.
if(p == cmsysProcess_Pipe_STDOUT && !output_quiet ||
p == cmsysProcess_Pipe_STDERR && !error_quiet && merge_output)
{
if(output_variable.empty())
{
cmSystemTools::Stdout(data, length);
}
else
{
tempOutput.insert(tempOutput.end(), data, data+length);
}
}
else if(p == cmsysProcess_Pipe_STDERR && !error_quiet)
{
if(!error_variable.empty())
{
tempError.insert(tempError.end(), data, data+length);
}
}
}
// All output has been read. Wait for the process to exit.
cmsysProcess_WaitForExit(cp, 0);
// Store the output obtained.
if(!output_variable.empty())
{
tempOutput.push_back('\0');
m_Makefile->AddDefinition(output_variable.c_str(), &*tempOutput.begin());
}
if(!merge_output && !error_variable.empty())
{
tempError.push_back('\0');
m_Makefile->AddDefinition(error_variable.c_str(), &*tempError.begin());
}
// Store the result of running the process.
if(!result_variable.empty())
{
switch(cmsysProcess_GetState(cp))
{
case cmsysProcess_State_Exited:
{
int v = cmsysProcess_GetExitValue(cp);
char buf[100];
sprintf(buf, "%d", v);
m_Makefile->AddDefinition(result_variable.c_str(), buf);
}
break;
case cmsysProcess_State_Exception:
m_Makefile->AddDefinition(result_variable.c_str(),
cmsysProcess_GetExceptionString(cp));
break;
case cmsysProcess_State_Error:
m_Makefile->AddDefinition(result_variable.c_str(),
cmsysProcess_GetErrorString(cp));
break;
case cmsysProcess_State_Expired:
m_Makefile->AddDefinition(result_variable.c_str(),
"Process terminated due to timeout");
break;
}
}
// Delete the process instance.
cmsysProcess_Delete(cp);
return true;
}

View File

@ -0,0 +1,113 @@
/*=========================================================================
Program: CMake - Cross-Platform Makefile Generator
Module: $RCSfile$
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) 2002 Kitware, Inc., Insight Consortium. All rights reserved.
See Copyright.txt or http://www.cmake.org/HTML/Copyright.html for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notices for more information.
=========================================================================*/
#ifndef cmExecuteProcessCommand_h
#define cmExecuteProcessCommand_h
#include "cmCommand.h"
/** \class cmExecuteProcessCommand
* \brief Command that adds a target to the build system.
*
* cmExecuteProcessCommand is a CMake language interface to the KWSys
* Process Execution implementation.
*/
class cmExecuteProcessCommand : public cmCommand
{
public:
/**
* This is a virtual constructor for the command.
*/
virtual cmCommand* Clone()
{
return new cmExecuteProcessCommand;
}
/**
* This is called when the command is first encountered in
* the CMakeLists.txt file.
*/
virtual bool InitialPass(std::vector<std::string> const& args);
/**
* The name of the command as specified in CMakeList.txt.
*/
virtual const char* GetName()
{return "EXECUTE_PROCESS";}
/**
* This determines if the command is invoked when in script mode.
*/
virtual bool IsScriptable() { return true; }
/**
* Succinct documentation.
*/
virtual const char* GetTerseDocumentation()
{
return "Execute one or more child processes.";
}
/**
* More documentation.
*/
virtual const char* GetFullDocumentation()
{
return
" EXECUTE_PROCESS(COMMAND <cmd1> [args1...]]\n"
" [COMMAND <cmd2> [args2...] [...]]\n"
" [WORKING_DIRECTORY <directory>]\n"
" [TIMEOUT <seconds>]\n"
" [RESULT_VARIABLE <variable>]\n"
" [OUTPUT_VARIABLE <variable>]\n"
" [ERROR_VARIABLE <variable>]\n"
" [INPUT_FILE <file>]\n"
" [OUTPUT_FILE <file>]\n"
" [ERROR_FILE <file>]\n"
" [OUTPUT_QUIET]\n"
" [ERROR_QUIET])\n"
"Runs the given sequence of one or more commands with the standard "
"output of each process piped to the standard input of the next. "
"A single standard error pipe is used for all processes. "
"If WORKING_DIRECTORY is given the named directory will be set as "
"the current working directory of the child processes. "
"If TIMEOUT is given the child processes will be terminated if they "
"do not finish in the specified number of seconds "
"(fractions are allowed). "
"If RESULT_VARIABLE is given the variable will be set to contain "
"the result of running the processes. This will be an integer return "
"code from the last child or a string describing an error condition. "
"If OUTPUT_VARIABLE or ERROR_VARIABLE are given the variable named "
"will be set with the contents of the standard output and standard error "
"pipes respectively. If the same variable is named for both pipes "
"their output will be merged in the order produced. "
"If INPUT_FILE, OUTPUT_FILE, or ERROR_FILE is given the file named "
"will be attached to the standard input of the first process, "
"standard output of the last process, or standard error of all "
"processes respectively. "
"If OUTPUT_QUIET or ERROR_QUIET is given then the standard output "
"or standard error results will be quietly ignored. "
"If more than one OUTPUT_* or ERROR_* option is given for the same "
"pipe the precedence is not specified. "
"If no OUTPUT_* or ERROR_* options are given the output will be shared "
"with the corresponding pipes of the CMake process itself."
;
}
cmTypeMacro(cmExecuteProcessCommand, cmCommand);
};
#endif