| gvSIG desktop 1

/*

 * Created on 14-abr-2005

 *

 * gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 * DwgMemoryDriver 0.2. Driver del formato DWG para gvSIG

 *

 * Copyright (C) 2004 IVER T.I. and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

/* CVS MESSAGES:

 *

 * $Id: DwgMemoryDriver.java 10860 2007-03-22 19:56:12Z azabala $

 * $Log$

 * Revision 1.10  2007-03-22 19:56:12  azabala

 * changes in exceptinos api

 *

 * Revision 1.9  2007/03/20 20:21:19  azabala

 * added roi filtering, reordering of code

 *

 * Revision 1.7  2007/02/27 17:18:03  azabala

 * solved bug with text orientation

 *

 * Revision 1.6  2007/02/14 16:17:22  azabala

 * *** empty log message ***

 *

 * Revision 1.5  2007/01/20 18:30:37  azabala

 * refactoring of blocks

 *

 * Revision 1.4  2007/01/18 19:59:10  azabala

 * refactoring to optimize and simplify the code

 *

 * Revision 1.3  2007/01/18 13:36:42  azabala

 * Refactoring general para evitar dar tantas pasadas en la carga, y para incrementar

 * la legibilidad del codigo (demasiados if-else-if en vez de usar polimorfismo)

 *

 * Revision 1.2  2007/01/12 19:57:44  azabala

 * *** empty log message ***

 *

 * Revision 1.1  2007/01/11 20:31:05  azabala

 * *** empty log message ***

 *

 *

 */

package com.iver.cit.gvsig.drivers.dwg;

import java.awt.Color;

import java.awt.Font;

import java.awt.geom.Rectangle2D;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import com.hardcode.gdbms.driver.exceptions.InitializeDriverException;

import com.hardcode.gdbms.driver.exceptions.ReadDriverException;

import com.hardcode.gdbms.driver.exceptions.UnsupportedVersionDriverException;

import com.hardcode.gdbms.engine.data.DataSourceFactory;

import com.hardcode.gdbms.engine.data.driver.ObjectDriver;

import com.hardcode.gdbms.engine.data.edition.DataWare;

import com.hardcode.gdbms.engine.values.IntValue;

import com.hardcode.gdbms.engine.values.Value;

import com.hardcode.gdbms.engine.values.ValueFactory;

import com.iver.cit.gvsig.fmap.core.FShape;

import com.iver.cit.gvsig.fmap.core.IGeometry;

import com.iver.cit.gvsig.fmap.core.v02.FConstant;

import com.iver.cit.gvsig.fmap.core.v02.FSymbol;

import com.iver.cit.gvsig.fmap.drivers.AbstractCadMemoryDriver;

import com.iver.cit.gvsig.fmap.drivers.DriverAttributes;

import com.iver.cit.gvsig.fmap.drivers.VectorialFileDriver;

import com.iver.cit.gvsig.fmap.drivers.WithDefaultLegend;

import com.iver.cit.gvsig.fmap.rendering.Legend;

import com.iver.cit.gvsig.fmap.rendering.LegendFactory;

import com.iver.cit.gvsig.fmap.rendering.VectorialUniqueValueLegend;

import com.iver.cit.gvsig.fmap.rendering.styling.AttrInTableLabeling;

import com.iver.cit.gvsig.fmap.rendering.styling.ILabelingStrategy;

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;

import com.iver.cit.jdwglib.util.AcadColor;

/**

 * Driver that allows gvSIG to read files in DWG format Using this driver, a

 * gvSIG user can manipulate part of the information contained in a DWG file

 * This driver load the Dwg file in memory This driver uses jdwglib

 * 

 * @author jmorell

 */

public class DwgMemoryDriver extends AbstractCadMemoryDriver implements

                VectorialFileDriver, WithDefaultLegend {

        VectorialUniqueValueLegend defaultLegend;

        private File m_Fich;

        private String fileVersion;

        private DriverAttributes attr = new DriverAttributes();

        private boolean isInitialized = false;

        float heightText = 10;        

        //new WithDefaultLegend API

        private AttrInTableLabeling labeling;

        private boolean debug = false;

        /**

         * entities of the dwg file (once applied many transformation,

         * including block insertion) that are IDwgToFMap.

         * */

        ArrayList entities = new ArrayList();

        /**

         * Saves an original DWG entity associated to a FMap entity

         * by their index.

         * Only available in debug mode (if debug is false, doesnt save anything)

         * */

        private void saveEntity(DwgObject entity) {

                if(debug)

                        entities.add(entity);

        }

        public DwgObject getEntity(int i){

                if(debug)

                        return (DwgObject) entities.get(i);

                else

                        return null;

        }

        /*

         * (non-Javadoc)

         * 

         * @see com.iver.cit.gvsig.fmap.drivers.MemoryDriver#open(java.io.File)

         */

        public void open(File f) {

                m_Fich = f;

        }

        /**

         * Allows recovering of the DWG Drawing entity associated to the FMap

         * feature whose position is index.

         * If debug mode is false, it always returns null.

         * 

         * @param index

         *            position of the fmap feature whose dwg entity is required

         * 

         */

        public Object getCadSource(int index) {

                if(debug)

                        return getEntity(index);

                else

                        return null;

        }

        /**

         * This method load the DWG file in memory. First, it will be necessary to

         * create a DwgFile object with the DWG file path as the argument. Second,

         * the method read of DwgFile allows to read the objects inside the DWG

         * file. Third, it will be necessary to process some DWG objects like Layers

         * or Polylines Fourth, applyExtrusions() can change the location of the DWG

         * objects through the extrusion parameters. Fifth, the testDwg3D method

         * test if this DWG has elevation informacion. Sixth, we can extract the

         * objects contained inside the blocks through the blockManagement method.

         * And finally, we can read the objects Vector, and convert this objects to

         * the FMap object model.

         */

        public void initialize() throws InitializeDriverException {

                if (isInitialized)

                        return;

                else

                        isInitialized = true;

                attr.setLoadedInMemory(true);

                DwgFile dwg = new DwgFile(m_Fich.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 InitializeDriverException(

//                                        formatString(

//                                                        "La version (%s) del fichero DWG '%s' no esta soportada por gvSIG.\n%s\n%s\n%s",

//                                                        new String[] { e1.getDwgVersion().toString(),

//                                                                        m_Fich.getName(),

//                                                                        Messages.getText("Probe_to_convert"),

//                                                                        Messages.getText("Autodesk_converter"),

//                                                                        autodeskUrl }));

                        String description=formatString(

                                        "La version (%s) del fichero DWG '%s' no esta soportada por gvSIG.\n%s\n%s\n%s",

                                        new String[] { e1.getDwgVersion().toString(),

                                                        m_Fich.getName(),

                                                        Messages.getText("Probe_to_convert"),

                                                        Messages.getText("Autodesk_converter"),

                                                        autodeskUrl });

                        throw new UnsupportedVersionDriverException(getName(),e1,description);

                } catch (IOException e) {

                        throw new InitializeDriverException(getName(),e);

                }

                fileVersion = dwg.getDwgVersion();

                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];

                                Rectangle2D roi = new 

                                        Rectangle2D.Double(xmin-100, ymin-100, 

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

                                addRegionOfInterest(roi);

                        }

                }

                // Campos de las MemoryLayer:

                Value[] auxRow = new Value[10];

                ArrayList arrayFields = new ArrayList();

                arrayFields.add("ID");

                arrayFields.add("FShape");

                arrayFields.add("Entity");

                arrayFields.add("Layer");

                arrayFields.add("Color");

                arrayFields.add("Elevation");

                arrayFields.add("Thickness");

                arrayFields.add("Text");

                arrayFields.add("HeightText");

                arrayFields.add("RotationText");

                getTableModel().setColumnIdentifiers(arrayFields.toArray());

                labeling = new AttrInTableLabeling();

                ((AttrInTableLabeling) labeling).setTextFieldId(arrayFields.indexOf("Text"));

                ((AttrInTableLabeling) labeling).setRotationFieldId(arrayFields.indexOf("RotationText"));

                ((AttrInTableLabeling) labeling).setHeightFieldId(arrayFields.indexOf("HeightText"));

                boolean is3dFile = dwg.isDwg3DFile();

                try {

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

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

                                if (entity instanceof IDwg2FMap) {

                                        IDwg2FMap dwgEnt = (IDwg2FMap) entity;

                                        IGeometry geometry = dwgEnt.toFMapGeometry(is3dFile);

                                        if (geometry == null)

                                                continue;

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

                                        if(! checkRois(geometry))

                                                continue;

                                        String fmapStr = dwgEnt.toFMapString(is3dFile);

                                        // nombre del tipo de geometria fmap

                                        auxRow[ID_FIELD_FSHAPE] = ValueFactory.createValue(fmapStr);

                                        // indice del registro

                                        auxRow[ID_FIELD_ID] = ValueFactory.createValue(i);

                                        // nombre de entidad dwg

                                        auxRow[ID_FIELD_ENTITY] = ValueFactory.createValue(dwgEnt

                                                        .toString());

                                        String layerName = dwg.getLayerName(entity);

                                        auxRow[ID_FIELD_LAYER] = ValueFactory

                                                        .createValue(layerName);

                                        int colorByLayer = dwg.getColorByLayer(entity);

                                        int color = entity.getColor();

                                        if(color < 0)

                                                color = Math.abs(color);

                                        if(color > 255)

                                                color = colorByLayer;                                        

                                        auxRow[ID_FIELD_COLOR] = ValueFactory.createValue(color);

                                        auxRow[ID_FIELD_HEIGHTTEXT] = ValueFactory.createValue(0.0);

                                        auxRow[ID_FIELD_ROTATIONTEXT] = ValueFactory

                                                        .createValue(0.0);

                                        auxRow[ID_FIELD_TEXT] = ValueFactory.createNullValue();

                                        auxRow[ID_FIELD_THICKNESS] = ValueFactory.createValue(0.0);

                                        if (entity instanceof IDwg3DTestable) {

                                                auxRow[ID_FIELD_ELEVATION] = ValueFactory

                                                                .createValue(((IDwg3DTestable) entity).getZ());

                                        }

                                        if (entity instanceof DwgMText) {

                                                DwgMText mtext = (DwgMText) entity;

                                                auxRow[ID_FIELD_TEXT] = ValueFactory.createValue(mtext

                                                                .getText());

                                                auxRow[ID_FIELD_HEIGHTTEXT] = ValueFactory

                                                                .createValue(mtext.getHeight());

                                        } else if (entity instanceof DwgText) {

                                                DwgText text = (DwgText) entity;

                                                auxRow[ID_FIELD_TEXT] = ValueFactory.createValue(text

                                                                .getText());

                                                auxRow[ID_FIELD_THICKNESS] = ValueFactory

                                                                .createValue(text.getThickness());

                                                auxRow[ID_FIELD_HEIGHTTEXT] = ValueFactory

                                                                .createValue(text.getHeight());

                                                auxRow[ID_FIELD_ROTATIONTEXT] = ValueFactory

                                                                .createValue(Math.toDegrees(text

                                                                                .getRotationAngle()));

                                        }// if-else

                                        addGeometry(geometry, auxRow);

                                        if(debug)

                                                saveEntity(entity);

                                }

                        }// for

                } catch (Throwable t) {

                        t.printStackTrace();

                }

                dwgObjects = null;

                dwg = null;

                if(! debug)

                        dwg = null;

                System.gc();

                setSymbols();

        }

        /**

         * Sets a symbol for each dwg entity's derived feature based in

         * the DWG symbology info.

         * */

        private void setSymbols() throws InitializeDriverException{

                FSymbol myDefaultSymbol = new FSymbol(getShapeType());

                myDefaultSymbol.setShapeVisible(false);

                myDefaultSymbol.setFontSizeInPixels(false);

                myDefaultSymbol.setFont(new Font("SansSerif", Font.PLAIN, 9));

                myDefaultSymbol.setFontColor(Color.BLACK);

                myDefaultSymbol.setFontSize(heightText);

                myDefaultSymbol.setStyle(FConstant.SYMBOL_STYLE_DGNSPECIAL);

                myDefaultSymbol.setSize(3);

                myDefaultSymbol.setSizeInPixels(true);

                defaultLegend = LegendFactory

                                .createVectorialUniqueValueLegend(getShapeType());

                defaultLegend.setFieldName("Color");

                defaultLegend.setLabelField("Text");

                defaultLegend.setDefaultSymbol(myDefaultSymbol);

                defaultLegend.setLabelHeightField("HeightText");

                defaultLegend.setLabelRotationField("RotationText");

                ObjectDriver rs = this;

                IntValue clave;

                FSymbol theSymbol = null;

                try {

                        for (long j = 0; j < rs.getRowCount(); j++) {

                                clave = (IntValue) rs.getFieldValue(j, ID_FIELD_COLOR);

                                if (defaultLegend.getSymbolByValue(clave) == null) {

                                        theSymbol = new FSymbol(getShapeType());

                                        theSymbol.setDescription(clave.toString());

                                        theSymbol.setColor(AcadColor.getColor(clave.getValue()));

                                        // theSymbol.setFill(fillProv);

                                        // 050202, jmorell: Asigna los colores de Autocad a los

                                        // bordes

                                        // de los pol?gonos.

                                        theSymbol.setOutlineColor(AcadColor.getColor(clave

                                                        .getValue()));

                                        theSymbol.setStyle(FConstant.SYMBOL_STYLE_DGNSPECIAL);

                                        theSymbol.setSize(3);

                                        theSymbol.setSizeInPixels(true);

                                        defaultLegend.addSymbol(clave, theSymbol);

                                }// if

                        } // for

                } catch (ReadDriverException e) {

                        throw new InitializeDriverException(getName(),e);

                }

        }

        /**

         * checks if the given geometry intersects or its

         * contained in one of the driver regions of interest.

         * 

         * @param geometry feature geometry we want to check

         * against the drivers rois.

         * 

         *  @return true if feature is contained in any ROI.

         *  false if not.

         * */

        private boolean checkRois(IGeometry geometry){

                Rectangle2D rect = geometry.getBounds2D();

                int numRois = this.getNumOfRois();

                for(int i = 0; i < numRois; i++){

                        Rectangle2D roi = getRegionOfInterest(i);

                        if( checkIntersect(rect.getMinX(), rect.getMaxX(), rect.getMinY(), rect.getMaxY(),

                                        roi.getMinX(), roi.getMaxX(), roi.getMinY(),roi.getMaxY()) ||

                                checkContains(rect.getMinX(), rect.getMaxX(), rect.getMinY(), rect.getMaxY(),

                                                roi.getMinX(), roi.getMaxX(), roi.getMinY(),roi.getMaxY())

                        ){

                                return true;

                        }

//                        if(roi.intersects(rect) || roi.contains(rect))

//                                return true;

                }

                return false;

        }

        private boolean checkContains(double x1, double x2, double y1, double y2, 

                        double ax1, double ax2, double ay1, double ay2){

                 boolean solution = ( x1 >= ax1 &&

                x2 <= ax2 &&

                y1 >= ay1 &&

                y2 <= ay2);

                 return solution;

        }

        private boolean checkIntersect(double x1, double x2, double y1, double y2, 

                                                                double ax1, double ax2, double ay1, double ay2){

                 return !(x1 > ax2 ||

                        x2 < ax1 ||

                        y1 > ay2 ||

                        y2 < ay1);

        }

        public String getFileVersion() {

                return fileVersion;

        }

        private String formatString(String fmt, String[] params) {

                String ret = fmt;

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

                        ret = ret.replaceFirst("%s", params[i]);

                }

                return ret;

        }

        /*

         * (non-Javadoc)

         * 

         * @see com.iver.cit.gvsig.fmap.drivers.MemoryDriver#accept(java.io.File)

         */

        public boolean accept(File f) {

                return f.getName().toUpperCase().endsWith("DWG");

        }

        /*

         * (non-Javadoc)

         * 

         * @see com.iver.cit.gvsig.fmap.drivers.MemoryDriver#getShapeType()

         */

        public int getShapeType() {

                return FShape.MULTI;

        }

        /*

         * (non-Javadoc)

         * 

         * @see com.iver.cit.gvsig.fmap.drivers.MemoryDriver#getName()

         */

        public String getName() {

                return "gvSIG DWG Memory Driver";

        }

        /*

         * (non-Javadoc)

         * 

         * @see com.iver.cit.gvsig.fmap.drivers.WithDefaultLegend#getDefaultLegend()

         */

        public Legend getDefaultLegend() {

                return defaultLegend;

        }

        /*

         * (non-Javadoc)

         * 

         * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#getDriverAttributes()

         */

        public DriverAttributes getDriverAttributes() {

                return attr;

        }

        /*

         * (non-Javadoc)

         * 

         * @see com.hardcode.gdbms.engine.data.driver.ObjectDriver#getPrimaryKeys()

         */

        public int[] getPrimaryKeys() {

                // TODO Auto-generated method stub

                return null;

        }

        /*

         * (non-Javadoc)

         * 

         * @see com.hardcode.gdbms.engine.data.driver.ObjectDriver#write(com.hardcode.gdbms.engine.data.edition.DataWare)

         */

        public void write(DataWare arg0)  {

                // TODO Auto-generated method stub

        }

        public void setDataSourceFactory(DataSourceFactory arg0) {

                // TODO Auto-generated method stub

        }

        public void close()  {

                // TODO Auto-generated method stub

        }

        public File getFile() {

                return m_Fich;

        }

        public boolean isWritable() {

                return m_Fich.canWrite();

        }

        public boolean isDebug() {

                return debug;

        }

        public void setDebug(boolean debug) {

                this.debug = debug;

        }

        public ILabelingStrategy getDefaultLabelingStrategy() {

                return labeling;

        }

}