Unknown segments removing.

1 files changed, 7 insertions, 6 deletions

diff --git a/drape_frontend/drape_engine.cpp b/drape_frontend/drape_engine.cpp
index 4a89d0c7fc..c9a0485e8c 100644
--- a/drape_frontend/drape_engine.cpp
+++ b/drape_frontend/drape_engine.cpp
@@ -11,6 +11,7 @@
 #include "platform/settings.hpp"
 
 #include "std/bind.hpp"
+#include "std/utility.hpp"
 
 namespace
 {
@@ -547,13 +548,13 @@ void DrapeEngine::UpdateTraffic(traffic::TrafficInfo const & info)
   if (info.GetColoring().empty())
     return;
 
-  df::TrafficSegmentsColoring segmentsColoring;
-  auto & mwmColoring = segmentsColoring[info.GetMwmId()];
+#ifdef DEBUG
   for (auto const & segmentPair : info.GetColoring())
-  {
-    if (segmentPair.second != traffic::SpeedGroup::Unknown)
-      mwmColoring.insert(segmentPair);
-  }
+    ASSERT_NOT_EQUAL(segmentPair.second, traffic::SpeedGroup::Unknown, ());
+#endif
+
+  df::TrafficSegmentsColoring segmentsColoring;
+  segmentsColoring.insert(make_pair(info.GetMwmId(), info.GetColoring()));
 
   m_threadCommutator->PostMessage(ThreadsCommutator::ResourceUploadThread,
                                   make_unique_dp<UpdateTrafficMessage>(move(segmentsColoring), false),