Application: gvSIG desktop

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2016 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.fmap.dal.file.bsq;

import java.io.DataOutputStream;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.nio.ByteOrder;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.io.IOUtils;

import org.gvsig.basicformats.FormatsFile;

import org.gvsig.basicformats.HDRFile;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.NewDataStoreParameters;

import org.gvsig.fmap.dal.exception.CreateException;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.raster.NewRasterStoreParameters;

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemStoreParameters;

import org.gvsig.fmap.dal.serverexplorer.filesystem.impl.AbstractFilesystemServerExplorerProvider;

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

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

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

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

public class BSQFileSystemServerProvider extends AbstractFilesystemServerExplorerProvider {

    protected String[] fileExtensions;

    public BSQFileSystemServerProvider(

            String name,

            String description,

            String[] fileExtensions

    ) {

        super(name, description);

        this.fileExtensions = fileExtensions;

    }

    @Override

    public boolean canCreate() {

        return true;

    }

    @Override

    public NewDataStoreParameters getCreateParameters() throws DataException {

        throw new UnsupportedOperationException();

    }

    @Override

    public void create(NewDataStoreParameters parameters, boolean overwrite) throws CreateException {

        FileOutputStream fos = null;

        try {

            Buffer buffer = ((NewRasterStoreParameters) parameters).getBuffer();

            if (!buffer.areAllBandsOfTheSameType()) {

                throw new IllegalArgumentException("The BSQ format does not support bands of different types.");

            }

            BufferDimensions dimensions = buffer.getDimensions();

            File file = ((FilesystemStoreParameters) parameters).getFile();

            fos = new FileOutputStream(file);

            DataOutputStream dos = new DataOutputStream(fos);

            HDRFile hdr = FormatsFile.createHDRFile();

            hdr.setDataType(buffer.getBandTypes()[0]);

            hdr.setNBands(buffer.getBandCount());

            hdr.setByteOrder(ByteOrder.nativeOrder());

            hdr.setDimensions(

                    dimensions.getRows(),

                    dimensions.getColumns(),

                    dimensions.getEnvelope()

            );

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

                writeBand(dos, buffer.getBand(nband));

            }

            hdr.write(file);

        } catch (Exception ex) {

            throw new CreateException(this.getName(), ex);

        } finally {

            IOUtils.closeQuietly(fos);

        }

    }

    void writeBand(DataOutputStream dos, Band band) throws IOException {

        int dataType = band.getDataType();

        Object rowBuffer = band.createRowBuffer();

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

            band.fetchRow(row, rowBuffer);

            switch (dataType) {

                case BufferManager.TYPE_BYTE:

                    byte[] byteBuffer = (byte[]) rowBuffer;

                    for (int n = 0; n < byteBuffer.length; n++) {

                        dos.writeByte(byteBuffer[n]);

                    }

                    break;

                case BufferManager.TYPE_SHORT:

                case BufferManager.TYPE_USHORT:

                    short[] shortBuffer = (short[]) rowBuffer;

                    for (int n = 0; n < shortBuffer.length; n++) {

                        dos.writeShort(shortBuffer[n]);

                    }

                    break;

                case BufferManager.TYPE_INT:

                    int[] intBuffer = (int[]) rowBuffer;

                    for (int n = 0; n < intBuffer.length; n++) {

                        dos.writeInt(intBuffer[n]);

                    }

                    break;

                case BufferManager.TYPE_FLOAT:

                    float[] floatBuffer = (float[]) rowBuffer;

                    for (int n = 0; n < floatBuffer.length; n++) {

                        dos.writeFloat(floatBuffer[n]);

                    }

                    break;

                case BufferManager.TYPE_DOUBLE:

                    double[] doubleBuffer = (double[]) rowBuffer;

                    for (int n = 0; n < doubleBuffer.length; n++) {

                        dos.writeDouble(doubleBuffer[n]);

                    }

                    break;

                default:

                    throw new IllegalArgumentException("Data Type " + dataType + " not supported.");

            }

        }

    }

    @Override

    public int getMode() {

        return DataServerExplorer.MODE_RASTER;

    }

    @Override

    public boolean accept(File pathname) {

        return FilenameUtils.isExtension(pathname.getName().toLowerCase(), this.fileExtensions);

    }

}