CMake/Source/cmCreateTestSourceList.cxx

309 lines
8.5 KiB
C++
Raw Normal View History

/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: $RCSfile$
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) 2002 Insight Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm 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 "cmCreateTestSourceList.h"
2002-09-28 00:24:10 +04:00
#include "cmSourceFile.h"
// cmCreateTestSourceList
bool cmCreateTestSourceList::InitialPass(std::vector<std::string> const& argsIn)
{
if (argsIn.size() < 3)
{
this->SetError("called with wrong number of arguments.");
return false;
}
std::vector<std::string> args;
cmSystemTools::ExpandListArguments(argsIn, args);
std::vector<std::string>::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 = *i;
}
else if(*i == "FUNCTION")
{
++i;
if(i == args.end())
{
this->SetError("incorrect arguments to FUNCTION");
return false;
}
function = *i;
}
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
2002-07-31 00:19:14 +04:00
// make sure they specified an extension
2002-07-31 19:07:09 +04:00
if (cmSystemTools::GetFilenameExtension(*i).size() < 2)
2002-07-31 00:19:14 +04:00
{
this->SetError("You must specify a file extenion for the test driver file.");
return false;
}
std::string driver = m_Makefile->GetCurrentOutputDirectory();
driver += "/";
driver += *i;
++i;
std::ofstream fout(driver.c_str());
if (!fout)
{
std::string err = "Could not create file ";
err += driver;
err += " for cmCreateTestSourceList command.";
this->SetError(err.c_str());
return false;
}
// Create the test driver file
fout <<
"#include <ctype.h>\n"
"#include <stdio.h>\n"
"#include <string.h>\n"
"#include <stdlib.h>\n";
if(extraInclude.size())
{
fout << "#include \"" << extraInclude << "\"\n";
}
fout <<
"\n"
"// Forward declare test functions\n"
"\n";
std::vector<std::string>::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 '_'
for(i = testsBegin; i != tests.end(); ++i)
{
if(*i == "EXTRA_INCLUDE")
{
break;
}
2002-07-31 00:19:14 +04:00
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);
2002-07-31 00:19:14 +04:00
fout << "int " << func_name << "(int, char*[]);\n";
}
fout <<
"\n"
"// Create map\n"
"\n"
2002-07-31 00:19:14 +04:00
"typedef int (*MainFuncPointer)(int , char*[]);\n"
2002-07-31 19:07:09 +04:00
"typedef struct\n"
"{\n"
" const char* name;\n"
" MainFuncPointer func;\n"
2002-07-31 19:07:09 +04:00
"} functionMapEntry;\n"
"\n"
"functionMapEntry cmakeGeneratedFunctionMapEntries[] = {\n";
int numTests = 0;
std::vector<std::string>::iterator j;
for(i = testsBegin, j = tests_func_name.begin(); i != tests.end(); ++i, ++j)
{
2002-07-31 00:19:14 +04:00
std::string func_name;
if (cmSystemTools::GetFilenamePath(*i).size() > 0)
{
func_name = cmSystemTools::GetFilenamePath(*i) + "/" +
cmSystemTools::GetFilenameWithoutLastExtension(*i);
}
else
{
func_name = cmSystemTools::GetFilenameWithoutLastExtension(*i);
}
fout <<
" {\n"
2002-07-31 00:19:14 +04:00
" \"" << func_name << "\",\n"
" " << *j << "\n"
" },\n";
numTests++;
}
fout <<
"};\n"
"\n"
"// Allocate and create a lowercased copy of string\n"
"// (note that it has to be free'd manually)\n"
"\n"
"char* lowercase(const char *string)\n"
"{\n"
" char *new_string, *p;\n"
" new_string = (char *)malloc(sizeof(char) * (size_t)(strlen(string) + 1));\n"
" if (!new_string)\n"
" {\n"
" return NULL;\n"
" }\n"
" strcpy(new_string, string);\n"
" p = new_string;\n"
2002-03-30 00:25:39 +03:00
" while (*p != 0)\n"
" {\n"
2002-08-07 18:30:38 +04:00
" *p = (char)tolower(*p);\n"
" ++p;\n"
" }\n"
" return new_string;\n"
"}\n"
"\n"
2002-07-31 00:19:14 +04:00
"int main(int ac, char *av[])\n"
"{\n"
2002-08-07 19:01:06 +04:00
" int i, NumTests, testNum, partial_match;\n"
" char *arg, *test_name;\n"
" \n"
" NumTests = " << numTests << ";\n"
" \n"
" // If no test name was given\n";
if(function.size())
{
fout << " // process command line with user function\n"
<< " " << function << "(&ac, &av);\n";
}
fout <<
" if (ac < 2)\n"
" {\n"
" // Ask for a test\n"
" printf(\"Available tests:\\n\");\n"
" for (i =0; i < NumTests; ++i)\n"
" {\n"
2002-03-27 23:52:52 +03:00
" printf(\"%3d. %s\\n\", i, cmakeGeneratedFunctionMapEntries[i].name);\n"
" }\n"
" printf(\"To run a test, enter the test number: \");\n"
" testNum = 0;\n"
" scanf(\"%d\", &testNum);\n"
" if (testNum >= NumTests)\n"
" {\n"
2002-03-27 23:52:52 +03:00
" printf(\"%3d is an invalid test number.\\n\", testNum);\n"
" return -1;\n"
" }\n"
" return (*cmakeGeneratedFunctionMapEntries[testNum].func)(ac-1, av+1);\n"
" }\n"
" \n"
" // If partial match is requested\n"
" partial_match = (strcmp(av[1], \"-R\") == 0) ? 1 : 0;\n"
" if (partial_match && ac < 3)\n"
" {\n"
" printf(\"-R needs an additional parameter.\\n\");\n"
" return -1;\n"
" }\n"
" \n"
" arg = lowercase(av[1 + partial_match]);\n"
" for (i =0; i < NumTests; ++i)\n"
" {\n"
" test_name = lowercase(cmakeGeneratedFunctionMapEntries[i].name);\n"
" if (partial_match && strstr(test_name, arg) != NULL)\n"
" {\n"
2002-08-08 16:51:40 +04:00
" free(arg);\n"
2002-08-07 18:30:38 +04:00
" free(test_name);\n"
" return (*cmakeGeneratedFunctionMapEntries[i].func)(ac - 2, av + 2);\n"
" }\n"
" else if (!partial_match && strcmp(test_name, arg) == 0)\n"
" {\n"
2002-08-08 16:51:40 +04:00
" free(arg);\n"
2002-08-07 18:30:38 +04:00
" free(test_name);\n"
" return (*cmakeGeneratedFunctionMapEntries[i].func)(ac - 1, av + 1);\n"
" }\n"
" free(test_name);\n"
" }\n"
" free(arg);\n"
" \n"
" // Nothing was run, display the test names\n"
" printf(\"Available tests:\\n\");\n"
" for (i =0; i < NumTests; ++i)\n"
" {\n"
2002-03-27 23:52:52 +03:00
" printf(\"%3d. %s\\n\", i, cmakeGeneratedFunctionMapEntries[i].name);\n"
" }\n"
" printf(\"Failed: %s is an invalid test name.\\n\", av[1]);\n"
" \n"
" return -1;\n"
"}\n";
fout.close();
// Create the source list
cmSourceFile cfile;
std::string sourceListValue;
cfile.SetProperty("ABSTRACT","0");
2002-07-31 00:19:14 +04:00
cfile.SetName(cmSystemTools::GetFilenameWithoutExtension(args[1]).c_str(),
m_Makefile->GetCurrentOutputDirectory(),
2002-07-31 19:07:09 +04:00
cmSystemTools::GetFilenameExtension(args[1]).c_str()+1,
false);
m_Makefile->AddSource(cfile);
2002-07-31 00:19:14 +04:00
sourceListValue = args[1];
for(i = testsBegin; i != tests.end(); ++i)
{
cmSourceFile cfile;
cfile.SetProperty("ABSTRACT","0");
cfile.SetName(i->c_str(),
m_Makefile->GetCurrentDirectory(),
m_Makefile->GetSourceExtensions(),
m_Makefile->GetHeaderExtensions());
m_Makefile->AddSource(cfile);
sourceListValue += ";";
sourceListValue += *i;
}
m_Makefile->AddDefinition(sourceList, sourceListValue.c_str());
return true;
}