1 files changed, 499 insertions, 0 deletions
diff --git a/src/XMLNode.cpp b/src/XMLNode.cpp
new file mode 100644
index 0000000..e47c3e7
--- /dev/null
+++ b/src/XMLNode.cpp
@@ -0,0 +1,499 @@
+/*
+-----------------------------------------------------------------------
+Copyright 2012 iMinds-Vision Lab, University of Antwerp
+
+Contact: astra@ua.ac.be
+Website: http://astra.ua.ac.be
+
+
+This file is part of the
+All Scale Tomographic Reconstruction Antwerp Toolbox ("ASTRA Toolbox").
+
+The ASTRA Toolbox is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+The ASTRA Toolbox is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with the ASTRA Toolbox. If not, see <http://www.gnu.org/licenses/>.
+
+-----------------------------------------------------------------------
+$Id$
+*/
+
+#include "astra/XMLNode.h"
+
+#ifdef _MSC_VER
+#include "rapidxml/rapidxml.hpp"
+#include "rapidxml/rapidxml_print.hpp"
+#else
+#include "rapidxml.hpp"
+#include "rapidxml_print.hpp"
+#endif
+
+#include <boost/lexical_cast.hpp>
+
+using namespace rapidxml;
+using namespace astra;
+using namespace std;
+
+
+//-----------------------------------------------------------------------------
+// Utility function to delete a list of nodes
+static void deleteNodes(list<XMLNode*>& nodes)
+{
+	for (list<XMLNode*>::iterator i = nodes.begin(); i != nodes.end(); ++i)
+		delete (*i);
+
+	nodes.clear();
+}
+
+
+//-----------------------------------------------------------------------------
+// default constructor
+XMLNode::XMLNode() 
+{
+
+}
+
+//-----------------------------------------------------------------------------	
+// private constructor
+XMLNode::XMLNode(xml_node<>* node) 
+{
+	fDOMElement = node;
+}
+
+//-----------------------------------------------------------------------------	
+// destructor
+XMLNode::~XMLNode() 
+{
+
+}
+
+//-----------------------------------------------------------------------------	
+// set DOM node (private)
+void XMLNode::setDOMNode(xml_node<>* n) 
+{
+	fDOMElement = n;
+}
+
+//-----------------------------------------------------------------------------	
+// print XML Node
+void XMLNode::print()
+{
+	std::cout << fDOMElement;
+}
+
+//-----------------------------------------------------------------------------	
+// print XML Node
+std::string XMLNode::toString()
+{
+	std::string s;
+	::print(std::back_inserter(s), *fDOMElement, 0);
+	return s;
+}
+
+//-----------------------------------------------------------------------------	
+// Get single node
+XMLNode* XMLNode::getSingleNode(string name) 
+{
+	xml_node<> *node = fDOMElement->first_node(name.c_str());
+
+	if (node)
+		return new XMLNode(node);
+	else
+		return 0;
+}
+
+//-----------------------------------------------------------------------------	
+// Get list of nodes
+list<XMLNode*> XMLNode::getNodes(string name) 
+{	
+	list<XMLNode*> result;
+	xml_node<> *iter;
+	for (iter = fDOMElement->first_node(name.c_str()); iter; iter = iter->next_sibling(name.c_str())) {
+		result.push_back(new XMLNode(iter));
+	}
+	return result;
+}
+
+//-----------------------------------------------------------------------------	
+// Get list of nodes
+list<XMLNode*> XMLNode::getNodes() 
+{	
+	list<XMLNode*> result;
+	xml_node<> *iter;
+	for (iter = fDOMElement->first_node(); iter; iter = iter->next_sibling()) {
+		result.push_back(new XMLNode(iter));
+	}
+	return result;
+}
+
+//-----------------------------------------------------------------------------	
+// Get name of this node
+std::string XMLNode::getName()
+{
+	return fDOMElement->name();
+}
+
+//-----------------------------------------------------------------------------	
+// Get node content - STRING
+string XMLNode::getContent() 
+{
+	return fDOMElement->value();
+}
+
+//-----------------------------------------------------------------------------	
+// Get node content - NUMERICAL
+float32 XMLNode::getContentNumerical() 
+{
+	return boost::lexical_cast<float32>(getContent());
+}
+
+//-----------------------------------------------------------------------------	
+// Get node content - BOOLEAN
+bool XMLNode::getContentBool() 
+{
+	string res = getContent();
+	return ((res == "1") || (res == "yes") || (res == "true") || (res == "on"));
+}
+
+//-----------------------------------------------------------------------------	
+// Get node content - STRING LIST
+vector<string> XMLNode::getContentArray()
+{
+	// get listsize
+	int iSize = boost::lexical_cast<int>(getAttribute("listsize"));
+	// create result array
+	vector<string> res(iSize);
+	// loop all list item nodes
+	list<XMLNode*> nodes = getNodes("ListItem");
+	for (list<XMLNode*>::iterator it = nodes.begin(); it != nodes.end(); it++) {
+		int iIndex = (*it)->getAttributeNumerical("index");
+		string sValue = (*it)->getAttribute("value");
+		ASTRA_ASSERT(iIndex < iSize);
+		res[iIndex] = sValue;
+	}
+	deleteNodes(nodes);
+
+	// return 
+	return res;
+}
+
+//-----------------------------------------------------------------------------	
+// Get node content - NUMERICAL LIST
+vector<float32> XMLNode::getContentNumericalArray()
+{
+	// is scalar
+	if (!hasAttribute("listsize")) {
+		vector<float32> res(1);
+		res[0] = getContentNumerical();
+		return res;
+	}
+
+	int iSize = boost::lexical_cast<int>(getAttribute("listsize"));
+	// create result array
+	vector<float32> res(iSize);
+	// loop all list item nodes
+	list<XMLNode*> nodes = getNodes("ListItem");
+	for (list<XMLNode*>::iterator it = nodes.begin(); it != nodes.end(); it++) {
+		int iIndex = (*it)->getAttributeNumerical("index");
+		float32 fValue = (*it)->getAttributeNumerical("value");
+		ASTRA_ASSERT(iIndex < iSize);
+		res[iIndex] = fValue;
+	}
+	deleteNodes(nodes);
+	// return 
+	return res;
+}
+
+vector<double> XMLNode::getContentNumericalArrayDouble()
+{
+	// is scalar
+	if (!hasAttribute("listsize")) {
+		vector<double> res(1);
+		res[0] = getContentNumerical();
+		return res;
+	}
+
+	int iSize = boost::lexical_cast<int>(getAttribute("listsize"));
+	// create result array
+	vector<double> res(iSize);
+	// loop all list item nodes
+	list<XMLNode*> nodes = getNodes("ListItem");
+	for (list<XMLNode*>::iterator it = nodes.begin(); it != nodes.end(); it++) {
+		int iIndex = (*it)->getAttributeNumerical("index");
+		double fValue = (*it)->getAttributeNumericalDouble("value");
+		ASTRA_ASSERT(iIndex < iSize);
+		res[iIndex] = fValue;
+	}
+	deleteNodes(nodes);
+	// return 
+	return res;
+}
+
+//-----------------------------------------------------------------------------	
+// Get node content - NUMERICAL LIST 2
+void XMLNode::getContentNumericalArray(float32*& _pfData, int& _iSize)
+{
+	// is scalar
+	if (!hasAttribute("listsize")) {
+		_iSize = 1;
+		_pfData = new float32[_iSize];
+		_pfData[0] = getContentNumerical();
+		return;
+	}
+	// get listsize
+	_iSize = boost::lexical_cast<int>(getAttribute("listsize"));
+	// create result array
+	_pfData = new float32[_iSize];
+	// loop all list item nodes
+	list<XMLNode*> nodes = getNodes("ListItem");
+	for (list<XMLNode*>::iterator it = nodes.begin(); it != nodes.end(); it++) {
+		int iIndex = (*it)->getAttributeNumerical("index");
+		float32 fValue = (*it)->getAttributeNumerical("value");
+		ASTRA_ASSERT(iIndex < _iSize);
+		_pfData[iIndex] = fValue;
+	}
+	deleteNodes(nodes);
+}
+
+//-----------------------------------------------------------------------------	
+// Is attribute?
+bool XMLNode::hasAttribute(string _sName)
+{
+	xml_attribute<> *attr = fDOMElement->first_attribute(_sName.c_str());
+	return (attr != 0);
+}
+
+//-----------------------------------------------------------------------------	
+// Get attribute - STRING
+string XMLNode::getAttribute(string _sName, string _sDefaultValue)
+{
+	xml_attribute<> *attr = fDOMElement->first_attribute(_sName.c_str());
+
+	if (!attr) return _sDefaultValue;
+
+	return attr->value();
+}
+
+//-----------------------------------------------------------------------------	
+// Get attribute - NUMERICAL
+float32 XMLNode::getAttributeNumerical(string _sName, float32 _fDefaultValue)
+{
+	if (!hasAttribute(_sName)) return _fDefaultValue;
+	return boost::lexical_cast<float32>(getAttribute(_sName));
+}
+double XMLNode::getAttributeNumericalDouble(string _sName, double _fDefaultValue)
+{
+	if (!hasAttribute(_sName)) return _fDefaultValue;
+	return boost::lexical_cast<double>(getAttribute(_sName));
+}
+
+//-----------------------------------------------------------------------------	
+// Get attribute - BOOLEAN
+bool XMLNode::getAttributeBool(string _sName, bool _bDefaultValue)
+{
+	if (!hasAttribute(_sName)) return _bDefaultValue;
+	string res = getAttribute(_sName);
+	return ((res == "1") || (res == "yes") || (res == "true") || (res == "on"));
+}
+
+//-----------------------------------------------------------------------------	
+// Has option?
+bool XMLNode::hasOption(string _sKey) 
+{
+	xml_node<> *iter;
+	for (iter = fDOMElement->first_node("Option"); iter; iter = iter->next_sibling("Option")) {
+		xml_attribute<> *attr = iter->first_attribute("key");
+		if (attr && _sKey == attr->value())
+			return true;
+	}
+	return false;
+}
+
+//-----------------------------------------------------------------------------	
+// Get option - STRING
+string XMLNode::getOption(string _sKey, string _sDefaultValue) 
+{
+	xml_node<> *iter;
+	for (iter = fDOMElement->first_node("Option"); iter; iter = iter->next_sibling("Option")) {
+		xml_attribute<> *attr = iter->first_attribute("key");
+		if (attr && _sKey == attr->value()) {
+			attr = iter->first_attribute("value");
+			if (!attr)
+				return "";
+			return attr->value();
+		}
+	}
+	return _sDefaultValue;
+}
+
+//-----------------------------------------------------------------------------	
+// Get option - NUMERICAL
+float32 XMLNode::getOptionNumerical(string _sKey, float32 _fDefaultValue) 
+{
+	if (!hasOption(_sKey)) return _fDefaultValue;
+	return boost::lexical_cast<float32>(getOption(_sKey));
+}
+
+//-----------------------------------------------------------------------------	
+// Get option - BOOL
+bool XMLNode::getOptionBool(string _sKey, bool _bDefaultValue)
+{
+	bool bHasOption = hasOption(_sKey);
+	if (!bHasOption) return _bDefaultValue;
+	string res = getOption(_sKey);
+	return ((res == "1") || (res == "yes") || (res == "true") || (res == "on"));
+}
+
+//-----------------------------------------------------------------------------	
+// Get option - NUMERICAL ARRAY
+vector<float32> XMLNode::getOptionNumericalArray(string _sKey)
+{
+	if (!hasOption(_sKey)) return vector<float32>();
+
+	list<XMLNode*> nodes = getNodes("Option");
+	for (list<XMLNode*>::iterator it = nodes.begin(); it != nodes.end(); it++) {
+		if ((*it)->getAttribute("key") == _sKey) {
+			vector<float32> vals = (*it)->getContentNumericalArray();
+			deleteNodes(nodes);
+			return vals;
+		}
+	}
+
+	deleteNodes(nodes);
+	return vector<float32>();
+}
+
+//-----------------------------------------------------------------------------	
+
+
+
+
+
+
+
+
+
+
+
+
+//-----------------------------------------------------------------------------	
+// BUILD NODE
+//-----------------------------------------------------------------------------	
+
+//-----------------------------------------------------------------------------
+// Add child node - EMPTY
+XMLNode* XMLNode::addChildNode(string _sNodeName) 
+{
+	xml_document<> *doc = fDOMElement->document();
+	char *node_name = doc->allocate_string(_sNodeName.c_str());
+	xml_node<> *node = doc->allocate_node(node_element, node_name);
+	fDOMElement->append_node(node);
+
+	// TODO: clean up: this 'new' requires callers to do memory management
+	return new XMLNode(node);
+}
+
+//-----------------------------------------------------------------------------
+// Add child node - STRING
+XMLNode* XMLNode::addChildNode(string _sNodeName, string _sText) 
+{
+	XMLNode* res = addChildNode(_sNodeName);
+	res->setContent(_sText);
+	return res;
+}
+
+//-----------------------------------------------------------------------------
+// Add child node - FLOAT
+XMLNode* XMLNode::addChildNode(string _sNodeName, float32 _fValue) 
+{
+	XMLNode* res = addChildNode(_sNodeName);
+	res->setContent(_fValue);
+	return res;
+}
+
+//-----------------------------------------------------------------------------
+// Add child node - LIST
+XMLNode* XMLNode::addChildNode(string _sNodeName, float32* _pfList, int _iSize) 
+{
+	XMLNode* res = addChildNode(_sNodeName);
+	res->setContent(_pfList, _iSize);
+	return res;
+}
+
+//-----------------------------------------------------------------------------	
+// Set content - STRING
+void XMLNode::setContent(string _sText) 
+{
+	xml_document<> *doc = fDOMElement->document();
+	char *text = doc->allocate_string(_sText.c_str());
+	fDOMElement->value(text);
+}
+
+//-----------------------------------------------------------------------------	
+// Set content - FLOAT
+void XMLNode::setContent(float32 _fValue) 
+{
+	setContent(boost::lexical_cast<string>(_fValue));
+}
+
+//-----------------------------------------------------------------------------	
+// Set content - LIST
+void XMLNode::setContent(float32* pfList, int _iSize) 
+{
+	addAttribute("listsize", _iSize);
+	for (int i = 0; i < _iSize; i++) {
+		XMLNode* item = addChildNode("ListItem");
+		item->addAttribute("index", i);
+		item->addAttribute("value",pfList[i]);
+		delete item;
+	}
+}
+
+//-----------------------------------------------------------------------------	
+// Add attribute - STRING
+void XMLNode::addAttribute(string _sName, string _sText) 
+{
+	xml_document<> *doc = fDOMElement->document();
+	char *name = doc->allocate_string(_sName.c_str());
+	char *text = doc->allocate_string(_sText.c_str());
+	xml_attribute<> *attr = doc->allocate_attribute(name, text);
+	fDOMElement->append_attribute(attr);
+}
+
+//-----------------------------------------------------------------------------	
+// Add attribute - FLOAT
+void XMLNode::addAttribute(string _sName, float32 _fValue) 
+{
+	addAttribute(_sName, boost::lexical_cast<string>(_fValue)); 
+}
+
+//-----------------------------------------------------------------------------	
+// Add option - STRING
+void XMLNode::addOption(string _sName, string _sText) 
+{
+	XMLNode* node = addChildNode("Option");
+	node->addAttribute("key", _sName);
+	node->addAttribute("value", _sText);
+	delete node;
+}
+
+//-----------------------------------------------------------------------------	
+// Add option - FLOAT
+void XMLNode::addOption(string _sName, float32 _sText) 
+{
+	XMLNode* node = addChildNode("Option");
+	node->addAttribute("key", _sName);
+	node->addAttribute("value", _sText);
+	delete node;
+}
+//-----------------------------------------------------------------------------	
+
+