Ticket #1163: 1163.5.diff

source/tools/atlas/GameInterface/Handlers/ObjectHandlers.cpp

@@ -1 @@
-/* Copyright (C) 2014 Wildfire Games.
+/* Copyright (C) 2015 Wildfire Games.
  * This file is part of 0 A.D.
  *
  * 0 A.D. is free software: you can redistribute it and/or modify
@@ -48 +48 @@
 #include "simulation2/components/ICmpSelectable.h"
 #include "simulation2/components/ICmpTemplateManager.h"
 #include "simulation2/components/ICmpVisual.h"
+#include "simulation2/components/ICmpObstruction.h"
 #include "simulation2/helpers/Selection.h"
 #include "ps/XML/XMLWriter.h"
 
@@ -61 +62 @@
     }
 }
 
+// Helpers for object constraints
+bool CanPlaceObjectEntityHelper(entity_id_t ent)
+{
+    CmpPtr<ICmpObstruction> cmpObstruction(*g_Game->GetSimulation2(), ent);
+    if (cmpObstruction)
+    {
+        ICmpObstruction::EFoundationCheck result = cmpObstruction->CheckFoundation("default");
+        if (result != ICmpObstruction::FOUNDATION_CHECK_SUCCESS)
+            return false;
+    }
+    return true;
+}
+
+void CheckObstructionAndUpdateVisual(entity_id_t id)
+{
+    // Check build Constraint
+    CmpPtr<ICmpVisual> cmpVisual(*g_Game->GetSimulation2(), id);
+    if (cmpVisual)
+    {
+        if (!CanPlaceObjectEntityHelper(id))
+            cmpVisual->SetShadingColour(fixed::FromDouble(1.4), fixed::FromDouble(0.4), fixed::FromDouble(0.4), fixed::FromDouble(1));
+        else
+            cmpVisual->SetShadingColour(fixed::FromDouble(1), fixed::FromDouble(1), fixed::FromDouble(1), fixed::FromDouble(1));
+    }
+}
+
 QUERYHANDLER(GetObjectsList)
 {
     std::vector<sObjectsListItem> objects;
@@ -473 +500 @@
             posFinal = posFixed + dir;
         }
         cmpPosition->JumpTo(posFinal.X, posFinal.Z);
+
+        // Check build Constraint
+        CheckObstructionAndUpdateVisual(id);
     }
 }
 
@@ -543 +573 @@
         CmpPtr<ICmpOwnership> cmpOwnership(*g_Game->GetSimulation2(), g_PreviewEntityID);
         if (cmpOwnership)
             cmpOwnership->SetOwner((player_id_t)msg->settings->player);
+
+        // Check build Constraint
+        CheckObstructionAndUpdateVisual(g_PreviewEntityID);
     }
 }
 
@@ -599 +632 @@
         if (cmpVisual)
         {
             cmpVisual->SetActorSeed(m_ActorSeed);
+            CheckObstructionAndUpdateVisual(m_EntityID);
             // TODO: variation/selection strings
         }
     }
@@ -736 +770 @@
         ObjectPositionMap::iterator it;
         for (it = map.begin(); it != map.end(); ++it)
         {
-            CmpPtr<ICmpPosition> cmpPosition(*g_Game->GetSimulation2(), (entity_id_t)it->first);
+            entity_id_t id = (entity_id_t)it->first;
+            CmpPtr<ICmpPosition> cmpPosition(*g_Game->GetSimulation2(), id);
             if (!cmpPosition)
                 return;
 
@@ -743 +778 @@
             // Set 2D position, ignoring height
             CVector3D pos = it->second;
             cmpPosition->JumpTo(entity_pos_t::FromFloat(pos.X), entity_pos_t::FromFloat(pos.Z));
+
+            // Check build Constraint
+            CheckObstructionAndUpdateVisual(id);
         }
     }