Refactor: Add generic poll ED_transverts_poll

1 files changed, 12 insertions, 0 deletions

diff --git a/source/blender/editors/util/ed_transverts.c b/source/blender/editors/util/ed_transverts.c
index 4123132ad03..9f41642280b 100644
--- a/source/blender/editors/util/ed_transverts.c
+++ b/source/blender/editors/util/ed_transverts.c
@@ -43,6 +43,7 @@
 #include "BKE_lattice.h"
 #include "BKE_editmesh.h"
 #include "BKE_DerivedMesh.h"
+#include "BKE_context.h"
 
 #include "ED_armature.h"
 
@@ -460,3 +461,14 @@ void ED_transverts_free(TransVertStore *tvs)
 	MEM_SAFE_FREE(tvs->transverts);
 	tvs->transverts_tot = 0;
 }
+
+int ED_transverts_poll(bContext *C)
+{
+	Object *obedit = CTX_data_edit_object(C);
+	if (obedit) {
+		if (ED_transverts_check_obedit(obedit)) {
+			return true;
+		}
+	}
+	return false;
+}