1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

#include "TableLoader.h"
#include <assert.h>
#include <iostream>

void TableWalker::OnStartElement(const XML_Char *name, const XML_Char **atts)
{
	if (curRec) {
		wcsncpy_s(curField, MAX_XML_TAG_NAME_SIZE, name, MAX_XML_TAG_NAME_SIZE-1);
		return;
	}
	if (!_wcsicmp(recName, name))
		CreateRecord();
}

void TableWalker::OnEndElement(const XML_Char *name)
{
	if (!_wcsicmp(recName, name)) {
		AddRecord();
		return;
	}
	if (!_wcsicmp(curField, name))
		ClearField();
}

/** \brief Parse XML and fill in the table */
BOOL TableLoader::Load(ISax2dTable* table)
{
	assert(table);

	// Check file and record name
	const TableDesc& desc = table->GetDesc();
	if (desc.fileName.length() == 0 || desc.recName.length() == 0)
		return FALSE;

	// Open file
	FILE*	pFile = NULL;
	errno_t	err = _wfopen_s(&pFile, desc.fileName.c_str(), L"r");
	if (pFile == NULL)
		return FALSE;
	int len = fread(m_xml, sizeof(char), sizeof(m_xml), pFile);
	if (len == 0)
		return FALSE;

	// Init table walker
	m_walker.pTable = table;
	wcsncpy_s(m_walker.recName, MAX_XML_TAG_NAME_SIZE, desc.recName.c_str(), desc.recName.length());
	m_walker.curRec = NULL;
	m_walker.ClearField();

#ifdef EXPAT
	// Parse by Expat
	XML_Parser			parser	= XML_ParserCreate(XML_INPUT_INCODING);
	XML_SetUserData			(parser, &m_walker);
	XML_SetElementHandler		(parser, &TableLoader::OnStartElement, &TableLoader::OnEndElement);
	XML_SetCharacterDataHandler	(parser, &TableLoader::OnCharacterData);
	XML_Status status = XML_Parse(parser, m_xml, len, TRUE);
	if (status != XML_STATUS_OK)
		return HandleLoadingError(XML_GetErrorCode(parser));
#endif
	return TRUE;
}

#ifdef EXPAT
BOOL TableLoader::HandleLoadingError(XML_Error error)
{
	switch (error)
	{
	case XML_ERROR_NO_ELEMENTS:
		return TRUE;
	default:
		// TODO: report error
		return FALSE;
	}
	return FALSE;
}

/** \brief Expat start element handler */
void XMLCALL TableLoader::OnStartElement(void *userData, const XML_Char *name, const XML_Char **atts)
{
	assert(userData);
	TableWalker* pWalker = static_cast<TableWalker*>(userData);
	pWalker->OnStartElement(name, atts);
#ifdef _DEBUG_MSG
	std::wcout<<L"OnStartElement: "<<name<<L"\n";
#endif
}

/** \brief Expat end element handler */
void XMLCALL TableLoader::OnEndElement(void *userData, const XML_Char *name)
{
	assert(userData);
	TableWalker* pWalker = static_cast<TableWalker*>(userData);
	pWalker->OnEndElement(name);
#ifdef _DEBUG_MSG
	std::wcout<<L"OnEndElement:   "<<name<<L"\n";
#endif
}

/** \brief Expat CDATA handler */
void XMLCALL TableLoader::OnCharacterData(void *userData, const XML_Char *s, int len)
{
	assert(userData);
	TableWalker* pWalker = static_cast<TableWalker*>(userData);
	if (pWalker->HasCurField() && s)
	{
		wchar_t	buf[MAX_XML_TAG_NAME_SIZE] = {0,};
		wcsncpy_s(buf, MAX_XML_TAG_NAME_SIZE, s, len);
		pWalker->AddField(buf);
#ifdef _DEBUG_MSG
		std::wcout<<L"OnCDATA: "<<buf<<L"\n";
#endif
	}
}
#endif // EXPAT