/*
|
* Copyright (C) 2018 The Android Open Source Project
|
*
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
*/
|
|
#include "annotator/annotator_jni_common.h"
|
|
#include "utils/java/jni-base.h"
|
#include "utils/java/scoped_local_ref.h"
|
|
namespace libtextclassifier3 {
|
namespace {
|
|
std::unordered_set<std::string> EntityTypesFromJObject(JNIEnv* env,
|
const jobject& jobject) {
|
std::unordered_set<std::string> entity_types;
|
jobjectArray jentity_types = reinterpret_cast<jobjectArray>(jobject);
|
const int size = env->GetArrayLength(jentity_types);
|
for (int i = 0; i < size; ++i) {
|
jstring jentity_type =
|
reinterpret_cast<jstring>(env->GetObjectArrayElement(jentity_types, i));
|
entity_types.insert(ToStlString(env, jentity_type));
|
}
|
return entity_types;
|
}
|
|
template <typename T>
|
T FromJavaOptionsInternal(JNIEnv* env, jobject joptions,
|
const std::string& class_name) {
|
if (!joptions) {
|
return {};
|
}
|
|
const ScopedLocalRef<jclass> options_class(env->FindClass(class_name.c_str()),
|
env);
|
if (!options_class) {
|
return {};
|
}
|
|
const std::pair<bool, jobject> status_or_locales = CallJniMethod0<jobject>(
|
env, joptions, options_class.get(), &JNIEnv::CallObjectMethod,
|
"getLocale", "Ljava/lang/String;");
|
const std::pair<bool, jobject> status_or_reference_timezone =
|
CallJniMethod0<jobject>(env, joptions, options_class.get(),
|
&JNIEnv::CallObjectMethod, "getReferenceTimezone",
|
"Ljava/lang/String;");
|
const std::pair<bool, int64> status_or_reference_time_ms_utc =
|
CallJniMethod0<int64>(env, joptions, options_class.get(),
|
&JNIEnv::CallLongMethod, "getReferenceTimeMsUtc",
|
"J");
|
const std::pair<bool, jobject> status_or_detected_text_language_tags =
|
CallJniMethod0<jobject>(
|
env, joptions, options_class.get(), &JNIEnv::CallObjectMethod,
|
"getDetectedTextLanguageTags", "Ljava/lang/String;");
|
const std::pair<bool, int> status_or_annotation_usecase =
|
CallJniMethod0<int>(env, joptions, options_class.get(),
|
&JNIEnv::CallIntMethod, "getAnnotationUsecase", "I");
|
|
if (!status_or_locales.first || !status_or_reference_timezone.first ||
|
!status_or_reference_time_ms_utc.first ||
|
!status_or_detected_text_language_tags.first ||
|
!status_or_annotation_usecase.first) {
|
return {};
|
}
|
|
T options;
|
options.locales =
|
ToStlString(env, reinterpret_cast<jstring>(status_or_locales.second));
|
options.reference_timezone = ToStlString(
|
env, reinterpret_cast<jstring>(status_or_reference_timezone.second));
|
options.reference_time_ms_utc = status_or_reference_time_ms_utc.second;
|
options.detected_text_language_tags = ToStlString(
|
env,
|
reinterpret_cast<jstring>(status_or_detected_text_language_tags.second));
|
options.annotation_usecase =
|
static_cast<AnnotationUsecase>(status_or_annotation_usecase.second);
|
return options;
|
}
|
} // namespace
|
|
SelectionOptions FromJavaSelectionOptions(JNIEnv* env, jobject joptions) {
|
if (!joptions) {
|
return {};
|
}
|
|
const ScopedLocalRef<jclass> options_class(
|
env->FindClass(TC3_PACKAGE_PATH TC3_ANNOTATOR_CLASS_NAME_STR
|
"$SelectionOptions"),
|
env);
|
const std::pair<bool, jobject> status_or_locales = CallJniMethod0<jobject>(
|
env, joptions, options_class.get(), &JNIEnv::CallObjectMethod,
|
"getLocales", "Ljava/lang/String;");
|
const std::pair<bool, int> status_or_annotation_usecase =
|
CallJniMethod0<int>(env, joptions, options_class.get(),
|
&JNIEnv::CallIntMethod, "getAnnotationUsecase", "I");
|
if (!status_or_locales.first || !status_or_annotation_usecase.first) {
|
return {};
|
}
|
|
SelectionOptions options;
|
options.locales =
|
ToStlString(env, reinterpret_cast<jstring>(status_or_locales.second));
|
options.annotation_usecase =
|
static_cast<AnnotationUsecase>(status_or_annotation_usecase.second);
|
|
return options;
|
}
|
|
ClassificationOptions FromJavaClassificationOptions(JNIEnv* env,
|
jobject joptions) {
|
return FromJavaOptionsInternal<ClassificationOptions>(
|
env, joptions,
|
TC3_PACKAGE_PATH TC3_ANNOTATOR_CLASS_NAME_STR "$ClassificationOptions");
|
}
|
|
AnnotationOptions FromJavaAnnotationOptions(JNIEnv* env, jobject joptions) {
|
if (!joptions) return {};
|
const ScopedLocalRef<jclass> options_class(
|
env->FindClass(TC3_PACKAGE_PATH TC3_ANNOTATOR_CLASS_NAME_STR
|
"$AnnotationOptions"),
|
env);
|
if (!options_class) return {};
|
const std::pair<bool, jobject> status_or_entity_types =
|
CallJniMethod0<jobject>(env, joptions, options_class.get(),
|
&JNIEnv::CallObjectMethod, "getEntityTypes",
|
"[Ljava/lang/String;");
|
if (!status_or_entity_types.first) return {};
|
const std::pair<bool, bool> status_or_enable_serialized_entity_data =
|
CallJniMethod0<bool>(env, joptions, options_class.get(),
|
&JNIEnv::CallBooleanMethod,
|
"isSerializedEntityDataEnabled", "Z");
|
if (!status_or_enable_serialized_entity_data.first) return {};
|
AnnotationOptions annotation_options =
|
FromJavaOptionsInternal<AnnotationOptions>(
|
env, joptions,
|
TC3_PACKAGE_PATH TC3_ANNOTATOR_CLASS_NAME_STR "$AnnotationOptions");
|
annotation_options.entity_types =
|
EntityTypesFromJObject(env, status_or_entity_types.second);
|
annotation_options.is_serialized_entity_data_enabled =
|
status_or_enable_serialized_entity_data.second;
|
return annotation_options;
|
}
|
|
} // namespace libtextclassifier3
|
