1 files changed, 26 insertions, 0 deletions
diff --git a/src/treebuilder/element-type.h b/src/treebuilder/element-type.h
index 08f58de..a8e33d9 100644
--- a/src/treebuilder/element-type.h
+++ b/src/treebuilder/element-type.h
@@ -9,6 +9,7 @@
 #define hubbub_treebuilder_element_type_h_
 
 #include "treebuilder/treebuilder.h"
+#include "utils/utils.h"
 
 typedef enum
 {
@@ -41,9 +42,34 @@ struct element_type_map {
 	element_type type;
 };
 
+/* Generated by gperf */
 const struct element_type_map *hubbub_element_type_lookup(
 		register const char *str,
 		register size_t len);
 
+/**
+ * Convert an element name into an element type
+ *
+ * \param treebuilder  The treebuilder instance
+ * \param tag_name     The tag name to consider
+ * \return The corresponding element type
+ */
+static inline element_type element_type_from_name(
+		hubbub_treebuilder *treebuilder,
+		const hubbub_string *tag_name)
+{
+	const struct element_type_map *value;
+
+	UNUSED(treebuilder);
+
+	value = hubbub_element_type_lookup((const char *)tag_name->ptr,
+			tag_name->len);
+	if (value == NULL) {
+		return UNKNOWN;
+	}
+
+	return value->type;
+}
+
 #endif