supercell-wx/scwx-qt/source/scwx/qt/map/map_widget.cpp

#include <scwx/qt/map/map_widget.hpp>
#include <scwx/qt/gl/gl.hpp>
#include <scwx/qt/manager/radar_product_manager.hpp>
#include <scwx/qt/manager/settings_manager.hpp>
#include <scwx/qt/map/overlay_layer.hpp>
#include <scwx/qt/map/radar_product_layer.hpp>
#include <scwx/qt/map/radar_range_layer.hpp>
#include <scwx/qt/view/radar_product_view_factory.hpp>

#include <QApplication>
#include <QColor>
#include <QDebug>
#include <QFile>
#include <QIcon>
#include <QKeyEvent>
#include <QMouseEvent>
#include <QString>

namespace scwx
{
namespace qt
{
namespace map
{

typedef std::pair<std::string, std::string> MapStyle;

// clang-format off
static const MapStyle streets          { "mapbox://styles/mapbox/streets-v11",           "Streets"};
static const MapStyle outdoors         { "mapbox://styles/mapbox/outdoors-v11",          "Outdoors"};
static const MapStyle light            { "mapbox://styles/mapbox/light-v10",             "Light"};
static const MapStyle dark             { "mapbox://styles/mapbox/dark-v10",              "Dark" };
static const MapStyle satellite        { "mapbox://styles/mapbox/satellite-v9",          "Satellite" };
static const MapStyle satelliteStreets { "mapbox://styles/mapbox/satellite-streets-v11", "Satellite Streets" };
// clang-format on

static const std::array<MapStyle, 6> mapboxStyles_ = {
   {streets, outdoors, light, dark, satellite, satelliteStreets}};

class MapWidgetImpl
{
public:
   explicit MapWidgetImpl(const QMapboxGLSettings& settings) :
       gl_(),
       settings_(settings),
       map_(),
       radarProductManager_ {std::make_shared<manager::RadarProductManager>()},
       radarProductLayer_ {nullptr},
       radarProductView_ {nullptr},
       overlayLayer_ {nullptr},
       lastPos_(),
       frameDraws_(0)
   {
   }
   ~MapWidgetImpl() = default;

   gl::OpenGLFunctions gl_;

   QMapboxGLSettings          settings_;
   std::shared_ptr<QMapboxGL> map_;

   std::shared_ptr<manager::RadarProductManager> radarProductManager_;

   std::shared_ptr<common::ColorTable>     colorTable_;
   std::shared_ptr<view::RadarProductView> radarProductView_;
   std::shared_ptr<RadarProductLayer>      radarProductLayer_;
   std::shared_ptr<OverlayLayer>           overlayLayer_;

   QPointF lastPos_;

   uint64_t frameDraws_;
};

MapWidget::MapWidget(const QMapboxGLSettings& settings) :
    p(std::make_unique<MapWidgetImpl>(settings))
{
   setFocusPolicy(Qt::StrongFocus);

   p->radarProductManager_->Initialize();

   SelectRadarProduct(common::Level2Product::Reflectivity);
}

MapWidget::~MapWidget()
{
   // Make sure we have a valid context so we
   // can delete the QMapboxGL.
   makeCurrent();
}

float MapWidget::GetElevation() const
{
   return p->radarProductView_->elevation();
}

std::vector<float> MapWidget::GetElevationCuts() const
{
   return p->radarProductView_->GetElevationCuts();
}

void MapWidget::SelectElevation(float elevation)
{
   p->radarProductView_->SelectElevation(elevation);
}

void MapWidget::SelectRadarProduct(common::Level2Product product)
{
   float currentElevation = 0.0f;

   if (p->radarProductView_ != nullptr)
   {
      currentElevation = p->radarProductView_->elevation();
   }

   p->radarProductView_ = view::RadarProductViewFactory::Create(
      product, currentElevation, p->radarProductManager_);

   connect(
      p->radarProductView_.get(),
      &view::RadarProductView::ColorTableUpdated,
      this,
      [&]() { update(); },
      Qt::QueuedConnection);
   connect(
      p->radarProductView_.get(),
      &view::RadarProductView::SweepComputed,
      this,
      [&]() {
         RadarRangeLayer::Update(p->map_, p->radarProductView_->range());
         update();
         emit RadarSweepUpdated();
      },
      Qt::QueuedConnection);

   p->radarProductView_->Initialize();

   std::string colorTableFile =
      manager::SettingsManager::palette_settings()->palette(
         common::GetLevel2Palette(product));
   if (!colorTableFile.empty())
   {
      std::shared_ptr<common::ColorTable> colorTable =
         common::ColorTable::Load(colorTableFile);
      p->radarProductView_->LoadColorTable(colorTable);
   }

   if (p->map_ != nullptr)
   {
      AddLayers();
   }
}

qreal MapWidget::pixelRatio()
{
   return devicePixelRatioF();
}

void MapWidget::changeStyle()
{
   static uint8_t currentStyleIndex = 0;

   auto& styles = mapboxStyles_;

   p->map_->setStyleUrl(styles[currentStyleIndex].first.c_str());
   setWindowTitle(QString("Mapbox GL: ") +
                  styles[currentStyleIndex].second.c_str());

   if (++currentStyleIndex == styles.size())
   {
      currentStyleIndex = 0;
   }
}

void MapWidget::AddLayers()
{
   // TODO: Improve this
   if (p->map_->layerExists("radar"))
   {
      p->map_->removeLayer("radar");
   }
   if (p->map_->layerExists("overlay"))
   {
      p->map_->removeLayer("overlay");
   }

   // QMapboxGL::addCustomLayer will take ownership of the QScopedPointer
   QScopedPointer<QMapbox::CustomLayerHostInterface> pHost(
      new RadarProductLayer(p->radarProductView_, p->gl_));
   QScopedPointer<QMapbox::CustomLayerHostInterface> pOverlayHost(
      new OverlayLayer(p->radarProductView_, p->gl_));

   QString before = "ferry";

   for (const QString& layer : p->map_->layerIds())
   {
      // Draw below tunnels, ferries and roads
      if (layer.startsWith("tunnel") || layer.startsWith("ferry") ||
          layer.startsWith("road"))
      {
         before = layer;
         break;
      }
   }

   p->map_->addCustomLayer("radar", pHost, before);
   RadarRangeLayer::Add(p->map_, p->radarProductView_->range(), before);
   p->map_->addCustomLayer("overlay", pOverlayHost);
}

void MapWidget::keyPressEvent(QKeyEvent* ev)
{
   switch (ev->key())
   {
   case Qt::Key_S: changeStyle(); break;
   case Qt::Key_L:
   {
      for (const QString& layer : p->map_->layerIds())
      {
         qDebug() << "Layer: " << layer;
      }
   }
   break;
   default: break;
   }

   ev->accept();
}

void MapWidget::mousePressEvent(QMouseEvent* ev)
{
   p->lastPos_ = ev->position();

   if (ev->type() == QEvent::MouseButtonPress)
   {
      if (ev->buttons() == (Qt::LeftButton | Qt::RightButton))
      {
         changeStyle();
      }
   }

   if (ev->type() == QEvent::MouseButtonDblClick)
   {
      if (ev->buttons() == Qt::LeftButton)
      {
         p->map_->scaleBy(2.0, p->lastPos_);
      }
      else if (ev->buttons() == Qt::RightButton)
      {
         p->map_->scaleBy(0.5, p->lastPos_);
      }
   }

   ev->accept();
}

void MapWidget::mouseMoveEvent(QMouseEvent* ev)
{
   QPointF delta = ev->position() - p->lastPos_;

   if (!delta.isNull())
   {
      if (ev->buttons() == Qt::LeftButton &&
          ev->modifiers() & Qt::ShiftModifier)
      {
         p->map_->pitchBy(delta.y());
      }
      else if (ev->buttons() == Qt::LeftButton)
      {
         p->map_->moveBy(delta);
      }
      else if (ev->buttons() == Qt::RightButton)
      {
         p->map_->rotateBy(p->lastPos_, ev->position());
      }
   }

   p->lastPos_ = ev->position();
   ev->accept();
}

void MapWidget::wheelEvent(QWheelEvent* ev)
{
   if (ev->angleDelta().y() == 0)
   {
      return;
   }

   float factor = ev->angleDelta().y() / 1200.;
   if (ev->angleDelta().y() < 0)
   {
      factor = factor > -1 ? factor : 1 / factor;
   }

   p->map_->scaleBy(1 + factor, ev->position());

   ev->accept();
}

void MapWidget::initializeGL()
{
   makeCurrent();
   p->gl_.initializeOpenGLFunctions();

   p->map_.reset(new QMapboxGL(nullptr, p->settings_, size(), pixelRatio()));
   connect(p->map_.get(), SIGNAL(needsRendering()), this, SLOT(update()));

   // Set default location to KLSX.
   p->map_->setCoordinateZoom(QMapbox::Coordinate(38.6986, -90.6828), 11);

   QString styleUrl = qgetenv("MAPBOX_STYLE_URL");
   if (styleUrl.isEmpty())
   {
      changeStyle();
   }
   else
   {
      p->map_->setStyleUrl(styleUrl);
      setWindowTitle(QString("Mapbox GL: ") + styleUrl);
   }

   connect(p->map_.get(), &QMapboxGL::mapChanged, this, &MapWidget::mapChanged);
}

void MapWidget::paintGL()
{
   p->frameDraws_++;
   p->map_->resize(size());
   p->map_->setFramebufferObject(defaultFramebufferObject(),
                                 size() * pixelRatio());
   p->map_->render();
}

void MapWidget::mapChanged(QMapboxGL::MapChange mapChange)
{
   switch (mapChange)
   {
   case QMapboxGL::MapChangeDidFinishLoadingStyle: AddLayers(); break;
   }
}

} // namespace map
} // namespace qt
} // namespace scwx