/*****************************************************************************
 *
 * This file is part of Mapnik (c++ mapping toolkit)
 *
 * Copyright (C) 2006 Artem Pavlenko
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 *****************************************************************************/
// $Id$

// mapnik
#include <mapnik/save_map.hpp>
#include <mapnik/image_util.hpp>
#include <mapnik/ptree_helpers.hpp>
#include <mapnik/expression_string.hpp>
#include <mapnik/raster_colorizer.hpp>

// boost
#include <boost/algorithm/string.hpp>
#include <boost/lexical_cast.hpp>
#include <boost/tokenizer.hpp>
#include <boost/optional.hpp>
#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/xml_parser.hpp>

// stl
#include <iostream>

namespace mapnik
{
using boost::property_tree::ptree;
using boost::optional;

void serialize_raster_colorizer(ptree & sym_node,
				raster_colorizer_ptr const& colorizer,
				bool explicit_defaults)
{
    ptree & col_node = sym_node.push_back(
	ptree::value_type("RasterColorizer", ptree() ))->second;

    unsigned i;
    color_bands const &cb = colorizer->get_color_bands();
    for (i=0; i<cb.size(); i++) {
	if (!cb[i].is_interpolated()) {
	    ptree & band_node = col_node.push_back(
		ptree::value_type("ColorBand", ptree())
		)->second;
	    set_attr(band_node, "value", cb[i].get_value());    
	    set_attr(band_node, "midpoints", cb[i].get_midpoints());    
	    optional<color> c = cb[i].get_color();
	    if (c) set_attr(band_node, "color", * c);    
	}
    }

}

class serialize_symbolizer : public boost::static_visitor<>
{
public:
    serialize_symbolizer( ptree & r , bool explicit_defaults):
	rule_(r),
	explicit_defaults_(explicit_defaults) {}

    void operator () ( const  point_symbolizer & sym )
    {
	ptree & sym_node = rule_.push_back(
	    ptree::value_type("PointSymbolizer", ptree()))->second;

	add_image_attributes( sym_node, sym );

	point_symbolizer dfl;
	if (sym.get_allow_overlap() != dfl.get_allow_overlap() || explicit_defaults_ )
	{
	    set_attr( sym_node, "allow_overlap", sym.get_allow_overlap() );
	}
	if ( sym.get_opacity() != dfl.get_opacity() || explicit_defaults_ )
	{
	    set_attr( sym_node, "opacity", sym.get_opacity() );
	}
    }

    void operator () ( const line_symbolizer & sym )
    {
	ptree & sym_node = rule_.push_back(
	    ptree::value_type("LineSymbolizer", ptree()))->second;
	const stroke & strk = sym.get_stroke();
	stroke dfl = stroke();
	
	if ( strk.get_color() != dfl.get_color() || explicit_defaults_ )
	{
	    set_attr( sym_node, "stroke", strk.get_color() );
	}
	if ( strk.get_width() != dfl.get_width() || explicit_defaults_ )
	{
	    set_attr( sym_node, "stroke-width", strk.get_width() );
	}
	if ( strk.get_opacity() != dfl.get_opacity() || explicit_defaults_ )
	{
	    set_attr( sym_node, "stroke-opacity", strk.get_opacity() );
	}
	if ( strk.get_line_join() != dfl.get_line_join() || explicit_defaults_ )
	{
	    set_attr( sym_node, "stroke-linejoin", strk.get_line_join() );
	}
	if ( strk.get_line_cap() != dfl.get_line_cap() || explicit_defaults_ )
	{
	    set_attr( sym_node, "stroke-linecap", strk.get_line_cap() );
	}
	if ( ! strk.get_dash_array().empty() )
	{
	    std::ostringstream os;
	    const dash_array & dashes = strk.get_dash_array();
	    for (unsigned i = 0; i < dashes.size(); ++i) {
		os << dashes[i].first << ", " << dashes[i].second;
		if ( i + 1 < dashes.size() ) os << ", ";
	    }
	    set_attr( sym_node, "stroke-dasharray", os.str() );
	}
	if ( strk.dash_offset() != dfl.dash_offset() || explicit_defaults_ )
	{
	    set_attr( sym_node, "stroke-dashoffset", strk.dash_offset());
	}
    }
	
    void operator () ( const line_pattern_symbolizer & sym )
    {
	ptree & sym_node = rule_.push_back(
	    ptree::value_type("LinePatternSymbolizer",
			      ptree()))->second;

	add_image_attributes( sym_node, sym );
    }

    void operator () ( const polygon_symbolizer & sym )
    {
	ptree & sym_node = rule_.push_back(
	    ptree::value_type("PolygonSymbolizer", ptree()))->second;
	polygon_symbolizer dfl;

	if ( sym.get_fill() != dfl.get_fill() || explicit_defaults_ )
	{
	    set_css( sym_node, "fill", sym.get_fill() );
	}
	if ( sym.get_opacity() != dfl.get_opacity() || explicit_defaults_ )
	{
	    set_css( sym_node, "fill-opacity", sym.get_opacity() );
	}
	if ( sym.get_gamma() != dfl.get_gamma() || explicit_defaults_ )
	{
	    set_css( sym_node, "gamma", sym.get_gamma() );
	}
    }

    void operator () ( const polygon_pattern_symbolizer & sym )
    {
	ptree & sym_node = rule_.push_back(
	    ptree::value_type("PolygonPatternSymbolizer",
			      ptree()))->second;

	add_image_attributes( sym_node, sym );
    }

    void operator () ( const raster_symbolizer & sym )
    {
	ptree & sym_node = rule_.push_back(
	    ptree::value_type("RasterSymbolizer", ptree()))->second;
	raster_symbolizer dfl;

	if ( sym.get_mode() != dfl.get_mode() || explicit_defaults_ )
	{
	    set_css( sym_node, "mode", sym.get_mode() );
	}
	if ( sym.get_scaling() != dfl.get_scaling() || explicit_defaults_ )
	{
	    set_css( sym_node, "scaling", sym.get_scaling() );
	}
	if ( sym.get_opacity() != dfl.get_opacity() || explicit_defaults_ )
	{
	    set_css( sym_node, "opacity", sym.get_opacity() );
	}
	if (sym.get_colorizer()) {
	    serialize_raster_colorizer(sym_node, sym.get_colorizer(),
				       explicit_defaults_);
	}
                    
    }

    void operator () ( const shield_symbolizer & sym )
    {
	ptree & sym_node = rule_.push_back(
	    ptree::value_type("ShieldSymbolizer",
			      ptree()))->second;

	add_font_attributes( sym_node, sym);
	add_image_attributes( sym_node, sym);

	// pseudo-default-construct a shield_symbolizer. It is used
	// to avoid printing of attributes with default values without
	// repeating the default values here.
	// maybe add a real, explicit default-ctor?

	//FIXME pls
	/*
	  shield_symbolizer sym_dfl("<no default>", "<no default>", 0, color(0,0,0), "<no default>", "<no default>");
	  if (sym.get_unlock_image() != sym_dfl.get_unlock_image() || explicit_defaults_ )
	  {
	  set_attr( sym_node, "unlock_image", sym.get_unlock_image() );
	  }
	  if (sym.get_no_text() != sym_dfl.get_no_text() || explicit_defaults_ )
	  {
	  set_attr( sym_node, "no_text", sym.get_no_text() );
	  }
	*/
    }

    void operator () ( const text_symbolizer & sym )
    {
	ptree & sym_node = rule_.push_back(
	    ptree::value_type("TextSymbolizer",
			      ptree()))->second;

	add_font_attributes( sym_node, sym);

    }

    void operator () ( const building_symbolizer & sym )
    {
	ptree & sym_node = rule_.push_back(
	    ptree::value_type("BuildingSymbolizer", ptree()))->second;
	building_symbolizer dfl;

	if ( sym.get_fill() != dfl.get_fill() || explicit_defaults_ )
	{
	    set_css( sym_node, "fill", sym.get_fill() );
	}
	if ( sym.get_opacity() != dfl.get_opacity() || explicit_defaults_ )
	{
	    set_css( sym_node, "fill-opacity", sym.get_opacity() );
	}
	if ( sym.height() != dfl.height() || explicit_defaults_ )
	{
	    set_css( sym_node, "height", sym.height() );
	}
    }

    void operator () ( markers_symbolizer const& )
    {
	// FIXME!!!!!
    }

    void operator () ( glyph_symbolizer const& sym)
    {
	ptree &node = rule_.push_back(
	    ptree::value_type("GlyphSymbolizer", ptree())
	    )->second;

	// face_name
	set_attr( node, "face_name", sym.get_face_name() );    

	// char
	if (sym.get_char()) {
	    const std::string &str =
		to_expression_string(*sym.get_char());
	    set_attr( node, "char", str );
	}

	// angle
	if (sym.get_angle()) {
	    const std::string &str =
		to_expression_string(*sym.get_angle());
	    set_attr( node, "angle", str );
	}

	// value
	if (sym.get_value()) {
	    const std::string &str =
		to_expression_string(*sym.get_value());
	    set_attr( node, "value", str );
	}

	// size
	if (sym.get_size()) {
	    const std::string &str =
		to_expression_string(*sym.get_size());
	    set_attr( node, "size", str );
	}

	// color
	if (sym.get_color()) {
	    const std::string &str =
		to_expression_string(*sym.get_color());
	    set_attr( node, "color", str );
	}

	// colorizer
	if (sym.get_colorizer()) {
	    serialize_raster_colorizer(node, sym.get_colorizer(),
				       explicit_defaults_);
	}

	// allow_overlap
	set_attr( node, "allow_overlap", sym.get_allow_overlap() );    

	// avoid_edges
	set_attr( node, "avoid_edges", sym.get_avoid_edges() );    

	// displacement
	position displacement = sym.get_displacement();
	set_attr( node, "dx", displacement.get<0>() );    
	set_attr( node, "dy", displacement.get<1>() );    

	// halo fill & radius
	set_attr( node, "halo_fill", sym.get_halo_fill() );    
	set_attr( node, "halo_radius", sym.get_halo_radius() );    

	// angle_mode
	set_attr( node, "angle_mode", sym.get_angle_mode() );

    }

private:
    serialize_symbolizer();
    void add_image_attributes(ptree & node, const symbolizer_with_image & sym)
    {
	const std::string & filename = path_processor_type::to_string( *sym.get_filename());
	if ( ! filename.empty() ) {
	    set_attr( node, "file", filename );
	}
    }
    void add_font_attributes(ptree & node, const text_symbolizer & sym)
    {
	expression_ptr const& expr = sym.get_name();
	const std::string & name = to_expression_string(*expr);
		
	if ( ! name.empty() ) {
	    set_attr( node, "name", name );
	}
	const std::string & face_name = sym.get_face_name();
	if ( ! face_name.empty() ) {
	    set_attr( node, "face_name", face_name );
	}
	const std::string & fontset_name = sym.get_fontset().get_name();
	if ( ! fontset_name.empty() ) {
	    set_attr( node, "fontset_name", fontset_name );
	}

	set_attr( node, "size", sym.get_text_size() );
	set_attr( node, "fill", sym.get_fill() );

	// pseudo-default-construct a text_symbolizer. It is used
	// to avoid printing ofattributes with default values without
	// repeating the default values here.
	// maybe add a real, explicit default-ctor?
	// FIXME
	text_symbolizer dfl(expression_ptr(), "<no default>",
			    0, color(0,0,0) );

	position displacement = sym.get_displacement();
	if ( displacement.get<0>() != dfl.get_displacement().get<0>() || explicit_defaults_ )
	{
	    set_attr( node, "dx", displacement.get<0>() );
	}
	if ( displacement.get<1>() != dfl.get_displacement().get<1>() || explicit_defaults_ )
	{
	    set_attr( node, "dy", displacement.get<1>() );
	}

	if (sym.get_label_placement() != dfl.get_label_placement() || explicit_defaults_ )
	{
	    set_attr( node, "placement", sym.get_label_placement() );
	}

	if (sym.get_vertical_alignment() != dfl.get_vertical_alignment() || explicit_defaults_ )
	{
	    set_attr( node, "vertical_alignment", sym.get_vertical_alignment() );
	}

	if (sym.get_halo_radius() != dfl.get_halo_radius() || explicit_defaults_ )
	{
	    set_attr( node, "halo_radius", sym.get_halo_radius() );
	}
	const color & c = sym.get_halo_fill();
	if ( c != dfl.get_halo_fill() || explicit_defaults_ )
	{
	    set_attr( node, "halo_fill", c );
	}
	if (sym.get_text_ratio() != dfl.get_text_ratio() || explicit_defaults_ )
	{
	    set_attr( node, "text_ratio", sym.get_text_ratio() );
	}
	if (sym.get_wrap_width() != dfl.get_wrap_width() || explicit_defaults_ )
	{
	    set_attr( node, "wrap_width", sym.get_wrap_width() );
	}
	if (sym.get_wrap_before() != dfl.get_wrap_before() || explicit_defaults_ )
	{
	    set_attr( node, "wrap_before", sym.get_wrap_before() );
	}
	if (sym.get_wrap_char() != dfl.get_wrap_char() || explicit_defaults_ )
	{
	    set_attr( node, "wrap_character", std::string(1, sym.get_wrap_char()) );
	}
	if (sym.get_text_convert() != dfl.get_text_convert() || explicit_defaults_ )
	{
	    set_attr( node, "text_convert", sym.get_text_convert() );
	}
	if (sym.get_line_spacing() != dfl.get_line_spacing() || explicit_defaults_ )
	{
	    set_attr( node, "line_spacing", sym.get_line_spacing() );
	}
	if (sym.get_character_spacing() != dfl.get_character_spacing() || explicit_defaults_ )
	{
	    set_attr( node, "character_spacing", sym.get_character_spacing() );
	}
	if (sym.get_label_spacing() != dfl.get_label_spacing() || explicit_defaults_ )
	{
	    set_attr( node, "spacing", sym.get_label_spacing() );
	}
	if (sym.get_minimum_distance() != dfl.get_minimum_distance() || explicit_defaults_ )
	{
	    set_attr( node, "min_distance", sym.get_minimum_distance() );
	}
	if (sym.get_allow_overlap() != dfl.get_allow_overlap() || explicit_defaults_ )
	{
	    set_attr( node, "allow_overlap", sym.get_allow_overlap() );
	}
	if (sym.get_avoid_edges() != dfl.get_avoid_edges() || explicit_defaults_ )
	{
	    set_attr( node, "avoid_edges", sym.get_avoid_edges() );
	}
	if (sym.get_opacity() != dfl.get_opacity() || explicit_defaults_ )
	{
	    set_attr( node, "opacity", sym.get_opacity() );
	}
	if (sym.get_horizontal_alignment() != dfl.get_horizontal_alignment() || explicit_defaults_ )
	{
	    set_attr( node, "horizontal_alignment", sym.get_horizontal_alignment() );
	}
	if (sym.get_justify_alignment() != dfl.get_justify_alignment() || explicit_defaults_ )
	{
	    set_attr( node, "justify_alignment", sym.get_justify_alignment() );
	}
    }
    ptree & rule_;
    bool explicit_defaults_;
};

void serialize_rule( ptree & style_node, const rule_type & rule, bool explicit_defaults)
{
    ptree & rule_node = style_node.push_back(
	ptree::value_type("Rule", ptree() ))->second;

    rule_type dfl;
    if ( rule.get_name() != dfl.get_name() )
    {
	set_attr(rule_node, "name", rule.get_name());
    }
    if ( rule.get_title() != dfl.get_title() )
    {
	set_attr(rule_node, "title", rule.get_title());
    }

    if ( rule.has_else_filter() )
    {
	rule_node.push_back( ptree::value_type(
				 "ElseFilter", ptree()));
    }
    else
    {
	// filters were not comparable, perhaps should now compare expressions?
	expression_ptr const& expr = rule.get_filter();
	std::string filter = mapnik::to_expression_string(*expr);
	std::string default_filter = mapnik::to_expression_string(*dfl.get_filter());
            
	if ( filter != default_filter)
	{
	    rule_node.push_back( ptree::value_type(
				     "Filter", ptree()))->second.put_value( filter );
	}
    }

    if (rule.get_min_scale() != dfl.get_min_scale() )
    {
	ptree & min_scale = rule_node.push_back( ptree::value_type(
						     "MinScaleDenominator", ptree()))->second;
	min_scale.put_value( rule.get_min_scale() );
    }

    if (rule.get_max_scale() != dfl.get_max_scale() )
    {
	ptree & max_scale = rule_node.push_back( ptree::value_type(
						     "MaxScaleDenominator", ptree()))->second;
	max_scale.put_value( rule.get_max_scale() );
    }

    rule_type::symbolizers::const_iterator begin = rule.get_symbolizers().begin();
    rule_type::symbolizers::const_iterator end = rule.get_symbolizers().end();
    serialize_symbolizer serializer( rule_node, explicit_defaults);
    std::for_each( begin, end , boost::apply_visitor( serializer ));
}

void serialize_style( ptree & map_node, Map::const_style_iterator style_it, bool explicit_defaults )
{
    const feature_type_style & style = style_it->second;
    const std::string & name = style_it->first;

    ptree & style_node = map_node.push_back(
	ptree::value_type("Style", ptree()))->second;

    set_attr(style_node, "name", name);

    rules::const_iterator it = style.get_rules().begin();
    rules::const_iterator end = style.get_rules().end();
    for (; it != end; ++it)
    {
	serialize_rule( style_node, * it , explicit_defaults);
    }

}

void serialize_fontset( ptree & map_node, Map::const_fontset_iterator fontset_it )
{
    const font_set & fontset = fontset_it->second;
    const std::string & name = fontset_it->first;

    ptree & fontset_node = map_node.push_back(
	ptree::value_type("FontSet", ptree()))->second;

    set_attr(fontset_node, "name", name);

    std::vector<std::string>::const_iterator it = fontset.get_face_names().begin();
    std::vector<std::string>::const_iterator end = fontset.get_face_names().end();
    for (; it != end; ++it)
    {
	ptree & font_node = fontset_node.push_back(
	    ptree::value_type("Font", ptree()))->second;
	set_attr(font_node, "face_name", *it);
    }

}

void serialize_datasource( ptree & layer_node, datasource_ptr datasource)
{
    ptree & datasource_node = layer_node.push_back(
	ptree::value_type("Datasource", ptree()))->second;

    parameters::const_iterator it = datasource->params().begin();
    parameters::const_iterator end = datasource->params().end();
    for (; it != end; ++it)
    {
	boost::property_tree::ptree & param_node = datasource_node.push_back(
	    boost::property_tree::ptree::value_type("Parameter",
						    boost::property_tree::ptree()))->second;
	param_node.put("<xmlattr>.name", it->first );
	param_node.put_value( it->second );

    }
}

void serialize_layer( ptree & map_node, const layer & layer, bool explicit_defaults )
{
    ptree & layer_node = map_node.push_back(
	ptree::value_type("Layer", ptree()))->second;
    if ( layer.name() != "" )
    {
	set_attr( layer_node, "name", layer.name() );
    }

    if ( layer.abstract() != "" )
    {
	set_attr( layer_node, "abstract", layer.abstract() );
    }

    if ( layer.title() != "" )
    {
	set_attr( layer_node, "title", layer.title() );
    }

    if ( layer.srs() != "" )
    {
	set_attr( layer_node, "srs", layer.srs() );
    }

    if ( !layer.isActive() || explicit_defaults )
    {
	set_attr/*<bool>*/( layer_node, "status", layer.isActive() );
    }
        
    if ( layer.clear_label_cache() || explicit_defaults )
    {        
	set_attr/*<bool>*/( layer_node, "clear_label_cache", layer.clear_label_cache() );
    }

    if ( layer.getMinZoom() )
    {
	set_attr( layer_node, "minzoom", layer.getMinZoom() );
    }

    if ( layer.getMaxZoom() != std::numeric_limits<double>::max() )
    {
	set_attr( layer_node, "maxzoom", layer.getMaxZoom() );
    }

    if ( layer.isQueryable() || explicit_defaults )
    {
	set_attr( layer_node, "queryable", layer.isQueryable() );
    }

    std::vector<std::string> const& style_names = layer.styles();
    for (unsigned i = 0; i < style_names.size(); ++i)
    {
	boost::property_tree::ptree & style_node = layer_node.push_back(
	    boost::property_tree::ptree::value_type("StyleName",
						    boost::property_tree::ptree()))->second;
	style_node.put_value( style_names[i] );
    }

    datasource_ptr datasource = layer.datasource();
    if ( datasource )
    {
	serialize_datasource( layer_node, datasource );
    }
}

void serialize_map(ptree & pt, Map const & map, bool explicit_defaults)
{

    ptree & map_node = pt.push_back(ptree::value_type("Map", ptree() ))->second;

    set_attr( map_node, "srs", map.srs() );

    optional<color> c = map.background();
    if ( c )
    {
	set_attr( map_node, "bgcolor", * c );
    }

    unsigned buffer_size = map.buffer_size();
    if ( buffer_size || explicit_defaults)
    {
	set_attr( map_node, "buffer_size", buffer_size ); 
    }

    {
	Map::const_fontset_iterator it = map.fontsets().begin();
	Map::const_fontset_iterator end = map.fontsets().end();
	for (; it != end; ++it)
	{
	    serialize_fontset( map_node, it);
	}
    }

    Map::const_style_iterator it = map.styles().begin();
    Map::const_style_iterator end = map.styles().end();
    for (; it != end; ++it)
    {
	serialize_style( map_node, it, explicit_defaults);
    }

    std::vector<layer> const & layers = map.layers();
    for (unsigned i = 0; i < layers.size(); ++i )
    {
	serialize_layer( map_node, layers[i], explicit_defaults );
    }
}

void save_map(Map const & map, std::string const& filename, bool explicit_defaults)
{
    ptree pt;
    serialize_map(pt,map,explicit_defaults);
    write_xml(filename,pt,std::locale(),boost::property_tree::xml_writer_make_settings(' ',4));
}

std::string save_map_to_string(Map const & map, bool explicit_defaults)
{
    ptree pt;
    serialize_map(pt,map,explicit_defaults);
    std::ostringstream ss;
    write_xml(ss,pt,boost::property_tree::xml_writer_make_settings(' ',4));
    return ss.str();
}

}