CompoundSelection.java

package org.chesmapper.view.cluster;

import java.util.HashMap;

import org.chesmapper.map.dataInterface.CompoundGroupWithProperties;
import org.chesmapper.map.dataInterface.CompoundProperty;
import org.chesmapper.map.dataInterface.NominalProperty;
import org.chesmapper.map.dataInterface.NumericProperty;
import org.mg.javalib.util.ArraySummary;
import org.mg.javalib.util.CountedSet;
import org.mg.javalib.util.DoubleArraySummary;

public class CompoundSelection implements CompoundGroupWithProperties
{
	private Compound[] compounds;
	HashMap<CompoundProperty, ArraySummary> summarys = new HashMap<CompoundProperty, ArraySummary>();
	HashMap<CompoundProperty, ArraySummary> formattedSummarys = new HashMap<CompoundProperty, ArraySummary>();

	public CompoundSelection(Compound[] c)
	{
		this.compounds = c;
	}

	@Override
	public String toString()
	{
		return "Selection of " + getNumCompounds() + " compounds";
	}

	public Compound[] getCompounds()
	{
		return compounds;
	}

	@Override
	public Double getDoubleValue(NumericProperty p)
	{
		if (!summarys.containsKey(p))
			updateNumeric(p);
		if (summarys.get(p).isAllNull())
			return null;
		return ((DoubleArraySummary) summarys.get(p)).getMean();
	}

	private void updateNumeric(NumericProperty p)
	{
		Double d[] = new Double[getNumCompounds()];
		int i = 0;
		for (Compound m : compounds)
			d[i++] = m.getDoubleValue(p);
		summarys.put(p, DoubleArraySummary.create(d));
	}

	private void updateNominal(NominalProperty p)
	{
		String s[] = new String[getNumCompounds()];
		int i = 0;
		for (Compound m : compounds)
			s[i++] = m.getStringValue(p);
		CountedSet<String> set = CountedSet.create(s);
		summarys.put(p, set);
		CountedSet<String> fSet = set.copy();
		for (String key : fSet.values())
			fSet.rename(key, p.getFormattedValue(key));
		fSet.setToBack(p.getFormattedNullValue());
		formattedSummarys.put(p, fSet);
	}

	@SuppressWarnings("unchecked")
	@Override
	public CountedSet<String> getNominalSummary(NominalProperty p)
	{
		if (!summarys.containsKey(p))
			updateNominal(p);
		return (CountedSet<String>) summarys.get(p);
	}

	@Override
	public String getFormattedValue(CompoundProperty p)
	{
		if (!summarys.containsKey(p))
			if (p instanceof NumericProperty)
				updateNumeric((NumericProperty) p);
			else
				updateNominal((NominalProperty) p);
		if (p instanceof NumericProperty)
		{
			if (summarys.get(p).isAllNull())
				return p.getFormattedNullValue();
			else
				return summarys.get(p).toString(false);
		}
		else
			return formattedSummarys.get(p).toString(false);
	}

	public int getNumCompounds()
	{
		return compounds.length;
	}

}