Application: gvSIG desktop

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

*

*/

/*

* AUTHORS (In addition to CIT):

* 2008 IVER T.I   {{Task}}

*/

/**

 *

 */

package org.gvsig.fmap.data.feature.file.dwg;

import java.io.File;

import java.io.IOException;

import java.util.List;

import java.util.TreeSet;

import org.gvsig.fmap.data.DataException;

import org.gvsig.fmap.data.OpenException;

import org.gvsig.fmap.data.ReadException;

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

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

import org.gvsig.fmap.data.feature.UnsupportedVersionException;

import org.gvsig.fmap.data.feature.file.FileMemoryResource;

import org.gvsig.fmap.data.feature.file.FileStoreParameters;

import org.gvsig.fmap.data.feature.operation.LegendLabelingManager;

import org.gvsig.fmap.data.feature.operation.LegendLabelingManagerFactory;

import org.gvsig.fmap.geom.Geometry;

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

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

import org.gvsig.i18n.Messages;

import com.iver.cit.jdwglib.dwg.DwgFile;

import com.iver.cit.jdwglib.dwg.DwgObject;

import com.iver.cit.jdwglib.dwg.DwgVersionNotSupportedException;

import com.iver.cit.jdwglib.dwg.IDwg2FMap;

import com.iver.cit.jdwglib.dwg.IDwg3DTestable;

import com.iver.cit.jdwglib.dwg.objects.DwgMText;

import com.iver.cit.jdwglib.dwg.objects.DwgText;

/**

 * @author jmvivo

 *

 */

public class DWGResource extends FileMemoryResource {

        private static LegendLabelingManagerFactory legendLabelingManagerFactory = null;

        public static final int ID_FIELD_ID = 0;

        public static final int ID_FIELD_FSHAPE = 1;

        public static final int ID_FIELD_ENTITY = 2;

        public static final int ID_FIELD_LAYER = 3;

        public static final int ID_FIELD_COLOR = 4;

        public static final int ID_FIELD_ELEVATION = 5;

        public static final int ID_FIELD_THICKNESS = 6;

        public static final int ID_FIELD_HEIGHTTEXT = 7;

        public static final int ID_FIELD_ROTATIONTEXT = 8;

        public static final int ID_FIELD_TEXT = 9;

        public static final int ID_FIELD_GEOMETRY = 10;

        private FeatureType featureType;

        private LegendLabelingManager legendlabelingManager;

        private Envelope fullEnvelope;

        /**

         * @param params

         */

        DWGResource(FileStoreParameters params) {

                super(params);

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.data.Resource#getName()

         */

        public String getName() {

                return "DWG";

        }

        private void initializeFeatureType() throws DataException {

                this.featureType= DWGStore.newFeatureType();

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.data.feature.file.FileMemoryResource#load()

         */

        protected void load() throws DataException {

//                float heightText = 10;

//                attr.setLoadedInMemory(true);

                initializeFeatureType();

                if (legendLabelingManagerFactory != null){

                        legendlabelingManager = legendLabelingManagerFactory.newManager();

                        legendlabelingManager.startLoading(null);

                }else{

                        legendlabelingManager = null;

                }

                this.fullEnvelope = new DefaultEnvelope(2);

                File file = this.getFile();

                DwgFile dwg = new DwgFile(file.getAbsolutePath());

                try {

                        dwg.read();

                } catch (DwgVersionNotSupportedException e1) {

                        String autodeskUrl = "<a href=\"http://usa.autodesk.com/adsk/servlet/item?siteID=123112&id=7024151\">"

                                        + "http://usa.autodesk.com/adsk/servlet/item?siteID=123112&id=7024151</a>";

                        throw new UnsupportedVersionException("la version ("+e1.getDwgVersion().toString()+

                                        ") del fichero '"+file.getName()+"' no esta soportada por gvSIG.\n"+

                                        Messages.getText("Probe_to_convert")+"\n"+

                                        Messages.getText("Autodesk_converter")+"\n"+autodeskUrl, e1);

                } catch (IOException e) {

                        throw new ReadException(DWGStore.DATASTORE_NAME, e);

                }

                dwg.calculateGisModelDwgPolylines();

                dwg.blockManagement2();

                List dwgObjects = dwg.getDwgObjects();

                //Each dwg file has two headers vars, EXTMIN and EXTMAX

                //that marks bounds of the file.

                //Also, it could have spureus entities (deleted objects

                //dont removed from the DWG database)

                double[] extMin = (double[]) dwg.getHeader("MSPACE_EXTMIN");

                double[] extMax = (double[]) dwg.getHeader("MSPACE_EXTMAX");

                if(extMin != null && extMax != null){

                        if(extMin.length >= 2 && extMax.length >= 2){

                                double xmin = extMin[0];

                                double ymin = extMin[1];

                                double xmax = extMax[0];

                                double ymax = extMax[1];

//                                Envelope roi = new

//                                        DefaultEnvelope(xmin-100, ymin-100,

//                                                        (xmax-xmin)+100, (ymax-ymin)+100); // ????

//                                Envelope roi = new DefaultEnvelope(xmin-100, ymin-100,

//                                                xmax+100, ymax+100);

// No dando margen

                                Envelope roi = new DefaultEnvelope(xmin, ymin,

                                                xmax, ymax);

                                this.fullEnvelope = roi;

                        }

                }

                TreeSet colorSet=new TreeSet();

                boolean is3dFile = dwg.isDwg3DFile();

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

                        try {

                                DwgObject entity = (DwgObject) dwgObjects.get(i);

                                if(entity instanceof IDwg2FMap){

                                        IDwg2FMap dwgEnt = (IDwg2FMap)entity;

                                        Geometry geometry = dwgEnt.toFMapGeometry(is3dFile);

                                        if (geometry == null ){ //|| geometry.getGeneralPath() == null) {

                                                continue;

                                        }

                                        //FIXME: Estas 6 lineas es por si la geometr??a no se ha creado correctamente

                                        Envelope gEnvelope = null;

                                        try{

                                                gEnvelope = geometry.getEnvelope();

                                        } catch (Exception e){

                                                gEnvelope = null;

                                        }

                                        if (gEnvelope == null) continue;

                                        //we check for Region of Interest of the CAD file

                                        if(!this.fullEnvelope.intersects(geometry.getEnvelope())) {

                                                continue;

                                        }

                                        DWGFeature feature = new DWGFeature(this.featureType, false);

                                        feature.loading();

                                        feature.set(ID_FIELD_ID, i);

                                        feature.set(ID_FIELD_ENTITY, dwgEnt.toString());

                                        feature.set(ID_FIELD_FSHAPE, geometry.getGeometryType().getName());

                                        feature.set(ID_FIELD_LAYER, dwg.getLayerName(entity));

                                        int colorByLayer = dwg.getColorByLayer(entity);

                                        int color = entity.getColor();

                                        if(color < 0) {

                                                color = Math.abs(color);

                                        }

                                        if(color > 255) {

                                                color = colorByLayer;

                                        }

                                        feature.set(ID_FIELD_COLOR, color);

                                        colorSet.add(new Integer(color));

                                        if(entity instanceof IDwg3DTestable){

                                                feature.set(ID_FIELD_ELEVATION, ((IDwg3DTestable)entity).getZ());

                                        }

                                        feature.set(ID_FIELD_THICKNESS, 0.0);

                                        feature.set(ID_FIELD_HEIGHTTEXT, 0.0);

                                        feature.set(ID_FIELD_ROTATIONTEXT, 0.0);

                                        feature.set(ID_FIELD_TEXT, "");

                                        if(entity instanceof DwgMText){

                                                DwgMText mtext = (DwgMText) entity;

                                                feature.set(ID_FIELD_HEIGHTTEXT, mtext.getHeight());

                                                feature.set(ID_FIELD_TEXT, mtext.getText());

                                        }else if(entity instanceof DwgText){

                                                DwgText text = (DwgText) entity;

                                                feature.set(ID_FIELD_THICKNESS, text.getThickness());

                                                feature.set(ID_FIELD_HEIGHTTEXT, text.getHeight());

                                                feature.set(ID_FIELD_ROTATIONTEXT, text.getRotationAngle());

                                                feature.set(ID_FIELD_TEXT, text.getText());

                                        }//if-else

                                        feature.setGeometry(ID_FIELD_GEOMETRY, geometry);

                                        this.addFeature(feature);

                                        if (this.legendlabelingManager != null){

                                                this.legendlabelingManager.add(feature, null);

                                        }

                                }//if instanceof

                        } catch (ReadException e) {

                                throw new OpenException(this.description(),e);

                        }

                }//for

                if (legendlabelingManager != null) {

                        legendlabelingManager.endLoading();

                }

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.data.feature.file.IFileResource#getTotalFiles()

         */

        public int getTotalFiles() {

                return 1;

        }

         protected FeatureType getFeatureType() throws ReadException{

                 this.checkOpen();

                 return this.featureType;

         }

        /**

         * @return

         * @throws ReadException

         */

        protected Object getDefaultLegend() throws ReadException {

                this.checkOpen();

                if (legendlabelingManager == null){

                        return null;

                }

                return legendlabelingManager.getDefaultLegend();

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.data.feature.file.FileMemoryResource#getFeatureCount()

         */

        protected int getFeatureCount() throws ReadException {

                return super.getFeatureCount();

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.data.feature.file.FileMemoryResource#getFullExtent()

         */

        protected Envelope getFullExtent() throws ReadException {

                return super.getFullExtent();

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.data.feature.file.FileMemoryResource#getFeature(int)

         */

        protected Feature getFeature(int index) throws ReadException {

                return super.getFeature(index);

        }

        protected Object getDefaultLabelingStrategy()  throws ReadException {

                this.checkOpen();

                if (legendlabelingManager == null){

                        return null;

                }

                return legendlabelingManager.getLabeling();

        }

        public static void setLegendLabelingManagerFactory(LegendLabelingManagerFactory factory) {

                legendLabelingManagerFactory = factory;

        }

        public static LegendLabelingManagerFactory getLegendLabelingManagerFactory() {

                return legendLabelingManagerFactory;

        }

}