1 /* 2 * Copyright (C) 2014 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.inputmethod.latin.utils; 18 19 import com.android.inputmethod.annotations.UsedForTesting; 20 import com.android.inputmethod.latin.BinaryDictionary; 21 import com.android.inputmethod.latin.common.StringUtils; 22 import com.android.inputmethod.latin.makedict.DictionaryHeader; 23 import com.android.inputmethod.latin.makedict.UnsupportedFormatException; 24 25 import java.io.File; 26 import java.io.IOException; 27 import java.util.Locale; 28 import java.util.Map; 29 import java.util.regex.Matcher; 30 import java.util.regex.Pattern; 31 32 public final class BinaryDictionaryUtils { 33 private static final String TAG = BinaryDictionaryUtils.class.getSimpleName(); 34 BinaryDictionaryUtils()35 private BinaryDictionaryUtils() { 36 // This utility class is not publicly instantiable. 37 } 38 39 static { JniUtils.loadNativeLibrary()40 JniUtils.loadNativeLibrary(); 41 } 42 43 @UsedForTesting createEmptyDictFileNative(String filePath, long dictVersion, String locale, String[] attributeKeyStringArray, String[] attributeValueStringArray)44 private static native boolean createEmptyDictFileNative(String filePath, long dictVersion, 45 String locale, String[] attributeKeyStringArray, String[] attributeValueStringArray); calcNormalizedScoreNative(int[] before, int[] after, int score)46 private static native float calcNormalizedScoreNative(int[] before, int[] after, int score); setCurrentTimeForTestNative(int currentTime)47 private static native int setCurrentTimeForTestNative(int currentTime); 48 getHeader(final File dictFile)49 public static DictionaryHeader getHeader(final File dictFile) 50 throws IOException, UnsupportedFormatException { 51 return getHeaderWithOffsetAndLength(dictFile, 0 /* offset */, dictFile.length()); 52 } 53 getHeaderWithOffsetAndLength(final File dictFile, final long offset, final long length)54 public static DictionaryHeader getHeaderWithOffsetAndLength(final File dictFile, 55 final long offset, final long length) throws IOException, UnsupportedFormatException { 56 // dictType is never used for reading the header. Passing an empty string. 57 final BinaryDictionary binaryDictionary = new BinaryDictionary( 58 dictFile.getAbsolutePath(), offset, length, 59 true /* useFullEditDistance */, null /* locale */, "" /* dictType */, 60 false /* isUpdatable */); 61 final DictionaryHeader header = binaryDictionary.getHeader(); 62 binaryDictionary.close(); 63 if (header == null) { 64 throw new IOException(); 65 } 66 return header; 67 } 68 renameDict(final File dictFile, final File newDictFile)69 public static boolean renameDict(final File dictFile, final File newDictFile) { 70 if (dictFile.isFile()) { 71 return dictFile.renameTo(newDictFile); 72 } else if (dictFile.isDirectory()) { 73 final String dictName = dictFile.getName(); 74 final String newDictName = newDictFile.getName(); 75 if (newDictFile.exists()) { 76 return false; 77 } 78 for (final File file : dictFile.listFiles()) { 79 if (!file.isFile()) { 80 continue; 81 } 82 final String fileName = file.getName(); 83 final String newFileName = fileName.replaceFirst( 84 Pattern.quote(dictName), Matcher.quoteReplacement(newDictName)); 85 if (!file.renameTo(new File(dictFile, newFileName))) { 86 return false; 87 } 88 } 89 return dictFile.renameTo(newDictFile); 90 } 91 return false; 92 } 93 94 @UsedForTesting createEmptyDictFile(final String filePath, final long dictVersion, final Locale locale, final Map<String, String> attributeMap)95 public static boolean createEmptyDictFile(final String filePath, final long dictVersion, 96 final Locale locale, final Map<String, String> attributeMap) { 97 final String[] keyArray = new String[attributeMap.size()]; 98 final String[] valueArray = new String[attributeMap.size()]; 99 int index = 0; 100 for (final String key : attributeMap.keySet()) { 101 keyArray[index] = key; 102 valueArray[index] = attributeMap.get(key); 103 index++; 104 } 105 return createEmptyDictFileNative(filePath, dictVersion, locale.toString(), keyArray, 106 valueArray); 107 } 108 calcNormalizedScore(final String before, final String after, final int score)109 public static float calcNormalizedScore(final String before, final String after, 110 final int score) { 111 return calcNormalizedScoreNative(StringUtils.toCodePointArray(before), 112 StringUtils.toCodePointArray(after), score); 113 } 114 115 /** 116 * Control the current time to be used in the native code. If currentTime >= 0, this method sets 117 * the current time and gets into test mode. 118 * In test mode, set timestamp is used as the current time in the native code. 119 * If currentTime < 0, quit the test mode and returns to using time() to get the current time. 120 * 121 * @param currentTime seconds since the unix epoch 122 * @return current time got in the native code. 123 */ 124 @UsedForTesting setCurrentTimeForTest(final int currentTime)125 public static int setCurrentTimeForTest(final int currentTime) { 126 return setCurrentTimeForTestNative(currentTime); 127 } 128 } 129