File indexing completed on 2024-04-14 03:47:54

 // SPDX-License-Identifier: LGPL-2.1-or-later
 //
 // SPDX-FileCopyrightText: 2006-2007 Torsten Rahn <tackat@kde.org>
 // SPDX-FileCopyrightText: 2007 Inge Wallin <ingwa@kde.org>
 // SPDX-FileCopyrightText: 2008, 2009, 2010 Jens-Michael Hoffmann <jmho@c-xx.com>
 // SPDX-FileCopyrightText: 2008-2009 Patrick Spendrin <ps_ml@gmx.de>
 // SPDX-FileCopyrightText: 2010-2013 Bernhard Beschow <bbeschow@cs.tu-berlin.de>
 // SPDX-FileCopyrightText: 2014 Abhinav Gangwar <abhgang@gmail.com>
 //
 
 #include "MarbleModel.h"
 
 #include <cmath>
 
 #include <QAtomicInt>
 #include <QPointer>
 #include <QAbstractItemModel>
 #include <QItemSelectionModel>
 #include <QSortFilterProxyModel>
 #include <QTextDocument>
 
 #include "kdescendantsproxymodel.h"
 
 #include "MapThemeManager.h"
 #include "MarbleDebug.h"
 
 #include "GeoSceneDocument.h"
 #include "GeoSceneGeodata.h"
 #include "GeoSceneHead.h"
 #include "GeoSceneLayer.h"
 #include "GeoSceneMap.h"
 #include "GeoScenePalette.h"
 #include "GeoSceneTileDataset.h"
 #include "GeoSceneVector.h"
 
 #include "GeoDataDocument.h"
 #include "GeoDataFeature.h"
 #include "GeoDataPlacemark.h"
 #include "GeoDataPoint.h"
 #include "GeoDataStyle.h"
 #include "GeoDataStyleMap.h"
 #include "GeoDataTrack.h"
 #include "GeoDataLineStyle.h"
 #include "GeoDataPolyStyle.h"
 #include "GeoDataTypes.h"
 
 #include "DgmlAuxillaryDictionary.h"
 #include "MarbleClock.h"
 #include "FileStoragePolicy.h"
 #include "FileStorageWatcher.h"
 #include "PositionTracking.h"
 #include "HttpDownloadManager.h"
 #include "MarbleDirs.h"
 #include "FileManager.h"
 #include "GeoDataTreeModel.h"
 #include "PlacemarkPositionProviderPlugin.h"
 #include "Planet.h"
 #include "PlanetFactory.h"
 #include "PluginManager.h"
 #include "StoragePolicy.h"
 #include "SunLocator.h"
 #include "TileCreator.h"
 #include "TileCreatorDialog.h"
 #include "TileLoader.h"
 #include "routing/RoutingManager.h"
 #include "RouteSimulationPositionProviderPlugin.h"
 #include "BookmarkManager.h"
 #include "ElevationModel.h"
 
 namespace Marble
 {
 
 class MarbleModelPrivate
 {
  public:
     MarbleModelPrivate()
         : m_clock(),
           m_planet(PlanetFactory::construct(QStringLiteral("earth"))),
           m_sunLocator( &m_clock, &m_planet ),
           m_pluginManager(),
           m_homePoint( -9.4, 54.8, 0.0, GeoDataCoordinates::Degree ),  // Some point that tackat defined. :-)
           m_homeZoom( 1050 ),
           m_mapTheme( nullptr ),
           m_storagePolicy( MarbleDirs::localPath() ),
           m_downloadManager( &m_storagePolicy ),
           m_storageWatcher( MarbleDirs::localPath() ),
           m_treeModel(),
           m_descendantProxy(),
           m_placemarkProxyModel(),
           m_placemarkSelectionModel( nullptr ),
           m_fileManager( &m_treeModel, &m_pluginManager ),
           m_positionTracking( &m_treeModel ),
           m_trackedPlacemark( nullptr ),
           m_bookmarkManager( &m_treeModel ),
           m_routingManager( nullptr ),
           m_legend( nullptr ),
           m_workOffline( false ),
           m_elevationModel( &m_downloadManager, &m_pluginManager )
     {
         m_descendantProxy.setSourceModel( &m_treeModel );
 
         m_placemarkProxyModel.setFilterFixedString( GeoDataTypes::GeoDataPlacemarkType );
         m_placemarkProxyModel.setFilterKeyColumn( 1 );
         m_placemarkProxyModel.setSourceModel( &m_descendantProxy );
         m_placemarkSelectionModel.setModel(&m_placemarkProxyModel);
 
         m_groundOverlayProxyModel.setFilterFixedString( GeoDataTypes::GeoDataGroundOverlayType );
         m_groundOverlayProxyModel.setFilterKeyColumn( 1 );
         m_groundOverlayProxyModel.setSourceModel( &m_descendantProxy );
     }
 
     ~MarbleModelPrivate()
     {
         delete m_mapTheme;
         delete m_legend;
     }
     /**
      * @brief Assigns each placemark an inline
      * style based on the color values specified
      * by colorMap attribute under <brush> element
      * in theme file.
      */
     void assignFillColors( const QString &filePath );
     void assignFillColors(GeoDataDocument *doc, const GeoSceneGeodata &data) const;
 
     void addHighlightStyle(GeoDataDocument *doc) const;
 
     // Misc stuff.
     MarbleClock              m_clock;
     Planet                   m_planet;
     SunLocator               m_sunLocator;
 
     PluginManager            m_pluginManager;
 
     // The home position
     GeoDataCoordinates       m_homePoint;
     int                      m_homeZoom;
 
     // View and paint stuff
     GeoSceneDocument        *m_mapTheme;
 
     FileStoragePolicy        m_storagePolicy;
     HttpDownloadManager      m_downloadManager;
 
     // Cache related
     FileStorageWatcher       m_storageWatcher;
 
     // Places on the map
     GeoDataTreeModel         m_treeModel;
     KDescendantsProxyModel   m_descendantProxy;
     QSortFilterProxyModel    m_placemarkProxyModel;
     QSortFilterProxyModel    m_groundOverlayProxyModel;
 
     // Selection handling
     QItemSelectionModel      m_placemarkSelectionModel;
 
     FileManager              m_fileManager;
 
     //Gps Stuff
     PositionTracking         m_positionTracking;
 
     const GeoDataPlacemark  *m_trackedPlacemark;
 
     BookmarkManager          m_bookmarkManager;
     RoutingManager          *m_routingManager;
     QTextDocument           *m_legend;
 
     bool                     m_workOffline;
 
     ElevationModel           m_elevationModel;
 };
 
 MarbleModel::MarbleModel( QObject *parent )
     : QObject( parent ),
       d( new MarbleModelPrivate() )
 {
     // connect the StoragePolicy used by the download manager to the FileStorageWatcher
     connect( &d->m_storagePolicy, SIGNAL(cleared()),
              &d->m_storageWatcher, SLOT(resetCurrentSize()) );
     connect( &d->m_storagePolicy, SIGNAL(sizeChanged(qint64)),
              &d->m_storageWatcher, SLOT(addToCurrentSize(qint64)) );
 
     connect( &d->m_fileManager, SIGNAL(fileAdded(QString)),
              this, SLOT(assignFillColors(QString)) );
 
     d->m_routingManager = new RoutingManager( this, this );
 
     connect(&d->m_clock,   SIGNAL(timeChanged()),
             &d->m_sunLocator, SLOT(update()) );
 
     d->m_pluginManager.addPositionProviderPlugin(new PlacemarkPositionProviderPlugin(this, this));
     d->m_pluginManager.addPositionProviderPlugin(new RouteSimulationPositionProviderPlugin(this, this));
 }
 
 MarbleModel::~MarbleModel()
 {
     delete d;
 
     mDebug() << "Model deleted:" << this;
 }
 
 BookmarkManager *MarbleModel::bookmarkManager()
 {
     return &d->m_bookmarkManager;
 }
 
 QString MarbleModel::mapThemeId() const
 {
     QString mapThemeId;
 
     if (d->m_mapTheme)
         mapThemeId = d->m_mapTheme->head()->mapThemeId();
 
     return mapThemeId;
 }
 
 GeoSceneDocument *MarbleModel::mapTheme()
 {
     return d->m_mapTheme;
 }
 
 const GeoSceneDocument *MarbleModel::mapTheme() const
 {
     return d->m_mapTheme;
 }
 
 // Set a particular theme for the map and load the appropriate tile level.
 // If the tiles (for the lowest tile level) haven't been created already
 // then create them here and now.
 //
 // FIXME: Move the tile creation dialogs out of this function.  Change
 //        them into signals instead.
 // FIXME: Get rid of 'currentProjection' here.  It's totally misplaced.
 //
 
 void MarbleModel::setMapThemeId( const QString &mapThemeId )
 {
     if ( !mapThemeId.isEmpty() && mapThemeId == this->mapThemeId() )
         return;
 
     GeoSceneDocument *mapTheme = MapThemeManager::loadMapTheme( mapThemeId );
     setMapTheme( mapTheme );
 }
 
 void MarbleModel::setMapTheme( GeoSceneDocument *document )
 {
     GeoSceneDocument *mapTheme = document;
     if ( !mapTheme ) {
         // Check whether the previous theme works
         if ( d->m_mapTheme ){
             qWarning() << "Selected theme doesn't work, so we stick to the previous one";
             return;
         }
 
         // Fall back to default theme
         QString defaultTheme = "earth/srtm/srtm.dgml";
         qWarning() << "Falling back to default theme:" << defaultTheme;
         mapTheme = MapThemeManager::loadMapTheme( defaultTheme );
     }
 
     // If this last resort doesn't work either shed a tear and exit
     if ( !mapTheme ) {
         qWarning() << "Couldn't find a valid DGML map.";
         return;
     }
 
     // find the list of previous theme's geodata
     QList<GeoSceneGeodata> currentDatasets;
     if ( d->m_mapTheme ) {
         for ( GeoSceneLayer *layer: d->m_mapTheme->map()->layers() ) {
             if ( layer->backend() != dgml::dgmlValue_geodata
                  && layer->backend() != dgml::dgmlValue_vector )
                 continue;
 
             // look for documents
             for ( GeoSceneAbstractDataset *dataset: layer->datasets() ) {
                 GeoSceneGeodata *data = dynamic_cast<GeoSceneGeodata*>( dataset );
                 Q_ASSERT( data );
                 currentDatasets << *data;
             }
         }
     }
 
     delete d->m_mapTheme;
     d->m_mapTheme = mapTheme;
 
     addDownloadPolicies( d->m_mapTheme );
 
     // Some output to show how to use this stuff ...
     mDebug() << "DGML2 Name       : " << d->m_mapTheme->head()->name();
 /*
     mDebug() << "DGML2 Description: " << d->m_mapTheme->head()->description();
 
     if ( d->m_mapTheme->map()->hasTextureLayers() )
         mDebug() << "Contains texture layers! ";
     else
         mDebug() << "Does not contain any texture layers! ";
 
     mDebug() << "Number of SRTM textures: " << d->m_mapTheme->map()->layer("srtm")->datasets().count();
 
     if ( d->m_mapTheme->map()->hasVectorLayers() )
         mDebug() << "Contains vector layers! ";
     else
         mDebug() << "Does not contain any vector layers! ";
 */
     //Don't change the planet unless we have to...
     qreal const radiusAttributeValue = d->m_mapTheme->head()->radius();
     if( d->m_mapTheme->head()->target().toLower() != d->m_planet.id() || radiusAttributeValue != d->m_planet.radius() ) {
         mDebug() << "Changing Planet";
         d->m_planet = PlanetFactory::construct(d->m_mapTheme->head()->target().toLower());
         if ( radiusAttributeValue > 0.0 ) {
             d->m_planet.setRadius( radiusAttributeValue );
         }
         sunLocator()->setPlanet( &d->m_planet );
     }
 
     QStringList fileList;
     QStringList propertyList;
     QList<GeoDataStyle::Ptr> styleList;
     QList<int> renderOrderList;
 
     for ( GeoSceneLayer *layer: d->m_mapTheme->map()->layers() ) {
         if ( layer->backend() != dgml::dgmlValue_geodata
              && layer->backend() != dgml::dgmlValue_vector )
             continue;
 
         // look for datasets which are different from currentDatasets
         for ( const GeoSceneAbstractDataset *dataset: layer->datasets() ) {
             const GeoSceneGeodata *data = dynamic_cast<const GeoSceneGeodata*>( dataset );
             Q_ASSERT( data );
             bool skip = false;
             GeoDataDocument *doc = nullptr;
             for ( int i = 0; i < currentDatasets.size(); ++i ) {
                 if ( currentDatasets[i] == *data ) {
                     currentDatasets.removeAt( i );
                     skip = true;
                     break;
                 }
                 /*
                  * If the sourcefile of data matches any in the currentDatasets then there
                  * is no need to parse the file again just update the style
                  * i.e. <brush> and <pen> values of already parsed file. assignNewStyle() does that
                  */
                 if ( currentDatasets[i].sourceFile() == data->sourceFile() ) {
                     doc = d->m_fileManager.at(data->sourceFile());
                     currentDatasets.removeAt(i);
                 }
             }
             if ( skip ) {
                 continue;
             }
 
             if (doc) {
                 d->assignFillColors(doc, *data);
             }
             else {
                 const QString filename = data->sourceFile();
                 const QString property = data->property();
                 const QPen pen = data->pen();
                 const QBrush brush = data->brush();
                 GeoDataStyle::Ptr style;
                 const int renderOrder = data->renderOrder();
 
                 /*
                  * data->colors() are the colorMap values from dgml file. If this is not
                  * empty then we are supposed to assign every placemark a different style
                  * by giving it a color from colorMap values based on color index
                  * of that placemark. See assignFillColors() for details. So, we need to
                  * send an empty style to fileManeger otherwise the FileLoader::createFilterProperties()
                  * will overwrite the parsed value of color index ( GeoDataPolyStyle::d->m_colorIndex ).
                  */
                 if ( data->colors().isEmpty() ) {
                     GeoDataLineStyle lineStyle( pen.color() );
                     lineStyle.setPenStyle( pen.style() );
                     lineStyle.setWidth( pen.width() );
                     GeoDataPolyStyle polyStyle( brush.color() );
                     polyStyle.setFill( true );
                     style = GeoDataStyle::Ptr(new GeoDataStyle);
                     style->setLineStyle( lineStyle );
                     style->setPolyStyle( polyStyle );
                     style->setId(QStringLiteral("default"));
                 }
 
                 fileList << filename;
                 propertyList << property;
                 styleList << style;
                 renderOrderList << renderOrder;
             }
         }
     }
     // unload old currentDatasets which are not part of the new map
     for(const GeoSceneGeodata &data: currentDatasets) {
         d->m_fileManager.removeFile( data.sourceFile() );
     }
     // load new datasets
     for ( int i = 0 ; i < fileList.size(); ++i ) {
         d->m_fileManager.addFile( fileList.at(i), propertyList.at(i), styleList.at(i), MapDocument, renderOrderList.at(i) );
     }
 
     mDebug() << "THEME CHANGED: ***" << mapTheme->head()->mapThemeId();
     emit themeChanged( mapTheme->head()->mapThemeId() );
 }
 
 void MarbleModelPrivate::addHighlightStyle(GeoDataDocument *doc) const
 {
     if ( doc ) {
         /*
          * Add a highlight style to GeoDataDocument if
          *the theme file specifies any highlight color.
          */
         QColor highlightBrushColor = m_mapTheme->map()->highlightBrushColor();
         QColor highlightPenColor = m_mapTheme->map()->highlightPenColor();
 
         GeoDataStyle::Ptr highlightStyle(new GeoDataStyle);
         highlightStyle->setId(QStringLiteral("highlight"));
 
         if ( highlightBrushColor.isValid() ) {
             GeoDataPolyStyle highlightPolyStyle;
             highlightPolyStyle.setColor( highlightBrushColor );
             highlightPolyStyle.setFill( true );
             highlightStyle->setPolyStyle( highlightPolyStyle );
         }
         if ( highlightPenColor.isValid() ) {
             GeoDataLineStyle highlightLineStyle( highlightPenColor );
             highlightStyle->setLineStyle( highlightLineStyle );
         }
         if ( highlightBrushColor.isValid()
             || highlightPenColor.isValid() )
         {
             GeoDataStyleMap styleMap = doc->styleMap(QStringLiteral("default-map"));
             styleMap.insert(QStringLiteral("highlight"), QLatin1Char('#') + highlightStyle->id());
             doc->addStyle( highlightStyle );
             doc->addStyleMap( styleMap );
         }
     }
 }
 
 void MarbleModel::home( qreal &lon, qreal &lat, int& zoom ) const
 {
     d->m_homePoint.geoCoordinates( lon, lat, GeoDataCoordinates::Degree );
     zoom = d->m_homeZoom;
 }
 
 void MarbleModel::setHome( qreal lon, qreal lat, int zoom )
 {
     d->m_homePoint = GeoDataCoordinates( lon, lat, 0, GeoDataCoordinates::Degree );
     d->m_homeZoom = zoom;
     emit homeChanged( d->m_homePoint );
 }
 
 void MarbleModel::setHome( const GeoDataCoordinates& homePoint, int zoom )
 {
     d->m_homePoint = homePoint;
     d->m_homeZoom = zoom;
     emit homeChanged( d->m_homePoint );
 }
 
 HttpDownloadManager *MarbleModel::downloadManager()
 {
     return &d->m_downloadManager;
 }
 
 const HttpDownloadManager *MarbleModel::downloadManager() const
 {
     return &d->m_downloadManager;
 }
 
 
 GeoDataTreeModel *MarbleModel::treeModel()
 {
     return &d->m_treeModel;
 }
 
 const GeoDataTreeModel *MarbleModel::treeModel() const
 {
     return &d->m_treeModel;
 }
 
 QAbstractItemModel *MarbleModel::placemarkModel()
 {
     return &d->m_placemarkProxyModel;
 }
 
 const QAbstractItemModel *MarbleModel::placemarkModel() const
 {
     return &d->m_placemarkProxyModel;
 }
 
 QAbstractItemModel *MarbleModel::groundOverlayModel()
 {
     return &d->m_groundOverlayProxyModel;
 }
 
 const QAbstractItemModel *MarbleModel::groundOverlayModel() const
 {
     return &d->m_groundOverlayProxyModel;
 }
 
 QItemSelectionModel *MarbleModel::placemarkSelectionModel()
 {
     return &d->m_placemarkSelectionModel;
 }
 
 PositionTracking *MarbleModel::positionTracking() const
 {
     return &d->m_positionTracking;
 }
 
 FileManager *MarbleModel::fileManager()
 {
     return &d->m_fileManager;
 }
 
 qreal MarbleModel::planetRadius()   const
 {
     return d->m_planet.radius();
 }
 
 QString MarbleModel::planetName()   const
 {
     return d->m_planet.name();
 }
 
 QString MarbleModel::planetId() const
 {
     return d->m_planet.id();
 }
 
 MarbleClock *MarbleModel::clock()
 {
     return &d->m_clock;
 }
 
 const MarbleClock *MarbleModel::clock() const
 {
     return &d->m_clock;
 }
 
 SunLocator *MarbleModel::sunLocator()
 {
     return &d->m_sunLocator;
 }
 
 const SunLocator *MarbleModel::sunLocator() const
 {
     return &d->m_sunLocator;
 }
 
 quint64 MarbleModel::persistentTileCacheLimit() const
 {
     return d->m_storageWatcher.cacheLimit() / 1024;
 }
 
 void MarbleModel::clearPersistentTileCache()
 {
     d->m_storagePolicy.clearCache();
 
     // Now create base tiles again if needed
     if ( d->m_mapTheme->map()->hasTextureLayers() || d->m_mapTheme->map()->hasVectorLayers() ) {
         // If the tiles aren't already there, put up a progress dialog
         // while creating them.
 
         // As long as we don't have an Layer Management Class we just lookup
         // the name of the layer that has the same name as the theme ID
         QString themeID = d->m_mapTheme->head()->theme();
 
         const GeoSceneLayer *layer =
             static_cast<const GeoSceneLayer*>( d->m_mapTheme->map()->layer( themeID ) );
         const GeoSceneTileDataset *texture =
             static_cast<const GeoSceneTileDataset*>( layer->groundDataset() );
 
         QString sourceDir = texture->sourceDir();
         QString installMap = texture->installMap();
         QString role = d->m_mapTheme->map()->layer( themeID )->role();
 
         if ( !TileLoader::baseTilesAvailable( *texture )
             && !installMap.isEmpty() )
         {
             mDebug() << "Base tiles not available. Creating Tiles ... \n"
                      << "SourceDir: " << sourceDir << "InstallMap:" << installMap;
             MarbleDirs::debug();
 
             TileCreator *tileCreator = new TileCreator(
                                      sourceDir,
                                      installMap,
                                      (role == QLatin1String("dem")) ? "true" : "false" );
             tileCreator->setTileFormat( texture->fileFormat().toLower() );
 
             QPointer<TileCreatorDialog> tileCreatorDlg = new TileCreatorDialog( tileCreator, nullptr );
             tileCreatorDlg->setSummary( d->m_mapTheme->head()->name(),
                                         d->m_mapTheme->head()->description() );
             tileCreatorDlg->exec();
             qDebug("Tile creation completed");
             delete tileCreatorDlg;
         }
     }
 }
 
 void MarbleModel::setPersistentTileCacheLimit(quint64 kiloBytes)
 {
     d->m_storageWatcher.setCacheLimit( kiloBytes * 1024 );
 
     if( kiloBytes != 0 )
     {
         if( !d->m_storageWatcher.isRunning() )
             d->m_storageWatcher.start( QThread::IdlePriority );
     }
     else
     {
         d->m_storageWatcher.quit();
     }
     // TODO: trigger update
 }
 
 void MarbleModel::setTrackedPlacemark( const GeoDataPlacemark *placemark )
 {
     d->m_trackedPlacemark = placemark;
     emit trackedPlacemarkChanged( placemark );
 }
 
 const GeoDataPlacemark* MarbleModel::trackedPlacemark() const
 {
     return d->m_trackedPlacemark;
 }
 
 const PluginManager* MarbleModel::pluginManager() const
 {
     return &d->m_pluginManager;
 }
 
 PluginManager* MarbleModel::pluginManager()
 {
     return &d->m_pluginManager;
 }
 
 const Planet *MarbleModel::planet() const
 {
     return &d->m_planet;
 }
 
 void MarbleModel::addDownloadPolicies( const GeoSceneDocument *mapTheme )
 {
     if ( !mapTheme )
         return;
     if ( !mapTheme->map()->hasTextureLayers() && !mapTheme->map()->hasVectorLayers() )
         return;
 
     // As long as we don't have an Layer Management Class we just lookup
     // the name of the layer that has the same name as the theme ID
     const QString themeId = mapTheme->head()->theme();
     const GeoSceneLayer * const layer = static_cast<const GeoSceneLayer*>( mapTheme->map()->layer( themeId ));
     if ( !layer )
         return;
 
     const GeoSceneTileDataset * const texture = static_cast<const GeoSceneTileDataset*>( layer->groundDataset() );
     if ( !texture )
         return;
 
     QList<const DownloadPolicy *> policies = texture->downloadPolicies();
     QList<const DownloadPolicy *>::const_iterator pos = policies.constBegin();
     QList<const DownloadPolicy *>::const_iterator const end = policies.constEnd();
     for (; pos != end; ++pos ) {
         d->m_downloadManager.addDownloadPolicy( **pos );
     }
 }
 
 RoutingManager* MarbleModel::routingManager()
 {
     return d->m_routingManager;
 }
 
 const RoutingManager* MarbleModel::routingManager() const
 {
     return d->m_routingManager;
 }
 
 void MarbleModel::setClockDateTime( const QDateTime& datetime )
 {
     d->m_clock.setDateTime( datetime );
 }
 
 QDateTime MarbleModel::clockDateTime() const
 {
     return d->m_clock.dateTime();
 }
 
 int MarbleModel::clockSpeed() const
 {
     return d->m_clock.speed();
 }
 
 void MarbleModel::setClockSpeed( int speed )
 {
     d->m_clock.setSpeed( speed );
 }
 
 void MarbleModel::setClockTimezone( int timeInSec )
 {
     d->m_clock.setTimezone( timeInSec );
 }
 
 int MarbleModel::clockTimezone() const
 {
     return d->m_clock.timezone();
 }
 
 QTextDocument * MarbleModel::legend()
 {
     return d->m_legend;
 }
 
 void MarbleModel::setLegend( QTextDocument * legend )
 {
     delete d->m_legend;
     d->m_legend = legend;
 }
 
 void MarbleModel::addGeoDataFile( const QString& filename )
 {
     d->m_fileManager.addFile( filename, filename, GeoDataStyle::Ptr(), UserDocument, 0, true );
 }
 
 void MarbleModel::addGeoDataString( const QString& data, const QString& key )
 {
     d->m_fileManager.addData( key, data, UserDocument );
 }
 
 void MarbleModel::removeGeoData( const QString& fileName )
 {
     d->m_fileManager.removeFile( fileName );
 }
 
 void MarbleModel::updateProperty( const QString &property, bool value )
 {
     for( GeoDataFeature *feature: d->m_treeModel.rootDocument()->featureList()) {
         if (auto document = geodata_cast<GeoDataDocument>(feature)) {
             if( document->property() == property ){
                 document->setVisible( value );
                 d->m_treeModel.updateFeature( document );
             }
         }
     }
 }
 
 void MarbleModelPrivate::assignFillColors(const QString &filePath)
 {
     const GeoSceneGeodata *data = nullptr;
 
     for (auto layer : m_mapTheme->map()->layers()) {
         if (layer->backend() != dgml::dgmlValue_geodata
             && layer->backend() != dgml::dgmlValue_vector) {
             continue;
         }
 
         for (auto dataset: layer->datasets()) {
             auto sceneData = dynamic_cast<const GeoSceneGeodata *>(dataset);
             if (sceneData != nullptr && sceneData->sourceFile() == filePath) {
                 data = sceneData;
                 break;
             }
         }
 
         if (data) {
             break;
         }
     }
 
     if (data == nullptr) {
         return;
     }
 
     GeoDataDocument *doc = m_fileManager.at(filePath);
     Q_ASSERT( doc );
 
     assignFillColors(doc, *data);
 }
 
 void MarbleModelPrivate::assignFillColors(GeoDataDocument *doc, const GeoSceneGeodata &data) const
 {
     addHighlightStyle( doc );
 
     const QPen pen = data.pen();
     const QBrush brush = data.brush();
     const QVector<QColor> colors = data.colors();
     GeoDataLineStyle lineStyle( pen.color() );
     lineStyle.setPenStyle( pen.style() );
     lineStyle.setWidth( pen.width() );
 
     if (!colors.isEmpty()) {
         const qreal alpha = data.alpha();
         for (auto feature : *doc) {
             auto placemark = geodata_cast<GeoDataPlacemark>(feature);
             if (placemark == nullptr) {
                 continue;
             }
 
             GeoDataStyle::Ptr style(new GeoDataStyle);
             style->setId(QStringLiteral("normal"));
             style->setLineStyle( lineStyle );
             quint8 colorIndex = placemark->style()->polyStyle().colorIndex();
             GeoDataPolyStyle polyStyle;
             // Set the colorIndex so that it's not lost after setting new style.
             polyStyle.setColorIndex( colorIndex );
             QColor color;
             // color index having value 99 is undefined
             Q_ASSERT( colors.size() );
             if (colorIndex > colors.size() || (colorIndex - 1) < 0) {
                 color = colors[0];      // Assign the first color as default
             }
             else {
                 color = colors[colorIndex-1];
             }
             color.setAlphaF( alpha );
             polyStyle.setColor( color );
             polyStyle.setFill( true );
             style->setPolyStyle( polyStyle );
             placemark->setStyle( style );
         }
     }
     else {
         GeoDataStyle::Ptr style(new GeoDataStyle);
         GeoDataPolyStyle polyStyle( brush.color() );
         polyStyle.setFill( true );
         style->setLineStyle( lineStyle );
         style->setPolyStyle( polyStyle );
         style->setId(QStringLiteral("default"));
         GeoDataStyleMap styleMap;
         styleMap.setId(QStringLiteral("default-map"));
         styleMap.insert(QStringLiteral("normal"), QLatin1Char('#') + style->id());
         doc->addStyle( style );
         doc->addStyleMap( styleMap );
 
         const QString styleUrl = QLatin1Char('#') + styleMap.id();
 
         for (auto feature : *doc) {
             auto placemark = geodata_cast<GeoDataPlacemark>(feature);
             if (placemark == nullptr) {
                 continue;
             }
 
             if (geodata_cast<GeoDataTrack>(placemark->geometry()) ||
                 geodata_cast<GeoDataPoint>(placemark->geometry())) {
                 continue;
             }
 
             placemark->setStyleUrl(styleUrl);
         }
     }
 }
 
 bool MarbleModel::workOffline() const
 {
     return d->m_workOffline;
 }
 
 void MarbleModel::setWorkOffline( bool workOffline )
 {
     if ( d->m_workOffline != workOffline ) {
         downloadManager()->setDownloadEnabled( !workOffline );
 
         d->m_workOffline = workOffline;
         emit workOfflineChanged();
     }
 }
 
 ElevationModel* MarbleModel::elevationModel()
 {
     return &d->m_elevationModel;
 }
 
 const ElevationModel* MarbleModel::elevationModel() const
 {
     return &d->m_elevationModel;
 }
 
 }
 
 #include "moc_MarbleModel.cpp"