| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058 |
- #include "tinyxml2.h"
- #include <string.h>
- #include <stdlib.h>
- #include <stdio.h>
- #include <ctype.h>
- #include <new.h>
- //#pragma warning ( disable : 4291 )
- using namespace tinyxml2;
- static const char LINE_FEED = (char)0x0a; // all line endings are normalized to LF
- static const char LF = LINE_FEED;
- static const char CARRIAGE_RETURN = (char)0x0d; // CR gets filtered out
- static const char CR = CARRIAGE_RETURN;
- static const char SINGLE_QUOTE = '\'';
- static const char DOUBLE_QUOTE = '\"';
- #define DELETE_NODE( node ) { MemPool* pool = node->memPool; node->~XMLNode(); pool->Free( node ); }
- #define DELETE_ATTRIBUTE( attrib ) { MemPool* pool = attrib->memPool; attrib->~XMLAttribute(); pool->Free( attrib ); }
- struct Entity {
- const char* pattern;
- int length;
- char value;
- };
- static const int NUM_ENTITIES = 5;
- static const Entity entities[NUM_ENTITIES] =
- {
- { "quot", 4, DOUBLE_QUOTE },
- { "amp", 3, '&' },
- { "apos", 4, SINGLE_QUOTE },
- { "lt", 2, '<' },
- { "gt", 2, '>' }
- };
- const char* StrPair::GetStr()
- {
- if ( flags & NEEDS_FLUSH ) {
- *end = 0;
- flags ^= NEEDS_FLUSH;
- if ( flags ) {
- char* p = start;
- char* q = start;
- while( p < end ) {
- if ( (flags & NEEDS_NEWLINE_NORMALIZATION) && *p == CR ) {
- // CR-LF pair becomes LF
- // CR alone becomes LF
- // LF-CR becomes LF
- if ( *(p+1) == LF ) {
- p += 2;
- }
- else {
- ++p;
- }
- *q++ = LF;
- }
- else if ( (flags & NEEDS_NEWLINE_NORMALIZATION) && *p == LF ) {
- if ( *(p+1) == CR ) {
- p += 2;
- }
- else {
- ++p;
- }
- *q++ = LF;
- }
- else if ( (flags & NEEDS_ENTITY_PROCESSING) && *p == '&' ) {
- int i=0;
- for( i=0; i<NUM_ENTITIES; ++i ) {
- if ( strncmp( p+1, entities[i].pattern, entities[i].length ) == 0
- && *(p+entities[i].length+1) == ';' )
- {
- // Found an entity convert;
- *q = entities[i].value;
- ++q;
- p += entities[i].length + 2;
- break;
- }
- }
- if ( i == NUM_ENTITIES ) {
- // fixme: treat as error?
- ++p;
- ++q;
- }
- }
- else {
- *q = *p;
- ++p;
- ++q;
- }
- }
- *q = 0;
- }
- flags = 0;
- }
- return start;
- }
- /*
- const char* StringPool::Intern( const char* str )
- {
- // Treat the array as a linear, inplace hash table.
- // Nothing can get deleted, so that's handy.
- if ( size > pool.Size()*3/4 ) {
- DynArray< const char*, 20 > store;
- for( int i=0; i<pool.Size(); ++i ) {
- if ( pool[i] != 0 ) {
- store.Push( pool[i] );
- }
- }
- int newSize = pool.Size() * 2;
- pool.PopArr( pool.Size() );
- const char** mem = pool.PushArr( newSize );
- memset( (void*)mem, 0, sizeof(char)*newSize );
- while ( !store.Empty() ) {
- Intern( store.Pop() );
- }
- }
- }
- */
- // --------- XMLUtil ----------- //
- char* StrPair::ParseText( char* p, const char* endTag, int strFlags )
- {
- TIXMLASSERT( endTag && *endTag );
- char* start = p;
- char endChar = *endTag;
- int length = strlen( endTag );
- // Inner loop of text parsing.
- while ( *p ) {
- if ( *p == endChar && strncmp( p, endTag, length ) == 0 ) {
- Set( start, p, strFlags );
- return p + length;
- }
- ++p;
- }
- return p;
- }
- char* StrPair::ParseName( char* p )
- {
- char* start = p;
- start = p;
- if ( !start || !(*start) ) {
- return 0;
- }
- if ( !XMLUtil::IsAlpha( *p ) ) {
- return 0;
- }
- while( *p && (
- XMLUtil::IsAlphaNum( (unsigned char) *p )
- || *p == '_'
- || *p == '-'
- || *p == '.'
- || *p == ':' ))
- {
- ++p;
- }
- if ( p > start ) {
- Set( start, p, 0 );
- return p;
- }
- return 0;
- }
- char* XMLDocument::Identify( char* p, XMLNode** node )
- {
- XMLNode* returnNode = 0;
- char* start = p;
- p = XMLUtil::SkipWhiteSpace( p );
- if( !p || !*p )
- {
- return 0;
- }
- // What is this thing?
- // - Elements start with a letter or underscore, but xml is reserved.
- // - Comments: <!--
- // - Decleration: <?xml
- // - Everthing else is unknown to tinyxml.
- //
- static const char* xmlHeader = { "<?xml" };
- static const char* commentHeader = { "<!--" };
- static const char* dtdHeader = { "<!" };
- static const char* cdataHeader = { "<![CDATA[" };
- static const char* elementHeader = { "<" }; // and a header for everything else; check last.
- static const int xmlHeaderLen = 5;
- static const int commentHeaderLen = 4;
- static const int dtdHeaderLen = 2;
- static const int cdataHeaderLen = 9;
- static const int elementHeaderLen = 1;
- TIXMLASSERT( sizeof( XMLComment ) == sizeof( XMLUnknown ) ); // use same memory pool
- TIXMLASSERT( sizeof( XMLComment ) == sizeof( XMLDeclaration ) ); // use same memory pool
- if ( XMLUtil::StringEqual( p, xmlHeader, xmlHeaderLen ) ) {
- returnNode = new (commentPool.Alloc()) XMLDeclaration( this );
- returnNode->memPool = &commentPool;
- p += xmlHeaderLen;
- }
- else if ( XMLUtil::StringEqual( p, commentHeader, commentHeaderLen ) ) {
- returnNode = new (commentPool.Alloc()) XMLComment( this );
- returnNode->memPool = &commentPool;
- p += commentHeaderLen;
- }
- else if ( XMLUtil::StringEqual( p, cdataHeader, cdataHeaderLen ) ) {
- XMLText* text = new (textPool.Alloc()) XMLText( this );
- returnNode = text;
- returnNode->memPool = &textPool;
- p += cdataHeaderLen;
- text->SetCData( true );
- }
- else if ( XMLUtil::StringEqual( p, dtdHeader, dtdHeaderLen ) ) {
- returnNode = new (commentPool.Alloc()) XMLUnknown( this );
- returnNode->memPool = &commentPool;
- p += dtdHeaderLen;
- }
- else if ( XMLUtil::StringEqual( p, elementHeader, elementHeaderLen ) ) {
- returnNode = new (elementPool.Alloc()) XMLElement( this );
- returnNode->memPool = &elementPool;
- p += elementHeaderLen;
- }
- else if ( (*p != '<') && XMLUtil::IsAlphaNum( *p ) ) {
- returnNode = new (textPool.Alloc()) XMLText( this );
- returnNode->memPool = &textPool;
- p = start; // Back it up, all the text counts.
- }
- else {
- this->SetError( ERROR_IDENTIFYING_TAG, p, 0 );
- p = 0;
- returnNode = 0;
- }
- *node = returnNode;
- return p;
- }
- bool XMLDocument::Accept( XMLVisitor* visitor ) const
- {
- if ( visitor->VisitEnter( *this ) )
- {
- for ( const XMLNode* node=FirstChild(); node; node=node->NextSibling() )
- {
- if ( !node->Accept( visitor ) )
- break;
- }
- }
- return visitor->VisitExit( *this );
- }
- // --------- XMLNode ----------- //
- XMLNode::XMLNode( XMLDocument* doc ) :
- document( doc ),
- parent( 0 ),
- firstChild( 0 ), lastChild( 0 ),
- prev( 0 ), next( 0 )
- {
- }
- XMLNode::~XMLNode()
- {
- ClearChildren();
- if ( parent ) {
- parent->Unlink( this );
- }
- }
- void XMLNode::ClearChildren()
- {
- while( firstChild ) {
- XMLNode* node = firstChild;
- Unlink( node );
-
- DELETE_NODE( node );
- }
- firstChild = lastChild = 0;
- }
- void XMLNode::Unlink( XMLNode* child )
- {
- TIXMLASSERT( child->parent == this );
- if ( child == firstChild )
- firstChild = firstChild->next;
- if ( child == lastChild )
- lastChild = lastChild->prev;
- if ( child->prev ) {
- child->prev->next = child->next;
- }
- if ( child->next ) {
- child->next->prev = child->prev;
- }
- child->parent = 0;
- }
- XMLNode* XMLNode::InsertEndChild( XMLNode* addThis )
- {
- if ( lastChild ) {
- TIXMLASSERT( firstChild );
- TIXMLASSERT( lastChild->next == 0 );
- lastChild->next = addThis;
- addThis->prev = lastChild;
- lastChild = addThis;
- addThis->parent = this;
- addThis->next = 0;
- }
- else {
- TIXMLASSERT( firstChild == 0 );
- firstChild = lastChild = addThis;
- addThis->parent = this;
- addThis->prev = 0;
- addThis->next = 0;
- }
- return addThis;
- }
- XMLNode* XMLNode::InsertFirstChild( XMLNode* addThis )
- {
- if ( firstChild ) {
- TIXMLASSERT( lastChild );
- TIXMLASSERT( firstChild->prev == 0 );
- firstChild->prev = addThis;
- addThis->next = firstChild;
- firstChild = addThis;
- addThis->parent = this;
- addThis->prev = 0;
- }
- else {
- TIXMLASSERT( lastChild == 0 );
- firstChild = lastChild = addThis;
- addThis->parent = this;
- addThis->prev = 0;
- addThis->next = 0;
- }
- return addThis;
- }
- XMLNode* XMLNode::InsertAfterChild( XMLNode* afterThis, XMLNode* addThis )
- {
- TIXMLASSERT( afterThis->parent == this );
- if ( afterThis->parent != this )
- return 0;
- if ( afterThis->next == 0 ) {
- // The last node or the only node.
- return InsertEndChild( addThis );
- }
- addThis->prev = afterThis;
- addThis->next = afterThis->next;
- afterThis->next->prev = addThis;
- afterThis->next = addThis;
- return addThis;
- }
- const XMLElement* XMLNode::FirstChildElement( const char* value ) const
- {
- for( XMLNode* node=firstChild; node; node=node->next ) {
- XMLElement* element = node->ToElement();
- if ( element ) {
- if ( !value || XMLUtil::StringEqual( element->Name(), value ) ) {
- return element;
- }
- }
- }
- return 0;
- }
- const XMLElement* XMLNode::LastChildElement( const char* value ) const
- {
- for( XMLNode* node=lastChild; node; node=node->prev ) {
- XMLElement* element = node->ToElement();
- if ( element ) {
- if ( !value || XMLUtil::StringEqual( element->Name(), value ) ) {
- return element;
- }
- }
- }
- return 0;
- }
- void XMLNode::DeleteChild( XMLNode* node )
- {
- TIXMLASSERT( node->parent == this );
- TIXMLASSERT( 0 );
- }
- char* XMLNode::ParseDeep( char* p )
- {
- while( p && *p ) {
- XMLNode* node = 0;
- p = document->Identify( p, &node );
- if ( p && node ) {
- p = node->ParseDeep( p );
- // FIXME: is it the correct closing element?
- if ( node->IsClosingElement() ) {
- DELETE_NODE( node );
- return p;
- }
- this->InsertEndChild( node );
- }
- }
- return 0;
- }
- // --------- XMLText ---------- //
- char* XMLText::ParseDeep( char* p )
- {
- if ( this->CData() ) {
- p = value.ParseText( p, "]]>", StrPair::NEEDS_NEWLINE_NORMALIZATION );
- return p;
- }
- else {
- p = value.ParseText( p, "<", StrPair::TEXT_ELEMENT );
- // consumes the end tag.
- if ( p && *p ) {
- return p-1;
- }
- }
- return 0;
- }
- bool XMLText::Accept( XMLVisitor* visitor ) const
- {
- return visitor->Visit( *this );
- }
- // --------- XMLComment ---------- //
- XMLComment::XMLComment( XMLDocument* doc ) : XMLNode( doc )
- {
- }
- XMLComment::~XMLComment()
- {
- //printf( "~XMLComment\n" );
- }
- char* XMLComment::ParseDeep( char* p )
- {
- // Comment parses as text.
- return value.ParseText( p, "-->", StrPair::COMMENT );
- }
- bool XMLComment::Accept( XMLVisitor* visitor ) const
- {
- return visitor->Visit( *this );
- }
- // --------- XMLDeclaration ---------- //
- XMLDeclaration::XMLDeclaration( XMLDocument* doc ) : XMLNode( doc )
- {
- }
- XMLDeclaration::~XMLDeclaration()
- {
- //printf( "~XMLDeclaration\n" );
- }
- char* XMLDeclaration::ParseDeep( char* p )
- {
- // Declaration parses as text.
- return value.ParseText( p, ">", StrPair::NEEDS_NEWLINE_NORMALIZATION );
- }
- bool XMLDeclaration::Accept( XMLVisitor* visitor ) const
- {
- return visitor->Visit( *this );
- }
- // --------- XMLUnknown ---------- //
- XMLUnknown::XMLUnknown( XMLDocument* doc ) : XMLNode( doc )
- {
- }
- XMLUnknown::~XMLUnknown()
- {
- }
- char* XMLUnknown::ParseDeep( char* p )
- {
- // Unknown parses as text.
- return value.ParseText( p, ">", StrPair::NEEDS_NEWLINE_NORMALIZATION );
- }
- bool XMLUnknown::Accept( XMLVisitor* visitor ) const
- {
- return visitor->Visit( *this );
- }
- // --------- XMLAttribute ---------- //
- char* XMLAttribute::ParseDeep( char* p )
- {
- p = name.ParseText( p, "=", StrPair::ATTRIBUTE_NAME );
- if ( !p || !*p ) return 0;
- char endTag[2] = { *p, 0 };
- ++p;
- p = value.ParseText( p, endTag, StrPair::ATTRIBUTE_VALUE );
- if ( value.Empty() ) return 0;
- return p;
- }
- int XMLAttribute::QueryIntAttribute( int* value ) const
- {
- if ( TIXML_SSCANF( Value(), "%d", value ) == 1 )
- return ATTRIBUTE_SUCCESS;
- return WRONG_ATTRIBUTE_TYPE;
- }
- int XMLAttribute::QueryUnsignedAttribute( unsigned int* value ) const
- {
- if ( TIXML_SSCANF( Value(), "%u", value ) == 1 )
- return ATTRIBUTE_SUCCESS;
- return WRONG_ATTRIBUTE_TYPE;
- }
- int XMLAttribute::QueryBoolAttribute( bool* value ) const
- {
- int ival = -1;
- QueryIntAttribute( &ival );
- if ( ival > 0 || XMLUtil::StringEqual( Value(), "true" ) ) {
- *value = true;
- return ATTRIBUTE_SUCCESS;
- }
- else if ( ival == 0 || XMLUtil::StringEqual( Value(), "false" ) ) {
- *value = false;
- return ATTRIBUTE_SUCCESS;
- }
- return WRONG_ATTRIBUTE_TYPE;
- }
- int XMLAttribute::QueryDoubleAttribute( double* value ) const
- {
- if ( TIXML_SSCANF( Value(), "%lf", value ) == 1 )
- return ATTRIBUTE_SUCCESS;
- return WRONG_ATTRIBUTE_TYPE;
- }
- int XMLAttribute::QueryFloatAttribute( float* value ) const
- {
- if ( TIXML_SSCANF( Value(), "%f", value ) == 1 )
- return ATTRIBUTE_SUCCESS;
- return WRONG_ATTRIBUTE_TYPE;
- }
- void XMLAttribute::SetAttribute( const char* v )
- {
- value.SetInternedStr( v );
- }
- /*
- void XMLAttribute::SetAttribute( int v )
- {
- char buf[BUF_SIZE];
- TIXML_SNPRINTF( buf, BUF_SIZE-1, "%d" );
- value.SetInternedStr( v );
- }
- */
- // --------- XMLElement ---------- //
- XMLElement::XMLElement( XMLDocument* doc ) : XMLNode( doc ),
- closing( false ),
- rootAttribute( 0 ),
- lastAttribute( 0 )
- {
- }
- XMLElement::~XMLElement()
- {
- XMLAttribute* attribute = rootAttribute;
- while( attribute ) {
- XMLAttribute* next = attribute->next;
- DELETE_ATTRIBUTE( attribute );
- attribute = next;
- }
- }
- char* XMLElement::ParseAttributes( char* p, bool* closedElement )
- {
- const char* start = p;
- *closedElement = false;
- // Read the attributes.
- while( p ) {
- p = XMLUtil::SkipWhiteSpace( p );
- if ( !p || !(*p) ) {
- document->SetError( XMLDocument::ERROR_PARSING_ELEMENT, start, Name() );
- return 0;
- }
- // attribute.
- if ( XMLUtil::IsAlpha( *p ) ) {
- XMLAttribute* attrib = new (document->attributePool.Alloc() ) XMLAttribute( this );
- attrib->memPool = &document->attributePool;
- p = attrib->ParseDeep( p );
- if ( !p ) {
- DELETE_ATTRIBUTE( attrib );
- document->SetError( XMLDocument::ERROR_PARSING_ATTRIBUTE, start, p );
- return 0;
- }
- if ( rootAttribute ) {
- TIXMLASSERT( lastAttribute );
- lastAttribute->next = attrib;
- lastAttribute = attrib;
- }
- else {
- rootAttribute = lastAttribute = attrib;
- }
- }
- // end of the tag
- else if ( *p == '/' && *(p+1) == '>' ) {
- if ( closing ) {
- document->SetError( XMLDocument::ERROR_PARSING_ELEMENT, start, p );
- return 0;
- }
- *closedElement = true;
- return p+2; // done; sealed element.
- }
- // end of the tag
- else if ( *p == '>' ) {
- ++p;
- break;
- }
- else {
- document->SetError( XMLDocument::ERROR_PARSING_ELEMENT, start, p );
- return 0;
- }
- }
- return p;
- }
- //
- // <ele></ele>
- // <ele>foo<b>bar</b></ele>
- //
- char* XMLElement::ParseDeep( char* p )
- {
- // Read the element name.
- p = XMLUtil::SkipWhiteSpace( p );
- if ( !p ) return 0;
- const char* start = p;
- // The closing element is the </element> form. It is
- // parsed just like a regular element then deleted from
- // the DOM.
- if ( *p == '/' ) {
- closing = true;
- ++p;
- }
- p = value.ParseName( p );
- if ( value.Empty() ) return 0;
- bool elementClosed=false;
- p = ParseAttributes( p, &elementClosed );
- if ( !p || !*p || elementClosed || closing )
- return p;
- p = XMLNode::ParseDeep( p );
- return p;
- }
- bool XMLElement::Accept( XMLVisitor* visitor ) const
- {
- if ( visitor->VisitEnter( *this, rootAttribute ) )
- {
- for ( const XMLNode* node=FirstChild(); node; node=node->NextSibling() )
- {
- if ( !node->Accept( visitor ) )
- break;
- }
- }
- return visitor->VisitExit( *this );
- }
- // --------- XMLDocument ----------- //
- XMLDocument::XMLDocument() :
- XMLNode( 0 ),
- charBuffer( 0 )
- {
- document = this; // avoid warning about 'this' in initializer list
- }
- XMLDocument::~XMLDocument()
- {
- ClearChildren();
- delete [] charBuffer;
- #if 0
- textPool.Trace( "text" );
- elementPool.Trace( "element" );
- commentPool.Trace( "comment" );
- attributePool.Trace( "attribute" );
- #endif
- TIXMLASSERT( textPool.CurrentAllocs() == 0 );
- TIXMLASSERT( elementPool.CurrentAllocs() == 0 );
- TIXMLASSERT( commentPool.CurrentAllocs() == 0 );
- TIXMLASSERT( attributePool.CurrentAllocs() == 0 );
- }
- void XMLDocument::InitDocument()
- {
- errorID = NO_ERROR;
- errorStr1 = 0;
- errorStr2 = 0;
- delete [] charBuffer;
- charBuffer = 0;
- }
- XMLElement* XMLDocument::NewElement( const char* name )
- {
- XMLElement* ele = new (elementPool.Alloc()) XMLElement( this );
- ele->memPool = &elementPool;
- ele->SetName( name );
- return ele;
- }
- XMLComment* XMLDocument::NewComment( const char* str )
- {
- XMLComment* comment = new (commentPool.Alloc()) XMLComment( this );
- comment->memPool = &commentPool;
- comment->SetValue( str );
- return comment;
- }
- XMLText* XMLDocument::NewText( const char* str )
- {
- XMLText* Text = new (textPool.Alloc()) XMLText( this );
- Text->memPool = &textPool;
- Text->SetValue( str );
- return Text;
- }
- int XMLDocument::Parse( const char* p )
- {
- ClearChildren();
- InitDocument();
- if ( !p || !*p ) {
- return true; // correctly parse an empty string?
- }
- size_t len = strlen( p );
- charBuffer = new char[ len+1 ];
- memcpy( charBuffer, p, len+1 );
- XMLNode* node = 0;
-
- char* q = ParseDeep( charBuffer );
- return errorID;
- }
- void XMLDocument::Print( XMLStreamer* streamer )
- {
- XMLStreamer stdStreamer( stdout );
- if ( !streamer )
- streamer = &stdStreamer;
- //for( XMLNode* node = firstChild; node; node=node->next ) {
- // node->Print( streamer );
- //}
- Accept( streamer );
- }
- void XMLDocument::SetError( int error, const char* str1, const char* str2 )
- {
- errorID = error;
- printf( "ERROR: id=%d '%s' '%s'\n", error, str1, str2 ); // fixme: remove
- errorStr1 = str1;
- errorStr2 = str2;
- }
- /*
- StringStack::StringStack()
- {
- nPositive = 0;
- mem.Push( 0 ); // start with null. makes later code simpler.
- }
- StringStack::~StringStack()
- {
- }
- void StringStack::Push( const char* str ) {
- int needed = strlen( str ) + 1;
- char* p = mem.PushArr( needed );
- strcpy( p, str );
- if ( needed > 1 )
- nPositive++;
- }
- const char* StringStack::Pop() {
- TIXMLASSERT( mem.Size() > 1 );
- const char* p = mem.Mem() + mem.Size() - 2; // end of final string.
- if ( *p ) {
- nPositive--;
- }
- while( *p ) { // stack starts with a null, don't need to check for 'mem'
- TIXMLASSERT( p > mem.Mem() );
- --p;
- }
- mem.PopArr( strlen(p)+1 );
- return p+1;
- }
- */
- XMLStreamer::XMLStreamer( FILE* file ) : fp( file ), depth( 0 ), elementJustOpened( false ), textDepth( -1 )
- {
- for( int i=0; i<ENTITY_RANGE; ++i ) {
- entityFlag[i] = false;
- }
- for( int i=0; i<NUM_ENTITIES; ++i ) {
- TIXMLASSERT( entities[i].value < ENTITY_RANGE );
- if ( entities[i].value < ENTITY_RANGE ) {
- entityFlag[ entities[i].value ] = true;
- }
- }
- }
- void XMLStreamer::PrintSpace( int depth )
- {
- for( int i=0; i<depth; ++i ) {
- fprintf( fp, " " );
- }
- }
- void XMLStreamer::PrintString( const char* p )
- {
- // Look for runs of bytes between entities to print.
- const char* q = p;
- while ( *q ) {
- if ( *q < ENTITY_RANGE ) {
- // Check for entities. If one is found, flush
- // the stream up until the entity, write the
- // entity, and keep looking.
- if ( entityFlag[*q] ) {
- while ( p < q ) {
- fputc( *p, fp );
- ++p;
- }
- for( int i=0; i<NUM_ENTITIES; ++i ) {
- if ( entities[i].value == *q ) {
- fprintf( fp, "&%s;", entities[i].pattern );
- break;
- }
- }
- ++p;
- }
- }
- ++q;
- }
- // Flush the remaining string. This will be the entire
- // string if an entity wasn't found.
- if ( q-p > 0 ) {
- fprintf( fp, "%s", p );
- }
- }
- void XMLStreamer::OpenElement( const char* name )
- {
- if ( elementJustOpened ) {
- SealElement();
- }
- stack.Push( name );
- if ( textDepth < 0 && depth > 0) {
- fprintf( fp, "\n" );
- PrintSpace( depth );
- }
- fprintf( fp, "<%s", name );
- elementJustOpened = true;
- ++depth;
- }
- void XMLStreamer::PushAttribute( const char* name, const char* value )
- {
- TIXMLASSERT( elementJustOpened );
- fprintf( fp, " %s=\"", name );
- PrintString( value );
- fprintf( fp, "\"" );
- }
- void XMLStreamer::CloseElement()
- {
- --depth;
- const char* name = stack.Pop();
- if ( elementJustOpened ) {
- fprintf( fp, "/>" );
- }
- else {
- if ( textDepth < 0 ) {
- fprintf( fp, "\n" );
- PrintSpace( depth );
- }
- fprintf( fp, "</%s>", name );
- }
- if ( textDepth == depth )
- textDepth = -1;
- if ( depth == 0 )
- fprintf( fp, "\n" );
- elementJustOpened = false;
- }
- void XMLStreamer::SealElement()
- {
- elementJustOpened = false;
- fprintf( fp, ">" );
- }
- void XMLStreamer::PushText( const char* text, bool cdata )
- {
- textDepth = depth-1;
- if ( elementJustOpened ) {
- SealElement();
- }
- if ( cdata )
- fprintf( fp, "<![CDATA[" );
- PrintString( text );
- if ( cdata )
- fprintf( fp, "]]>" );
- }
- void XMLStreamer::PushComment( const char* comment )
- {
- if ( elementJustOpened ) {
- SealElement();
- }
- PrintSpace( depth );
- fprintf( fp, "<!--%s-->\n", comment );
- }
- bool XMLStreamer::VisitEnter( const XMLElement& element, const XMLAttribute* attribute )
- {
- OpenElement( element.Name() );
- while ( attribute ) {
- PushAttribute( attribute->Name(), attribute->Value() );
- attribute = attribute->Next();
- }
- return true;
- }
- bool XMLStreamer::VisitExit( const XMLElement& element )
- {
- CloseElement();
- return true;
- }
- bool XMLStreamer::Visit( const XMLText& text )
- {
- PushText( text.Value() );
- return true;
- }
- bool XMLStreamer::Visit( const XMLComment& comment )
- {
- PushComment( comment.Value() );
- return true;
- }
|