Application: gvSIG desktop

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2018 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.buffer.spi.operations;

import java.util.ArrayList;

import java.util.List;

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

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

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.operations.SpecifiedBandsOperation;

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

/**

 * @author fdiaz

 *

 */

public abstract class AbstractSpecifiedBandsOperation extends AbstractOperation implements SpecifiedBandsOperation {

    private List<Integer> bandsToProcess;

    /**

     *

     * @param factory

     */

    public AbstractSpecifiedBandsOperation(OperationFactory factory) {

        super(factory);

    }

    public List<Integer> getBandsToProcess() {

        return this.bandsToProcess;

    }

    @SuppressWarnings("unchecked")

    @Override

    public void preProcess() throws BufferOperationException {

        super.preProcess();

        if(this.getParameters().getDynClass().getDynField(BANDS_TO_PROCESS_PARAM)!=null) {

            bandsToProcess = (List<Integer>)this.getParameters().getDynValue(BANDS_TO_PROCESS_PARAM);

        }

        if(this.bandsToProcess==null) {

            bandsToProcess = new ArrayList<>(this.getInputBuffer().getBandCount());

        }

        if(this.bandsToProcess.isEmpty()){

            for (int i=0; i<this.getInputBuffer().getBandCount(); i++){

                bandsToProcess.add(i);

            }

        }

    }

    /**

     * @param band

     * @return

     */

    protected boolean isProcessableBand(int band) {

        return bandsToProcess.contains(band);

    }

    @Override

    public void process() throws ProcessingOperationException {

        super.process();

    }

    @Override

    public void postProcess() throws BufferOperationException {

        super.postProcess();

    }

    /**

     * This is a utility method that return a copy of the NoData of 

     * the processable band of the input buffer.

     * 

     * @return List of NoDatas

     */

    protected List<NoData> getProcessableBandNoDatas() {

        int bands = this.getInputBuffer().getBandCount();

        List<NoData> noDatas = new ArrayList<>();

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

            if (this.isProcessableBand(band)) {

                noDatas.add(this.getInputBuffer().getBandNoData()[band]);

            }

        }

        return noDatas;

    }

    /**

     * This is a utility method that return a copy of the NoData of 

     * the processable band of the input buffer.

     * 

     * @return array of NoDatas

     */

    protected NoData[] getProcessableBandNoDatasAsArray() {

        List<NoData> l = this.getProcessableBandNoDatas();

        NoData[] noDatas = l.toArray(new NoData[l.size()]);

        return noDatas;

    }

    /**

     * This is a utility method that return a copy of the types of 

     * the processable band of the input buffer.

     * 

     * @return list of types

     */

    protected List<Integer> getProcessableBandTypes() {

        int bands = this.getInputBuffer().getBandCount();

        List<Integer> types = new ArrayList<>();

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

            if (this.isProcessableBand(band)) {

                types.add(this.getInputBuffer().getBandTypes()[band]);

            }

        }

        return types;

    }

    /**

     * This is a utility method that return a copy of the types of 

     * the processable band of the input buffer.

     * 

     * @return Array of types

     */

    protected int[] getProcessableBandTypesAsArray() {

        int[] types = this.getTypesAsArray(this.getProcessableBandTypes());

        return types;

    }

    protected int[] getTypesAsArray(List<Integer> theTypes) {

        int[] types = new int[theTypes.size()];

        int n = 0;

        for (Integer type : theTypes) {

            types[n++] = type ;

        }

        return types;

    }

}