Source/cmVTKWrapPythonCommand.cxx - third_party/cmake - Git at Google

 /*=========================================================================

   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 "cmVTKWrapPythonCommand.h"

 // cmVTKWrapPythonCommand
 bool cmVTKWrapPythonCommand::InitialPass(std::vector<std::string> const&
                                          argsIn)
 {
   if(argsIn.size() < 3 )
     {
     this->SetError("called with incorrect number of arguments");
     return false;
     }
   std::vector<std::string> args;
   this->Makefile->ExpandSourceListArguments(argsIn, args, 2);

   // Now check and see if the value has been stored in the cache
   // already, if so use that value and don't look for the program
   if(!this->Makefile->IsOn("VTK_WRAP_PYTHON"))
     {
     return true;
     }


   // what is the current source dir
   std::string cdir = this->Makefile->GetCurrentDirectory();

   // keep the library name
   this->LibraryName = args[0];
   this->SourceList = args[1];
   std::string sourceListValue;
   // was the list already populated
   const char *def = this->Makefile->GetDefinition(this->SourceList.c_str());
   if (def)
     {
     sourceListValue = def;
     sourceListValue += ";";
     }

   // Create the init file
   std::string res = this->LibraryName;
   res += "Init.cxx";

   // add the init file
   std::string initName = this->LibraryName;
   initName += "Init";
   sourceListValue += initName + ".cxx";

   // get the list of classes for this library
   for(std::vector<std::string>::iterator j = (args.begin() + 2);
       j != args.end(); ++j)
     {
     cmSourceFile *curr = this->Makefile->GetSource(j->c_str());

     // if we should wrap the class
     if (!curr || !curr->GetPropertyAsBool("WRAP_EXCLUDE"))
       {
       cmSourceFile file;
       if (curr)
         {
         file.SetProperty("ABSTRACT",curr->GetProperty("ABSTRACT"));
         }
       std::string srcName = cmSystemTools::GetFilenameWithoutExtension(*j);
       std::string newName = srcName + "Python";
       file.SetName(newName.c_str(),
                    this->Makefile->GetCurrentOutputDirectory(),
                    "cxx",false);
       std::string hname = cdir + "/" + srcName + ".h";
       this->WrapHeaders.push_back(hname);
       // add starting depends
       file.GetDepends().push_back(hname);
       this->WrapClasses.push_back(file);
       sourceListValue += ";";
       sourceListValue += newName + ".cxx";
       }
     }

   cmSourceFile cfile;
   cfile.SetProperty("ABSTRACT","0");
   this->CreateInitFile(res);
   cfile.SetName(initName.c_str(), this->Makefile->GetCurrentOutputDirectory(),
                 "cxx",false);
   this->Makefile->AddSource(cfile);
   this->Makefile->AddDefinition(this->SourceList.c_str(),
                                 sourceListValue.c_str());
   return true;
 }

 void cmVTKWrapPythonCommand::FinalPass()
 {
   // first we add the rules for all the .h to Python.cxx files
   size_t lastClass = this->WrapClasses.size();
   std::vector<std::string> depends;
   const char* wpython =
     this->Makefile->GetRequiredDefinition("VTK_WRAP_PYTHON_EXE");
   const char* hints = this->Makefile->GetDefinition("VTK_WRAP_HINTS");

   // wrap all the .h files
   depends.push_back(wpython);
   if(hints)
     {
     depends.push_back(hints);
     }
   for(size_t classNum = 0; classNum < lastClass; classNum++)
     {
     this->Makefile->AddSource(this->WrapClasses[classNum]);
     std::string res = this->Makefile->GetCurrentOutputDirectory();
     res += "/";
     res += this->WrapClasses[classNum].GetSourceName() + ".cxx";
     cmCustomCommandLine commandLine;
     commandLine.push_back(wpython);
     commandLine.push_back(this->WrapHeaders[classNum]);
     if(hints)
       {
       commandLine.push_back(hints);
       }
     commandLine.push_back((this->WrapClasses[classNum].
                            GetPropertyAsBool("ABSTRACT") ? "0" : "1"));
     commandLine.push_back(res);

     cmCustomCommandLines commandLines;
     commandLines.push_back(commandLine);
     std::vector<std::string> outputs;
     outputs.push_back(res);
     const char* no_comment = 0;
     this->Makefile->AddCustomCommandOldStyle(this->LibraryName.c_str(),
                                          outputs,
                                          depends,
                                          this->WrapHeaders[classNum].c_str(),
                                          commandLines,
                                          no_comment);
     }
 }

 bool cmVTKWrapPythonCommand::CreateInitFile(std::string& res)
 {
   std::vector<std::string> classes;
   size_t lastClass = this->WrapHeaders.size();
   size_t classNum;
   for(classNum = 0; classNum < lastClass; classNum++)
     {
     std::string cls = this->WrapHeaders[classNum];
     cls = cls.substr(0,cls.size()-2);
     std::string::size_type pos = cls.rfind('/');
     if(pos != std::string::npos)
       {
       cls = cls.substr(pos+1);
       }
     classes.push_back(cls);
     }

   // open the init file
   std::string outFileName =
     this->Makefile->GetCurrentOutputDirectory();
   outFileName += "/" + res;

   return this->WriteInit(this->LibraryName.c_str(), outFileName, classes);
 }


 /* warning this code is also in getclasses.cxx under pcmaker */
 bool cmVTKWrapPythonCommand::WriteInit(const char *kitName,
                                        std::string& outFileName,
                                        std::vector<std::string>& classes)
 {
   unsigned int i;

   std::string tempOutputFile = outFileName + ".tmp";
   FILE *fout = fopen(tempOutputFile.c_str(),"w");
   if (!fout)
     {
     cmSystemTools::ReportLastSystemError("cmVTKWrapPythonCommand error:");
     return false;
     }

   fprintf(fout,"// Generated by cmVTKWrapPythonCommand in CMake\n\n");
   fprintf(fout,"#include <string.h>\n");
   fprintf(fout,"#include \"Python.h\"\n\n");
   fprintf(fout,"// Handle compiler warning messages, etc.\n"
           "#if defined( _MSC_VER ) && !defined(VTK_DISPLAY_WIN32_WARNINGS)\n"
           "#pragma warning ( disable : 4706 )\n"
           "#endif // Windows Warnings \n\n");

   for (i = 0; i < classes.size(); i++)
     {
 #ifdef _WIN32
     fprintf(fout, "extern  \"C\" {__declspec( dllexport) "
             "PyObject *PyVTKClass_%sNew(char *); }\n", classes[i].c_str());
 #else
     fprintf(fout,"extern  \"C\" {PyObject *PyVTKClass_%sNew(char *); }\n",
             classes[i].c_str());
 #endif
     }

   fprintf(fout,"\nstatic PyMethodDef Py%s_ClassMethods[] = {\n",
           kitName);
   fprintf(fout,"{NULL, NULL, 0, NULL}};\n\n");

 #ifdef _WIN32
   fprintf(fout,"extern  \"C\" {__declspec( dllexport) void init%s();}\n\n",
           kitName);
   fprintf(fout,"void init%s()\n{\n",kitName);
 #else
   fprintf(fout,"extern  \"C\" {void initlib%s();}\n\n",kitName);
   fprintf(fout,"void initlib%s()\n{\n",kitName);
 #endif


   /* module init function */
   fprintf(fout,"  PyObject *m, *d, *c;\n\n");
 #ifdef _WIN32
   fprintf(fout,"  static char modulename[] = \"%s\";\n",kitName);
 #else
   fprintf(fout,"  static char modulename[] = \"lib%s\";\n",kitName);
 #endif
   fprintf(fout,"  m = Py_InitModule(modulename, Py%s_ClassMethods);\n",
           kitName);

   fprintf(fout,"  d = PyModule_GetDict(m);\n");
   fprintf(fout,"  if (!d) Py_FatalError"
           "(\"can't get dictionary for module %s!\");\n\n", kitName);

   for (i = 0; i < classes.size(); i++)
     {
     fprintf(fout,"  if ((c = PyVTKClass_%sNew(modulename)))\n",
             classes[i].c_str());
     fprintf(fout,"    if (-1 == PyDict_SetItemString(d, \"%s\", c))\n",
             classes[i].c_str());
     fprintf(fout,
             "      Py_FatalError(\"can't add class %s to dictionary!\");\n\n",
             classes[i].c_str());
     }
   fprintf(fout,"}\n\n");
   fclose(fout);


   // copy the file if different
   cmSystemTools::CopyFileIfDifferent(tempOutputFile.c_str(),
                                      outFileName.c_str());
   cmSystemTools::RemoveFile(tempOutputFile.c_str());
   return true;
 }
	/*=========================================================================

	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 "cmVTKWrapPythonCommand.h"

	// cmVTKWrapPythonCommand
	bool cmVTKWrapPythonCommand::InitialPass(std::vector<std::string> const&
	argsIn)
	{
	if(argsIn.size() < 3 )
	{
	this->SetError("called with incorrect number of arguments");
	return false;
	}
	std::vector<std::string> args;
	this->Makefile->ExpandSourceListArguments(argsIn, args, 2);

	// Now check and see if the value has been stored in the cache
	// already, if so use that value and don't look for the program
	if(!this->Makefile->IsOn("VTK_WRAP_PYTHON"))
	{
	return true;
	}


	// what is the current source dir
	std::string cdir = this->Makefile->GetCurrentDirectory();

	// keep the library name
	this->LibraryName = args[0];
	this->SourceList = args[1];
	std::string sourceListValue;
	// was the list already populated
	const char *def = this->Makefile->GetDefinition(this->SourceList.c_str());
	if (def)
	{
	sourceListValue = def;
	sourceListValue += ";";
	}

	// Create the init file
	std::string res = this->LibraryName;
	res += "Init.cxx";

	// add the init file
	std::string initName = this->LibraryName;
	initName += "Init";
	sourceListValue += initName + ".cxx";

	// get the list of classes for this library
	for(std::vector<std::string>::iterator j = (args.begin() + 2);
	j != args.end(); ++j)
	{
	cmSourceFile *curr = this->Makefile->GetSource(j->c_str());

	// if we should wrap the class
	if (!curr \|\| !curr->GetPropertyAsBool("WRAP_EXCLUDE"))
	{
	cmSourceFile file;
	if (curr)
	{
	file.SetProperty("ABSTRACT",curr->GetProperty("ABSTRACT"));
	}
	std::string srcName = cmSystemTools::GetFilenameWithoutExtension(*j);
	std::string newName = srcName + "Python";
	file.SetName(newName.c_str(),
	this->Makefile->GetCurrentOutputDirectory(),
	"cxx",false);
	std::string hname = cdir + "/" + srcName + ".h";
	this->WrapHeaders.push_back(hname);
	// add starting depends
	file.GetDepends().push_back(hname);
	this->WrapClasses.push_back(file);
	sourceListValue += ";";
	sourceListValue += newName + ".cxx";
	}
	}

	cmSourceFile cfile;
	cfile.SetProperty("ABSTRACT","0");
	this->CreateInitFile(res);
	cfile.SetName(initName.c_str(), this->Makefile->GetCurrentOutputDirectory(),
	"cxx",false);
	this->Makefile->AddSource(cfile);
	this->Makefile->AddDefinition(this->SourceList.c_str(),
	sourceListValue.c_str());
	return true;
	}

	void cmVTKWrapPythonCommand::FinalPass()
	{
	// first we add the rules for all the .h to Python.cxx files
	size_t lastClass = this->WrapClasses.size();
	std::vector<std::string> depends;
	const char* wpython =
	this->Makefile->GetRequiredDefinition("VTK_WRAP_PYTHON_EXE");
	const char* hints = this->Makefile->GetDefinition("VTK_WRAP_HINTS");

	// wrap all the .h files
	depends.push_back(wpython);
	if(hints)
	{
	depends.push_back(hints);
	}
	for(size_t classNum = 0; classNum < lastClass; classNum++)
	{
	this->Makefile->AddSource(this->WrapClasses[classNum]);
	std::string res = this->Makefile->GetCurrentOutputDirectory();
	res += "/";
	res += this->WrapClasses[classNum].GetSourceName() + ".cxx";
	cmCustomCommandLine commandLine;
	commandLine.push_back(wpython);
	commandLine.push_back(this->WrapHeaders[classNum]);
	if(hints)
	{
	commandLine.push_back(hints);
	}
	commandLine.push_back((this->WrapClasses[classNum].
	GetPropertyAsBool("ABSTRACT") ? "0" : "1"));
	commandLine.push_back(res);

	cmCustomCommandLines commandLines;
	commandLines.push_back(commandLine);
	std::vector<std::string> outputs;
	outputs.push_back(res);
	const char* no_comment = 0;
	this->Makefile->AddCustomCommandOldStyle(this->LibraryName.c_str(),
	outputs,
	depends,
	this->WrapHeaders[classNum].c_str(),
	commandLines,
	no_comment);
	}
	}

	bool cmVTKWrapPythonCommand::CreateInitFile(std::string& res)
	{
	std::vector<std::string> classes;
	size_t lastClass = this->WrapHeaders.size();
	size_t classNum;
	for(classNum = 0; classNum < lastClass; classNum++)
	{
	std::string cls = this->WrapHeaders[classNum];
	cls = cls.substr(0,cls.size()-2);
	std::string::size_type pos = cls.rfind('/');
	if(pos != std::string::npos)
	{
	cls = cls.substr(pos+1);
	}
	classes.push_back(cls);
	}

	// open the init file
	std::string outFileName =
	this->Makefile->GetCurrentOutputDirectory();
	outFileName += "/" + res;

	return this->WriteInit(this->LibraryName.c_str(), outFileName, classes);
	}


	/* warning this code is also in getclasses.cxx under pcmaker */
	bool cmVTKWrapPythonCommand::WriteInit(const char *kitName,
	std::string& outFileName,
	std::vector<std::string>& classes)
	{
	unsigned int i;

	std::string tempOutputFile = outFileName + ".tmp";
	FILE *fout = fopen(tempOutputFile.c_str(),"w");
	if (!fout)
	{
	cmSystemTools::ReportLastSystemError("cmVTKWrapPythonCommand error:");
	return false;
	}

	fprintf(fout,"// Generated by cmVTKWrapPythonCommand in CMake\n\n");
	fprintf(fout,"#include <string.h>\n");
	fprintf(fout,"#include \"Python.h\"\n\n");
	fprintf(fout,"// Handle compiler warning messages, etc.\n"
	"#if defined( _MSC_VER ) && !defined(VTK_DISPLAY_WIN32_WARNINGS)\n"
	"#pragma warning ( disable : 4706 )\n"
	"#endif // Windows Warnings \n\n");

	for (i = 0; i < classes.size(); i++)
	{
	#ifdef _WIN32
	fprintf(fout, "extern \"C\" {__declspec( dllexport) "
	"PyObject PyVTKClass_%sNew(char ); }\n", classes[i].c_str());
	#else
	fprintf(fout,"extern \"C\" {PyObject PyVTKClass_%sNew(char ); }\n",
	classes[i].c_str());
	#endif
	}

	fprintf(fout,"\nstatic PyMethodDef Py%s_ClassMethods[] = {\n",
	kitName);
	fprintf(fout,"{NULL, NULL, 0, NULL}};\n\n");

	#ifdef _WIN32
	fprintf(fout,"extern \"C\" {__declspec( dllexport) void init%s();}\n\n",
	kitName);
	fprintf(fout,"void init%s()\n{\n",kitName);
	#else
	fprintf(fout,"extern \"C\" {void initlib%s();}\n\n",kitName);
	fprintf(fout,"void initlib%s()\n{\n",kitName);
	#endif


	/* module init function */
	fprintf(fout," PyObject m, d, *c;\n\n");
	#ifdef _WIN32
	fprintf(fout," static char modulename[] = \"%s\";\n",kitName);
	#else
	fprintf(fout," static char modulename[] = \"lib%s\";\n",kitName);
	#endif
	fprintf(fout," m = Py_InitModule(modulename, Py%s_ClassMethods);\n",
	kitName);

	fprintf(fout," d = PyModule_GetDict(m);\n");
	fprintf(fout," if (!d) Py_FatalError"
	"(\"can't get dictionary for module %s!\");\n\n", kitName);

	for (i = 0; i < classes.size(); i++)
	{
	fprintf(fout," if ((c = PyVTKClass_%sNew(modulename)))\n",
	classes[i].c_str());
	fprintf(fout," if (-1 == PyDict_SetItemString(d, \"%s\", c))\n",
	classes[i].c_str());
	fprintf(fout,
	" Py_FatalError(\"can't add class %s to dictionary!\");\n\n",
	classes[i].c_str());
	}
	fprintf(fout,"}\n\n");
	fclose(fout);


	// copy the file if different
	cmSystemTools::CopyFileIfDifferent(tempOutputFile.c_str(),
	outFileName.c_str());
	cmSystemTools::RemoveFile(tempOutputFile.c_str());
	return true;
	}