| /* |
| * 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. |
| */ |
| |
| package com.google.android.textclassifier; |
| |
| import android.content.res.AssetFileDescriptor; |
| import java.util.concurrent.atomic.AtomicBoolean; |
| |
| /** |
| * Java wrapper for LangId native library interface. This class is used to detect languages in text. |
| * |
| * @hide |
| */ |
| public final class LangIdModel implements AutoCloseable { |
| private final AtomicBoolean isClosed = new AtomicBoolean(false); |
| |
| static { |
| System.loadLibrary("textclassifier"); |
| } |
| |
| private long modelPtr; |
| |
| /** Creates a new instance of LangId predictor, using the provided model image. */ |
| public LangIdModel(int fd) { |
| modelPtr = nativeNew(fd); |
| if (modelPtr == 0L) { |
| throw new IllegalArgumentException("Couldn't initialize LangId from given file descriptor."); |
| } |
| } |
| |
| /** Creates a new instance of LangId predictor, using the provided model image. */ |
| public LangIdModel(String modelPath) { |
| modelPtr = nativeNewFromPath(modelPath); |
| if (modelPtr == 0L) { |
| throw new IllegalArgumentException("Couldn't initialize LangId from given file."); |
| } |
| } |
| |
| /** |
| * Creates a new instance of LangId predictor, using the provided model image, given as an {@link |
| * AssetFileDescriptor}. |
| */ |
| public LangIdModel(AssetFileDescriptor assetFileDescriptor) { |
| modelPtr = |
| nativeNewWithOffset( |
| assetFileDescriptor.getParcelFileDescriptor().getFd(), |
| assetFileDescriptor.getStartOffset(), |
| assetFileDescriptor.getLength()); |
| if (modelPtr == 0L) { |
| throw new IllegalArgumentException("Couldn't initialize LangId from asset file descriptor."); |
| } |
| } |
| |
| /** Creates a new instance of LangId predictor, using the provided model image. */ |
| public LangIdModel(int fd, long offset, long size) { |
| modelPtr = nativeNewWithOffset(fd, offset, size); |
| if (modelPtr == 0L) { |
| throw new IllegalArgumentException("Couldn't initialize LangId from given file descriptor."); |
| } |
| } |
| |
| /** Detects the languages for given text. */ |
| public LanguageResult[] detectLanguages(String text) { |
| return nativeDetectLanguages(modelPtr, text); |
| } |
| |
| /** Frees up the allocated memory. */ |
| @Override |
| public void close() { |
| if (isClosed.compareAndSet(false, true)) { |
| nativeClose(modelPtr); |
| modelPtr = 0L; |
| } |
| } |
| |
| @Override |
| protected void finalize() throws Throwable { |
| try { |
| close(); |
| } finally { |
| super.finalize(); |
| } |
| } |
| |
| /** Result for detectLanguages method. */ |
| public static final class LanguageResult { |
| final String mLanguage; |
| final float mScore; |
| |
| LanguageResult(String language, float score) { |
| mLanguage = language; |
| mScore = score; |
| } |
| |
| public final String getLanguage() { |
| return mLanguage; |
| } |
| |
| public final float getScore() { |
| return mScore; |
| } |
| } |
| |
| /** Returns the version of the LangId model used. */ |
| public int getVersion() { |
| return nativeGetVersion(modelPtr); |
| } |
| |
| public static int getVersion(int fd) { |
| return nativeGetVersionFromFd(fd); |
| } |
| |
| /** Returns the version of the model. */ |
| public static int getVersion(AssetFileDescriptor assetFileDescriptor) { |
| return nativeGetVersionWithOffset( |
| assetFileDescriptor.getParcelFileDescriptor().getFd(), |
| assetFileDescriptor.getStartOffset(), |
| assetFileDescriptor.getLength()); |
| } |
| |
| public float getLangIdThreshold() { |
| return nativeGetLangIdThreshold(modelPtr); |
| } |
| |
| /** Retrieves the pointer to the native object. */ |
| long getNativePointer() { |
| return modelPtr; |
| } |
| |
| // Visible for testing. |
| float getLangIdNoiseThreshold() { |
| return nativeGetLangIdNoiseThreshold(modelPtr); |
| } |
| |
| // Visible for testing. |
| int getMinTextSizeInBytes() { |
| return nativeGetMinTextSizeInBytes(modelPtr); |
| } |
| |
| /** |
| * Returns the pointer to the native object. Note: Need to keep the LangIdModel alive as long as |
| * the pointer is used. |
| */ |
| long getNativeLangIdPointer() { |
| return modelPtr; |
| } |
| |
| private static native long nativeNew(int fd); |
| |
| private static native long nativeNewFromPath(String path); |
| |
| private static native long nativeNewWithOffset(int fd, long offset, long size); |
| |
| private native LanguageResult[] nativeDetectLanguages(long nativePtr, String text); |
| |
| private native void nativeClose(long nativePtr); |
| |
| private native int nativeGetVersion(long nativePtr); |
| |
| private static native int nativeGetVersionFromFd(int fd); |
| |
| private native float nativeGetLangIdThreshold(long nativePtr); |
| |
| private native float nativeGetLangIdNoiseThreshold(long nativePtr); |
| |
| private native int nativeGetMinTextSizeInBytes(long nativePtr); |
| |
| private static native int nativeGetVersionWithOffset(int fd, long offset, long size); |
| } |