Application: gvSIG desktop » gvSIG raster

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2017 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * 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 program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.raster.lib.legend.impl.operations.histogrammatching;

import org.gvsig.fmap.geom.exception.CreateEnvelopeException;

import org.gvsig.raster.lib.buffer.api.Band;

import org.gvsig.raster.lib.buffer.api.BufferLocator;

import org.gvsig.raster.lib.buffer.api.BufferManager;

import org.gvsig.raster.lib.buffer.api.NoData;

import org.gvsig.raster.lib.buffer.api.exceptions.BandException;

import org.gvsig.raster.lib.buffer.api.exceptions.BufferException;

import org.gvsig.raster.lib.buffer.api.exceptions.BufferOperationException;

import org.gvsig.raster.lib.buffer.api.operations.OperationFactory;

import org.gvsig.raster.lib.buffer.api.statistics.Statistics;

import org.gvsig.raster.lib.buffer.impl.exceptions.ProcessingOperationException;

import org.gvsig.raster.lib.buffer.spi.operations.AbstractOperation;

import org.gvsig.raster.lib.legend.api.colorinterpretation.ColorInterpretation;

import org.gvsig.tools.locator.LocatorException;

/**

 * @author fdiaz

 *

 */

public class HistogramMatchingOperation extends AbstractOperation{

    static public String COLOR_INTERPRETATION_PARAM = "color_interpretation";

    static public String STATISTICS_PARAM = "statistics";

    protected ColorInterpretation colorInterpretation;

    protected Statistics statistics;

    private RowProcessor rowProcessor;

    /**

     * @param factory

     *

     */

    public HistogramMatchingOperation(OperationFactory factory) {

        this.factory = factory;

    }

    @Override

    public void preProcess() throws BufferOperationException {

        BufferManager manager = BufferLocator.getBufferManager();

        colorInterpretation = (ColorInterpretation) this.parameters.getDynValue(COLOR_INTERPRETATION_PARAM);

        statistics = (Statistics)this.parameters.getDynValue(STATISTICS_PARAM);

        int bands = this.buffer.getBandCount();

        NoData[] noData = this.buffer.getBandNoData();

        try {

            this.outputBuffer = manager.createBuffer(

                this.buffer.getRows(),

                this.buffer.getColumns(),

                this.buffer.getBandTypes(),

                this.buffer.getBandNoData(),

                this.buffer.getProjection(),

                this.buffer.getEnvelope());

        } catch (LocatorException | BufferException | CreateEnvelopeException e) {

            throw new ProcessingOperationException(e);

        }

    }

    @Override

    public void process() throws ProcessingOperationException {

        for (int band=0; band<this.buffer.getBandCount(); band++){

            rowProcessor = new ByteRowProcessor(band);

            if (isRGBorGrayBand(band)) {

                Band bufferBand = this.buffer.getBand(band);

                Band outputBufferBand = this.outputBuffer.getBand(band);

                for (int row = 0; row < this.buffer.getRows(); row++) {

                    Object rowBuffer = bufferBand.createRowBuffer();

                    bufferBand.fetchRow(row, rowBuffer);

                    Object outputRowBuffer = outputBufferBand.createRowBuffer();

                    outputBufferBand.fetchRow(row, outputRowBuffer);

                    rowProcessor.processRow(rowBuffer, outputRowBuffer);

                    outputBufferBand.putRow(row, outputRowBuffer);

                }

            } else {

                try {

                    this.outputBuffer.getBand(band).copyFrom(this.buffer.getBand(band));

                } catch (BandException e) {

                    throw new ProcessingOperationException(e);

                }

            }

        }

    }

    private boolean isRGBorGrayBand(int band) {

        String bandColorInterpretation = colorInterpretation.get(band);

        return (bandColorInterpretation.equals(ColorInterpretation.RED_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.GREEN_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.BLUE_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.GRAY_BAND));

    }

    @Override

    public void postProcess() throws BufferOperationException {

        // TODO Auto-generated method stub

    }

    interface RowProcessor {

        void processRow(Object inputRow, Object outputRow);

        byte processValue(Object value);

    };

    private abstract class AbstractRowProcessor implements RowProcessor {

//        int band;

        int maxResult = 255;

        int minResult = 0;

        NoData noData;

        public AbstractRowProcessor(int band) {

//            this.band = band;

            noData = buffer.getBand(band).getNoData();

            if(noData.isDefined()) {

                minResult = (byte)1;

            }

        }

    }

    private class ByteRowProcessor extends AbstractRowProcessor {

        public ByteRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, Object outputRow) {

            byte[] inputByteRow = (byte[])inputRow;

            byte[] outputByteRow = (byte[])outputRow;

            for (int i = 0; i < inputByteRow.length; i++) {

                outputByteRow[i] = processValue(inputByteRow[i]);

            }

        }

        @Override

        public byte processValue(Object value) {

            if(noData.isDefined() && noData.getValue().equals(value)){

                return (byte)value;

            }

            int iValue = 0xFF & ((Byte) value).byteValue();

            int result = iValue; // + brightness;

            if(result>maxResult){

                result = maxResult;

            }

            if(result<minResult){

                result = minResult;

            }

            return (byte)result;

        }

    }

}

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2017 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * 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 program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.raster.lib.legend.impl.operations.histogrammatching;

import java.io.IOException;

import java.io.InputStream;

import java.util.Map;

import org.xmlpull.v1.XmlPullParserException;

import org.gvsig.raster.lib.buffer.api.Buffer;

import org.gvsig.raster.lib.buffer.api.BufferManager;

import org.gvsig.raster.lib.buffer.api.operations.Operation;

import org.gvsig.raster.lib.buffer.api.operations.OperationFactory;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.dynobject.DynObjectManager;

import org.gvsig.tools.dynobject.DynStruct;

/**

 * @author fdiaz

 *

 */

public class HistogramMatchingOperationFactory implements OperationFactory {

    public static String NAME="HistogramMatching";

    public static String DESCRIPTION="Histogram matching operation";//FIXME:

    private DynStruct definition;

    /**

     *

     */

    public HistogramMatchingOperationFactory() {

        // TODO Auto-generated constructor stub

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.lib.buffer.api.operations.OperationFactory#createParameters()

     */

    @Override

    public DynObject createParameters() {

        DynObjectManager manager = ToolsLocator.getDynObjectManager();

        if (definition == null) {

            definition = createParametersDefinition();

        }

        DynObject parameters = manager.createDynObject(definition);

        return parameters;

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.lib.buffer.api.operations.OperationFactory#create()

     */

    @Override

    public Operation create() {

        return new HistogramMatchingOperation(this);

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.lib.buffer.api.operations.OperationFactory#createParametersDefinition()

     */

    @Override

    public DynStruct createParametersDefinition() {

        return registerParametersDefinition("HistogramMatchingOperationParameters", "org/gvsig/raster/lib/legend/impl/operations/histogrammatching/HistogramMatchingOperationParameters.xml");

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.lib.buffer.api.operations.OperationFactory#getName()

     */

    @Override

    public String getName() {

        return NAME;

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.lib.buffer.api.operations.OperationFactory#getDescription()

     */

    @Override

    public String getDescription() {

        return DESCRIPTION;

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.lib.buffer.api.operations.OperationFactory#isApplicable(org.gvsig.raster.lib.buffer.api.Buffer)

     */

    @Override

    public Applicable isApplicable(Buffer buffer) {

        int bandsCount = buffer.getBandCount();

        int byteBandsCount = 0;

        for(int band = 0; band<bandsCount; band++){

            if (buffer.getBandTypes()[band]==BufferManager.TYPE_BYTE) {

                byteBandsCount++;

            }

        }

        if(byteBandsCount==bandsCount){

            return Applicable.YES;

        } else if (byteBandsCount==0) {

            return Applicable.NO;

        }

        return Applicable.UNKNOW;

    }

    /* (non-Javadoc)

     * @see org.gvsig.raster.lib.buffer.api.operations.OperationFactory#isApplicable(org.gvsig.raster.lib.buffer.api.operations.Operation)

     */

    @Override

    public Applicable isApplicable(Operation operation) {

        //FIXME:

        return Applicable.YES;

    }

    protected DynStruct registerParametersDefinition(String name, String filename) {

        DynStruct definition = null;

        DynObjectManager manager = ToolsLocator.getDynObjectManager();

        definition = manager.get(name);

        if ( definition == null) {

            InputStream stream = this.getClass().getResourceAsStream(filename);

            if( stream == null ) {

                throw new IllegalArgumentException("File '"+filename+"' not found");

            }

            Map<String, DynStruct> definitions = null;

            try {

                definitions = manager.importDynClassDefinitions(stream, this.getClass().getClassLoader());

            } catch (XmlPullParserException e) {

                throw new IllegalArgumentException("DynClass '"+name+"' can't parse file '"+filename+"'",e);

            } catch (IOException e) {

                throw new IllegalArgumentException("DynClass '"+name+"' can't read file '"+filename+"'",e);

            }

            definition = definitions.get(name);

            if ( definition == null) {

                throw new IllegalArgumentException("DynClass '"+name+"' not found in file '"+filename+"'");

            }

        }

        return definition;

    }

}

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2017 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * 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 program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.raster.lib.legend.impl.operations.histogrammatching;

import org.gvsig.fmap.geom.exception.CreateEnvelopeException;

import org.gvsig.raster.lib.buffer.api.Band;

import org.gvsig.raster.lib.buffer.api.BufferLocator;

import org.gvsig.raster.lib.buffer.api.BufferManager;

import org.gvsig.raster.lib.buffer.api.NoData;

import org.gvsig.raster.lib.buffer.api.exceptions.BandException;

import org.gvsig.raster.lib.buffer.api.exceptions.BufferException;

import org.gvsig.raster.lib.buffer.api.exceptions.BufferOperationException;

import org.gvsig.raster.lib.buffer.api.operations.OperationFactory;

import org.gvsig.raster.lib.buffer.api.statistics.Statistics;

import org.gvsig.raster.lib.buffer.impl.exceptions.ProcessingOperationException;

import org.gvsig.raster.lib.buffer.spi.operations.AbstractOperation;

import org.gvsig.raster.lib.legend.api.colorinterpretation.ColorInterpretation;

import org.gvsig.tools.locator.LocatorException;

/**

 * @author fdiaz

 *

 */

public class HistogramMatchingOperation extends AbstractOperation{

    static public String COLOR_INTERPRETATION_PARAM = "color_interpretation";

    static public String STATISTICS_PARAM = "statistics";

    protected ColorInterpretation colorInterpretation;

    protected Statistics statistics;

    private RowProcessor rowProcessor;

    /**

     * @param factory

     *

     */

    public HistogramMatchingOperation(OperationFactory factory) {

        this.factory = factory;

    }

    @Override

    public void preProcess() throws BufferOperationException {

        BufferManager manager = BufferLocator.getBufferManager();

        colorInterpretation = (ColorInterpretation) this.parameters.getDynValue(COLOR_INTERPRETATION_PARAM);

        statistics = (Statistics)this.parameters.getDynValue(STATISTICS_PARAM);

        int bands = this.buffer.getBandCount();

        NoData[] noData = this.buffer.getBandNoData();

        try {

            this.outputBuffer = manager.createBuffer(

                this.buffer.getRows(),

                this.buffer.getColumns(),

                this.buffer.getBandTypes(),

                this.buffer.getBandNoData(),

                this.buffer.getProjection(),

                this.buffer.getEnvelope());

        } catch (LocatorException | BufferException | CreateEnvelopeException e) {

            throw new ProcessingOperationException(e);

        }

    }

    @Override

    public void process() throws ProcessingOperationException {

        for (int band=0; band<this.buffer.getBandCount(); band++){

            rowProcessor = new ByteRowProcessor(band);

            if (isRGBorGrayBand(band)) {

                Band bufferBand = this.buffer.getBand(band);

                Band outputBufferBand = this.outputBuffer.getBand(band);

                for (int row = 0; row < this.buffer.getRows(); row++) {

                    Object rowBuffer = bufferBand.createRowBuffer();

                    bufferBand.fetchRow(row, rowBuffer);

                    Object outputRowBuffer = outputBufferBand.createRowBuffer();

                    outputBufferBand.fetchRow(row, outputRowBuffer);

                    rowProcessor.processRow(rowBuffer, outputRowBuffer);

                    outputBufferBand.putRow(row, outputRowBuffer);

                }

            } else {

                try {

                    this.outputBuffer.getBand(band).copyFrom(this.buffer.getBand(band));

                } catch (BandException e) {

                    throw new ProcessingOperationException(e);

                }

            }

        }

    }

    private boolean isRGBorGrayBand(int band) {

        String bandColorInterpretation = colorInterpretation.get(band);

        return (bandColorInterpretation.equals(ColorInterpretation.RED_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.GREEN_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.BLUE_BAND) ||

            bandColorInterpretation.equals(ColorInterpretation.GRAY_BAND));

    }

    @Override

    public void postProcess() throws BufferOperationException {

        // TODO Auto-generated method stub

    }

    interface RowProcessor {

        void processRow(Object inputRow, Object outputRow);

        byte processValue(Object value);

    };

    private abstract class AbstractRowProcessor implements RowProcessor {

//        int band;

        int maxResult = 255;

        int minResult = 0;

        NoData noData;

        public AbstractRowProcessor(int band) {

//            this.band = band;

            noData = buffer.getBand(band).getNoData();

            if(noData.isDefined()) {

                minResult = (byte)1;

            }

        }

    }

    private class ByteRowProcessor extends AbstractRowProcessor {

        public ByteRowProcessor(int band) {

            super(band);

        }

        @Override

        public void processRow(Object inputRow, Object outputRow) {

            byte[] inputByteRow = (byte[])inputRow;

            byte[] outputByteRow = (byte[])outputRow;

            for (int i = 0; i < inputByteRow.length; i++) {

                outputByteRow[i] = processValue(inputByteRow[i]);

            }

        }

        @Override

        public byte processValue(Object value) {

            if(noData.isDefined() && noData.getValue().equals(value)){

                return (byte)value;

            }

            int iValue = 0xFF & ((Byte) value).byteValue();

            int result = iValue; // + brightness;

            if(result>maxResult){

                result = maxResult;

            }

            if(result<minResult){

                result = minResult;

            }

            return (byte)result;

        }

    }

}