#pragma once

#include "scene/gui/button.h"
#include "terrain_editor/terrain_mesh_generator.h"
#include "terrain_editor/terrain_primitive.h"

class AddPrimitiveButton : public Button {
	GDCLASS(AddPrimitiveButton, Button);
	static void _bind_methods();
	void pressed() override;

public:
	void set_primitive_blueprint(Ref<TerrainPrimitive> primitive);
	Ref<TerrainPrimitive> get_primitive_blueprint() const;
	void set_terrain(TerrainMeshGenerator *terrain);
	TerrainMeshGenerator *get_terrain() const;

private:
	Ref<TerrainPrimitive> primitive_blueprint{};
	TerrainMeshGenerator *terrain{ nullptr };
};