| Absolute File Name: | /home/qt/qt5_coco/qt5/qtbase/src/gui/image/qpnghandler.cpp | 
| Switch to Source code | Preprocessed file | 
| Line | Source | Count | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 1 | - | |||||||||||||
| 2 | - | |||||||||||||
| 3 | - | |||||||||||||
| 4 | - | |||||||||||||
| 5 | class QPngHandlerPrivate | - | ||||||||||||
| 6 | { | - | ||||||||||||
| 7 | public: | - | ||||||||||||
| 8 | enum State { | - | ||||||||||||
| 9 | Ready, | - | ||||||||||||
| 10 | ReadHeader, | - | ||||||||||||
| 11 | ReadingEnd, | - | ||||||||||||
| 12 | Error | - | ||||||||||||
| 13 | }; | - | ||||||||||||
| 14 | - | |||||||||||||
| 15 | QPngHandlerPrivate(QPngHandler *qq) | - | ||||||||||||
| 16 | : gamma(0.0), fileGamma(0.0), quality(2), png_ptr(0), info_ptr(0), end_info(0), state(Ready), q(qq) | - | ||||||||||||
| 17 | { } | - | ||||||||||||
| 18 | - | |||||||||||||
| 19 | float gamma; | - | ||||||||||||
| 20 | float fileGamma; | - | ||||||||||||
| 21 | int quality; | - | ||||||||||||
| 22 | QString description; | - | ||||||||||||
| 23 | QSize scaledSize; | - | ||||||||||||
| 24 | QStringList readTexts; | - | ||||||||||||
| 25 | - | |||||||||||||
| 26 | png_struct *png_ptr; | - | ||||||||||||
| 27 | png_info *info_ptr; | - | ||||||||||||
| 28 | png_info *end_info; | - | ||||||||||||
| 29 | - | |||||||||||||
| 30 | bool readPngHeader(); | - | ||||||||||||
| 31 | bool readPngImage(QImage *image); | - | ||||||||||||
| 32 | void readPngTexts(png_info *info); | - | ||||||||||||
| 33 | - | |||||||||||||
| 34 | QImage::Format readImageFormat(); | - | ||||||||||||
| 35 | - | |||||||||||||
| 36 | struct AllocatedMemoryPointers { | - | ||||||||||||
| 37 | AllocatedMemoryPointers() | - | ||||||||||||
| 38 | : row_pointers(0), accRow(0), inRow(0), outRow(0) | - | ||||||||||||
| 39 | { } | - | ||||||||||||
| 40 | void deallocate() | - | ||||||||||||
| 41 | { | - | ||||||||||||
| 42 | delete [] row_pointers; | - | ||||||||||||
| 43 | row_pointers = 0; | - | ||||||||||||
| 44 | delete [] accRow; | - | ||||||||||||
| 45 | accRow = 0; | - | ||||||||||||
| 46 | delete [] inRow; | - | ||||||||||||
| 47 | inRow = 0; | - | ||||||||||||
| 48 | delete [] outRow; | - | ||||||||||||
| 49 | outRow = 0; | - | ||||||||||||
| 50 | } | - | ||||||||||||
| 51 | - | |||||||||||||
| 52 | png_byte **row_pointers; | - | ||||||||||||
| 53 | quint32 *accRow; | - | ||||||||||||
| 54 | png_byte *inRow; | - | ||||||||||||
| 55 | uchar *outRow; | - | ||||||||||||
| 56 | }; | - | ||||||||||||
| 57 | - | |||||||||||||
| 58 | AllocatedMemoryPointers amp; | - | ||||||||||||
| 59 | - | |||||||||||||
| 60 | State state; | - | ||||||||||||
| 61 | - | |||||||||||||
| 62 | QPngHandler *q; | - | ||||||||||||
| 63 | }; | - | ||||||||||||
| 64 | - | |||||||||||||
| 65 | - | |||||||||||||
| 66 | class QPNGImageWriter { | - | ||||||||||||
| 67 | public: | - | ||||||||||||
| 68 | explicit QPNGImageWriter(QIODevice*); | - | ||||||||||||
| 69 | ~QPNGImageWriter(); | - | ||||||||||||
| 70 | - | |||||||||||||
| 71 | enum DisposalMethod { Unspecified, NoDisposal, RestoreBackground, RestoreImage }; | - | ||||||||||||
| 72 | void setDisposalMethod(DisposalMethod); | - | ||||||||||||
| 73 | void setLooping(int loops=0); | - | ||||||||||||
| 74 | void setFrameDelay(int msecs); | - | ||||||||||||
| 75 | void setGamma(float); | - | ||||||||||||
| 76 | - | |||||||||||||
| 77 | bool writeImage(const QImage& img, int x, int y); | - | ||||||||||||
| 78 | bool writeImage(const QImage& img, volatile int quality, const QString &description, int x, int y); | - | ||||||||||||
| 79 | bool writeImage(const QImage& img) | - | ||||||||||||
| 80 | { return writeImage(img, 0, 0); } | - | ||||||||||||
| 81 | bool writeImage(const QImage& img, int quality, const QString &description) | - | ||||||||||||
| 82 | { return writeImage(img, quality, description, 0, 0); } | - | ||||||||||||
| 83 | - | |||||||||||||
| 84 | QIODevice* device() { return dev; } | - | ||||||||||||
| 85 | - | |||||||||||||
| 86 | private: | - | ||||||||||||
| 87 | QIODevice* dev; | - | ||||||||||||
| 88 | int frames_written; | - | ||||||||||||
| 89 | DisposalMethod disposal; | - | ||||||||||||
| 90 | int looping; | - | ||||||||||||
| 91 | int ms_delay; | - | ||||||||||||
| 92 | float gamma; | - | ||||||||||||
| 93 | }; | - | ||||||||||||
| 94 | - | |||||||||||||
| 95 | extern "C" { | - | ||||||||||||
| 96 | static | - | ||||||||||||
| 97 | void iod_read_fn(png_structp png_ptr, png_bytep data, png_size_t length) | - | ||||||||||||
| 98 | { | - | ||||||||||||
| 99 | QPngHandlerPrivate *d = (QPngHandlerPrivate *)png_get_io_ptr(png_ptr); | - | ||||||||||||
| 100 | QIODevice *in = d->q->device(); | - | ||||||||||||
| 101 | - | |||||||||||||
| 102 | if (d->state == QPngHandlerPrivate::ReadingEnd && !in->isSequential() && (in->size() - in->pos()) < 4 && length == 4) { | - | ||||||||||||
| 103 | - | |||||||||||||
| 104 | uchar endcrc[4] = { 0xae, 0x42, 0x60, 0x82 }; | - | ||||||||||||
| 105 | memcpy(data, endcrc, 4); | - | ||||||||||||
| 106 | in->seek(in->size()); | - | ||||||||||||
| 107 | return; | - | ||||||||||||
| 108 | } | - | ||||||||||||
| 109 | - | |||||||||||||
| 110 | while (length) { | - | ||||||||||||
| 111 | int nr = in->read((char*)data, length); | - | ||||||||||||
| 112 | if (nr <= 0) { | - | ||||||||||||
| 113 | png_error(png_ptr, "Read Error"); | - | ||||||||||||
| 114 | return; | - | ||||||||||||
| 115 | } | - | ||||||||||||
| 116 | length -= nr; | - | ||||||||||||
| 117 | } | - | ||||||||||||
| 118 | } | - | ||||||||||||
| 119 | - | |||||||||||||
| 120 | - | |||||||||||||
| 121 | static | - | ||||||||||||
| 122 | void qpiw_write_fn(png_structp png_ptr, png_bytep data, png_size_t length) | - | ||||||||||||
| 123 | { | - | ||||||||||||
| 124 | QPNGImageWriter* qpiw = (QPNGImageWriter*)png_get_io_ptr(png_ptr); | - | ||||||||||||
| 125 | QIODevice* out = qpiw->device(); | - | ||||||||||||
| 126 | - | |||||||||||||
| 127 | uint nr = out->write((char*)data, length); | - | ||||||||||||
| 128 | if (nr != length) { | - | ||||||||||||
| 129 | png_error(png_ptr, "Write Error"); | - | ||||||||||||
| 130 | return; | - | ||||||||||||
| 131 | } | - | ||||||||||||
| 132 | } | - | ||||||||||||
| 133 | - | |||||||||||||
| 134 | - | |||||||||||||
| 135 | static | - | ||||||||||||
| 136 | void qpiw_flush_fn(png_structp ) | - | ||||||||||||
| 137 | { | - | ||||||||||||
| 138 | } | - | ||||||||||||
| 139 | - | |||||||||||||
| 140 | } | - | ||||||||||||
| 141 | - | |||||||||||||
| 142 | static | - | ||||||||||||
| 143 | void setup_qt(QImage& image, png_structp png_ptr, png_infop info_ptr, QSize scaledSize, bool *doScaledRead, float screen_gamma=0.0, float file_gamma=0.0) | - | ||||||||||||
| 144 | { | - | ||||||||||||
| 145 | if (screen_gamma != 0.0 && file_gamma != 0.0) | - | ||||||||||||
| 146 | png_set_gamma(png_ptr, 1.0f / screen_gamma, file_gamma); | - | ||||||||||||
| 147 | - | |||||||||||||
| 148 | png_uint_32 width; | - | ||||||||||||
| 149 | png_uint_32 height; | - | ||||||||||||
| 150 | int bit_depth; | - | ||||||||||||
| 151 | int color_type; | - | ||||||||||||
| 152 | png_bytep trans_alpha = 0; | - | ||||||||||||
| 153 | png_color_16p trans_color_p = 0; | - | ||||||||||||
| 154 | int num_trans; | - | ||||||||||||
| 155 | png_colorp palette = 0; | - | ||||||||||||
| 156 | int num_palette; | - | ||||||||||||
| 157 | int interlace_method; | - | ||||||||||||
| 158 | png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type, &interlace_method, 0, 0); | - | ||||||||||||
| 159 | png_set_interlace_handling(png_ptr); | - | ||||||||||||
| 160 | - | |||||||||||||
| 161 | if (color_type == 0) { | - | ||||||||||||
| 162 | - | |||||||||||||
| 163 | if (bit_depth == 1 && png_get_channels(png_ptr, info_ptr) == 1) { | - | ||||||||||||
| 164 | png_set_invert_mono(png_ptr); | - | ||||||||||||
| 165 | png_read_update_info(png_ptr, info_ptr); | - | ||||||||||||
| 166 | if (image.size() != QSize(width, height) || image.format() != QImage::Format_Mono) { | - | ||||||||||||
| 167 | image = QImage(width, height, QImage::Format_Mono); | - | ||||||||||||
| 168 | if (image.isNull()) | - | ||||||||||||
| 169 | return; | - | ||||||||||||
| 170 | } | - | ||||||||||||
| 171 | image.setColorCount(2); | - | ||||||||||||
| 172 | image.setColor(1, qRgb(0,0,0)); | - | ||||||||||||
| 173 | image.setColor(0, qRgb(255,255,255)); | - | ||||||||||||
| 174 | if (png_get_tRNS(png_ptr, info_ptr, &trans_alpha, &num_trans, &trans_color_p) && trans_color_p) { | - | ||||||||||||
| 175 | const int g = trans_color_p->gray; | - | ||||||||||||
| 176 | - | |||||||||||||
| 177 | - | |||||||||||||
| 178 | if (g == 0) | - | ||||||||||||
| 179 | image.setColor(1, qRgba(0, 0, 0, 0)); | - | ||||||||||||
| 180 | else if (g == 1) | - | ||||||||||||
| 181 | image.setColor(0, qRgba(255, 255, 255, 0)); | - | ||||||||||||
| 182 | } | - | ||||||||||||
| 183 | } else if (bit_depth == 16 && png_get_valid(png_ptr, info_ptr, 0x0010)) { | - | ||||||||||||
| 184 | png_set_expand(png_ptr); | - | ||||||||||||
| 185 | png_set_strip_16(png_ptr); | - | ||||||||||||
| 186 | png_set_gray_to_rgb(png_ptr); | - | ||||||||||||
| 187 | if (image.size() != QSize(width, height) || image.format() != QImage::Format_ARGB32) { | - | ||||||||||||
| 188 | image = QImage(width, height, QImage::Format_ARGB32); | - | ||||||||||||
| 189 | if (image.isNull()) | - | ||||||||||||
| 190 | return; | - | ||||||||||||
| 191 | } | - | ||||||||||||
| 192 | if (QSysInfo::ByteOrder == QSysInfo::BigEndian) | - | ||||||||||||
| 193 | png_set_swap_alpha(png_ptr); | - | ||||||||||||
| 194 | - | |||||||||||||
| 195 | png_read_update_info(png_ptr, info_ptr); | - | ||||||||||||
| 196 | } else if (bit_depth == 8 && !png_get_valid(png_ptr, info_ptr, 0x0010)) { | - | ||||||||||||
| 197 | png_set_expand(png_ptr); | - | ||||||||||||
| 198 | if (image.size() != QSize(width, height) || image.format() != QImage::Format_Grayscale8) { | - | ||||||||||||
| 199 | image = QImage(width, height, QImage::Format_Grayscale8); | - | ||||||||||||
| 200 | if (image.isNull()) | - | ||||||||||||
| 201 | return; | - | ||||||||||||
| 202 | } | - | ||||||||||||
| 203 | - | |||||||||||||
| 204 | png_read_update_info(png_ptr, info_ptr); | - | ||||||||||||
| 205 | } else { | - | ||||||||||||
| 206 | if (bit_depth == 16) | - | ||||||||||||
| 207 | png_set_strip_16(png_ptr); | - | ||||||||||||
| 208 | else if (bit_depth < 8) | - | ||||||||||||
| 209 | png_set_packing(png_ptr); | - | ||||||||||||
| 210 | int ncols = bit_depth < 8 ? 1 << bit_depth : 256; | - | ||||||||||||
| 211 | png_read_update_info(png_ptr, info_ptr); | - | ||||||||||||
| 212 | if (image.size() != QSize(width, height) || image.format() != QImage::Format_Indexed8) { | - | ||||||||||||
| 213 | image = QImage(width, height, QImage::Format_Indexed8); | - | ||||||||||||
| 214 | if (image.isNull()) | - | ||||||||||||
| 215 | return; | - | ||||||||||||
| 216 | } | - | ||||||||||||
| 217 | image.setColorCount(ncols); | - | ||||||||||||
| 218 | for (int i=0; i<ncols; i++) { | - | ||||||||||||
| 219 | int c = i*255/(ncols-1); | - | ||||||||||||
| 220 | image.setColor(i, qRgba(c,c,c,0xff)); | - | ||||||||||||
| 221 | } | - | ||||||||||||
| 222 | if (png_get_tRNS(png_ptr, info_ptr, &trans_alpha, &num_trans, &trans_color_p) && trans_color_p) { | - | ||||||||||||
| 223 | const int g = trans_color_p->gray; | - | ||||||||||||
| 224 | if (g < ncols) { | - | ||||||||||||
| 225 | image.setColor(g, 0); | - | ||||||||||||
| 226 | } | - | ||||||||||||
| 227 | } | - | ||||||||||||
| 228 | } | - | ||||||||||||
| 229 | } else if (color_type == (2 | 1) | - | ||||||||||||
| 230 | && png_get_PLTE(png_ptr, info_ptr, &palette, &num_palette) | - | ||||||||||||
| 231 | && num_palette <= 256) | - | ||||||||||||
| 232 | { | - | ||||||||||||
| 233 | - | |||||||||||||
| 234 | if (bit_depth != 1) | - | ||||||||||||
| 235 | png_set_packing(png_ptr); | - | ||||||||||||
| 236 | png_read_update_info(png_ptr, info_ptr); | - | ||||||||||||
| 237 | png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type, 0, 0, 0); | - | ||||||||||||
| 238 | QImage::Format format = bit_depth == 1 ? QImage::Format_Mono : QImage::Format_Indexed8; | - | ||||||||||||
| 239 | if (image.size() != QSize(width, height) || image.format() != format) { | - | ||||||||||||
| 240 | image = QImage(width, height, format); | - | ||||||||||||
| 241 | if (image.isNull()) | - | ||||||||||||
| 242 | return; | - | ||||||||||||
| 243 | } | - | ||||||||||||
| 244 | png_get_PLTE(png_ptr, info_ptr, &palette, &num_palette); | - | ||||||||||||
| 245 | image.setColorCount(num_palette); | - | ||||||||||||
| 246 | int i = 0; | - | ||||||||||||
| 247 | if (png_get_tRNS(png_ptr, info_ptr, &trans_alpha, &num_trans, &trans_color_p) && trans_alpha) { | - | ||||||||||||
| 248 | while (i < num_trans) { | - | ||||||||||||
| 249 | image.setColor(i, qRgba( | - | ||||||||||||
| 250 | palette[i].red, | - | ||||||||||||
| 251 | palette[i].green, | - | ||||||||||||
| 252 | palette[i].blue, | - | ||||||||||||
| 253 | trans_alpha[i] | - | ||||||||||||
| 254 | ) | - | ||||||||||||
| 255 | ); | - | ||||||||||||
| 256 | i++; | - | ||||||||||||
| 257 | } | - | ||||||||||||
| 258 | } | - | ||||||||||||
| 259 | while (i < num_palette) { | - | ||||||||||||
| 260 | image.setColor(i, qRgba( | - | ||||||||||||
| 261 | palette[i].red, | - | ||||||||||||
| 262 | palette[i].green, | - | ||||||||||||
| 263 | palette[i].blue, | - | ||||||||||||
| 264 | 0xff | - | ||||||||||||
| 265 | ) | - | ||||||||||||
| 266 | ); | - | ||||||||||||
| 267 | i++; | - | ||||||||||||
| 268 | } | - | ||||||||||||
| 269 | } else { | - | ||||||||||||
| 270 | - | |||||||||||||
| 271 | if (bit_depth == 16) | - | ||||||||||||
| 272 | png_set_strip_16(png_ptr); | - | ||||||||||||
| 273 | - | |||||||||||||
| 274 | png_set_expand(png_ptr); | - | ||||||||||||
| 275 | - | |||||||||||||
| 276 | if (color_type == (4)) | - | ||||||||||||
| 277 | png_set_gray_to_rgb(png_ptr); | - | ||||||||||||
| 278 | - | |||||||||||||
| 279 | QImage::Format format = QImage::Format_ARGB32; | - | ||||||||||||
| 280 | - | |||||||||||||
| 281 | if (!(color_type & 4) | - | ||||||||||||
| 282 | && !png_get_valid(png_ptr, info_ptr, 0x0010)) { | - | ||||||||||||
| 283 | png_set_filler(png_ptr, 0xff, QSysInfo::ByteOrder == QSysInfo::BigEndian ? | - | ||||||||||||
| 284 | 0 : 1); | - | ||||||||||||
| 285 | - | |||||||||||||
| 286 | format = QImage::Format_RGB32; | - | ||||||||||||
| 287 | } | - | ||||||||||||
| 288 | QSize outSize(width,height); | - | ||||||||||||
| 289 | if (!scaledSize.isEmpty() && quint32(scaledSize.width()) <= width && | - | ||||||||||||
| 290 | quint32(scaledSize.height()) <= height && interlace_method == 0) { | - | ||||||||||||
| 291 | - | |||||||||||||
| 292 | outSize = scaledSize; | - | ||||||||||||
| 293 | if (doScaledRead) | - | ||||||||||||
| 294 | *doScaledRead = true; | - | ||||||||||||
| 295 | } | - | ||||||||||||
| 296 | if (image.size() != outSize || image.format() != format) { | - | ||||||||||||
| 297 | image = QImage(outSize, format); | - | ||||||||||||
| 298 | if (image.isNull()) | - | ||||||||||||
| 299 | return; | - | ||||||||||||
| 300 | } | - | ||||||||||||
| 301 | - | |||||||||||||
| 302 | if (QSysInfo::ByteOrder == QSysInfo::BigEndian) | - | ||||||||||||
| 303 | png_set_swap_alpha(png_ptr); | - | ||||||||||||
| 304 | - | |||||||||||||
| 305 | png_read_update_info(png_ptr, info_ptr); | - | ||||||||||||
| 306 | } | - | ||||||||||||
| 307 | - | |||||||||||||
| 308 | - | |||||||||||||
| 309 | if (QSysInfo::ByteOrder == QSysInfo::LittleEndian) { | - | ||||||||||||
| 310 | png_set_bgr(png_ptr); | - | ||||||||||||
| 311 | } | - | ||||||||||||
| 312 | } | - | ||||||||||||
| 313 | - | |||||||||||||
| 314 | static void read_image_scaled(QImage *outImage, png_structp png_ptr, png_infop info_ptr, | - | ||||||||||||
| 315 | QPngHandlerPrivate::AllocatedMemoryPointers &, QSize scaledSize) | - | ||||||||||||
| 316 | { | - | ||||||||||||
| 317 | - | |||||||||||||
| 318 | png_uint_32 width = 0; | - | ||||||||||||
| 319 | png_uint_32 height = 0; | - | ||||||||||||
| 320 | png_int_32 offset_x = 0; | - | ||||||||||||
| 321 | png_int_32 offset_y = 0; | - | ||||||||||||
| 322 | - | |||||||||||||
| 323 | int bit_depth = 0; | - | ||||||||||||
| 324 | int color_type = 0; | - | ||||||||||||
| 325 | int unit_type = 0; | - | ||||||||||||
| 326 | png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type, 0, 0, 0); | - | ||||||||||||
| 327 | png_get_oFFs(png_ptr, info_ptr, &offset_x, &offset_y, &unit_type); | - | ||||||||||||
| 328 | uchar *data = outImage->bits(); | - | ||||||||||||
| 329 | int bpl = outImage->bytesPerLine(); | - | ||||||||||||
| 330 | - | |||||||||||||
| 331 | if (scaledSize.isEmpty() || !width || !height) | - | ||||||||||||
| 332 | return; | - | ||||||||||||
| 333 | - | |||||||||||||
| 334 | const quint32 iysz = height; | - | ||||||||||||
| 335 | const quint32 ixsz = width; | - | ||||||||||||
| 336 | const quint32 oysz = scaledSize.height(); | - | ||||||||||||
| 337 | const quint32 oxsz = scaledSize.width(); | - | ||||||||||||
| 338 | const quint32 ibw = 4*width; | - | ||||||||||||
| 339 | amp.accRow = new quint32[ibw]; | - | ||||||||||||
| 340 | memset(amp.accRow, 0, ibw*sizeof(quint32)); | - | ||||||||||||
| 341 | amp.inRow = new png_byte[ibw]; | - | ||||||||||||
| 342 | memset(amp.inRow, 0, ibw*sizeof(png_byte)); | - | ||||||||||||
| 343 | amp.outRow = new uchar[ibw]; | - | ||||||||||||
| 344 | memset(amp.outRow, 0, ibw*sizeof(uchar)); | - | ||||||||||||
| 345 | qint32 rval = 0; | - | ||||||||||||
| 346 | for (quint32 oy=0; oy<oysz; oy++) { | - | ||||||||||||
| 347 | - | |||||||||||||
| 348 | for (quint32 i=0; i < ibw; i++) | - | ||||||||||||
| 349 | amp.accRow[i] = rval*amp.inRow[i]; | - | ||||||||||||
| 350 | - | |||||||||||||
| 351 | for (rval = iysz-rval; rval > 0; rval-=oysz) { | - | ||||||||||||
| 352 | png_read_row(png_ptr, amp.inRow, __null); | - | ||||||||||||
| 353 | quint32 fact = qMin(oysz, quint32(rval)); | - | ||||||||||||
| 354 | for (quint32 i=0; i < ibw; i++) | - | ||||||||||||
| 355 | amp.accRow[i] += fact*amp.inRow[i]; | - | ||||||||||||
| 356 | } | - | ||||||||||||
| 357 | rval *= -1; | - | ||||||||||||
| 358 | - | |||||||||||||
| 359 | - | |||||||||||||
| 360 | for (quint32 i=0; i < ibw; i++) | - | ||||||||||||
| 361 | amp.outRow[i] = uchar(amp.accRow[i]/iysz); | - | ||||||||||||
| 362 | - | |||||||||||||
| 363 | quint32 a[4] = {0, 0, 0, 0}; | - | ||||||||||||
| 364 | qint32 cval = oxsz; | - | ||||||||||||
| 365 | quint32 ix = 0; | - | ||||||||||||
| 366 | for (quint32 ox=0; ox<oxsz; ox++) { | - | ||||||||||||
| 367 | for (quint32 i=0; i < 4; i++) | - | ||||||||||||
| 368 | a[i] = cval * amp.outRow[ix+i]; | - | ||||||||||||
| 369 | for (cval = ixsz - cval; cval > 0; cval-=oxsz) { | - | ||||||||||||
| 370 | ix += 4; | - | ||||||||||||
| 371 | if (ix >= ibw) | - | ||||||||||||
| 372 | break; | - | ||||||||||||
| 373 | quint32 fact = qMin(oxsz, quint32(cval)); | - | ||||||||||||
| 374 | for (quint32 i=0; i < 4; i++) | - | ||||||||||||
| 375 | a[i] += fact * amp.outRow[ix+i]; | - | ||||||||||||
| 376 | } | - | ||||||||||||
| 377 | cval *= -1; | - | ||||||||||||
| 378 | for (quint32 i=0; i < 4; i++) | - | ||||||||||||
| 379 | data[(4*ox)+i] = uchar(a[i]/ixsz); | - | ||||||||||||
| 380 | } | - | ||||||||||||
| 381 | data += bpl; | - | ||||||||||||
| 382 | } | - | ||||||||||||
| 383 | amp.deallocate(); | - | ||||||||||||
| 384 | - | |||||||||||||
| 385 | outImage->setDotsPerMeterX((png_get_x_pixels_per_meter(png_ptr,info_ptr)*oxsz)/ixsz); | - | ||||||||||||
| 386 | outImage->setDotsPerMeterY((png_get_y_pixels_per_meter(png_ptr,info_ptr)*oysz)/iysz); | - | ||||||||||||
| 387 | - | |||||||||||||
| 388 | if (unit_type == 0) | - | ||||||||||||
| 389 | outImage->setOffset(QPoint(offset_x*oxsz/ixsz, offset_y*oysz/iysz)); | - | ||||||||||||
| 390 | - | |||||||||||||
| 391 | } | - | ||||||||||||
| 392 | - | |||||||||||||
| 393 | extern "C" { | - | ||||||||||||
| 394 | static void qt_png_warning(png_structp , png_const_charp message) | - | ||||||||||||
| 395 | { | - | ||||||||||||
| 396 | QMessageLogger(__FILE__, 486492, __PRETTY_FUNCTION__).warning("libpng warning: %s", message); | - | ||||||||||||
| 397 | } | - | ||||||||||||
| 398 | - | |||||||||||||
| 399 | } | - | ||||||||||||
| 400 | - | |||||||||||||
| 401 | - | |||||||||||||
| 402 | void QPngHandlerPrivate::readPngTexts(png_info *info) | - | ||||||||||||
| 403 | { | - | ||||||||||||
| 404 | png_textp text_ptr; | - | ||||||||||||
| 405 | int num_text=0; | - | ||||||||||||
| 406 | png_get_text(png_ptr, info, &text_ptr, &num_text); | - | ||||||||||||
| 407 | - | |||||||||||||
| 408 | while (num_text--) { | - | ||||||||||||
| 409 | QString key, value; | - | ||||||||||||
| 410 | key = QString::fromLatin1(text_ptr->key); | - | ||||||||||||
| 411 | - | |||||||||||||
| 412 | - | |||||||||||||
| 413 | - | |||||||||||||
| 414 | - | |||||||||||||
| 415 | - | |||||||||||||
| 416 | { | - | ||||||||||||
| 417 | value = QString::fromLatin1(text_ptr->text, int(text_ptr->text_length)); | - | ||||||||||||
| 418 | } | - | ||||||||||||
| 419 | if (!description.isEmpty()) | - | ||||||||||||
| 420 | description += QLatin1String("\n\n"); | - | ||||||||||||
| 421 | description += key + QLatin1String(": ") + value.simplified(); | - | ||||||||||||
| 422 | readTexts.append(key); | - | ||||||||||||
| 423 | readTexts.append(value); | - | ||||||||||||
| 424 | text_ptr++; | - | ||||||||||||
| 425 | } | - | ||||||||||||
| 426 | } | - | ||||||||||||
| 427 | - | |||||||||||||
| 428 | - | |||||||||||||
| 429 | bool QPngHandlerPrivate::readPngHeader() | - | ||||||||||||
| 430 | { | - | ||||||||||||
| 431 | state = Error; | - | ||||||||||||
| 432 | png_ptr = png_create_read_struct("1.2.50",0,0,0); | - | ||||||||||||
| 433 |     if (!png_ptr
  | 0 | ||||||||||||
| 434 |         return never executed:   false;return false;never executed:  return false; | 0 | ||||||||||||
| 435 | - | |||||||||||||
| 436 | png_set_error_fn(png_ptr, 0, 0, qt_png_warning); | - | ||||||||||||
| 437 | - | |||||||||||||
| 438 | info_ptr = png_create_info_struct(png_ptr); | - | ||||||||||||
| 439 |     if (!info_ptr
  | 0 | ||||||||||||
| 440 | png_destroy_read_struct(&png_ptr, 0, 0); | - | ||||||||||||
| 441 | png_ptr = 0; | - | ||||||||||||
| 442 |         return never executed:   false;return false;never executed:  return false; | 0 | ||||||||||||
| 443 | } | - | ||||||||||||
| 444 | - | |||||||||||||
| 445 | end_info = png_create_info_struct(png_ptr); | - | ||||||||||||
| 446 |     if (!end_info
  | 0 | ||||||||||||
| 447 | png_destroy_read_struct(&png_ptr, &info_ptr, 0); | - | ||||||||||||
| 448 | png_ptr = 0; | - | ||||||||||||
| 449 |         return never executed:   false;return false;never executed:  return false; | 0 | ||||||||||||
| 450 | } | - | ||||||||||||
| 451 | - | |||||||||||||
| 452 |     if (_setjmp (((png_ptr)->jmpbuf))
  | 0 | ||||||||||||
| 453 | png_destroy_read_struct(&png_ptr, &info_ptr, &end_info); | - | ||||||||||||
| 454 | png_ptr = 0; | - | ||||||||||||
| 455 |         return never executed:   false;return false;never executed:  return false; | 0 | ||||||||||||
| 456 | } | - | ||||||||||||
| 457 | - | |||||||||||||
| 458 | png_set_read_fn(png_ptr, this, iod_read_fn); | - | ||||||||||||
| 459 | png_read_info(png_ptr, info_ptr); | - | ||||||||||||
| 460 | - | |||||||||||||
| 461 | readPngTexts(info_ptr); | - | ||||||||||||
| 462 | - | |||||||||||||
| 463 |     if (png_get_valid(png_ptr, info_ptr, 0x0001)
  | 0 | ||||||||||||
| 464 | double file_gamma = 0.0; | - | ||||||||||||
| 465 | png_get_gAMA(png_ptr, info_ptr, &file_gamma); | - | ||||||||||||
| 466 | fileGamma = file_gamma; | - | ||||||||||||
| 467 |     } never executed:  end of block | 0 | ||||||||||||
| 468 | - | |||||||||||||
| 469 | state = ReadHeader; | - | ||||||||||||
| 470 |     return never executed:   true;return true;never executed:  return true; | 0 | ||||||||||||
| 471 | } | - | ||||||||||||
| 472 | - | |||||||||||||
| 473 | bool QPngHandlerPrivate::readPngImage(QImage *outImage) | - | ||||||||||||
| 474 | { | - | ||||||||||||
| 475 | if (state == Error) | - | ||||||||||||
| 476 | return false; | - | ||||||||||||
| 477 | - | |||||||||||||
| 478 | if (state == Ready && !readPngHeader()) { | - | ||||||||||||
| 479 | state = Error; | - | ||||||||||||
| 480 | return false; | - | ||||||||||||
| 481 | } | - | ||||||||||||
| 482 | - | |||||||||||||
| 483 | if (_setjmp (((png_ptr)->jmpbuf))) { | - | ||||||||||||
| 484 | png_destroy_read_struct(&png_ptr, &info_ptr, &end_info); | - | ||||||||||||
| 485 | png_ptr = 0; | - | ||||||||||||
| 486 | amp.deallocate(); | - | ||||||||||||
| 487 | state = Error; | - | ||||||||||||
| 488 | return false; | - | ||||||||||||
| 489 | } | - | ||||||||||||
| 490 | - | |||||||||||||
| 491 | bool doScaledRead = false; | - | ||||||||||||
| 492 | setup_qt(*outImage, png_ptr, info_ptr, scaledSize, &doScaledRead, gamma, fileGamma); | - | ||||||||||||
| 493 | - | |||||||||||||
| 494 | if (outImage->isNull()) { | - | ||||||||||||
| 495 | png_destroy_read_struct(&png_ptr, &info_ptr, &end_info); | - | ||||||||||||
| 496 | png_ptr = 0; | - | ||||||||||||
| 497 | amp.deallocate(); | - | ||||||||||||
| 498 | state = Error; | - | ||||||||||||
| 499 | return false; | - | ||||||||||||
| 500 | } | - | ||||||||||||
| 501 | - | |||||||||||||
| 502 | if (doScaledRead) { | - | ||||||||||||
| 503 | read_image_scaled(outImage, png_ptr, info_ptr, amp, scaledSize); | - | ||||||||||||
| 504 | } else { | - | ||||||||||||
| 505 | png_uint_32 width = 0; | - | ||||||||||||
| 506 | png_uint_32 height = 0; | - | ||||||||||||
| 507 | png_int_32 offset_x = 0; | - | ||||||||||||
| 508 | png_int_32 offset_y = 0; | - | ||||||||||||
| 509 | - | |||||||||||||
| 510 | int bit_depth = 0; | - | ||||||||||||
| 511 | int color_type = 0; | - | ||||||||||||
| 512 | int unit_type = 0; | - | ||||||||||||
| 513 | png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type, 0, 0, 0); | - | ||||||||||||
| 514 | png_get_oFFs(png_ptr, info_ptr, &offset_x, &offset_y, &unit_type); | - | ||||||||||||
| 515 | uchar *data = outImage->bits(); | - | ||||||||||||
| 516 | int bpl = outImage->bytesPerLine(); | - | ||||||||||||
| 517 | amp.row_pointers = new png_bytep[height]; | - | ||||||||||||
| 518 | - | |||||||||||||
| 519 | for (uint y = 0; y < height; y++) | - | ||||||||||||
| 520 | amp.row_pointers[y] = data + y * bpl; | - | ||||||||||||
| 521 | - | |||||||||||||
| 522 | png_read_image(png_ptr, amp.row_pointers); | - | ||||||||||||
| 523 | amp.deallocate(); | - | ||||||||||||
| 524 | - | |||||||||||||
| 525 | outImage->setDotsPerMeterX(png_get_x_pixels_per_meter(png_ptr,info_ptr)); | - | ||||||||||||
| 526 | outImage->setDotsPerMeterY(png_get_y_pixels_per_meter(png_ptr,info_ptr)); | - | ||||||||||||
| 527 | - | |||||||||||||
| 528 | if (unit_type == 0) | - | ||||||||||||
| 529 | outImage->setOffset(QPoint(offset_x, offset_y)); | - | ||||||||||||
| 530 | - | |||||||||||||
| 531 | - | |||||||||||||
| 532 | if (color_type == (2 | 1) && outImage->format() == QImage::Format_Indexed8) { | - | ||||||||||||
| 533 | int color_table_size = outImage->colorCount(); | - | ||||||||||||
| 534 | for (int y=0; y<(int)height; ++y) { | - | ||||||||||||
| 535 | uchar *p = (data + (y) * bpl); | - | ||||||||||||
| 536 | uchar *end = p + width; | - | ||||||||||||
| 537 | while (p < end) { | - | ||||||||||||
| 538 | if (*p >= color_table_size) | - | ||||||||||||
| 539 | *p = 0; | - | ||||||||||||
| 540 | ++p; | - | ||||||||||||
| 541 | } | - | ||||||||||||
| 542 | } | - | ||||||||||||
| 543 | } | - | ||||||||||||
| 544 | } | - | ||||||||||||
| 545 | - | |||||||||||||
| 546 | state = ReadingEnd; | - | ||||||||||||
| 547 | png_read_end(png_ptr, end_info); | - | ||||||||||||
| 548 | - | |||||||||||||
| 549 | readPngTexts(end_info); | - | ||||||||||||
| 550 | for (int i = 0; i < readTexts.size()-1; i+=2) | - | ||||||||||||
| 551 | outImage->setText(readTexts.at(i), readTexts.at(i+1)); | - | ||||||||||||
| 552 | - | |||||||||||||
| 553 | png_destroy_read_struct(&png_ptr, &info_ptr, &end_info); | - | ||||||||||||
| 554 | png_ptr = 0; | - | ||||||||||||
| 555 | amp.deallocate(); | - | ||||||||||||
| 556 | state = Ready; | - | ||||||||||||
| 557 | - | |||||||||||||
| 558 | if (scaledSize.isValid() && outImage->size() != scaledSize) | - | ||||||||||||
| 559 | *outImage = outImage->scaled(scaledSize, Qt::IgnoreAspectRatio, Qt::SmoothTransformation); | - | ||||||||||||
| 560 | - | |||||||||||||
| 561 | return true; | - | ||||||||||||
| 562 | } | - | ||||||||||||
| 563 | - | |||||||||||||
| 564 | QImage::Format QPngHandlerPrivate::readImageFormat() | - | ||||||||||||
| 565 | { | - | ||||||||||||
| 566 | QImage::Format format = QImage::Format_Invalid; | - | ||||||||||||
| 567 | png_uint_32 width, height; | - | ||||||||||||
| 568 | int bit_depth, color_type; | - | ||||||||||||
| 569 | png_colorp palette; | - | ||||||||||||
| 570 | int num_palette; | - | ||||||||||||
| 571 | png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type, 0, 0, 0); | - | ||||||||||||
| 572 | if (color_type == 0) { | - | ||||||||||||
| 573 | - | |||||||||||||
| 574 | if (bit_depth == 1 && png_get_channels(png_ptr, info_ptr) == 1) { | - | ||||||||||||
| 575 | format = QImage::Format_Mono; | - | ||||||||||||
| 576 | } else if (bit_depth == 16 && png_get_valid(png_ptr, info_ptr, 0x0010)) { | - | ||||||||||||
| 577 | format = QImage::Format_ARGB32; | - | ||||||||||||
| 578 | } else if (bit_depth == 8 && !png_get_valid(png_ptr, info_ptr, 0x0010)) { | - | ||||||||||||
| 579 | format = QImage::Format_Grayscale8; | - | ||||||||||||
| 580 | } else { | - | ||||||||||||
| 581 | format = QImage::Format_Indexed8; | - | ||||||||||||
| 582 | } | - | ||||||||||||
| 583 | } else if (color_type == (2 | 1) | - | ||||||||||||
| 584 | && png_get_PLTE(png_ptr, info_ptr, &palette, &num_palette) | - | ||||||||||||
| 585 | && num_palette <= 256) | - | ||||||||||||
| 586 | { | - | ||||||||||||
| 587 | - | |||||||||||||
| 588 | format = bit_depth == 1 ? QImage::Format_Mono : QImage::Format_Indexed8; | - | ||||||||||||
| 589 | } else { | - | ||||||||||||
| 590 | - | |||||||||||||
| 591 | format = QImage::Format_ARGB32; | - | ||||||||||||
| 592 | - | |||||||||||||
| 593 | if (!(color_type & 4) | - | ||||||||||||
| 594 | && !png_get_valid(png_ptr, info_ptr, 0x0010)) { | - | ||||||||||||
| 595 | - | |||||||||||||
| 596 | format = QImage::Format_RGB32; | - | ||||||||||||
| 597 | } | - | ||||||||||||
| 598 | } | - | ||||||||||||
| 599 | - | |||||||||||||
| 600 | return format; | - | ||||||||||||
| 601 | } | - | ||||||||||||
| 602 | - | |||||||||||||
| 603 | QPNGImageWriter::QPNGImageWriter(QIODevice* iod) : | - | ||||||||||||
| 604 | dev(iod), | - | ||||||||||||
| 605 | frames_written(0), | - | ||||||||||||
| 606 | disposal(Unspecified), | - | ||||||||||||
| 607 | looping(-1), | - | ||||||||||||
| 608 | ms_delay(-1), | - | ||||||||||||
| 609 | gamma(0.0) | - | ||||||||||||
| 610 | { | - | ||||||||||||
| 611 | } | - | ||||||||||||
| 612 | - | |||||||||||||
| 613 | QPNGImageWriter::~QPNGImageWriter() | - | ||||||||||||
| 614 | { | - | ||||||||||||
| 615 | } | - | ||||||||||||
| 616 | - | |||||||||||||
| 617 | void QPNGImageWriter::setDisposalMethod(DisposalMethod dm) | - | ||||||||||||
| 618 | { | - | ||||||||||||
| 619 | disposal = dm; | - | ||||||||||||
| 620 | } | - | ||||||||||||
| 621 | - | |||||||||||||
| 622 | void QPNGImageWriter::setLooping(int loops) | - | ||||||||||||
| 623 | { | - | ||||||||||||
| 624 | looping = loops; | - | ||||||||||||
| 625 | } | - | ||||||||||||
| 626 | - | |||||||||||||
| 627 | void QPNGImageWriter::setFrameDelay(int msecs) | - | ||||||||||||
| 628 | { | - | ||||||||||||
| 629 | ms_delay = msecs; | - | ||||||||||||
| 630 | } | - | ||||||||||||
| 631 | - | |||||||||||||
| 632 | void QPNGImageWriter::setGamma(float g) | - | ||||||||||||
| 633 | { | - | ||||||||||||
| 634 | gamma = g; | - | ||||||||||||
| 635 | } | - | ||||||||||||
| 636 | - | |||||||||||||
| 637 | - | |||||||||||||
| 638 | static void set_text(const QImage &image, png_structp png_ptr, png_infop info_ptr, | - | ||||||||||||
| 639 | const QString &description) | - | ||||||||||||
| 640 | { | - | ||||||||||||
| 641 | QMap<QString, QString> text; | - | ||||||||||||
| 642 | for (QForeachContainer<typename QtPrivate::remove_reference<decltype(image.textKeys())>::type> _container_((image.textKeys())); _container_.control && _container_.i != _container_.e; ++_container_.i, _container_.control ^= 1) for (const QString &key = *_container_.i; _container_.control; _container_.control = 0) { | - | ||||||||||||
| 643 |         if (!key.isEmpty()
  | 0 | ||||||||||||
| 644 |             text.insert(key, image.text(key)); never executed:  text.insert(key, image.text(key)); | 0 | ||||||||||||
| 645 |     } never executed:  end of block | 0 | ||||||||||||
| 646 | for (QForeachContainer<typename QtPrivate::remove_reference<decltype(description.split(QLatin1String("\n\n")))>::type> _container_((description.split(QLatin1String("\n\n")))); _container_.control && _container_.i != _container_.e; ++_container_.i, _container_.control ^= 1) for (const QString &pair = *_container_.i; _container_.control; _container_.control = 0) { | - | ||||||||||||
| 647 | int index = pair.indexOf(QLatin1Char(':')); | - | ||||||||||||
| 648 |         if (index >= 0
 
  | 0 | ||||||||||||
| 649 | QString s = pair.simplified(); | - | ||||||||||||
| 650 |             if (!s.isEmpty()
  | 0 | ||||||||||||
| 651 |                 text.insert(QLatin1String("Description"), s); never executed:  text.insert(QLatin1String("Description"), s); | 0 | ||||||||||||
| 652 |         } never executed:   else {end of block | 0 | ||||||||||||
| 653 | QString key = pair.left(index); | - | ||||||||||||
| 654 |             if (!key.simplified().isEmpty()
  | 0 | ||||||||||||
| 655 |                 text.insert(key, pair.mid(index + 2).simplified()); never executed:  text.insert(key, pair.mid(index + 2).simplified()); | 0 | ||||||||||||
| 656 |         } never executed:  end of block | 0 | ||||||||||||
| 657 | } | - | ||||||||||||
| 658 | - | |||||||||||||
| 659 |     if (text.isEmpty()
  | 0 | ||||||||||||
| 660 |         return; never executed:  return; | 0 | ||||||||||||
| 661 | - | |||||||||||||
| 662 | png_textp text_ptr = new png_text[text.size()]; | - | ||||||||||||
| 663 | memset(text_ptr, 0, text.size() * sizeof(png_text)); | - | ||||||||||||
| 664 | - | |||||||||||||
| 665 | QMap<QString, QString>::ConstIterator it = text.constBegin(); | - | ||||||||||||
| 666 | int i = 0; | - | ||||||||||||
| 667 |     while (it != text.constEnd()
  | 0 | ||||||||||||
| 668 | text_ptr[i].key = qstrdup(it.key().leftleftRef(79).toLatin1().constData()); | - | ||||||||||||
| 669 | bool noCompress = (it.value().length() < 40); | - | ||||||||||||
| 670 | { | - | ||||||||||||
| 671 |             text_ptr[i].compression = noCompress
  | 0 | ||||||||||||
| 672 | QByteArray value = it.value().toLatin1(); | - | ||||||||||||
| 673 | text_ptr[i].text = qstrdup(value.constData()); | - | ||||||||||||
| 674 | text_ptr[i].text_length = value.size(); | - | ||||||||||||
| 675 | } | - | ||||||||||||
| 676 | ++i; | - | ||||||||||||
| 677 | ++it; | - | ||||||||||||
| 678 |     } never executed:  end of block | 0 | ||||||||||||
| 679 | - | |||||||||||||
| 680 | png_set_text(png_ptr, info_ptr, text_ptr, i); | - | ||||||||||||
| 681 |     for (i = 0; i < text.size()
  | 0 | ||||||||||||
| 682 | delete [] text_ptr[i].key; | - | ||||||||||||
| 683 | delete [] text_ptr[i].text; | - | ||||||||||||
| 684 | - | |||||||||||||
| 685 | - | |||||||||||||
| 686 | - | |||||||||||||
| 687 |     } never executed:  end of block | 0 | ||||||||||||
| 688 | delete [] text_ptr; | - | ||||||||||||
| 689 | } never executed:  end of block | 0 | ||||||||||||
| 690 | - | |||||||||||||
| 691 | bool QPNGImageWriter::writeImage(const QImage& image, int off_x, int off_y) | - | ||||||||||||
| 692 | { | - | ||||||||||||
| 693 | return writeImage(image, -1, QString(), off_x, off_y); | - | ||||||||||||
| 694 | } | - | ||||||||||||
| 695 | - | |||||||||||||
| 696 | bool QPNGImageWriter::writeImage(const QImage& image, volatile int quality_in, const QString &description, | - | ||||||||||||
| 697 | int off_x_in, int off_y_in) | - | ||||||||||||
| 698 | { | - | ||||||||||||
| 699 | QPoint offset = image.offset(); | - | ||||||||||||
| 700 | int off_x = off_x_in + offset.x(); | - | ||||||||||||
| 701 | int off_y = off_y_in + offset.y(); | - | ||||||||||||
| 702 | - | |||||||||||||
| 703 | png_structp png_ptr; | - | ||||||||||||
| 704 | png_infop info_ptr; | - | ||||||||||||
| 705 | - | |||||||||||||
| 706 | png_ptr = png_create_write_struct("1.2.50",0,0,0); | - | ||||||||||||
| 707 | if (!png_ptr) { | - | ||||||||||||
| 708 | return false; | - | ||||||||||||
| 709 | } | - | ||||||||||||
| 710 | - | |||||||||||||
| 711 | png_set_error_fn(png_ptr, 0, 0, qt_png_warning); | - | ||||||||||||
| 712 | - | |||||||||||||
| 713 | info_ptr = png_create_info_struct(png_ptr); | - | ||||||||||||
| 714 | if (!info_ptr) { | - | ||||||||||||
| 715 | png_destroy_write_struct(&png_ptr, 0); | - | ||||||||||||
| 716 | return false; | - | ||||||||||||
| 717 | } | - | ||||||||||||
| 718 | - | |||||||||||||
| 719 | if (_setjmp (((png_ptr)->jmpbuf))) { | - | ||||||||||||
| 720 | png_destroy_write_struct(&png_ptr, &info_ptr); | - | ||||||||||||
| 721 | return false; | - | ||||||||||||
| 722 | } | - | ||||||||||||
| 723 | - | |||||||||||||
| 724 | int quality = quality_in; | - | ||||||||||||
| 725 | if (quality >= 0) { | - | ||||||||||||
| 726 | if (quality > 9) { | - | ||||||||||||
| 727 | QMessageLogger(__FILE__, 844, __PRETTY_FUNCTION__).warning("PNG: Quality %d out of range", quality); | - | ||||||||||||
| 728 | quality = 9; | - | ||||||||||||
| 729 | } | - | ||||||||||||
| 730 | png_set_compression_level(png_ptr, quality); | - | ||||||||||||
| 731 | } | - | ||||||||||||
| 732 | - | |||||||||||||
| 733 | png_set_write_fn(png_ptr, (void*)this, qpiw_write_fn, qpiw_flush_fn); | - | ||||||||||||
| 734 | - | |||||||||||||
| 735 | - | |||||||||||||
| 736 | int color_type = 0; | - | ||||||||||||
| 737 | if (image.colorCount()) { | - | ||||||||||||
| 738 | if (image.isGrayscale()) | - | ||||||||||||
| 739 | color_type = 0; | - | ||||||||||||
| 740 | else | - | ||||||||||||
| 741 | color_type = (2 | 1); | - | ||||||||||||
| 742 | } | - | ||||||||||||
| 743 | else if (image.format() == QImage::Format_Grayscale8) | - | ||||||||||||
| 744 | color_type = 0; | - | ||||||||||||
| 745 | else if (image.hasAlphaChannel()) | - | ||||||||||||
| 746 | color_type = (2 | 4); | - | ||||||||||||
| 747 | else | - | ||||||||||||
| 748 | color_type = (2); | - | ||||||||||||
| 749 | - | |||||||||||||
| 750 | png_set_IHDR(png_ptr, info_ptr, image.width(), image.height(), | - | ||||||||||||
| 751 | image.depth() == 1 ? 1 : 8, | - | ||||||||||||
| 752 | color_type, 0, 0, 0); | - | ||||||||||||
| 753 | - | |||||||||||||
| 754 | if (gamma != 0.0) { | - | ||||||||||||
| 755 | png_set_gAMA(png_ptr, info_ptr, 1.0/gamma); | - | ||||||||||||
| 756 | } | - | ||||||||||||
| 757 | - | |||||||||||||
| 758 | if (image.format() == QImage::Format_MonoLSB) | - | ||||||||||||
| 759 | png_set_packswap(png_ptr); | - | ||||||||||||
| 760 | - | |||||||||||||
| 761 | if (color_type == (2 | 1)) { | - | ||||||||||||
| 762 | - | |||||||||||||
| 763 | int num_palette = qMin(256, image.colorCount()); | - | ||||||||||||
| 764 | png_color palette[256]; | - | ||||||||||||
| 765 | png_byte trans[256]; | - | ||||||||||||
| 766 | int num_trans = 0; | - | ||||||||||||
| 767 | for (int i=0; i<num_palette; i++) { | - | ||||||||||||
| 768 | QRgb rgba=image.color(i); | - | ||||||||||||
| 769 | palette[i].red = qRed(rgba); | - | ||||||||||||
| 770 | palette[i].green = qGreen(rgba); | - | ||||||||||||
| 771 | palette[i].blue = qBlue(rgba); | - | ||||||||||||
| 772 | trans[i] = qAlpha(rgba); | - | ||||||||||||
| 773 | if (trans[i] < 255) { | - | ||||||||||||
| 774 | num_trans = i+1; | - | ||||||||||||
| 775 | } | - | ||||||||||||
| 776 | } | - | ||||||||||||
| 777 | png_set_PLTE(png_ptr, info_ptr, palette, num_palette); | - | ||||||||||||
| 778 | - | |||||||||||||
| 779 | if (num_trans) { | - | ||||||||||||
| 780 | png_set_tRNS(png_ptr, info_ptr, trans, num_trans, 0); | - | ||||||||||||
| 781 | } | - | ||||||||||||
| 782 | } | - | ||||||||||||
| 783 | - | |||||||||||||
| 784 | - | |||||||||||||
| 785 | - | |||||||||||||
| 786 | if (QSysInfo::ByteOrder == QSysInfo::BigEndian) { | - | ||||||||||||
| 787 | png_set_swap_alpha(png_ptr); | - | ||||||||||||
| 788 | } | - | ||||||||||||
| 789 | - | |||||||||||||
| 790 | - | |||||||||||||
| 791 | if (QSysInfo::ByteOrder == QSysInfo::LittleEndian | - | ||||||||||||
| 792 | && image.format() != QImage::Format_RGB888) { | - | ||||||||||||
| 793 | png_set_bgr(png_ptr); | - | ||||||||||||
| 794 | } | - | ||||||||||||
| 795 | - | |||||||||||||
| 796 | if (off_x || off_y) { | - | ||||||||||||
| 797 | png_set_oFFs(png_ptr, info_ptr, off_x, off_y, 0); | - | ||||||||||||
| 798 | } | - | ||||||||||||
| 799 | - | |||||||||||||
| 800 | if (frames_written > 0) | - | ||||||||||||
| 801 | png_set_sig_bytes(png_ptr, 8); | - | ||||||||||||
| 802 | - | |||||||||||||
| 803 | if (image.dotsPerMeterX() > 0 || image.dotsPerMeterY() > 0) { | - | ||||||||||||
| 804 | png_set_pHYs(png_ptr, info_ptr, | - | ||||||||||||
| 805 | image.dotsPerMeterX(), image.dotsPerMeterY(), | - | ||||||||||||
| 806 | 1); | - | ||||||||||||
| 807 | } | - | ||||||||||||
| 808 | - | |||||||||||||
| 809 | set_text(image, png_ptr, info_ptr, description); | - | ||||||||||||
| 810 | - | |||||||||||||
| 811 | png_write_info(png_ptr, info_ptr); | - | ||||||||||||
| 812 | - | |||||||||||||
| 813 | if (image.depth() != 1) | - | ||||||||||||
| 814 | png_set_packing(png_ptr); | - | ||||||||||||
| 815 | - | |||||||||||||
| 816 | if (color_type == (2) && image.format() != QImage::Format_RGB888) | - | ||||||||||||
| 817 | png_set_filler(png_ptr, 0, | - | ||||||||||||
| 818 | QSysInfo::ByteOrder == QSysInfo::BigEndian ? | - | ||||||||||||
| 819 | 0 : 1); | - | ||||||||||||
| 820 | - | |||||||||||||
| 821 | if (looping >= 0 && frames_written == 0) { | - | ||||||||||||
| 822 | uchar data[13] = "NETSCAPE2.0"; | - | ||||||||||||
| 823 | - | |||||||||||||
| 824 | data[0xB] = looping%0x100; | - | ||||||||||||
| 825 | data[0xC] = looping/0x100; | - | ||||||||||||
| 826 | png_write_chunk(png_ptr, const_cast<png_bytep>((const png_byte *)"gIFx"), data, 13); | - | ||||||||||||
| 827 | } | - | ||||||||||||
| 828 | if (ms_delay >= 0 || disposal!=Unspecified) { | - | ||||||||||||
| 829 | uchar data[4]; | - | ||||||||||||
| 830 | data[0] = disposal; | - | ||||||||||||
| 831 | data[1] = 0; | - | ||||||||||||
| 832 | data[2] = (ms_delay/10)/0x100; | - | ||||||||||||
| 833 | data[3] = (ms_delay/10)%0x100; | - | ||||||||||||
| 834 | png_write_chunk(png_ptr, const_cast<png_bytep>((const png_byte *)"gIFg"), data, 4); | - | ||||||||||||
| 835 | } | - | ||||||||||||
| 836 | - | |||||||||||||
| 837 | int height = image.height(); | - | ||||||||||||
| 838 | int width = image.width(); | - | ||||||||||||
| 839 | switch (image.format()) { | - | ||||||||||||
| 840 | case QImage::Format_Mono: | - | ||||||||||||
| 841 | case QImage::Format_MonoLSB: | - | ||||||||||||
| 842 | case QImage::Format_Indexed8: | - | ||||||||||||
| 843 | case QImage::Format_Grayscale8: | - | ||||||||||||
| 844 | case QImage::Format_RGB32: | - | ||||||||||||
| 845 | case QImage::Format_ARGB32: | - | ||||||||||||
| 846 | case QImage::Format_RGB888: | - | ||||||||||||
| 847 | { | - | ||||||||||||
| 848 | png_bytep* row_pointers = new png_bytep[height]; | - | ||||||||||||
| 849 | for (int y=0; y<height; y++) | - | ||||||||||||
| 850 | row_pointers[y] = const_cast<png_bytep>(image.constScanLine(y)); | - | ||||||||||||
| 851 | png_write_image(png_ptr, row_pointers); | - | ||||||||||||
| 852 | delete [] row_pointers; | - | ||||||||||||
| 853 | } | - | ||||||||||||
| 854 | break; | - | ||||||||||||
| 855 | default: | - | ||||||||||||
| 856 | { | - | ||||||||||||
| 857 | QImage::Format fmt = image.hasAlphaChannel() ? QImage::Format_ARGB32 : QImage::Format_RGB32; | - | ||||||||||||
| 858 | QImage row; | - | ||||||||||||
| 859 | png_bytep row_pointers[1]; | - | ||||||||||||
| 860 | for (int y=0; y<height; y++) { | - | ||||||||||||
| 861 | row = image.copy(0, y, width, 1).convertToFormat(fmt); | - | ||||||||||||
| 862 | row_pointers[0] = const_cast<png_bytep>(row.constScanLine(0)); | - | ||||||||||||
| 863 | png_write_rows(png_ptr, row_pointers, 1); | - | ||||||||||||
| 864 | } | - | ||||||||||||
| 865 | } | - | ||||||||||||
| 866 | break; | - | ||||||||||||
| 867 | } | - | ||||||||||||
| 868 | - | |||||||||||||
| 869 | png_write_end(png_ptr, info_ptr); | - | ||||||||||||
| 870 | frames_written++; | - | ||||||||||||
| 871 | - | |||||||||||||
| 872 | png_destroy_write_struct(&png_ptr, &info_ptr); | - | ||||||||||||
| 873 | - | |||||||||||||
| 874 | return true; | - | ||||||||||||
| 875 | } | - | ||||||||||||
| 876 | - | |||||||||||||
| 877 | static bool write_png_image(const QImage &image, QIODevice *device, | - | ||||||||||||
| 878 | int quality, float gamma, const QString &description) | - | ||||||||||||
| 879 | { | - | ||||||||||||
| 880 | QPNGImageWriter writer(device); | - | ||||||||||||
| 881 | if (quality >= 0) { | - | ||||||||||||
| 882 | quality = qMin(quality, 100); | - | ||||||||||||
| 883 | quality = (100-quality) * 9 / 91; | - | ||||||||||||
| 884 | } | - | ||||||||||||
| 885 | writer.setGamma(gamma); | - | ||||||||||||
| 886 | return writer.writeImage(image, quality, description); | - | ||||||||||||
| 887 | } | - | ||||||||||||
| 888 | - | |||||||||||||
| 889 | QPngHandler::QPngHandler() | - | ||||||||||||
| 890 | : d(new QPngHandlerPrivate(this)) | - | ||||||||||||
| 891 | { | - | ||||||||||||
| 892 | } | - | ||||||||||||
| 893 | - | |||||||||||||
| 894 | QPngHandler::~QPngHandler() | - | ||||||||||||
| 895 | { | - | ||||||||||||
| 896 | if (d->png_ptr) | - | ||||||||||||
| 897 | png_destroy_read_struct(&d->png_ptr, &d->info_ptr, &d->end_info); | - | ||||||||||||
| 898 | delete d; | - | ||||||||||||
| 899 | } | - | ||||||||||||
| 900 | - | |||||||||||||
| 901 | bool QPngHandler::canRead() const | - | ||||||||||||
| 902 | { | - | ||||||||||||
| 903 | if (d->state == QPngHandlerPrivate::Ready && !canRead(device())) | - | ||||||||||||
| 904 | return false; | - | ||||||||||||
| 905 | - | |||||||||||||
| 906 | if (d->state != QPngHandlerPrivate::Error) { | - | ||||||||||||
| 907 | setFormat("png"); | - | ||||||||||||
| 908 | return true; | - | ||||||||||||
| 909 | } | - | ||||||||||||
| 910 | - | |||||||||||||
| 911 | return false; | - | ||||||||||||
| 912 | } | - | ||||||||||||
| 913 | - | |||||||||||||
| 914 | bool QPngHandler::canRead(QIODevice *device) | - | ||||||||||||
| 915 | { | - | ||||||||||||
| 916 | if (!device) { | - | ||||||||||||
| 917 | QMessageLogger(__FILE__, 1034, __PRETTY_FUNCTION__).warning("QPngHandler::canRead() called with no device"); | - | ||||||||||||
| 918 | return false; | - | ||||||||||||
| 919 | } | - | ||||||||||||
| 920 | - | |||||||||||||
| 921 | return device->peek(8) == "\x89\x50\x4E\x47\x0D\x0A\x1A\x0A"; | - | ||||||||||||
| 922 | } | - | ||||||||||||
| 923 | - | |||||||||||||
| 924 | bool QPngHandler::read(QImage *image) | - | ||||||||||||
| 925 | { | - | ||||||||||||
| 926 | if (!canRead()) | - | ||||||||||||
| 927 | return false; | - | ||||||||||||
| 928 | return d->readPngImage(image); | - | ||||||||||||
| 929 | } | - | ||||||||||||
| 930 | - | |||||||||||||
| 931 | bool QPngHandler::write(const QImage &image) | - | ||||||||||||
| 932 | { | - | ||||||||||||
| 933 | return write_png_image(image, device(), d->quality, d->gamma, d->description); | - | ||||||||||||
| 934 | } | - | ||||||||||||
| 935 | - | |||||||||||||
| 936 | bool QPngHandler::supportsOption(ImageOption option) const | - | ||||||||||||
| 937 | { | - | ||||||||||||
| 938 | return option == Gamma | - | ||||||||||||
| 939 | || option == Description | - | ||||||||||||
| 940 | || option == ImageFormat | - | ||||||||||||
| 941 | || option == Quality | - | ||||||||||||
| 942 | || option == Size | - | ||||||||||||
| 943 | || option == ScaledSize; | - | ||||||||||||
| 944 | } | - | ||||||||||||
| 945 | - | |||||||||||||
| 946 | QVariant QPngHandler::option(ImageOption option) const | - | ||||||||||||
| 947 | { | - | ||||||||||||
| 948 | if (d->state == QPngHandlerPrivate::Error) | - | ||||||||||||
| 949 | return QVariant(); | - | ||||||||||||
| 950 | if (d->state == QPngHandlerPrivate::Ready && !d->readPngHeader()) | - | ||||||||||||
| 951 | return QVariant(); | - | ||||||||||||
| 952 | - | |||||||||||||
| 953 | if (option == Gamma) | - | ||||||||||||
| 954 | return d->gamma == 0.0 ? d->fileGamma : d->gamma; | - | ||||||||||||
| 955 | else if (option == Quality) | - | ||||||||||||
| 956 | return d->quality; | - | ||||||||||||
| 957 | else if (option == Description) | - | ||||||||||||
| 958 | return d->description; | - | ||||||||||||
| 959 | else if (option == Size) | - | ||||||||||||
| 960 | return QSize(png_get_image_width(d->png_ptr, d->info_ptr), | - | ||||||||||||
| 961 | png_get_image_height(d->png_ptr, d->info_ptr)); | - | ||||||||||||
| 962 | else if (option == ScaledSize) | - | ||||||||||||
| 963 | return d->scaledSize; | - | ||||||||||||
| 964 | else if (option == ImageFormat) | - | ||||||||||||
| 965 | return d->readImageFormat(); | - | ||||||||||||
| 966 | return QVariant(); | - | ||||||||||||
| 967 | } | - | ||||||||||||
| 968 | - | |||||||||||||
| 969 | void QPngHandler::setOption(ImageOption option, const QVariant &value) | - | ||||||||||||
| 970 | { | - | ||||||||||||
| 971 | if (option == Gamma) | - | ||||||||||||
| 972 | d->gamma = value.toFloat(); | - | ||||||||||||
| 973 | else if (option == Quality) | - | ||||||||||||
| 974 | d->quality = value.toInt(); | - | ||||||||||||
| 975 | else if (option == Description) | - | ||||||||||||
| 976 | d->description = value.toString(); | - | ||||||||||||
| 977 | else if (option == ScaledSize) | - | ||||||||||||
| 978 | d->scaledSize = value.toSize(); | - | ||||||||||||
| 979 | } | - | ||||||||||||
| 980 | - | |||||||||||||
| 981 | QByteArray QPngHandler::name() const | - | ||||||||||||
| 982 | { | - | ||||||||||||
| 983 | return "png"; | - | ||||||||||||
| 984 | } | - | ||||||||||||
| 985 | - | |||||||||||||
| 986 | - | |||||||||||||
| Switch to Source code | Preprocessed file |