mirror of
				https://github.com/ciphervance/supercell-wx.git
				synced 2025-10-31 10:20:04 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			1122 lines
		
	
	
	
		
			33 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			1122 lines
		
	
	
	
		
			33 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| #include <scwx/qt/model/layer_model.hpp>
 | |
| #include <scwx/qt/manager/placefile_manager.hpp>
 | |
| #include <scwx/qt/types/layer_types.hpp>
 | |
| #include <scwx/qt/types/qt_types.hpp>
 | |
| #include <scwx/qt/util/json.hpp>
 | |
| #include <scwx/util/logger.hpp>
 | |
| 
 | |
| #include <filesystem>
 | |
| #include <set>
 | |
| #include <variant>
 | |
| 
 | |
| #include <QApplication>
 | |
| #include <QCheckBox>
 | |
| #include <QFontMetrics>
 | |
| #include <QIODevice>
 | |
| #include <QMimeData>
 | |
| #include <QStyle>
 | |
| #include <QStyleOption>
 | |
| #include <QStandardPaths>
 | |
| #include <boost/container/stable_vector.hpp>
 | |
| #include <boost/json.hpp>
 | |
| 
 | |
| namespace scwx
 | |
| {
 | |
| namespace qt
 | |
| {
 | |
| namespace model
 | |
| {
 | |
| 
 | |
| static const std::string logPrefix_ = "scwx::qt::model::layer_model";
 | |
| static const auto        logger_    = scwx::util::Logger::Create(logPrefix_);
 | |
| 
 | |
| static constexpr int kFirstColumn = static_cast<int>(LayerModel::Column::Order);
 | |
| static constexpr int kLastColumn =
 | |
|    static_cast<int>(LayerModel::Column::Description);
 | |
| static constexpr int kNumColumns = kLastColumn - kFirstColumn + 1;
 | |
| 
 | |
| static constexpr std::size_t kMapCount_ = 4u;
 | |
| 
 | |
| static const QString kMimeFormat {"application/x.scwx-layer-model"};
 | |
| 
 | |
| static const std::string kTypeName_ {"type"};
 | |
| static const std::string kDescriptionName_ {"description"};
 | |
| static const std::string kMovableName_ {"movable"};
 | |
| static const std::string kDisplayedName_ {"displayed"};
 | |
| 
 | |
| typedef std::variant<std::monostate,
 | |
|                      types::DataLayer,
 | |
|                      types::InformationLayer,
 | |
|                      types::MapLayer,
 | |
|                      awips::Phenomenon,
 | |
|                      std::string>
 | |
|    LayerDescription;
 | |
| 
 | |
| struct LayerInfo
 | |
| {
 | |
|    types::LayerType             type_;
 | |
|    LayerDescription             description_;
 | |
|    bool                         movable_ {true};
 | |
|    std::array<bool, kMapCount_> displayed_ {true, true, true, true};
 | |
| };
 | |
| 
 | |
| typedef boost::container::stable_vector<LayerInfo> LayerVector;
 | |
| 
 | |
| static const std::vector<LayerInfo> kDefaultLayers_ {
 | |
|    {types::LayerType::Information, types::InformationLayer::MapOverlay, false},
 | |
|    {types::LayerType::Information, types::InformationLayer::ColorTable, false},
 | |
|    {types::LayerType::Data, types::DataLayer::RadarRange, true},
 | |
|    {types::LayerType::Alert, awips::Phenomenon::Tornado, true},
 | |
|    {types::LayerType::Alert, awips::Phenomenon::SnowSquall, true},
 | |
|    {types::LayerType::Alert, awips::Phenomenon::SevereThunderstorm, true},
 | |
|    {types::LayerType::Alert, awips::Phenomenon::FlashFlood, true},
 | |
|    {types::LayerType::Alert, awips::Phenomenon::Marine, true},
 | |
|    {types::LayerType::Map, types::MapLayer::MapSymbology, false},
 | |
|    {types::LayerType::Radar, std::monostate {}, true},
 | |
|    {types::LayerType::Map, types::MapLayer::MapUnderlay, false},
 | |
| };
 | |
| 
 | |
| static const std::vector<LayerInfo> kImmovableLayers_ {
 | |
|    {types::LayerType::Information, types::InformationLayer::MapOverlay, false},
 | |
|    {types::LayerType::Information, types::InformationLayer::ColorTable, false},
 | |
|    {types::LayerType::Map, types::MapLayer::MapSymbology, false},
 | |
|    {types::LayerType::Map, types::MapLayer::MapUnderlay, false},
 | |
| };
 | |
| 
 | |
| static const std::array<awips::Phenomenon, 5> kAlertPhenomena_ {
 | |
|    awips::Phenomenon::Tornado,
 | |
|    awips::Phenomenon::SnowSquall,
 | |
|    awips::Phenomenon::SevereThunderstorm,
 | |
|    awips::Phenomenon::FlashFlood,
 | |
|    awips::Phenomenon::Marine};
 | |
| 
 | |
| class LayerModel::Impl
 | |
| {
 | |
| public:
 | |
|    explicit Impl(LayerModel* self) : self_ {self} {}
 | |
|    ~Impl() = default;
 | |
| 
 | |
|    void AddPlacefile(const std::string& name);
 | |
|    void InitializeLayerSettings();
 | |
|    void ReadLayerSettings();
 | |
|    void WriteLayerSettings();
 | |
| 
 | |
|    static void ValidateLayerSettings(LayerVector& layers);
 | |
| 
 | |
|    LayerModel* self_;
 | |
| 
 | |
|    std::string layerSettingsPath_ {};
 | |
| 
 | |
|    std::shared_ptr<manager::PlacefileManager> placefileManager_ {
 | |
|       manager::PlacefileManager::Instance()};
 | |
| 
 | |
|    LayerVector layers_ {};
 | |
| };
 | |
| 
 | |
| LayerModel::LayerModel(QObject* parent) :
 | |
|     QAbstractTableModel(parent), p(std::make_unique<Impl>(this))
 | |
| {
 | |
|    connect(p->placefileManager_.get(),
 | |
|            &manager::PlacefileManager::PlacefileEnabled,
 | |
|            this,
 | |
|            &LayerModel::HandlePlacefileUpdate);
 | |
| 
 | |
|    connect(p->placefileManager_.get(),
 | |
|            &manager::PlacefileManager::PlacefileRemoved,
 | |
|            this,
 | |
|            &LayerModel::HandlePlacefileRemoved);
 | |
| 
 | |
|    connect(p->placefileManager_.get(),
 | |
|            &manager::PlacefileManager::PlacefileRenamed,
 | |
|            this,
 | |
|            &LayerModel::HandlePlacefileRenamed);
 | |
| 
 | |
|    connect(p->placefileManager_.get(),
 | |
|            &manager::PlacefileManager::PlacefileUpdated,
 | |
|            this,
 | |
|            &LayerModel::HandlePlacefileUpdate);
 | |
| 
 | |
|    p->InitializeLayerSettings();
 | |
|    p->ReadLayerSettings();
 | |
| 
 | |
|    if (p->layers_.empty())
 | |
|    {
 | |
|       p->layers_.assign(kDefaultLayers_.cbegin(), kDefaultLayers_.cend());
 | |
|    }
 | |
| }
 | |
| 
 | |
| LayerModel::~LayerModel()
 | |
| {
 | |
|    // Write layer settings on shutdown
 | |
|    p->WriteLayerSettings();
 | |
| };
 | |
| 
 | |
| void LayerModel::Impl::InitializeLayerSettings()
 | |
| {
 | |
|    std::string appDataPath {
 | |
|       QStandardPaths::writableLocation(QStandardPaths::AppDataLocation)
 | |
|          .toStdString()};
 | |
| 
 | |
|    if (!std::filesystem::exists(appDataPath))
 | |
|    {
 | |
|       if (!std::filesystem::create_directories(appDataPath))
 | |
|       {
 | |
|          logger_->error("Unable to create application data directory: \"{}\"",
 | |
|                         appDataPath);
 | |
|       }
 | |
|    }
 | |
| 
 | |
|    layerSettingsPath_ = appDataPath + "/layers.json";
 | |
| }
 | |
| 
 | |
| void LayerModel::Impl::ReadLayerSettings()
 | |
| {
 | |
|    logger_->info("Reading layer settings");
 | |
| 
 | |
|    boost::json::value layerJson = nullptr;
 | |
|    LayerVector        newLayers {};
 | |
| 
 | |
|    // Determine if layer settings exists
 | |
|    if (std::filesystem::exists(layerSettingsPath_))
 | |
|    {
 | |
|       layerJson = util::json::ReadJsonFile(layerSettingsPath_);
 | |
|    }
 | |
| 
 | |
|    // If layer settings was successfully read
 | |
|    if (layerJson != nullptr && layerJson.is_array())
 | |
|    {
 | |
|       // For each layer entry
 | |
|       auto& layerArray = layerJson.as_array();
 | |
|       for (auto& layerEntry : layerArray)
 | |
|       {
 | |
|          try
 | |
|          {
 | |
|             // Convert layer entry to a LayerInfo record, and add to new layers
 | |
|             newLayers.emplace_back(
 | |
|                boost::json::value_to<LayerInfo>(layerEntry));
 | |
|          }
 | |
|          catch (const std::exception& ex)
 | |
|          {
 | |
|             logger_->warn("Invalid layer entry: {}", ex.what());
 | |
|          }
 | |
|       }
 | |
| 
 | |
|       // Validate and correct read layers
 | |
|       ValidateLayerSettings(newLayers);
 | |
| 
 | |
|       // Assign read layers
 | |
|       layers_.swap(newLayers);
 | |
|    }
 | |
| }
 | |
| 
 | |
| void LayerModel::Impl::ValidateLayerSettings(LayerVector& layers)
 | |
| {
 | |
|    // Validate layer properties
 | |
|    for (auto it = layers.begin(); it != layers.end();)
 | |
|    {
 | |
|       // If the layer is invalid, remove it
 | |
|       if (it->type_ == types::LayerType::Unknown ||
 | |
|           (std::holds_alternative<types::DataLayer>(it->description_) &&
 | |
|            std::get<types::DataLayer>(it->description_) ==
 | |
|               types::DataLayer::Unknown) ||
 | |
|           (std::holds_alternative<types::InformationLayer>(it->description_) &&
 | |
|            std::get<types::InformationLayer>(it->description_) ==
 | |
|               types::InformationLayer::Unknown) ||
 | |
|           (std::holds_alternative<types::MapLayer>(it->description_) &&
 | |
|            std::get<types::MapLayer>(it->description_) ==
 | |
|               types::MapLayer::Unknown) ||
 | |
|           (std::holds_alternative<awips::Phenomenon>(it->description_) &&
 | |
|            std::get<awips::Phenomenon>(it->description_) ==
 | |
|               awips::Phenomenon::Unknown))
 | |
|       {
 | |
|          // Erase the current layer and continue
 | |
|          it = layers.erase(it);
 | |
|          continue;
 | |
|       }
 | |
| 
 | |
|       // Ensure layers are appropriately marked movable
 | |
|       it->movable_ = (it->type_ != types::LayerType::Information &&
 | |
|                       it->type_ != types::LayerType::Map);
 | |
| 
 | |
|       // Continue to the next layer
 | |
|       ++it;
 | |
|    }
 | |
| 
 | |
|    // Validate immovable layers
 | |
|    std::vector<LayerVector::iterator> immovableIterators {};
 | |
|    LayerVector::iterator              colorTableIterator {};
 | |
|    LayerVector::iterator              mapSymbologyIterator {};
 | |
|    LayerVector::iterator              mapUnderlayIterator {};
 | |
|    for (auto& immovableLayer : kImmovableLayers_)
 | |
|    {
 | |
|       // Set the default displayed state for a layer that is not found
 | |
|       std::array<bool, kMapCount_> displayed {true, true, true, true};
 | |
| 
 | |
|       // Find the immovable layer
 | |
|       auto it = std::find_if(layers.begin(),
 | |
|                              layers.end(),
 | |
|                              [&immovableLayer](const LayerInfo& layer)
 | |
|                              {
 | |
|                                 return layer.type_ == immovableLayer.type_ &&
 | |
|                                        layer.description_ ==
 | |
|                                           immovableLayer.description_;
 | |
|                              });
 | |
| 
 | |
|       // If the immovable layer is out of order
 | |
|       if (!immovableIterators.empty() && immovableIterators.back() > it)
 | |
|       {
 | |
|          // Save the displayed state of the immovable layer
 | |
|          displayed = it->displayed_;
 | |
| 
 | |
|          // Remove the layer from the list, to re-add it later
 | |
|          layers.erase(it);
 | |
| 
 | |
|          // Treat the layer as not found
 | |
|          it = layers.end();
 | |
|       }
 | |
| 
 | |
|       // If the immovable layer is not found
 | |
|       if (it == layers.end())
 | |
|       {
 | |
|          // If this is the first immovable layer, insert at the beginning,
 | |
|          // otherwise, insert after the previous immovable layer
 | |
|          LayerVector::iterator insertPosition =
 | |
|             immovableIterators.empty() ? layers.begin() :
 | |
|                                          immovableIterators.back() + 1;
 | |
|          it = layers.insert(insertPosition, immovableLayer);
 | |
| 
 | |
|          // Restore the displayed state of the immovable layer
 | |
|          it->displayed_ = displayed;
 | |
|       }
 | |
| 
 | |
|       // Store positional iterators
 | |
|       if (it->type_ == types::LayerType::Information)
 | |
|       {
 | |
|          switch (std::get<types::InformationLayer>(it->description_))
 | |
|          {
 | |
|          case types::InformationLayer::ColorTable:
 | |
|             colorTableIterator = it;
 | |
|             break;
 | |
| 
 | |
|          default:
 | |
|             break;
 | |
|          }
 | |
|       }
 | |
|       else if (it->type_ == types::LayerType::Map)
 | |
|       {
 | |
|          switch (std::get<types::MapLayer>(it->description_))
 | |
|          {
 | |
|          case types::MapLayer::MapSymbology:
 | |
|             mapSymbologyIterator = it;
 | |
|             break;
 | |
| 
 | |
|          case types::MapLayer::MapUnderlay:
 | |
|             mapUnderlayIterator = it;
 | |
|             break;
 | |
| 
 | |
|          default:
 | |
|             break;
 | |
|          }
 | |
|       }
 | |
| 
 | |
|       // Add the immovable iterator to the list
 | |
|       immovableIterators.push_back(it);
 | |
|    }
 | |
| 
 | |
|    // Validate data layers
 | |
|    std::vector<LayerVector::iterator> dataIterators {};
 | |
|    for (const auto& dataLayer : types::DataLayerIterator())
 | |
|    {
 | |
|       // Find the data layer
 | |
|       auto it = std::find_if(layers.begin(),
 | |
|                              layers.end(),
 | |
|                              [&dataLayer](const LayerInfo& layer)
 | |
|                              {
 | |
|                                 return layer.type_ == types::LayerType::Data &&
 | |
|                                        std::get<types::DataLayer>(
 | |
|                                           layer.description_) == dataLayer;
 | |
|                              });
 | |
| 
 | |
|       if (it == layers.end())
 | |
|       {
 | |
|          // If this is the first data layer, insert after the color table layer,
 | |
|          // otherwise, insert after the previous data layer
 | |
|          LayerVector::iterator insertPosition = dataIterators.empty() ?
 | |
|                                                    colorTableIterator + 1 :
 | |
|                                                    dataIterators.back() + 1;
 | |
|          it =
 | |
|             layers.insert(insertPosition, {types::LayerType::Data, dataLayer});
 | |
|       }
 | |
| 
 | |
|       dataIterators.push_back(it);
 | |
|    }
 | |
| 
 | |
|    // Validate alert layers
 | |
|    std::vector<LayerVector::iterator> alertIterators {};
 | |
|    for (auto& phenomenon : kAlertPhenomena_)
 | |
|    {
 | |
|       // Find the alert layer
 | |
|       auto it = std::find_if(layers.begin(),
 | |
|                              layers.end(),
 | |
|                              [&phenomenon](const LayerInfo& layer)
 | |
|                              {
 | |
|                                 return layer.type_ == types::LayerType::Alert &&
 | |
|                                        std::get<awips::Phenomenon>(
 | |
|                                           layer.description_) == phenomenon;
 | |
|                              });
 | |
| 
 | |
|       if (it == layers.end())
 | |
|       {
 | |
|          // Insert before the map symbology layer
 | |
|          it = layers.insert(mapSymbologyIterator,
 | |
|                             {types::LayerType::Alert, phenomenon});
 | |
|       }
 | |
| 
 | |
|       alertIterators.push_back(it);
 | |
|    }
 | |
| 
 | |
|    // Validate the radar layer
 | |
|    auto it = std::find_if(layers.begin(),
 | |
|                           layers.end(),
 | |
|                           [](const LayerInfo& layer)
 | |
|                           { return layer.type_ == types::LayerType::Radar; });
 | |
|    if (it == layers.end())
 | |
|    {
 | |
|       // Insert before the map underlay layer
 | |
|       it = layers.insert(mapUnderlayIterator,
 | |
|                          {types::LayerType::Radar, std::monostate {}});
 | |
|    }
 | |
| }
 | |
| 
 | |
| void LayerModel::Impl::WriteLayerSettings()
 | |
| {
 | |
|    logger_->info("Saving layer settings");
 | |
| 
 | |
|    auto layerJson = boost::json::value_from(layers_);
 | |
|    util::json::WriteJsonFile(layerSettingsPath_, layerJson);
 | |
| }
 | |
| 
 | |
| void LayerModel::ResetLayers()
 | |
| {
 | |
|    // Initialize a new layer vector from the default
 | |
|    LayerVector newLayers {};
 | |
|    newLayers.assign(kDefaultLayers_.cbegin(), kDefaultLayers_.cend());
 | |
| 
 | |
|    auto colorTableIterator = std::find_if(
 | |
|       newLayers.begin(),
 | |
|       newLayers.end(),
 | |
|       [](const LayerInfo& layerInfo)
 | |
|       {
 | |
|          return std::holds_alternative<types::InformationLayer>(
 | |
|                    layerInfo.description_) &&
 | |
|                 std::get<types::InformationLayer>(layerInfo.description_) ==
 | |
|                    types::InformationLayer::ColorTable;
 | |
|       });
 | |
| 
 | |
|    // Add all existing placefile layers
 | |
|    for (auto it = p->layers_.rbegin(); it != p->layers_.rend(); ++it)
 | |
|    {
 | |
|       if (it->type_ == types::LayerType::Placefile)
 | |
|       {
 | |
|          newLayers.insert(
 | |
|             colorTableIterator + 1,
 | |
|             {it->type_, it->description_, it->movable_, it->displayed_});
 | |
|       }
 | |
|    }
 | |
| 
 | |
|    // Swap the model
 | |
|    beginResetModel();
 | |
|    p->layers_.swap(newLayers);
 | |
|    endResetModel();
 | |
| }
 | |
| 
 | |
| int LayerModel::rowCount(const QModelIndex& parent) const
 | |
| {
 | |
|    return parent.isValid() ? 0 : static_cast<int>(p->layers_.size());
 | |
| }
 | |
| 
 | |
| int LayerModel::columnCount(const QModelIndex& parent) const
 | |
| {
 | |
|    return parent.isValid() ? 0 : kNumColumns;
 | |
| }
 | |
| 
 | |
| Qt::ItemFlags LayerModel::flags(const QModelIndex& index) const
 | |
| {
 | |
|    Qt::ItemFlags flags = QAbstractTableModel::flags(index);
 | |
| 
 | |
|    if (!index.isValid() || index.row() < 0 ||
 | |
|        static_cast<std::size_t>(index.row()) >= p->layers_.size())
 | |
|    {
 | |
|       return flags;
 | |
|    }
 | |
| 
 | |
|    const auto& layer = p->layers_.at(index.row());
 | |
| 
 | |
|    switch (index.column())
 | |
|    {
 | |
|    case static_cast<int>(Column::DisplayMap1):
 | |
|    case static_cast<int>(Column::DisplayMap2):
 | |
|    case static_cast<int>(Column::DisplayMap3):
 | |
|    case static_cast<int>(Column::DisplayMap4):
 | |
|       if (layer.type_ != types::LayerType::Map)
 | |
|       {
 | |
|          flags |=
 | |
|             Qt::ItemFlag::ItemIsUserCheckable | Qt::ItemFlag::ItemIsEditable;
 | |
|       }
 | |
|       break;
 | |
| 
 | |
|    default:
 | |
|       break;
 | |
|    }
 | |
| 
 | |
|    if (layer.movable_)
 | |
|    {
 | |
|       flags |= Qt::ItemFlag::ItemIsDragEnabled;
 | |
|    }
 | |
| 
 | |
|    flags |= Qt::ItemFlag::ItemIsDropEnabled;
 | |
| 
 | |
|    return flags;
 | |
| }
 | |
| 
 | |
| Qt::DropActions LayerModel::supportedDropActions() const
 | |
| {
 | |
|    return Qt::DropAction::MoveAction;
 | |
| }
 | |
| 
 | |
| bool LayerModel::IsMovable(int row) const
 | |
| {
 | |
|    bool movable = false;
 | |
| 
 | |
|    if (0 <= row && static_cast<std::size_t>(row) < p->layers_.size())
 | |
|    {
 | |
|       movable = p->layers_.at(row).movable_;
 | |
|    }
 | |
| 
 | |
|    return movable;
 | |
| }
 | |
| 
 | |
| QVariant LayerModel::data(const QModelIndex& index, int role) const
 | |
| {
 | |
|    static const QString enabledString  = QObject::tr("Enabled");
 | |
|    static const QString disabledString = QObject::tr("Disabled");
 | |
| 
 | |
|    static const QString displayedString = QObject::tr("Displayed");
 | |
|    static const QString hiddenString    = QObject::tr("Hidden");
 | |
| 
 | |
|    if (!index.isValid() || index.row() < 0 ||
 | |
|        static_cast<std::size_t>(index.row()) >= p->layers_.size())
 | |
|    {
 | |
|       return QVariant();
 | |
|    }
 | |
| 
 | |
|    const auto& layer = p->layers_.at(index.row());
 | |
| 
 | |
|    switch (index.column())
 | |
|    {
 | |
|    case static_cast<int>(Column::Order):
 | |
|       if (role == Qt::ItemDataRole::DisplayRole)
 | |
|       {
 | |
|          return index.row() + 1;
 | |
|       }
 | |
|       break;
 | |
| 
 | |
|    case static_cast<int>(Column::DisplayMap1):
 | |
|    case static_cast<int>(Column::DisplayMap2):
 | |
|    case static_cast<int>(Column::DisplayMap3):
 | |
|    case static_cast<int>(Column::DisplayMap4):
 | |
|       if (layer.type_ != types::LayerType::Map)
 | |
|       {
 | |
|          bool displayed =
 | |
|             layer.displayed_[index.column() -
 | |
|                              static_cast<int>(Column::DisplayMap1)];
 | |
| 
 | |
|          if (role == Qt::ItemDataRole::ToolTipRole)
 | |
|          {
 | |
|             return displayed ? displayedString : hiddenString;
 | |
|          }
 | |
|          else if (role == Qt::ItemDataRole::CheckStateRole)
 | |
|          {
 | |
|             return static_cast<int>(displayed ? Qt::CheckState::Checked :
 | |
|                                                 Qt::CheckState::Unchecked);
 | |
|          }
 | |
|       }
 | |
|       break;
 | |
| 
 | |
|    case static_cast<int>(Column::Type):
 | |
|       if (role == Qt::ItemDataRole::DisplayRole ||
 | |
|           role == Qt::ItemDataRole::ToolTipRole)
 | |
|       {
 | |
|          return QString::fromStdString(types::GetLayerTypeName(layer.type_));
 | |
|       }
 | |
|       break;
 | |
| 
 | |
|    case static_cast<int>(Column::Enabled):
 | |
|       if (role == Qt::ItemDataRole::DisplayRole ||
 | |
|           role == Qt::ItemDataRole::ToolTipRole)
 | |
|       {
 | |
|          if (layer.type_ == types::LayerType::Placefile)
 | |
|          {
 | |
|             return p->placefileManager_->placefile_enabled(
 | |
|                       std::get<std::string>(layer.description_)) ?
 | |
|                       enabledString :
 | |
|                       disabledString;
 | |
|          }
 | |
|       }
 | |
|       break;
 | |
| 
 | |
|    case static_cast<int>(Column::Description):
 | |
|       if (role == Qt::ItemDataRole::DisplayRole ||
 | |
|           role == Qt::ItemDataRole::ToolTipRole)
 | |
|       {
 | |
|          if (layer.type_ == types::LayerType::Placefile)
 | |
|          {
 | |
|             std::string placefileName =
 | |
|                std::get<std::string>(layer.description_);
 | |
|             std::string description = placefileName;
 | |
|             std::string title =
 | |
|                p->placefileManager_->placefile_title(placefileName);
 | |
|             if (!title.empty())
 | |
|             {
 | |
|                description = title + '\n' + description;
 | |
|             }
 | |
| 
 | |
|             return QString::fromStdString(description);
 | |
|          }
 | |
|          else
 | |
|          {
 | |
|             if (std::holds_alternative<std::string>(layer.description_))
 | |
|             {
 | |
|                return QString::fromStdString(
 | |
|                   std::get<std::string>(layer.description_));
 | |
|             }
 | |
|             else if (std::holds_alternative<types::DataLayer>(
 | |
|                         layer.description_))
 | |
|             {
 | |
|                return QString::fromStdString(types::GetDataLayerName(
 | |
|                   std::get<types::DataLayer>(layer.description_)));
 | |
|             }
 | |
|             else if (std::holds_alternative<types::InformationLayer>(
 | |
|                         layer.description_))
 | |
|             {
 | |
|                return QString::fromStdString(types::GetInformationLayerName(
 | |
|                   std::get<types::InformationLayer>(layer.description_)));
 | |
|             }
 | |
|             else if (std::holds_alternative<types::MapLayer>(
 | |
|                         layer.description_))
 | |
|             {
 | |
|                return QString::fromStdString(types::GetMapLayerName(
 | |
|                   std::get<types::MapLayer>(layer.description_)));
 | |
|             }
 | |
|             else if (std::holds_alternative<awips::Phenomenon>(
 | |
|                         layer.description_))
 | |
|             {
 | |
|                return QString::fromStdString(awips::GetPhenomenonText(
 | |
|                   std::get<awips::Phenomenon>(layer.description_)));
 | |
|             }
 | |
|          }
 | |
|       }
 | |
|       break;
 | |
| 
 | |
|    default:
 | |
|       break;
 | |
|    }
 | |
| 
 | |
|    return QVariant();
 | |
| }
 | |
| 
 | |
| QVariant
 | |
| LayerModel::headerData(int section, Qt::Orientation orientation, int role) const
 | |
| {
 | |
|    if (role == Qt::ItemDataRole::DisplayRole)
 | |
|    {
 | |
|       if (orientation == Qt::Horizontal)
 | |
|       {
 | |
|          switch (section)
 | |
|          {
 | |
|          case static_cast<int>(Column::DisplayMap1):
 | |
|             return tr("1");
 | |
| 
 | |
|          case static_cast<int>(Column::DisplayMap2):
 | |
|             return tr("2");
 | |
| 
 | |
|          case static_cast<int>(Column::DisplayMap3):
 | |
|             return tr("3");
 | |
| 
 | |
|          case static_cast<int>(Column::DisplayMap4):
 | |
|             return tr("4");
 | |
| 
 | |
|          case static_cast<int>(Column::Type):
 | |
|             return tr("Type");
 | |
| 
 | |
|          case static_cast<int>(Column::Enabled):
 | |
|             return tr("Enabled");
 | |
| 
 | |
|          case static_cast<int>(Column::Description):
 | |
|             return tr("Description");
 | |
| 
 | |
|          default:
 | |
|             break;
 | |
|          }
 | |
|       }
 | |
|    }
 | |
|    else if (role == Qt::ItemDataRole::ToolTipRole)
 | |
|    {
 | |
|       switch (section)
 | |
|       {
 | |
|       case static_cast<int>(Column::Order):
 | |
|          return tr("Order");
 | |
| 
 | |
|       case static_cast<int>(Column::DisplayMap1):
 | |
|          return tr("Display on Map 1");
 | |
| 
 | |
|       case static_cast<int>(Column::DisplayMap2):
 | |
|          return tr("Display on Map 2");
 | |
| 
 | |
|       case static_cast<int>(Column::DisplayMap3):
 | |
|          return tr("Display on Map 3");
 | |
| 
 | |
|       case static_cast<int>(Column::DisplayMap4):
 | |
|          return tr("Display on Map 4");
 | |
| 
 | |
|       default:
 | |
|          break;
 | |
|       }
 | |
|    }
 | |
|    else if (role == Qt::ItemDataRole::SizeHintRole)
 | |
|    {
 | |
|       switch (section)
 | |
|       {
 | |
|       case static_cast<int>(Column::DisplayMap1):
 | |
|       case static_cast<int>(Column::DisplayMap2):
 | |
|       case static_cast<int>(Column::DisplayMap3):
 | |
|       case static_cast<int>(Column::DisplayMap4):
 | |
|       {
 | |
|          static const QCheckBox checkBox {};
 | |
|          QStyleOptionButton     option {};
 | |
|          option.initFrom(&checkBox);
 | |
| 
 | |
|          // Width values from QCheckBox
 | |
|          return QApplication::style()->sizeFromContents(
 | |
|             QStyle::ContentsType::CT_CheckBox,
 | |
|             &option,
 | |
|             {option.iconSize.width() + 4, 0});
 | |
|       }
 | |
| 
 | |
|       default:
 | |
|          break;
 | |
|       }
 | |
|    }
 | |
| 
 | |
|    return QVariant();
 | |
| }
 | |
| 
 | |
| bool LayerModel::setData(const QModelIndex& index,
 | |
|                          const QVariant&    value,
 | |
|                          int                role)
 | |
| {
 | |
|    if (!index.isValid() || index.row() < 0 ||
 | |
|        static_cast<std::size_t>(index.row()) >= p->layers_.size())
 | |
|    {
 | |
|       return false;
 | |
|    }
 | |
| 
 | |
|    auto& layer  = p->layers_.at(index.row());
 | |
|    bool  result = false;
 | |
| 
 | |
|    switch (index.column())
 | |
|    {
 | |
|    case static_cast<int>(Column::DisplayMap1):
 | |
|    case static_cast<int>(Column::DisplayMap2):
 | |
|    case static_cast<int>(Column::DisplayMap3):
 | |
|    case static_cast<int>(Column::DisplayMap4):
 | |
|       if (role == Qt::ItemDataRole::CheckStateRole)
 | |
|       {
 | |
|          layer.displayed_[index.column() -
 | |
|                           static_cast<int>(Column::DisplayMap1)] =
 | |
|             value.toBool();
 | |
|          result = true;
 | |
|       }
 | |
|       break;
 | |
| 
 | |
|    default:
 | |
|       break;
 | |
|    }
 | |
| 
 | |
|    if (result)
 | |
|    {
 | |
|       Q_EMIT dataChanged(index, index);
 | |
|    }
 | |
| 
 | |
|    return result;
 | |
| }
 | |
| 
 | |
| QStringList LayerModel::mimeTypes() const
 | |
| {
 | |
|    return {kMimeFormat};
 | |
| }
 | |
| 
 | |
| QMimeData* LayerModel::mimeData(const QModelIndexList& indexes) const
 | |
| {
 | |
|    // Get parent QMimeData
 | |
|    QMimeData* mimeData = QAbstractTableModel::mimeData(indexes);
 | |
| 
 | |
|    // Generate LayerModel data
 | |
|    QByteArray    data {};
 | |
|    QDataStream   stream(&data, QIODevice::WriteOnly);
 | |
|    std::set<int> rows {};
 | |
| 
 | |
|    for (auto& index : indexes)
 | |
|    {
 | |
|       if (!rows.contains(index.row()))
 | |
|       {
 | |
|          rows.insert(index.row());
 | |
|          stream << index.row();
 | |
|       }
 | |
|    }
 | |
| 
 | |
|    // Set LayerModel data in QMimeData
 | |
|    mimeData->setData(kMimeFormat, data);
 | |
| 
 | |
|    return mimeData;
 | |
| }
 | |
| 
 | |
| bool LayerModel::dropMimeData(const QMimeData* data,
 | |
|                               Qt::DropAction /* action */,
 | |
|                               int /* row */,
 | |
|                               int /* column */,
 | |
|                               const QModelIndex& parent)
 | |
| {
 | |
|    QByteArray       mimeData = data->data(kMimeFormat);
 | |
|    QDataStream      stream(&mimeData, QIODevice::ReadOnly);
 | |
|    std::vector<int> sourceRows {};
 | |
| 
 | |
|    // Read source rows from QMimeData
 | |
|    while (!stream.atEnd())
 | |
|    {
 | |
|       int sourceRow;
 | |
|       stream >> sourceRow;
 | |
|       sourceRows.push_back(sourceRow);
 | |
|    }
 | |
| 
 | |
|    // Ensure rows are in numerical order
 | |
|    std::sort(sourceRows.begin(), sourceRows.end());
 | |
| 
 | |
|    if (sourceRows.back() >= p->layers_.size())
 | |
|    {
 | |
|       logger_->error("Cannot perform drop action, invalid source rows");
 | |
|       return false;
 | |
|    }
 | |
| 
 | |
|    // Nothing to insert
 | |
|    if (sourceRows.empty())
 | |
|    {
 | |
|       return false;
 | |
|    }
 | |
| 
 | |
|    // Create a copy of the layers to insert (don't insert in-place)
 | |
|    std::vector<LayerInfo> newLayers {};
 | |
|    for (auto& sourceRow : sourceRows)
 | |
|    {
 | |
|       newLayers.push_back(p->layers_.at(sourceRow));
 | |
|    }
 | |
| 
 | |
|    // Insert the copied layers
 | |
|    auto insertPosition = p->layers_.begin() + parent.row();
 | |
|    beginInsertRows(QModelIndex(),
 | |
|                    parent.row(),
 | |
|                    parent.row() + static_cast<int>(sourceRows.size()) - 1);
 | |
|    p->layers_.insert(insertPosition, newLayers.begin(), newLayers.end());
 | |
|    endInsertRows();
 | |
| 
 | |
|    return true;
 | |
| }
 | |
| 
 | |
| bool LayerModel::removeRows(int row, int count, const QModelIndex& parent)
 | |
| {
 | |
|    // Validate count
 | |
|    if (count <= 0)
 | |
|    {
 | |
|       return false;
 | |
|    }
 | |
| 
 | |
|    // Remove rows
 | |
|    auto erasePosition = p->layers_.begin() + row;
 | |
|    for (int i = 0; i < count; ++i)
 | |
|    {
 | |
|       if (erasePosition->movable_)
 | |
|       {
 | |
|          // Remove the current row if movable
 | |
|          beginRemoveRows(parent, row, row);
 | |
|          erasePosition = p->layers_.erase(erasePosition);
 | |
|          endRemoveRows();
 | |
|       }
 | |
|       else
 | |
|       {
 | |
|          // Don't remove immovable rows
 | |
|          ++erasePosition;
 | |
|          ++row;
 | |
|       }
 | |
|    }
 | |
| 
 | |
|    return true;
 | |
| }
 | |
| 
 | |
| bool LayerModel::moveRows(const QModelIndex& sourceParent,
 | |
|                           int                sourceRow,
 | |
|                           int                count,
 | |
|                           const QModelIndex& destinationParent,
 | |
|                           int                destinationChild)
 | |
| {
 | |
|    bool moved = false;
 | |
| 
 | |
|    if (sourceParent != destinationParent ||     // Only accept internal moves
 | |
|        count < 1 ||                             // Minimum selection size of 1
 | |
|        sourceRow < 0 ||                         // Valid source row (start)
 | |
|        sourceRow + count > p->layers_.size() || // Valid source row (end)
 | |
|        destinationChild < 0 ||                  // Valid destination row
 | |
|        destinationChild > p->layers_.size())
 | |
|    {
 | |
|       return false;
 | |
|    }
 | |
| 
 | |
|    if (destinationChild < sourceRow)
 | |
|    {
 | |
|       // Move up
 | |
|       auto first  = p->layers_.begin() + destinationChild;
 | |
|       auto middle = p->layers_.begin() + sourceRow;
 | |
|       auto last   = middle + count;
 | |
| 
 | |
|       beginMoveRows(sourceParent,
 | |
|                     sourceRow,
 | |
|                     sourceRow + count - 1,
 | |
|                     destinationParent,
 | |
|                     destinationChild);
 | |
|       std::rotate(first, middle, last);
 | |
|       endMoveRows();
 | |
| 
 | |
|       moved = true;
 | |
|    }
 | |
|    else if (sourceRow + count < destinationChild)
 | |
|    {
 | |
|       // Move down
 | |
|       auto first  = p->layers_.begin() + sourceRow;
 | |
|       auto middle = first + count;
 | |
|       auto last   = p->layers_.begin() + destinationChild;
 | |
| 
 | |
|       beginMoveRows(sourceParent,
 | |
|                     sourceRow,
 | |
|                     sourceRow + count - 1,
 | |
|                     destinationParent,
 | |
|                     destinationChild);
 | |
|       std::rotate(first, middle, last);
 | |
|       endMoveRows();
 | |
| 
 | |
|       moved = true;
 | |
|    }
 | |
| 
 | |
|    return moved;
 | |
| }
 | |
| 
 | |
| void LayerModel::HandlePlacefileRemoved(const std::string& name)
 | |
| {
 | |
|    auto it =
 | |
|       std::find_if(p->layers_.begin(),
 | |
|                    p->layers_.end(),
 | |
|                    [&name](const auto& layer)
 | |
|                    {
 | |
|                       return layer.type_ == types::LayerType::Placefile &&
 | |
|                              std::get<std::string>(layer.description_) == name;
 | |
|                    });
 | |
| 
 | |
|    if (it != p->layers_.end())
 | |
|    {
 | |
|       // Placefile exists, delete row
 | |
|       const int row = std::distance(p->layers_.begin(), it);
 | |
| 
 | |
|       beginRemoveRows(QModelIndex(), row, row);
 | |
|       p->layers_.erase(it);
 | |
|       endRemoveRows();
 | |
|    }
 | |
| }
 | |
| 
 | |
| void LayerModel::HandlePlacefileRenamed(const std::string& oldName,
 | |
|                                         const std::string& newName)
 | |
| {
 | |
|    auto it = std::find_if(
 | |
|       p->layers_.begin(),
 | |
|       p->layers_.end(),
 | |
|       [&oldName](const auto& layer)
 | |
|       {
 | |
|          return layer.type_ == types::LayerType::Placefile &&
 | |
|                 std::get<std::string>(layer.description_) == oldName;
 | |
|       });
 | |
| 
 | |
|    if (it != p->layers_.end())
 | |
|    {
 | |
|       // Placefile exists, mark row as updated
 | |
|       const int   row         = std::distance(p->layers_.begin(), it);
 | |
|       QModelIndex topLeft     = createIndex(row, kFirstColumn);
 | |
|       QModelIndex bottomRight = createIndex(row, kLastColumn);
 | |
| 
 | |
|       // Rename placefile
 | |
|       it->description_ = newName;
 | |
| 
 | |
|       Q_EMIT dataChanged(topLeft, bottomRight);
 | |
|    }
 | |
|    else
 | |
|    {
 | |
|       // Placefile doesn't exist, add row
 | |
|       p->AddPlacefile(newName);
 | |
|    }
 | |
| }
 | |
| 
 | |
| void LayerModel::HandlePlacefileUpdate(const std::string& name)
 | |
| {
 | |
|    auto it =
 | |
|       std::find_if(p->layers_.begin(),
 | |
|                    p->layers_.end(),
 | |
|                    [&name](const auto& layer)
 | |
|                    {
 | |
|                       return layer.type_ == types::LayerType::Placefile &&
 | |
|                              std::get<std::string>(layer.description_) == name;
 | |
|                    });
 | |
| 
 | |
|    if (it != p->layers_.end())
 | |
|    {
 | |
|       // Placefile exists, mark row as updated
 | |
|       const int   row         = std::distance(p->layers_.begin(), it);
 | |
|       QModelIndex topLeft     = createIndex(row, kFirstColumn);
 | |
|       QModelIndex bottomRight = createIndex(row, kLastColumn);
 | |
| 
 | |
|       Q_EMIT dataChanged(topLeft, bottomRight);
 | |
|    }
 | |
|    else
 | |
|    {
 | |
|       // Placefile doesn't exist, add row
 | |
|       p->AddPlacefile(name);
 | |
|    }
 | |
| }
 | |
| 
 | |
| void LayerModel::Impl::AddPlacefile(const std::string& name)
 | |
| {
 | |
|    // Insert after color table
 | |
|    auto insertPosition = std::find_if(
 | |
|       layers_.begin(),
 | |
|       layers_.end(),
 | |
|       [](const LayerInfo& layerInfo)
 | |
|       {
 | |
|          return std::holds_alternative<types::InformationLayer>(
 | |
|                    layerInfo.description_) &&
 | |
|                 std::get<types::InformationLayer>(layerInfo.description_) ==
 | |
|                    types::InformationLayer::ColorTable;
 | |
|       });
 | |
|    if (insertPosition != layers_.end())
 | |
|    {
 | |
|       ++insertPosition;
 | |
|    }
 | |
| 
 | |
|    // Placefile is new, add row
 | |
|    self_->beginInsertRows(QModelIndex(), 0, 0);
 | |
|    layers_.insert(insertPosition, {types::LayerType::Placefile, name, true});
 | |
|    self_->endInsertRows();
 | |
| }
 | |
| 
 | |
| void tag_invoke(boost::json::value_from_tag,
 | |
|                 boost::json::value& jv,
 | |
|                 const LayerInfo&    record)
 | |
| {
 | |
|    std::string description {};
 | |
| 
 | |
|    if (std::holds_alternative<awips::Phenomenon>(record.description_))
 | |
|    {
 | |
|       description = awips::GetPhenomenonCode(
 | |
|          std::get<awips::Phenomenon>(record.description_));
 | |
|    }
 | |
|    else if (std::holds_alternative<types::DataLayer>(record.description_))
 | |
|    {
 | |
|       description = types::GetDataLayerName(
 | |
|          std::get<types::DataLayer>(record.description_));
 | |
|    }
 | |
|    else if (std::holds_alternative<types::InformationLayer>(
 | |
|                record.description_))
 | |
|    {
 | |
|       description = types::GetInformationLayerName(
 | |
|          std::get<types::InformationLayer>(record.description_));
 | |
|    }
 | |
|    else if (std::holds_alternative<types::MapLayer>(record.description_))
 | |
|    {
 | |
|       description =
 | |
|          types::GetMapLayerName(std::get<types::MapLayer>(record.description_));
 | |
|    }
 | |
|    else if (std::holds_alternative<std::string>(record.description_))
 | |
|    {
 | |
|       description = std::get<std::string>(record.description_);
 | |
|    }
 | |
| 
 | |
|    jv = {{kTypeName_, types::GetLayerTypeName(record.type_)},
 | |
|          {kDescriptionName_, description},
 | |
|          {kMovableName_, record.movable_},
 | |
|          {kDisplayedName_, boost::json::value_from(record.displayed_)}};
 | |
| }
 | |
| 
 | |
| template<typename T, std::size_t n>
 | |
| std::array<T, n> tag_invoke(boost::json::value_to_tag<std::array<T, n>>,
 | |
|                             const boost::json::value& jv)
 | |
| {
 | |
|    std::array<T, n>   array {};
 | |
|    boost::json::array jsonArray = jv.as_array();
 | |
| 
 | |
|    for (std::size_t i = 0; i < n && i < jsonArray.size(); ++i)
 | |
|    {
 | |
|       array[i] = jsonArray[i];
 | |
|    }
 | |
| 
 | |
|    return array;
 | |
| }
 | |
| 
 | |
| LayerInfo tag_invoke(boost::json::value_to_tag<LayerInfo>,
 | |
|                      const boost::json::value& jv)
 | |
| {
 | |
|    const types::LayerType layerType = types::GetLayerType(
 | |
|       boost::json::value_to<std::string>(jv.at(kTypeName_)));
 | |
|    const std::string descriptionName =
 | |
|       boost::json::value_to<std::string>(jv.at(kDescriptionName_));
 | |
| 
 | |
|    LayerDescription description {};
 | |
| 
 | |
|    if (layerType == types::LayerType::Map)
 | |
|    {
 | |
|       description = types::GetMapLayer(descriptionName);
 | |
|    }
 | |
|    else if (layerType == types::LayerType::Information)
 | |
|    {
 | |
|       description = types::GetInformationLayer(descriptionName);
 | |
|    }
 | |
|    else if (layerType == types::LayerType::Data)
 | |
|    {
 | |
|       description = types::GetDataLayer(descriptionName);
 | |
|    }
 | |
|    else if (layerType == types::LayerType::Radar)
 | |
|    {
 | |
|       description = std::monostate {};
 | |
|    }
 | |
|    else if (layerType == types::LayerType::Alert)
 | |
|    {
 | |
|       description = awips::GetPhenomenon(descriptionName);
 | |
|    }
 | |
|    else
 | |
|    {
 | |
|       description = descriptionName;
 | |
|    }
 | |
| 
 | |
|    return LayerInfo {
 | |
|       layerType,
 | |
|       description,
 | |
|       jv.at(kMovableName_).as_bool(),
 | |
|       boost::json::value_to<std::array<bool, 4>>(jv.at(kDisplayedName_))};
 | |
| }
 | |
| 
 | |
| } // namespace model
 | |
| } // namespace qt
 | |
| } // namespace scwx
 | 
