- Use strictier naming scheme for pyloader

git-svn-id: svn://svn.gna.org/svn/sgpemv2/trunk@812 3ecf2c5c-341e-0410-92b4-d18e462d057c

plugins/pyloader/src/python_cpu_policy_manager.cc

@@ -0,0 +1,143 @@
+// src/python_cpu_policy_manager.cc - Copyright 2005, 2006, University
+//               of Padova, dept. of Pure and Applied
+//               Mathematics
+//
+// This file is part of SGPEMv2.
+//
+// This is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation; either version 2 of the License, or
+// (at your option) any later version.
+//
+// SGPEMv2 is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with SGPEMv2; if not, write to the Free Software
+// Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+
+#include "python_cpu_policy_manager.hh"
+#include "global_preferences.hh"
+#include "cpu_policies_gatekeeper.hh"
+
+#include <Python.h>
+#include <glibmm/ustring.h>
+#include <glibmm/timer.h>
+#include <glibmm/fileutils.h>
+#include <glibmm/pattern.h>
+
+#include <algorithm>
+#include <cassert>
+#include <functional>
+#include <iostream>
+#include <string>
+#include <iostream>
+#include <unistd.h>
+
+using namespace sgpem;
+using std::vector;
+using std::cout;
+using std::endl;
+
+
+// Concatenate a string with all the policies directories
+struct pol_dirs_concat : public std::unary_function<void, const Glib::ustring&>
+{
+public:
+  pol_dirs_concat(Glib::ustring& cat) : _cat(cat) {}
+  void operator()(const Glib::ustring& add)
+  {
+    // Please note that this string will end finishing with
+    // and additional ","!
+    _cat += "'" + add + "', ";
+  }
+private:
+  Glib::ustring& _cat;
+};
+
+
+PythonCPUPolicyManager::PythonCPUPolicyManager()
+    : _initialized(false)
+{
+  PyEval_InitThreads();
+}
+
+
+PythonCPUPolicyManager::~PythonCPUPolicyManager()
+{
+  for_each(_policies.begin(), _policies.end(), ptr_fun(operator delete));
+}
+
+
+void
+PythonCPUPolicyManager::init()
+{
+  if(_initialized)
+    // No-op
+    return;
+
+  Py_Initialize();
+  _initialized = true;
+
+  // The following lines are ugly, but necessary if we use
+  // non-standard installation directories. Theoretically,
+  // it should be up to the user to set correct
+  // environment variables.
+
+  GlobalPreferences& prefs = GlobalPreferences::get_instance();
+  Glib::ustring importdirs = "import sys\n"
+                             "sys.path[:0] = [ ";
+  for_each(prefs.policies_dir_begin(),
+           prefs.policies_dir_end(),
+           pol_dirs_concat(importdirs));
+  importdirs += " '" SHAREDIR "' ]\n";
+
+  PyRun_SimpleString(importdirs.c_str());
+
+  // Okay, here we go.
+  // Black magic at work.
+  collect_policies();
+}
+
+const vector<CPUPolicy*>&
+PythonCPUPolicyManager::get_avail_policies()
+{
+  return _policies;
+}
+
+void
+PythonCPUPolicyManager::collect_policies()
+{
+  GlobalPreferences& prefs = GlobalPreferences::get_instance();
+  GlobalPreferences::dir_iterator dir_it = prefs.policies_dir_begin();
+  GlobalPreferences::dir_iterator dir_end = prefs.policies_dir_end();
+
+  for(; dir_it != dir_end; ++dir_it)
+  {
+    Glib::Dir dir(dir_it->c_str());
+
+    cout << "Opening directory " << *dir_it << "..." << endl;
+
+    for(Glib::DirIterator file_it = dir.begin(); file_it != dir.end(); ++file_it)
+    {
+      cout << "\tChecking if " << *file_it << " is a Python script... " << endl;
+
+      Glib::PatternSpec dot_py("*.py");
+
+      if(dot_py.match(*file_it))
+      {
+        cout << "\t\tIt is.\n";
+
+        //strip extension
+        std::string policy_name = (*file_it).substr(0, (*file_it).size() - 3);
+
+        PythonCPUPolicy *pypolicy = new PythonCPUPolicy(policy_name.c_str());
+
+        _policies.push_back(pypolicy);
+      }
+    }
+  }
+}
+