Reader.h

浏览该文件的文档.

/************************************************************************
* @file Reader.h
*
* @brief 读取文件相关类
*
* @details 读取文件相关类
*
* @author wukx
*
* @version  v1.0
*
* @date  2016.9.29
*
* @license 北京华科软科技有限公司
*
*************************************************************************/
 
#ifndef BIMHOMEBASE_READER_H
#define BIMHOMEBASE_READER_H
 
#include <bitset>
#include <map>
#include <memory>
#include <sstream>
#include <string>
 
#include <xercesc/framework/XMLPScanToken.hpp>
#include <xercesc/sax2/Attributes.hpp>
#include <xercesc/sax2/DefaultHandler.hpp>
 
 
#include <Base/Common/FileInfo.h>
 
 
namespace zipios
{
    class ZipInputStream;
}
 
XERCES_CPP_NAMESPACE_BEGIN
class DefaultHandler;
class SAX2XMLReader;
XERCES_CPP_NAMESPACE_END
 
namespace Base
{
    class Persistence;
 
    class BaseExport XMLReader : public XERCES_CPP_NAMESPACE_QUALIFIER DefaultHandler
    {
    public:
        enum ReaderStatus
        {
            PartialRestore = 0,                     
            PartialRestoreInDocumentObject = 1,     
            PartialRestoreInProperty = 2,           
            PartialRestoreInObject = 3              
        };
 
        enum 
        {
            None = 0,                           
            Chars,                              
            StartDocument,                      
            EndDocument,                        
            StartElement,                       
            StartEndElement,                    
            EndElement,                         
            StartCDATA,                         
            EndCDATA                            
        }ReadType;
 
        struct FileEntry
        {
            std::string FileName;
            Base::Persistence* Object;
        };
 
        XMLReader(const char* FileName, std::istream&);
        ~XMLReader();
 
        bool isValid() const { return _valid; }
 
        bool isVerbose() const { return _verbose; }
 
        void setVerbose(bool on) { _verbose = on; }
 
        const char* localName() const;
 
        int level() const;
 
        void readElement(const char* ElementName = nullptr);
 
        void readEndElement(const char* ElementName = nullptr, int level = -1);
 
        bool readNextElement();
 
        void readCharacters();
 
        void readBinFile(const char*);
 
        unsigned int getAttributeCount() const;
 
        bool hasAttribute(const char* AttrName) const;
 
        long getAttributeAsInteger(const char* AttrName) const;
 
        unsigned long getAttributeAsUnsigned(const char* AttrName) const;
 
        double getAttributeAsFloat(const char* AttrName) const;
 
        const char* getAttribute(const char* AttrName) const;
 
        const char* addFile(const char* Name, Base::Persistence* Object);
 
        void readFiles(zipios::ZipInputStream& zipstream) const;
 
        void readFile(zipios::ZipInputStream& zipstream, std::string fileName) const;
        const std::vector<std::string>& getFilenames() const;
        bool isRegistered(Base::Persistence* Object) const;
        virtual void addName(const char*, const char*);
        virtual const char* getName(const char*) const;
        virtual bool doNameMapping() const;
 
        void setPartialRestore(bool on);
        void clearPartialRestoreDocumentObject();
        void clearPartialRestoreProperty();
        void clearPartialRestoreObject();
 
        bool testStatus(ReaderStatus pos) const;
 
        void setStatus(ReaderStatus pos, bool on);
        FileInfo getFileInfo() { return _File; }
 
        
    protected:
        bool read();
 
        // -----------------------------------------------------------------------
        //  SAX ContentHandler 接口
        // -----------------------------------------------------------------------
        virtual void startDocument();
        virtual void endDocument();
        virtual void startElement(const XMLCh* const uri, const XMLCh* const localname, const XMLCh* const qname, const XERCES_CPP_NAMESPACE_QUALIFIER Attributes& attrs);
        virtual void endElement(const XMLCh* const uri, const XMLCh* const localname, const XMLCh* const qname);
 
#if (XERCES_VERSION_MAJOR == 2)
        virtual void characters(const XMLCh* const chars, const unsigned int length);
#else
        virtual void characters(const XMLCh* const chars, const XMLSize_t length);
#endif
 
#if (XERCES_VERSION_MAJOR == 2)
        virtual void ignorableWhitespace(const XMLCh* const chars, const unsigned int length);
#else
        virtual void ignorableWhitespace(const XMLCh* const chars, const XMLSize_t length);
#endif
 
        virtual void startCDATA();
        virtual void endCDATA();
 
        virtual void resetDocument();
 
        void warning(const XERCES_CPP_NAMESPACE_QUALIFIER SAXParseException& exc);
        void error(const XERCES_CPP_NAMESPACE_QUALIFIER SAXParseException& exc);
        void fatalError(const XERCES_CPP_NAMESPACE_QUALIFIER SAXParseException& exc);
        void resetErrors();
 
    public:
        int DocumentSchema;                                         
        std::string ProgramVersion;                                 
        int FileVersion;                                            
        std::vector<FileEntry> FileList;                            
    protected:
        int Level;                                                  
        std::string LocalName;                                      
        std::string Characters;                                     
        unsigned int CharacterCount;                                
        std::map<std::string, std::string> AttrMap;                 
        typedef std::map<std::string, std::string> AttrMapType;     
        FileInfo _File;                                             
        XERCES_CPP_NAMESPACE_QUALIFIER SAX2XMLReader* parser;       
        XERCES_CPP_NAMESPACE_QUALIFIER XMLPScanToken token;         
        bool _valid;                                                
        bool _verbose;                                              
        std::vector<std::string> FileNames;                         
        std::map<std::string, std::string> nameMap;                 
        std::bitset<32> StatusBits;                                 
    };
    class BaseExport Reader : public std::istream
    {
    public:
        Reader(std::istream& stream, const std::string& name, int version);
 
        std::istream& getStream();
 
        std::string getFileName() const;
 
        int getFileVersion() const;
 
        void initLocalReader(std::shared_ptr<Base::XMLReader> reader);
 
        std::shared_ptr<Base::XMLReader> getLocalReader() const;
 
    private:
        std::istream& _str;                                 
        std::string _name;                                  
        int fileVersion;                                    
        std::shared_ptr<Base::XMLReader> localreader;       
    };
   
 
}
 
 
#endif //BIMHOMEBASE_READER_H