File: BaseSvgRenderer.cpp

package info (click to toggle)
qtop 2.3.4-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 3,976 kB
  • sloc: cpp: 40,477; makefile: 7
file content (210 lines) | stat: -rw-r--r-- 7,073 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
/******************************************************************************
*
* Copyright (C) 2002 Hugo PEREIRA <mailto: hugo.pereira@free.fr>
*
* This is free software; you can redistribute it and/or modify it under the
* terms of the GNU General Public License as published by the Free Software
* Foundation; either version 2 of the License, or (at your option) any later
* version.
*
* This software is distributed in the hope that it will be useful, but WITHOUT
* Any WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
* for more details.
*
* You should have received a copy of the GNU General Public License along with
* this program.  If not, see <http://www.gnu.org/licenses/>.
*
*******************************************************************************/

#include "BaseSvgRenderer.h"

#include <QDomDocument>
#include <QFile>

#if WITH_ZLIB
#include <zlib.h>
#endif

namespace Svg
{

    //________________________________________________
    BaseSvgRenderer::BaseSvgRenderer():
        QSvgRenderer(),
        Counter( "Svg::BaseSvgRenderer" )
    {}

    //________________________________________________
    void BaseSvgRenderer::createStyleSheet( QPalette palette )
    {
        palette.setCurrentColorGroup( QPalette::Active );
        styleSheet_.clear();

        auto addColor = [](QString name, QColor color )
        { return QString( ".ColorScheme-%1 {\n  color:%2;\n  stop-color:%2;\n}\n" ).arg( name, color.name()); };

        styleSheet_ += addColor( "Text", palette.color( QPalette::WindowText ) );
        styleSheet_ += addColor( "Background", palette.color( QPalette::Window ) );
        styleSheet_ += addColor( "Highlight", palette.color( QPalette::Highlight ) );
        styleSheet_ += addColor( "ViewText", palette.color( QPalette::Text ) );
        styleSheet_ += addColor( "ViewBackground", palette.color( QPalette::Base ) );
        styleSheet_ += addColor( "ViewHover", palette.color( QPalette::Highlight ) );
        styleSheet_ += addColor( "ViewFocus", palette.color( QPalette::Highlight ) );
        styleSheet_ += addColor( "ButtonText", palette.color( QPalette::ButtonText ) );
        styleSheet_ += addColor( "ButtonBackground", palette.color( QPalette::Button ) );
        styleSheet_ += addColor( "ButtonHover", palette.color( QPalette::Highlight ) );
        styleSheet_ += addColor( "ButtonFocus", palette.color( QPalette::Highlight ) );
    }

    //________________________________________________
    bool BaseSvgRenderer::load( const QString& filename )
    {

        // ret
        styleSheetIsUsed_ = false;

        #if WITH_ZLIB

        // open file
        QFile in( filename );
        in.open( QIODevice::ReadOnly );
        if( !( in.isOpen() && in.isReadable() ) ) return false;

        // when zlib is available we try to uncompress the file manually before passing to QSvgRenderer
        // this will allow to alter the file and in particular to change colors using current palette.
        auto content = _tryUncompress( in );
        if( content.isEmpty() )
        {

            in.seek(0);
            content = in.readAll();

        }

        // update stylesheet
        styleSheetIsUsed_ = content.contains( "current-color-scheme" );
        if( !styleSheet_.isEmpty() && styleSheetIsUsed_ )
        {
            QDomDocument document;
            document.setContent( content );

            // create new style element
            QDomElement element = document.createElement( QLatin1String( "style" ) );
            element.setAttribute( QLatin1String( "type" ), QLatin1String( "text/css" ) );
            element.setAttribute( QLatin1String( "id" ), QLatin1String( "current-color-scheme" ) );
            element.appendChild( document.createTextNode( styleSheet_ ) );

            // find child in document
            auto&& children( document.elementsByTagName( element.tagName() ) );
            bool replaced( false );
            for( int i = 0; i < children.size() && !replaced; ++i )
            {

                auto&& child( children.at(i) );
                if( child.toElement().attribute( QLatin1String( "id" ) ) == QLatin1String( "current-color-scheme" ) )
                {
                    auto parent( child.parentNode() );
                    parent.replaceChild( element, child );
                    replaced = true;
                    break;
                }
            }

            if( replaced ) content = document.toByteArray();

        }

        bool loaded( QSvgRenderer::load( content ) );

        #else

        // load filename directly
        bool loaded( QSvgRenderer::load( filename ) );

        #endif

        return loaded;

    }

    //________________________________________________
    QByteArray BaseSvgRenderer::_tryUncompress( QIODevice& in ) const
    {

        QByteArray out;
        if( !( in.isOpen() && in.isReadable() ) ) return out;

        #if WITH_ZLIB

        // initialize zlib stream
        z_stream stream;
        stream.zalloc = nullptr;
        stream.zfree = nullptr;
        stream.opaque = nullptr;
        stream.avail_in = 0;
        stream.next_in = nullptr;

        if( inflateInit2(&stream, MAX_WBITS + 16) != Z_OK) return out;

        // read chunks of max size chunksize from input source
        const int chunkSize = 16384;
        forever
        {

            // read chunk from input and store in stream
            auto source = in.read( chunkSize );
            if( source.isEmpty() )
            {
                out.chop(stream.avail_out);
                break;
            }

            stream.avail_in = source.size();
            stream.next_in = reinterpret_cast<Bytef*>(source.data());

            // uncompress chunk and append to output stream
            int result = Z_OK;
            do
            {

                // prepare destination buffer
                auto size = out.size();
                out.resize( size+chunkSize );
                stream.avail_out = chunkSize;
                stream.next_out = reinterpret_cast<Bytef*>(out.data() + size);

                result = inflate(&stream, Z_NO_FLUSH);
                switch( result )
                {
                    case Z_NEED_DICT:
                    case Z_DATA_ERROR:
                    case Z_STREAM_ERROR:
                    case Z_MEM_ERROR:
                    {
                        inflateEnd( &stream );
                        out.chop(stream.avail_out);
                        return out;
                    }
                }


            } while( stream.avail_out == 0 );

            // Chop off trailing space in the buffer
            out.chop(stream.avail_out);

            if( result == Z_STREAM_END )
            {
                // Make sure there are no more members to process before exiting
                if( !(stream.avail_in && inflateReset(&stream) == Z_OK ) ) break;
            }

        }

        inflateEnd(&stream);
        #endif
        return out;

    }
}