| /*============================================================================ |
| CMake - Cross Platform Makefile Generator |
| Copyright 2000-2009 Kitware, Inc., Insight Software Consortium |
| |
| Distributed under the OSI-approved BSD License (the "License"); |
| see accompanying file Copyright.txt for details. |
| |
| This software is distributed WITHOUT ANY WARRANTY; without even the |
| implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. |
| See the License for more information. |
| ============================================================================*/ |
| #include "cmCreateTestSourceList.h" |
| #include "cmSourceFile.h" |
| |
| // cmCreateTestSourceList |
| bool cmCreateTestSourceList |
| ::InitialPass(std::vector<std::string> const& args, cmExecutionStatus &) |
| { |
| if (args.size() < 3) |
| { |
| this->SetError("called with wrong number of arguments."); |
| return false; |
| } |
| |
| |
| std::vector<std::string>::const_iterator i = args.begin(); |
| std::string extraInclude; |
| std::string function; |
| std::vector<std::string> tests; |
| // extract extra include and function ot |
| for(; i != args.end(); i++) |
| { |
| if(*i == "EXTRA_INCLUDE") |
| { |
| ++i; |
| if(i == args.end()) |
| { |
| this->SetError("incorrect arguments to EXTRA_INCLUDE"); |
| return false; |
| } |
| extraInclude = "#include \""; |
| extraInclude += *i; |
| extraInclude += "\"\n"; |
| } |
| else if(*i == "FUNCTION") |
| { |
| ++i; |
| if(i == args.end()) |
| { |
| this->SetError("incorrect arguments to FUNCTION"); |
| return false; |
| } |
| function = *i; |
| function += "(&ac, &av);\n"; |
| } |
| else |
| { |
| tests.push_back(*i); |
| } |
| } |
| i = tests.begin(); |
| |
| // Name of the source list |
| |
| const char* sourceList = i->c_str(); |
| ++i; |
| |
| // Name of the test driver |
| // make sure they specified an extension |
| if (cmSystemTools::GetFilenameExtension(*i).size() < 2) |
| { |
| this->SetError( |
| "You must specify a file extension for the test driver file."); |
| return false; |
| } |
| std::string driver = this->Makefile->GetCurrentOutputDirectory(); |
| driver += "/"; |
| driver += *i; |
| ++i; |
| |
| std::string configFile = |
| this->Makefile->GetRequiredDefinition("CMAKE_ROOT"); |
| |
| configFile += "/Templates/TestDriver.cxx.in"; |
| // Create the test driver file |
| |
| std::vector<std::string>::const_iterator testsBegin = i; |
| std::vector<std::string> tests_func_name; |
| |
| // The rest of the arguments consist of a list of test source files. |
| // Sadly, they can be in directories. Let's find a unique function |
| // name for the corresponding test, and push it to the tests_func_name |
| // list. |
| // For the moment: |
| // - replace spaces ' ', ':' and '/' with underscores '_' |
| std::string forwardDeclareCode; |
| for(i = testsBegin; i != tests.end(); ++i) |
| { |
| if(*i == "EXTRA_INCLUDE") |
| { |
| break; |
| } |
| std::string func_name; |
| if (cmSystemTools::GetFilenamePath(*i).size() > 0) |
| { |
| func_name = cmSystemTools::GetFilenamePath(*i) + "/" + |
| cmSystemTools::GetFilenameWithoutLastExtension(*i); |
| } |
| else |
| { |
| func_name = cmSystemTools::GetFilenameWithoutLastExtension(*i); |
| } |
| cmSystemTools::ConvertToUnixSlashes(func_name); |
| cmSystemTools::ReplaceString(func_name, " ", "_"); |
| cmSystemTools::ReplaceString(func_name, "/", "_"); |
| cmSystemTools::ReplaceString(func_name, ":", "_"); |
| tests_func_name.push_back(func_name); |
| forwardDeclareCode += "int "; |
| forwardDeclareCode += func_name; |
| forwardDeclareCode += "(int, char*[]);\n"; |
| } |
| |
| std::string functionMapCode; |
| int numTests = 0; |
| std::vector<std::string>::iterator j; |
| for(i = testsBegin, j = tests_func_name.begin(); i != tests.end(); ++i, ++j) |
| { |
| std::string func_name; |
| if (cmSystemTools::GetFilenamePath(*i).size() > 0) |
| { |
| func_name = cmSystemTools::GetFilenamePath(*i) + "/" + |
| cmSystemTools::GetFilenameWithoutLastExtension(*i); |
| } |
| else |
| { |
| func_name = cmSystemTools::GetFilenameWithoutLastExtension(*i); |
| } |
| functionMapCode += " {\n" |
| " \""; |
| functionMapCode += func_name; |
| functionMapCode += "\",\n" |
| " "; |
| functionMapCode += *j; |
| functionMapCode += "\n" |
| " },\n"; |
| numTests++; |
| } |
| if(extraInclude.size()) |
| { |
| this->Makefile->AddDefinition("CMAKE_TESTDRIVER_EXTRA_INCLUDES", |
| extraInclude.c_str()); |
| } |
| if(function.size()) |
| { |
| this->Makefile->AddDefinition("CMAKE_TESTDRIVER_ARGVC_FUNCTION", |
| function.c_str()); |
| } |
| this->Makefile->AddDefinition("CMAKE_FORWARD_DECLARE_TESTS", |
| forwardDeclareCode.c_str()); |
| this->Makefile->AddDefinition("CMAKE_FUNCTION_TABLE_ENTIRES", |
| functionMapCode.c_str()); |
| bool res = true; |
| if ( !this->Makefile->ConfigureFile(configFile.c_str(), driver.c_str(), |
| false, true, false) ) |
| { |
| res = false; |
| } |
| |
| // Construct the source list. |
| std::string sourceListValue; |
| { |
| cmSourceFile* sf = this->Makefile->GetOrCreateSource(driver); |
| sf->SetProperty("ABSTRACT","0"); |
| sourceListValue = args[1]; |
| } |
| for(i = testsBegin; i != tests.end(); ++i) |
| { |
| cmSourceFile* sf = this->Makefile->GetOrCreateSource(*i); |
| sf->SetProperty("ABSTRACT","0"); |
| sourceListValue += ";"; |
| sourceListValue += *i; |
| } |
| |
| this->Makefile->AddDefinition(sourceList, sourceListValue.c_str()); |
| return res; |
| } |
| |
| |
| |