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 /** 31 * 32 */ 33 package org.openimaj.image.segmentation; 34 35 import java.util.List; 36 37 import org.openimaj.image.FImage; 38 import org.openimaj.image.connectedcomponent.ConnectedComponentLabeler; 39 import org.openimaj.image.connectedcomponent.ConnectedComponentLabeler.Algorithm; 40 import org.openimaj.image.pixel.ConnectedComponent; 41 import org.openimaj.image.pixel.ConnectedComponent.ConnectMode; 42 import org.openimaj.image.processor.Processor; 43 44 /** 45 * Simple wrapper to make thresholding algorithms into {@link Segmenter}s by 46 * applying the thresholding operation and then applying connected component 47 * labeling. This class will produce components for both the foreground and 48 * background elements of thresholded input image. 49 * 50 * @author Jonathon Hare (jsh2@ecs.soton.ac.uk) 51 * 52 */ 53 public class ConnectedThresholdSegmenter extends ThresholdSegmenter { 54 private ConnectMode mode; 55 private Algorithm algorithm; 56 57 /** 58 * Construct with the given thresholding algorithm implementation and 59 * connection mode. 60 * 61 * @param thresholder 62 * the thresholding algorithm 63 * @param mode 64 * the connection mode 65 */ 66 public ConnectedThresholdSegmenter(Processor<FImage> thresholder, ConnectMode mode) { 67 this(thresholder, ConnectedComponentLabeler.Algorithm.TWO_PASS, mode); 68 } 69 70 /** 71 * Construct with the given thresholding algorithm implementation. 72 * 73 * @param thresholder 74 * the thresholding algorithm 75 * @param algorithm 76 * the connected component labeling algorithm to use 77 * @param mode 78 * the connection mode 79 */ 80 public ConnectedThresholdSegmenter(Processor<FImage> thresholder, Algorithm algorithm, ConnectMode mode) { 81 super(thresholder); 82 this.mode = mode; 83 this.algorithm = algorithm; 84 } 85 86 @Override 87 public List<ConnectedComponent> segment(FImage image) { 88 final FImage timg = image.process(thresholder); 89 90 return algorithm.findComponents(timg, 0, mode); 91 } 92 }