/* * GeoTools - The Open Source Java GIS Toolkit * http://geotools.org * * (C) 2014, Open Source Geospatial Foundation (OSGeo) * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; * version 2.1 of the License. * * This library 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 * Lesser General Public License for more details. */ package org.geotools.process.spatialstatistics; import java.util.HashMap; import java.util.Map; import java.util.logging.Level; import java.util.logging.Logger; import org.geotools.data.DataUtilities; import org.geotools.data.simple.SimpleFeatureCollection; import org.geotools.process.Process; import org.geotools.process.ProcessException; import org.geotools.process.ProcessFactory; import org.geotools.process.spatialstatistics.core.Params; import org.geotools.process.spatialstatistics.transformation.FieldCalculationFeatureCollection; import org.geotools.util.logging.Logging; import org.opengis.filter.expression.Expression; import org.opengis.util.ProgressListener; /** * Calculates field values using filter expression. * * @author Minpa Lee, MangoSystem * * @source $URL$ */ public class CalculateFieldProcess extends AbstractStatisticsProcess { protected static final Logger LOGGER = Logging.getLogger(CalculateFieldProcess.class); public CalculateFieldProcess(ProcessFactory factory) { super(factory); } public ProcessFactory getFactory() { return factory; } public static SimpleFeatureCollection process(SimpleFeatureCollection inputFeatures, Expression expression, String fieldName, ProgressListener monitor) { Map<String, Object> map = new HashMap<String, Object>(); map.put(CalculateFieldProcessFactory.inputFeatures.key, inputFeatures); map.put(CalculateFieldProcessFactory.expression.key, expression); map.put(CalculateFieldProcessFactory.fieldName.key, fieldName); Process process = new CalculateFieldProcess(null); Map<String, Object> resultMap; try { resultMap = process.execute(map, monitor); return (SimpleFeatureCollection) resultMap.get(CalculateFieldProcessFactory.RESULT.key); } catch (ProcessException e) { LOGGER.log(Level.FINER, e.getMessage(), e); } return null; } @Override public Map<String, Object> execute(Map<String, Object> input, ProgressListener monitor) throws ProcessException { SimpleFeatureCollection inputFeatures = (SimpleFeatureCollection) Params.getValue(input, CalculateFieldProcessFactory.inputFeatures, null); Expression expression = (Expression) Params.getValue(input, CalculateFieldProcessFactory.expression, null); String fieldName = (String) Params.getValue(input, CalculateFieldProcessFactory.fieldName, CalculateFieldProcessFactory.fieldName.sample); if (inputFeatures == null || expression == null) { throw new NullPointerException("All parameters required"); } // start process SimpleFeatureCollection resultFc = DataUtilities .simple(new FieldCalculationFeatureCollection(inputFeatures, expression, fieldName)); // end process Map<String, Object> resultMap = new HashMap<String, Object>(); resultMap.put(CalculateFieldProcessFactory.RESULT.key, resultFc); return resultMap; } }