1 /** 2 * Copyright (c) 2011, The University of Southampton and the individual contributors. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without modification, 6 * are permitted provided that the following conditions are met: 7 * 8 * * Redistributions of source code must retain the above copyright notice, 9 * this list of conditions and the following disclaimer. 10 * 11 * * Redistributions in binary form must reproduce the above copyright notice, 12 * this list of conditions and the following disclaimer in the documentation 13 * and/or other materials provided with the distribution. 14 * 15 * * Neither the name of the University of Southampton nor the names of its 16 * contributors may be used to endorse or promote products derived from this 17 * software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 21 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 22 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR 23 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 24 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 25 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON 26 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 28 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 29 */ 30 package org.openimaj.image.contour; 31 32 import java.util.ArrayList; 33 import java.util.List; 34 35 import org.openimaj.image.MBFImage; 36 import org.openimaj.image.colour.RGBColour; 37 import org.openimaj.image.renderer.MBFImageRenderer; 38 39 /** 40 * {@link MBFImageRenderer} for drawing {@link Contour}s. 41 * 42 * @author Sina Samnagooei (ss@ecs.soton.ac.uk) 43 * 44 */ 45 public class ContourRenderer extends MBFImageRenderer { 46 47 /** 48 * Construct with the given target image. 49 * 50 * @param targetImage 51 * the target image 52 */ 53 public ContourRenderer(MBFImage targetImage) { 54 super(targetImage); 55 } 56 57 /** 58 * Utility function to quickly draw the contour to the image. 59 * 60 * @param imgC 61 * the target image 62 * @param root 63 * the contour 64 * @return the target image 65 */ 66 public static MBFImage drawContours(MBFImage imgC, Contour root) { 67 new ContourRenderer(imgC).drawContours(root); 68 return imgC; 69 } 70 71 /** 72 * Draw the given contour. 73 * 74 * @param root 75 * the contour. 76 */ 77 public void drawContours(Contour root) { 78 final List<Contour> toDraw = new ArrayList<Contour>(); 79 toDraw.add(root); 80 while (!toDraw.isEmpty()) { 81 final Contour next = toDraw.remove(toDraw.size() - 1); 82 Float[] c = null; 83 switch (next.type) { 84 case HOLE: 85 c = RGBColour.BLUE; 86 break; 87 case OUTER: 88 c = RGBColour.RED; 89 break; 90 } 91 this.drawShape(next, 3, c); 92 93 toDraw.addAll(next.children); 94 } 95 } 96 }