many fixes and cleanup and features

1 files changed, 213 insertions, 0 deletions

diff --git a/Source/cmVTKWrapPythonCommand.cxx b/Source/cmVTKWrapPythonCommand.cxx
new file mode 100644
index 0000000..0a26f8de
--- /dev/null
+++ b/Source/cmVTKWrapPythonCommand.cxx
@@ -0,0 +1,213 @@
+/*=========================================================================
+
+  Program:   Insight Segmentation & Registration Toolkit
+  Module:    $RCSfile$
+  Language:  C++
+  Date:      $Date$
+  Version:   $Revision$
+
+
+  Copyright (c) 2000 National Library of Medicine
+  All rights reserved.
+
+  See COPYRIGHT.txt for copyright details.
+
+=========================================================================*/
+#include "cmVTKWrapPythonCommand.h"
+
+// cmVTKWrapPythonCommand
+bool cmVTKWrapPythonCommand::Invoke(std::vector<std::string>& args)
+{
+  if(args.size() < 3 )
+    {
+    this->SetError("called with incorrect number of arguments");
+    return false;
+    }
+
+  // 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
+  const char* cacheValue
+    = cmCacheManager::GetInstance()->GetCacheValue("VTK_WRAP_PYTHON");
+  if(!cacheValue || !strcmp(cacheValue,"0"))
+    {
+    return true;
+    }
+
+  // add in a depend in the vtkVTKWrapPython executable
+  m_Makefile->AddUtility("vtkWrapPython");
+  
+  // what is the current source dir
+  std::string cdir = m_Makefile->GetCurrentDirectory();
+
+  // keep the library name
+  m_LibraryName = args[0];
+  m_SourceList = args[1];
+  
+  // get the list of classes for this library
+  cmMakefile::SourceMap &Classes = m_Makefile->GetSources();
+  for(std::vector<std::string>::iterator j = (args.begin() + 2);
+      j != args.end(); ++j)
+    {   
+    for(cmMakefile::SourceMap::iterator l = Classes.begin(); 
+        l != Classes.end(); l++)
+      {
+      for(std::vector<cmSourceFile>::iterator i = l->second.begin(); 
+          i != l->second.end(); i++)
+        {
+        cmSourceFile &curr = *i;
+        // if we should wrap the class
+        if (!curr.GetWrapExclude())
+          {
+          cmSourceFile file;
+          file.SetIsAnAbstractClass(curr.IsAnAbstractClass());
+          std::string newName = curr.GetSourceName() + "Python";
+          file.SetName(newName.c_str(), m_Makefile->GetCurrentOutputDirectory(),
+                       "cxx",false);
+          std::string hname = cdir + "/" + curr.GetSourceName() + ".h";
+          m_WrapHeaders.push_back(hname);
+          // add starting depends
+          file.GetDepends().push_back(hname);
+          m_WrapClasses.push_back(file);
+          }
+        }
+      }
+    }
+  
+  return true;
+}
+
+void cmVTKWrapPythonCommand::FinalPass() 
+{
+  // first we add the rules for all the .h to Python.cxx files
+  int lastClass = m_WrapClasses.size();
+  std::vector<std::string> depends;
+  std::string wpython = "${VTK_WRAP_PYTHON_EXE}";
+  std::string hints = "${VTK_WRAP_HINTS}";
+  
+  // Create the init file 
+  std::string res = m_LibraryName;
+  res += "Init.cxx";
+  this->CreateInitFile(res);
+  
+  // add the init file
+  cmSourceFile cfile;
+  cfile.SetIsAnAbstractClass(false);
+  std::string newName = m_LibraryName;
+  newName += "Init";
+  cfile.SetName(newName.c_str(), m_Makefile->GetCurrentOutputDirectory(),
+                "cxx",false);
+  m_Makefile->AddSource(cfile,m_SourceList.c_str());
+  
+  // wrap all the .h files
+  depends.push_back(wpython);
+  for(int classNum = 0; classNum < lastClass; classNum++)
+    {
+    m_Makefile->AddSource(m_WrapClasses[classNum],m_SourceList.c_str());
+    std::string res = m_WrapClasses[classNum].GetSourceName() + ".cxx";
+    std::string cmd = wpython + " " + m_WrapHeaders[classNum] + " "
+      + hints + (m_WrapClasses[classNum].IsAnAbstractClass() ? " 0 " : " 1 ") + " > " + m_WrapClasses[classNum].GetSourceName() + ".cxx";
+    m_Makefile->AddCustomCommand(m_WrapHeaders[classNum].c_str(),
+                                 cmd.c_str(), depends, 
+                                 res.c_str(), m_LibraryName.c_str());
+    }
+  
+}
+
+bool cmVTKWrapPythonCommand::CreateInitFile(std::string& res) 
+{
+  unsigned int i;
+  
+  /* we have to make sure that the name is the correct case */
+  std::string kitName = m_LibraryName;
+  if (kitName[0] > 90) kitName[0] -= 32;
+  for (i = 1; i < kitName.size(); i++)
+    {
+    if ((kitName[i] > 64)&&(kitName[i] < 91))
+      {
+      kitName[i] += 32;
+      }
+    }
+  
+  std::vector<std::string> classes;
+  int lastClass = m_WrapHeaders.size();
+  int classNum;
+  for(classNum = 0; classNum < lastClass; classNum++)
+    {
+    if (!m_WrapClasses[classNum].IsAnAbstractClass())
+      {
+      std::string cls = m_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 = 
+    m_Makefile->GetCurrentOutputDirectory();
+  outFileName += "/" + res;
+  
+  return this->WriteInit(kitName.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;
+  
+  FILE *fout = fopen(outFileName.c_str(),"w");
+  if (!fout)
+    {
+    return false;
+    }
+  
+  fprintf(fout,"#include <string.h>\n");
+  fprintf(fout,"#include \"Python.h\"\n\n");
+
+  for (i = 0; i < classes.size(); i++)
+    {
+    fprintf(fout,"extern  \"C\" {__declspec( dllexport) PyObject *PyVTKClass_%sNew(char *); }\n",classes[i].c_str());
+    }
+
+  fprintf(fout,"\nstatic PyMethodDef Py%s_ClassMethods[] = {\n",
+	  kitName);
+  fprintf(fout,"{NULL, NULL}};\n\n");
+  
+  fprintf(fout,"extern  \"C\" {__declspec( dllexport) void init%s();}\n\n",kitName);
+  
+
+  /* module init function */
+  fprintf(fout,"void init%s()\n{\n",kitName);
+  fprintf(fout,"  PyObject *m, *d, *c;\n\n");
+  fprintf(fout,"  static char modulename[] = \"%s\";\n",kitName);
+  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);
+  
+  
+  return true;
+}
+
+