WSL/SLF GitLab Repository

PNGIO.cc 28.5 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
/***********************************************************************************/
/*  Copyright 2009 WSL Institute for Snow and Avalanche Research    SLF-DAVOS      */
/***********************************************************************************/
/* This file is part of MeteoIO.
    MeteoIO is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    MeteoIO 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 Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with MeteoIO.  If not, see <http://www.gnu.org/licenses/>.
*/
#include "PNGIO.h"
#include <meteoio/ResamplingAlgorithms2D.h>
20
#include <meteoio/Graphics.h>
21
#include <meteoio/meteoLaws/Meteoconst.h>
22

23
#include <cstring>
24
#include <algorithm>
25
#include <errno.h>
26
#include <zlib.h>
27
28
29
30
31

using namespace std;

namespace mio {
/**
32
 * @page pngio PNGIO
33
 * @section pngio_format Format
34
35
36
37
 * This plugin write data to the Portable Network Graphics format (see https://secure.wikimedia.org/wikipedia/en/wiki/Portable_Network_Graphics).
 * No data read has been implemented, because reading an existing file would require the exact knowlege of the color gradient that has been used
 * to create it. When writing grids, various color gradients will be used depending on the parameter that the data represents. Nodata values
 * are represented by transparent pixels (transparency is acheived through a transparent color instead of a true alpha channel for size and performance).
38
 * If a grid containing no data (ie: size 0x0) is sent to the plugin, then no file will be written.
39
 * Finally, the naming scheme for meteo grids should be: YYYY-MM-DDTHH.mm_{MeteoGrids::Parameters}.png
40
 *
41
 * @section pngio_units Units
42
 * All units are MKSA except temperatures that are expressed in celcius.
43
 *
44
 * @section pngio_keywords Keywords
45
 * This plugin uses the following keywords:
46
47
 * - COORDSYS: input coordinate system (see Coords) specified in the [Output] section
 * - COORDPARAM: extra input coordinates parameters (see Coords) specified in the [Output] section
48
49
50
51
52
53
54
 * - GRID2DPATH: meteo grids directory where to read the grids; [Output] section
 * - PNG_LEGEND: plot legend on the side of the graph? (default: true)
 * - PNG_MIN_SIZE: guarantee that a 2D plot will have at least the given size
 * - PNG_MAX_SIZE: guarantee that a 2D plot will have at most the given size
 * - PNG_SCALING: scaling algorithm, either nearest or bilinear (default=bilinear)
 * - PNG_AUTOSCALE: autoscale for the color gradient? (default=true)
 * - PNG_WORLD_FILE: create world file with each file? (default=false)
55
56
 *
 * Advanced parameters (ie: don't mess up with them if you don't know what you're doing):
57
 * - PNG_INDEXED: create an indexed PNG? (default=true)
58
 * - PNG_NR_LEVELS: number of colors to use (less=smaller files, but it must be at least 5 and less than 255. default=30)
59
 * - PNG_SPEED_OPTIMIZE: optimize file creation for speed? (default=true, otherwise optimize for file size)
60
61
 *
 * The size are specified as width followed by height, with the separator being either a space, 'x' or '*'. If a minimum and a maximum size are given, the average of the smallest and largest permissible sizes will be used.
62
63
64
65
66
67
68
69
70
71
72
 * The world file is used for geolocalization and goes alongside the graphics output. By convention,
 * the file has the same name as the image file, with the third letter of the extension jammed with a w: tif->tfw, jpg->jqw.
 * The format is the following:
 * @code
 *    5.000000000000 (size of pixel in x direction)
 *    0.000000000000 (rotation term for row)
 *    0.000000000000 (rotation term for column)
 *    -5.000000000000 (size of pixel in y direction)
 *    492169.690845528910 (x coordinate of centre of upper left pixel in map units)
 *    5426523.318065105000 (y coordinate of centre of upper left pixel in map units)
 * @endcode
73
 *
74
 * @section pngio_example Example use
75
76
77
78
79
80
 * @code
 * GRID2D = PNG
 * png_legend = false
 * png_min_size = 400x400
 * png_max_size = 1366*768
 * @endcode
81
 *
82
 * @section pngio_compilation
83
84
85
86
87
88
89
90
91
92
93
94
 * In order to compile this plugin, you need libpng and zlib. For Linux, please select both the libraries and their development files in your package manager.
 *
 * For Windows, you can find zlib at http://switch.dl.sourceforge.net/project/gnuwin32/zlib/1.2.3/zlib-1.2.3.exe
 * and libpng at http://switch.dl.sourceforge.net/project/gnuwin32/libpng/1.2.37/libpng-1.2.37-setup.exe . Once this has been installed, if you plan on using
 * Visual c++, you also need to edit the file zconf.h in the libpng installation directory and transform the line 287:
 * @code
 * #if 0           // HAVE_UNISTD_H etc etc
 * @endcode
 * should become
 * @code
 * #if 1           // HAVE_UNISTD_H etc etc
 * @endcode
95
96
97
 */

const double PNGIO::plugin_nodata = -999.; //plugin specific nodata value. It can also be read by the plugin (depending on what is appropriate)
98
99
100
const unsigned char PNGIO::channel_depth = 8;
const unsigned char PNGIO::channel_max_color = 255;
const unsigned char PNGIO::transparent_grey = channel_max_color;
101

102
PNGIO::PNGIO(const std::string& configfile)
103
       : cfg(configfile),
104
105
106
107
108
         fp(NULL), autoscale(true), has_legend(true), has_world_file(false), optimize_for_speed(true),
         indexed_png(true), nr_levels(30),
         coordout(), coordoutparam(), grid2dpath(),
         scaling("bilinear"), min_w(IOUtils::unodata), min_h(IOUtils::unodata), max_w(IOUtils::unodata), max_h(IOUtils::unodata),
         metadata_key(), metadata_text()
109
{
110
	setOptions();
111
112
}

113
PNGIO::PNGIO(const Config& cfgreader)
114
       : cfg(cfgreader),
115
116
117
118
119
         fp(NULL), autoscale(true), has_legend(true), has_world_file(false), optimize_for_speed(true),
         indexed_png(true), nr_levels(30),
         coordout(), coordoutparam(), grid2dpath(),
         scaling("bilinear"), min_w(IOUtils::unodata), min_h(IOUtils::unodata), max_w(IOUtils::unodata), max_h(IOUtils::unodata),
         metadata_key(), metadata_text()
120
{
121
122
123
	setOptions();
}

124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
PNGIO& PNGIO::operator=(const PNGIO& source) {
	if(this != &source) {
		fp = NULL;
		autoscale = source.autoscale;
		has_legend = source.has_legend;
		has_world_file = source.has_world_file;
		optimize_for_speed = source.optimize_for_speed;
		indexed_png = source.indexed_png;
		nr_levels = source.nr_levels;
		coordout = source.coordout;
		coordoutparam = source.coordoutparam;
		grid2dpath = source.grid2dpath;
		scaling = source.scaling;
		min_w = source.min_w;
		min_h = source.min_h;
		max_w = source.max_w;
		max_h = source.max_h;
		metadata_key = source.metadata_key;
		metadata_text = source.metadata_text;
	}
	return *this;
}

147
148
void PNGIO::setOptions()
{
149
	//default values have been set by the constructors
150
	cfg.getValue("COORDSYS", "Output", coordout);
151
	cfg.getValue("COORDPARAM", "Output", coordoutparam, IOUtils::nothrow);
152
	cfg.getValue("GRID2DPATH", "Output", grid2dpath);
153
	//cfg.getValue("TIME_ZONE", "Output", tz_out, IOUtils::nothrow);
154

155
156
	//get size specifications
	std::string min_size, max_size;
157
	cfg.getValue("PNG_MIN_SIZE", "Output", min_size, IOUtils::nothrow);
158
	if(!min_size.empty()) parse_size(min_size, min_w, min_h);
159
	cfg.getValue("PNG_MAX_SIZE", "Output", max_size, IOUtils::nothrow);
160
	if(!max_size.empty()) parse_size(max_size, max_w, max_h);
161

162
163
164
165
	cfg.getValue("PNG_AUTOSCALE", "Output", autoscale, IOUtils::nothrow);
	cfg.getValue("PNG_LEGEND", "Output", has_legend, IOUtils::nothrow);
	cfg.getValue("PNG_SCALING", "Output", scaling, IOUtils::nothrow);
	cfg.getValue("PNG_WORLD_FILE", "Output", has_world_file, IOUtils::nothrow);
166
167
168
169
170

	if(has_legend) { //we need to save room for the legend
		if(min_w!=IOUtils::unodata) min_w -= legend::getLegendWidth();
		if(max_w!=IOUtils::unodata) max_w -= legend::getLegendWidth();
	}
171

172
173
	cfg.getValue("PNG_INDEXED", "Output", indexed_png, IOUtils::nothrow);
	cfg.getValue("PNG_SPEED_OPTIMIZE", "Output", optimize_for_speed, IOUtils::nothrow);
174
	unsigned int tmp=IOUtils::unodata;
175
	cfg.getValue("PNG_NR_LEVELS", "Output", tmp, IOUtils::nothrow);
176
	if(tmp!=IOUtils::unodata && (tmp>255 || tmp<5)) {
177
178
		throw InvalidFormatException("PNG_NR_LEVELS must be between 5 and 255!", AT);
	}
179
	if(tmp!=IOUtils::unodata) nr_levels=static_cast<unsigned char>(tmp);
180
181
}

182
void PNGIO::parse_size(const std::string& size_spec, size_t& width, size_t& height)
183
184
{
	char rest[32] = "";
185
186
187
188
	unsigned int w,h;
	if(sscanf(size_spec.c_str(), "%u %u%31s", &w, &h, rest) < 2)
	if(sscanf(size_spec.c_str(), "%u*%u%31s", &w, &h, rest) < 2)
	if(sscanf(size_spec.c_str(), "%ux%u%31s", &w, &h, rest) < 2) {
189
		std::ostringstream ss;
190
191
192
		ss << "Can not parse PNGIO size specification \"" << size_spec << "\"";
		throw InvalidFormatException(ss.str(), AT);
	}
193
194
195
196

	width = static_cast<size_t>(w);
	height = static_cast<size_t>(h);

197
198
199
	std::string tmp(rest);
	IOUtils::trim(tmp);
	if ((tmp.length() > 0) && tmp[0] != '#' && tmp[0] != ';') {//if line holds more than one value it's invalid
200
		std::ostringstream ss;
201
202
203
204
205
		ss << "Invalid PNGIO size specification \"" << size_spec << "\"";
		throw InvalidFormatException(ss.str(), AT);
	}
}

206
double PNGIO::getScaleFactor(const size_t& grid_w, const size_t& grid_h)
207
208
209
210
211
212
213
214
215
216
217
{
	if(grid_w==0 || grid_h==0) {
		return 1.;
	}

	double min_factor = IOUtils::nodata;
	if(min_w!=IOUtils::unodata) { //min_w & min_w are read together
		const double min_w_factor = (double)min_w / (double)grid_w;
		const double min_h_factor = (double)min_h / (double)grid_h;
		min_factor = std::max(min_w_factor, min_h_factor);
	}
218

219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
	double max_factor = IOUtils::nodata;
	if(max_w!=IOUtils::unodata) { //max_w & max_h are read together
		const double max_w_factor = (double)max_w / (double)grid_w;
		const double max_h_factor = (double)max_h / (double)grid_h;
		max_factor = std::min(max_w_factor, max_h_factor);
	}

	if(min_factor==IOUtils::nodata && max_factor==IOUtils::nodata)
		return 1.; //no user given specification
	if(min_factor!=IOUtils::nodata && max_factor!=IOUtils::nodata)
		return (min_factor+max_factor)/2.; //both min & max -> average

	//only one size specification provided -> return its matching factor
	if(min_factor!=IOUtils::nodata)
		return min_factor;
	else
		return max_factor;
236
237
}

238
PNGIO::~PNGIO() throw() {
239
	if(fp!=NULL) fclose(fp); fp=NULL;
240
241
}

242
243
244
245
246
247
248
void PNGIO::read2DGrid(Grid2DObject&, const std::string&)
{
	//Nothing so far
	throw IOException("Nothing implemented here", AT);
}

void PNGIO::read2DGrid(Grid2DObject&, const MeteoGrids::Parameters& , const Date&)
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
{
	//Nothing so far
	throw IOException("Nothing implemented here", AT);
}

void PNGIO::readDEM(DEMObject& /*dem_out*/)
{
	//Nothing so far
	throw IOException("Nothing implemented here", AT);
}

void PNGIO::readLanduse(Grid2DObject& /*landuse_out*/)
{
	//Nothing so far
	throw IOException("Nothing implemented here", AT);
}

void PNGIO::readAssimilationData(const Date& /*date_in*/, Grid2DObject& /*da_out*/)
{
	//Nothing so far
	throw IOException("Nothing implemented here", AT);
}

void PNGIO::readStationData(const Date&, std::vector<StationData>& /*vecStation*/)
{
	//Nothing so far
	throw IOException("Nothing implemented here", AT);
}

void PNGIO::readMeteoData(const Date& /*dateStart*/, const Date& /*dateEnd*/,
                             std::vector< std::vector<MeteoData> >& /*vecMeteo*/,
                             const size_t&)
{
	//Nothing so far
	throw IOException("Nothing implemented here", AT);
}

void PNGIO::writeMeteoData(const std::vector< std::vector<MeteoData> >& /*vecMeteo*/,
                              const std::string&)
{
	//Nothing so far
	throw IOException("Nothing implemented here", AT);
}

293
void PNGIO::readPOI(std::vector<Coords>&)
294
295
296
297
298
{
	//Nothing so far
	throw IOException("Nothing implemented here", AT);
}

299
300
Grid2DObject PNGIO::scaleGrid(const Grid2DObject& grid_in)
{ //scale input image
301
	const double factor = getScaleFactor(grid_in.getNx(), grid_in.getNy());
302
	if(scaling=="nearest")
303
		return ResamplingAlgorithms2D::NearestNeighbour(grid_in, factor);
304
	else if(scaling=="bilinear")
305
		return ResamplingAlgorithms2D::BilinearResampling(grid_in, factor);
306
	else {
307
		ostringstream ss;
308
309
310
		ss << "Grid scaling algorithm \"" << scaling << "\" unknown";
		throw UnknownValueException(ss.str(), AT);
	}
311
}
312

313
void PNGIO::setFile(const std::string& filename, png_structp& png_ptr, png_infop& info_ptr, const size_t &width, const size_t &height)
314
{
315
316
317
318
	// Open file for writing (binary mode)
	if (!IOUtils::validFileName(filename)) {
		throw InvalidFileNameException(filename, AT);
	}
319
	errno=0;
320
321
	fp = fopen(filename.c_str(), "wb");
	if (fp == NULL) {
322
		ostringstream ss;
323
		ss << "Error opening file \"" << filename << "\", possible reason: " << strerror(errno);
324
		throw FileAccessException(ss.str(), AT);
325
326
327
328
329
	}

	// Initialize write structure
	png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
	if (png_ptr == NULL) {
330
		fclose(fp); fp=NULL;
331
		throw IOException("Could not allocate write structure. Are you running with the same version of libpng that you linked to?", AT);
332
333
334
335
336
	}

	// Initialize info structure
	info_ptr = png_create_info_struct(png_ptr);
	if (info_ptr == NULL) {
337
		fclose(fp); fp=NULL;
338
		png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
339
		free(png_ptr);
340
341
342
343
		throw IOException("Could not allocate info structure", AT);
	}

	// Setup Exception handling
344
#ifdef _MSC_VER
345
346
	#pragma warning(disable:4611) //the setjmp of libpng has been set up so that it can safely be called from c++
#endif
347
	if (setjmp(png_jmpbuf(png_ptr))) {
348
		closePNG(png_ptr, info_ptr, NULL);
349
		throw IOException("Error during png creation. Can not set jump pointer (I have no clue what it means too!)", AT);
350
351
352
353
	}

	png_init_io(png_ptr, fp);

354
	if(optimize_for_speed) png_set_compression_level(png_ptr, Z_BEST_SPEED);
355
356
	else png_set_compression_level(png_ptr, Z_BEST_COMPRESSION);

357
358
	png_set_filter(png_ptr, PNG_FILTER_TYPE_BASE, PNG_FILTER_SUB|PNG_FILTER_UP); //any other filter is costly and brings close to nothing...
	if(indexed_png) png_set_compression_strategy(png_ptr, Z_RLE); //Z_DEFAULT_STRATEGY, Z_FILTERED, Z_HUFFMAN_ONLY, Z_RLE
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376

	// Write header (8 bit colour depth). Full alpha channel with PNG_COLOR_TYPE_RGB_ALPHA
	if(indexed_png) {
		png_set_IHDR(png_ptr, info_ptr, width, height,
			channel_depth, PNG_COLOR_TYPE_PALETTE, PNG_INTERLACE_NONE,
			PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
		//set transparent color (ie: cheap transparency: leads to smaller files and shorter run times)
		png_byte trans = 0; //by convention, the gradient define it as color 0
		png_set_tRNS(png_ptr, info_ptr, &trans, 1, 0);
	} else {
		png_set_IHDR(png_ptr, info_ptr, width, height,
			channel_depth, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,
			PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
		//set transparent color (ie: cheap transparency: leads to smaller files and shorter run times)
		png_color_16 trans_rgb_value = {transparent_grey, transparent_grey, transparent_grey, transparent_grey, transparent_grey};
		png_set_tRNS(png_ptr, info_ptr, 0, 0, &trans_rgb_value);
	}

377
378
379
	//set background color to help applications show the picture when no background is present
	png_color_16 background = {channel_max_color, channel_max_color, channel_max_color, channel_max_color, channel_max_color};
	png_set_background(png_ptr, &background, PNG_BACKGROUND_GAMMA_SCREEN, true, 1.0);
380
381
}

382
size_t PNGIO::setLegend(const size_t &ncols, const size_t &nrows, const double &min, const double &max, Array2D<double> &legend_array)
383
{
384
	if(has_legend) {
385
		const legend leg(static_cast<unsigned int>(nrows), min, max);
386
		legend_array = leg.getLegend();
387
		const size_t nx = legend_array.getNx();
388
389
390
		return (ncols+nx);
	} else {
		return ncols;
391
	}
392
393
}

394
void PNGIO::writeDataSection(const Grid2DObject& grid, const Array2D<double>& legend_array, const Gradient& gradient, const size_t& full_width, const png_structp& png_ptr, png_infop& info_ptr)
395
{
396
397
	const size_t ncols = grid.getNx();
	const size_t nrows = grid.getNy();
398

399
	// Allocate memory for one row (3 bytes per pixel - RGB)
400
	const unsigned char channels = (indexed_png)? 1 : 3; //4 for rgba
401
402
	png_bytep row = (png_bytep)calloc(channels*sizeof(png_byte), full_width);
	if(row==NULL) {
403
404
405
		fclose(fp); fp=NULL;
		png_free_data(png_ptr, info_ptr, PNG_FREE_ALL, -1);
		free(png_ptr);
406
		throw IOException("Can not allocate row memory for PNG!", AT);
407
	}
408
409

	// Write image data
410
	if(indexed_png) {
411
		for(size_t y=(nrows-1) ; y-- > 0; ) {
412
			size_t x=0;
413
			for(; x<ncols ; x++) {
414
				const size_t i=x*channels;
415
				unsigned char index;
416
				gradient.getColor(grid(x,y), index);
417
				row[i]=static_cast<png_byte>(index);
418
419
			}
			for(; x<full_width; x++) {
420
				const size_t i=x*channels;
421
				unsigned char index;
422
				gradient.getColor(legend_array(x-ncols,y), index);
423
				row[i]=static_cast<png_byte>(index);
424
			}
425
			png_write_row(png_ptr, row);
426
		}
427
	} else {
428
		for(size_t y=(nrows-1) ; y -- > 0; ) {
429
			size_t x=0;
430
			for(; x<ncols ; x++) {
431
				const size_t i=x*channels;
432
433
434
435
				unsigned char r,g,b;
				bool a;
				gradient.getColor(grid(x,y), r,g,b,a);
				if(a==true) {
436
					row[i]=static_cast<png_byte>(transparent_grey); row[i+1]=static_cast<png_byte>(transparent_grey); row[i+2]=static_cast<png_byte>(transparent_grey);
437
				} else {
438
					row[i]=static_cast<png_byte>(r); row[i+1]=static_cast<png_byte>(g); row[i+2]=static_cast<png_byte>(b);
439
440
441
				}
			}
			for(; x<full_width; x++) {
442
				const size_t i=x*channels;
443
444
445
446
				unsigned char r,g,b;
				bool a;
				gradient.getColor(legend_array(x-ncols,y), r,g,b,a);
				if(a==true) {
447
					row[i]=static_cast<png_byte>(transparent_grey); row[i+1]=static_cast<png_byte>(transparent_grey); row[i+2]=static_cast<png_byte>(transparent_grey);
448
				} else {
449
					row[i]=static_cast<png_byte>(r); row[i+1]=static_cast<png_byte>(g); row[i+2]=static_cast<png_byte>(b);
450
				}
451
			}
452
			png_write_row(png_ptr, row);
453
454
		}
	}
455

456
457
	png_write_flush(png_ptr);
	png_free(png_ptr, row);
458
459
}

460
void PNGIO::setPalette(const Gradient &gradient, png_structp& png_ptr, png_infop& info_ptr, png_color *palette)
461
{
462
463
464
	std::vector<unsigned char> pal;
	size_t nr_colors;
	gradient.getPalette(pal, nr_colors);
465
	palette = (png_color*)calloc(sizeof (png_color), nr_colors); //ie: three png_bytes, each being an unsigned char
466
	for(size_t ii=0; ii<nr_colors; ii++) {
467
468
469
470
		const size_t interlace = ii*3; //colors from Gradient interlaced
		palette[ii].red = static_cast<png_byte>(pal[interlace]);
		palette[ii].green = static_cast<png_byte>(pal[interlace+1]);
		palette[ii].blue = static_cast<png_byte>(pal[interlace+2]);
471
	}
472
	png_set_PLTE(png_ptr, info_ptr, palette, static_cast<int>(nr_colors));
473
474
}

475
void PNGIO::closePNG(png_structp& png_ptr, png_infop& info_ptr, png_color *palette)
476
{
477
	png_free_data(png_ptr, info_ptr, PNG_FREE_ALL, -1);
478
	if(indexed_png && palette!=NULL) free(palette);
479
	png_destroy_write_struct(&png_ptr, &info_ptr);
480
	fclose(fp); fp=NULL;
481
482
	free(info_ptr);
	free(png_ptr);
483
484
}

485
void PNGIO::write2DGrid(const Grid2DObject& grid_in, const std::string& filename)
486
{
487
	const string full_name = grid2dpath+"/"+filename;
488
	fp=NULL;
489
	png_color *palette=NULL;
490
491
492
493
494
	png_structp png_ptr=NULL;
	png_infop info_ptr=NULL;

	//scale input image
	const Grid2DObject grid = scaleGrid(grid_in);
495
	const size_t ncols = grid.getNx(), nrows = grid.getNy();
496
497
	if(ncols==0 || nrows==0) return;

498
499
500
501
	const double min = grid.grid2D.getMin();
	const double max = grid.grid2D.getMax();

	Gradient gradient(Gradient::heat, min, max, autoscale);
502
	if(indexed_png) gradient.setNrOfLevels(nr_levels);
503
504

	Array2D<double> legend_array; //it will remain empty if there is no legend
505
	const size_t full_width = setLegend(ncols, nrows, min, max, legend_array);
506

507
	setFile(full_name, png_ptr, info_ptr, full_width, nrows);
508
	if(indexed_png) setPalette(gradient, png_ptr, info_ptr, palette);
509
	if(has_world_file) writeWorldFile(grid, full_name);
510
511
512
513
514
515

	createMetadata(grid);
	metadata_key.push_back("Title"); //adding generic title
	metadata_text.push_back("Unknown Gridded data");
	writeMetadata(png_ptr, info_ptr);

516
	writeDataSection(grid, legend_array, gradient, full_width, png_ptr, info_ptr);
517
	png_write_end(png_ptr, NULL);
518

519
	closePNG(png_ptr, info_ptr, palette);
520
521
}

522
void PNGIO::write2DGrid(const Grid2DObject& grid_in, const MeteoGrids::Parameters& parameter, const Date& date)
523
{
524
525
	std::string filename;
	if(parameter==MeteoGrids::DEM || parameter==MeteoGrids::SLOPE || parameter==MeteoGrids::AZI)
526
		filename = grid2dpath + "/" + MeteoGrids::getParameterName(parameter) + ".png";
527
528
529
530
531
	else {
		std::string date_str = date.toString(Date::ISO);
		std::replace( date_str.begin(), date_str.end(), ':', '.');
		filename = grid2dpath + "/" + date_str + "_" + MeteoGrids::getParameterName(parameter) + ".png";
	}
532

533
	fp=NULL;
534
	png_color *palette=NULL;
535
536
537
538
539
	png_structp png_ptr=NULL;
	png_infop info_ptr=NULL;

	//scale input image
	Grid2DObject grid = scaleGrid(grid_in);
540
	const size_t ncols = grid.getNx(), nrows = grid.getNy();
541
542
	if(ncols==0 || nrows==0) return;

543
544
545
546
547
	double min = grid.grid2D.getMin();
	double max = grid.grid2D.getMax();

	Gradient gradient;
	if(parameter==MeteoGrids::DEM) {
548
549
		if(!autoscale) {
			min = 0.; //we want a 3000 snow line with a full scale legend
550
551
			max = 3500.;
			gradient.set(Gradient::terrain, min, max, autoscale); //max used as snow line reference
552
553
		} else
			gradient.set(Gradient::terrain, min, max, autoscale);
554
555
	} else if(parameter==MeteoGrids::SLOPE) {
		gradient.set(Gradient::slope, min, max, autoscale);
556
557
558
559
560
	} else if(parameter==MeteoGrids::SHADE) {
		if(!autoscale) {
			min = 0.; max = 1.;
		}
		gradient.set(Gradient::blktowhite, min, max, autoscale);
561
	} else if(parameter==MeteoGrids::AZI) {
562
563
564
565
		if(!autoscale) {
			min = 0.;
			max = 360.;
		}
566
		gradient.set(Gradient::azi, min, max, autoscale);
567
568
569
570
571
572
	} else if(parameter==MeteoGrids::DW) {
		if(!autoscale) {
			min = 0.;
			max = 360.;
		}
		gradient.set(Gradient::azi, min, max, autoscale);
573
	} else if(parameter==MeteoGrids::HS) {
574
		if(!autoscale) {
575
			min = 0.; max = 2.5;
576
577
		}
		gradient.set(Gradient::blue, min, max, autoscale);
578
579
	} else if(parameter==MeteoGrids::TA) {
		grid.grid2D -= Cst::t_water_freezing_pt; //convert to celsius
580
		if(!autoscale) {
581
			min = -15.; max = 15.;
582
583
584
585
		} else {
			min -= Cst::t_water_freezing_pt;
			max -= Cst::t_water_freezing_pt;
		}
586
		gradient.set(Gradient::heat, min, max, autoscale);
587
588
589
	} else if(parameter==MeteoGrids::TSS) {
		grid.grid2D -= Cst::t_water_freezing_pt; //convert to celsius
		if(!autoscale) {
590
			min = -20.; max = 5.;
591
592
593
594
595
		} else {
			min -= Cst::t_water_freezing_pt;
			max -= Cst::t_water_freezing_pt;
		}
		gradient.set(Gradient::freeze, min, max, autoscale);
596
	} else if(parameter==MeteoGrids::RH) {
597
598
599
		if(!autoscale) {
			min = 0.; max = 1.;
		}
600
		gradient.set(Gradient::bg_isomorphic, min, max, autoscale);
601
602
603
604
605
606
607
608
609
610
611
	} else if(parameter==MeteoGrids::P) {
		if(!autoscale) {
			//lowest and highest pressures ever recorded on Earth: 87000 and 108570
			min = 87000.; max = 115650.; //centered around 1 atm
			gradient.set(Gradient::bluewhitered, min, max, autoscale);
		} else {
			const double delta1 = fabs(Cst::std_press-min);
			const double delta2 = fabs(max - Cst::std_press);
			const double delta = (delta1>delta2)?delta1:delta2;
			gradient.set(Gradient::bluewhitered, Cst::std_press-delta, Cst::std_press+delta, autoscale);
		}
612
613
614
615
	} else if(parameter==MeteoGrids::ALB) {
		if(!autoscale) {
			min = 0.; max = 1.;
		}
616
		gradient.set(Gradient::blktowhite, min, max, autoscale);
617
	} else if(parameter==MeteoGrids::TAU_CLD) {
618
619
620
621
		if(!autoscale) {
			min = 0.; max = 1.;
		}
		gradient.set(Gradient::blktowhite, min, max, autoscale);
622
	} else if(parameter==MeteoGrids::ISWR) {
623
		if(!autoscale) {
624
			min = 0.; max = 800.;
625
626
627
628
629
630
631
		}
		gradient.set(Gradient::heat, min, max, autoscale);
	} else if(parameter==MeteoGrids::ILWR) {
		if(!autoscale) {
			min = 200.; max = 500.;
		}
		gradient.set(Gradient::heat, min, max, autoscale);
632
	} else if(parameter==MeteoGrids::SWE) {
633
		if(!autoscale) {
634
			min = 0.; max = 250.;
635
		}
636
		gradient.set(Gradient::blue_pink, min, max, autoscale);
637
638
639
	} else {
		gradient.set(Gradient::heat, min, max, autoscale);
	}
640
	gradient.setNrOfLevels(nr_levels);
641
642

	Array2D<double> legend_array; //it will remain empty if there is no legend
643
	const size_t full_width = setLegend(ncols, nrows, min, max, legend_array);
644

645
	setFile(filename, png_ptr, info_ptr, full_width, nrows);
646
	if(indexed_png) setPalette(gradient, png_ptr, info_ptr, palette);
647
	if(has_world_file) writeWorldFile(grid, filename);
648
649
650
651

	createMetadata(grid);
	metadata_key.push_back("Title"); //adding title
	metadata_text.push_back( MeteoGrids::getParameterName(parameter)+" on "+date.toString(Date::ISO) );
652
653
	metadata_key.push_back("Simulation Date");
	metadata_text.push_back( date.toString(Date::ISO) );
654
655
	metadata_key.push_back("Simulation Parameter");
	metadata_text.push_back( MeteoGrids::getParameterName(parameter) );
656
657
	writeMetadata(png_ptr, info_ptr);

658
	writeDataSection(grid, legend_array, gradient, full_width, png_ptr, info_ptr);
659
	png_write_end(png_ptr, NULL);
660

661
	closePNG(png_ptr, info_ptr, palette);
662
663
664
665
666
667
}

void PNGIO::writeWorldFile(const Grid2DObject& grid_in, const std::string& filename)
{
	const string world_file = IOUtils::removeExtension(filename)+".pnw";
	const double cellsize = grid_in.cellsize;
668
	Coords world_ref = grid_in.llcorner;
669
	world_ref.setProj(coordout, coordoutparam);
670
	world_ref.moveByXY(.5*cellsize, (double(grid_in.getNy())+.5)*cellsize); //moving to center of upper left cell
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690

	std::ofstream fout;
	fout.open(world_file.c_str());
	if (fout.fail()) {
		throw FileAccessException(world_file, AT);
	}

	try {
		fout << std::setprecision(12) << cellsize << "\n";
		fout << "0.000000000000\n";
		fout << "0.000000000000\n";
		fout << std::setprecision(12) << -cellsize << "\n";
		fout << std::setprecision(12) << world_ref.getEasting() << "\n";
		fout << std::setprecision(12) << world_ref.getNorthing() << "\n";
	} catch(...) {
		fout.close();
		throw FileAccessException("Failed when writing to PNG world file \""+world_file+"\"", AT);
	}

	fout.close();
691
692
}

693
void PNGIO::createMetadata(const Grid2DObject& grid)
694
{
695
696
	const double lat = grid.llcorner.getLat();
	const double lon = grid.llcorner.getLon();
697
	ostringstream ss;
698

699
700
701
	metadata_key.clear();
	metadata_text.clear();

702
703
704
705
	metadata_key.push_back("Creation Time");
	Date cr_date;
	cr_date.setFromSys();
	metadata_text.push_back( cr_date.toString(Date::ISO) );
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
	metadata_key.push_back("Author");
	metadata_text.push_back(IOUtils::getLogName());
	metadata_key.push_back("Software");
	metadata_text.push_back("MeteoIO "+getLibVersion());
	metadata_key.push_back("Position");
	metadata_text.push_back("llcorner");
	metadata_key.push_back("Cellsize");
	ss.str(""); ss << fixed << setprecision(2) << grid.cellsize;
	metadata_text.push_back(ss.str());
	metadata_key.push_back("Latitude");
	ss.str(""); ss << fixed << setprecision(6) << lat;
	metadata_text.push_back(ss.str());
	metadata_key.push_back("Longitude");
	ss.str(""); ss << fixed << setprecision(6) << lon;
	metadata_text.push_back(ss.str());
721

722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
	if(lat<0.) {
		metadata_key.push_back("LatitudeRef");
		metadata_text.push_back("S");
		metadata_key.push_back("GPSLatitude");
		metadata_text.push_back(decimal_to_dms(-lat));
	} else {
		metadata_key.push_back("LatitudeRef");
		metadata_text.push_back("N");
		metadata_key.push_back("GPSLatitude");
		metadata_text.push_back(decimal_to_dms(lat));
	}
	if(lon<0.) {
		metadata_key.push_back("LongitudeRef");
		metadata_text.push_back("W");
		metadata_key.push_back("GPSLongitude");
		metadata_text.push_back(decimal_to_dms(-lon));
	} else {
		metadata_key.push_back("LongitudeRef");
		metadata_text.push_back("E");
		metadata_key.push_back("GPSLongitude");
		metadata_text.push_back(decimal_to_dms(lon));
	}
}

void PNGIO::writeMetadata(png_structp &png_ptr, png_infop &info_ptr)
{
748
	const size_t max_len = 79; //according to the official specs' recommendation
749
	const size_t nr = metadata_key.size();
750
751
752
	png_text *info_text = (png_text *)calloc(sizeof(png_text), nr);
	char **key = (char**)calloc(sizeof(char)*max_len, nr);
	char **text = (char**)calloc(sizeof(char)*max_len, nr);
753
754

	for(size_t ii=0; ii<nr; ii++) {
755
756
757
758
		key[ii] = (char *)calloc(sizeof(char), max_len);
		text[ii] = (char *)calloc(sizeof(char), max_len);
		strncpy(key[ii], metadata_key[ii].c_str(), max_len);
		strncpy(text[ii], metadata_text[ii].c_str(), max_len);
759
760
761
762
		info_text[ii].key = key[ii];
		info_text[ii].text = text[ii];
		info_text[ii].compression = PNG_TEXT_COMPRESSION_NONE;
	}
763

764
	png_set_text(png_ptr, info_ptr, info_text, static_cast<int>(nr));
765
766
	png_write_info(png_ptr, info_ptr);

767
768
769
770
771
772
773
774
775
776
777
778
779
780
	free(info_text);
	for(size_t ii=0; ii<nr; ii++) {
		free(key[ii]);
		free(text[ii]);
	}
	free(key);
	free(text);
}

std::string PNGIO::decimal_to_dms(const double& decimal) {
	const int d = static_cast<int>( floor(decimal) );
	const double m = floor( ((decimal - (double)d)*60.)*100. ) / 100.;
	const double s = 3600.*(decimal - (double)d) - 60.*m;

781
	std::ostringstream dms;
782
783
	dms << d << "/1 " << static_cast<int>(m*100) << "/100 " << fixed << setprecision(6) << s << "/1";
	return dms.str();
784
785
786
}

} //namespace