Branch data Line data Source code
1 : : /***************************************************************************
2 : : qgsstyle.cpp
3 : : ---------------------
4 : : begin : November 2009
5 : : copyright : (C) 2009 by Martin Dobias
6 : : email : wonder dot sk at gmail dot com
7 : : ***************************************************************************
8 : : * *
9 : : * This program is free software; you can redistribute it and/or modify *
10 : : * it under the terms of the GNU General Public License as published by *
11 : : * the Free Software Foundation; either version 2 of the License, or *
12 : : * (at your option) any later version. *
13 : : * *
14 : : ***************************************************************************/
15 : :
16 : : #include "qgsstyle.h"
17 : :
18 : : #include "qgssymbol.h"
19 : : #include "qgscolorramp.h"
20 : : #include "qgssymbollayerregistry.h"
21 : : #include "qgsapplication.h"
22 : : #include "qgslogger.h"
23 : : #include "qgsreadwritecontext.h"
24 : : #include "qgssettings.h"
25 : : #include "qgslegendpatchshape.h"
26 : : #include "qgslinestring.h"
27 : : #include "qgspolygon.h"
28 : : #include "qgsmarkersymbollayer.h"
29 : : #include "qgslinesymbollayer.h"
30 : : #include "qgsfillsymbollayer.h"
31 : : #include "qgsruntimeprofiler.h"
32 : : #include "qgsabstract3dsymbol.h"
33 : : #include "qgs3dsymbolregistry.h"
34 : :
35 : : #include <QDomDocument>
36 : : #include <QDomElement>
37 : : #include <QDomNode>
38 : : #include <QDomNodeList>
39 : : #include <QFile>
40 : : #include <QTextStream>
41 : : #include <QByteArray>
42 : : #include <QFileInfo>
43 : :
44 : : #include <sqlite3.h>
45 : : #include "qgssqliteutils.h"
46 : :
47 : : #define STYLE_CURRENT_VERSION "2"
48 : :
49 : : /**
50 : : * Columns available in the legend patch table.
51 : : */
52 : : enum LegendPatchTable
53 : : {
54 : : LegendPatchTableId, //!< Legend patch ID
55 : : LegendPatchTableName, //!< Legend patch name
56 : : LegendPatchTableXML, //!< Legend patch definition (as XML)
57 : : LegendPatchTableFavoriteId, //!< Legend patch is favorite flag
58 : : };
59 : :
60 : : /**
61 : : * Columns available in the 3d symbol table.
62 : : */
63 : : enum Symbol3DTable
64 : : {
65 : : Symbol3DTableId, //!< 3d symbol ID
66 : : Symbol3DTableName, //!< 3d symbol name
67 : : Symbol3DTableXML, //!< 3d symbol definition (as XML)
68 : : Symbol3DTableFavoriteId, //!< 3d symbol is favorite flag
69 : : };
70 : :
71 : :
72 : : QgsStyle *QgsStyle::sDefaultStyle = nullptr;
73 : :
74 : 5 : QgsStyle::QgsStyle()
75 : 10 : {
76 : 10 : std::unique_ptr< QgsSimpleMarkerSymbolLayer > simpleMarker = std::make_unique< QgsSimpleMarkerSymbolLayer >( QgsSimpleMarkerSymbolLayerBase::Circle,
77 : 5 : 1.6, 0, QgsSymbol::ScaleArea, QColor( 84, 176, 74 ), QColor( 61, 128, 53 ) );
78 : 5 : simpleMarker->setStrokeWidth( 0.4 );
79 : 5 : mPatchMarkerSymbol = std::make_unique< QgsMarkerSymbol >( QgsSymbolLayerList() << simpleMarker.release() );
80 : :
81 : 5 : std::unique_ptr< QgsSimpleLineSymbolLayer > simpleLine = std::make_unique< QgsSimpleLineSymbolLayer >( QColor( 84, 176, 74 ), 0.6 );
82 : 5 : mPatchLineSymbol = std::make_unique< QgsLineSymbol >( QgsSymbolLayerList() << simpleLine.release() );
83 : :
84 : 5 : std::unique_ptr< QgsGradientFillSymbolLayer > gradientFill = std::make_unique< QgsGradientFillSymbolLayer >( QColor( 66, 150, 63 ), QColor( 84, 176, 74 ) );
85 : 5 : std::unique_ptr< QgsSimpleLineSymbolLayer > simpleOutline = std::make_unique< QgsSimpleLineSymbolLayer >( QColor( 56, 128, 54 ), 0.26 );
86 : 5 : mPatchFillSymbol = std::make_unique< QgsFillSymbol >( QgsSymbolLayerList() << gradientFill.release() << simpleOutline.release() );
87 : 5 : }
88 : :
89 : 6 : QgsStyle::~QgsStyle()
90 : 6 : {
91 : 3 : clear();
92 : 6 : }
93 : :
94 : 0 : bool QgsStyle::addEntity( const QString &name, const QgsStyleEntityInterface *entity, bool update )
95 : : {
96 : 0 : switch ( entity->type() )
97 : : {
98 : : case SymbolEntity:
99 : 0 : if ( !static_cast< const QgsStyleSymbolEntity * >( entity )->symbol() )
100 : 0 : return false;
101 : 0 : return addSymbol( name, static_cast< const QgsStyleSymbolEntity * >( entity )->symbol()->clone(), update );
102 : :
103 : : case ColorrampEntity:
104 : 0 : if ( !static_cast< const QgsStyleColorRampEntity * >( entity )->ramp() )
105 : 0 : return false;
106 : 0 : return addColorRamp( name, static_cast< const QgsStyleColorRampEntity * >( entity )->ramp()->clone(), update );
107 : :
108 : : case TextFormatEntity:
109 : 0 : return addTextFormat( name, static_cast< const QgsStyleTextFormatEntity * >( entity )->format(), update );
110 : :
111 : : case LabelSettingsEntity:
112 : 0 : return addLabelSettings( name, static_cast< const QgsStyleLabelSettingsEntity * >( entity )->settings(), update );
113 : :
114 : : case LegendPatchShapeEntity:
115 : 0 : return addLegendPatchShape( name, static_cast< const QgsStyleLegendPatchShapeEntity * >( entity )->shape(), update );
116 : :
117 : : case Symbol3DEntity:
118 : 0 : return addSymbol3D( name, static_cast< const QgsStyleSymbol3DEntity * >( entity )->symbol()->clone(), update );
119 : :
120 : : case TagEntity:
121 : : case SmartgroupEntity:
122 : 0 : break;
123 : :
124 : : }
125 : 0 : return false;
126 : 0 : }
127 : :
128 : 5 : QgsStyle *QgsStyle::defaultStyle() // static
129 : : {
130 : 5 : if ( !sDefaultStyle )
131 : : {
132 : 5 : QgsScopedRuntimeProfile profile( tr( "Load default style database" ) );
133 : 5 : QString styleFilename = QgsApplication::userStylePath();
134 : :
135 : : // copy default style if user style doesn't exist
136 : 5 : if ( !QFile::exists( styleFilename ) )
137 : : {
138 : 0 : sDefaultStyle = new QgsStyle;
139 : 0 : sDefaultStyle->createDatabase( styleFilename );
140 : 0 : if ( QFile::exists( QgsApplication::defaultStylePath() ) )
141 : : {
142 : 0 : sDefaultStyle->importXml( QgsApplication::defaultStylePath() );
143 : 0 : }
144 : 0 : }
145 : : else
146 : : {
147 : 5 : sDefaultStyle = new QgsStyle;
148 : 5 : if ( sDefaultStyle->load( styleFilename ) )
149 : : {
150 : 5 : sDefaultStyle->upgradeIfRequired();
151 : 5 : }
152 : : }
153 : 5 : }
154 : 5 : return sDefaultStyle;
155 : 0 : }
156 : :
157 : 3 : void QgsStyle::cleanDefaultStyle() // static
158 : : {
159 : 3 : delete sDefaultStyle;
160 : 3 : sDefaultStyle = nullptr;
161 : 3 : }
162 : :
163 : 3 : void QgsStyle::clear()
164 : : {
165 : 3 : qDeleteAll( mSymbols );
166 : 3 : qDeleteAll( mColorRamps );
167 : 3 : qDeleteAll( m3dSymbols );
168 : :
169 : 3 : mSymbols.clear();
170 : 3 : mColorRamps.clear();
171 : 3 : mTextFormats.clear();
172 : 3 : m3dSymbols.clear();
173 : :
174 : 3 : mCachedTags.clear();
175 : 3 : mCachedFavorites.clear();
176 : 3 : }
177 : :
178 : 0 : bool QgsStyle::addSymbol( const QString &name, QgsSymbol *symbol, bool update )
179 : : {
180 : 0 : if ( !symbol || name.isEmpty() )
181 : 0 : return false;
182 : :
183 : : // delete previous symbol (if any)
184 : 0 : if ( mSymbols.contains( name ) )
185 : : {
186 : : // TODO remove groups and tags?
187 : 0 : delete mSymbols.value( name );
188 : 0 : mSymbols.insert( name, symbol );
189 : 0 : if ( update )
190 : 0 : updateSymbol( SymbolEntity, name );
191 : 0 : }
192 : : else
193 : : {
194 : 0 : mSymbols.insert( name, symbol );
195 : 0 : if ( update )
196 : 0 : saveSymbol( name, symbol, false, QStringList() );
197 : : }
198 : :
199 : 0 : return true;
200 : 0 : }
201 : :
202 : 0 : bool QgsStyle::saveSymbol( const QString &name, QgsSymbol *symbol, bool favorite, const QStringList &tags )
203 : : {
204 : : // TODO add support for groups
205 : 0 : QDomDocument doc( QStringLiteral( "dummy" ) );
206 : 0 : QDomElement symEl = QgsSymbolLayerUtils::saveSymbol( name, symbol, doc, QgsReadWriteContext() );
207 : 0 : if ( symEl.isNull() )
208 : : {
209 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't convert symbol to valid XML!" ) );
210 : 0 : return false;
211 : : }
212 : :
213 : 0 : QByteArray xmlArray;
214 : 0 : QTextStream stream( &xmlArray );
215 : 0 : stream.setCodec( "UTF-8" );
216 : 0 : symEl.save( stream, 4 );
217 : 0 : QString query = qgs_sqlite3_mprintf( "INSERT INTO symbol VALUES (NULL, '%q', '%q', %d);",
218 : 0 : name.toUtf8().constData(), xmlArray.constData(), ( favorite ? 1 : 0 ) );
219 : :
220 : 0 : if ( !runEmptyQuery( query ) )
221 : : {
222 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't insert symbol into the database!" ) );
223 : 0 : return false;
224 : : }
225 : :
226 : 0 : mCachedFavorites[ SymbolEntity ].insert( name, favorite );
227 : :
228 : 0 : tagSymbol( SymbolEntity, name, tags );
229 : :
230 : 0 : emit symbolSaved( name, symbol );
231 : 0 : emit entityAdded( SymbolEntity, name );
232 : :
233 : 0 : return true;
234 : 0 : }
235 : :
236 : 0 : bool QgsStyle::removeSymbol( const QString &name )
237 : : {
238 : 0 : return removeEntityByName( SymbolEntity, name );
239 : : }
240 : :
241 : 0 : bool QgsStyle::renameEntity( QgsStyle::StyleEntity type, const QString &oldName, const QString &newName )
242 : : {
243 : 0 : switch ( type )
244 : : {
245 : : case SymbolEntity:
246 : 0 : return renameSymbol( oldName, newName );
247 : :
248 : : case ColorrampEntity:
249 : 0 : return renameColorRamp( oldName, newName );
250 : :
251 : : case TextFormatEntity:
252 : 0 : return renameTextFormat( oldName, newName );
253 : :
254 : : case LabelSettingsEntity:
255 : 0 : return renameLabelSettings( oldName, newName );
256 : :
257 : : case LegendPatchShapeEntity:
258 : 0 : return renameLegendPatchShape( oldName, newName );
259 : :
260 : : case Symbol3DEntity:
261 : 0 : return renameSymbol3D( oldName, newName );
262 : :
263 : : case TagEntity:
264 : : case SmartgroupEntity:
265 : 0 : return false;
266 : : }
267 : 0 : return false;
268 : 0 : }
269 : :
270 : 0 : QgsSymbol *QgsStyle::symbol( const QString &name )
271 : : {
272 : 0 : const QgsSymbol *symbol = symbolRef( name );
273 : 0 : return symbol ? symbol->clone() : nullptr;
274 : : }
275 : :
276 : 0 : const QgsSymbol *QgsStyle::symbolRef( const QString &name ) const
277 : : {
278 : 0 : return mSymbols.value( name );
279 : : }
280 : :
281 : 0 : int QgsStyle::symbolCount()
282 : : {
283 : 0 : return mSymbols.count();
284 : : }
285 : :
286 : 5 : QStringList QgsStyle::symbolNames() const
287 : : {
288 : 5 : return mSymbols.keys();
289 : : }
290 : :
291 : :
292 : 0 : bool QgsStyle::addColorRamp( const QString &name, QgsColorRamp *colorRamp, bool update )
293 : : {
294 : 0 : if ( !colorRamp || name.isEmpty() )
295 : 0 : return false;
296 : :
297 : : // delete previous color ramps (if any)
298 : 0 : if ( mColorRamps.contains( name ) )
299 : : {
300 : : // TODO remove groups and tags?
301 : 0 : delete mColorRamps.value( name );
302 : 0 : mColorRamps.insert( name, colorRamp );
303 : 0 : if ( update )
304 : 0 : updateSymbol( ColorrampEntity, name );
305 : 0 : }
306 : : else
307 : : {
308 : 0 : mColorRamps.insert( name, colorRamp );
309 : 0 : if ( update )
310 : 0 : saveColorRamp( name, colorRamp, false, QStringList() );
311 : : }
312 : :
313 : 0 : return true;
314 : 0 : }
315 : :
316 : 0 : bool QgsStyle::addTextFormat( const QString &name, const QgsTextFormat &format, bool update )
317 : : {
318 : : // delete previous text format (if any)
319 : 0 : if ( mTextFormats.contains( name ) )
320 : : {
321 : : // TODO remove groups and tags?
322 : 0 : mTextFormats.remove( name );
323 : 0 : mTextFormats.insert( name, format );
324 : 0 : if ( update )
325 : 0 : updateSymbol( TextFormatEntity, name );
326 : 0 : }
327 : : else
328 : : {
329 : 0 : mTextFormats.insert( name, format );
330 : 0 : if ( update )
331 : 0 : saveTextFormat( name, format, false, QStringList() );
332 : : }
333 : :
334 : 0 : return true;
335 : 0 : }
336 : :
337 : 0 : bool QgsStyle::addLabelSettings( const QString &name, const QgsPalLayerSettings &settings, bool update )
338 : : {
339 : : // delete previous label settings (if any)
340 : 0 : if ( mLabelSettings.contains( name ) )
341 : : {
342 : : // TODO remove groups and tags?
343 : 0 : mLabelSettings.remove( name );
344 : 0 : mLabelSettings.insert( name, settings );
345 : 0 : if ( update )
346 : 0 : updateSymbol( LabelSettingsEntity, name );
347 : 0 : }
348 : : else
349 : : {
350 : 0 : mLabelSettings.insert( name, settings );
351 : 0 : if ( update )
352 : 0 : saveLabelSettings( name, settings, false, QStringList() );
353 : : }
354 : :
355 : 0 : return true;
356 : 0 : }
357 : :
358 : 0 : bool QgsStyle::addLegendPatchShape( const QString &name, const QgsLegendPatchShape &shape, bool update )
359 : : {
360 : : // delete previous legend patch shape (if any)
361 : 0 : if ( mLegendPatchShapes.contains( name ) )
362 : : {
363 : : // TODO remove groups and tags?
364 : 0 : mLegendPatchShapes.remove( name );
365 : 0 : mLegendPatchShapes.insert( name, shape );
366 : 0 : if ( update )
367 : 0 : updateSymbol( LegendPatchShapeEntity, name );
368 : 0 : }
369 : : else
370 : : {
371 : 0 : mLegendPatchShapes.insert( name, shape );
372 : 0 : if ( update )
373 : 0 : saveLegendPatchShape( name, shape, false, QStringList() );
374 : : }
375 : :
376 : 0 : return true;
377 : 0 : }
378 : :
379 : 0 : bool QgsStyle::addSymbol3D( const QString &name, QgsAbstract3DSymbol *symbol, bool update )
380 : : {
381 : : // delete previous symbol (if any)
382 : 0 : if ( m3dSymbols.contains( name ) )
383 : : {
384 : : // TODO remove groups and tags?
385 : 0 : delete m3dSymbols.take( name );
386 : 0 : m3dSymbols.insert( name, symbol );
387 : 0 : if ( update )
388 : 0 : updateSymbol( Symbol3DEntity, name );
389 : 0 : }
390 : : else
391 : : {
392 : 0 : m3dSymbols.insert( name, symbol );
393 : 0 : if ( update )
394 : 0 : saveSymbol3D( name, symbol, false, QStringList() );
395 : : }
396 : :
397 : 0 : return true;
398 : 0 : }
399 : :
400 : 0 : bool QgsStyle::saveColorRamp( const QString &name, QgsColorRamp *ramp, bool favorite, const QStringList &tags )
401 : : {
402 : : // insert it into the database
403 : 0 : QDomDocument doc( QStringLiteral( "dummy" ) );
404 : 0 : QDomElement rampEl = QgsSymbolLayerUtils::saveColorRamp( name, ramp, doc );
405 : :
406 : 0 : if ( rampEl.isNull() )
407 : : {
408 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't convert color ramp to valid XML!" ) );
409 : 0 : return false;
410 : : }
411 : :
412 : 0 : QByteArray xmlArray;
413 : 0 : QTextStream stream( &xmlArray );
414 : 0 : stream.setCodec( "UTF-8" );
415 : 0 : rampEl.save( stream, 4 );
416 : 0 : QString query = qgs_sqlite3_mprintf( "INSERT INTO colorramp VALUES (NULL, '%q', '%q', %d);",
417 : 0 : name.toUtf8().constData(), xmlArray.constData(), ( favorite ? 1 : 0 ) );
418 : 0 : if ( !runEmptyQuery( query ) )
419 : : {
420 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't insert colorramp into the database!" ) );
421 : 0 : return false;
422 : : }
423 : :
424 : 0 : mCachedFavorites[ ColorrampEntity ].insert( name, favorite );
425 : :
426 : 0 : tagSymbol( ColorrampEntity, name, tags );
427 : :
428 : 0 : emit rampAdded( name );
429 : 0 : emit entityAdded( ColorrampEntity, name );
430 : :
431 : 0 : return true;
432 : 0 : }
433 : :
434 : 0 : bool QgsStyle::removeColorRamp( const QString &name )
435 : : {
436 : 0 : return removeEntityByName( ColorrampEntity, name );
437 : : }
438 : :
439 : 0 : QgsColorRamp *QgsStyle::colorRamp( const QString &name ) const
440 : : {
441 : 0 : const QgsColorRamp *ramp = colorRampRef( name );
442 : 0 : return ramp ? ramp->clone() : nullptr;
443 : : }
444 : :
445 : 0 : const QgsColorRamp *QgsStyle::colorRampRef( const QString &name ) const
446 : : {
447 : 0 : return mColorRamps.value( name );
448 : : }
449 : :
450 : 0 : int QgsStyle::colorRampCount()
451 : : {
452 : 0 : return mColorRamps.count();
453 : : }
454 : :
455 : 5 : QStringList QgsStyle::colorRampNames() const
456 : : {
457 : 5 : return mColorRamps.keys();
458 : : }
459 : :
460 : 0 : void QgsStyle::handleDeferred3DSymbolCreation()
461 : : {
462 : 0 : for ( auto it = mDeferred3DsymbolElements.constBegin(); it != mDeferred3DsymbolElements.constEnd(); ++it )
463 : : {
464 : 0 : const QString symbolType = it.value().attribute( QStringLiteral( "type" ) );
465 : 0 : std::unique_ptr< QgsAbstract3DSymbol > symbol( QgsApplication::symbol3DRegistry()->createSymbol( symbolType ) );
466 : 0 : if ( symbol )
467 : : {
468 : 0 : symbol->readXml( it.value(), QgsReadWriteContext() );
469 : 0 : addSymbol3D( it.key(), symbol.release(), false );
470 : 0 : emit entityAdded( Symbol3DEntity, it.key() );
471 : 0 : }
472 : : else
473 : : {
474 : 0 : QgsDebugMsg( "Cannot open 3d symbol " + it.key() );
475 : 0 : continue;
476 : : }
477 : 0 : }
478 : 0 : mDeferred3DsymbolElements.clear();
479 : 0 : }
480 : :
481 : 5 : bool QgsStyle::openDatabase( const QString &filename )
482 : : {
483 : 5 : int rc = mCurrentDB.open( filename );
484 : 5 : if ( rc )
485 : : {
486 : 0 : mErrorString = QStringLiteral( "Couldn't open the style database: %1" ).arg( mCurrentDB.errorMessage() );
487 : 0 : return false;
488 : : }
489 : :
490 : 5 : return true;
491 : 5 : }
492 : :
493 : 0 : bool QgsStyle::createDatabase( const QString &filename )
494 : : {
495 : 0 : mErrorString.clear();
496 : 0 : if ( !openDatabase( filename ) )
497 : : {
498 : 0 : mErrorString = QStringLiteral( "Unable to create database" );
499 : 0 : QgsDebugMsg( mErrorString );
500 : 0 : return false;
501 : : }
502 : :
503 : 0 : createTables();
504 : :
505 : 0 : return true;
506 : 0 : }
507 : :
508 : 0 : bool QgsStyle::createMemoryDatabase()
509 : : {
510 : 0 : mErrorString.clear();
511 : 0 : if ( !openDatabase( QStringLiteral( ":memory:" ) ) )
512 : : {
513 : 0 : mErrorString = QStringLiteral( "Unable to create temporary memory database" );
514 : 0 : QgsDebugMsg( mErrorString );
515 : 0 : return false;
516 : : }
517 : :
518 : 0 : createTables();
519 : :
520 : 0 : return true;
521 : 0 : }
522 : :
523 : 0 : void QgsStyle::createTables()
524 : : {
525 : 0 : QString query = qgs_sqlite3_mprintf( "CREATE TABLE symbol("\
526 : : "id INTEGER PRIMARY KEY,"\
527 : : "name TEXT UNIQUE,"\
528 : : "xml TEXT,"\
529 : : "favorite INTEGER);"\
530 : : "CREATE TABLE colorramp("\
531 : : "id INTEGER PRIMARY KEY,"\
532 : : "name TEXT UNIQUE,"\
533 : : "xml TEXT,"\
534 : : "favorite INTEGER);"\
535 : : "CREATE TABLE textformat("\
536 : : "id INTEGER PRIMARY KEY,"\
537 : : "name TEXT UNIQUE,"\
538 : : "xml TEXT,"\
539 : : "favorite INTEGER);"\
540 : : "CREATE TABLE labelsettings("\
541 : : "id INTEGER PRIMARY KEY,"\
542 : : "name TEXT UNIQUE,"\
543 : : "xml TEXT,"\
544 : : "favorite INTEGER);"\
545 : : "CREATE TABLE legendpatchshapes("\
546 : : "id INTEGER PRIMARY KEY,"\
547 : : "name TEXT UNIQUE,"\
548 : : "xml TEXT,"\
549 : : "favorite INTEGER);"\
550 : : "CREATE TABLE symbol3d("\
551 : : "id INTEGER PRIMARY KEY,"\
552 : : "name TEXT UNIQUE,"\
553 : : "xml TEXT,"\
554 : : "favorite INTEGER);"\
555 : : "CREATE TABLE tag("\
556 : : "id INTEGER PRIMARY KEY,"\
557 : : "name TEXT);"\
558 : : "CREATE TABLE tagmap("\
559 : : "tag_id INTEGER NOT NULL,"\
560 : : "symbol_id INTEGER);"\
561 : : "CREATE TABLE ctagmap("\
562 : : "tag_id INTEGER NOT NULL,"\
563 : : "colorramp_id INTEGER);"\
564 : : "CREATE TABLE tftagmap("\
565 : : "tag_id INTEGER NOT NULL,"\
566 : : "textformat_id INTEGER);"\
567 : : "CREATE TABLE lstagmap("\
568 : : "tag_id INTEGER NOT NULL,"\
569 : : "labelsettings_id INTEGER);"\
570 : : "CREATE TABLE lpstagmap("\
571 : : "tag_id INTEGER NOT NULL,"\
572 : : "legendpatchshape_id INTEGER);"\
573 : : "CREATE TABLE symbol3dtagmap("\
574 : : "tag_id INTEGER NOT NULL,"\
575 : : "symbol3d_id INTEGER);"\
576 : : "CREATE TABLE smartgroup("\
577 : : "id INTEGER PRIMARY KEY,"\
578 : : "name TEXT,"\
579 : : "xml TEXT);" );
580 : 0 : runEmptyQuery( query );
581 : 0 : }
582 : :
583 : 5 : bool QgsStyle::load( const QString &filename )
584 : : {
585 : 5 : mErrorString.clear();
586 : :
587 : : // Open the sqlite database
588 : 5 : if ( !openDatabase( filename ) )
589 : : {
590 : 0 : mErrorString = QStringLiteral( "Unable to open database file specified" );
591 : 0 : QgsDebugMsg( mErrorString );
592 : 0 : return false;
593 : : }
594 : :
595 : : // make sure text format table exists
596 : 5 : QString query = qgs_sqlite3_mprintf( "SELECT name FROM sqlite_master WHERE name='textformat'" );
597 : 5 : sqlite3_statement_unique_ptr statement;
598 : : int rc;
599 : 5 : statement = mCurrentDB.prepare( query, rc );
600 : 5 : if ( rc != SQLITE_OK || sqlite3_step( statement.get() ) != SQLITE_ROW )
601 : : {
602 : 0 : query = qgs_sqlite3_mprintf( "CREATE TABLE textformat("\
603 : : "id INTEGER PRIMARY KEY,"\
604 : : "name TEXT UNIQUE,"\
605 : : "xml TEXT,"\
606 : : "favorite INTEGER);"\
607 : : "CREATE TABLE tftagmap("\
608 : : "tag_id INTEGER NOT NULL,"\
609 : : "textformat_id INTEGER);" );
610 : 0 : runEmptyQuery( query );
611 : 0 : }
612 : : // make sure label settings table exists
613 : 5 : query = qgs_sqlite3_mprintf( "SELECT name FROM sqlite_master WHERE name='labelsettings'" );
614 : 5 : statement = mCurrentDB.prepare( query, rc );
615 : 5 : if ( rc != SQLITE_OK || sqlite3_step( statement.get() ) != SQLITE_ROW )
616 : : {
617 : 0 : query = qgs_sqlite3_mprintf( "CREATE TABLE labelsettings("\
618 : : "id INTEGER PRIMARY KEY,"\
619 : : "name TEXT UNIQUE,"\
620 : : "xml TEXT,"\
621 : : "favorite INTEGER);"\
622 : : "CREATE TABLE lstagmap("\
623 : : "tag_id INTEGER NOT NULL,"\
624 : : "labelsettings_id INTEGER);" );
625 : 0 : runEmptyQuery( query );
626 : 0 : }
627 : : // make sure legend patch shape table exists
628 : 5 : query = qgs_sqlite3_mprintf( "SELECT name FROM sqlite_master WHERE name='legendpatchshapes'" );
629 : 5 : statement = mCurrentDB.prepare( query, rc );
630 : 5 : if ( rc != SQLITE_OK || sqlite3_step( statement.get() ) != SQLITE_ROW )
631 : : {
632 : 0 : query = qgs_sqlite3_mprintf( "CREATE TABLE legendpatchshapes("\
633 : : "id INTEGER PRIMARY KEY,"\
634 : : "name TEXT UNIQUE,"\
635 : : "xml TEXT,"\
636 : : "favorite INTEGER);"\
637 : : "CREATE TABLE lpstagmap("\
638 : : "tag_id INTEGER NOT NULL,"\
639 : : "legendpatchshape_id INTEGER);" );
640 : 0 : runEmptyQuery( query );
641 : 0 : }
642 : : // make sure 3d symbol table exists
643 : 5 : query = qgs_sqlite3_mprintf( "SELECT name FROM sqlite_master WHERE name='symbol3d'" );
644 : 5 : statement = mCurrentDB.prepare( query, rc );
645 : 5 : if ( rc != SQLITE_OK || sqlite3_step( statement.get() ) != SQLITE_ROW )
646 : : {
647 : 0 : query = qgs_sqlite3_mprintf( "CREATE TABLE symbol3d("\
648 : : "id INTEGER PRIMARY KEY,"\
649 : : "name TEXT UNIQUE,"\
650 : : "xml TEXT,"\
651 : : "favorite INTEGER);"\
652 : : "CREATE TABLE symbol3dtagmap("\
653 : : "tag_id INTEGER NOT NULL,"\
654 : : "symbol3d_id INTEGER);" );
655 : 0 : runEmptyQuery( query );
656 : 0 : }
657 : :
658 : : // Make sure there are no Null fields in parenting symbols and groups
659 : 5 : query = qgs_sqlite3_mprintf( "UPDATE symbol SET favorite=0 WHERE favorite IS NULL;"
660 : : "UPDATE colorramp SET favorite=0 WHERE favorite IS NULL;"
661 : : "UPDATE textformat SET favorite=0 WHERE favorite IS NULL;"
662 : : "UPDATE labelsettings SET favorite=0 WHERE favorite IS NULL;"
663 : : "UPDATE legendpatchshapes SET favorite=0 WHERE favorite IS NULL;"
664 : : "UPDATE symbol3d SET favorite=0 WHERE favorite IS NULL;"
665 : : );
666 : 5 : runEmptyQuery( query );
667 : :
668 : : {
669 : 5 : QgsScopedRuntimeProfile profile( tr( "Load symbols" ) );
670 : : // First create all the main symbols
671 : 5 : query = qgs_sqlite3_mprintf( "SELECT * FROM symbol" );
672 : 5 : statement = mCurrentDB.prepare( query, rc );
673 : :
674 : 580 : while ( rc == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
675 : : {
676 : 575 : QDomDocument doc;
677 : 575 : QString symbolName = statement.columnAsText( SymbolName );
678 : 575 : QgsScopedRuntimeProfile profile( symbolName );
679 : 575 : QString xmlstring = statement.columnAsText( SymbolXML );
680 : 575 : if ( !doc.setContent( xmlstring ) )
681 : : {
682 : 0 : QgsDebugMsg( "Cannot open symbol " + symbolName );
683 : 0 : continue;
684 : : }
685 : :
686 : 575 : QDomElement symElement = doc.documentElement();
687 : 575 : QgsSymbol *symbol = QgsSymbolLayerUtils::loadSymbol( symElement, QgsReadWriteContext() );
688 : 575 : if ( symbol )
689 : 575 : mSymbols.insert( symbolName, symbol );
690 : 575 : }
691 : 5 : }
692 : :
693 : : {
694 : 5 : QgsScopedRuntimeProfile profile( tr( "Load color ramps" ) );
695 : 5 : query = qgs_sqlite3_mprintf( "SELECT * FROM colorramp" );
696 : 5 : statement = mCurrentDB.prepare( query, rc );
697 : 160 : while ( rc == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
698 : : {
699 : 155 : QDomDocument doc;
700 : 155 : const QString rampName = statement.columnAsText( ColorrampName );
701 : 155 : QgsScopedRuntimeProfile profile( rampName );
702 : 155 : QString xmlstring = statement.columnAsText( ColorrampXML );
703 : 155 : if ( !doc.setContent( xmlstring ) )
704 : : {
705 : 0 : QgsDebugMsg( "Cannot open symbol " + rampName );
706 : 0 : continue;
707 : : }
708 : 155 : QDomElement rampElement = doc.documentElement();
709 : 155 : QgsColorRamp *ramp = QgsSymbolLayerUtils::loadColorRamp( rampElement );
710 : 155 : if ( ramp )
711 : 155 : mColorRamps.insert( rampName, ramp );
712 : 155 : }
713 : 5 : }
714 : :
715 : : {
716 : 5 : QgsScopedRuntimeProfile profile( tr( "Load text formats" ) );
717 : 5 : query = qgs_sqlite3_mprintf( "SELECT * FROM textformat" );
718 : 5 : statement = mCurrentDB.prepare( query, rc );
719 : 5 : while ( rc == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
720 : : {
721 : 0 : QDomDocument doc;
722 : 0 : const QString formatName = statement.columnAsText( TextFormatName );
723 : 0 : QgsScopedRuntimeProfile profile( formatName );
724 : 0 : const QString xmlstring = statement.columnAsText( TextFormatXML );
725 : 0 : if ( !doc.setContent( xmlstring ) )
726 : : {
727 : 0 : QgsDebugMsg( "Cannot open text format " + formatName );
728 : 0 : continue;
729 : : }
730 : 0 : QDomElement formatElement = doc.documentElement();
731 : 0 : QgsTextFormat format;
732 : 0 : format.readXml( formatElement, QgsReadWriteContext() );
733 : 0 : mTextFormats.insert( formatName, format );
734 : 0 : }
735 : 5 : }
736 : :
737 : : {
738 : 5 : QgsScopedRuntimeProfile profile( tr( "Load label settings" ) );
739 : 5 : query = qgs_sqlite3_mprintf( "SELECT * FROM labelsettings" );
740 : 5 : statement = mCurrentDB.prepare( query, rc );
741 : 5 : while ( rc == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
742 : : {
743 : 0 : QDomDocument doc;
744 : 0 : const QString settingsName = statement.columnAsText( LabelSettingsName );
745 : 0 : QgsScopedRuntimeProfile profile( settingsName );
746 : 0 : const QString xmlstring = statement.columnAsText( LabelSettingsXML );
747 : 0 : if ( !doc.setContent( xmlstring ) )
748 : : {
749 : 0 : QgsDebugMsg( "Cannot open label settings " + settingsName );
750 : 0 : continue;
751 : : }
752 : 0 : QDomElement settingsElement = doc.documentElement();
753 : 0 : QgsPalLayerSettings settings;
754 : 0 : settings.readXml( settingsElement, QgsReadWriteContext() );
755 : 0 : mLabelSettings.insert( settingsName, settings );
756 : 0 : }
757 : 5 : }
758 : :
759 : : {
760 : 5 : QgsScopedRuntimeProfile profile( tr( "Load legend patch shapes" ) );
761 : 5 : query = qgs_sqlite3_mprintf( "SELECT * FROM legendpatchshapes" );
762 : 5 : statement = mCurrentDB.prepare( query, rc );
763 : 5 : while ( rc == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
764 : : {
765 : 0 : QDomDocument doc;
766 : 0 : const QString settingsName = statement.columnAsText( LegendPatchTableName );
767 : 0 : QgsScopedRuntimeProfile profile( settingsName );
768 : 0 : const QString xmlstring = statement.columnAsText( LegendPatchTableXML );
769 : 0 : if ( !doc.setContent( xmlstring ) )
770 : : {
771 : 0 : QgsDebugMsg( "Cannot open legend patch shape " + settingsName );
772 : 0 : continue;
773 : : }
774 : 0 : QDomElement settingsElement = doc.documentElement();
775 : 0 : QgsLegendPatchShape shape;
776 : 0 : shape.readXml( settingsElement, QgsReadWriteContext() );
777 : 0 : mLegendPatchShapes.insert( settingsName, shape );
778 : 0 : }
779 : 5 : }
780 : :
781 : : {
782 : 5 : QgsScopedRuntimeProfile profile( tr( "Load 3D symbols shapes" ) );
783 : 5 : query = qgs_sqlite3_mprintf( "SELECT * FROM symbol3d" );
784 : 5 : statement = mCurrentDB.prepare( query, rc );
785 : :
786 : 5 : const bool registry3dPopulated = !QgsApplication::symbol3DRegistry()->symbolTypes().empty();
787 : :
788 : 5 : while ( rc == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
789 : : {
790 : 0 : QDomDocument doc;
791 : 0 : const QString settingsName = statement.columnAsText( Symbol3DTableName );
792 : 0 : QgsScopedRuntimeProfile profile( settingsName );
793 : 0 : const QString xmlstring = statement.columnAsText( Symbol3DTableXML );
794 : 0 : if ( !doc.setContent( xmlstring ) )
795 : : {
796 : 0 : QgsDebugMsg( "Cannot open 3d symbol " + settingsName );
797 : 0 : continue;
798 : : }
799 : 0 : QDomElement settingsElement = doc.documentElement();
800 : :
801 : 0 : if ( !registry3dPopulated )
802 : : {
803 : 0 : mDeferred3DsymbolElements.insert( settingsName, settingsElement );
804 : 0 : }
805 : : else
806 : : {
807 : 0 : const QString symbolType = settingsElement.attribute( QStringLiteral( "type" ) );
808 : 0 : std::unique_ptr< QgsAbstract3DSymbol > symbol( QgsApplication::symbol3DRegistry()->createSymbol( symbolType ) );
809 : 0 : if ( symbol )
810 : : {
811 : 0 : symbol->readXml( settingsElement, QgsReadWriteContext() );
812 : 0 : m3dSymbols.insert( settingsName, symbol.release() );
813 : 0 : }
814 : : else
815 : : {
816 : 0 : QgsDebugMsg( "Cannot open 3d symbol " + settingsName );
817 : 0 : continue;
818 : : }
819 : 0 : }
820 : 0 : }
821 : 5 : }
822 : :
823 : 5 : mFileName = filename;
824 : 5 : return true;
825 : 5 : }
826 : :
827 : :
828 : :
829 : 0 : bool QgsStyle::save( QString filename )
830 : : {
831 : 0 : mErrorString.clear();
832 : :
833 : 0 : if ( filename.isEmpty() )
834 : 0 : filename = mFileName;
835 : :
836 : : // TODO evaluate the requirement of this function and change implementation accordingly
837 : : // TODO remove QEXPECT_FAIL from TestStyle::testSaveLoad() when done
838 : : #if 0
839 : : QDomDocument doc( "qgis_style" );
840 : : QDomElement root = doc.createElement( "qgis_style" );
841 : : root.setAttribute( "version", STYLE_CURRENT_VERSION );
842 : : doc.appendChild( root );
843 : :
844 : : QDomElement symbolsElem = QgsSymbolLayerUtils::saveSymbols( mSymbols, "symbols", doc );
845 : :
846 : : QDomElement rampsElem = doc.createElement( "colorramps" );
847 : :
848 : : // save color ramps
849 : : for ( QMap<QString, QgsColorRamp *>::iterator itr = mColorRamps.begin(); itr != mColorRamps.end(); ++itr )
850 : : {
851 : : QDomElement rampEl = QgsSymbolLayerUtils::saveColorRamp( itr.key(), itr.value(), doc );
852 : : rampsElem.appendChild( rampEl );
853 : : }
854 : :
855 : : root.appendChild( symbolsElem );
856 : : root.appendChild( rampsElem );
857 : :
858 : : // save
859 : : QFile f( filename );
860 : : if ( !f.open( QFile::WriteOnly ) )
861 : : {
862 : : mErrorString = "Couldn't open file for writing: " + filename;
863 : : return false;
864 : : }
865 : : QTextStream ts( &f );
866 : : ts.setCodec( "UTF-8" );
867 : : doc.save( ts, 2 );
868 : : f.close();
869 : : #endif
870 : :
871 : 0 : mFileName = filename;
872 : 0 : return true;
873 : : }
874 : :
875 : 0 : bool QgsStyle::renameSymbol( const QString &oldName, const QString &newName )
876 : : {
877 : 0 : if ( mSymbols.contains( newName ) )
878 : : {
879 : 0 : QgsDebugMsg( QStringLiteral( "Symbol of new name already exists" ) );
880 : 0 : return false;
881 : : }
882 : :
883 : 0 : QgsSymbol *symbol = mSymbols.take( oldName );
884 : 0 : if ( !symbol )
885 : 0 : return false;
886 : :
887 : 0 : mSymbols.insert( newName, symbol );
888 : :
889 : 0 : if ( !mCurrentDB )
890 : : {
891 : 0 : QgsDebugMsg( QStringLiteral( "Sorry! Cannot open database to tag." ) );
892 : 0 : return false;
893 : : }
894 : :
895 : 0 : int symbolid = symbolId( oldName );
896 : 0 : if ( !symbolid )
897 : : {
898 : 0 : QgsDebugMsg( QStringLiteral( "No such symbol for tagging in database: " ) + oldName );
899 : 0 : return false;
900 : : }
901 : :
902 : 0 : mCachedTags[ SymbolEntity ].remove( oldName );
903 : 0 : mCachedFavorites[ SymbolEntity ].remove( oldName );
904 : :
905 : 0 : const bool result = rename( SymbolEntity, symbolid, newName );
906 : 0 : if ( result )
907 : : {
908 : 0 : emit symbolRenamed( oldName, newName );
909 : 0 : emit entityRenamed( SymbolEntity, oldName, newName );
910 : 0 : }
911 : :
912 : 0 : return result;
913 : 0 : }
914 : :
915 : 0 : bool QgsStyle::renameColorRamp( const QString &oldName, const QString &newName )
916 : : {
917 : 0 : if ( mColorRamps.contains( newName ) )
918 : : {
919 : 0 : QgsDebugMsg( QStringLiteral( "Color ramp of new name already exists." ) );
920 : 0 : return false;
921 : : }
922 : :
923 : 0 : QgsColorRamp *ramp = mColorRamps.take( oldName );
924 : 0 : if ( !ramp )
925 : 0 : return false;
926 : :
927 : 0 : mColorRamps.insert( newName, ramp );
928 : 0 : mCachedTags[ ColorrampEntity ].remove( oldName );
929 : 0 : mCachedFavorites[ ColorrampEntity ].remove( oldName );
930 : :
931 : 0 : int rampid = 0;
932 : 0 : sqlite3_statement_unique_ptr statement;
933 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT id FROM colorramp WHERE name='%q'", oldName.toUtf8().constData() );
934 : : int nErr;
935 : 0 : statement = mCurrentDB.prepare( query, nErr );
936 : 0 : if ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
937 : : {
938 : 0 : rampid = sqlite3_column_int( statement.get(), 0 );
939 : 0 : }
940 : 0 : const bool result = rename( ColorrampEntity, rampid, newName );
941 : 0 : if ( result )
942 : : {
943 : 0 : emit rampRenamed( oldName, newName );
944 : 0 : emit entityRenamed( ColorrampEntity, oldName, newName );
945 : 0 : }
946 : :
947 : 0 : return result;
948 : 0 : }
949 : :
950 : 0 : bool QgsStyle::saveTextFormat( const QString &name, const QgsTextFormat &format, bool favorite, const QStringList &tags )
951 : : {
952 : : // insert it into the database
953 : 0 : QDomDocument doc( QStringLiteral( "dummy" ) );
954 : 0 : QDomElement formatElem = format.writeXml( doc, QgsReadWriteContext() );
955 : :
956 : 0 : if ( formatElem.isNull() )
957 : : {
958 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't convert text format to valid XML!" ) );
959 : 0 : return false;
960 : : }
961 : :
962 : 0 : QByteArray xmlArray;
963 : 0 : QTextStream stream( &xmlArray );
964 : 0 : stream.setCodec( "UTF-8" );
965 : 0 : formatElem.save( stream, 4 );
966 : 0 : QString query = qgs_sqlite3_mprintf( "INSERT INTO textformat VALUES (NULL, '%q', '%q', %d);",
967 : 0 : name.toUtf8().constData(), xmlArray.constData(), ( favorite ? 1 : 0 ) );
968 : 0 : if ( !runEmptyQuery( query ) )
969 : : {
970 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't insert text format into the database!" ) );
971 : 0 : return false;
972 : : }
973 : :
974 : 0 : mCachedFavorites[ TextFormatEntity ].insert( name, favorite );
975 : :
976 : 0 : tagSymbol( TextFormatEntity, name, tags );
977 : :
978 : 0 : emit textFormatAdded( name );
979 : 0 : emit entityAdded( TextFormatEntity, name );
980 : :
981 : 0 : return true;
982 : 0 : }
983 : :
984 : 0 : bool QgsStyle::removeTextFormat( const QString &name )
985 : : {
986 : 0 : return removeEntityByName( TextFormatEntity, name );
987 : : }
988 : :
989 : 0 : bool QgsStyle::renameTextFormat( const QString &oldName, const QString &newName )
990 : : {
991 : 0 : if ( mTextFormats.contains( newName ) )
992 : : {
993 : 0 : QgsDebugMsg( QStringLiteral( "Text format of new name already exists." ) );
994 : 0 : return false;
995 : : }
996 : :
997 : 0 : if ( !mTextFormats.contains( oldName ) )
998 : 0 : return false;
999 : 0 : QgsTextFormat format = mTextFormats.take( oldName );
1000 : :
1001 : 0 : mTextFormats.insert( newName, format );
1002 : 0 : mCachedTags[ TextFormatEntity ].remove( oldName );
1003 : 0 : mCachedFavorites[ TextFormatEntity ].remove( oldName );
1004 : :
1005 : 0 : int textFormatId = 0;
1006 : 0 : sqlite3_statement_unique_ptr statement;
1007 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT id FROM textformat WHERE name='%q'", oldName.toUtf8().constData() );
1008 : : int nErr;
1009 : 0 : statement = mCurrentDB.prepare( query, nErr );
1010 : 0 : if ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1011 : : {
1012 : 0 : textFormatId = sqlite3_column_int( statement.get(), 0 );
1013 : 0 : }
1014 : 0 : const bool result = rename( TextFormatEntity, textFormatId, newName );
1015 : 0 : if ( result )
1016 : : {
1017 : 0 : emit textFormatRenamed( oldName, newName );
1018 : 0 : emit entityRenamed( TextFormatEntity, oldName, newName );
1019 : 0 : }
1020 : :
1021 : 0 : return result;
1022 : 0 : }
1023 : :
1024 : 0 : bool QgsStyle::saveLabelSettings( const QString &name, const QgsPalLayerSettings &settings, bool favorite, const QStringList &tags )
1025 : : {
1026 : : // insert it into the database
1027 : 0 : QDomDocument doc( QStringLiteral( "dummy" ) );
1028 : 0 : QDomElement settingsElem = settings.writeXml( doc, QgsReadWriteContext() );
1029 : :
1030 : 0 : if ( settingsElem.isNull() )
1031 : : {
1032 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't convert label settings to valid XML!" ) );
1033 : 0 : return false;
1034 : : }
1035 : :
1036 : 0 : QByteArray xmlArray;
1037 : 0 : QTextStream stream( &xmlArray );
1038 : 0 : stream.setCodec( "UTF-8" );
1039 : 0 : settingsElem.save( stream, 4 );
1040 : 0 : QString query = qgs_sqlite3_mprintf( "INSERT INTO labelsettings VALUES (NULL, '%q', '%q', %d);",
1041 : 0 : name.toUtf8().constData(), xmlArray.constData(), ( favorite ? 1 : 0 ) );
1042 : 0 : if ( !runEmptyQuery( query ) )
1043 : : {
1044 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't insert label settings into the database!" ) );
1045 : 0 : return false;
1046 : : }
1047 : :
1048 : 0 : mCachedFavorites[ LabelSettingsEntity ].insert( name, favorite );
1049 : :
1050 : 0 : tagSymbol( LabelSettingsEntity, name, tags );
1051 : :
1052 : 0 : emit labelSettingsAdded( name );
1053 : 0 : emit entityAdded( LabelSettingsEntity, name );
1054 : :
1055 : 0 : return true;
1056 : 0 : }
1057 : :
1058 : 0 : bool QgsStyle::removeLabelSettings( const QString &name )
1059 : : {
1060 : 0 : return removeEntityByName( LabelSettingsEntity, name );
1061 : : }
1062 : :
1063 : 0 : bool QgsStyle::renameLabelSettings( const QString &oldName, const QString &newName )
1064 : : {
1065 : 0 : if ( mLabelSettings.contains( newName ) )
1066 : : {
1067 : 0 : QgsDebugMsg( QStringLiteral( "Label settings of new name already exists." ) );
1068 : 0 : return false;
1069 : : }
1070 : :
1071 : 0 : if ( !mLabelSettings.contains( oldName ) )
1072 : 0 : return false;
1073 : 0 : QgsPalLayerSettings settings = mLabelSettings.take( oldName );
1074 : :
1075 : 0 : mLabelSettings.insert( newName, settings );
1076 : 0 : mCachedTags[ LabelSettingsEntity ].remove( oldName );
1077 : 0 : mCachedFavorites[ LabelSettingsEntity ].remove( oldName );
1078 : :
1079 : 0 : int labelSettingsId = 0;
1080 : 0 : sqlite3_statement_unique_ptr statement;
1081 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT id FROM labelsettings WHERE name='%q'", oldName.toUtf8().constData() );
1082 : : int nErr;
1083 : 0 : statement = mCurrentDB.prepare( query, nErr );
1084 : 0 : if ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1085 : : {
1086 : 0 : labelSettingsId = sqlite3_column_int( statement.get(), 0 );
1087 : 0 : }
1088 : 0 : const bool result = rename( LabelSettingsEntity, labelSettingsId, newName );
1089 : 0 : if ( result )
1090 : : {
1091 : 0 : emit labelSettingsRenamed( oldName, newName );
1092 : 0 : emit entityRenamed( LabelSettingsEntity, oldName, newName );
1093 : 0 : }
1094 : :
1095 : 0 : return result;
1096 : 0 : }
1097 : :
1098 : 0 : bool QgsStyle::saveLegendPatchShape( const QString &name, const QgsLegendPatchShape &shape, bool favorite, const QStringList &tags )
1099 : : {
1100 : : // insert it into the database
1101 : 0 : QDomDocument doc( QStringLiteral( "dummy" ) );
1102 : 0 : QDomElement shapeElem = doc.createElement( QStringLiteral( "shape" ) );
1103 : 0 : shape.writeXml( shapeElem, doc, QgsReadWriteContext() );
1104 : :
1105 : 0 : QByteArray xmlArray;
1106 : 0 : QTextStream stream( &xmlArray );
1107 : 0 : stream.setCodec( "UTF-8" );
1108 : 0 : shapeElem.save( stream, 4 );
1109 : 0 : QString query = qgs_sqlite3_mprintf( "INSERT INTO legendpatchshapes VALUES (NULL, '%q', '%q', %d);",
1110 : 0 : name.toUtf8().constData(), xmlArray.constData(), ( favorite ? 1 : 0 ) );
1111 : 0 : if ( !runEmptyQuery( query ) )
1112 : : {
1113 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't insert legend patch shape into the database!" ) );
1114 : 0 : return false;
1115 : : }
1116 : :
1117 : 0 : mCachedFavorites[ LegendPatchShapeEntity ].insert( name, favorite );
1118 : :
1119 : 0 : tagSymbol( LegendPatchShapeEntity, name, tags );
1120 : :
1121 : 0 : emit entityAdded( LegendPatchShapeEntity, name );
1122 : :
1123 : 0 : return true;
1124 : 0 : }
1125 : :
1126 : 0 : bool QgsStyle::renameLegendPatchShape( const QString &oldName, const QString &newName )
1127 : : {
1128 : 0 : if ( mLegendPatchShapes.contains( newName ) )
1129 : : {
1130 : 0 : QgsDebugMsg( QStringLiteral( "Legend patch shape of new name already exists." ) );
1131 : 0 : return false;
1132 : : }
1133 : :
1134 : 0 : if ( !mLegendPatchShapes.contains( oldName ) )
1135 : 0 : return false;
1136 : 0 : QgsLegendPatchShape shape = mLegendPatchShapes.take( oldName );
1137 : :
1138 : 0 : mLegendPatchShapes.insert( newName, shape );
1139 : 0 : mCachedTags[ LegendPatchShapeEntity ].remove( oldName );
1140 : 0 : mCachedFavorites[ LegendPatchShapeEntity ].remove( oldName );
1141 : :
1142 : 0 : int labelSettingsId = 0;
1143 : 0 : sqlite3_statement_unique_ptr statement;
1144 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT id FROM legendpatchshapes WHERE name='%q'", oldName.toUtf8().constData() );
1145 : : int nErr;
1146 : 0 : statement = mCurrentDB.prepare( query, nErr );
1147 : 0 : if ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1148 : : {
1149 : 0 : labelSettingsId = sqlite3_column_int( statement.get(), 0 );
1150 : 0 : }
1151 : 0 : const bool result = rename( LegendPatchShapeEntity, labelSettingsId, newName );
1152 : 0 : if ( result )
1153 : : {
1154 : 0 : emit entityRenamed( LegendPatchShapeEntity, oldName, newName );
1155 : 0 : }
1156 : :
1157 : 0 : return result;
1158 : 0 : }
1159 : :
1160 : 0 : QgsLegendPatchShape QgsStyle::defaultPatch( QgsSymbol::SymbolType type, QSizeF size ) const
1161 : : {
1162 : 0 : if ( type == QgsSymbol::Hybrid )
1163 : 0 : return QgsLegendPatchShape();
1164 : :
1165 : 0 : if ( mDefaultPatchCache[ type ].contains( size ) )
1166 : 0 : return mDefaultPatchCache[ type ].value( size );
1167 : :
1168 : 0 : QgsGeometry geom;
1169 : 0 : switch ( type )
1170 : : {
1171 : : case QgsSymbol::Marker:
1172 : 0 : geom = QgsGeometry( std::make_unique< QgsPoint >( static_cast< int >( size.width() ) / 2, static_cast< int >( size.height() ) / 2 ) );
1173 : 0 : break;
1174 : :
1175 : : case QgsSymbol::Line:
1176 : : {
1177 : : // we're adding 0.5 to get rid of blurred preview:
1178 : : // drawing antialiased lines of width 1 at (x,0)-(x,100) creates 2px line
1179 : 0 : double y = static_cast< int >( size.height() ) / 2 + 0.5;
1180 : 0 : geom = QgsGeometry( std::make_unique< QgsLineString >( ( QVector< double >() << 0 << size.width() ),
1181 : 0 : ( QVector< double >() << y << y ) ) );
1182 : 0 : break;
1183 : : }
1184 : :
1185 : : case QgsSymbol::Fill:
1186 : : {
1187 : 0 : geom = QgsGeometry( std::make_unique< QgsPolygon >(
1188 : 0 : new QgsLineString( QVector< double >() << 0 << static_cast< int >( size.width() ) << static_cast< int >( size.width() ) << 0 << 0,
1189 : 0 : QVector< double >() << static_cast< int >( size.height() ) << static_cast< int >( size.height() ) << 0 << 0 << static_cast< int >( size.height() ) ) ) );
1190 : 0 : break;
1191 : : }
1192 : :
1193 : : case QgsSymbol::Hybrid:
1194 : 0 : break;
1195 : : }
1196 : :
1197 : 0 : QgsLegendPatchShape res = QgsLegendPatchShape( type, geom, false );
1198 : 0 : mDefaultPatchCache[ type ][size ] = res;
1199 : 0 : return res;
1200 : 0 : }
1201 : :
1202 : 0 : QList<QList<QPolygonF> > QgsStyle::defaultPatchAsQPolygonF( QgsSymbol::SymbolType type, QSizeF size ) const
1203 : : {
1204 : 0 : if ( type == QgsSymbol::Hybrid )
1205 : 0 : return QList<QList<QPolygonF> >();
1206 : :
1207 : 0 : if ( mDefaultPatchQPolygonFCache[ type ].contains( size ) )
1208 : 0 : return mDefaultPatchQPolygonFCache[ type ].value( size );
1209 : :
1210 : 0 : QList<QList<QPolygonF> > res = defaultPatch( type, size ).toQPolygonF( type, size );
1211 : 0 : mDefaultPatchQPolygonFCache[ type ][size ] = res;
1212 : 0 : return res;
1213 : 0 : }
1214 : :
1215 : 0 : bool QgsStyle::saveSymbol3D( const QString &name, QgsAbstract3DSymbol *symbol, bool favorite, const QStringList &tags )
1216 : : {
1217 : : // insert it into the database
1218 : 0 : QDomDocument doc( QStringLiteral( "dummy" ) );
1219 : 0 : QDomElement elem = doc.createElement( QStringLiteral( "symbol" ) );
1220 : 0 : elem.setAttribute( QStringLiteral( "type" ), symbol->type() );
1221 : 0 : symbol->writeXml( elem, QgsReadWriteContext() );
1222 : :
1223 : 0 : QByteArray xmlArray;
1224 : 0 : QTextStream stream( &xmlArray );
1225 : 0 : stream.setCodec( "UTF-8" );
1226 : 0 : elem.save( stream, 4 );
1227 : 0 : QString query = qgs_sqlite3_mprintf( "INSERT INTO symbol3d VALUES (NULL, '%q', '%q', %d);",
1228 : 0 : name.toUtf8().constData(), xmlArray.constData(), ( favorite ? 1 : 0 ) );
1229 : 0 : if ( !runEmptyQuery( query ) )
1230 : : {
1231 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't insert 3d symbol into the database!" ) );
1232 : 0 : return false;
1233 : : }
1234 : :
1235 : 0 : mCachedFavorites[ Symbol3DEntity ].insert( name, favorite );
1236 : :
1237 : 0 : tagSymbol( Symbol3DEntity, name, tags );
1238 : :
1239 : 0 : emit entityAdded( Symbol3DEntity, name );
1240 : :
1241 : 0 : return true;
1242 : 0 : }
1243 : :
1244 : 0 : bool QgsStyle::renameSymbol3D( const QString &oldName, const QString &newName )
1245 : : {
1246 : 0 : if ( m3dSymbols.contains( newName ) )
1247 : : {
1248 : 0 : QgsDebugMsg( QStringLiteral( "3d symbol of new name already exists." ) );
1249 : 0 : return false;
1250 : : }
1251 : :
1252 : 0 : if ( !m3dSymbols.contains( oldName ) )
1253 : 0 : return false;
1254 : 0 : QgsAbstract3DSymbol *symbol = m3dSymbols.take( oldName );
1255 : :
1256 : 0 : m3dSymbols.insert( newName, symbol );
1257 : 0 : mCachedTags[Symbol3DEntity ].remove( oldName );
1258 : 0 : mCachedFavorites[ Symbol3DEntity ].remove( oldName );
1259 : :
1260 : 0 : int labelSettingsId = 0;
1261 : 0 : sqlite3_statement_unique_ptr statement;
1262 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT id FROM symbol3d WHERE name='%q'", oldName.toUtf8().constData() );
1263 : : int nErr;
1264 : 0 : statement = mCurrentDB.prepare( query, nErr );
1265 : 0 : if ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1266 : : {
1267 : 0 : labelSettingsId = sqlite3_column_int( statement.get(), 0 );
1268 : 0 : }
1269 : 0 : const bool result = rename( Symbol3DEntity, labelSettingsId, newName );
1270 : 0 : if ( result )
1271 : : {
1272 : 0 : emit entityRenamed( Symbol3DEntity, oldName, newName );
1273 : 0 : }
1274 : :
1275 : 0 : return result;
1276 : 0 : }
1277 : :
1278 : 5 : QStringList QgsStyle::symbol3DNames() const
1279 : : {
1280 : 5 : return m3dSymbols.keys();
1281 : : }
1282 : :
1283 : 0 : QStringList QgsStyle::symbolsOfFavorite( StyleEntity type ) const
1284 : : {
1285 : 0 : if ( !mCurrentDB )
1286 : : {
1287 : 0 : QgsDebugMsg( QStringLiteral( "Cannot Open database for getting favorite symbols" ) );
1288 : 0 : return QStringList();
1289 : : }
1290 : :
1291 : 0 : QString query;
1292 : 0 : switch ( type )
1293 : : {
1294 : : case TagEntity:
1295 : : case SmartgroupEntity:
1296 : 0 : QgsDebugMsg( QStringLiteral( "No such style entity" ) );
1297 : 0 : return QStringList();
1298 : :
1299 : : default:
1300 : 0 : query = qgs_sqlite3_mprintf( QStringLiteral( "SELECT name FROM %1 WHERE favorite=1" ).arg( entityTableName( type ) ).toLocal8Bit().data() );
1301 : 0 : break;
1302 : : }
1303 : :
1304 : : int nErr;
1305 : 0 : sqlite3_statement_unique_ptr statement;
1306 : 0 : statement = mCurrentDB.prepare( query, nErr );
1307 : :
1308 : 0 : QStringList symbols;
1309 : 0 : while ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1310 : : {
1311 : 0 : symbols << statement.columnAsText( 0 );
1312 : : }
1313 : :
1314 : 0 : return symbols;
1315 : 0 : }
1316 : :
1317 : 0 : QStringList QgsStyle::symbolsWithTag( StyleEntity type, int tagid ) const
1318 : : {
1319 : 0 : if ( !mCurrentDB )
1320 : : {
1321 : 0 : QgsDebugMsg( QStringLiteral( "Cannot open database to get symbols of tagid %1" ).arg( tagid ) );
1322 : 0 : return QStringList();
1323 : : }
1324 : :
1325 : 0 : QString subquery;
1326 : 0 : switch ( type )
1327 : : {
1328 : : case TagEntity:
1329 : : case SmartgroupEntity:
1330 : 0 : QgsDebugMsg( QStringLiteral( "Unknown Entity" ) );
1331 : 0 : return QStringList();
1332 : :
1333 : : default:
1334 : 0 : subquery = qgs_sqlite3_mprintf( QStringLiteral( "SELECT %1 FROM %2 WHERE tag_id=%d" ).arg( tagmapEntityIdFieldName( type ),
1335 : 0 : tagmapTableName( type ) ).toLocal8Bit().data(), tagid );
1336 : 0 : break;
1337 : : }
1338 : :
1339 : : int nErr;
1340 : 0 : sqlite3_statement_unique_ptr statement;
1341 : 0 : statement = mCurrentDB.prepare( subquery, nErr );
1342 : :
1343 : : // get the symbol <-> tag connection from the tag map table
1344 : 0 : QStringList symbols;
1345 : 0 : while ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1346 : : {
1347 : 0 : int id = sqlite3_column_int( statement.get(), 0 );
1348 : :
1349 : 0 : const QString query = qgs_sqlite3_mprintf( QStringLiteral( "SELECT name FROM %1 WHERE id=%d" ).arg( entityTableName( type ) ).toLocal8Bit().data(), id );
1350 : :
1351 : : int rc;
1352 : 0 : sqlite3_statement_unique_ptr statement2;
1353 : 0 : statement2 = mCurrentDB.prepare( query, rc );
1354 : 0 : while ( rc == SQLITE_OK && sqlite3_step( statement2.get() ) == SQLITE_ROW )
1355 : : {
1356 : 0 : symbols << statement2.columnAsText( 0 );
1357 : : }
1358 : 0 : }
1359 : :
1360 : 0 : return symbols;
1361 : 0 : }
1362 : :
1363 : 0 : int QgsStyle::addTag( const QString &tagname )
1364 : : {
1365 : 0 : if ( !mCurrentDB )
1366 : 0 : return 0;
1367 : 0 : sqlite3_statement_unique_ptr statement;
1368 : :
1369 : 0 : QString query = qgs_sqlite3_mprintf( "INSERT INTO tag VALUES (NULL, '%q')", tagname.toUtf8().constData() );
1370 : : int nErr;
1371 : 0 : statement = mCurrentDB.prepare( query, nErr );
1372 : 0 : if ( nErr == SQLITE_OK )
1373 : 0 : ( void )sqlite3_step( statement.get() );
1374 : :
1375 : 0 : QgsSettings settings;
1376 : 0 : settings.setValue( QStringLiteral( "qgis/symbolsListGroupsIndex" ), 0 );
1377 : :
1378 : 0 : emit groupsModified();
1379 : :
1380 : 0 : return static_cast< int >( sqlite3_last_insert_rowid( mCurrentDB.get() ) );
1381 : 0 : }
1382 : :
1383 : 0 : QStringList QgsStyle::tags() const
1384 : : {
1385 : 0 : if ( !mCurrentDB )
1386 : 0 : return QStringList();
1387 : :
1388 : 0 : sqlite3_statement_unique_ptr statement;
1389 : :
1390 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT name FROM tag" );
1391 : : int nError;
1392 : 0 : statement = mCurrentDB.prepare( query, nError );
1393 : :
1394 : 0 : QStringList tagList;
1395 : 0 : while ( nError == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1396 : : {
1397 : 0 : tagList << statement.columnAsText( 0 );
1398 : : }
1399 : :
1400 : 0 : return tagList;
1401 : 0 : }
1402 : :
1403 : 0 : bool QgsStyle::rename( StyleEntity type, int id, const QString &newName )
1404 : : {
1405 : 0 : const QString query = qgs_sqlite3_mprintf( QStringLiteral( "UPDATE %1 SET name='%q' WHERE id=%d" ).arg( entityTableName( type ) ).toLocal8Bit().data(), newName.toUtf8().constData(), id );
1406 : :
1407 : 0 : const bool result = runEmptyQuery( query );
1408 : 0 : if ( !result )
1409 : : {
1410 : 0 : mErrorString = QStringLiteral( "Could not rename!" );
1411 : 0 : }
1412 : : else
1413 : : {
1414 : 0 : mCachedTags.clear();
1415 : 0 : mCachedFavorites.clear();
1416 : :
1417 : 0 : switch ( type )
1418 : : {
1419 : : case TagEntity:
1420 : : {
1421 : 0 : emit groupsModified();
1422 : 0 : break;
1423 : : }
1424 : :
1425 : : case SmartgroupEntity:
1426 : : {
1427 : 0 : emit groupsModified();
1428 : 0 : break;
1429 : : }
1430 : :
1431 : : default:
1432 : 0 : break;
1433 : : }
1434 : : }
1435 : 0 : return result;
1436 : 0 : }
1437 : :
1438 : 0 : bool QgsStyle::remove( StyleEntity type, int id )
1439 : : {
1440 : 0 : bool groupRemoved = false;
1441 : 0 : QString query;
1442 : 0 : switch ( type )
1443 : : {
1444 : : case TagEntity:
1445 : 0 : query = qgs_sqlite3_mprintf( "DELETE FROM tag WHERE id=%d; DELETE FROM tagmap WHERE tag_id=%d", id, id );
1446 : 0 : groupRemoved = true;
1447 : 0 : break;
1448 : : case SmartgroupEntity:
1449 : 0 : query = qgs_sqlite3_mprintf( "DELETE FROM smartgroup WHERE id=%d", id );
1450 : 0 : groupRemoved = true;
1451 : 0 : break;
1452 : :
1453 : : default:
1454 : 0 : query = qgs_sqlite3_mprintf( QStringLiteral( "DELETE FROM %1 WHERE id=%d; DELETE FROM %2 WHERE %3=%d" ).arg(
1455 : 0 : entityTableName( type ),
1456 : 0 : tagmapTableName( type ),
1457 : 0 : tagmapEntityIdFieldName( type )
1458 : 0 : ).toLocal8Bit().data(), id, id );
1459 : 0 : break;
1460 : : }
1461 : :
1462 : 0 : bool result = false;
1463 : 0 : if ( !runEmptyQuery( query ) )
1464 : : {
1465 : 0 : QgsDebugMsg( QStringLiteral( "Could not delete entity!" ) );
1466 : 0 : }
1467 : : else
1468 : : {
1469 : 0 : mCachedTags.clear();
1470 : 0 : mCachedFavorites.clear();
1471 : :
1472 : 0 : if ( groupRemoved )
1473 : : {
1474 : 0 : QgsSettings settings;
1475 : 0 : settings.setValue( QStringLiteral( "qgis/symbolsListGroupsIndex" ), 0 );
1476 : :
1477 : 0 : emit groupsModified();
1478 : 0 : }
1479 : 0 : result = true;
1480 : : }
1481 : 0 : return result;
1482 : 0 : }
1483 : :
1484 : 0 : bool QgsStyle::removeEntityByName( QgsStyle::StyleEntity type, const QString &name )
1485 : : {
1486 : 0 : switch ( type )
1487 : : {
1488 : : case QgsStyle::TagEntity:
1489 : : case QgsStyle::SmartgroupEntity:
1490 : 0 : return false;
1491 : :
1492 : : case QgsStyle::SymbolEntity:
1493 : : {
1494 : 0 : std::unique_ptr< QgsSymbol > symbol( mSymbols.take( name ) );
1495 : 0 : if ( !symbol )
1496 : 0 : return false;
1497 : :
1498 : 0 : break;
1499 : 0 : }
1500 : :
1501 : : case QgsStyle::Symbol3DEntity:
1502 : : {
1503 : 0 : std::unique_ptr< QgsAbstract3DSymbol > symbol( m3dSymbols.take( name ) );
1504 : 0 : if ( !symbol )
1505 : 0 : return false;
1506 : :
1507 : 0 : break;
1508 : 0 : }
1509 : :
1510 : : case QgsStyle::ColorrampEntity:
1511 : : {
1512 : 0 : std::unique_ptr< QgsColorRamp > ramp( mColorRamps.take( name ) );
1513 : 0 : if ( !ramp )
1514 : 0 : return false;
1515 : 0 : break;
1516 : 0 : }
1517 : :
1518 : : case QgsStyle::TextFormatEntity:
1519 : : {
1520 : 0 : if ( !mTextFormats.contains( name ) )
1521 : 0 : return false;
1522 : :
1523 : 0 : mTextFormats.remove( name );
1524 : 0 : break;
1525 : : }
1526 : :
1527 : : case QgsStyle::LabelSettingsEntity:
1528 : : {
1529 : 0 : if ( !mLabelSettings.contains( name ) )
1530 : 0 : return false;
1531 : :
1532 : 0 : mLabelSettings.remove( name );
1533 : 0 : break;
1534 : : }
1535 : :
1536 : : case QgsStyle::LegendPatchShapeEntity:
1537 : : {
1538 : 0 : if ( !mLegendPatchShapes.contains( name ) )
1539 : 0 : return false;
1540 : :
1541 : 0 : mLegendPatchShapes.remove( name );
1542 : 0 : break;
1543 : : }
1544 : : }
1545 : :
1546 : 0 : if ( !mCurrentDB )
1547 : : {
1548 : 0 : QgsDebugMsg( QStringLiteral( "Sorry! Cannot open database to modify." ) );
1549 : 0 : return false;
1550 : : }
1551 : :
1552 : 0 : const int id = entityId( type, name );
1553 : 0 : if ( !id )
1554 : : {
1555 : 0 : QgsDebugMsg( "No matching entity for deleting in database: " + name );
1556 : 0 : }
1557 : :
1558 : 0 : const bool result = remove( type, id );
1559 : 0 : if ( result )
1560 : : {
1561 : 0 : mCachedTags[ type ].remove( name );
1562 : 0 : mCachedFavorites[ type ].remove( name );
1563 : :
1564 : 0 : switch ( type )
1565 : : {
1566 : : case SymbolEntity:
1567 : 0 : emit symbolRemoved( name );
1568 : 0 : break;
1569 : :
1570 : : case ColorrampEntity:
1571 : 0 : emit rampRemoved( name );
1572 : 0 : break;
1573 : :
1574 : : case TextFormatEntity:
1575 : 0 : emit textFormatRemoved( name );
1576 : 0 : break;
1577 : :
1578 : : case LabelSettingsEntity:
1579 : 0 : emit labelSettingsRemoved( name );
1580 : 0 : break;
1581 : :
1582 : : default:
1583 : : // these specific signals should be discouraged -- don't add them for new entity types!
1584 : 0 : break;
1585 : : }
1586 : 0 : emit entityRemoved( type, name );
1587 : 0 : }
1588 : 0 : return result;
1589 : 0 : }
1590 : :
1591 : 5 : bool QgsStyle::runEmptyQuery( const QString &query )
1592 : : {
1593 : 5 : if ( !mCurrentDB )
1594 : 0 : return false;
1595 : :
1596 : 5 : char *zErr = nullptr;
1597 : 5 : int nErr = sqlite3_exec( mCurrentDB.get(), query.toUtf8().constData(), nullptr, nullptr, &zErr );
1598 : :
1599 : 5 : if ( nErr != SQLITE_OK )
1600 : : {
1601 : 0 : QgsDebugMsg( zErr );
1602 : 0 : sqlite3_free( zErr );
1603 : 0 : }
1604 : :
1605 : 5 : return nErr == SQLITE_OK;
1606 : 5 : }
1607 : :
1608 : 0 : bool QgsStyle::addFavorite( StyleEntity type, const QString &name )
1609 : : {
1610 : 0 : QString query;
1611 : :
1612 : 0 : switch ( type )
1613 : : {
1614 : : case TagEntity:
1615 : : case SmartgroupEntity:
1616 : 0 : QgsDebugMsg( QStringLiteral( "Wrong entity value. cannot apply group" ) );
1617 : 0 : return false;
1618 : :
1619 : : default:
1620 : 0 : query = qgs_sqlite3_mprintf( QStringLiteral( "UPDATE %1 SET favorite=1 WHERE name='%q'" ).arg( entityTableName( type ) ).toLocal8Bit().data(),
1621 : 0 : name.toUtf8().constData() );
1622 : 0 : break;
1623 : : }
1624 : :
1625 : 0 : const bool res = runEmptyQuery( query );
1626 : 0 : if ( res )
1627 : : {
1628 : 0 : switch ( type )
1629 : : {
1630 : : case TagEntity:
1631 : : case SmartgroupEntity:
1632 : 0 : break;
1633 : :
1634 : : default:
1635 : 0 : mCachedFavorites[ type ].insert( name, true );
1636 : 0 : break;
1637 : : }
1638 : 0 : emit favoritedChanged( type, name, true );
1639 : 0 : }
1640 : :
1641 : 0 : return res;
1642 : 0 : }
1643 : :
1644 : 0 : bool QgsStyle::removeFavorite( StyleEntity type, const QString &name )
1645 : : {
1646 : 0 : QString query;
1647 : :
1648 : 0 : switch ( type )
1649 : : {
1650 : : case TagEntity:
1651 : : case SmartgroupEntity:
1652 : 0 : QgsDebugMsg( QStringLiteral( "Wrong entity value. cannot apply group" ) );
1653 : 0 : return false;
1654 : :
1655 : : default:
1656 : 0 : query = qgs_sqlite3_mprintf( QStringLiteral( "UPDATE %1 SET favorite=0 WHERE name='%q'" ).arg( entityTableName( type ) ).toLocal8Bit().data(), name.toUtf8().constData() );
1657 : 0 : break;
1658 : : }
1659 : :
1660 : 0 : const bool res = runEmptyQuery( query );
1661 : 0 : if ( res )
1662 : : {
1663 : 0 : mCachedFavorites[ type ].insert( name, false );
1664 : 0 : emit favoritedChanged( type, name, false );
1665 : 0 : }
1666 : :
1667 : 0 : return res;
1668 : 0 : }
1669 : :
1670 : 0 : QStringList QgsStyle::findSymbols( StyleEntity type, const QString &qword )
1671 : : {
1672 : 0 : if ( !mCurrentDB )
1673 : : {
1674 : 0 : QgsDebugMsg( QStringLiteral( "Sorry! Cannot open database to search" ) );
1675 : 0 : return QStringList();
1676 : : }
1677 : :
1678 : : // first find symbols with matching name
1679 : 0 : QString item;
1680 : 0 : switch ( type )
1681 : : {
1682 : : case TagEntity:
1683 : : case SmartgroupEntity:
1684 : 0 : return QStringList();
1685 : :
1686 : : default:
1687 : 0 : item = entityTableName( type );
1688 : 0 : break;
1689 : : }
1690 : :
1691 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT name FROM %q WHERE name LIKE '%%%q%%'",
1692 : 0 : item.toUtf8().constData(), qword.toUtf8().constData() );
1693 : :
1694 : 0 : sqlite3_statement_unique_ptr statement;
1695 : 0 : int nErr; statement = mCurrentDB.prepare( query, nErr );
1696 : :
1697 : 0 : QSet< QString > symbols;
1698 : 0 : while ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1699 : : {
1700 : 0 : symbols << statement.columnAsText( 0 );
1701 : : }
1702 : :
1703 : : // next add symbols with matching tags
1704 : 0 : query = qgs_sqlite3_mprintf( "SELECT id FROM tag WHERE name LIKE '%%%q%%'", qword.toUtf8().constData() );
1705 : 0 : statement = mCurrentDB.prepare( query, nErr );
1706 : :
1707 : 0 : QStringList tagids;
1708 : 0 : while ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1709 : : {
1710 : 0 : tagids << statement.columnAsText( 0 );
1711 : : }
1712 : :
1713 : 0 : QString dummy = tagids.join( QLatin1String( ", " ) );
1714 : 0 : query = qgs_sqlite3_mprintf( QStringLiteral( "SELECT %1 FROM %2 WHERE tag_id IN (%q)" ).arg( tagmapEntityIdFieldName( type ),
1715 : 0 : tagmapTableName( type ) ).toLocal8Bit().data(), dummy.toUtf8().constData() );
1716 : :
1717 : 0 : statement = mCurrentDB.prepare( query, nErr );
1718 : :
1719 : 0 : QStringList symbolids;
1720 : 0 : while ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1721 : : {
1722 : 0 : symbolids << statement.columnAsText( 0 );
1723 : : }
1724 : :
1725 : 0 : dummy = symbolids.join( QLatin1String( ", " ) );
1726 : 0 : query = qgs_sqlite3_mprintf( "SELECT name FROM %q WHERE id IN (%q)",
1727 : 0 : item.toUtf8().constData(), dummy.toUtf8().constData() );
1728 : 0 : statement = mCurrentDB.prepare( query, nErr );
1729 : 0 : while ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1730 : : {
1731 : 0 : symbols << statement.columnAsText( 0 );
1732 : : }
1733 : :
1734 : 0 : return qgis::setToList( symbols );
1735 : 0 : }
1736 : :
1737 : 0 : bool QgsStyle::tagSymbol( StyleEntity type, const QString &symbol, const QStringList &tags )
1738 : : {
1739 : 0 : if ( !mCurrentDB )
1740 : : {
1741 : 0 : QgsDebugMsg( QStringLiteral( "Sorry! Cannot open database to tag." ) );
1742 : 0 : return false;
1743 : : }
1744 : :
1745 : 0 : int symbolid = 0;
1746 : 0 : switch ( type )
1747 : : {
1748 : : case TagEntity:
1749 : : case SmartgroupEntity:
1750 : 0 : return false;
1751 : :
1752 : : default:
1753 : 0 : symbolid = entityId( type, symbol );
1754 : 0 : break;
1755 : : }
1756 : :
1757 : 0 : if ( !symbolid )
1758 : : {
1759 : 0 : QgsDebugMsg( QStringLiteral( "No such symbol for tagging in database: " ) + symbol );
1760 : 0 : return false;
1761 : : }
1762 : :
1763 : 0 : QString tag;
1764 : 0 : const auto constTags = tags;
1765 : 0 : for ( const QString &t : constTags )
1766 : : {
1767 : 0 : tag = t.trimmed();
1768 : 0 : if ( !tag.isEmpty() )
1769 : : {
1770 : : // sql: gets the id of the tag if present or insert the tag and get the id of the tag
1771 : 0 : int tagid( tagId( tag ) );
1772 : 0 : if ( ! tagid )
1773 : : {
1774 : 0 : tagid = addTag( tag );
1775 : 0 : }
1776 : :
1777 : : // Now map the tag to the symbol if it's not already tagged
1778 : 0 : if ( !symbolHasTag( type, symbol, tag ) )
1779 : : {
1780 : 0 : QString query = qgs_sqlite3_mprintf( QStringLiteral( "INSERT INTO %1 VALUES (%d,%d)" ).arg( tagmapTableName( type ) ).toLocal8Bit().data(), tagid, symbolid );
1781 : :
1782 : 0 : char *zErr = nullptr;
1783 : : int nErr;
1784 : 0 : nErr = sqlite3_exec( mCurrentDB.get(), query.toUtf8().constData(), nullptr, nullptr, &zErr );
1785 : 0 : if ( nErr )
1786 : : {
1787 : 0 : QgsDebugMsg( zErr );
1788 : 0 : sqlite3_free( zErr );
1789 : 0 : }
1790 : 0 : }
1791 : 0 : }
1792 : : }
1793 : :
1794 : 0 : clearCachedTags( type, symbol );
1795 : 0 : emit entityTagsChanged( type, symbol, tagsOfSymbol( type, symbol ) );
1796 : :
1797 : 0 : return true;
1798 : 0 : }
1799 : :
1800 : 0 : bool QgsStyle::detagSymbol( StyleEntity type, const QString &symbol, const QStringList &tags )
1801 : : {
1802 : 0 : if ( !mCurrentDB )
1803 : : {
1804 : 0 : QgsDebugMsg( QStringLiteral( "Sorry! Cannot open database for detagging." ) );
1805 : 0 : return false;
1806 : : }
1807 : :
1808 : 0 : switch ( type )
1809 : : {
1810 : : case TagEntity:
1811 : : case SmartgroupEntity:
1812 : 0 : return false;
1813 : :
1814 : : default:
1815 : 0 : break;
1816 : : }
1817 : :
1818 : 0 : const int symbolid = entityId( type, symbol );
1819 : 0 : if ( symbolid == 0 )
1820 : 0 : return false;
1821 : :
1822 : : int nErr;
1823 : 0 : QString query;
1824 : 0 : const auto constTags = tags;
1825 : 0 : for ( const QString &tag : constTags )
1826 : : {
1827 : 0 : query = qgs_sqlite3_mprintf( "SELECT id FROM tag WHERE name='%q'", tag.toUtf8().constData() );
1828 : :
1829 : 0 : sqlite3_statement_unique_ptr statement2;
1830 : 0 : statement2 = mCurrentDB.prepare( query, nErr );
1831 : :
1832 : 0 : int tagid = 0;
1833 : 0 : if ( nErr == SQLITE_OK && sqlite3_step( statement2.get() ) == SQLITE_ROW )
1834 : : {
1835 : 0 : tagid = sqlite3_column_int( statement2.get(), 0 );
1836 : 0 : }
1837 : :
1838 : 0 : if ( tagid )
1839 : : {
1840 : : // remove from the tagmap
1841 : 0 : const QString query = qgs_sqlite3_mprintf( QStringLiteral( "DELETE FROM %1 WHERE tag_id=%d AND %2=%d" ).arg( tagmapTableName( type ), tagmapEntityIdFieldName( type ) ).toLocal8Bit().data(), tagid, symbolid );
1842 : 0 : runEmptyQuery( query );
1843 : 0 : }
1844 : 0 : }
1845 : :
1846 : 0 : clearCachedTags( type, symbol );
1847 : 0 : emit entityTagsChanged( type, symbol, tagsOfSymbol( type, symbol ) );
1848 : :
1849 : : // TODO Perform tag cleanup
1850 : : // check the number of entries for a given tag in the tagmap
1851 : : // if the count is 0, then remove( TagEntity, tagid )
1852 : 0 : return true;
1853 : 0 : }
1854 : :
1855 : 0 : bool QgsStyle::detagSymbol( StyleEntity type, const QString &symbol )
1856 : : {
1857 : 0 : if ( !mCurrentDB )
1858 : : {
1859 : 0 : QgsDebugMsg( QStringLiteral( "Sorry! Cannot open database for detagging." ) );
1860 : 0 : return false;
1861 : : }
1862 : :
1863 : 0 : switch ( type )
1864 : : {
1865 : : case TagEntity:
1866 : : case SmartgroupEntity:
1867 : 0 : return false;
1868 : :
1869 : : default:
1870 : 0 : break;
1871 : : }
1872 : :
1873 : 0 : const int symbolid = entityId( type, symbol );
1874 : 0 : if ( symbolid == 0 )
1875 : : {
1876 : 0 : return false;
1877 : : }
1878 : :
1879 : : // remove all tags
1880 : 0 : const QString query = qgs_sqlite3_mprintf( QStringLiteral( "DELETE FROM %1 WHERE %2=%d" ).arg( tagmapTableName( type ),
1881 : 0 : tagmapEntityIdFieldName( type ) ).toLocal8Bit().data(), symbolid );
1882 : 0 : runEmptyQuery( query );
1883 : :
1884 : 0 : clearCachedTags( type, symbol );
1885 : 0 : emit entityTagsChanged( type, symbol, QStringList() );
1886 : :
1887 : : // TODO Perform tag cleanup
1888 : : // check the number of entries for a given tag in the tagmap
1889 : : // if the count is 0, then remove( TagEntity, tagid )
1890 : 0 : return true;
1891 : 0 : }
1892 : :
1893 : 0 : QStringList QgsStyle::tagsOfSymbol( StyleEntity type, const QString &symbol )
1894 : : {
1895 : 0 : switch ( type )
1896 : : {
1897 : : case TagEntity:
1898 : : case SmartgroupEntity:
1899 : 0 : return QStringList();
1900 : :
1901 : : default:
1902 : 0 : if ( mCachedTags[ type ].contains( symbol ) )
1903 : 0 : return mCachedTags[ type ].value( symbol );
1904 : 0 : break;
1905 : : }
1906 : :
1907 : 0 : if ( !mCurrentDB )
1908 : : {
1909 : 0 : QgsDebugMsg( QStringLiteral( "Sorry! Cannot open database for getting the tags." ) );
1910 : 0 : return QStringList();
1911 : : }
1912 : :
1913 : 0 : int symbolid = entityId( type, symbol );
1914 : 0 : if ( !symbolid )
1915 : 0 : return QStringList();
1916 : :
1917 : : // get the ids of tags for the symbol
1918 : 0 : const QString query = qgs_sqlite3_mprintf( QStringLiteral( "SELECT tag_id FROM %1 WHERE %2=%d" ).arg( tagmapTableName( type ),
1919 : 0 : tagmapEntityIdFieldName( type ) ).toLocal8Bit().data(), symbolid );
1920 : :
1921 : 0 : sqlite3_statement_unique_ptr statement;
1922 : 0 : int nErr; statement = mCurrentDB.prepare( query, nErr );
1923 : :
1924 : 0 : QStringList tagList;
1925 : 0 : while ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
1926 : : {
1927 : 0 : QString subquery = qgs_sqlite3_mprintf( "SELECT name FROM tag WHERE id=%d", sqlite3_column_int( statement.get(), 0 ) );
1928 : :
1929 : 0 : sqlite3_statement_unique_ptr statement2;
1930 : : int pErr;
1931 : 0 : statement2 = mCurrentDB.prepare( subquery, pErr );
1932 : 0 : if ( pErr == SQLITE_OK && sqlite3_step( statement2.get() ) == SQLITE_ROW )
1933 : : {
1934 : 0 : tagList << statement2.columnAsText( 0 );
1935 : 0 : }
1936 : 0 : }
1937 : :
1938 : : // update cache
1939 : 0 : mCachedTags[ type ].insert( symbol, tagList );
1940 : :
1941 : 0 : return tagList;
1942 : 0 : }
1943 : :
1944 : 0 : bool QgsStyle::isFavorite( QgsStyle::StyleEntity type, const QString &name )
1945 : : {
1946 : 0 : if ( !mCurrentDB )
1947 : : {
1948 : 0 : QgsDebugMsg( QStringLiteral( "Sorry! Cannot open database for getting the tags." ) );
1949 : 0 : return false;
1950 : : }
1951 : :
1952 : 0 : switch ( type )
1953 : : {
1954 : : case TagEntity:
1955 : : case SmartgroupEntity:
1956 : 0 : return false;
1957 : :
1958 : : default:
1959 : 0 : if ( mCachedFavorites[ type ].contains( name ) )
1960 : 0 : return mCachedFavorites[ type ].value( name );
1961 : 0 : break;
1962 : : }
1963 : :
1964 : 0 : const QStringList names = allNames( type );
1965 : 0 : if ( !names.contains( name ) )
1966 : 0 : return false; // entity doesn't exist
1967 : :
1968 : : // for efficiency, retrieve names of all favorited symbols and store them in cache
1969 : 0 : const QStringList favorites = symbolsOfFavorite( type );
1970 : 0 : bool res = false;
1971 : 0 : for ( const QString &n : names )
1972 : : {
1973 : 0 : const bool isFav = favorites.contains( n );
1974 : 0 : if ( n == name )
1975 : 0 : res = isFav;
1976 : :
1977 : 0 : mCachedFavorites[ type ].insert( n, isFav );
1978 : : }
1979 : 0 : return res;
1980 : 0 : }
1981 : :
1982 : 0 : bool QgsStyle::symbolHasTag( StyleEntity type, const QString &symbol, const QString &tag )
1983 : : {
1984 : 0 : if ( !mCurrentDB )
1985 : : {
1986 : 0 : QgsDebugMsg( QStringLiteral( "Sorry! Cannot open database for getting the tags." ) );
1987 : 0 : return false;
1988 : : }
1989 : :
1990 : 0 : int symbolid = 0;
1991 : 0 : switch ( type )
1992 : : {
1993 : : case TagEntity:
1994 : : case SmartgroupEntity:
1995 : 0 : return false;
1996 : :
1997 : : default:
1998 : 0 : symbolid = entityId( type, symbol );
1999 : 0 : break;
2000 : : }
2001 : :
2002 : 0 : if ( !symbolid )
2003 : : {
2004 : 0 : return false;
2005 : : }
2006 : 0 : int tagid = tagId( tag );
2007 : 0 : if ( !tagid )
2008 : : {
2009 : 0 : return false;
2010 : : }
2011 : :
2012 : : // get the ids of tags for the symbol
2013 : 0 : const QString query = qgs_sqlite3_mprintf( QStringLiteral( "SELECT tag_id FROM %1 WHERE tag_id=%d AND %2=%d" ).arg( tagmapTableName( type ),
2014 : 0 : tagmapEntityIdFieldName( type ) ).toLocal8Bit().data(), tagid, symbolid );
2015 : :
2016 : 0 : sqlite3_statement_unique_ptr statement;
2017 : 0 : int nErr; statement = mCurrentDB.prepare( query, nErr );
2018 : :
2019 : 0 : return ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW );
2020 : 0 : }
2021 : :
2022 : 0 : QString QgsStyle::tag( int id ) const
2023 : : {
2024 : 0 : if ( !mCurrentDB )
2025 : 0 : return QString();
2026 : :
2027 : 0 : sqlite3_statement_unique_ptr statement;
2028 : :
2029 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT name FROM tag WHERE id=%d", id );
2030 : : int nError;
2031 : 0 : statement = mCurrentDB.prepare( query, nError );
2032 : :
2033 : 0 : QString tag;
2034 : 0 : if ( nError == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
2035 : : {
2036 : 0 : tag = statement.columnAsText( 0 );
2037 : 0 : }
2038 : :
2039 : 0 : return tag;
2040 : 0 : }
2041 : :
2042 : 0 : int QgsStyle::getId( const QString &table, const QString &name )
2043 : : {
2044 : 0 : QString lowerName( name.toLower() );
2045 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT id FROM %q WHERE LOWER(name)='%q'", table.toUtf8().constData(), lowerName.toUtf8().constData() );
2046 : :
2047 : 0 : sqlite3_statement_unique_ptr statement;
2048 : 0 : int nErr; statement = mCurrentDB.prepare( query, nErr );
2049 : :
2050 : 0 : int id = 0;
2051 : 0 : if ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
2052 : : {
2053 : 0 : id = sqlite3_column_int( statement.get(), 0 );
2054 : 0 : }
2055 : : else
2056 : : {
2057 : : // Try the name without lowercase conversion
2058 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT id FROM %q WHERE name='%q'", table.toUtf8().constData(), name.toUtf8().constData() );
2059 : :
2060 : 0 : sqlite3_statement_unique_ptr statement;
2061 : 0 : int nErr; statement = mCurrentDB.prepare( query, nErr );
2062 : 0 : if ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
2063 : : {
2064 : 0 : id = sqlite3_column_int( statement.get(), 0 );
2065 : 0 : }
2066 : 0 : }
2067 : :
2068 : 0 : return id;
2069 : 0 : }
2070 : :
2071 : 0 : QString QgsStyle::getName( const QString &table, int id ) const
2072 : : {
2073 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT name FROM %q WHERE id='%q'", table.toUtf8().constData(), QString::number( id ).toUtf8().constData() );
2074 : :
2075 : 0 : sqlite3_statement_unique_ptr statement;
2076 : 0 : int nErr; statement = mCurrentDB.prepare( query, nErr );
2077 : :
2078 : 0 : QString name;
2079 : 0 : if ( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
2080 : : {
2081 : 0 : name = statement.columnAsText( 0 );
2082 : 0 : }
2083 : :
2084 : 0 : return name;
2085 : 0 : }
2086 : :
2087 : 0 : int QgsStyle::symbolId( const QString &name )
2088 : : {
2089 : 0 : return getId( QStringLiteral( "symbol" ), name );
2090 : 0 : }
2091 : :
2092 : 0 : int QgsStyle::entityId( QgsStyle::StyleEntity type, const QString &name )
2093 : : {
2094 : 0 : return getId( entityTableName( type ), name );
2095 : 0 : }
2096 : :
2097 : 0 : int QgsStyle::colorrampId( const QString &name )
2098 : : {
2099 : 0 : return getId( QStringLiteral( "colorramp" ), name );
2100 : 0 : }
2101 : :
2102 : 0 : QgsTextFormat QgsStyle::textFormat( const QString &name ) const
2103 : : {
2104 : 0 : return mTextFormats.value( name );
2105 : 0 : }
2106 : :
2107 : 0 : int QgsStyle::textFormatCount() const
2108 : : {
2109 : 0 : return mTextFormats.count();
2110 : : }
2111 : :
2112 : 5 : QStringList QgsStyle::textFormatNames() const
2113 : : {
2114 : 5 : return mTextFormats.keys();
2115 : : }
2116 : :
2117 : 0 : int QgsStyle::textFormatId( const QString &name )
2118 : : {
2119 : 0 : return getId( QStringLiteral( "textformat" ), name );
2120 : 0 : }
2121 : :
2122 : 0 : QgsPalLayerSettings QgsStyle::labelSettings( const QString &name ) const
2123 : : {
2124 : 0 : return mLabelSettings.value( name );
2125 : 0 : }
2126 : :
2127 : 0 : QgsLegendPatchShape QgsStyle::legendPatchShape( const QString &name ) const
2128 : : {
2129 : 0 : return mLegendPatchShapes.value( name );
2130 : 0 : }
2131 : :
2132 : 0 : int QgsStyle::legendPatchShapesCount() const
2133 : : {
2134 : 0 : return mLegendPatchShapes.count();
2135 : : }
2136 : :
2137 : 0 : QgsSymbol::SymbolType QgsStyle::legendPatchShapeSymbolType( const QString &name ) const
2138 : : {
2139 : 0 : if ( !mLegendPatchShapes.contains( name ) )
2140 : 0 : return QgsSymbol::Hybrid;
2141 : :
2142 : 0 : return mLegendPatchShapes.value( name ).symbolType();
2143 : 0 : }
2144 : :
2145 : 0 : QgsAbstract3DSymbol *QgsStyle::symbol3D( const QString &name ) const
2146 : : {
2147 : 0 : return m3dSymbols.contains( name ) ? m3dSymbols.value( name )->clone() : nullptr;
2148 : : }
2149 : :
2150 : 0 : int QgsStyle::symbol3DCount() const
2151 : : {
2152 : 0 : return m3dSymbols.count();
2153 : : }
2154 : :
2155 : 0 : QList<QgsWkbTypes::GeometryType> QgsStyle::symbol3DCompatibleGeometryTypes( const QString &name ) const
2156 : : {
2157 : 0 : if ( !m3dSymbols.contains( name ) )
2158 : 0 : return QList<QgsWkbTypes::GeometryType>();
2159 : :
2160 : 0 : return m3dSymbols.value( name )->compatibleGeometryTypes();
2161 : 0 : }
2162 : :
2163 : 0 : QgsWkbTypes::GeometryType QgsStyle::labelSettingsLayerType( const QString &name ) const
2164 : : {
2165 : 0 : if ( !mLabelSettings.contains( name ) )
2166 : 0 : return QgsWkbTypes::UnknownGeometry;
2167 : :
2168 : 0 : return mLabelSettings.value( name ).layerType;
2169 : 0 : }
2170 : :
2171 : 0 : int QgsStyle::labelSettingsCount() const
2172 : : {
2173 : 0 : return mLabelSettings.count();
2174 : : }
2175 : :
2176 : 5 : QStringList QgsStyle::labelSettingsNames() const
2177 : : {
2178 : 5 : return mLabelSettings.keys();
2179 : : }
2180 : :
2181 : 0 : int QgsStyle::labelSettingsId( const QString &name )
2182 : : {
2183 : 0 : return getId( QStringLiteral( "labelsettings" ), name );
2184 : 0 : }
2185 : :
2186 : 5 : QStringList QgsStyle::legendPatchShapeNames() const
2187 : : {
2188 : 5 : return mLegendPatchShapes.keys();
2189 : : }
2190 : :
2191 : 0 : const QgsSymbol *QgsStyle::previewSymbolForPatchShape( const QgsLegendPatchShape &shape ) const
2192 : : {
2193 : 0 : switch ( shape.symbolType() )
2194 : : {
2195 : : case QgsSymbol::Marker:
2196 : 0 : return mPatchMarkerSymbol.get();
2197 : :
2198 : : case QgsSymbol::Line:
2199 : 0 : return mPatchLineSymbol.get();
2200 : :
2201 : : case QgsSymbol::Fill:
2202 : 0 : return mPatchFillSymbol.get();
2203 : :
2204 : : case QgsSymbol::Hybrid:
2205 : 0 : break;
2206 : : }
2207 : 0 : return nullptr;
2208 : 0 : }
2209 : :
2210 : 0 : int QgsStyle::tagId( const QString &name )
2211 : : {
2212 : 0 : return getId( QStringLiteral( "tag" ), name );
2213 : 0 : }
2214 : :
2215 : 0 : int QgsStyle::smartgroupId( const QString &name )
2216 : : {
2217 : 0 : return getId( QStringLiteral( "smartgroup" ), name );
2218 : 0 : }
2219 : :
2220 : 30 : QStringList QgsStyle::allNames( QgsStyle::StyleEntity type ) const
2221 : : {
2222 : 30 : switch ( type )
2223 : : {
2224 : : case SymbolEntity:
2225 : 5 : return symbolNames();
2226 : :
2227 : : case ColorrampEntity:
2228 : 5 : return colorRampNames();
2229 : :
2230 : : case TextFormatEntity:
2231 : 5 : return textFormatNames();
2232 : :
2233 : : case LabelSettingsEntity:
2234 : 5 : return labelSettingsNames();
2235 : :
2236 : : case LegendPatchShapeEntity:
2237 : 5 : return legendPatchShapeNames();
2238 : :
2239 : : case Symbol3DEntity:
2240 : 5 : return symbol3DNames();
2241 : :
2242 : : case TagEntity:
2243 : 0 : return tags();
2244 : :
2245 : : case SmartgroupEntity:
2246 : 0 : return smartgroupNames();
2247 : : }
2248 : 0 : return QStringList();
2249 : 30 : }
2250 : :
2251 : 0 : int QgsStyle::addSmartgroup( const QString &name, const QString &op, const QgsSmartConditionMap &conditions )
2252 : : {
2253 : 0 : return addSmartgroup( name, op, conditions.values( QStringLiteral( "tag" ) ),
2254 : 0 : conditions.values( QStringLiteral( "!tag" ) ),
2255 : 0 : conditions.values( QStringLiteral( "name" ) ),
2256 : 0 : conditions.values( QStringLiteral( "!name" ) ) );
2257 : 0 : }
2258 : :
2259 : 0 : int QgsStyle::addSmartgroup( const QString &name, const QString &op, const QStringList &matchTag, const QStringList &noMatchTag, const QStringList &matchName, const QStringList &noMatchName )
2260 : : {
2261 : 0 : QDomDocument doc( QStringLiteral( "dummy" ) );
2262 : 0 : QDomElement smartEl = doc.createElement( QStringLiteral( "smartgroup" ) );
2263 : 0 : smartEl.setAttribute( QStringLiteral( "name" ), name );
2264 : 0 : smartEl.setAttribute( QStringLiteral( "operator" ), op );
2265 : :
2266 : 0 : auto addCondition = [&doc, &smartEl]( const QString & constraint, const QStringList & parameters )
2267 : : {
2268 : 0 : for ( const QString ¶m : parameters )
2269 : : {
2270 : 0 : QDomElement condEl = doc.createElement( QStringLiteral( "condition" ) );
2271 : 0 : condEl.setAttribute( QStringLiteral( "constraint" ), constraint );
2272 : 0 : condEl.setAttribute( QStringLiteral( "param" ), param );
2273 : 0 : smartEl.appendChild( condEl );
2274 : 0 : }
2275 : 0 : };
2276 : 0 : addCondition( QStringLiteral( "tag" ), matchTag );
2277 : 0 : addCondition( QStringLiteral( "!tag" ), noMatchTag );
2278 : 0 : addCondition( QStringLiteral( "name" ), matchName );
2279 : 0 : addCondition( QStringLiteral( "!name" ), noMatchName );
2280 : :
2281 : 0 : QByteArray xmlArray;
2282 : 0 : QTextStream stream( &xmlArray );
2283 : 0 : stream.setCodec( "UTF-8" );
2284 : 0 : smartEl.save( stream, 4 );
2285 : 0 : QString query = qgs_sqlite3_mprintf( "INSERT INTO smartgroup VALUES (NULL, '%q', '%q')",
2286 : 0 : name.toUtf8().constData(), xmlArray.constData() );
2287 : :
2288 : 0 : if ( runEmptyQuery( query ) )
2289 : : {
2290 : 0 : QgsSettings settings;
2291 : 0 : settings.setValue( QStringLiteral( "qgis/symbolsListGroupsIndex" ), 0 );
2292 : :
2293 : 0 : emit groupsModified();
2294 : 0 : return static_cast< int >( sqlite3_last_insert_rowid( mCurrentDB.get() ) );
2295 : 0 : }
2296 : : else
2297 : : {
2298 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't insert symbol into the database!" ) );
2299 : 0 : return 0;
2300 : : }
2301 : 0 : }
2302 : :
2303 : 0 : QgsSymbolGroupMap QgsStyle::smartgroupsListMap()
2304 : : {
2305 : 0 : if ( !mCurrentDB )
2306 : : {
2307 : 0 : QgsDebugMsg( QStringLiteral( "Cannot open database for listing groups" ) );
2308 : 0 : return QgsSymbolGroupMap();
2309 : : }
2310 : :
2311 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT * FROM smartgroup" );
2312 : :
2313 : : // Now run the query and retrieve the group names
2314 : 0 : sqlite3_statement_unique_ptr statement;
2315 : : int nError;
2316 : 0 : statement = mCurrentDB.prepare( query, nError );
2317 : :
2318 : 0 : QgsSymbolGroupMap groupNames;
2319 : 0 : while ( nError == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
2320 : : {
2321 : 0 : QString group = statement.columnAsText( SmartgroupName );
2322 : 0 : groupNames.insert( sqlite3_column_int( statement.get(), SmartgroupId ), group );
2323 : 0 : }
2324 : :
2325 : 0 : return groupNames;
2326 : 0 : }
2327 : :
2328 : 0 : QStringList QgsStyle::smartgroupNames() const
2329 : : {
2330 : 0 : if ( !mCurrentDB )
2331 : : {
2332 : 0 : QgsDebugMsg( QStringLiteral( "Cannot open database for listing groups" ) );
2333 : 0 : return QStringList();
2334 : : }
2335 : :
2336 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT name FROM smartgroup" );
2337 : :
2338 : : // Now run the query and retrieve the group names
2339 : 0 : sqlite3_statement_unique_ptr statement;
2340 : : int nError;
2341 : 0 : statement = mCurrentDB.prepare( query, nError );
2342 : :
2343 : 0 : QStringList groups;
2344 : 0 : while ( nError == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
2345 : : {
2346 : 0 : groups << statement.columnAsText( 0 );
2347 : : }
2348 : :
2349 : 0 : return groups;
2350 : 0 : }
2351 : :
2352 : 0 : QStringList QgsStyle::symbolsOfSmartgroup( StyleEntity type, int id )
2353 : : {
2354 : 0 : QStringList symbols;
2355 : :
2356 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT xml FROM smartgroup WHERE id=%d", id );
2357 : :
2358 : 0 : sqlite3_statement_unique_ptr statement;
2359 : 0 : int nErr; statement = mCurrentDB.prepare( query, nErr );
2360 : 0 : if ( !( nErr == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW ) )
2361 : : {
2362 : 0 : return QStringList();
2363 : : }
2364 : : else
2365 : : {
2366 : 0 : QDomDocument doc;
2367 : 0 : QString xmlstr = statement.columnAsText( 0 );
2368 : 0 : if ( !doc.setContent( xmlstr ) )
2369 : : {
2370 : 0 : QgsDebugMsg( QStringLiteral( "Cannot open smartgroup id: %1" ).arg( id ) );
2371 : 0 : }
2372 : 0 : QDomElement smartEl = doc.documentElement();
2373 : 0 : QString op = smartEl.attribute( QStringLiteral( "operator" ) );
2374 : 0 : QDomNodeList conditionNodes = smartEl.childNodes();
2375 : :
2376 : 0 : bool firstSet = true;
2377 : 0 : for ( int i = 0; i < conditionNodes.count(); i++ )
2378 : : {
2379 : 0 : QDomElement condEl = conditionNodes.at( i ).toElement();
2380 : 0 : QString constraint = condEl.attribute( QStringLiteral( "constraint" ) );
2381 : 0 : QString param = condEl.attribute( QStringLiteral( "param" ) );
2382 : :
2383 : 0 : QStringList resultNames;
2384 : : // perform suitable action for the given constraint
2385 : 0 : if ( constraint == QLatin1String( "tag" ) )
2386 : : {
2387 : 0 : resultNames = symbolsWithTag( type, tagId( param ) );
2388 : 0 : }
2389 : 0 : else if ( constraint == QLatin1String( "name" ) )
2390 : : {
2391 : 0 : resultNames = allNames( type ).filter( param, Qt::CaseInsensitive );
2392 : 0 : }
2393 : 0 : else if ( constraint == QLatin1String( "!tag" ) )
2394 : : {
2395 : 0 : resultNames = allNames( type );
2396 : 0 : const QStringList unwanted = symbolsWithTag( type, tagId( param ) );
2397 : 0 : for ( const QString &name : unwanted )
2398 : : {
2399 : 0 : resultNames.removeAll( name );
2400 : : }
2401 : 0 : }
2402 : 0 : else if ( constraint == QLatin1String( "!name" ) )
2403 : : {
2404 : 0 : const QStringList all = allNames( type );
2405 : 0 : for ( const QString &str : all )
2406 : : {
2407 : 0 : if ( !str.contains( param, Qt::CaseInsensitive ) )
2408 : 0 : resultNames << str;
2409 : : }
2410 : 0 : }
2411 : :
2412 : : // not apply the operator
2413 : 0 : if ( firstSet )
2414 : : {
2415 : 0 : symbols = resultNames;
2416 : 0 : firstSet = false;
2417 : 0 : }
2418 : : else
2419 : : {
2420 : 0 : if ( op == QLatin1String( "OR" ) )
2421 : : {
2422 : 0 : symbols << resultNames;
2423 : 0 : }
2424 : 0 : else if ( op == QLatin1String( "AND" ) )
2425 : : {
2426 : 0 : QStringList dummy = symbols;
2427 : 0 : symbols.clear();
2428 : 0 : for ( const QString &result : std::as_const( resultNames ) )
2429 : : {
2430 : 0 : if ( dummy.contains( result ) )
2431 : 0 : symbols << result;
2432 : : }
2433 : 0 : }
2434 : : }
2435 : 0 : } // DOM loop ends here
2436 : 0 : }
2437 : :
2438 : : // return sorted, unique list
2439 : 0 : QStringList unique = qgis::setToList( qgis::listToSet( symbols ) );
2440 : 0 : std::sort( unique.begin(), unique.end() );
2441 : 0 : return unique;
2442 : 0 : }
2443 : :
2444 : 0 : QgsSmartConditionMap QgsStyle::smartgroup( int id )
2445 : : {
2446 : 0 : if ( !mCurrentDB )
2447 : : {
2448 : 0 : QgsDebugMsg( QStringLiteral( "Cannot open database for listing groups" ) );
2449 : 0 : return QgsSmartConditionMap();
2450 : : }
2451 : :
2452 : 0 : QgsSmartConditionMap condition;
2453 : :
2454 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT xml FROM smartgroup WHERE id=%d", id );
2455 : :
2456 : 0 : sqlite3_statement_unique_ptr statement;
2457 : : int nError;
2458 : 0 : statement = mCurrentDB.prepare( query, nError );
2459 : 0 : if ( nError == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
2460 : : {
2461 : 0 : QDomDocument doc;
2462 : 0 : QString xmlstr = statement.columnAsText( 0 );
2463 : 0 : if ( !doc.setContent( xmlstr ) )
2464 : : {
2465 : 0 : QgsDebugMsg( QStringLiteral( "Cannot open smartgroup id: %1" ).arg( id ) );
2466 : 0 : }
2467 : :
2468 : 0 : QDomElement smartEl = doc.documentElement();
2469 : 0 : QDomNodeList conditionNodes = smartEl.childNodes();
2470 : :
2471 : 0 : for ( int i = 0; i < conditionNodes.count(); i++ )
2472 : : {
2473 : 0 : QDomElement condEl = conditionNodes.at( i ).toElement();
2474 : 0 : QString constraint = condEl.attribute( QStringLiteral( "constraint" ) );
2475 : 0 : QString param = condEl.attribute( QStringLiteral( "param" ) );
2476 : :
2477 : 0 : condition.insert( constraint, param );
2478 : 0 : }
2479 : 0 : }
2480 : :
2481 : 0 : return condition;
2482 : 0 : }
2483 : :
2484 : 0 : QString QgsStyle::smartgroupOperator( int id )
2485 : : {
2486 : 0 : if ( !mCurrentDB )
2487 : : {
2488 : 0 : QgsDebugMsg( QStringLiteral( "Cannot open database for listing groups" ) );
2489 : 0 : return QString();
2490 : : }
2491 : :
2492 : 0 : QString op;
2493 : :
2494 : 0 : QString query = qgs_sqlite3_mprintf( "SELECT xml FROM smartgroup WHERE id=%d", id );
2495 : :
2496 : : int nError;
2497 : 0 : sqlite3_statement_unique_ptr statement;
2498 : 0 : statement = mCurrentDB.prepare( query, nError );
2499 : 0 : if ( nError == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
2500 : : {
2501 : 0 : QDomDocument doc;
2502 : 0 : QString xmlstr = statement.columnAsText( 0 );
2503 : 0 : if ( !doc.setContent( xmlstr ) )
2504 : : {
2505 : 0 : QgsDebugMsg( QStringLiteral( "Cannot open smartgroup id: %1" ).arg( id ) );
2506 : 0 : }
2507 : 0 : QDomElement smartEl = doc.documentElement();
2508 : 0 : op = smartEl.attribute( QStringLiteral( "operator" ) );
2509 : 0 : }
2510 : :
2511 : 0 : return op;
2512 : 0 : }
2513 : :
2514 : 0 : bool QgsStyle::exportXml( const QString &filename )
2515 : : {
2516 : 0 : if ( filename.isEmpty() )
2517 : : {
2518 : 0 : QgsDebugMsg( QStringLiteral( "Invalid filename for style export." ) );
2519 : 0 : return false;
2520 : : }
2521 : :
2522 : 0 : QDomDocument doc( QStringLiteral( "qgis_style" ) );
2523 : 0 : QDomElement root = doc.createElement( QStringLiteral( "qgis_style" ) );
2524 : 0 : root.setAttribute( QStringLiteral( "version" ), QStringLiteral( STYLE_CURRENT_VERSION ) );
2525 : 0 : doc.appendChild( root );
2526 : :
2527 : 0 : const QStringList favoriteSymbols = symbolsOfFavorite( SymbolEntity );
2528 : 0 : const QStringList favoriteColorramps = symbolsOfFavorite( ColorrampEntity );
2529 : 0 : const QStringList favoriteTextFormats = symbolsOfFavorite( TextFormatEntity );
2530 : 0 : const QStringList favoriteLegendShapes = symbolsOfFavorite( LegendPatchShapeEntity );
2531 : 0 : const QStringList favorite3DSymbols = symbolsOfFavorite( Symbol3DEntity );
2532 : :
2533 : : // save symbols and attach tags
2534 : 0 : QDomElement symbolsElem = QgsSymbolLayerUtils::saveSymbols( mSymbols, QStringLiteral( "symbols" ), doc, QgsReadWriteContext() );
2535 : 0 : QDomNodeList symbolsList = symbolsElem.elementsByTagName( QStringLiteral( "symbol" ) );
2536 : 0 : int nbSymbols = symbolsList.count();
2537 : 0 : for ( int i = 0; i < nbSymbols; ++i )
2538 : : {
2539 : 0 : QDomElement symbol = symbolsList.at( i ).toElement();
2540 : 0 : QString name = symbol.attribute( QStringLiteral( "name" ) );
2541 : 0 : QStringList tags = tagsOfSymbol( SymbolEntity, name );
2542 : 0 : if ( tags.count() > 0 )
2543 : : {
2544 : 0 : symbol.setAttribute( QStringLiteral( "tags" ), tags.join( ',' ) );
2545 : 0 : }
2546 : 0 : if ( favoriteSymbols.contains( name ) )
2547 : : {
2548 : 0 : symbol.setAttribute( QStringLiteral( "favorite" ), QStringLiteral( "1" ) );
2549 : 0 : }
2550 : 0 : }
2551 : :
2552 : : // save color ramps
2553 : 0 : QDomElement rampsElem = doc.createElement( QStringLiteral( "colorramps" ) );
2554 : 0 : for ( QMap<QString, QgsColorRamp *>::const_iterator itr = mColorRamps.constBegin(); itr != mColorRamps.constEnd(); ++itr )
2555 : : {
2556 : 0 : QDomElement rampEl = QgsSymbolLayerUtils::saveColorRamp( itr.key(), itr.value(), doc );
2557 : 0 : QStringList tags = tagsOfSymbol( ColorrampEntity, itr.key() );
2558 : 0 : if ( tags.count() > 0 )
2559 : : {
2560 : 0 : rampEl.setAttribute( QStringLiteral( "tags" ), tags.join( ',' ) );
2561 : 0 : }
2562 : 0 : if ( favoriteColorramps.contains( itr.key() ) )
2563 : : {
2564 : 0 : rampEl.setAttribute( QStringLiteral( "favorite" ), QStringLiteral( "1" ) );
2565 : 0 : }
2566 : 0 : rampsElem.appendChild( rampEl );
2567 : 0 : }
2568 : :
2569 : : // save text formats
2570 : 0 : QDomElement textFormatsElem = doc.createElement( QStringLiteral( "textformats" ) );
2571 : 0 : for ( auto it = mTextFormats.constBegin(); it != mTextFormats.constEnd(); ++it )
2572 : : {
2573 : 0 : QDomElement textFormatEl = doc.createElement( QStringLiteral( "textformat" ) );
2574 : 0 : textFormatEl.setAttribute( QStringLiteral( "name" ), it.key() );
2575 : 0 : QDomElement textStyleEl = it.value().writeXml( doc, QgsReadWriteContext() );
2576 : 0 : textFormatEl.appendChild( textStyleEl );
2577 : 0 : QStringList tags = tagsOfSymbol( TextFormatEntity, it.key() );
2578 : 0 : if ( tags.count() > 0 )
2579 : : {
2580 : 0 : textFormatEl.setAttribute( QStringLiteral( "tags" ), tags.join( ',' ) );
2581 : 0 : }
2582 : 0 : if ( favoriteTextFormats.contains( it.key() ) )
2583 : : {
2584 : 0 : textFormatEl.setAttribute( QStringLiteral( "favorite" ), QStringLiteral( "1" ) );
2585 : 0 : }
2586 : 0 : textFormatsElem.appendChild( textFormatEl );
2587 : 0 : }
2588 : :
2589 : : // save label settings
2590 : 0 : QDomElement labelSettingsElem = doc.createElement( QStringLiteral( "labelsettings" ) );
2591 : 0 : for ( auto it = mLabelSettings.constBegin(); it != mLabelSettings.constEnd(); ++it )
2592 : : {
2593 : 0 : QDomElement labelSettingsEl = doc.createElement( QStringLiteral( "labelsetting" ) );
2594 : 0 : labelSettingsEl.setAttribute( QStringLiteral( "name" ), it.key() );
2595 : 0 : QDomElement defEl = it.value().writeXml( doc, QgsReadWriteContext() );
2596 : 0 : labelSettingsEl.appendChild( defEl );
2597 : 0 : QStringList tags = tagsOfSymbol( LabelSettingsEntity, it.key() );
2598 : 0 : if ( tags.count() > 0 )
2599 : : {
2600 : 0 : labelSettingsEl.setAttribute( QStringLiteral( "tags" ), tags.join( ',' ) );
2601 : 0 : }
2602 : 0 : if ( favoriteTextFormats.contains( it.key() ) )
2603 : : {
2604 : 0 : labelSettingsEl.setAttribute( QStringLiteral( "favorite" ), QStringLiteral( "1" ) );
2605 : 0 : }
2606 : 0 : labelSettingsElem.appendChild( labelSettingsEl );
2607 : 0 : }
2608 : :
2609 : : // save legend patch shapes
2610 : 0 : QDomElement legendPatchShapesElem = doc.createElement( QStringLiteral( "legendpatchshapes" ) );
2611 : 0 : for ( auto it = mLegendPatchShapes.constBegin(); it != mLegendPatchShapes.constEnd(); ++it )
2612 : : {
2613 : 0 : QDomElement legendPatchShapeEl = doc.createElement( QStringLiteral( "legendpatchshape" ) );
2614 : 0 : legendPatchShapeEl.setAttribute( QStringLiteral( "name" ), it.key() );
2615 : 0 : QDomElement defEl = doc.createElement( QStringLiteral( "definition" ) );
2616 : 0 : it.value().writeXml( defEl, doc, QgsReadWriteContext() );
2617 : 0 : legendPatchShapeEl.appendChild( defEl );
2618 : 0 : QStringList tags = tagsOfSymbol( LegendPatchShapeEntity, it.key() );
2619 : 0 : if ( tags.count() > 0 )
2620 : : {
2621 : 0 : legendPatchShapeEl.setAttribute( QStringLiteral( "tags" ), tags.join( ',' ) );
2622 : 0 : }
2623 : 0 : if ( favoriteLegendShapes.contains( it.key() ) )
2624 : : {
2625 : 0 : legendPatchShapeEl.setAttribute( QStringLiteral( "favorite" ), QStringLiteral( "1" ) );
2626 : 0 : }
2627 : 0 : legendPatchShapesElem.appendChild( legendPatchShapeEl );
2628 : 0 : }
2629 : :
2630 : : // save symbols and attach tags
2631 : 0 : QDomElement symbols3DElem = doc.createElement( QStringLiteral( "symbols3d" ) );
2632 : 0 : for ( auto it = m3dSymbols.constBegin(); it != m3dSymbols.constEnd(); ++it )
2633 : : {
2634 : 0 : QDomElement symbolEl = doc.createElement( QStringLiteral( "symbol3d" ) );
2635 : 0 : symbolEl.setAttribute( QStringLiteral( "name" ), it.key() );
2636 : 0 : QDomElement defEl = doc.createElement( QStringLiteral( "definition" ) );
2637 : 0 : defEl.setAttribute( QStringLiteral( "type" ), it.value()->type() );
2638 : 0 : it.value()->writeXml( defEl, QgsReadWriteContext() );
2639 : 0 : symbolEl.appendChild( defEl );
2640 : 0 : QStringList tags = tagsOfSymbol( Symbol3DEntity, it.key() );
2641 : 0 : if ( tags.count() > 0 )
2642 : : {
2643 : 0 : symbolEl.setAttribute( QStringLiteral( "tags" ), tags.join( ',' ) );
2644 : 0 : }
2645 : 0 : if ( favorite3DSymbols.contains( it.key() ) )
2646 : : {
2647 : 0 : symbolEl.setAttribute( QStringLiteral( "favorite" ), QStringLiteral( "1" ) );
2648 : 0 : }
2649 : 0 : symbols3DElem.appendChild( symbolEl );
2650 : 0 : }
2651 : :
2652 : 0 : root.appendChild( symbolsElem );
2653 : 0 : root.appendChild( rampsElem );
2654 : 0 : root.appendChild( textFormatsElem );
2655 : 0 : root.appendChild( labelSettingsElem );
2656 : 0 : root.appendChild( legendPatchShapesElem );
2657 : 0 : root.appendChild( symbols3DElem );
2658 : :
2659 : : // save
2660 : 0 : QFile f( filename );
2661 : 0 : if ( !f.open( QFile::WriteOnly | QIODevice::Truncate ) )
2662 : : {
2663 : 0 : mErrorString = "Couldn't open file for writing: " + filename;
2664 : 0 : return false;
2665 : : }
2666 : :
2667 : 0 : QTextStream ts( &f );
2668 : 0 : ts.setCodec( "UTF-8" );
2669 : 0 : doc.save( ts, 2 );
2670 : 0 : f.close();
2671 : :
2672 : 0 : mFileName = filename;
2673 : 0 : return true;
2674 : 0 : }
2675 : :
2676 : 0 : bool QgsStyle::importXml( const QString &filename )
2677 : : {
2678 : 0 : return importXml( filename, -1 );
2679 : : }
2680 : :
2681 : 1 : bool QgsStyle::importXml( const QString &filename, int sinceVersion )
2682 : : {
2683 : 1 : mErrorString = QString();
2684 : 2 : QDomDocument doc( QStringLiteral( "style" ) );
2685 : 1 : QFile f( filename );
2686 : 1 : if ( !f.open( QFile::ReadOnly ) )
2687 : : {
2688 : 2 : mErrorString = QStringLiteral( "Unable to open the specified file" );
2689 : 1 : QgsDebugMsg( QStringLiteral( "Error opening the style XML file." ) );
2690 : 1 : return false;
2691 : : }
2692 : :
2693 : 0 : if ( !doc.setContent( &f ) )
2694 : : {
2695 : 0 : mErrorString = QStringLiteral( "Unable to understand the style file: %1" ).arg( filename );
2696 : 0 : QgsDebugMsg( QStringLiteral( "XML Parsing error" ) );
2697 : 0 : f.close();
2698 : 0 : return false;
2699 : : }
2700 : 0 : f.close();
2701 : :
2702 : 0 : QDomElement docEl = doc.documentElement();
2703 : 0 : if ( docEl.tagName() != QLatin1String( "qgis_style" ) )
2704 : : {
2705 : 0 : mErrorString = "Incorrect root tag in style: " + docEl.tagName();
2706 : 0 : return false;
2707 : : }
2708 : :
2709 : 0 : const QString version = docEl.attribute( QStringLiteral( "version" ) );
2710 : 0 : if ( version != QLatin1String( STYLE_CURRENT_VERSION ) && version != QLatin1String( "0" ) && version != QLatin1String( "1" ) )
2711 : : {
2712 : 0 : mErrorString = "Unknown style file version: " + version;
2713 : 0 : return false;
2714 : : }
2715 : :
2716 : 0 : QgsSymbolMap symbols;
2717 : :
2718 : 0 : QDomElement symbolsElement = docEl.firstChildElement( QStringLiteral( "symbols" ) );
2719 : 0 : QDomElement e = symbolsElement.firstChildElement();
2720 : :
2721 : : // gain speed by re-grouping the INSERT statements in a transaction
2722 : 0 : QString query = qgs_sqlite3_mprintf( "BEGIN TRANSACTION;" );
2723 : 0 : runEmptyQuery( query );
2724 : :
2725 : 0 : if ( version == QLatin1String( STYLE_CURRENT_VERSION ) || version == QLatin1String( "1" ) )
2726 : : {
2727 : : // For the new style, load symbols individually
2728 : 0 : while ( !e.isNull() )
2729 : : {
2730 : 0 : const int entityAddedVersion = e.attribute( QStringLiteral( "addedVersion" ) ).toInt();
2731 : 0 : if ( entityAddedVersion != 0 && sinceVersion != -1 && entityAddedVersion <= sinceVersion )
2732 : : {
2733 : : // skip the symbol, should already be present
2734 : 0 : continue;
2735 : : }
2736 : :
2737 : 0 : if ( e.tagName() == QLatin1String( "symbol" ) )
2738 : : {
2739 : 0 : QString name = e.attribute( QStringLiteral( "name" ) );
2740 : 0 : QStringList tags;
2741 : 0 : if ( e.hasAttribute( QStringLiteral( "tags" ) ) )
2742 : : {
2743 : 0 : tags = e.attribute( QStringLiteral( "tags" ) ).split( ',' );
2744 : 0 : }
2745 : 0 : bool favorite = false;
2746 : 0 : if ( e.hasAttribute( QStringLiteral( "favorite" ) ) && e.attribute( QStringLiteral( "favorite" ) ) == QLatin1String( "1" ) )
2747 : : {
2748 : 0 : favorite = true;
2749 : 0 : }
2750 : :
2751 : 0 : QgsSymbol *symbol = QgsSymbolLayerUtils::loadSymbol( e, QgsReadWriteContext() );
2752 : 0 : if ( symbol )
2753 : : {
2754 : 0 : addSymbol( name, symbol );
2755 : 0 : if ( mCurrentDB )
2756 : : {
2757 : 0 : saveSymbol( name, symbol, favorite, tags );
2758 : 0 : }
2759 : 0 : }
2760 : 0 : }
2761 : : else
2762 : : {
2763 : 0 : QgsDebugMsg( "unknown tag: " + e.tagName() );
2764 : : }
2765 : 0 : e = e.nextSiblingElement();
2766 : : }
2767 : 0 : }
2768 : : else
2769 : : {
2770 : : // for the old version, use the utility function to solve @symbol@layer subsymbols
2771 : 0 : symbols = QgsSymbolLayerUtils::loadSymbols( symbolsElement, QgsReadWriteContext() );
2772 : :
2773 : : // save the symbols with proper name
2774 : 0 : for ( QMap<QString, QgsSymbol *>::iterator it = symbols.begin(); it != symbols.end(); ++it )
2775 : : {
2776 : 0 : addSymbol( it.key(), it.value() );
2777 : 0 : }
2778 : : }
2779 : :
2780 : : // load color ramps
2781 : 0 : QDomElement rampsElement = docEl.firstChildElement( QStringLiteral( "colorramps" ) );
2782 : 0 : e = rampsElement.firstChildElement();
2783 : 0 : while ( !e.isNull() )
2784 : : {
2785 : 0 : const int entityAddedVersion = e.attribute( QStringLiteral( "addedVersion" ) ).toInt();
2786 : 0 : if ( entityAddedVersion != 0 && sinceVersion != -1 && entityAddedVersion <= sinceVersion )
2787 : : {
2788 : : // skip the ramp, should already be present
2789 : 0 : continue;
2790 : : }
2791 : :
2792 : 0 : if ( e.tagName() == QLatin1String( "colorramp" ) )
2793 : : {
2794 : 0 : QString name = e.attribute( QStringLiteral( "name" ) );
2795 : 0 : QStringList tags;
2796 : 0 : if ( e.hasAttribute( QStringLiteral( "tags" ) ) )
2797 : : {
2798 : 0 : tags = e.attribute( QStringLiteral( "tags" ) ).split( ',' );
2799 : 0 : }
2800 : 0 : bool favorite = false;
2801 : 0 : if ( e.hasAttribute( QStringLiteral( "favorite" ) ) && e.attribute( QStringLiteral( "favorite" ) ) == QLatin1String( "1" ) )
2802 : : {
2803 : 0 : favorite = true;
2804 : 0 : }
2805 : :
2806 : 0 : QgsColorRamp *ramp = QgsSymbolLayerUtils::loadColorRamp( e );
2807 : 0 : if ( ramp )
2808 : : {
2809 : 0 : addColorRamp( name, ramp );
2810 : 0 : if ( mCurrentDB )
2811 : : {
2812 : 0 : saveColorRamp( name, ramp, favorite, tags );
2813 : 0 : }
2814 : 0 : }
2815 : 0 : }
2816 : : else
2817 : : {
2818 : 0 : QgsDebugMsg( "unknown tag: " + e.tagName() );
2819 : : }
2820 : 0 : e = e.nextSiblingElement();
2821 : : }
2822 : :
2823 : : // load text formats
2824 : 0 : if ( version == STYLE_CURRENT_VERSION )
2825 : : {
2826 : 0 : const QDomElement textFormatElement = docEl.firstChildElement( QStringLiteral( "textformats" ) );
2827 : 0 : e = textFormatElement.firstChildElement();
2828 : 0 : while ( !e.isNull() )
2829 : : {
2830 : 0 : const int entityAddedVersion = e.attribute( QStringLiteral( "addedVersion" ) ).toInt();
2831 : 0 : if ( entityAddedVersion != 0 && sinceVersion != -1 && entityAddedVersion <= sinceVersion )
2832 : : {
2833 : : // skip the format, should already be present
2834 : 0 : continue;
2835 : : }
2836 : :
2837 : 0 : if ( e.tagName() == QLatin1String( "textformat" ) )
2838 : : {
2839 : 0 : QString name = e.attribute( QStringLiteral( "name" ) );
2840 : 0 : QStringList tags;
2841 : 0 : if ( e.hasAttribute( QStringLiteral( "tags" ) ) )
2842 : : {
2843 : 0 : tags = e.attribute( QStringLiteral( "tags" ) ).split( ',' );
2844 : 0 : }
2845 : 0 : bool favorite = false;
2846 : 0 : if ( e.hasAttribute( QStringLiteral( "favorite" ) ) && e.attribute( QStringLiteral( "favorite" ) ) == QLatin1String( "1" ) )
2847 : : {
2848 : 0 : favorite = true;
2849 : 0 : }
2850 : :
2851 : 0 : QgsTextFormat format;
2852 : 0 : const QDomElement styleElem = e.firstChildElement();
2853 : 0 : format.readXml( styleElem, QgsReadWriteContext() );
2854 : 0 : addTextFormat( name, format );
2855 : 0 : if ( mCurrentDB )
2856 : : {
2857 : 0 : saveTextFormat( name, format, favorite, tags );
2858 : 0 : }
2859 : 0 : }
2860 : : else
2861 : : {
2862 : 0 : QgsDebugMsg( "unknown tag: " + e.tagName() );
2863 : : }
2864 : 0 : e = e.nextSiblingElement();
2865 : : }
2866 : 0 : }
2867 : :
2868 : : // load label settings
2869 : 0 : if ( version == STYLE_CURRENT_VERSION )
2870 : : {
2871 : 0 : const QDomElement labelSettingsElement = docEl.firstChildElement( QStringLiteral( "labelsettings" ) );
2872 : 0 : e = labelSettingsElement.firstChildElement();
2873 : 0 : while ( !e.isNull() )
2874 : : {
2875 : 0 : const int entityAddedVersion = e.attribute( QStringLiteral( "addedVersion" ) ).toInt();
2876 : 0 : if ( entityAddedVersion != 0 && sinceVersion != -1 && entityAddedVersion <= sinceVersion )
2877 : : {
2878 : : // skip the settings, should already be present
2879 : 0 : continue;
2880 : : }
2881 : :
2882 : 0 : if ( e.tagName() == QLatin1String( "labelsetting" ) )
2883 : : {
2884 : 0 : QString name = e.attribute( QStringLiteral( "name" ) );
2885 : 0 : QStringList tags;
2886 : 0 : if ( e.hasAttribute( QStringLiteral( "tags" ) ) )
2887 : : {
2888 : 0 : tags = e.attribute( QStringLiteral( "tags" ) ).split( ',' );
2889 : 0 : }
2890 : 0 : bool favorite = false;
2891 : 0 : if ( e.hasAttribute( QStringLiteral( "favorite" ) ) && e.attribute( QStringLiteral( "favorite" ) ) == QLatin1String( "1" ) )
2892 : : {
2893 : 0 : favorite = true;
2894 : 0 : }
2895 : :
2896 : 0 : QgsPalLayerSettings settings;
2897 : 0 : const QDomElement styleElem = e.firstChildElement();
2898 : 0 : settings.readXml( styleElem, QgsReadWriteContext() );
2899 : 0 : addLabelSettings( name, settings );
2900 : 0 : if ( mCurrentDB )
2901 : : {
2902 : 0 : saveLabelSettings( name, settings, favorite, tags );
2903 : 0 : }
2904 : 0 : }
2905 : : else
2906 : : {
2907 : 0 : QgsDebugMsg( "unknown tag: " + e.tagName() );
2908 : : }
2909 : 0 : e = e.nextSiblingElement();
2910 : : }
2911 : 0 : }
2912 : :
2913 : : // load legend patch shapes
2914 : 0 : if ( version == STYLE_CURRENT_VERSION )
2915 : : {
2916 : 0 : const QDomElement legendPatchShapesElement = docEl.firstChildElement( QStringLiteral( "legendpatchshapes" ) );
2917 : 0 : e = legendPatchShapesElement.firstChildElement();
2918 : 0 : while ( !e.isNull() )
2919 : : {
2920 : 0 : const int entityAddedVersion = e.attribute( QStringLiteral( "addedVersion" ) ).toInt();
2921 : 0 : if ( entityAddedVersion != 0 && sinceVersion != -1 && entityAddedVersion <= sinceVersion )
2922 : : {
2923 : : // skip the shape, should already be present
2924 : 0 : continue;
2925 : : }
2926 : :
2927 : 0 : if ( e.tagName() == QLatin1String( "legendpatchshape" ) )
2928 : : {
2929 : 0 : QString name = e.attribute( QStringLiteral( "name" ) );
2930 : 0 : QStringList tags;
2931 : 0 : if ( e.hasAttribute( QStringLiteral( "tags" ) ) )
2932 : : {
2933 : 0 : tags = e.attribute( QStringLiteral( "tags" ) ).split( ',' );
2934 : 0 : }
2935 : 0 : bool favorite = false;
2936 : 0 : if ( e.hasAttribute( QStringLiteral( "favorite" ) ) && e.attribute( QStringLiteral( "favorite" ) ) == QLatin1String( "1" ) )
2937 : : {
2938 : 0 : favorite = true;
2939 : 0 : }
2940 : :
2941 : 0 : QgsLegendPatchShape shape;
2942 : 0 : const QDomElement shapeElem = e.firstChildElement();
2943 : 0 : shape.readXml( shapeElem, QgsReadWriteContext() );
2944 : 0 : addLegendPatchShape( name, shape );
2945 : 0 : if ( mCurrentDB )
2946 : : {
2947 : 0 : saveLegendPatchShape( name, shape, favorite, tags );
2948 : 0 : }
2949 : 0 : }
2950 : : else
2951 : : {
2952 : 0 : QgsDebugMsg( "unknown tag: " + e.tagName() );
2953 : : }
2954 : 0 : e = e.nextSiblingElement();
2955 : : }
2956 : 0 : }
2957 : :
2958 : : // load 3d symbols
2959 : 0 : if ( version == STYLE_CURRENT_VERSION )
2960 : : {
2961 : 0 : const QDomElement symbols3DElement = docEl.firstChildElement( QStringLiteral( "symbols3d" ) );
2962 : 0 : e = symbols3DElement.firstChildElement();
2963 : 0 : while ( !e.isNull() )
2964 : : {
2965 : 0 : const int entityAddedVersion = e.attribute( QStringLiteral( "addedVersion" ) ).toInt();
2966 : 0 : if ( entityAddedVersion != 0 && sinceVersion != -1 && entityAddedVersion <= sinceVersion )
2967 : : {
2968 : : // skip the symbol, should already be present
2969 : 0 : continue;
2970 : : }
2971 : :
2972 : 0 : if ( e.tagName() == QLatin1String( "symbol3d" ) )
2973 : : {
2974 : 0 : QString name = e.attribute( QStringLiteral( "name" ) );
2975 : 0 : QStringList tags;
2976 : 0 : if ( e.hasAttribute( QStringLiteral( "tags" ) ) )
2977 : : {
2978 : 0 : tags = e.attribute( QStringLiteral( "tags" ) ).split( ',' );
2979 : 0 : }
2980 : 0 : bool favorite = false;
2981 : 0 : if ( e.hasAttribute( QStringLiteral( "favorite" ) ) && e.attribute( QStringLiteral( "favorite" ) ) == QLatin1String( "1" ) )
2982 : : {
2983 : 0 : favorite = true;
2984 : 0 : }
2985 : :
2986 : 0 : const QDomElement symbolElem = e.firstChildElement();
2987 : 0 : const QString type = symbolElem.attribute( QStringLiteral( "type" ) );
2988 : 0 : std::unique_ptr< QgsAbstract3DSymbol > sym( QgsApplication::symbol3DRegistry()->createSymbol( type ) );
2989 : 0 : if ( sym )
2990 : : {
2991 : 0 : sym->readXml( symbolElem, QgsReadWriteContext() );
2992 : 0 : QgsAbstract3DSymbol *newSym = sym.get();
2993 : 0 : addSymbol3D( name, sym.release() );
2994 : 0 : if ( mCurrentDB )
2995 : : {
2996 : 0 : saveSymbol3D( name, newSym, favorite, tags );
2997 : 0 : }
2998 : 0 : }
2999 : 0 : }
3000 : : else
3001 : : {
3002 : 0 : QgsDebugMsg( "unknown tag: " + e.tagName() );
3003 : : }
3004 : 0 : e = e.nextSiblingElement();
3005 : : }
3006 : 0 : }
3007 : :
3008 : 0 : query = qgs_sqlite3_mprintf( "COMMIT TRANSACTION;" );
3009 : 0 : runEmptyQuery( query );
3010 : :
3011 : 0 : mFileName = filename;
3012 : 0 : return true;
3013 : 1 : }
3014 : :
3015 : 0 : bool QgsStyle::isXmlStyleFile( const QString &path )
3016 : : {
3017 : 0 : QFileInfo fileInfo( path );
3018 : :
3019 : 0 : if ( fileInfo.suffix().compare( QLatin1String( "xml" ), Qt::CaseInsensitive ) != 0 )
3020 : 0 : return false;
3021 : :
3022 : : // sniff the first line of the file to see if it's a style file
3023 : 0 : if ( !QFile::exists( path ) )
3024 : 0 : return false;
3025 : :
3026 : 0 : QFile inputFile( path );
3027 : 0 : if ( !inputFile.open( QIODevice::ReadOnly ) )
3028 : 0 : return false;
3029 : :
3030 : 0 : QTextStream stream( &inputFile );
3031 : 0 : const QString line = stream.readLine();
3032 : 0 : return line == QLatin1String( "<!DOCTYPE qgis_style>" );
3033 : 0 : }
3034 : :
3035 : 0 : bool QgsStyle::updateSymbol( StyleEntity type, const QString &name )
3036 : : {
3037 : 0 : QDomDocument doc( QStringLiteral( "dummy" ) );
3038 : 0 : QDomElement symEl;
3039 : 0 : QByteArray xmlArray;
3040 : 0 : QTextStream stream( &xmlArray );
3041 : 0 : stream.setCodec( "UTF-8" );
3042 : :
3043 : 0 : QString query;
3044 : :
3045 : 0 : switch ( type )
3046 : : {
3047 : : case SymbolEntity:
3048 : : {
3049 : : // check if it is an existing symbol
3050 : 0 : if ( !symbolNames().contains( name ) )
3051 : : {
3052 : 0 : QgsDebugMsg( QStringLiteral( "Update request received for unavailable symbol" ) );
3053 : 0 : return false;
3054 : : }
3055 : :
3056 : 0 : symEl = QgsSymbolLayerUtils::saveSymbol( name, symbol( name ), doc, QgsReadWriteContext() );
3057 : 0 : if ( symEl.isNull() )
3058 : : {
3059 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't convert symbol to valid XML!" ) );
3060 : 0 : return false;
3061 : : }
3062 : 0 : symEl.save( stream, 4 );
3063 : 0 : query = qgs_sqlite3_mprintf( "UPDATE symbol SET xml='%q' WHERE name='%q';",
3064 : 0 : xmlArray.constData(), name.toUtf8().constData() );
3065 : 0 : break;
3066 : : }
3067 : :
3068 : : case Symbol3DEntity:
3069 : : {
3070 : : // check if it is an existing symbol
3071 : 0 : if ( !symbol3DNames().contains( name ) )
3072 : : {
3073 : 0 : QgsDebugMsg( QStringLiteral( "Update request received for unavailable symbol" ) );
3074 : 0 : return false;
3075 : : }
3076 : :
3077 : 0 : symEl = doc.createElement( QStringLiteral( "symbol" ) );
3078 : 0 : symEl.setAttribute( QStringLiteral( "type" ), m3dSymbols.value( name )->type() );
3079 : 0 : m3dSymbols.value( name )->writeXml( symEl, QgsReadWriteContext() );
3080 : 0 : if ( symEl.isNull() )
3081 : : {
3082 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't convert symbol to valid XML!" ) );
3083 : 0 : return false;
3084 : : }
3085 : 0 : symEl.save( stream, 4 );
3086 : 0 : query = qgs_sqlite3_mprintf( "UPDATE symbol3d SET xml='%q' WHERE name='%q';",
3087 : 0 : xmlArray.constData(), name.toUtf8().constData() );
3088 : 0 : break;
3089 : : }
3090 : :
3091 : : case ColorrampEntity:
3092 : : {
3093 : 0 : if ( !colorRampNames().contains( name ) )
3094 : : {
3095 : 0 : QgsDebugMsg( QStringLiteral( "Update requested for unavailable color ramp." ) );
3096 : 0 : return false;
3097 : : }
3098 : :
3099 : 0 : std::unique_ptr< QgsColorRamp > ramp( colorRamp( name ) );
3100 : 0 : symEl = QgsSymbolLayerUtils::saveColorRamp( name, ramp.get(), doc );
3101 : 0 : if ( symEl.isNull() )
3102 : : {
3103 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't convert color ramp to valid XML!" ) );
3104 : 0 : return false;
3105 : : }
3106 : 0 : symEl.save( stream, 4 );
3107 : 0 : query = qgs_sqlite3_mprintf( "UPDATE colorramp SET xml='%q' WHERE name='%q';",
3108 : 0 : xmlArray.constData(), name.toUtf8().constData() );
3109 : 0 : break;
3110 : 0 : }
3111 : :
3112 : : case TextFormatEntity:
3113 : : {
3114 : 0 : if ( !textFormatNames().contains( name ) )
3115 : : {
3116 : 0 : QgsDebugMsg( QStringLiteral( "Update requested for unavailable text format." ) );
3117 : 0 : return false;
3118 : : }
3119 : :
3120 : 0 : QgsTextFormat format( textFormat( name ) );
3121 : 0 : symEl = format.writeXml( doc, QgsReadWriteContext() );
3122 : 0 : if ( symEl.isNull() )
3123 : : {
3124 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't convert text format to valid XML!" ) );
3125 : 0 : return false;
3126 : : }
3127 : 0 : symEl.save( stream, 4 );
3128 : 0 : query = qgs_sqlite3_mprintf( "UPDATE textformat SET xml='%q' WHERE name='%q';",
3129 : 0 : xmlArray.constData(), name.toUtf8().constData() );
3130 : 0 : break;
3131 : 0 : }
3132 : :
3133 : : case LabelSettingsEntity:
3134 : : {
3135 : 0 : if ( !labelSettingsNames().contains( name ) )
3136 : : {
3137 : 0 : QgsDebugMsg( QStringLiteral( "Update requested for unavailable label settings." ) );
3138 : 0 : return false;
3139 : : }
3140 : :
3141 : 0 : QgsPalLayerSettings settings( labelSettings( name ) );
3142 : 0 : symEl = settings.writeXml( doc, QgsReadWriteContext() );
3143 : 0 : if ( symEl.isNull() )
3144 : : {
3145 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't convert label settings to valid XML!" ) );
3146 : 0 : return false;
3147 : : }
3148 : 0 : symEl.save( stream, 4 );
3149 : 0 : query = qgs_sqlite3_mprintf( "UPDATE labelsettings SET xml='%q' WHERE name='%q';",
3150 : 0 : xmlArray.constData(), name.toUtf8().constData() );
3151 : 0 : break;
3152 : 0 : }
3153 : :
3154 : : case LegendPatchShapeEntity:
3155 : : {
3156 : 0 : if ( !legendPatchShapeNames().contains( name ) )
3157 : : {
3158 : 0 : QgsDebugMsg( QStringLiteral( "Update requested for unavailable legend patch shape." ) );
3159 : 0 : return false;
3160 : : }
3161 : :
3162 : 0 : QgsLegendPatchShape shape( legendPatchShape( name ) );
3163 : 0 : symEl = doc.createElement( QStringLiteral( "shape" ) );
3164 : 0 : shape.writeXml( symEl, doc, QgsReadWriteContext() );
3165 : 0 : symEl.save( stream, 4 );
3166 : 0 : query = qgs_sqlite3_mprintf( "UPDATE legendpatchshapes SET xml='%q' WHERE name='%q';",
3167 : 0 : xmlArray.constData(), name.toUtf8().constData() );
3168 : : break;
3169 : 0 : }
3170 : :
3171 : : case TagEntity:
3172 : : case SmartgroupEntity:
3173 : : {
3174 : 0 : QgsDebugMsg( QStringLiteral( "Updating the unsupported StyleEntity" ) );
3175 : 0 : return false;
3176 : : }
3177 : : }
3178 : :
3179 : :
3180 : 0 : if ( !runEmptyQuery( query ) )
3181 : : {
3182 : 0 : QgsDebugMsg( QStringLiteral( "Couldn't insert symbol into the database!" ) );
3183 : 0 : return false;
3184 : : }
3185 : : else
3186 : : {
3187 : 0 : switch ( type )
3188 : : {
3189 : : case SymbolEntity:
3190 : 0 : emit symbolChanged( name );
3191 : 0 : break;
3192 : :
3193 : : case ColorrampEntity:
3194 : 0 : emit rampChanged( name );
3195 : 0 : break;
3196 : :
3197 : : case TextFormatEntity:
3198 : 0 : emit textFormatChanged( name );
3199 : 0 : break;
3200 : :
3201 : : case LabelSettingsEntity:
3202 : 0 : emit labelSettingsChanged( name );
3203 : 0 : break;
3204 : :
3205 : : case LegendPatchShapeEntity:
3206 : : case TagEntity:
3207 : : case SmartgroupEntity:
3208 : : case Symbol3DEntity:
3209 : 0 : break;
3210 : : }
3211 : 0 : emit entityChanged( type, name );
3212 : : }
3213 : 0 : return true;
3214 : 0 : }
3215 : :
3216 : 0 : void QgsStyle::clearCachedTags( QgsStyle::StyleEntity type, const QString &name )
3217 : : {
3218 : 0 : mCachedTags[ type ].remove( name );
3219 : 0 : }
3220 : :
3221 : 5 : void QgsStyle::upgradeIfRequired()
3222 : : {
3223 : : // make sure metadata table exists
3224 : 5 : QString query = qgs_sqlite3_mprintf( "SELECT name FROM sqlite_master WHERE name='stylemetadata'" );
3225 : 5 : sqlite3_statement_unique_ptr statement;
3226 : : int rc;
3227 : 5 : int dbVersion = 0;
3228 : 5 : statement = mCurrentDB.prepare( query, rc );
3229 : :
3230 : 5 : if ( rc != SQLITE_OK || sqlite3_step( statement.get() ) != SQLITE_ROW )
3231 : : {
3232 : : // no metadata table
3233 : 0 : query = qgs_sqlite3_mprintf( "CREATE TABLE stylemetadata("\
3234 : : "id INTEGER PRIMARY KEY,"\
3235 : : "key TEXT UNIQUE,"\
3236 : : "value TEXT);" );
3237 : 0 : runEmptyQuery( query );
3238 : 0 : query = qgs_sqlite3_mprintf( "INSERT INTO stylemetadata VALUES (NULL, '%q', '%q')", "version", "31200" );
3239 : 0 : runEmptyQuery( query );
3240 : :
3241 : 0 : dbVersion = 31200;
3242 : 0 : }
3243 : : else
3244 : : {
3245 : 5 : query = qgs_sqlite3_mprintf( "SELECT value FROM stylemetadata WHERE key='version'" );
3246 : 5 : statement = mCurrentDB.prepare( query, rc );
3247 : 5 : if ( rc == SQLITE_OK && sqlite3_step( statement.get() ) == SQLITE_ROW )
3248 : : {
3249 : 5 : dbVersion = statement.columnAsText( 0 ).toInt();
3250 : 5 : }
3251 : : }
3252 : :
3253 : 5 : if ( dbVersion < Qgis::versionInt() )
3254 : : {
3255 : : // do upgrade
3256 : 1 : if ( importXml( QgsApplication::defaultStylePath(), dbVersion ) )
3257 : : {
3258 : 0 : query = qgs_sqlite3_mprintf( "UPDATE stylemetadata SET value='%q' WHERE key='version'", QString::number( Qgis::versionInt() ).toUtf8().constData() );
3259 : 0 : runEmptyQuery( query );
3260 : 0 : }
3261 : 1 : }
3262 : 5 : }
3263 : :
3264 : 0 : QString QgsStyle::entityTableName( QgsStyle::StyleEntity type )
3265 : : {
3266 : 0 : switch ( type )
3267 : : {
3268 : : case SymbolEntity:
3269 : 0 : return QStringLiteral( "symbol" );
3270 : :
3271 : : case ColorrampEntity:
3272 : 0 : return QStringLiteral( "colorramp" );
3273 : :
3274 : : case TextFormatEntity:
3275 : 0 : return QStringLiteral( "textformat" );
3276 : :
3277 : : case LabelSettingsEntity:
3278 : 0 : return QStringLiteral( "labelsettings" );
3279 : :
3280 : : case LegendPatchShapeEntity:
3281 : 0 : return QStringLiteral( "legendpatchshapes" );
3282 : :
3283 : : case Symbol3DEntity:
3284 : 0 : return QStringLiteral( "symbol3d" );
3285 : :
3286 : : case TagEntity:
3287 : 0 : return QStringLiteral( "tag" );
3288 : :
3289 : : case SmartgroupEntity:
3290 : 0 : return QStringLiteral( "smartgroup" );
3291 : : }
3292 : 0 : return QString();
3293 : 0 : }
3294 : :
3295 : 0 : QString QgsStyle::tagmapTableName( QgsStyle::StyleEntity type )
3296 : : {
3297 : 0 : switch ( type )
3298 : : {
3299 : : case SymbolEntity:
3300 : 0 : return QStringLiteral( "tagmap" );
3301 : :
3302 : : case ColorrampEntity:
3303 : 0 : return QStringLiteral( "ctagmap" );
3304 : :
3305 : : case TextFormatEntity:
3306 : 0 : return QStringLiteral( "tftagmap" );
3307 : :
3308 : : case LabelSettingsEntity:
3309 : 0 : return QStringLiteral( "lstagmap" );
3310 : :
3311 : : case LegendPatchShapeEntity:
3312 : 0 : return QStringLiteral( "lpstagmap" );
3313 : :
3314 : : case Symbol3DEntity:
3315 : 0 : return QStringLiteral( "symbol3dtagmap" );
3316 : :
3317 : : case TagEntity:
3318 : : case SmartgroupEntity:
3319 : 0 : break;
3320 : : }
3321 : 0 : return QString();
3322 : 0 : }
3323 : :
3324 : 0 : QString QgsStyle::tagmapEntityIdFieldName( QgsStyle::StyleEntity type )
3325 : : {
3326 : 0 : switch ( type )
3327 : : {
3328 : : case SymbolEntity:
3329 : 0 : return QStringLiteral( "symbol_id" );
3330 : :
3331 : : case ColorrampEntity:
3332 : 0 : return QStringLiteral( "colorramp_id" );
3333 : :
3334 : : case TextFormatEntity:
3335 : 0 : return QStringLiteral( "textformat_id" );
3336 : :
3337 : : case LabelSettingsEntity:
3338 : 0 : return QStringLiteral( "labelsettings_id" );
3339 : :
3340 : : case LegendPatchShapeEntity:
3341 : 0 : return QStringLiteral( "legendpatchshape_id" );
3342 : :
3343 : : case Symbol3DEntity:
3344 : 0 : return QStringLiteral( "symbol3d_id" );
3345 : :
3346 : : case TagEntity:
3347 : : case SmartgroupEntity:
3348 : 0 : break;
3349 : : }
3350 : 0 : return QString();
3351 : 0 : }
3352 : :
3353 : 0 : QgsStyle::StyleEntity QgsStyleSymbolEntity::type() const
3354 : : {
3355 : 0 : return QgsStyle::SymbolEntity;
3356 : : }
3357 : :
3358 : 0 : QgsStyle::StyleEntity QgsStyleColorRampEntity::type() const
3359 : : {
3360 : 0 : return QgsStyle::ColorrampEntity;
3361 : : }
3362 : :
3363 : 0 : QgsStyle::StyleEntity QgsStyleTextFormatEntity::type() const
3364 : : {
3365 : 0 : return QgsStyle::TextFormatEntity;
3366 : : }
3367 : :
3368 : 0 : QgsStyle::StyleEntity QgsStyleLabelSettingsEntity::type() const
3369 : : {
3370 : 0 : return QgsStyle::LabelSettingsEntity;
3371 : : }
3372 : :
3373 : 0 : QgsStyle::StyleEntity QgsStyleLegendPatchShapeEntity::type() const
3374 : : {
3375 : 0 : return QgsStyle::LegendPatchShapeEntity;
3376 : : }
3377 : :
3378 : 0 : QgsStyle::StyleEntity QgsStyleSymbol3DEntity::type() const
3379 : : {
3380 : 0 : return QgsStyle::Symbol3DEntity;
3381 : : }
|