RasterClassifierDescriptor.java

/* JAI-Ext - OpenSource Java Advanced Image Extensions Library
 *    http://www.geo-solutions.it/
 *    Copyright 2014 GeoSolutions


 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at

 * http://www.apache.org/licenses/LICENSE-2.0

 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package it.geosolutions.jaiext.classifier;

import it.geosolutions.jaiext.piecewise.Domain1D;
import it.geosolutions.jaiext.range.Range;

import java.awt.RenderingHints;
import java.awt.image.RenderedImage;
import java.awt.image.renderable.ParameterBlock;

import javax.media.jai.JAI;
import javax.media.jai.OperationDescriptorImpl;
import javax.media.jai.ParameterBlockJAI;
import javax.media.jai.ROI;
import javax.media.jai.RenderedOp;
import javax.media.jai.registry.RenderedRegistryMode;

/**
 * {@link OperationDescriptorImpl} describing the RasterClassifier operation
 * 
 * @author Nicola Lagomarsini geosolutions
 * 
 */
public class RasterClassifierDescriptor extends OperationDescriptorImpl {

    /**
     * UID
     */
    private static final long serialVersionUID = 7954257625240335874L;

    /**
     * Construct the descriptor.
     */
    public RasterClassifierDescriptor() {
        super(new String[][] { { "GlobalName", RasterClassifierOpImage.OPERATION_NAME },
                { "LocalName", RasterClassifierOpImage.OPERATION_NAME },
                { "Vendor", "it.geosolutions.jaiext" },
                { "Description", "Transformation from sample to geophysics values" },
                { "DocURL", "" }, { "Version", "1.0" } },
                new String[] { RenderedRegistryMode.MODE_NAME }, 1, new String[] { "Domain1D",
                        "bandIndex", "roi", "nodata" }, // Argument names
                new Class[] { ColorMapTransform.class, Integer.class, javax.media.jai.ROI.class,
                        it.geosolutions.jaiext.range.Range.class }, // Argument classes
                new Object[] { NO_PARAMETER_DEFAULT, Integer.valueOf(-1), null, null }, // Default values for parameters,
                null // No restriction on valid parameter values.
        );
    }

    /**
     * Returns {@code true} if the parameters are valids. This implementation check that the number of bands in the source image is equals to the
     * number of supplied sample dimensions, and that all sample dimensions has pieces.
     * 
     * @param modeName The mode name (usually "Rendered").
     * @param param The parameter block for the operation to performs.
     * @param message A buffer for formatting an error message if any.
     */
    protected boolean validateParameters(final String modeName, final ParameterBlock param,
            final StringBuffer message) {
        if (!super.validateParameters(modeName, param, message)) {
            return false;
        }
        final RenderedImage source = (RenderedImage) param.getSource(0);
        final Domain1D lic = (Domain1D) param.getObjectParameter(0);
        if (lic == null)
            return false;
        final int numBands = source.getSampleModel().getNumBands();
        final int bandIndex = param.getIntParameter(1);
        if (bandIndex == -1)
            return true;
        if (bandIndex < 0 || bandIndex >= numBands) {
            return false;
        }
        return true;
    }

    public RenderedOp create(RenderedImage source0, ColorMapTransform domain1D, Integer bandIndex,
            ROI roi, Range nodata, RenderingHints hints) {
        ParameterBlockJAI pb = new ParameterBlockJAI(RasterClassifierOpImage.OPERATION_NAME,
                RenderedRegistryMode.MODE_NAME);
        // Setting source
        pb.setSource(source0, 0);
        // Setting parameters
        pb.setParameter("Domain1D", domain1D);
        pb.setParameter("bandIndex", bandIndex);
        pb.setParameter("roi", roi);
        pb.setParameter("nodata", nodata);

        return JAI.create(RasterClassifierOpImage.OPERATION_NAME, pb, hints);
    }
}