vectorix/pnm_handler.h at master · hadrava/vectorix · GitHub

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
/*
 * Vectorix -- line-based image vectorizer
 * (c) 2016 Jan Hadrava <had@atrey.karlin.mff.cuni.cz>
 */
#ifndef VECTORIX__PNM_HANDLER_H
#define VECTORIX__PNM_HANDLER_H

// Manipulation with Netpbm format images

#include <cstdio>
#include <cstring>
#include "logger.h"
#include "config.h"
#include "parameters.h"

namespace vectorix {

typedef unsigned char pnm_data_t; // PNM image data type for one pixel (1 channel)

enum class pnm_variant_type {
	ascii_pbm  = 1, // ASCII bitmap
	ascii_pgm  = 2, // ASCII grayscale image
	ascii_ppm  = 3, // ASCII 3channel image
	binary_pbm = 4, // binary bitmap
	binary_pgm = 5, // binary grayscale image
	binary_ppm = 6  // binary 3channel image
};

class pnm_image {
public:
	pnm_image(parameters &params): width(0), height(0), type(pnm_variant_type::ascii_pbm), maxvalue(1), data(NULL), par(&params) {
		int *param_pnm_verbosity;
		par->bind_param(param_pnm_verbosity, "pnm_verbosity", 0);
		log.set_verbosity((log_level) *param_pnm_verbosity);
	}; // empty image
	pnm_image(int _width, int _height, pnm_variant_type _type, parameters &params): pnm_image(params) { // create empty image with given image size and type
		width = _width;
		height = _height;
		type = _type;
		maxvalue = guess_maxvalue();
		data = new pnm_data_t[size()];
	};
	pnm_image(const pnm_image & copy): width(copy.width), height(copy.height), type(copy.type), maxvalue(copy.maxvalue), log(copy.log), par(copy.par) { // copy constructor
		data = new pnm_data_t[size()];
		std::memcpy(data, copy.data, sizeof(pnm_data_t) * size());
	};
	~pnm_image();
	void read(FILE *fd);
	void write(FILE *fd);
	void convert(pnm_variant_type new_type); // Convert between two image types
	void erase_image(); // Fill image with white
	int width;
	int height;
	pnm_variant_type type;
	pnm_data_t maxvalue; // Maximal value of one pixel
	pnm_data_t *data; // Raw data
	pnm_image(pnm_image &&) = default;
	pnm_image &operator=(pnm_image &&move);
private:
	int fscanf_comment(FILE *stream, const char *format, ...); // Read line from filedescriptor, ignore comments (#)
	bool read_header(FILE *fd); // Parse header
	void write_header(FILE *fd); // Write image header
	int size(); // Calculate buffersize for new data (given type and image dimensions)
	int guess_maxvalue(); // Get common maxvalue for image type (1 for bitmap, 255 for others)

	logger log;
	parameters *par;
};

}; // namespace

#endif