Transit scheme.

1 files changed, 67 insertions, 0 deletions

diff --git a/drape_frontend/message_subclasses.hpp b/drape_frontend/message_subclasses.hpp
index 5c05a47cf1..85cd07be8e 100644
--- a/drape_frontend/message_subclasses.hpp
+++ b/drape_frontend/message_subclasses.hpp
@@ -17,6 +17,7 @@
 #include "drape_frontend/selection_shape.hpp"
 #include "drape_frontend/tile_utils.hpp"
 #include "drape_frontend/traffic_generator.hpp"
+#include "drape_frontend/transit_scheme_builder.hpp"
 #include "drape_frontend/user_event_stream.hpp"
 #include "drape_frontend/user_mark_shapes.hpp"
 #include "drape_frontend/user_marks_provider.hpp"
@@ -1019,6 +1020,72 @@ private:
   bool const m_isSimplified;
 };
 
+class EnableTransitSchemeMessage : public Message
+{
+public:
+  explicit EnableTransitSchemeMessage(bool enable)
+    : m_enable(enable)
+  {}
+
+  Type GetType() const override { return Message::EnableTransitScheme; }
+
+  bool Enable() { return m_enable; }
+
+private:
+  bool m_enable = false;
+};
+
+class ClearTransitSchemeDataMessage : public Message
+{
+public:
+  explicit ClearTransitSchemeDataMessage(MwmSet::MwmId const & mwmId)
+    : m_mwmId(mwmId)
+  {}
+
+  Type GetType() const override { return Message::ClearTransitSchemeData; }
+
+  MwmSet::MwmId const & GetMwmId() { return m_mwmId; }
+
+private:
+  MwmSet::MwmId m_mwmId;
+};
+
+class UpdateTransitSchemeMessage : public Message
+{
+public:
+  UpdateTransitSchemeMessage(TransitDisplayInfos && transitInfos,
+                             std::vector<MwmSet::MwmId> const & visibleMwms)
+    : m_transitInfos(move(transitInfos)), m_visibleMwms(visibleMwms)
+  {}
+
+  Type GetType() const override { return Message::UpdateTransitScheme; }
+
+  TransitDisplayInfos & GetTransitDisplayInfos() { return m_transitInfos; }
+  std::vector<MwmSet::MwmId> const & GetVisibleMwms() const { return m_visibleMwms; }
+
+private:
+  TransitDisplayInfos m_transitInfos;
+  std::vector<MwmSet::MwmId> m_visibleMwms;
+};
+
+class RegenerateTransitMessage : public Message
+{
+public:
+  Type GetType() const override { return Message::RegenerateTransitScheme; }
+};
+
+using FlushTransitSchemeMessage = FlushRenderDataMessage<TransitRenderData,
+                                                         Message::FlushTransitScheme>;
+
+using FlushTransitMarkersMessage = FlushRenderDataMessage<TransitRenderData,
+                                                          Message::FlushTransitMarkers>;
+
+using FlushTransitTextMessage = FlushRenderDataMessage<TransitRenderData,
+                                                       Message::FlushTransitText>;
+
+using FlushTransitStubsMessage = FlushRenderDataMessage<TransitRenderData,
+                                                        Message::FlushTransitStubs>;
+
 class DrapeApiAddLinesMessage : public Message
 {
 public: