CoverageTree.java

package hudson.plugins.cobertura.targets;

import java.io.Serializable;
import java.util.Map;
import java.util.Map.Entry;

import hudson.plugins.cobertura.Ratio;

import org.kohsuke.stapler.export.Exported;
import org.kohsuke.stapler.export.ExportedBean;

@ExportedBean
public class CoverageTree implements Serializable {

	/**
	 * Generated 
	 */
	private static final long serialVersionUID = 5112467356061418891L;
	
	private Map<CoverageMetric, Ratio> aggregateResults;
	private Map<String, CoverageResult> children;
	private String name;

	public CoverageTree(String name, Map<CoverageMetric, Ratio> aggregateResults,
			Map<String, CoverageResult> children) {
		this.name=name;
		this.aggregateResults=aggregateResults;
		this.children=children;
	}
	
	@Exported
	public String getName(){
		return name;
	}
	
	@Exported
	public CoverageTreeElement[] getElements(){
		CoverageTreeElement[] cte=new CoverageTreeElement[aggregateResults.size()];
		int current=0;
		for (Entry<CoverageMetric, Ratio> e: aggregateResults.entrySet()){
			cte[current]=new CoverageTreeElement(e.getKey(), e.getValue());
			current++;
		}
		return cte;
	}

	@Exported
	public CoverageTree[] getChildren(){
		CoverageTree[] ct=new CoverageTree[children.size()];
		int current=0;
		for (Entry<String, CoverageResult> e: children.entrySet()){
			ct[current]=new CoverageTree(e.getKey(), e.getValue().getResults(), e.getValue().getChildrenReal());
			current++;
		}
		return ct;
	}
}