org.nlpcn.commons.lang.dat.DoubleArrayTire Maven / Gradle / Ivy

Go to download
package org.nlpcn.commons.lang.dat;

import org.nlpcn.commons.lang.util.FileIterator;
import org.nlpcn.commons.lang.util.IOUtil;
import org.nlpcn.commons.lang.util.StringUtil;

import java.io.*;

/**
 * 双数组使用
 *
 * @author ansj
 */
public class DoubleArrayTire {

	private Item[] dat;

	public int arrayLength;

	private DoubleArrayTire() {
	}

	/**
	 * 获得dat数组
	 */
	public Item[] getDAT() {
		return dat;
	}

	/**
	 * 一个词在词典中的id
	 */
	public int getId(String str) {
		final Item item = getItem(str);
		return item != null ? item.index : 0;
	}

	/**
	 * 获得一个词语的item
	 */
	@SuppressWarnings("unchecked")
	public  T getItem(String str) {
		if (StringUtil.isBlank(str)) {
			return null;
		}
		if (str.length() == 1) {
			return (T) dat[str.charAt(0)];
		}

		Item item = dat[str.charAt(0)];
		if (item == null) {
			return null;
		}
		for (int i = 1; i < str.length(); i++) {
			final int checkValue = item.index;
			if (item.base + str.charAt(i) > dat.length - 1)
				return null;

			item = dat[item.base + str.charAt(i)];
			if (item == null) {
				return null;
			}
			if (item.check != -1 && item.check != checkValue) {
				return null;
			}
		}
		return (T) item;
	}

	@SuppressWarnings("unchecked")
	public  T getItem(int id) {
		return (T) dat[id];
	}

	public static DoubleArrayTire load(final String filePath) throws FileNotFoundException, IOException, ClassNotFoundException {
		ObjectInputStream ois = null;
		try {
			ois = new ObjectInputStream(new BufferedInputStream(new FileInputStream(filePath)));
			final DoubleArrayTire instance = new DoubleArrayTire();
			instance.dat = new Item[ois.readInt()];
			instance.arrayLength = ois.readInt();
			for (int i = 0; i < instance.arrayLength; i++) {
				final Item item = (Item) ois.readObject();
				instance.dat[item.index] = item;
			}
			return instance;
		} finally {
			if (ois != null) {
				ois.close();
			}
		}
	}

	/**
	 * 从文本中加载模型
	 * 
	 * @throws IllegalAccessException
	 * @throws InstantiationException
	 * @throws FileNotFoundException
	 */
	public static DoubleArrayTire loadText(String filePath, Class cla) throws FileNotFoundException, InstantiationException,
			IllegalAccessException {
		return loadText(IOUtil.getInputStream(filePath), cla);
	}

	/**
	 * 从文本中加载模型
	 * 
	 * @throws FileNotFoundException
	 * @throws IllegalAccessException
	 * @throws InstantiationException
	 */
	public static DoubleArrayTire loadText(InputStream is, Class cla) throws FileNotFoundException, InstantiationException,
			IllegalAccessException {
		final DoubleArrayTire obj = new DoubleArrayTire();
		final FileIterator it = IOUtil.instanceFileIterator(is, IOUtil.UTF8);
		if (it == null) {
			throw new FileNotFoundException();
		}
		String temp = it.next();
		obj.arrayLength = Integer.parseInt(temp);
		obj.dat = new Item[obj.arrayLength];
		while (it.hasNext()) {
			temp = it.next();
			final Item item = cla.newInstance();
			item.initValue(temp.split("\t"));
			obj.dat[item.index] = item;
		}
		return obj;
	}

	/**
	 * 从文本中加载模型
	 * 
	 * @throws IllegalAccessException
	 * @throws InstantiationException
	 * @throws FileNotFoundException
	 */
	public static DoubleArrayTire loadText(String filePath) throws FileNotFoundException, InstantiationException, IllegalAccessException {
		return loadText(filePath, BasicItem.class);
	}
}