pikotools/space/spacetojson.h

313 lines
7.2 KiB
C++

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