204 lines
7.0 KiB
C++
204 lines
7.0 KiB
C++
/******************************************************************************
|
|
** libDXFrw - Library to read/write DXF files (ascii & binary) **
|
|
** **
|
|
** Copyright (C) 2011-2015 José F. Soriano, rallazz@gmail.com **
|
|
** **
|
|
** This library is free software, licensed 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. **
|
|
** You should have received a copy of the GNU General Public License **
|
|
** along with this program. If not, see <http://www.gnu.org/licenses/>. **
|
|
******************************************************************************/
|
|
|
|
#ifndef DWGREADER_H
|
|
#define DWGREADER_H
|
|
|
|
#include <map>
|
|
#include <list>
|
|
#include "drw_textcodec.h"
|
|
#include "dwgutil.h"
|
|
#include "dwgbuffer.h"
|
|
#include "../libdwgr.h"
|
|
|
|
class objHandle{
|
|
public:
|
|
objHandle(){ handle = type = loc = 0; }
|
|
objHandle(duint32 t, duint32 h, duint32 l){
|
|
type = t;
|
|
handle = h;
|
|
loc = l;
|
|
}
|
|
duint32 type;
|
|
duint32 handle;
|
|
duint32 loc;
|
|
};
|
|
|
|
//until 2000 = 2000-
|
|
//since 2004 except 2007 = 2004+
|
|
// 2007 = 2007
|
|
// pages of section
|
|
/* 2000-: No pages, only sections
|
|
* 2004+: Id, page number (index)
|
|
* size, size of page in file stream
|
|
* address, address in file stream
|
|
* dataSize, data size for this page
|
|
* startOffset, start offset for this page
|
|
* cSize, compresed size of data
|
|
* uSize, uncompresed size of data
|
|
* 2007: page Id, pageCount & pages
|
|
* size, size in file
|
|
* dataSize
|
|
* startOffset, start position in decompresed data stream
|
|
* cSize, compresed size of data
|
|
* uSize, uncompresed size of data
|
|
* address, address in file stream
|
|
* */
|
|
class dwgPageInfo {
|
|
public:
|
|
dwgPageInfo(){}
|
|
dwgPageInfo(duint64 i, duint64 ad, duint32 sz){
|
|
Id=i; address=ad; size=sz;
|
|
}
|
|
~dwgPageInfo(){}
|
|
duint64 Id;
|
|
duint64 address; //in file stream, for rd18, rd21
|
|
duint64 size; //in file stream, for rd18, rd21
|
|
duint64 dataSize; //for rd18, rd21
|
|
duint32 startOffset; //for rd18, rd21
|
|
duint64 cSize; //compresed page size, for rd21
|
|
duint64 uSize; //uncompresed page size, for rd21
|
|
};
|
|
|
|
// sections of file
|
|
/* 2000-: No pages, only section Id, size & address in file
|
|
* 2004+: Id, Section Id
|
|
* size, total size of uncompresed data
|
|
* pageCount & pages, number of pages in section
|
|
* maxSize, max decompressed Size per page
|
|
* compresed, (1 = no, 2 = yes, normally 2)
|
|
* encrypted, (0 = no, 1 = yes, 2 = unknown)
|
|
* name, read & stored but not used
|
|
* 2007: same as 2004+ except encoding, saved in compresed field
|
|
* */
|
|
class dwgSectionInfo {
|
|
public:
|
|
dwgSectionInfo(){
|
|
compresed = 1;//1=no, 2=yes
|
|
encrypted = 0;//???
|
|
pageCount = 0;
|
|
Id=-1;
|
|
}
|
|
~dwgSectionInfo(){}
|
|
dint32 Id; //section Id, 2000- rd15 rd18
|
|
std::string name; //section name rd18
|
|
duint32 compresed;//is compresed? 1=no, 2=yes rd18, rd21(encoding)
|
|
duint32 encrypted;//encrypted (doc: 0=no, 1=yes, 2=unkn) on read: objects 0 and encrypted yes rd18
|
|
std::map<duint32, dwgPageInfo >pages;//index, size, offset
|
|
duint64 size;//size of section, 2000- rd15, rd18, rd21 (data size)
|
|
duint64 pageCount; //number of pages (dwgPageInfo) in section rd18, rd21
|
|
duint64 maxSize; //max decompressed size (needed??) rd18 rd21
|
|
duint64 address; //address (seek) , 2000-
|
|
};
|
|
|
|
|
|
//! Class to handle dwg obj control entries
|
|
/*!
|
|
* Class to handle dwg obj control entries
|
|
* @author Rallaz
|
|
*/
|
|
class DRW_ObjControl : public DRW_TableEntry {
|
|
public:
|
|
DRW_ObjControl() { reset();}
|
|
|
|
void reset(){
|
|
}
|
|
bool parseDwg(DRW::Version version, dwgBuffer *buf, duint32 bs=0);
|
|
std::list<duint32>hadlesList;
|
|
};
|
|
|
|
|
|
class dwgReader {
|
|
friend class dwgR;
|
|
public:
|
|
dwgReader(std::istream *stream, dwgR *p){
|
|
fileBuf = new dwgBuffer(stream);
|
|
parent = p;
|
|
decoder.setVersion(DRW::AC1021, false);//default 2007 in utf8(no convert)
|
|
decoder.setCodePage("UTF-16", false);
|
|
// blockCtrl=0; //RLZ: temporary
|
|
// blockCtrl=layerCtrl=styleCtrl=linetypeCtrl=viewCtrl=0;
|
|
// ucsCtrl=vportCtrl=appidCtrl=dimstyleCtrl=vpEntHeaderCtrl=0;
|
|
nextEntLink = prevEntLink = 0;
|
|
maintenanceVersion=0;
|
|
}
|
|
virtual ~dwgReader();
|
|
|
|
protected:
|
|
virtual bool readMetaData() = 0;
|
|
virtual bool readPreview(){return false;}
|
|
virtual bool readFileHeader() = 0;
|
|
virtual bool readDwgHeader(DRW_Header& hdr)=0;
|
|
virtual bool readDwgClasses() = 0;
|
|
virtual bool readDwgHandles() = 0;
|
|
virtual bool readDwgTables(DRW_Header& hdr)=0;
|
|
virtual bool readDwgBlocks(DRW_Interface& intfa) = 0;
|
|
virtual bool readDwgEntities(DRW_Interface& intfa) = 0;
|
|
virtual bool readDwgObjects(DRW_Interface& intfa) = 0;
|
|
|
|
virtual bool readDwgEntity(dwgBuffer *dbuf, objHandle& obj, DRW_Interface& intfa);
|
|
bool readDwgObject(dwgBuffer *dbuf, objHandle& obj, DRW_Interface& intfa);
|
|
void parseAttribs(DRW_Entity* e);
|
|
std::string findTableName(DRW::TTYPE table, dint32 handle);
|
|
|
|
void setCodePage(std::string *c){decoder.setCodePage(c, false);}
|
|
std::string getCodePage(){ return decoder.getCodePage();}
|
|
bool readDwgHeader(DRW_Header& hdr, dwgBuffer *buf, dwgBuffer *hBuf);
|
|
bool readDwgHandles(dwgBuffer *dbuf, duint32 offset, duint32 size);
|
|
bool readDwgTables(DRW_Header& hdr, dwgBuffer *dbuf);
|
|
bool checkSentinel(dwgBuffer *buf, enum secEnum::DWGSection, bool start);
|
|
|
|
bool readDwgBlocks(DRW_Interface& intfa, dwgBuffer *dbuf);
|
|
bool readDwgEntities(DRW_Interface& intfa, dwgBuffer *dbuf);
|
|
bool readDwgObjects(DRW_Interface& intfa, dwgBuffer *dbuf);
|
|
bool readPlineVertex(DRW_Polyline& pline, dwgBuffer *dbuf);
|
|
|
|
public:
|
|
std::map<duint32, objHandle>ObjectMap;
|
|
std::map<duint32, objHandle>objObjectMap; //stores the ojects & entities not read in readDwgEntities
|
|
std::map<duint32, objHandle>remainingMap; //stores the ojects & entities not read in all proces, for debug only
|
|
std::map<duint32, DRW_LType*> ltypemap;
|
|
std::map<duint32, DRW_Layer*> layermap;
|
|
std::map<duint32, DRW_Block*> blockmap;
|
|
std::map<duint32, DRW_Textstyle*> stylemap;
|
|
std::map<duint32, DRW_Dimstyle*> dimstylemap;
|
|
std::map<duint32, DRW_Vport*> vportmap;
|
|
std::map<duint32, DRW_Block_Record*> blockRecordmap;
|
|
std::map<duint32, DRW_AppId*> appIdmap;
|
|
// duint32 currBlock;
|
|
duint8 maintenanceVersion;
|
|
|
|
protected:
|
|
dwgBuffer *fileBuf;
|
|
dwgR *parent;
|
|
DRW::Version version;
|
|
|
|
//seeker (position) for the beginning sentinel of the image data (R13 to R15)
|
|
duint32 previewImagePos;
|
|
|
|
//sections map
|
|
std::map<enum secEnum::DWGSection, dwgSectionInfo >sections;
|
|
std::map<duint32, DRW_Class*> classesmap;
|
|
|
|
protected:
|
|
DRW_TextCodec decoder;
|
|
|
|
protected:
|
|
// duint32 blockCtrl;
|
|
duint32 nextEntLink;
|
|
duint32 prevEntLink;
|
|
};
|
|
|
|
|
|
|
|
#endif // DWGREADER_H
|