1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*
* Copyright (C) 2017 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 android.view.textclassifier;
/**
* Java wrapper for SmartSelection native library interface.
* This library is used for detecting entities in text.
*/
final class SmartSelection {
static {
System.loadLibrary("textclassifier");
}
/** Hints the classifier that this may be a url. */
static final int HINT_FLAG_URL = 0x01;
/** Hints the classifier that this may be an email. */
static final int HINT_FLAG_EMAIL = 0x02;
private final long mCtx;
/**
* Creates a new instance of SmartSelect predictor, using the provided model image,
* given as a file descriptor.
*/
SmartSelection(int fd) {
mCtx = nativeNew(fd);
}
/**
* Given a string context and current selection, computes the SmartSelection suggestion.
*
* The begin and end are character indices into the context UTF8 string. selectionBegin is the
* character index where the selection begins, and selectionEnd is the index of one character
* past the selection span.
*
* The return value is an array of two ints: suggested selection beginning and end, with the
* same semantics as the input selectionBeginning and selectionEnd.
*/
public int[] suggest(String context, int selectionBegin, int selectionEnd) {
return nativeSuggest(mCtx, context, selectionBegin, selectionEnd);
}
/**
* Given a string context and current selection, classifies the type of the selected text.
*
* The begin and end params are character indices in the context string.
*
* Returns an array of ClassificationResult objects with the probability
* scores for different collections.
*/
public ClassificationResult[] classifyText(
String context, int selectionBegin, int selectionEnd, int hintFlags) {
return nativeClassifyText(mCtx, context, selectionBegin, selectionEnd, hintFlags);
}
/**
* Frees up the allocated memory.
*/
public void close() {
nativeClose(mCtx);
}
/**
* Returns the language of the model.
*/
public static String getLanguage(int fd) {
return nativeGetLanguage(fd);
}
/**
* Returns the version of the model.
*/
public static int getVersion(int fd) {
return nativeGetVersion(fd);
}
private static native long nativeNew(int fd);
private static native int[] nativeSuggest(
long context, String text, int selectionBegin, int selectionEnd);
private static native ClassificationResult[] nativeClassifyText(
long context, String text, int selectionBegin, int selectionEnd, int hintFlags);
private static native void nativeClose(long context);
private static native String nativeGetLanguage(int fd);
private static native int nativeGetVersion(int fd);
/** Classification result for classifyText method. */
static final class ClassificationResult {
final String mCollection;
/** float range: 0 - 1 */
final float mScore;
ClassificationResult(String collection, float score) {
mCollection = collection;
mScore = score;
}
}
}
|