WatershedPixel.java

Go to the documentation of this file.

package watershed;

import java.util.ArrayList;

/*
 * Watershed algorithm
 *
 * Copyright (c) 2003 by Christopher Mei (christopher.mei@sophia.inria.fr)
 *
 * This plugin is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this plugin; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */


public class WatershedPixel implements Comparable {
        final static int INIT = -1;

        final static int MASK = -2;

        final static int WSHED = 0;

        final static int FICTITIOUS = -3;

        private int x;
        private int y;
        private short height;
        private int label;
        private int dist;

        private ArrayList neighbours;

        public WatershedPixel(int x, int y, short height) {
                this.x = x;
                this.y = y;
                this.height = height;
                label = INIT;
                dist = 0;
                neighbours = new ArrayList(8);
        }

        public WatershedPixel() {
                label = FICTITIOUS;
        }

        public void addNeighbour(WatershedPixel neighbour) {
                neighbours.add(neighbour);
        }

        public ArrayList getNeighbours() {
                return neighbours;
        }

        public final short getHeight() {
                return height;
        }

        public final int getIntHeight() {
                return height & 0xff;
        }

        public final int getX() {
                return x;
        }

        public final int getY() {
                return y;
        }

        public int compareTo(Object o) {
                WatershedPixel obj = (WatershedPixel) o;
                return (getIntHeight()) - (obj.getIntHeight());
        }

        public void setLabel(int label) {
                this.label = label;
        }

        public void setLabelToINIT() {
                label = INIT;
        }

        public void setLabelToMASK() {
                label = MASK;
        }

        public void setLabelToWSHED() {
                label = WSHED;
        }

        public boolean isLabelINIT() {
                return label == INIT;
        }

        public boolean isLabelMASK() {
                return label == MASK;
        }

        public boolean isLabelWSHED() {
                return label == WSHED;
        }

        public int getLabel() {
                return label;
        }

        public void setDistance(int distance) {
                dist = distance;
        }

        public int getDistance() {
                return dist;
        }

        public boolean isFICTITIOUS() {
                return label == FICTITIOUS;
        }

        public boolean allNeighboursAreWSHED() {
                for (int i = 0; i < neighbours.size(); i++) {
                        WatershedPixel r = (WatershedPixel) neighbours.get(i);

                        if (!r.isLabelWSHED())
                                return false;
                }
                return true;
        }
}