Application: gvSIG desktop

/* jdwglib. Java Library for reading Dwg files.

 * 

 * Author: Jose Morell Rama (jose.morell@gmail.com).

 * Port from the Pythoncad Dwg library by Art Haas.

 *

 * Copyright (C) 2005 Jose Morell, IVER TI S.A. 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:

 *

 * Jose Morell (jose.morell@gmail.com)

 * 

 * or

 *

 * IVER TI S.A.

 *  C/Salamanca, 50

 *  46005 Valencia

 *  Spain

 *  +34 963163400

 *  dac@iver.es

 */

package com.iver.cit.jdwglib.dwg.readers;

import java.io.IOException;

import java.nio.ByteBuffer;

import java.nio.ByteOrder;

import java.util.ArrayList;

import java.util.LinkedList;

import java.util.List;

import com.iver.cit.gvsig.fmap.drivers.dgn.ByteUtils;

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

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

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

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

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

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

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

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

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

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

import freenet.support.HexUtil;

/**

 * The DwgFileV15Reader reads the DWG version 15 format

 * 

 * @author jmorell

 */

public class DwgFileV15Reader implements IDwgFileReader {

        protected DwgFile dwgFile;

        protected ByteBuffer bb;

        /**

         * Reads the DWG version 15 format

         * 

         * @param dwgFile

         *            Represents the DWG file that we want to read

         * @throws IOException

         *             When DWG file path is wrong

         */

        public void read(DwgFile dwgFile, ByteBuffer bb) throws IOException {

                this.dwgFile = dwgFile;

                this.bb = bb;

                readDwgSectionOffsets();

                try {

                        readDwgObjectOffsets();

                        //readDwgClasses(bb);

                } catch (Exception e) {

                        //                    System.out.println("Error leyendo offsets y classes. Posible

                        // corrupci?n en" +

                        //                                    "el DWG file ...");

                }

                readDwgObjects();

                //checkSuperentities();

        }

        /*

         *TODO: Eliminar esto cuando terminemos con el testeo

         */

        protected void checkSuperentities(){

                System.out.println("***** CHEQUEANDO LAS SUPERENTIDADES *****");

                List objects = dwgFile.getDwgObjects();

                DwgObject obj;

                DwgObject superEnt;

                boolean buscado=false;

                boolean perdidos=false;

                int j=0;

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

                        obj = (DwgObject)objects.get(i);

                        if (obj.hasSubEntityHandle()){

                                buscado = true;

                                superEnt = dwgFile.getDwgSuperEntity(obj);

                                if (superEnt == null){

                                        j++;

                                }

                        }

                }

                if(j!=0) {

                        System.out.println("+++++ SE HAN PERDIDO LAS SUPERENTIDADES DE "+ j+" OBJETOS +++++");

                }

                System.out.println("***** FIN DEL CHEQUEO DE LAS SUPERENTIDADES *****");

        }

        /**

         * It read the SECTIONS from the header of the DWG file

         */

        protected void readDwgSectionOffsets() {

                bb.position(19);

                bb.order(ByteOrder.LITTLE_ENDIAN);

                short codePage = bb.getShort();

                int count = bb.getInt();

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

                        byte rec = bb.get();

                        int seek = bb.getInt();

                        int size = bb.getInt();

                        if (rec == 0) {

                                dwgFile.addDwgSectionOffset("HEADERS", seek, size);

                        } else if (rec == 1) {

                                dwgFile.addDwgSectionOffset("CLASSES", seek, size);

                        } else if (rec == 2) {

                                dwgFile.addDwgSectionOffset("OBJECTS", seek, size);

                        } else if (rec == 3) {

                                dwgFile.addDwgSectionOffset("UNKNOWN", seek, size);

                        } else if (rec == 4) {

                                dwgFile.addDwgSectionOffset("R14DATA", seek, size);

                        } else if (rec == 5) {

                                dwgFile.addDwgSectionOffset("R14REC5", seek, size);

                        } else {

                                //                                System.out.println("ERROR: C?digo de n?mero de registro

                                // no soportado: " + rec);

                        }

                }

        }

        /**

         * OpenDWG spec says: This section -object map- is a table which gives the

         * location of each object in the DWG file. This table is broken into

         * sections. It is basically a list of handle/file loc pairs. It could be

         * readed with this pseudocode:

         * 

         * Set lastHandle to all 0 and last loc to 0L. Repeat until section size ==

         * 2 section size = read short (in bigendian order) Repeat until out of data

         * for this section offset of this handle form last handle as modular char

         * offset of location in file from last location as modular char End repeat

         * End repeat

         *  

         */

        protected void readDwgObjectOffsets() throws Exception {

                int offset = dwgFile.getDwgSectionOffset("OBJECTS");

                bb.position(offset);

                while (true) {

                        bb.order(ByteOrder.BIG_ENDIAN);

                        /*

                         * We read the size of the next section. If size == 2, break (it is

                         * the last empty -except crc- section)

                         */

                        short size = bb.getShort();

                        if (size == 2)

                                break;

                        bb.order(ByteOrder.LITTLE_ENDIAN);

                        byte[] dataBytes = new byte[size];

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

                                dataBytes[i] = bb.get();

                        }

                        int[] data = DwgUtil.bytesToMachineBytes(dataBytes);

                        int lastHandle = 0;

                        int lastLoc = 0;

                        int bitPos = 0;

                        int bitMax = (size - 2) * 8;

                        while (bitPos < bitMax) {

                                ArrayList v = DwgUtil.getModularChar(data, bitPos);

                                bitPos = ((Integer) v.get(0)).intValue();

                                lastHandle = lastHandle + ((Integer) v.get(1)).intValue();

                                v = DwgUtil.getModularChar(data, bitPos);

                                bitPos = ((Integer) v.get(0)).intValue();

                                lastLoc = lastLoc + ((Integer) v.get(1)).intValue();

                                dwgFile.addDwgObjectOffset(lastHandle, lastLoc);

                        }//while

                }//while

        }

        /*

         * Unused.

         * 

         * Dwg spec says that drawing entities (objects) that has not a fixed type

         * must read its type from this section

         * 

         *  

         */

        protected void readDwgClasses() throws Exception {

                int offset = dwgFile.getDwgSectionOffset("CLASSES");

                // Por ahora nos saltamos los 16 bytes de control

                bb.position(offset + 16);

                bb.order(ByteOrder.LITTLE_ENDIAN);

                int size = bb.getInt();

                byte[] dataBytes = new byte[size];

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

                        dataBytes[i] = bb.get();

                }

                int[] data = DwgUtil.bytesToMachineBytes(dataBytes);

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

                        data[i] = (byte) ByteUtils.getUnsigned((byte) data[i]);

                }

                bb.position(bb.position() + 2 + 16);

                int maxbit = size * 8;

                int bitPos = 0;

                while ((bitPos + 8) < maxbit) {

                        ArrayList v = DwgUtil.getBitShort(data, bitPos);

                        bitPos = ((Integer) v.get(0)).intValue();

                        v = DwgUtil.getBitShort(data, bitPos);

                        bitPos = ((Integer) v.get(0)).intValue();

                        v = DwgUtil.getTextString(data, bitPos);

                        bitPos = ((Integer) v.get(0)).intValue();

                        v = DwgUtil.getTextString(data, bitPos);

                        bitPos = ((Integer) v.get(0)).intValue();

                        v = DwgUtil.getTextString(data, bitPos);

                        bitPos = ((Integer) v.get(0)).intValue();

                        v = DwgUtil.testBit(data, bitPos);

                        bitPos = ((Integer) v.get(0)).intValue();

                        v = DwgUtil.getBitShort(data, bitPos);

                        bitPos = ((Integer) v.get(0)).intValue();

                }

        }

        /**

         * Reads all the object referenced in the object map section of the DWG file

         * (using their object file obsets)

         */

        protected void readDwgObjects() {

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

                        DwgObjectOffset doo = (DwgObjectOffset) dwgFile

                                        .getDwgObjectOffsets().get(i);

                        DwgObject obj = readDwgObject(doo.getOffset(), i);

                        /*

                         * azabala: las entidades DWG no implementadas no nos aportan nada

                         * (aunque la sigo leyendo por si aparecen problemas de puntero de

                         * fichero) No considero por tanto los DwgObject if (obj != null) {

                         * dwgFile.addDwgObject(obj); }

                         *        

                         * paco: propongo reconsiderar esto. Si no cargamos todos los objetos

                         * podemos tener problemas con las subentities.

                         */

                        if (obj != null && obj.getClass() != DwgObject.class) {

                                dwgFile.addDwgObject(obj);

//                            DwgObject superEnt = dwgFile.getDwgSuperEntity(obj);

//                            if(superEnt != null){

//                              System.out.println(superEnt.getClass().getName() + "(TIPO " +Integer.toHexString(superEnt.getType())+") ES SUPERENTITY DE "+obj.getClass().getName());

                            } 

//                        }

                }

        }

        /**

         * Reads the header of an object in a DWG file Version 15

         * 

         * @param data

         *            Array of unsigned bytes obtained from the DWG binary file

         * @param offset

         *            The current bit offset where the value begins

         * @return int New offset

         */

        public int readObjectHeader(int[] data, int offset, DwgObject dwgObject) {

                int bitPos = offset;

                Integer mode = (Integer) DwgUtil.getBits(data, 2, bitPos);

                bitPos = bitPos + 2;

                dwgObject.setMode(mode.intValue());

                /*

                ArrayList v = DwgUtil.getBitLong(data, bitPos);

                bitPos = ((Integer) v.get(0)).intValue();

                int rnum = ((Integer) v.get(1)).intValue();

                dwgObject.setNumReactors(rnum);

                */

                ArrayList v = DwgUtil.getBitShort(data, bitPos);

                bitPos = ((Integer) v.get(0)).intValue();

                int rnum = ((Integer) v.get(1)).intValue();

                dwgObject.setNumReactors(rnum);

                v = DwgUtil.testBit(data, bitPos);

                bitPos = ((Integer) v.get(0)).intValue();

                boolean nolinks = ((Boolean) v.get(1)).booleanValue();

                dwgObject.setNoLinks(nolinks);

                v = DwgUtil.getBitShort(data, bitPos);

                bitPos = ((Integer) v.get(0)).intValue();

                int color = ((Integer) v.get(1)).intValue();

                dwgObject.setColor(color);

                v = DwgUtil.getBitDouble(data, bitPos);

                bitPos = ((Integer) v.get(0)).intValue();

                float ltscale = ((Double) v.get(1)).floatValue();

                Integer ltflag = (Integer) DwgUtil.getBits(data, 2, bitPos);

                bitPos = bitPos + 2;

                Integer psflag = (Integer) DwgUtil.getBits(data, 2, bitPos);

                bitPos = bitPos + 2;

                v = DwgUtil.getBitShort(data, bitPos);

                bitPos = ((Integer) v.get(0)).intValue();

                int invis = ((Integer) v.get(1)).intValue();

                v = DwgUtil.getRawChar(data, bitPos);

                bitPos = ((Integer) v.get(0)).intValue();

                int weight = ((Integer) v.get(1)).intValue();

                return bitPos;

        }

        /**

         * Reads the tail of an object in a DWG file Version 15

         * 

         * @param data

         *            Array of bytes obtained from the DWG binary file

         * @param offset

         *            Offset for this array of bytes

         * @return int New offset

         * @throws CorruptedDwgEntityException 

         * @throws RuntimeException 

         */

        public int readObjectTailer(int[] data, int offset, DwgObject dwgObject) throws RuntimeException, CorruptedDwgEntityException {

                int bitPos = offset;

                /*

                 * Subentity ref handle. Esto se aplica sobre VERTEX, ATTRIB, SEQEND

                 */

                if (dwgObject.getMode() == 0x0) {

                        DwgHandleReference subEntityHandle = new DwgHandleReference();

                        bitPos = subEntityHandle.read(data, bitPos);

                        dwgObject.setSubEntityHandle(subEntityHandle);

//                        System.out.println("subEntityHandle: code = "+subEntityHandle.getCode()+" offset = "+subEntityHandle.getOffset());

                }

                /*

                 * Reactors handles TODO No se est?n usando para setear nada en

                 * DwgObject

                 */

                DwgHandleReference reactorHandle;

                for (int i = 0; i < dwgObject.getNumReactors(); i++) {

                        reactorHandle = new DwgHandleReference();

                        bitPos = reactorHandle.read(data, bitPos);

                        dwgObject.addReactorHandle(reactorHandle);

//                        System.out.println("reactorHandle "+i+": code = "+reactorHandle.getCode()+" offset = "+reactorHandle.getOffset());

                }

                /*

                 * XDICOBJHANDLE

                 */

                DwgHandleReference xDicObjHandle = new DwgHandleReference();

                bitPos = xDicObjHandle.read(data, bitPos);

                dwgObject.setXDicObjHandle(xDicObjHandle);

//                System.out.println("xDicObjHandle: code = "+xDicObjHandle.getCode()+" offset = "+xDicObjHandle.getOffset());

                /*

                 * Layer Handle code

                 */

                DwgHandleReference handle = new DwgHandleReference();

                bitPos = handle.read(data, bitPos);

                dwgObject.setLayerHandle(handle);

//                System.out.println("layerHandle: code = "+handle.getCode()+" offset = "+handle.getOffset());

                if (!dwgObject.isNoLinks()) {

                        DwgHandleReference previousHandle = new DwgHandleReference();

                        bitPos = previousHandle.read(data, bitPos);

                        dwgObject.setPreviousHandle(previousHandle);

//                        System.out.println("previousHandle: code = "+previousHandle.getCode()+" offset = "+previousHandle.getOffset());

                        DwgHandleReference nextHandle = new DwgHandleReference();

                        bitPos = nextHandle.read(data, bitPos);

                        dwgObject.setNextHandle(nextHandle);

//                        System.out.println("nextHandle: code = "+nextHandle.getCode()+" offset = "+nextHandle.getOffset());

                }

                if (dwgObject.getLinetypeFlags() == 0x3) {

                        DwgHandleReference lineTypeHandle = new DwgHandleReference();

                        bitPos = lineTypeHandle.read(data, bitPos);

                        dwgObject.setLineTypeHandle(lineTypeHandle);

//                        System.out.println("lineTypeHandle: code = "+lineTypeHandle.getCode()+" offset = "+lineTypeHandle.getOffset());

                }

                if (dwgObject.getPlotstyleFlags() == 0x3) {

                        DwgHandleReference plotStyleHandle = new DwgHandleReference();

                        bitPos = plotStyleHandle.read(data, bitPos);

                        dwgObject.setPlotStyleHandle(plotStyleHandle);

//                        System.out.println("plotStyleHandle: code = "+plotStyleHandle.getCode()+" offset = "+plotStyleHandle.getOffset());

                }

                return bitPos;

        }

        /**

         * Return a dwg object from its index in the dwg file

         * 

         * @param index

         *            of the requested dwg object in the dwg file

         *  

         */

        public DwgObject getDwgObjectByIndex(int index) {

                DwgObjectOffset doo = (DwgObjectOffset) dwgFile.getDwgObjectOffsets()

                                .get(index);

                return readDwgObject(doo.getOffset(), index);

        }

        /**

         * Reads a dwg drawing entity (dwg object) given its offset in the file

         */

        protected DwgObject readDwgObject(int offset, int index) {

                DwgObject obj = null;

                try {

                        bb.position(offset);

                        int size = DwgUtil.getModularShort(bb);

                        bb.order(ByteOrder.LITTLE_ENDIAN);

                        byte[] dataBytes = new byte[size];

                        String[] dataMachValString = new String[size];

                        int[] data = new int[size];

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

                                dataBytes[i] = bb.get();

                                dataMachValString[i] = HexUtil

                                                .bytesToHex(new byte[] { dataBytes[i] });

                                Integer dataMachValShort = Integer.decode("0x"

                                                + dataMachValString[i]);

                                data[i] = dataMachValShort.byteValue();

                                data[i] = ByteUtils.getUnsigned((byte) data[i]);

                        }

                        int bitPos = 0;

                        ArrayList v = DwgUtil.getBitShort(data, bitPos);

                        bitPos = ((Integer) v.get(0)).intValue();

                        int type = ((Integer) v.get(1)).intValue();

                        obj = DwgObjectFactory.getInstance().create(type, index);

//                        System.out.println("SE HA CREADO UN OBJETO DEL TIPO = "+type+" =>> "+obj.getClass().getName());

                        if(obj == null)

                                return null;

                        v = DwgUtil.getRawLong(data, bitPos);

                        bitPos = ((Integer) v.get(0)).intValue();

                        int objBSize = ((Integer) v.get(1)).intValue();

                        obj.setSizeInBits(objBSize);

                        DwgHandleReference entityHandle = new DwgHandleReference();

                        bitPos = entityHandle.read(data, bitPos);

                        obj.setHandle(entityHandle);

//                        System.out.println("entityHandle:code = " + entityHandle.getCode() +" offset = "+entityHandle.getOffset());

                        v = DwgUtil.readExtendedData(data, bitPos);

                        bitPos = ((Integer) v.get(0)).intValue();

                        ArrayList extData = (ArrayList) v.get(1);

                        obj.setExtendedData(extData);

                        boolean gflag = false;

                        gflag = obj.isGraphicsFlag();

                        if (gflag) {

                                //lee un flag boolean

                                v = DwgUtil.testBit(data, bitPos);

                                bitPos = ((Integer) v.get(0)).intValue();

                                boolean val = ((Boolean) v.get(1)).booleanValue();

                                //si hay imagen asociada, se lee por completo

                                if (val) {

                                        v = DwgUtil.getRawLong(data, bitPos);

                                        bitPos = ((Integer) v.get(0)).intValue();

                                        size = ((Integer) v.get(1)).intValue();

                                        int bgSize = size * 8;

                                        Integer giData = (Integer) DwgUtil.getBits(data, bgSize,

                                                        bitPos);

                                        obj.setGraphicData(giData.intValue());

                                        bitPos = bitPos + bgSize;

                                }

                        }

                                if(obj.getClass() != DwgObject.class)

                                        readSpecificObject(obj, data, bitPos);

//                                else

//                                        System.out.println("Dwg Object desconocido:"+obj.getType());

                } catch (RuntimeException e) {

                        e.printStackTrace();

                } catch (CorruptedDwgEntityException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                        return null;        

                }

                return obj;

        }

        /*

         * TODO Esto est? pesimamente dise?ado. Cada objeto DwgObject debe tener un

         * metodo readSpecificObject(data,bitPos)

         *  

         */

        protected void readSpecificObject(DwgObject obj, int[] data, int bitPos)

                        throws RuntimeException, CorruptedDwgEntityException {

                        DwgObjectReaderPool pool = DwgObjectReaderPool.getInstance();

                        IDwgObjectReader reader = pool.get(obj, "15");

                        if(reader != null){

                                reader.setFileReader(this);

                                reader.readSpecificObj(data, bitPos, obj);

                        }else{

                                System.out.println("No se ha implementado la lectura de "+obj.getClass().getName()+", code="+obj.getType());

                        }

        }

}