001/*
002 * Copyright (c) 2014-2015 The Regents of the University of California.
003 * All rights reserved.
004 *
005 * '$Author: crawl $'
006 * '$Date: 2015-04-27 18:47:47 +0000 (Mon, 27 Apr 2015) $' 
007 * '$Revision: 33374 $'
008 * 
009 * Permission is hereby granted, without written agreement and without
010 * license or royalty fees, to use, copy, modify, and distribute this
011 * software and its documentation for any purpose, provided that the above
012 * copyright notice and the following two paragraphs appear in all copies
013 * of this software.
014 *
015 * IN NO EVENT SHALL THE UNIVERSITY OF CALIFORNIA BE LIABLE TO ANY PARTY
016 * FOR DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES
017 * ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF
018 * THE UNIVERSITY OF CALIFORNIA HAS BEEN ADVISED OF THE POSSIBILITY OF
019 * SUCH DAMAGE.
020 *
021 * THE UNIVERSITY OF CALIFORNIA SPECIFICALLY DISCLAIMS ANY WARRANTIES,
022 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
023 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE SOFTWARE
024 * PROVIDED HEREUNDER IS ON AN "AS IS" BASIS, AND THE UNIVERSITY OF
025 * CALIFORNIA HAS NO OBLIGATION TO PROVIDE MAINTENANCE, SUPPORT, UPDATES,
026 * ENHANCEMENTS, OR MODIFICATIONS.
027 *
028 */
029package org.kepler.gis.actor;
030
031import org.geotools.coverage.grid.GridCoverage2D;
032import org.geotools.data.simple.SimpleFeatureCollection;
033import org.geotools.process.raster.PolygonExtractionProcess;
034import org.kepler.gis.data.RasterToken;
035import org.kepler.gis.data.VectorToken;
036import org.opengis.coverage.grid.GridCoverage;
037
038import ptolemy.actor.lib.Transformer;
039import ptolemy.kernel.CompositeEntity;
040import ptolemy.kernel.util.IllegalActionException;
041import ptolemy.kernel.util.NameDuplicationException;
042
043/** An actor that converts a raster to a feature data set
044 *  by extracting the polygons.
045 *  
046 *  @author Daniel Crawl
047 *  @version $Id: ExtractPolygons.java 33374 2015-04-27 18:47:47Z crawl $
048 */
049public class ExtractPolygons extends Transformer {
050
051    public ExtractPolygons(CompositeEntity container, String name)
052            throws IllegalActionException, NameDuplicationException {
053        super(container, name);
054        input.setTypeEquals(RasterToken.RASTER);
055        output.setTypeEquals(VectorToken.VECTOR);
056    }
057
058    @Override
059    public void fire() throws IllegalActionException {
060        
061        super.fire();
062
063        PolygonExtractionProcess process = new PolygonExtractionProcess();
064        
065        GridCoverage coverage = ((RasterToken)input.get(0)).coverageValue();
066
067        if(!(coverage instanceof GridCoverage2D)) {
068            throw new IllegalActionException(this, "Input raster is not 2d.");
069        }
070        
071        SimpleFeatureCollection features = 
072                    process.execute((GridCoverage2D) coverage,
073                            null, false, null, null, null, null);
074        
075        output.broadcast(new VectorToken(features));
076    }
077}