You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
219 lines
7.6 KiB
219 lines
7.6 KiB
/*
|
|
* 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 "utils/calendar/calendar-javaicu.h"
|
|
|
|
#include "annotator/types.h"
|
|
#include "utils/base/statusor.h"
|
|
#include "utils/java/jni-base.h"
|
|
#include "utils/java/jni-helper.h"
|
|
|
|
namespace libtextclassifier3 {
|
|
namespace {
|
|
|
|
// Generic version of icu::Calendar::add with error checking.
|
|
bool CalendarAdd(JniCache* jni_cache, JNIEnv* jenv, jobject calendar,
|
|
jint field, jint value) {
|
|
return JniHelper::CallVoidMethod(jenv, calendar, jni_cache->calendar_add,
|
|
field, value)
|
|
.ok();
|
|
}
|
|
|
|
// Generic version of icu::Calendar::get with error checking.
|
|
bool CalendarGet(JniCache* jni_cache, JNIEnv* jenv, jobject calendar,
|
|
jint field, jint* value) {
|
|
TC3_ASSIGN_OR_RETURN_FALSE(
|
|
*value,
|
|
JniHelper::CallIntMethod(jenv, calendar, jni_cache->calendar_get, field));
|
|
return true;
|
|
}
|
|
|
|
// Generic version of icu::Calendar::set with error checking.
|
|
bool CalendarSet(JniCache* jni_cache, JNIEnv* jenv, jobject calendar,
|
|
jint field, jint value) {
|
|
return JniHelper::CallVoidMethod(jenv, calendar, jni_cache->calendar_set,
|
|
field, value)
|
|
.ok();
|
|
}
|
|
|
|
// Extracts the first tag from a BCP47 tag (e.g. "en" for "en-US").
|
|
std::string GetFirstBcp47Tag(const std::string& tag) {
|
|
for (size_t i = 0; i < tag.size(); ++i) {
|
|
if (tag[i] == '_' || tag[i] == '-') {
|
|
return std::string(tag, 0, i);
|
|
}
|
|
}
|
|
return tag;
|
|
}
|
|
|
|
} // anonymous namespace
|
|
|
|
Calendar::Calendar(JniCache* jni_cache)
|
|
: jni_cache_(jni_cache),
|
|
jenv_(jni_cache_ ? jni_cache->GetEnv() : nullptr),
|
|
calendar_(nullptr, jenv_) {}
|
|
|
|
bool Calendar::Initialize(const std::string& time_zone,
|
|
const std::string& locale, int64 time_ms_utc) {
|
|
if (!jni_cache_ || !jenv_) {
|
|
TC3_LOG(ERROR) << "Initialize without env";
|
|
return false;
|
|
}
|
|
|
|
// We'll assume the day indices match later on, so verify it here.
|
|
if (jni_cache_->calendar_sunday != kSunday ||
|
|
jni_cache_->calendar_monday != kMonday ||
|
|
jni_cache_->calendar_tuesday != kTuesday ||
|
|
jni_cache_->calendar_wednesday != kWednesday ||
|
|
jni_cache_->calendar_thursday != kThursday ||
|
|
jni_cache_->calendar_friday != kFriday ||
|
|
jni_cache_->calendar_saturday != kSaturday) {
|
|
TC3_LOG(ERROR) << "day of the week indices mismatch";
|
|
return false;
|
|
}
|
|
|
|
// Get the time zone.
|
|
TC3_ASSIGN_OR_RETURN_FALSE(ScopedLocalRef<jstring> java_time_zone_str,
|
|
JniHelper::NewStringUTF(jenv_, time_zone.c_str()));
|
|
TC3_ASSIGN_OR_RETURN_FALSE(
|
|
ScopedLocalRef<jobject> java_time_zone,
|
|
JniHelper::CallStaticObjectMethod(jenv_, jni_cache_->timezone_class.get(),
|
|
jni_cache_->timezone_get_timezone,
|
|
java_time_zone_str.get()));
|
|
if (java_time_zone == nullptr) {
|
|
TC3_LOG(ERROR) << "failed to get timezone";
|
|
return false;
|
|
}
|
|
|
|
// Get the locale.
|
|
ScopedLocalRef<jobject> java_locale(nullptr, jenv_);
|
|
if (jni_cache_->locale_for_language_tag) {
|
|
// API level 21+, we can actually parse language tags.
|
|
TC3_ASSIGN_OR_RETURN_FALSE(ScopedLocalRef<jstring> java_locale_str,
|
|
JniHelper::NewStringUTF(jenv_, locale.c_str()));
|
|
|
|
TC3_ASSIGN_OR_RETURN_FALSE(
|
|
java_locale,
|
|
JniHelper::CallStaticObjectMethod(jenv_, jni_cache_->locale_class.get(),
|
|
jni_cache_->locale_for_language_tag,
|
|
java_locale_str.get()));
|
|
} else {
|
|
// API level <21. We can't parse tags, so we just use the language.
|
|
TC3_ASSIGN_OR_RETURN_FALSE(
|
|
ScopedLocalRef<jstring> java_language_str,
|
|
JniHelper::NewStringUTF(jenv_, GetFirstBcp47Tag(locale).c_str()));
|
|
|
|
TC3_ASSIGN_OR_RETURN_FALSE(
|
|
java_locale, JniHelper::NewObject(jenv_, jni_cache_->locale_class.get(),
|
|
jni_cache_->locale_init_string,
|
|
java_language_str.get()));
|
|
}
|
|
if (java_locale == nullptr) {
|
|
TC3_LOG(ERROR) << "failed to get locale";
|
|
return false;
|
|
}
|
|
|
|
// Get the calendar.
|
|
TC3_ASSIGN_OR_RETURN_FALSE(
|
|
calendar_, JniHelper::CallStaticObjectMethod(
|
|
jenv_, jni_cache_->calendar_class.get(),
|
|
jni_cache_->calendar_get_instance, java_time_zone.get(),
|
|
java_locale.get()));
|
|
if (calendar_ == nullptr) {
|
|
TC3_LOG(ERROR) << "failed to get calendar";
|
|
return false;
|
|
}
|
|
|
|
// Set the time.
|
|
if (!JniHelper::CallVoidMethod(jenv_, calendar_.get(),
|
|
jni_cache_->calendar_set_time_in_millis,
|
|
time_ms_utc)
|
|
.ok()) {
|
|
TC3_LOG(ERROR) << "failed to set time";
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool Calendar::GetFirstDayOfWeek(int* value) const {
|
|
if (!jni_cache_ || !jenv_ || !calendar_) return false;
|
|
|
|
TC3_ASSIGN_OR_RETURN_FALSE(
|
|
*value,
|
|
JniHelper::CallIntMethod(jenv_, calendar_.get(),
|
|
jni_cache_->calendar_get_first_day_of_week));
|
|
return true;
|
|
}
|
|
|
|
bool Calendar::GetTimeInMillis(int64* value) const {
|
|
if (!jni_cache_ || !jenv_ || !calendar_) return false;
|
|
|
|
TC3_ASSIGN_OR_RETURN_FALSE(
|
|
*value,
|
|
JniHelper::CallLongMethod(jenv_, calendar_.get(),
|
|
jni_cache_->calendar_get_time_in_millis));
|
|
|
|
return true;
|
|
}
|
|
|
|
CalendarLib::CalendarLib() {
|
|
TC3_LOG(FATAL) << "Java ICU CalendarLib must be initialized with a JniCache.";
|
|
}
|
|
|
|
CalendarLib::CalendarLib(const std::shared_ptr<JniCache>& jni_cache)
|
|
: jni_cache_(jni_cache) {}
|
|
|
|
// Below is the boilerplate code for implementing the specialisations of
|
|
// get/set/add for the various field types.
|
|
#define TC3_DEFINE_FIELD_ACCESSOR(NAME, FIELD, KIND, TYPE) \
|
|
bool Calendar::KIND##NAME(TYPE value) const { \
|
|
if (!jni_cache_ || !jenv_ || !calendar_) return false; \
|
|
return Calendar##KIND(jni_cache_, jenv_, calendar_.get(), \
|
|
jni_cache_->calendar_##FIELD, value); \
|
|
}
|
|
#define TC3_DEFINE_ADD(NAME, CONST) \
|
|
TC3_DEFINE_FIELD_ACCESSOR(NAME, CONST, Add, int)
|
|
#define TC3_DEFINE_SET(NAME, CONST) \
|
|
TC3_DEFINE_FIELD_ACCESSOR(NAME, CONST, Set, int)
|
|
#define TC3_DEFINE_GET(NAME, CONST) \
|
|
TC3_DEFINE_FIELD_ACCESSOR(NAME, CONST, Get, int*)
|
|
|
|
TC3_DEFINE_ADD(Second, second)
|
|
TC3_DEFINE_ADD(Minute, minute)
|
|
TC3_DEFINE_ADD(HourOfDay, hour_of_day)
|
|
TC3_DEFINE_ADD(DayOfMonth, day_of_month)
|
|
TC3_DEFINE_ADD(Year, year)
|
|
TC3_DEFINE_ADD(Month, month)
|
|
TC3_DEFINE_GET(DayOfWeek, day_of_week)
|
|
TC3_DEFINE_SET(ZoneOffset, zone_offset)
|
|
TC3_DEFINE_SET(DstOffset, dst_offset)
|
|
TC3_DEFINE_SET(Year, year)
|
|
TC3_DEFINE_SET(Month, month)
|
|
TC3_DEFINE_SET(DayOfYear, day_of_year)
|
|
TC3_DEFINE_SET(DayOfMonth, day_of_month)
|
|
TC3_DEFINE_SET(DayOfWeek, day_of_week)
|
|
TC3_DEFINE_SET(HourOfDay, hour_of_day)
|
|
TC3_DEFINE_SET(Minute, minute)
|
|
TC3_DEFINE_SET(Second, second)
|
|
TC3_DEFINE_SET(Millisecond, millisecond)
|
|
|
|
#undef TC3_DEFINE_FIELD_ACCESSOR
|
|
#undef TC3_DEFINE_ADD
|
|
#undef TC3_DEFINE_SET
|
|
#undef TC3_DEFINE_GET
|
|
|
|
} // namespace libtextclassifier3
|