/*
-----------------------------------------------------------------------
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$
*/


#define BOOST_TEST_DYN_LINK
#include <boost/test/unit_test.hpp>
#include <boost/test/auto_unit_test.hpp>

#include "astra/XMLDocument.h"

BOOST_AUTO_TEST_CASE( testXMLDocument_Constructor1 )
{
	astra::XMLDocument *doc = astra::XMLDocument::createDocument("test");
	BOOST_REQUIRE(doc);

	astra::XMLNode *root = doc->getRootNode();

	BOOST_REQUIRE(root);

	BOOST_CHECK(root->getName() == "test");
	BOOST_CHECK(root->getContent().empty());
}

BOOST_AUTO_TEST_CASE( testXMLDocument_FileIO )
{
	astra::XMLDocument *doc = astra::XMLDocument::createDocument("test");

	doc->saveToFile("test.xml");

	astra::XMLDocument *doc2 = astra::XMLDocument::readFromFile("test.xml");
	astra::XMLNode *root = doc2->getRootNode();

	BOOST_REQUIRE(root);

	BOOST_CHECK(root->getName() == "test");
	BOOST_CHECK(root->getContent().empty());

}

BOOST_AUTO_TEST_CASE( testXMLDocument_CreateNodes )
{
	astra::XMLDocument *doc = astra::XMLDocument::createDocument("test");
	BOOST_REQUIRE(doc);

	astra::XMLNode *root = doc->getRootNode();
	BOOST_REQUIRE(root);

	astra::XMLNode *node = root->addChildNode("child");
	BOOST_REQUIRE(node);

	node->addAttribute("attr", "val");

	doc->saveToFile("test2.xml");

	delete node;
	delete root;
	delete doc;

	doc = astra::XMLDocument::readFromFile("test2.xml");
	BOOST_REQUIRE(doc);
	root = doc->getRootNode();
	BOOST_REQUIRE(node);
	node = root->getSingleNode("child");
	BOOST_REQUIRE(node);

	BOOST_CHECK(node->hasAttribute("attr"));
	BOOST_CHECK(node->getAttribute("attr") == "val");

	delete node;
	delete root;
	delete doc;
}

BOOST_AUTO_TEST_CASE( testXMLDocument_Options )
{
	astra::XMLDocument *doc = astra::XMLDocument::createDocument("test");
	BOOST_REQUIRE(doc);

	astra::XMLNode *root = doc->getRootNode();
	BOOST_REQUIRE(root);

	BOOST_CHECK(!root->hasOption("opt"));

	root->addOption("opt", "val");

	BOOST_CHECK(root->hasOption("opt"));

	BOOST_CHECK(root->getOption("opt") == "val");

}

BOOST_AUTO_TEST_CASE( testXMLDocument_List )
{
	astra::XMLDocument *doc = astra::XMLDocument::createDocument("test");
	BOOST_REQUIRE(doc);

	astra::XMLNode *root = doc->getRootNode();
	BOOST_REQUIRE(root);

	astra::XMLNode *node = root->addChildNode("child");
	BOOST_REQUIRE(node);


	float fl[] = { 1.0, 3.5, 2.0, 4.75 };

	node->setContent(fl, sizeof(fl)/sizeof(fl[0]));

	doc->saveToFile("test3.xml");

	delete node;
	delete root;
	delete doc;

	doc = astra::XMLDocument::readFromFile("test3.xml");
	BOOST_REQUIRE(doc);
	root = doc->getRootNode();
	BOOST_REQUIRE(root);
	node = root->getSingleNode("child");
	BOOST_REQUIRE(node);

	std::vector<astra::float32> f = node->getContentNumericalArray();

	BOOST_CHECK(f[0] == fl[0]);
	BOOST_CHECK(f[1] == fl[1]);
	BOOST_CHECK(f[2] == fl[2]);
	BOOST_CHECK(f[3] == fl[3]);

	delete node;
	delete root;
	delete doc;

}