Add get-size functions to SurfaceTool

Author: zhehangd

doc/classes/SurfaceTool.xml

@@ -163,6 +163,18 @@
 				Returns the format for custom [param channel_index] (currently up to 4). Returns [constant CUSTOM_MAX] if this custom channel is unused.
 			</description>
 		</method>
+		<method name="get_num_indices" qualifiers="const">
+			<return type="int" />
+			<description>
+				Returns the current number of indices.
+			</description>
+		</method>
+		<method name="get_num_vertices" qualifiers="const">
+			<return type="int" />
+			<description>
+				Returns the current number of vertices.
+			</description>
+		</method>
 		<method name="get_primitive_type" qualifiers="const">
 			<return type="int" enum="Mesh.PrimitiveType" />
 			<description>

scene/resources/surface_tool.cpp

@@ -223,6 +223,14 @@ void SurfaceTool::begin(Mesh::PrimitiveType p_primitive) {
 	first = true;
 }
 
+int SurfaceTool::get_num_vertices() const {
+	return vertex_array.size();
+}
+
+int SurfaceTool::get_num_indices() const {
+	return index_array.size();
+}
+
 void SurfaceTool::add_vertex(const Vector3 &p_vertex) {
 	ERR_FAIL_COND(!begun);
 
@@ -1346,6 +1354,8 @@ void SurfaceTool::_bind_methods() {
 	ClassDB::bind_method(D_METHOD("get_custom_format", "channel_index"), &SurfaceTool::get_custom_format);
 
 	ClassDB::bind_method(D_METHOD("begin", "primitive"), &SurfaceTool::begin);
+	ClassDB::bind_method(D_METHOD("get_num_vertices"), &SurfaceTool::get_num_vertices);
+	ClassDB::bind_method(D_METHOD("get_num_indices"), &SurfaceTool::get_num_indices);
 
 	ClassDB::bind_method(D_METHOD("add_vertex", "vertex"), &SurfaceTool::add_vertex);
 	ClassDB::bind_method(D_METHOD("set_color", "color"), &SurfaceTool::set_color);

scene/resources/surface_tool.h

@@ -209,6 +209,9 @@ class SurfaceTool : public RefCounted {
 	void set_weights(const Vector<float> &p_weights);
 	void set_smooth_group(uint32_t p_group);
 
+	int get_num_vertices() const;
+	int get_num_indices() const;
+
 	void add_vertex(const Vector3 &p_vertex);
 
 	void add_triangle_fan(const Vector<Vector3> &p_vertices, const Vector<Vector2> &p_uvs = Vector<Vector2>(), const Vector<Color> &p_colors = Vector<Color>(), const Vector<Vector2> &p_uv2s = Vector<Vector2>(), const Vector<Vector3> &p_normals = Vector<Vector3>(), const Vector<Plane> &p_tangents = Vector<Plane>());