00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030 #include <stdint.h>
00031
00032 #include "plugin_manager.h"
00033 #include "plugin.h"
00034
00035 #include <ros/console.h>
00036 #include <ros/package.h>
00037
00038 #include <wx/confbase.h>
00039
00040 namespace rviz
00041 {
00042
00043 PluginManager::PluginManager()
00044 {
00045
00046 }
00047
00048 PluginManager::~PluginManager()
00049 {
00050
00051 }
00052
00053 void PluginManager::loadDescriptions()
00054 {
00055 ros::package::V_string plugins;
00056 ros::package::getPlugins("rviz", "plugin", plugins);
00057
00058 ros::package::V_string::iterator it = plugins.begin();
00059 ros::package::V_string::iterator end = plugins.end();
00060 for (; it != end; ++it)
00061 {
00062 const std::string& plugin = *it;
00063 loadDescription(plugin);
00064 }
00065 }
00066
00067 PluginPtr PluginManager::getPlugin(const std::string& description_file)
00068 {
00069 L_Plugin::iterator it = plugins_.begin();
00070 L_Plugin::iterator end = plugins_.end();
00071 for (; it != end; ++it)
00072 {
00073 const PluginPtr& plugin = *it;
00074
00075 if (plugin->getDescriptionPath() == description_file)
00076 {
00077 return plugin;
00078 }
00079 }
00080
00081 return PluginPtr();
00082 }
00083
00084 PluginPtr PluginManager::getPluginByPackage(const std::string& package)
00085 {
00086 L_Plugin::iterator it = plugins_.begin();
00087 L_Plugin::iterator end = plugins_.end();
00088 for (; it != end; ++it)
00089 {
00090 const PluginPtr& plugin = *it;
00091
00092 if (plugin->getPackageName() == package)
00093 {
00094 return plugin;
00095 }
00096 }
00097
00098 return PluginPtr();
00099 }
00100
00101 PluginPtr PluginManager::getPluginByDisplayName(const std::string& display_name)
00102 {
00103 L_Plugin::iterator it = plugins_.begin();
00104 L_Plugin::iterator end = plugins_.end();
00105 for (; it != end; ++it)
00106 {
00107 const PluginPtr& plugin = *it;
00108
00109 if (plugin->getDisplayTypeInfoByDisplayName(display_name))
00110 {
00111 return plugin;
00112 }
00113 }
00114
00115 return PluginPtr();
00116 }
00117
00118 PluginPtr PluginManager::loadDescription(const std::string& description_path)
00119 {
00120 PluginPtr plugin = getPlugin(description_path);
00121 if (plugin)
00122 {
00123 return plugin;
00124 }
00125
00126 plugin.reset(new Plugin);
00127
00128 try
00129 {
00130 plugin->loadDescription(description_path);
00131 }
00132 catch (std::runtime_error& e)
00133 {
00134 ROS_ERROR("Error loading plugin from description [%s]: %s", description_path.c_str(), e.what());
00135 }
00136
00137 plugins_.push_back(plugin);
00138
00139 return plugin;
00140 }
00141
00142 void PluginManager::loadConfig(const boost::shared_ptr<wxConfigBase>& config)
00143 {
00144 int i = 0;
00145 while (1)
00146 {
00147 wxString package_key, autoload_key;
00148 package_key.Printf( wxT("Plugins/Plugin%d/Package"), i );
00149 autoload_key.Printf( wxT("Plugins/Plugin%d/AutoLoad"), i );
00150
00151 wxString package;
00152 bool autoload;
00153
00154 if (!config->Read(package_key, &package))
00155 {
00156 break;
00157 }
00158
00159 if (!config->Read(autoload_key, &autoload))
00160 {
00161 break;
00162 }
00163
00164 PluginPtr plugin = getPluginByPackage((const char*)package.char_str());
00165 if (plugin)
00166 {
00167 plugin->setAutoLoad(autoload);
00168 plugin->autoLoad();
00169 }
00170
00171 ++i;
00172 }
00173 }
00174
00175 void PluginManager::saveConfig(const boost::shared_ptr<wxConfigBase>& config)
00176 {
00177 int i = 0;
00178 L_Plugin::iterator it = plugins_.begin();
00179 L_Plugin::iterator end = plugins_.end();
00180 for (; it != end; ++it, ++i)
00181 {
00182 const PluginPtr& plugin = *it;
00183
00184 wxString package_key, autoload_key;
00185 package_key.Printf( wxT("Plugins/Plugin%d/Package"), i );
00186 autoload_key.Printf( wxT("Plugins/Plugin%d/AutoLoad"), i );
00187
00188 config->Write(package_key, wxString::FromAscii(plugin->getPackageName().c_str()));
00189 config->Write(autoload_key, plugin->isAutoLoad());
00190 }
00191 }
00192
00193 }