/*
 * This file is a part of PikoTools
 * and is distributed under the (new) BSD licence.
 * Author: Tomasz Sowa <t.sowa@ttmath.org>
 */

/* 
 * Copyright (c) 2012, Tomasz Sowa
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * 
 *  * Redistributions of source code must retain the above copyright notice,
 *    this list of conditions and the following disclaimer.
 *    
 *  * Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *    
 *  * Neither the name Tomasz Sowa nor the names of contributors to this
 *    project may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
 * THE POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef headerfile_picotools_space_spacetojson
#define headerfile_picotools_space_spacetojson

#include <string>
#include <vector>
#include <set>
#include "space.h"


namespace PT
{


class SpaceToJSON
{
public:

	void ChangeSpacesToTable(const wchar_t * space_name);
	void ChangeSpacesToTable(const std::wstring & space_name);

	void TreatAsNumeric(const wchar_t * name);
	void TreatAsNumeric(const std::wstring & name);

	void TreatAsBool(const wchar_t * name);
	void TreatAsBool(const std::wstring & name);



	template<class Stream>
	void Serialize(Space & space, Stream & out, bool use_indents = false, int level = 0, bool use_comma = false, bool skip_name = false);

	template<class Stream>
	void SerializeTableSingle(Space & space, Stream & out, bool use_indents, int level, bool use_comma);

	template<class Stream>
	void SerializeTableMulti(Space & space, Stream & out, bool use_indents, int level, bool use_comma);

	template<class Stream, class StringType>
	void PrintToken(Stream & out, const StringType & str, bool check_specials = false);

	template<class Stream>
	void PrintLevel(Stream & out, bool use_indents, int level);


private:

	bool HasUnnamedSpace(Space & space);
	bool HasSpace(Space & space, const std::wstring & name);

	bool IsNumeric(const std::wstring & name);
	bool IsBool(const std::wstring & name);

	std::vector<std::wstring> spaces_to_table;
	std::vector<bool> space_used;
	std::set<std::wstring> numeric, boolean;
};




template<class Stream>
void SpaceToJSON::PrintLevel(Stream & out, bool use_indents, int level)
{
	if( use_indents )
	{
		for(int i=0 ; i<level ; ++i)
			out << ' ';
	}
}




template<class Stream, class StringType>
void SpaceToJSON::PrintToken(Stream & out, const StringType & str, bool is_special)
{
	if( !is_special )
		out << '\"';

	for(size_t i=0 ; i<str.size() ; ++i)
	{
		switch(str[i])
		{
		case 0:		out << '\\';	out << '0';		break;	// may skip this character is better?
		case '\r':	out << '\\';	out << 'r';		break;
		case '\n':	out << '\\';	out << 'n';		break;
		case '\t':	out << '\\';	out << 't';		break;
		case 0x08:	out << '\\';	out << 'b';		break;
		case 0x0c:	out << '\\';	out << 'f';		break;
		case '\\':	out << '\\';	out << '\\';	break;
		case '"':	out << '\\';	out << '\"';	break;
		default:
			out << str[i];
		}
	}

	if( !is_special )
		out << '\"';
}






template<class Stream>
void SpaceToJSON::SerializeTableSingle(Space & space, Stream & out, bool use_indents, int level, bool use_comma)
{
Space::TableSingle::const_iterator i;
size_t index = 0;
bool is_special;

	if( use_comma && !space.table_single.empty() )
	{
		PrintLevel(out, use_indents, level);
		out << L",\n";
	}

	for(i=space.table_single.begin() ; i != space.table_single.end() ; ++i, ++index)
	{
		is_special = IsNumeric(i->first) || IsBool(i->first);

		PrintLevel(out, use_indents, level);
		PrintToken(out, i->first);
		out << L": ";
		PrintToken(out, i->second, is_special);

		if( index + 1 < space.table_single.size() )
			out << ',';

		out << '\n';
	}
}



template<class Stream>
void SpaceToJSON::SerializeTableMulti(Space & space, Stream & out, bool use_indents, int level, bool use_comma)
{
Space::Table::const_iterator i2;
size_t v;
size_t index = 0;
bool is_special;

	if( use_comma && !space.table.empty() )
	{
		PrintLevel(out, use_indents, level);
		out << L",\n";
	}

	for(i2 = space.table.begin() ; i2 != space.table.end() ; ++i2, ++index)
	{
		is_special = IsNumeric(i2->first) || IsBool(i2->first);

		PrintLevel(out, use_indents, level);
		PrintToken(out, i2->first);
		out << L": ";

		if( i2->second.size() != 1 )
			out << '[';

		for(v = 0 ; v < i2->second.size() ; ++v)
		{
			if( v > 0 )
				PrintLevel(out, use_indents, level + i2->first.size() + 3);

			PrintToken(out, i2->second[v], is_special);

			if( v + 1 < i2->second.size() )
				out << L",\n";
		}

		if( i2->second.size() != 1 )
			out << ']';

		if( index + 1 < space.table.size() )
			out << ',';

		out << '\n';
	}
}




template<class Stream>
void SpaceToJSON::Serialize(Space & space, Stream & out, bool use_indents, int level, bool use_comma, bool skip_name)
{
	if( use_comma )
	{
		PrintLevel(out, use_indents, level);
		out << L",\n";
	}

	PrintLevel(out, use_indents, level);

	if( !skip_name )
	{
		if( !space.name.empty() )
		{
			PrintToken(out, space.name);
			out << L": ";
		}
	}

	out << L"{\n";

	bool printed_something = false;

	SerializeTableSingle(space, out, use_indents, level, false);
	SerializeTableMulti(space, out, use_indents, level, !space.table_single.empty());

	if( !space.table_single.empty() || !space.table.empty() )
		printed_something = true;

	space_used.resize(space.spaces.size());

	for(size_t i=0 ; i<space_used.size() ; ++i)
		space_used[i] = false;


	for(size_t z=0 ; z<spaces_to_table.size() ; ++z)
	{
		if( HasSpace(space, spaces_to_table[z]) )
		{
			if( printed_something )
			{
				out << L",\n";
			}

			PrintToken(out, spaces_to_table[z]);
			out << L": [\n";

			for(size_t i=0 ; i<space.spaces.size() ; ++i)
			{
				if( space.spaces[i]->name == spaces_to_table[z] )
				{
					space_used[i] = true;
					Serialize(*space.spaces[i], out, use_indents, level+1, i>0, true);
				}
			}

			out << L"]\n";
			printed_something = true;
		}
	}


	for(size_t i=0 ; i<space.spaces.size() ; ++i)
	{
		if( !space_used[i] )
		{
			Serialize(*space.spaces[i], out, use_indents, level+1, printed_something || i>0);

			if( i + 1 < space.spaces.size() )
			{
				PrintLevel(out, use_indents, level);
				out << L",\n";
			}

			printed_something = true;
		}
	}

	PrintLevel(out, use_indents, level);
	out << L"}\n";
}




} // namespace

#endif