Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

* 

* 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.

* 

*/

package org.gvsig.raster.fmap.roi;

import java.awt.Color;

import java.io.File;

import java.util.ArrayList;

import java.util.HashMap;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.coverage.exception.FileNotExistsException;

import org.gvsig.fmap.dal.coverage.grid.Grid;

import org.gvsig.fmap.dal.coverage.grid.ROI;

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

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

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

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.dal.store.shp.SHPStoreParameters;

import org.gvsig.fmap.dal.store.shp.SHPStoreProvider;

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

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

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.tools.dataTypes.DataTypes;

import org.gvsig.tools.dispose.DisposableIterator;

public class VectorialROIsReader {

        private FLyrVect               fLyrVect     = null;

        private HashMap<String, ROI>   rois         = null;

        private Grid                   grid         = null;

        @SuppressWarnings("deprecation")

        public VectorialROIsReader(String filename, Grid grid, IProjection projection) throws LoadLayerException, FileNotExistsException {

                this.grid = grid;

                File file = new File(filename);

                if(file.exists()) {

                        DataManager datamanager = DALLocator.getDataManager();

                        SHPStoreParameters params = null;

                        try {

                                params = (SHPStoreParameters)datamanager.createStoreParameters(SHPStoreProvider.NAME);

                                params.setCRS(projection);

                        } catch (InitializeException e) {

                                throw new LoadLayerException(file.getName(),e);

                        } catch (ProviderNotRegisteredException e) {

                                throw new LoadLayerException(file.getName(),e);

                        }

                        params.setFile(file);

                        fLyrVect = (FLyrVect)LayerFactory.getInstance().createLayer("layer1", params);

                } else

                        throw new FileNotExistsException("file not found");

        }

        @SuppressWarnings("deprecation")

        public ArrayList<ROI> read(ArrayList<ROI> existingROIs) throws InvalidROIsShpException, DataException{

                FeatureStore featureStore = fLyrVect.getFeatureStore();

                // Validaci?n del .shp:

                FeatureType featureType=featureStore.getDefaultFeatureType();

                int nameFieldIndex = featureType.getIndex("name");

                int rFiledIndex = featureType.getIndex("R");

                int gFiledIndex = featureType.getIndex("G");

                int bFiledIndex = featureType.getIndex("B");

                int typeFieldIndex = featureType.getAttributeDescriptor(nameFieldIndex).getType();

                int typerFiledIndex = featureType.getAttributeDescriptor(rFiledIndex).getType();

                int typegFiledIndex = featureType.getAttributeDescriptor(gFiledIndex).getType();

                int typebFiledIndex = featureType.getAttributeDescriptor(bFiledIndex).getType();

                if (nameFieldIndex < 0 || rFiledIndex < 0 || gFiledIndex < 0 || bFiledIndex < 0)

                        throw new InvalidROIsShpException("");

                if (typeFieldIndex != DataTypes.STRING ||

                                (typerFiledIndex != DataTypes.DOUBLE && typerFiledIndex != DataTypes.INT)  ||

                                (typegFiledIndex != DataTypes.DOUBLE && typegFiledIndex != DataTypes.INT)  ||

                                (typebFiledIndex != DataTypes.DOUBLE && typebFiledIndex != DataTypes.INT) )

                        throw new InvalidROIsShpException("");

                if (existingROIs != null)

                        rois = new HashMap<String, ROI>();

                if (existingROIs != null)

                        for (int i = 0; i < existingROIs.size(); i++) {

                                VectorialROI roi = (VectorialROI)existingROIs.get(i);

                                rois.put(roi.getName(), roi);

                        }

                String roiName;

                int r, g, b;

                FeatureSet set = null;

                DisposableIterator features = null;

                try {

                        set = featureStore.getFeatureSet();

                        features = set.iterator();

                        while (features.hasNext()) {

                                Feature feature = (Feature) features.next();

                                // }

                                // for (int i = 0; i<dataSource.getRowCount(); i++) {

                                // IFeature feature = fLyrVect.getSource().getFeature(i);

                                roiName = feature.getString(nameFieldIndex).toString();

                                VectorialROI roi = null;

                                if (!rois.containsKey(roiName)) {

                                        roi = new VectorialROI(grid);

                                        roi.setName(roiName);

                                        r = ((Number) feature.get(rFiledIndex)).intValue();

                                        g = ((Number) feature.get(gFiledIndex)).intValue();

                                        b = ((Number) feature.get(bFiledIndex)).intValue();

                                        roi.setColor(new Color(r, g, b));

                                        rois.put(roi.getName(), roi);

                                } else

                                        roi = (VectorialROI) rois.get(roiName);

                                roi.addGeometry(feature.getDefaultGeometry());

                        }

                } finally {

                        if (features != null)

                                features.dispose();

                        if (set != null)

                                set.dispose();

                }

                return new ArrayList<ROI>(rois.values());

        }

}