org.nlpcn.commons.lang.util.MapCount Maven / Gradle / Ivy

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

import java.io.Serializable;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;

/**
 * 用map做的计数器.
 * 
 * @param 
 * @author ansj
 */
public class MapCount implements Serializable {
	private static final long serialVersionUID = 1L;
	private HashMap hm = null;

	public MapCount() {
		hm = new HashMap();
	}

	public MapCount(HashMap hm) {
		this.hm = hm;
	}

	public MapCount(int initialCapacity) {
		hm = new HashMap(initialCapacity);
	}

	/**
	 * 增加一个元素
	 * 
	 * @param t
	 * @param n
	 */
	public void add(T t, double n) {
		Double value = null;
		if ((value = hm.get(t)) != null) {
			hm.put(t, value + n);
		} else {
			hm.put(t, Double.valueOf(n));
		}
	}

	/**
	 * 兼容旧的api
	 * 
	 * @param t
	 * @param n
	 */
	public void add(T t, int n) {
		add(t, (double) n);
	}

	/**
	 * 计数增加.默认为1
	 * 
	 * @param t
	 */
	public void add(T t) {
		this.add(t, 1);
	}

	/**
	 * map的大小
	 * 
	 * @return
	 */
	public int size() {
		return hm.size();
	}

	/**
	 * 删除一个元素
	 * 
	 * @param t
	 */
	public void remove(T t) {
		hm.remove(t);
	}

	/**
	 * 得道内部的map
	 * 
	 * @return
	 */
	public HashMap get() {
		return this.hm;
	}

	/**
	 * 将map序列化为词典格式
	 * 
	 * @return
	 */
	public String getDic() {
		Iterator> iterator = this.hm.entrySet().iterator();
		StringBuilder sb = new StringBuilder();
		Entry next = null;
		while (iterator.hasNext()) {
			next = iterator.next();
			sb.append(next.getKey());
			sb.append("\t");
			sb.append(next.getValue());
			sb.append("\n");
		}
		return sb.toString();
	}

	/**
	 * 批量增加
	 * 
	 * @param hs
	 */
	public void addAll(Collection collection) {
		for (T t : collection) {
			this.add(t);
		}
	}

	/**
	 * 批量增加
	 * 
	 * @param hs
	 */
	public void addAll(Collection collection, double weight) {
		for (T t : collection) {
			this.add(t, weight);
		}
	}

	/**
	 * 批量增加
	 * 
	 * @param hs
	 */
	public void addAll(Map map) {
		for (Entry e : map.entrySet()) {
			this.add(e.getKey(), e.getValue());
		}
	}
}