Initial commit

Lts/Plugins/BuildingViewer/Common/BgViewer/viewerCtrl/roomdestructioncomponent.cpp

@@ -0,0 +1,142 @@
+#include "roomdestructioncomponent.h"
+
+#include "visitors.h"
+#include <dtEntity/core.h>
+#include <dtEntity/commandmessages.h>
+#include <dtEntity/systemmessages.h>
+#include <dtEntity/windowinterface.h>
+#include <dtEntityOSG/layercomponent.h>
+#include <dtEntityOSG/staticmeshcomponent.h>
+#include <osg/Depth>
+#include <osg/PolygonMode>
+#include <osg/BlendFunc>
+#include <osg/Material>
+#include <osg/NodeVisitor>
+#include <osg/Geode>
+#include <osg/Geometry>
+#include <osg/Node>
+#include <osgDB/FileUtils>
+#include <sstream>
+
+namespace VCtrl
+{
+
+
+////////////////////////////////////////////////////////////////////////////////
+const dtEntity::StringId RoomDestructionComponent::TYPE(dtEntity::SID("RoomDestruction"));
+
+////////////////////////////////////////////////////////////////////////////////
+void RoomDestructionComponent::OnEnteredWorld()
+{
+
+  osg::Shader *vShader = osg::Shader::readShaderFile(osg::Shader::VERTEX, osgDB::findDataFile("shaders/roomdestruction.vert"));
+  osg::Shader *fShader = osg::Shader::readShaderFile(osg::Shader::FRAGMENT, osgDB::findDataFile("shaders/roomdestruction.frag"));
+
+  if (vShader == nullptr || fShader == nullptr)
+  {
+    LOG_ERROR("Could not load building transparency shaders!");
+    Q_ASSERT(false);
+  }
+  osg::Program *p = new osg::Program();
+  p->setName("RoomDestructionShader");
+  p->addShader(vShader);
+  p->addShader(fShader);
+
+  dtEntityOSG::LayerComponent *lc;
+  if (mEntity->GetComponent(lc))
+  {
+    osg::Node *n = lc->GetAttachedComponentNode();
+    if (n != nullptr)
+    {
+      ClearVertexColorsVisitor v;
+      lc->GetAttachedComponentNode()->accept(v);
+      osg::StateSet *ss = n->getOrCreateStateSet();
+      ss->setAttributeAndModes(p);
+      ss->addUniform(new osg::Uniform("room_color", osg::Vec4(1, 1, 1, 1)));
+      ss->addUniform(new osg::Uniform("wall_color", osg::Vec4(1, 1, 1, 1)));
+    }
+  }
+
+  dtEntityOSG::StaticMeshComponent *smc;
+  if (mEntity->GetComponent(smc))
+  {
+    FindRoomNamesVisitor v;
+    smc->GetNode()->accept(v);
+    for (const auto &roomName: v.mNames)
+      mRoomDestructions[roomName] = 0;
+  }
+}
+
+////////////////////////////////////////////////////////////////////////////
+bool RoomDestructionComponent::SetRoomDestructionValue(const std::string &room, float value)
+{
+  if (mRoomDestructions.find(room) == mRoomDestructions.end())
+    return false;
+
+  FindRoomByNumberVisitor fn;
+  fn.setNameToFind(room);
+  dtEntityOSG::LayerComponent *lc;
+
+  if (!mEntity->GetComponent(lc))
+    return false;
+
+  osg::Node *n = lc->GetAttachedComponentNode();
+  if (n == nullptr)
+    return false;
+
+  lc->GetAttachedComponentNode()->accept(fn);
+
+
+  for (const auto &node : fn.getNodeList())
+  {
+    RoomDestructionSystem *bts;
+    mEntity->GetEntityManager().GetES(bts);
+
+    FindRoomComponentsVisitor frc;
+    node->accept(frc);
+
+    FindExactNameVisitor v(frc.mNamesFound);
+    //LOG_ERROR("Found " + std::to_string(frc.mNamesFound.size()) + " component names");
+    osg::Node *partNode = lc->GetAttachedComponentNode();
+    if (partNode)
+      partNode->accept(v);
+
+    for (const auto &visitedNodes : v.mNodes)
+    {
+      osg::StateSet *ss = visitedNodes->getOrCreateStateSet();
+      ss->addUniform(new osg::Uniform("room_color", osg::Vec4(1, 1 - value, 1 - value, 1)));
+    }
+  }
+  dtEntity::GetWindowInterface()->RequestRedraw(0);
+
+  mRoomDestructions[room] = value;
+  return true;
+}
+
+////////////////////////////////////////////////////////////////////////////
+bool RoomDestructionComponent::SetBuildingDestructionValue(float value)
+{
+  for (const auto &rd : mRoomDestructions)
+    SetRoomDestructionValue(rd.first, value);
+  return true;
+}
+
+////////////////////////////////////////////////////////////////////////////
+const dtEntity::StringId RoomDestructionSystem::TYPE(dtEntity::SID("RoomDestruction"));
+
+RoomDestructionSystem::RoomDestructionSystem(dtEntity::EntityManager &em)
+  : dtEntity::DefaultEntitySystem<RoomDestructionComponent>(em)
+{
+  mEnterWorldFunctor = dtEntity::MessageFunctor(this, &RoomDestructionSystem::OnEnterWorld);
+  em.RegisterForMessages(dtEntity::EntityAddedToSceneMessage::TYPE, mEnterWorldFunctor, "RoomDestructionSystem::OnEnterWorld");
+}
+
+////////////////////////////////////////////////////////////////////////////
+void RoomDestructionSystem::OnEnterWorld(const dtEntity::Message &msg)
+{
+  dtEntity::EntityId id = msg.Get(dtEntity::EntityAddedToSceneMessage::AboutEntityId)->UIntValue();
+  RoomDestructionComponent *bt;
+  if (GetEntityManager().GetComponent(id, bt))
+    bt->OnEnteredWorld();
+}
+}