Application: gvSIG desktop » gvSIG raster

/* 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.mapcontext.raster.impl;

import java.awt.Graphics2D;

import java.awt.image.BufferedImage;

import java.util.Observable;

import java.util.Observer;

import java.util.Set;

import java.util.TreeSet;

import org.cresques.cts.ICoordTrans;

import org.slf4j.LoggerFactory;

import org.gvsig.compat.print.PrintAttributes;

import org.gvsig.fmap.dal.DataStore;

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

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

import org.gvsig.fmap.dal.raster.api.RasterQuery;

import org.gvsig.fmap.dal.raster.api.RasterSet;

import org.gvsig.fmap.dal.raster.api.RasterStore;

import org.gvsig.fmap.dal.raster.api.RasterStoreNotification;

import org.gvsig.fmap.geom.Geometry.SUBTYPES;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

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

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.fmap.mapcontext.ViewPort;

import org.gvsig.fmap.mapcontext.exceptions.LegendLayerException;

import org.gvsig.fmap.mapcontext.exceptions.LoadLayerException;

import org.gvsig.fmap.mapcontext.layers.FLyrDefault;

import org.gvsig.fmap.mapcontext.raster.api.RasterLayer;

import org.gvsig.fmap.mapcontext.raster.api.RasterLegendChangedListener;

import org.gvsig.metadata.exceptions.MetadataException;

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

import org.gvsig.raster.lib.legend.api.RasterLegend;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.dynobject.exception.DynMethodException;

import org.gvsig.tools.dynobject.exception.DynMethodNotSupportedException;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.task.Cancellable;

import org.gvsig.tools.task.SimpleTaskStatus;

import org.gvsig.tools.task.TaskStatusManager;

/**

 * Capa b?sica Raster.

 *

 */

public class DefaultRasterLayer extends FLyrDefault implements RasterLayer, RasterLegendChangedListener, Observer {

    final static private org.slf4j.Logger logger

    = LoggerFactory.getLogger(DefaultRasterLayer.class);

    private RasterStore store;

    private RasterLegend legend;

    //private Cancellable cancel;

    private RasterQuery baseQuery;

    /**

     * Creates a new DefaultRasterLayer

     * @throws LoadLayerException

     */

    public DefaultRasterLayer() throws LoadLayerException{

        super();

    }

    /**

     * Creates a new DefaultRasterLayer

     * @param store

     * @param legend

     */

    public DefaultRasterLayer(RasterStore store, RasterLegend legend) {

        super();

        ToolsLocator.getDisposableManager().bind(store);

        this.store = store;

        this.legend = legend;

    }

    public void setLegend(RasterLegend legend) throws LegendLayerException {

        if (this.legend == legend) {

            return;

        }

        if (this.legend != null && this.legend.equals(legend)) {

            return;

        }

        RasterLegend oldLegend = this.legend;

        this.legend = legend;

        if (oldLegend != null) {

            //oldLegend.removeLegendListener( this);

            //oldLegend.deleteDrawingObserver(this);

        }

        if (legend != null) {

            //this.legend.addDrawingObserver(this);

            //this.legend.addLegendListener( this);

        }

//        LegendChangedEvent e = LegendChangedEvent.createLegendChangedEvent(oldLegend, this.legend);

//        e.setLayer(this);

        updateDrawVersion();

//        callLegendChanged(e);

    }

    /**

     * Devuelve la Leyenda de la capa.

     *

     * @return Leyenda.

     */

    public RasterLegend getLegend() {

        return legend;

    }

    @Override

    public DataStore getDataStore() {

        return this.store;

    }

    @Override

    public void setDataStore(DataStore dataStore) throws LoadLayerException {

        if (this.store != null) {

            ToolsLocator.getDisposableManager().release(this.store);

            DisposeUtils.disposeQuietly(this.store);

        }

        this.store = (RasterStore) dataStore;

        ToolsLocator.getDisposableManager().bind(dataStore);

        initLegend();

    }

    /**

     * Initializes the legend if it is not initalized yet

     * @throws LegendLayerException

     */

    private void initLegend() throws LegendLayerException {

        RasterLegend legend = null;

        if (this.legend == null) {

            // Should be done like this but must be corrected mapContextManager

            // MapContextManager mapContextManager = MapContextLocator.getMapContextManager();

            // Set the legend

            // IRasterLegend legend = (IRasterLegend) mapContextManager.getLegend(store);

            // Should be moved to mapContextManager when RasterLegend implements

            // ILegend

            if (legend == null) {

                try {

                    legend = (RasterLegend) store.invokeDynMethod(RasterStore.DYNMETHOD_GETLEGEND_NAME, null);

                } catch (DynMethodNotSupportedException e) {

                    logger.debug("This store {} does not provide a legend.", store.getName());

                } catch (DynMethodException e) {

                    logger.error("Can't load the specific legend provided for the store {}.", store.getName(), e);

                }

            }

            if (legend == null) {

                throw new LegendLayerException(this.getName());

            }

            this.setLegend(legend);

        }

    }

    @Override

    public Envelope getFullEnvelope() throws ReadException {

        Envelope rAux;

        try {

            rAux = getRasterStore().getEnvelope();

        } catch (BaseException e) {

            throw new ReadException(getName(), e);

        }

        // Esto es para cuando se crea una capa nueva con el fullExtent de ancho

        // y alto 0.

        if (rAux == null || rAux.isEmpty() || rAux.getMaximum(0) - rAux.getMinimum(0) == 0

                && rAux.getMaximum(1) - rAux.getMinimum(1) == 0) {

            try {

                GeometryManager geomManager = GeometryLocator.getGeometryManager();

                rAux = geomManager.createEnvelope(0, 0, 90, 90, SUBTYPES.GEOM2D);

            } catch (CreateEnvelopeException e) {

                logger.error("Error creating the envelope", e);

                e.printStackTrace();

            }

        }

        // Si existe reproyecci?n, reproyectar el extent

        ICoordTrans ct = getCoordTrans();

        if (ct != null) {

            rAux = rAux.convert(ct);

        }

        return rAux;

    }

    @Override

    public void draw(BufferedImage image, Graphics2D g, ViewPort viewPort,

        Cancellable cancel, double scale) throws ReadException {

        if (legend == null) {

            return;

        }

        if (!this.isWithinScale(scale)) {

            return;

        }

        if (cancel.isCanceled()) {

            return;

        }

        RasterQuery rasterQuery = createRasterQuery();

        Envelope reprojectedViewPortEnvelope = (Envelope) viewPort.getEnvelope();

        ICoordTrans ct = getCoordTrans();

        if (ct != null) {

            try {

                reprojectedViewPortEnvelope = (Envelope) reprojectedViewPortEnvelope.clone();

                reprojectedViewPortEnvelope = reprojectedViewPortEnvelope.convert(ct.getInverted());

            } catch (CloneNotSupportedException e) {

                this.setAvailable(false);

                this.setError(e);

                throw new ReadException(getName(), e);

            }

        }

        double pixelSize = reprojectedViewPortEnvelope.getLength(0) / (double)viewPort.getImageWidth();

        rasterQuery.setPixelSize(pixelSize);

        ColorInterpretation colorInterpretation = legend.getColorInterpretation();

        String[] renderBands = colorInterpretation.getValues();//.buildRenderBands();

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

            int band = colorInterpretation.getBand(renderBands[i]);

            rasterQuery.addBand(this.store.createBandQuery(band));

        }

        RasterSet rasterSet = null;

        try {

            long tini = System.currentTimeMillis();

            //TODO Task status should be used to cancel

            TaskStatusManager manager = ToolsLocator.getTaskStatusManager();

            SimpleTaskStatus taskStatus = manager.createDefaultSimpleTaskStatus("Draw "+getDataStore().getName());

            rasterSet = getRasterStore().getRasterSet(rasterQuery);

            ((RasterLegend) legend).draw(g, rasterSet, viewPort, taskStatus);

            logger.debug("Layer " + this.getName() + " drawn in "

                    + (System.currentTimeMillis() - tini) + " milliseconds.");

        } catch (DataException e) {

            this.setAvailable(false);

            this.setError(e);

            throw new ReadException(getName(), e);

        } finally {

            if(rasterSet!=null){

                DisposeUtils.dispose(rasterSet);

                rasterSet = null;

            }

        }

    }

    @Override

    public void print(Graphics2D g, ViewPort viewPort, Cancellable cancel,

        double scale, PrintAttributes properties) throws ReadException {

        //TODO legend doesn't have print method

        throw new UnsupportedOperationException();

    }

    @Override

    public Set<RasterStore> getMetadataChildren() throws MetadataException {

        Set<RasterStore> ret = new TreeSet<RasterStore>();

        ret.add(this.store);

        return ret;

    }

    @Override

    public Object getMetadataID() throws MetadataException {

        return "Layer(" + this.getName() + "):"

            + this.store.getMetadataID();

    }

    @Override

    public void update(Observable observable, Object notification) {

        if (observable.equals(this.store)) {

            if (notification instanceof RasterStoreNotification) {

                RasterStoreNotification event

                        = (RasterStoreNotification) notification;

                if (event.getType() == RasterStoreNotification.AFTER_REFRESH

                        || event.getType() == RasterStoreNotification.SELECTION_CHANGE

                        ) {

                    this.updateDrawVersion();

                } else if (event.getType() == RasterStoreNotification.RESOURCE_CHANGED) {

                    this.setAvailable(false);

                }

            }

        }

    }

    @Override

    public RasterStore getRasterStore() {

        return this.store;

    }

    @Override

    public void setBaseQuery(RasterQuery baseQuery) {

        this.baseQuery = baseQuery;

    }

    @Override

    public RasterQuery getBaseQuery() {

        return this.baseQuery;

    }

    @Override

    public RasterQuery createRasterQuery() {

        if( this.baseQuery==null ) {

            return this.getRasterStore().createRasterQuery();

        }

        try {

            return (RasterQuery) baseQuery.clone();

        } catch (CloneNotSupportedException ex) {

            throw new RuntimeException(ex);

        }

    }

    @Override

    protected void doDispose() throws BaseException {

        DisposeUtils.dispose(this.store);

        this.store = null;

        this.legend = null;

        this.baseQuery = null;

    }

}