#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace scwx { namespace qt { namespace map { static const std::string logPrefix_ = "scwx::qt::map::map_widget"; static const auto logger_ = scwx::util::Logger::Create(logPrefix_); typedef std::pair 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 mapboxStyles_ = { {streets, outdoors, light, dark, satellite, satelliteStreets}}; class MapWidgetImpl : public QObject { Q_OBJECT public: explicit MapWidgetImpl(MapWidget* widget, const QMapLibreGL::Settings& settings) : context_ {std::make_shared()}, widget_ {widget}, settings_(settings), map_(), layerList_ {}, imGuiRendererInitialized_ {false}, radarProductManager_ {nullptr}, radarProductLayer_ {nullptr}, alertLayer_ {std::make_shared(context_)}, overlayLayer_ {nullptr}, colorTableLayer_ {nullptr}, autoRefreshEnabled_ {true}, selectedLevel2Product_ {common::Level2Product::Unknown}, selectedTime_ {}, lastPos_(), currentStyleIndex_ {0}, frameDraws_(0), prevLatitude_ {0.0}, prevLongitude_ {0.0}, prevZoom_ {0.0}, prevBearing_ {0.0}, prevPitch_ {0.0} { SetRadarSite(scwx::qt::manager::SettingsManager::general_settings() .default_radar_site() .GetValue()); // Create ImGui Context static size_t currentMapId_ {0u}; imGuiContextName_ = std::format("Map {}", ++currentMapId_); imGuiContext_ = model::ImGuiContextModel::Instance().CreateContext(imGuiContextName_); // Initialize ImGui Qt backend ImGui_ImplQt_Init(); ImGui_ImplQt_RegisterWidget(widget_); } ~MapWidgetImpl() { // Set ImGui Context ImGui::SetCurrentContext(imGuiContext_); // Shutdown ImGui Context if (imGuiRendererInitialized_) { ImGui_ImplOpenGL3_Shutdown(); } ImGui_ImplQt_Shutdown(); // Destroy ImGui Context model::ImGuiContextModel::Instance().DestroyContext(imGuiContextName_); } void AddLayer(const std::string& id, std::shared_ptr layer, const std::string& before = {}); void InitializeNewRadarProductView(const std::string& colorPalette); void RadarProductManagerConnect(); void RadarProductManagerDisconnect(); void RadarProductViewConnect(); void RadarProductViewDisconnect(); void SetRadarSite(const std::string& radarSite); bool UpdateStoredMapParameters(); common::Level2Product GetLevel2ProductOrDefault(const std::string& productName) const; std::shared_ptr context_; MapWidget* widget_; QMapLibreGL::Settings settings_; std::shared_ptr map_; std::list layerList_; ImGuiContext* imGuiContext_; std::string imGuiContextName_; bool imGuiRendererInitialized_; std::shared_ptr radarProductManager_; std::shared_ptr colorTable_; std::shared_ptr radarProductLayer_; std::shared_ptr alertLayer_; std::shared_ptr overlayLayer_; std::shared_ptr colorTableLayer_; bool autoRefreshEnabled_; common::Level2Product selectedLevel2Product_; std::chrono::system_clock::time_point selectedTime_; QPointF lastPos_; uint8_t currentStyleIndex_; uint64_t frameDraws_; double prevLatitude_; double prevLongitude_; double prevZoom_; double prevBearing_; double prevPitch_; public slots: void Update(); }; MapWidget::MapWidget(const QMapLibreGL::Settings& settings) : p(std::make_unique(this, settings)) { setFocusPolicy(Qt::StrongFocus); ImGui_ImplQt_RegisterWidget(this); } MapWidget::~MapWidget() { // Make sure we have a valid context so we can delete the QMapLibreGL. makeCurrent(); } common::Level3ProductCategoryMap MapWidget::GetAvailableLevel3Categories() { if (p->radarProductManager_ != nullptr) { return p->radarProductManager_->GetAvailableLevel3Categories(); } else { return {}; } } float MapWidget::GetElevation() const { auto radarProductView = p->context_->radar_product_view(); if (radarProductView != nullptr) { return radarProductView->elevation(); } else { return 0.0f; } } std::vector MapWidget::GetElevationCuts() const { auto radarProductView = p->context_->radar_product_view(); if (radarProductView != nullptr) { return radarProductView->GetElevationCuts(); } else { return {}; } } common::Level2Product MapWidgetImpl::GetLevel2ProductOrDefault(const std::string& productName) const { common::Level2Product level2Product = common::GetLevel2Product(productName); if (level2Product == common::Level2Product::Unknown) { auto radarProductView = context_->radar_product_view(); if (radarProductView != nullptr) { level2Product = common::GetLevel2Product(radarProductView->GetRadarProductName()); } } if (level2Product == common::Level2Product::Unknown) { if (selectedLevel2Product_ != common::Level2Product::Unknown) { level2Product = selectedLevel2Product_; } else { level2Product = common::Level2Product::Reflectivity; } } return level2Product; } std::vector MapWidget::GetLevel3Products() { if (p->radarProductManager_ != nullptr) { return p->radarProductManager_->GetLevel3Products(); } else { return {}; } } common::RadarProductGroup MapWidget::GetRadarProductGroup() const { auto radarProductView = p->context_->radar_product_view(); if (radarProductView != nullptr) { return radarProductView->GetRadarProductGroup(); } else { return common::RadarProductGroup::Unknown; } } std::string MapWidget::GetRadarProductName() const { auto radarProductView = p->context_->radar_product_view(); if (radarProductView != nullptr) { return radarProductView->GetRadarProductName(); } else { return "?"; } } std::shared_ptr MapWidget::GetRadarSite() const { std::shared_ptr radarSite = nullptr; if (p->radarProductManager_ != nullptr) { radarSite = p->radarProductManager_->radar_site(); } return radarSite; } uint16_t MapWidget::GetVcp() const { auto radarProductView = p->context_->radar_product_view(); if (radarProductView != nullptr) { return radarProductView->vcp(); } else { return 0; } } void MapWidget::SelectElevation(float elevation) { auto radarProductView = p->context_->radar_product_view(); if (radarProductView != nullptr) { radarProductView->SelectElevation(elevation); radarProductView->Update(); } } void MapWidget::SelectRadarProduct(common::RadarProductGroup group, const std::string& product, int16_t productCode) { bool radarProductViewCreated = false; auto radarProductView = p->context_->radar_product_view(); std::string productName {product}; // Validate level 2 product, set to default if invalid if (group == common::RadarProductGroup::Level2) { common::Level2Product level2Product = p->GetLevel2ProductOrDefault(productName); productName = common::GetLevel2Name(level2Product); p->selectedLevel2Product_ = level2Product; } if (group == common::RadarProductGroup::Level3 && productCode == 0) { productCode = common::GetLevel3ProductCodeByAwipsId(productName); } if (radarProductView == nullptr || radarProductView->GetRadarProductGroup() != group || (radarProductView->GetRadarProductGroup() == common::RadarProductGroup::Level2 && radarProductView->GetRadarProductName() != productName) || p->context_->radar_product_code() != productCode) { p->RadarProductViewDisconnect(); radarProductView = view::RadarProductViewFactory::Create( group, productName, productCode, p->radarProductManager_); p->context_->set_radar_product_view(radarProductView); p->RadarProductViewConnect(); radarProductViewCreated = true; } else { radarProductView->SelectProduct(productName); } p->context_->set_radar_product_group(group); p->context_->set_radar_product(productName); p->context_->set_radar_product_code(productCode); if (radarProductView != nullptr) { // Always select the latest product available radarProductView->SelectTime({}); if (radarProductViewCreated) { const std::string palette = (group == common::RadarProductGroup::Level2) ? common::GetLevel2Palette(common::GetLevel2Product(productName)) : common::GetLevel3Palette(productCode); p->InitializeNewRadarProductView(palette); } else { radarProductView->Update(); } } if (p->autoRefreshEnabled_) { p->radarProductManager_->EnableRefresh(group, productName, true); } } void MapWidget::SelectRadarProduct( std::shared_ptr record) { const std::string radarId = record->radar_id(); common::RadarProductGroup group = record->radar_product_group(); const std::string product = record->radar_product(); std::chrono::system_clock::time_point time = record->time(); int16_t productCode = record->product_code(); logger_->debug("SelectRadarProduct: {}, {}, {}, {}", radarId, common::GetRadarProductGroupName(group), product, scwx::util::TimeString(time)); p->SetRadarSite(radarId); p->selectedTime_ = time; SelectRadarProduct(group, product, productCode); } void MapWidget::SelectRadarSite(const std::string& id, bool updateCoordinates) { logger_->debug("Selecting radar site: {}", id); std::shared_ptr radarSite = config::RadarSite::Get(id); SelectRadarSite(radarSite, updateCoordinates); } void MapWidget::SelectRadarSite(std::shared_ptr radarSite, bool updateCoordinates) { // Verify radar site is valid and has changed if (radarSite != nullptr && (p->radarProductManager_ == nullptr || radarSite->id() != p->radarProductManager_->radar_site()->id())) { auto radarProductView = p->context_->radar_product_view(); if (updateCoordinates) { p->map_->setCoordinate( {radarSite->latitude(), radarSite->longitude()}); } p->SetRadarSite(radarSite->id()); p->Update(); // Select products from new site if (radarProductView != nullptr) { radarProductView->set_radar_product_manager(p->radarProductManager_); SelectRadarProduct(radarProductView->GetRadarProductGroup(), radarProductView->GetRadarProductName(), 0); } AddLayers(); // TODO: Disable refresh from old site } } void MapWidget::SetActive(bool isActive) { p->context_->settings().isActive_ = isActive; update(); } void MapWidget::SetAutoRefresh(bool enabled) { if (p->autoRefreshEnabled_ != enabled) { p->autoRefreshEnabled_ = enabled; auto radarProductView = p->context_->radar_product_view(); if (p->autoRefreshEnabled_ && radarProductView != nullptr) { p->radarProductManager_->EnableRefresh( radarProductView->GetRadarProductGroup(), radarProductView->GetRadarProductName(), true); } } } void MapWidget::SetMapLocation(double latitude, double longitude, bool updateRadarSite) { if (p->map_ != nullptr && p->prevLatitude_ != latitude || p->prevLongitude_ != longitude) { // Update the map location p->map_->setCoordinate({latitude, longitude}); // If the radar site should be updated based on the new location if (updateRadarSite) { // Find the nearest WSR-88D radar std::shared_ptr nearestRadarSite = config::RadarSite::FindNearest(latitude, longitude, "wsr88d"); // If found, select it if (nearestRadarSite != nullptr) { SelectRadarSite(nearestRadarSite->id(), false); } } } } void MapWidget::SetMapParameters( double latitude, double longitude, double zoom, double bearing, double pitch) { if (p->map_ != nullptr && (p->prevLatitude_ != latitude || p->prevLongitude_ != longitude || p->prevZoom_ != zoom || p->prevBearing_ != bearing || p->prevPitch_ != pitch)) { p->map_->setCoordinateZoom({latitude, longitude}, zoom); p->map_->setBearing(bearing); p->map_->setPitch(pitch); } } qreal MapWidget::pixelRatio() { return devicePixelRatioF(); } void MapWidget::changeStyle() { auto& styles = mapboxStyles_; p->map_->setStyleUrl(styles[p->currentStyleIndex_].first.c_str()); setWindowTitle(QString("Mapbox GL: ") + styles[p->currentStyleIndex_].second.c_str()); if (++p->currentStyleIndex_ == styles.size()) { p->currentStyleIndex_ = 0; } } void MapWidget::AddLayers() { logger_->debug("AddLayers()"); // Clear custom layers for (const std::string& id : p->layerList_) { p->map_->removeLayer(id.c_str()); } p->layerList_.clear(); auto radarProductView = p->context_->radar_product_view(); if (radarProductView != nullptr) { p->radarProductLayer_ = std::make_shared(p->context_); p->colorTableLayer_ = std::make_shared(p->context_); std::shared_ptr radarSite = p->radarProductManager_->radar_site(); std::string 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.toStdString(); break; } } p->AddLayer("radar", p->radarProductLayer_, before); RadarRangeLayer::Add(p->map_, radarProductView->range(), {radarSite->latitude(), radarSite->longitude()}); p->AddLayer("colorTable", p->colorTableLayer_); } p->alertLayer_->AddLayers("colorTable"); p->overlayLayer_ = std::make_shared(p->context_); p->AddLayer("overlay", p->overlayLayer_); } void MapWidgetImpl::AddLayer(const std::string& id, std::shared_ptr layer, const std::string& before) { // QMapLibreGL::addCustomLayer will take ownership of the std::unique_ptr std::unique_ptr pHost = std::make_unique(layer); map_->addCustomLayer(id.c_str(), std::move(pHost), before.c_str()); layerList_.push_back(id); } 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() { logger_->debug("initializeGL()"); makeCurrent(); p->context_->gl().initializeOpenGLFunctions(); // Initialize ImGui OpenGL3 backend ImGui::SetCurrentContext(p->imGuiContext_); ImGui_ImplOpenGL3_Init(); p->imGuiRendererInitialized_ = true; p->map_.reset( new QMapLibreGL::Map(nullptr, p->settings_, size(), pixelRatio())); p->context_->set_map(p->map_); connect(p->map_.get(), &QMapLibreGL::Map::needsRendering, p.get(), &MapWidgetImpl::Update); // Set default location to radar site std::shared_ptr radarSite = p->radarProductManager_->radar_site(); p->map_->setCoordinateZoom({radarSite->latitude(), radarSite->longitude()}, 7); p->UpdateStoredMapParameters(); emit MapParametersChanged(p->prevLatitude_, p->prevLongitude_, p->prevZoom_, p->prevBearing_, p->prevPitch_); QString styleUrl = qgetenv("MAPBOX_STYLE_URL"); if (styleUrl.isEmpty()) { changeStyle(); } else { p->map_->setStyleUrl(styleUrl); setWindowTitle(QString("Mapbox GL: ") + styleUrl); } connect(p->map_.get(), &QMapLibreGL::Map::mapChanged, this, &MapWidget::mapChanged); } void MapWidget::paintGL() { p->frameDraws_++; // Setup ImGui Frame ImGui::SetCurrentContext(p->imGuiContext_); // Start ImGui Frame ImGui_ImplQt_NewFrame(this); ImGui_ImplOpenGL3_NewFrame(); ImGui::NewFrame(); // Render QMapLibreGL Map p->map_->resize(size()); p->map_->setFramebufferObject(defaultFramebufferObject(), size() * pixelRatio()); p->map_->render(); // Render ImGui Frame ImGui::Render(); ImGui_ImplOpenGL3_RenderDrawData(ImGui::GetDrawData()); } void MapWidget::mapChanged(QMapLibreGL::Map::MapChange mapChange) { switch (mapChange) { case QMapLibreGL::Map::MapChangeDidFinishLoadingStyle: AddLayers(); break; } } void MapWidgetImpl::RadarProductManagerConnect() { if (radarProductManager_ != nullptr) { connect(radarProductManager_.get(), &manager::RadarProductManager::Level3ProductsChanged, this, [&]() { emit widget_->Level3ProductsChanged(); }); connect( radarProductManager_.get(), &manager::RadarProductManager::NewDataAvailable, this, [&](common::RadarProductGroup group, const std::string& product, std::chrono::system_clock::time_point latestTime) { if (autoRefreshEnabled_ && context_->radar_product_group() == group && (group == common::RadarProductGroup::Level2 || context_->radar_product() == product)) { // Create file request std::shared_ptr request = std::make_shared(); // File request callback connect(request.get(), &request::NexradFileRequest::RequestComplete, this, [&](std::shared_ptr request) { // Select loaded record auto record = request->radar_product_record(); if (record != nullptr) { widget_->SelectRadarProduct(record); } }); // Load file scwx::util::async( [=]() { if (group == common::RadarProductGroup::Level2) { radarProductManager_->LoadLevel2Data(latestTime, request); } else { radarProductManager_->LoadLevel3Data( product, latestTime, request); } }); } }, Qt::QueuedConnection); } } void MapWidgetImpl::RadarProductManagerDisconnect() { if (radarProductManager_ != nullptr) { disconnect(radarProductManager_.get(), &manager::RadarProductManager::NewDataAvailable, this, nullptr); } } void MapWidgetImpl::InitializeNewRadarProductView( const std::string& colorPalette) { scwx::util::async( [=]() { auto radarProductView = context_->radar_product_view(); std::string colorTableFile = manager::SettingsManager::palette_settings() .palette(colorPalette) .GetValue(); if (!colorTableFile.empty()) { std::unique_ptr colorTableStream = util::OpenFile(colorTableFile); std::shared_ptr colorTable = common::ColorTable::Load(*colorTableStream); radarProductView->LoadColorTable(colorTable); } radarProductView->Initialize(); }); if (map_ != nullptr) { widget_->AddLayers(); } } void MapWidgetImpl::RadarProductViewConnect() { auto radarProductView = context_->radar_product_view(); if (radarProductView != nullptr) { connect( radarProductView.get(), &view::RadarProductView::ColorTableUpdated, this, [&]() { widget_->update(); }, Qt::QueuedConnection); connect( radarProductView.get(), &view::RadarProductView::SweepComputed, this, [=]() { std::shared_ptr radarSite = radarProductManager_->radar_site(); RadarRangeLayer::Update( map_, radarProductView->range(), {radarSite->latitude(), radarSite->longitude()}); widget_->update(); emit widget_->RadarSweepUpdated(); }, Qt::QueuedConnection); } } void MapWidgetImpl::RadarProductViewDisconnect() { auto radarProductView = context_->radar_product_view(); if (radarProductView != nullptr) { disconnect(radarProductView.get(), &view::RadarProductView::ColorTableUpdated, this, nullptr); disconnect(radarProductView.get(), &view::RadarProductView::SweepComputed, this, nullptr); } } void MapWidgetImpl::SetRadarSite(const std::string& radarSite) { // Check if radar site has changed if (radarProductManager_ == nullptr || radarSite != radarProductManager_->radar_site()->id()) { // Disconnect signals from old RadarProductManager RadarProductManagerDisconnect(); // Set new RadarProductManager radarProductManager_ = manager::RadarProductManager::Instance(radarSite); // Connect signals to new RadarProductManager RadarProductManagerConnect(); radarProductManager_->UpdateAvailableProducts(); } } void MapWidgetImpl::Update() { widget_->update(); if (UpdateStoredMapParameters()) { emit widget_->MapParametersChanged( prevLatitude_, prevLongitude_, prevZoom_, prevBearing_, prevPitch_); } } bool MapWidgetImpl::UpdateStoredMapParameters() { bool changed = false; double newLatitude = map_->latitude(); double newLongitude = map_->longitude(); double newZoom = map_->zoom(); double newBearing = map_->bearing(); double newPitch = map_->pitch(); if (prevLatitude_ != newLatitude || // prevLongitude_ != newLongitude || // prevZoom_ != newZoom || // prevBearing_ != newBearing || // prevPitch_ != newPitch) { prevLatitude_ = newLatitude; prevLongitude_ = newLongitude; prevZoom_ = newZoom; prevBearing_ = newBearing; prevPitch_ = newPitch; changed = true; } return changed; } } // namespace map } // namespace qt } // namespace scwx #include "map_widget.moc"