Add grn_obj_is_normalizer_proc()

groonga · Oct 19, 2015 · 5526432 · 5526432
1 parent 32c2eb9
commit 5526432
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 0 deletions.
diff --git a/include/groonga/obj.h b/include/groonga/obj.h
@@ -31,6 +31,7 @@ GRN_API grn_bool grn_obj_is_tokenizer_proc(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_function_proc(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_selector_proc(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_selector_only_proc(grn_ctx *ctx, grn_obj *obj);
+GRN_API grn_bool grn_obj_is_normalizer_proc(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_scorer_proc(grn_ctx *ctx, grn_obj *obj);
 
 GRN_API grn_rc grn_obj_cast(grn_ctx *ctx,

diff --git a/lib/obj.c b/lib/obj.c
@@ -128,6 +128,19 @@ grn_obj_is_selector_only_proc(grn_ctx *ctx, grn_obj *obj)
   return proc->funcs[PROC_INIT] == NULL;
 }
 
+grn_bool
+grn_obj_is_normalizer_proc(grn_ctx *ctx, grn_obj *obj)
+{
+  grn_proc *proc;
+
+  if (!grn_obj_is_proc(ctx, obj)) {
+    return GRN_FALSE;
+  }
+
+  proc = (grn_proc *)obj;
+  return proc->type == GRN_PROC_NORMALIZER;
+}
+
 grn_bool
 grn_obj_is_scorer_proc(grn_ctx *ctx, grn_obj *obj)
 {

diff --git a/test/unit/core/test-object.c b/test/unit/core/test-object.c
@@ -39,6 +39,8 @@ void data_is_function_proc(void);
 void test_is_function_proc(gconstpointer data);
 void data_is_selector_proc(void);
 void test_is_selector_proc(gconstpointer data);
+void data_is_normalizer_proc(void);
+void test_is_normalizer_proc(gconstpointer data);
 void data_is_scorer_proc(void);
 void test_is_scorer_proc(gconstpointer data);
 
@@ -329,6 +331,38 @@ test_is_selector_proc(gconstpointer data)
   }
 }
 
+void
+data_is_normalizer_proc(void)
+{
+#define ADD_DATUM(expected, name)                                       \
+  gcut_add_datum((expected ?                                            \
+                  "normalizer-proc - " name :                           \
+                  "not normalizer-proc - " name),                       \
+                 "expected", G_TYPE_BOOLEAN, expected,                  \
+                 "name", G_TYPE_STRING, name,                           \
+                 NULL)
+
+  ADD_DATUM(TRUE, "NormalizerAuto");
+  ADD_DATUM(FALSE, "TokenBigram");
+
+#undef ADD_DATUM
+}
+
+void
+test_is_normalizer_proc(gconstpointer data)
+{
+  const gchar *name;
+  grn_obj *object;
+
+  name = gcut_data_get_string(data, "name");
+  object = grn_ctx_get(context, name, strlen(name));
+  if (gcut_data_get_string(data, "expected")) {
+    cut_assert_true(grn_obj_is_normalizer_proc(context, object));
+  } else {
+    cut_assert_false(grn_obj_is_normalizer_proc(context, object));
+  }
+}
+
 void
 data_is_scorer_proc(void)
 {