Changeset 8079 for code/trunk/src/libraries/tools/Shader.cc
- Timestamp:
- Mar 15, 2011, 9:47:11 PM (13 years ago)
- Location:
- code/trunk
- Files:
-
- 2 edited
Legend:
- Unmodified
- Added
- Removed
-
code/trunk
- Property svn:mergeinfo changed
-
code/trunk/src/libraries/tools/Shader.cc
r6417 r8079 30 30 31 31 #include <OgreCompositorManager.h> 32 #include <OgreCompositorInstance.h>33 #include <OgreSceneManager.h>34 32 #include <OgreRoot.h> 35 33 #include <OgrePlugin.h> 36 #include <OgreMaterial.h>37 #include <OgreTechnique.h>38 #include <OgrePass.h>39 #include <OgreMaterialManager.h>40 34 41 35 #include "core/CoreIncludes.h" … … 45 39 namespace orxonox 46 40 { 47 bool Shader::bLoadedCgPlugin_s = false;48 Shader::MaterialMap Shader::parameters_s;49 41 /** 42 @brief Initializes the values and sets the scene manager. 43 */ 50 44 Shader::Shader(Ogre::SceneManager* scenemanager) : compositorInstance_(0) 51 45 { … … 53 47 54 48 this->scenemanager_ = scenemanager; 55 this->compositorInstance_ = 0;56 49 this->bVisible_ = true; 57 50 this->bLoadCompositor_ = GameMode::showsGraphics(); 58 this->bViewportInitialized_ = false; 59 60 if (this->bLoadCompositor_ && Ogre::Root::getSingletonPtr()) 61 { 62 Shader::bLoadedCgPlugin_s = false; 51 this->registeredAsListener_ = false; 52 53 static bool hasCgProgramManager = Shader::hasCgProgramManager(); 54 55 this->bLoadCompositor_ &= hasCgProgramManager; 56 } 57 58 /** 59 @brief Removes the compositor and frees the resources. 60 */ 61 Shader::~Shader() 62 { 63 if (this->compositorInstance_ && GraphicsManager::getInstance().getViewport()) 64 Ogre::CompositorManager::getSingleton().removeCompositor(GraphicsManager::getInstance().getViewport(), this->compositorName_); 65 } 66 67 /** 68 @brief Inherited from ViewportEventListener - called if the camera changes. 69 70 Since the new camera could be in a different scene, the shader has to make sure 71 it deactivates or activates itself accordingly. 72 73 Additionally the shader has to be turned off and on even if the camera stays in 74 the same scene to fix a weird behavior of Ogre. 75 */ 76 void Shader::cameraChanged(Ogre::Viewport* viewport, Ogre::Camera* oldCamera) 77 { 78 if (!this->bLoadCompositor_ || !this->scenemanager_) 79 return; 80 81 // load the compositor if not already done 82 if (!this->compositorName_.empty() && !this->compositorInstance_) 83 this->changedCompositorName(viewport); 84 85 // update compositor in viewport (shader should only be active if the current camera is in the same scene as the shader) 86 87 // Note: 88 // The shader needs also to be switched off and on after changing the camera in the 89 // same scene to avoid weird behaviour with active compositors while switching the 90 // camera (like freezing the image) 91 // 92 // Last known Ogre version needing this workaround: 93 // 1.4.8 94 // 1.7.2 95 96 if (oldCamera && this->scenemanager_ == oldCamera->getSceneManager()) 97 Ogre::CompositorManager::getSingleton().setCompositorEnabled(viewport, this->compositorName_, false); 98 99 if (viewport->getCamera() && this->scenemanager_ == viewport->getCamera()->getSceneManager()) 100 Ogre::CompositorManager::getSingleton().setCompositorEnabled(viewport, this->compositorName_, this->isVisible()); 101 } 102 103 /** 104 @brief Changes the compositor - default viewport. 105 */ 106 void Shader::changedCompositorName() 107 { 108 // For the moment, we get the viewport always from the graphics manager 109 // TODO: Try to support multiple viewports - note however that scenemanager_->getCurrentViewport() returns NULL 110 // after switching to a camera in a different scene (only for the first time this scene is displayed though) 111 this->changedCompositorName(GraphicsManager::getInstance().getViewport()); 112 } 113 114 /** 115 @brief Changes the compositor. 116 */ 117 void Shader::changedCompositorName(Ogre::Viewport* viewport) 118 { 119 if (this->bLoadCompositor_) 120 { 121 assert(viewport); 122 if (this->compositorInstance_) 123 { 124 // remove the old compositor, remove the listener 125 Ogre::CompositorManager::getSingleton().removeCompositor(viewport, this->oldcompositorName_); 126 this->compositorInstance_->removeListener(this); 127 this->compositorInstance_ = 0; 128 } 129 if (!this->compositorName_.empty()) 130 { 131 // add the new compositor 132 this->compositorInstance_ = Ogre::CompositorManager::getSingleton().addCompositor(viewport, this->compositorName_); 133 if (this->compositorInstance_) 134 { 135 // register as listener if required 136 if (this->registeredAsListener_) 137 this->compositorInstance_->addListener(this); 138 // set visibility according to the isVisible() and the camera/viewport 139 if (viewport->getCamera()) 140 Ogre::CompositorManager::getSingleton().setCompositorEnabled(viewport, this->compositorName_, this->isVisible() && viewport->getCamera() && this->scenemanager_ == viewport->getCamera()->getSceneManager()); 141 } 142 else 143 COUT(2) << "Warning: Couldn't load compositor with name \"" << this->compositorName_ << "\"." << std::endl; 144 } 145 this->oldcompositorName_ = this->compositorName_; 146 } 147 } 148 149 /** 150 @brief Changes the visibility of the shader. Doesn't free any resources if set to invisible. 151 */ 152 void Shader::updateVisibility() 153 { 154 if (this->compositorInstance_) 155 Ogre::CompositorManager::getSingleton().setCompositorEnabled(GraphicsManager::getInstance().getViewport(), this->compositorName_, this->isVisible()); 156 } 157 158 /** 159 @brief Defines a new integer value for a given parameter. The parameter will be updated if the compositor is rendered the next time. 160 */ 161 void Shader::setParameter(size_t technique, size_t pass, const std::string& parameter, int value) 162 { 163 ParameterContainer container = {technique, pass, parameter, value, 0.0f, MT_Type::Int}; 164 this->parameters_.push_back(container); 165 this->addAsListener(); 166 } 167 168 /** 169 @brief Defines a new float value for a given parameter. The parameter will be updated if the compositor is rendered the next time. 170 */ 171 void Shader::setParameter(size_t technique, size_t pass, const std::string& parameter, float value) 172 { 173 ParameterContainer container = {technique, pass, parameter, 0, value, MT_Type::Float}; 174 this->parameters_.push_back(container); 175 this->addAsListener(); 176 } 177 178 /** 179 @brief Registers the shader as CompositorInstance::Listener at the compositor. Used to change parameters. 180 */ 181 void Shader::addAsListener() 182 { 183 if (!this->registeredAsListener_) 184 { 185 this->registeredAsListener_ = true; 186 if (this->compositorInstance_) 187 this->compositorInstance_->addListener(this); 188 } 189 } 190 191 /** 192 @brief Inherited by Ogre::CompositorInstance::Listener, called whenever the material is rendered. Used to change parameters. 193 */ 194 void Shader::notifyMaterialRender(Ogre::uint32 pass_id, Ogre::MaterialPtr& materialPtr) 195 { 196 // iterate through the list of parameters 197 for (std::list<ParameterContainer>::iterator it = this->parameters_.begin(); it != this->parameters_.end(); ++it) 198 { 199 Ogre::Technique* techniquePtr = materialPtr->getTechnique(it->technique_); 200 if (techniquePtr) 201 { 202 Ogre::Pass* passPtr = techniquePtr->getPass(it->pass_); 203 if (passPtr) 204 { 205 // change the value of the parameter depending on its type 206 switch (it->valueType_) 207 { 208 case MT_Type::Int: 209 passPtr->getFragmentProgramParameters()->setNamedConstant(it->parameter_, it->valueInt_); 210 break; 211 case MT_Type::Float: 212 passPtr->getFragmentProgramParameters()->setNamedConstant(it->parameter_, it->valueFloat_); 213 break; 214 default: 215 break; 216 } 217 } 218 else 219 COUT(2) << "Warning: No pass " << it->pass_ << " in technique " << it->technique_ << " in compositor \"" << this->compositorName_ << "\" or pass has no shader." << std::endl; 220 } 221 else 222 COUT(2) << "Warning: No technique " << it->technique_ << " in compositor \"" << this->compositorName_ << "\" or technique has no pass with shader." << std::endl; 223 } 224 this->parameters_.clear(); 225 } 226 227 /** 228 @brief Detects if the Cg program manager plugin is active. 229 */ 230 /* static */ bool Shader::hasCgProgramManager() 231 { 232 if (Ogre::Root::getSingletonPtr()) 233 { 63 234 const Ogre::Root::PluginInstanceList& plugins = Ogre::Root::getSingleton().getInstalledPlugins(); 64 235 for (size_t i = 0; i < plugins.size(); ++i) 65 {66 236 if (plugins[i]->getName() == "Cg Program Manager") 67 {68 Shader::bLoadedCgPlugin_s = true;69 break;70 }71 }72 }73 74 this->bLoadCompositor_ &= Shader::bLoadedCgPlugin_s;75 }76 77 Shader::~Shader()78 {79 if (this->compositorInstance_ && this->bLoadCompositor_)80 {81 Ogre::Viewport* viewport = GraphicsManager::getInstance().getViewport();82 assert(viewport);83 Ogre::CompositorManager::getSingleton().removeCompositor(viewport, this->compositor_);84 }85 86 }87 88 void Shader::setSceneManager(Ogre::SceneManager* scenemanager)89 {90 this->scenemanager_ = scenemanager;91 this->bViewportInitialized_ = false;92 }93 94 void Shader::tick(float dt)95 {96 SUPER(Shader, tick, dt);97 98 if (this->bLoadCompositor_ && !this->bViewportInitialized_ && this->scenemanager_ && this->scenemanager_->getCurrentViewport())99 {100 this->bViewportInitialized_ = true;101 this->updateVisibility();102 }103 }104 105 void Shader::changedCompositor()106 {107 if (this->bLoadCompositor_)108 {109 Ogre::Viewport* viewport = GraphicsManager::getInstance().getViewport();110 assert(viewport);111 if (!this->oldcompositor_.empty())112 {113 Ogre::CompositorManager::getSingleton().removeCompositor(viewport, this->oldcompositor_);114 this->compositorInstance_ = 0;115 }116 if (!this->compositor_.empty())117 {118 this->compositorInstance_ = Ogre::CompositorManager::getSingleton().addCompositor(viewport, this->compositor_);119 if (!this->compositorInstance_)120 COUT(2) << "Warning: Couldn't load compositor with name \"" << this->compositor_ << "\"." << std::endl;121 Ogre::CompositorManager::getSingleton().setCompositorEnabled(viewport, this->compositor_, this->bViewportInitialized_ && this->isVisible());122 }123 this->oldcompositor_ = this->compositor_;124 }125 }126 127 void Shader::updateVisibility()128 {129 if (this->compositorInstance_ && this->scenemanager_)130 this->compositorInstance_->setEnabled(this->scenemanager_->getCurrentViewport() && this->isVisible());131 }132 133 void Shader::setParameter(const std::string& material, size_t technique, size_t pass, const std::string& parameter, float value)134 {135 if (Shader::_setParameter(material, technique, pass, parameter, value))136 {137 if (this->bViewportInitialized_ && this->compositorInstance_ && this->isVisible())138 {139 this->compositorInstance_->setEnabled(false);140 this->compositorInstance_->setEnabled(true);141 }142 }143 }144 145 void Shader::setParameter(const std::string& material, size_t technique, size_t pass, const std::string& parameter, int value)146 {147 if (Shader::_setParameter(material, technique, pass, parameter, value))148 {149 if (this->bViewportInitialized_ && this->compositorInstance_ && this->isVisible())150 {151 this->compositorInstance_->setEnabled(false);152 this->compositorInstance_->setEnabled(true);153 }154 }155 }156 157 bool Shader::_setParameter(const std::string& material, size_t technique, size_t pass, const std::string& parameter, float value)158 {159 ParameterPointer* pointer = Shader::getParameterPointer(material, technique, pass, parameter);160 if (pointer)161 {162 if (pointer->first)163 {164 if ((*static_cast<float*>(pointer->second)) != value)165 {166 (*static_cast<float*>(pointer->second)) = value;167 237 return true; 168 }169 }170 else171 {172 if ((*static_cast<int*>(pointer->second)) != static_cast<int>(value))173 {174 (*static_cast<int*>(pointer->second)) = static_cast<int>(value);175 return true;176 }177 }178 238 } 179 239 return false; 180 240 } 181 182 bool Shader::_setParameter(const std::string& material, size_t technique, size_t pass, const std::string& parameter, int value)183 {184 ParameterPointer* pointer = Shader::getParameterPointer(material, technique, pass, parameter);185 if (pointer)186 {187 if (pointer->first)188 {189 if ((*static_cast<float*>(pointer->second)) != static_cast<float>(value))190 {191 (*static_cast<float*>(pointer->second)) = static_cast<float>(value);192 return true;193 }194 }195 else196 {197 if ((*static_cast<int*>(pointer->second)) != value)198 {199 (*static_cast<int*>(pointer->second)) = value;200 return true;201 }202 }203 }204 return false;205 }206 207 float Shader::getParameter(const std::string& material, size_t technique, size_t pass, const std::string& parameter)208 {209 ParameterPointer* pointer = Shader::getParameterPointer(material, technique, pass, parameter);210 if (pointer)211 {212 if (pointer->first)213 return (*static_cast<float*>(pointer->second));214 else215 return static_cast<float>(*static_cast<int*>(pointer->second));216 }217 else218 return 0;219 }220 221 bool Shader::getParameterIsFloat(const std::string& material, size_t technique, size_t pass, const std::string& parameter)222 {223 ParameterPointer* pointer = Shader::getParameterPointer(material, technique, pass, parameter);224 if (pointer)225 return pointer->first;226 else227 return false;228 }229 230 bool Shader::getParameterIsInt(const std::string& material, size_t technique, size_t pass, const std::string& parameter)231 {232 ParameterPointer* pointer = Shader::getParameterPointer(material, technique, pass, parameter);233 if (pointer)234 return (!pointer->first);235 else236 return false;237 }238 239 Shader::ParameterPointer* Shader::getParameterPointer(const std::string& material, size_t technique, size_t pass, const std::string& parameter)240 {241 if (!GameMode::showsGraphics() || !Shader::bLoadedCgPlugin_s)242 return 0;243 244 MaterialMap::iterator material_iterator = Shader::parameters_s.find(material);245 if (material_iterator != Shader::parameters_s.end())246 {247 TechniqueVector& technique_vector = material_iterator->second;248 if (technique < technique_vector.size())249 {250 PassVector& pass_vector = technique_vector[technique];251 if (pass < pass_vector.size())252 {253 ParameterMap& parameter_map = pass_vector[pass];254 ParameterMap::iterator parameter_iterator = parameter_map.find(parameter);255 256 if (parameter_iterator != parameter_map.end())257 return (¶meter_iterator->second);258 else259 COUT(2) << "Warning: No shader parameter \"" << parameter << "\" in pass " << pass << " in technique " << technique << " in material \"" << material << "\"." << std::endl;260 }261 else262 COUT(2) << "Warning: No pass " << pass << " in technique " << technique << " in material \"" << material << "\" or pass has no shader." << std::endl;263 }264 else265 COUT(2) << "Warning: No technique " << technique << " in material \"" << material << "\" or technique has no pass with shader." << std::endl;266 }267 else268 {269 bool foundAtLeastOneShaderParameter = false;270 Ogre::MaterialManager::ResourceMapIterator iterator = Ogre::MaterialManager::getSingleton().getResourceIterator();271 Ogre::Material* material_pointer = 0;272 273 while (iterator.hasMoreElements())274 {275 Ogre::Resource* resource = iterator.getNext().get();276 if (resource->getName() == material)277 material_pointer = (Ogre::Material*)resource;278 }279 280 if (!material_pointer)281 {282 COUT(2) << "Warning: No material with name \"" << material << "\" found." << std::endl;283 return 0;284 }285 286 for (unsigned int t = 0; t < material_pointer->getNumTechniques(); ++t)287 {288 Ogre::Technique* technique_pointer = material_pointer->getTechnique(t);289 if (!technique_pointer)290 continue;291 292 for (unsigned int p = 0; p < technique_pointer->getNumPasses(); ++p)293 {294 Ogre::Pass* pass_pointer = technique_pointer->getPass(p);295 if (!pass_pointer)296 continue;297 298 if (!pass_pointer->getFragmentProgramName().empty())299 {300 Ogre::GpuProgramParameters* parameter_pointer = pass_pointer->getFragmentProgramParameters().get();301 if (!parameter_pointer)302 continue;303 304 const Ogre::GpuConstantDefinitionMap& constant_definitions = parameter_pointer->getConstantDefinitions().map;305 for (Ogre::GpuConstantDefinitionMap::const_iterator definition_iterator = constant_definitions.begin(); definition_iterator != constant_definitions.end(); ++definition_iterator)306 {307 void* temp = (definition_iterator->second.isFloat())308 ? static_cast<void*>(parameter_pointer->getFloatPointer(definition_iterator->second.physicalIndex))309 : static_cast<void*>(parameter_pointer->getIntPointer(definition_iterator->second.physicalIndex));310 ParameterPointer parameter_pointer = ParameterPointer(definition_iterator->second.isFloat(), temp);311 312 TechniqueVector& technique_vector = Shader::parameters_s[material];313 technique_vector.resize(technique + 1);314 PassVector& pass_vector = technique_vector[technique];315 pass_vector.resize(pass + 1);316 pass_vector[pass][definition_iterator->first] = parameter_pointer;317 foundAtLeastOneShaderParameter = true;318 }319 }320 }321 }322 323 // recursive call if the material was added to the map324 if (foundAtLeastOneShaderParameter)325 return Shader::getParameterPointer(material, technique, pass, parameter);326 }327 return 0;328 }329 241 }
Note: See TracChangeset
for help on using the changeset viewer.