+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
/*
* Config_ModuleReader.cpp
*
* Author: sbh
*/
+#include <pyconfig.h>
+
#include <Config_Keywords.h>
+#include <Config_Common.h>
#include <Config_ModuleReader.h>
#include <Config_FeatureReader.h>
+#include <Events_InfoMessage.h>
#include <libxml/parser.h>
#include <libxml/tree.h>
-#ifdef _DEBUG
+// Have to be included before std headers
+#include <Python.h>
+
+//Necessary for cerr
#include <iostream>
-#endif
+#include <algorithm>
+#ifdef WIN32
+#include <windows.h>
+#pragma warning(disable : 4996) // for getenv
+#else
+#include <dlfcn.h>
+#endif
+std::map<std::string, Config_ModuleReader::PluginType> Config_ModuleReader::myPluginTypes;
+std::set<std::string> Config_ModuleReader::myDependencyModules;
Config_ModuleReader::Config_ModuleReader(const char* theEventGenerated)
- : Config_XMLReader("plugins.xml"), myIsAutoImport(false), myEventGenerated(theEventGenerated)
+ : Config_XMLReader(PLUGIN_FILE),
+ myEventGenerated(theEventGenerated)
{
}
{
}
-/*
+const std::map<std::string, std::string>& Config_ModuleReader::featuresInFiles() const
+{
+ return myFeaturesInFiles;
+}
+
+const std::set<std::string>& Config_ModuleReader::modulePluginFiles() const
+{
+ return myPluginFiles;
+}
+
+/*!
* Get module name from plugins.xml
* (property "module")
*/
return getProperty(aRoot, PLUGINS_MODULE);
}
-/*
- *
- */
+
+void Config_ModuleReader::addFeature(const std::string& theFeatureName,
+ const std::string& thePluginConfig)
+{
+ if (myFeaturesInFiles.count(theFeatureName)) {
+ std::string anErrorMsg = "Can not register feature '%1' in plugin '%2'."
+ " There is a feature with the same ID.";
+ Events_InfoMessage("Config_ModuleReader", anErrorMsg)
+ .arg(theFeatureName).arg(thePluginConfig).send();
+ return;
+ }
+
+ myFeaturesInFiles[theFeatureName] = thePluginConfig;
+}
+
void Config_ModuleReader::processNode(xmlNodePtr theNode)
{
if (isNode(theNode, NODE_PLUGIN, NULL)) {
+ if (!hasRequiredModules(theNode))
+ return;
std::string aPluginConf = getProperty(theNode, PLUGIN_CONFIG);
+ if (!aPluginConf.empty())
+ myPluginFiles.insert(aPluginConf);
std::string aPluginLibrary = getProperty(theNode, PLUGIN_LIBRARY);
- if (myIsAutoImport)
- importPlugin(aPluginConf, aPluginLibrary);
- myPluginsMap[aPluginLibrary] = aPluginConf;
+ std::string aPluginScript = getProperty(theNode, PLUGIN_SCRIPT);
+ std::string aPluginName = addPlugin(aPluginLibrary, aPluginScript, aPluginConf);
+
+ std::list<std::string> aFeatures = importPlugin(aPluginName, aPluginConf);
+ std::list<std::string>::iterator it = aFeatures.begin();
+ for (; it != aFeatures.end(); it++) {
+ addFeature(*it, aPluginConf);
+ }
}
}
return isNode(theNode, NODE_PLUGINS, NULL);
}
-void Config_ModuleReader::importPlugin(const std::string& thePluginName,
- const std::string& thePluginLibrary)
+bool Config_ModuleReader::hasRequiredModules(xmlNodePtr theNode) const
+{
+ std::string aRequiredModule = normalize(getProperty(theNode, PLUGIN_DEPENDENCY));
+ if(aRequiredModule.empty())
+ return true;
+ std::set<std::string>::iterator it = myDependencyModules.begin();
+ for ( ; it != myDependencyModules.end(); it++ ) {
+ if (*it == aRequiredModule) return true;
+ }
+ return false;
+}
+
+std::list<std::string> Config_ModuleReader::importPlugin(const std::string& thePluginLibrary,
+ const std::string& thePluginXmlConf)
{
- Config_FeatureReader* aReader;
- if(thePluginLibrary.empty()) {
- aReader = new Config_FeatureReader(thePluginName);
- } else {
- aReader = new Config_FeatureReader(thePluginName, thePluginLibrary, myEventGenerated);
+ if (thePluginXmlConf.empty()) { //probably a third party library
+ loadLibrary(thePluginLibrary);
+ return std::list<std::string>();
}
- aReader->readAll();
+
+ Config_FeatureReader aReader = Config_FeatureReader(thePluginXmlConf,
+ thePluginLibrary,
+ myEventGenerated);
+ aReader.readAll();
+ return aReader.features();
}
-void Config_ModuleReader::setAutoImport(bool theEnabled)
+std::string Config_ModuleReader::addPlugin(const std::string& aPluginLibrary,
+ const std::string& aPluginScript,
+ const std::string& aPluginConf)
{
- myIsAutoImport = theEnabled;
+ PluginType aType = Config_ModuleReader::Binary;
+ std::string aPluginName;
+ if (!aPluginLibrary.empty()) {
+ aPluginName = aPluginLibrary;
+ if (aPluginConf.empty()) {
+ aType = Config_ModuleReader::Intrenal;
+ }
+ } else if (!aPluginScript.empty()) {
+ aPluginName = aPluginScript;
+ aType = Config_ModuleReader::Python;
+ }
+ if(!aPluginName.empty()) {
+ myPluginTypes[aPluginName] = aType;
+ }
+ addDependencyModule(aPluginName);
+ return aPluginName;
}
-const std::map<std::string, std::string>& Config_ModuleReader::plugins() const
+void Config_ModuleReader::loadPlugin(const std::string& thePluginName)
{
- return myPluginsMap;
+ PluginType aType = Config_ModuleReader::Binary;
+ if(myPluginTypes.find(thePluginName) != myPluginTypes.end()) {
+ aType = myPluginTypes.at(thePluginName);
+ }
+ switch (aType) {
+ case Config_ModuleReader::Python:
+ loadScript(thePluginName);
+ break;
+ case Config_ModuleReader::Binary:
+ case Config_ModuleReader::Intrenal:
+ default:
+ loadLibrary(thePluginName);
+ break;
+ }
}
+
+void Config_ModuleReader::loadScript(const std::string& theFileName, bool theSendErr)
+{
+ /* acquire python thread */
+ PyGILState_STATE gstate = PyGILState_Ensure();
+
+ PyObject* module = PyImport_ImportModule(theFileName.c_str());
+ if (!module) {
+ std::string anErrorMsg = "An error occurred while importing " + theFileName;
+ //Get detailed error message:
+ if (PyErr_Occurred()) {
+ PyObject *pstr, *ptype, *pvalue, *ptraceback;
+ PyErr_Fetch(&ptype, &pvalue, &ptraceback);
+ PyErr_NormalizeException(&ptype, &pvalue, &ptraceback);
+ pstr = PyObject_Str(pvalue);
+ std::string aPyError = std::string(PyString_AsString(pstr));
+ if (!aPyError.empty()) {
+ anErrorMsg += ":\n" + aPyError;
+ }
+ Py_XDECREF(pstr);
+ Py_XDECREF(ptype);
+ Py_XDECREF(pvalue);
+ Py_XDECREF(ptraceback);
+ }
+ if (theSendErr)
+ Events_InfoMessage("Config_ModuleReader", anErrorMsg).send();
+ }
+
+ /* release python thread */
+ PyGILState_Release(gstate);
+}
+
+void Config_ModuleReader::loadLibrary(const std::string& theLibName)
+{
+ std::string aFileName = library(theLibName);
+ if (aFileName.empty())
+ return;
+
+ #ifdef WIN32
+ HINSTANCE aModLib = ::LoadLibrary(aFileName.c_str());
+ #else
+ void* aModLib = dlopen( aFileName.c_str(), RTLD_LAZY | RTLD_GLOBAL );
+ #endif
+ if(!aModLib && theLibName != "DFBrowser") { // don't show error for internal debugging tool
+ std::string anErrorMsg = "Failed to load " + aFileName;
+ #ifdef WIN32
+ DWORD dwLastError = ::GetLastError();
+ LPSTR messageBuffer = NULL;
+ size_t size = ::FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER |
+ FORMAT_MESSAGE_FROM_SYSTEM |
+ FORMAT_MESSAGE_IGNORE_INSERTS,
+ NULL,
+ dwLastError,
+ MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
+ (LPSTR)&messageBuffer, 0, NULL);
+ anErrorMsg += ": " + std::string(messageBuffer, size);
+ #else
+ anErrorMsg += ": " + std::string(dlerror());
+ #endif
+ std::cerr << anErrorMsg << std::endl;
+ Events_InfoMessage("Config_ModuleReader", anErrorMsg).send();
+ }
+}
+
+void Config_ModuleReader::addDependencyModule(const std::string& theModuleName)
+{
+ myDependencyModules.insert(normalize(theModuleName));
+}
+
