package ie.dcu.apps.ist;

import ie.dcu.segment.painters.*;

import java.util.*;

public class PainterRegistry {
	private final LinkedHashMap<String, SegmentationPainter> painters;

	public PainterRegistry() {
		painters = new LinkedHashMap<String, SegmentationPainter>();
		init();
	}
	
	
	private void init() {
		add(new SegmentPainter());
		add(new OriginalPainter());
		/*
		 * Commented different views of segmentation. Currently original and Segment painter are enabled
		 */
		//add(new MarkupPainter());
		//add(new MaskPainter());
		//add(new ForegroundOnlyPainter());
		//add(new OutlineOverlayPainter());
	}
	
	
	public void add(SegmentationPainter painter) {
		painters.put(painter.getName(), painter);
	}
	
	
	public SegmentationPainter get(String painter) {
		return painters.get(painter);
	}
	
	
	public Collection<SegmentationPainter> values() {
		return painters.values();
	}
	
	
	public void dispose() {
		for (SegmentationPainter p : values()) {
			p.dispose();
		}
	}
}