height_map_brush.h

#ifndef HEIGHT_MAP_BRUSH_H
#define HEIGHT_MAP_BRUSH_H

#include "grid.h"
#include "height_map.h"

class HeightMapBrush {
public:
	enum Mode {
		MODE_ADD = 0,
		MODE_SUBTRACT,
		MODE_SMOOTH,
		MODE_FLATTEN,
		MODE_COLOR,
		MODE_SPLAT,
		MODE_MASK,
		MODE_COUNT
	};

	struct UndoData {
		PoolIntArray chunk_positions;
		Array undo;
		Array redo;
		int channel;
	};

	HeightMapBrush();

	void set_mode(Mode mode);
	Mode get_mode() const { return _mode; }

	void set_radius(int p_radius);
	int get_radius() const { return _radius; }

	void set_opacity(float opacity);
	float get_opacity() const { return _opacity; }

	void set_flatten_height(float flatten_height);
	float get_flatten_height() const { return _flatten_height; }

	static HeightMapData::Channel get_mode_channel(Mode mode);

	void set_texture_index(int tid);
	int get_texture_index() const { return _texture_index; }

	void set_color(Color c);
	Color get_color() const { return _color; }

	void paint(HeightMap &height_map, Point2i cell_pos, int override_mode);

	UndoData pop_undo_redo_data(const HeightMapData &heightmap_data);

private:
	struct UndoCache {
		HashMap< Point2i, Ref<Image> > chunks;
		void clear() {
			chunks.clear();
		}
	};

	void generate_procedural(int radius);

	void paint_height(HeightMapData &data, Point2i cell_pos, float speed);
	void smooth_height(HeightMapData &data, Point2i cell_pos, float speed);
	void flatten_height(HeightMapData &data, Point2i cell_pos);
	void paint_color(HeightMapData &data, Point2i cell_pos);
	void paint_splat(HeightMapData &data, Point2i origin);
	void paint_mask(HeightMapData &data, Point2i origin);

	static void backup_for_undo(const Image &im, UndoCache &undo_cache, Point2i rect_origin, Point2i rect_size);

private:
	int _radius;
	float _opacity;
	Grid2D<float> _shape;
	float _shape_sum;
	Mode _mode;
	float _flatten_height;
	int _texture_index;
	Color _color;
	UndoCache _undo_cache;
};


VARIANT_ENUM_CAST(HeightMapBrush::Mode)


#endif // HEIGHT_MAP_BRUSH_H