---

BXMLOutput.hpp

Go to the documentation of this file.

// This class will handle binary format of xml file.
// When you decompress the file with xdemill -b, you call this BXMLOutput class.
// This will call sax2tsax obj. and this will call tsax2bin obj. to create binary format file.
//
// Author : Jason Kim (jasonkim@cs.washington.edu)
//

#ifndef BXMLOutput_HPP
#define BXMLOutput_HPP

#include "xmltk.h"
#include "ISAXClient.hpp"

extern XTOKEN xmllabel;

class BXMLOutput : public ISAXClient
{
        // Initialize
        bool IUnknownCL_Init(IUnknownCL *punk, IStream* pws, bool bForStdout)
        {
                ITSAX2Bin *p2bin;
                bool bRet = punk->QueryInterface(&IID_ITSAX2Bin, (void**)&p2bin);

                if (bRet)
                {
                        bRet = p2bin->Init(pws, bForStdout);
                        p2bin->Release();
                }

                return bRet;
        }

        // Manage the buffer to decompress file
    void _CopyStringToBuffer(char *psz, int cch, bool fTrimWhitespace = false)
    {
        unsigned int cchBuf;

        m_psz = m_szBuf;
        cchBuf = ARRAYSIZE(m_szBuf);

        if ((unsigned int)cch >= ARRAYSIZE(m_szBuf)-1)
        {
            //
            // try to alloc a large enough block
            //
            if (m_pszBig)
                free(m_pszBig);

            m_pszBig = (char*)calloc(cch+1, sizeof(char));
            if (m_pszBig)
            {
                m_psz = m_pszBig;
                cchBuf = cch;
            }
            else
            {
                myassert(0);
                fprintf(stderr, "warning: low memory, data lost\n");
            }
        }

        m_cch = min((unsigned int)cch, cchBuf);
        strncpy(m_psz, psz, m_cch);
        m_psz[m_cch] = 0;
    }

        // Private method
    char m_szBuf[4096];
    char *m_pszBig;
    char *m_psz;
    int m_cch;

        ITSAXContentHandler *m_pch;
    XTOKEN m_xtAttribute;

public:

        // Constructor for this class
        BXMLOutput()
        {
                IFileStream* pfs;
                CreateFileStream(&IID_IFileStream, (void**)&pfs);

                if (pfs)
                {
                        // Want to save this output as stdout
                        pfs->SetFile(stdout);
                        CreateTSAX2Bin(&IID_ITSAXContentHandler, (void**)&m_pch);

                        if (m_pch)
                        {
                                IUnknownCL_Init(m_pch, pfs, true);
                        }
                        pfs->Release();
                }
                m_pch->AddRef();
                m_pch->startDocument();
        }

        // Destructor for this class
        virtual ~BXMLOutput()
        {
                m_pch->endDocument();
                m_pch->Release();
        }

        #ifndef XDEMILL_NOOUTPUT

        virtual void startElement(char *str,int len)
        {
                // HandleStartLabel
                _CopyStringToBuffer(str, len);
                XTOKEN xt = g_ptt->XTOKENFromStr(m_psz, XST_ELEMENT); // "bib"
                xmllabel = xt;
                m_pch->startElement(xt);
        }

        virtual void endElement(char *str,int len)
        {
                // Handles an end tag

                _CopyStringToBuffer(str, len);
                XTOKEN xt = g_ptt->XTOKENFromStr(m_psz, XST_ELEMENT);
                m_pch->endElement(xt);
        }

        virtual void endEmptyElement()
        {
                // Handles an end tag
                // Need more study for this case
                // IMPLEMENT LATER
                m_pch->endElement(xmllabel);
        }

        virtual void startAttribute(char *str,int len)
        {
                // Handles a given attribute name
                _CopyStringToBuffer(str, len);
                m_xtAttribute = g_ptt->XTOKENFromStr(m_psz, XST_ATTRIBUTE);

                // Handles an attribute value
                _CopyStringToBuffer(str, len);
                m_pch->attribute(m_xtAttribute, m_psz, m_cch);
        }

        virtual void endAttribute(char *str=NULL,int len=0)
        {
                // Nothing...
        }

        virtual void characters(char *str,int len)
        {
                // Handles a piece of text.
                // 'leftwslen' and 'rightwslen' specify how many white spaces (' ', '\n', ...)
                // are on the left and right end of the text.
                // If 'iscont=1', then 'rightwslen' must be zero and for the text piece
                // coming afterwards, 'leftwslen' must be zero.
                // It is also possible that 'len=leftwslen=rightwslen'

                //
                // hide the leading and trailing whitespace from the TSAX client
                //

                // Have to check int leftwslen,int rightwslen
                // str += leftwslen;
                // len = len - (leftwslen + rightwslen);

                len = max(0,len);

                _CopyStringToBuffer(str, len);

                if (m_cch > 0)
                {
                        m_pch->characters(m_psz, m_cch);
                }
        }

        virtual void whitespaces(char *str,int len)
        {
                // Nothing...
        }

        virtual void attribWhitespaces(char *str,int len)
        {
                // Nothing...
        }

        #else
        virtual void startElement(char *str,int len);
        virtual void endElement(char *str,int len);
        virtual void endEmptyElement();
        virtual void startAttribute(char *str,int len);
        virtual void endAttribute(char *str=NULL,int len=0);
        virtual void characters(char *str,int len);
        virtual void whitespaces(char *str,int len);
        virtual void attribWhitespaces(char *str,int len);
        #endif


}; // End of class

#endif

---