LHEInterface/src/XMLUtils.h

0001 #ifndef GeneratorInterface_LHEInterface_XMLUtils_h
0002 #define GeneratorInterface_LHEInterface_XMLUtils_h
0003
0004 #include <iostream>
0005 #include <string>
0006 #include <memory>
0007 #include <vector>
0008
0009 #include <xercesc/util/XMLString.hpp>
0010 #include <xercesc/util/XMLUni.hpp>
0011 #include <xercesc/util/XMLChar.hpp>
0012 #include <xercesc/util/BinInputStream.hpp>
0013 #include <xercesc/framework/XMLPScanToken.hpp>
0014 #include <xercesc/sax/InputSource.hpp>
0015 #include <xercesc/sax2/DefaultHandler.hpp>
0016 #include <xercesc/sax2/SAX2XMLReader.hpp>
0017 #include <lzma.h>
0018
0019 namespace edm::storage {
0020   class Storage;
0021 }
0022
0023 namespace lhef {
0024
0025   class StorageWrap {
0026   public:
0027     using Storage = edm::storage::Storage;
0028
0029     StorageWrap(std::unique_ptr<Storage> storage);
0030     ~StorageWrap();
0031
0032     Storage *operator->() { return storage.get(); }
0033     const Storage *operator->() const { return storage.get(); }
0034
0035   private:
0036     std::unique_ptr<Storage> storage;
0037   };
0038
0039   class XMLDocument {
0040   public:
0041     class Handler : public XERCES_CPP_NAMESPACE_QUALIFIER DefaultHandler {};
0042
0043     XMLDocument(std::unique_ptr<std::istream> &in, Handler &handler);
0044     XMLDocument(std::unique_ptr<StorageWrap> &in, Handler &handler);
0045     virtual ~XMLDocument();
0046
0047     bool parse();
0048
0049     static std::shared_ptr<void> platformHandle() { return std::make_shared<XercesPlatform>(); }
0050
0051   private:
0052     class XercesPlatform {
0053     public:
0054       XercesPlatform();
0055       // do not make any kind of copies
0056       XercesPlatform(const XercesPlatform &orig) = delete;
0057       XercesPlatform &operator=(const XercesPlatform &orig) = delete;
0058       ~XercesPlatform();
0059
0060     private:
0061       static unsigned int instances;
0062     };
0063
0064     void init(Handler &handler);
0065
0066     std::unique_ptr<XercesPlatform> platform;
0067
0068     std::unique_ptr<XERCES_CPP_NAMESPACE_QUALIFIER InputSource> source;
0069     std::unique_ptr<XERCES_CPP_NAMESPACE_QUALIFIER SAX2XMLReader> parser;
0070
0071     XERCES_CPP_NAMESPACE_QUALIFIER XMLPScanToken token;
0072
0073     bool done;
0074   };
0075
0076   class XMLSimpleStr {
0077   public:
0078     XMLSimpleStr(const XMLCh *str) : string(XERCES_CPP_NAMESPACE_QUALIFIER XMLString::transcode(str)) {}
0079
0080     ~XMLSimpleStr() { XERCES_CPP_NAMESPACE_QUALIFIER XMLString::release(&string); }
0081
0082     inline operator const char *() const { return string; }
0083
0084     inline static bool isAllSpaces(const XMLCh *str, unsigned int length) {
0085       return XERCES_CPP_NAMESPACE_QUALIFIER XMLChar1_0::isAllSpaces(str, length);
0086     }
0087
0088     inline static bool isSpace(XMLCh ch) { return XERCES_CPP_NAMESPACE_QUALIFIER XMLChar1_0::isWhitespace(ch); }
0089
0090   private:
0091     char *string;
0092   };
0093
0094   class XMLUniStr {
0095   public:
0096     XMLUniStr(const char *str) : unicode(XERCES_CPP_NAMESPACE_QUALIFIER XMLString::transcode(str)) {}
0097
0098     ~XMLUniStr() { XERCES_CPP_NAMESPACE_QUALIFIER XMLString::release(&unicode); }
0099
0100     operator const XMLCh *() const { return unicode; }
0101
0102   private:
0103     XMLCh *unicode;
0104   };
0105
0106   template <typename T>
0107   class XMLInputSourceWrapper : public XERCES_CPP_NAMESPACE_QUALIFIER InputSource {
0108   public:
0109     typedef typename T::Stream_t Stream_t;
0110
0111     XMLInputSourceWrapper(std::unique_ptr<Stream_t> &obj) : obj(std::move(obj)) {}
0112     ~XMLInputSourceWrapper() override {}
0113
0114     XERCES_CPP_NAMESPACE_QUALIFIER BinInputStream *makeStream() const override { return new T(*obj); }
0115
0116   private:
0117     std::unique_ptr<Stream_t> obj;
0118   };
0119
0120   class CBInputStream : public XERCES_CPP_NAMESPACE_QUALIFIER BinInputStream {
0121   public:
0122     class Reader {
0123     public:
0124       virtual ~Reader();
0125       virtual const std::string &data() = 0;
0126     };
0127
0128     typedef Reader Stream_t;
0129
0130     CBInputStream(Reader &in);
0131     ~CBInputStream() override;
0132
0133     XMLFilePos curPos() const override { return pos; }
0134
0135     XMLSize_t readBytes(XMLByte *const buf, const XMLSize_t size) override;
0136
0137     const XMLCh *getContentType() const override { return nullptr; }
0138
0139   private:
0140     Reader &reader;
0141     std::string buffer;
0142     unsigned int pos;
0143   };
0144
0145   class STLInputStream : public XERCES_CPP_NAMESPACE_QUALIFIER BinInputStream {
0146   public:
0147     typedef std::istream Stream_t;
0148
0149     STLInputStream(std::istream &in);
0150     ~STLInputStream() override;
0151
0152     XMLFilePos curPos() const override { return pos; }
0153
0154     XMLSize_t readBytes(XMLByte *const buf, const XMLSize_t size) override;
0155
0156     const XMLCh *getContentType() const override { return nullptr; }
0157
0158   private:
0159     std::istream &in;
0160     unsigned int pos;
0161   };
0162
0163   class StorageInputStream : public XERCES_CPP_NAMESPACE_QUALIFIER BinInputStream {
0164   public:
0165     typedef StorageWrap Stream_t;
0166
0167     StorageInputStream(StorageWrap &in);
0168     ~StorageInputStream() override;
0169
0170     XMLFilePos curPos() const override { return pos; }
0171
0172     XMLSize_t readBytes(XMLByte *const buf, const XMLSize_t size) override;
0173
0174     const XMLCh *getContentType() const override { return nullptr; }
0175
0176   private:
0177     StorageWrap &in;
0178     unsigned int pos;
0179     lzma_stream lstr;
0180     bool compression_;
0181     unsigned int lasttotal_;
0182
0183     unsigned int buffLoc_ = 0, buffTotal_ = 0;
0184     std::vector<uint8_t> buffer_;
0185     static constexpr unsigned bufferSize_ = 16 * 1024 * 1024;
0186   };
0187
0188   typedef XMLInputSourceWrapper<CBInputStream> CBInputSource;
0189   typedef XMLInputSourceWrapper<STLInputStream> STLInputSource;
0190   typedef XMLInputSourceWrapper<StorageInputStream> StorageInputSource;
0191
0192 }  // namespace lhef
0193
0194 #endif  // GeneratorInterface_LHEInterface_XMLUtils_h