package homework3;

import java.util.Comparator;
import java.util.Map;

public class IncidentComparator implements Comparator {

	public int compare(Object val0, Object val1) {
		Integer count0 = null;
		Integer count1 = null;
		
		Map.Entry m0 = (Map.Entry) val0;
		if (m0.getValue() instanceof IncidentEntry) {
			IncidentEntry ie = (IncidentEntry) m0.getValue();
			count0 = ie.getUcrCount();
		} else {
			count0 = (Integer) m0.getValue();
		}
			
		Map.Entry m1 = (Map.Entry) val1;
		if (m1.getValue() instanceof IncidentEntry) {
			IncidentEntry ie = (IncidentEntry) m1.getValue();
			count1 = ie.getUcrCount();
		} else {
			count1 = (Integer) m1.getValue();
		}
		
		return -1*count0.compareTo(count1);
	}

}

