/* This file is part of Mapnik (c++ mapping toolkit)
 * Copyright (C) 2006 Artem Pavlenko
 *
 * Mapnik 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 2
 * of the License, or any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

//$Id$

#include "agg_renderer.hpp"

#include "agg_basics.h"
#include "agg_rendering_buffer.h"
#include "agg_rasterizer_scanline_aa.h"
#include "agg_scanline_p.h"
#include "agg_scanline_u.h"
#include "agg_renderer_scanline.h"
#include "agg_pixfmt_rgba.h"
#include "agg_path_storage.h"
#include "agg_span_allocator.h"
#include "agg_span_pattern_rgba.h"
#include "agg_image_accessors.h"
#include "agg_conv_stroke.h"
#include "agg_conv_dash.h"
#include "agg_conv_contour.h"
#include "agg_vcgen_stroke.h"
#include "agg_conv_adaptor_vcgen.h"
#include "agg_conv_smooth_poly1.h"
#include "agg_conv_marker.h"
#include "agg_arrowhead.h"
#include "agg_vcgen_markers_term.h"
#include "agg_renderer_outline_aa.h"
#include "agg_rasterizer_outline_aa.h"
#include "agg_rasterizer_outline.h"
#include "agg_renderer_outline_image.h"
#include "agg_span_allocator.h"
#include "agg_span_pattern_rgba.h"
#include "agg_renderer_scanline.h"
#include "agg_pattern_filters_rgba.h"
#include "agg_renderer_outline_image.h"

#include <boost/utility.hpp>

#include <iostream>

namespace mapnik
{
    class pattern_source : private boost::noncopyable
    {
    public:
	pattern_source(ImageData32 const& pattern)
	    : pattern_(pattern) {}
	
	unsigned int width() const
	{
	    return pattern_.width();
	}
	unsigned int height() const
	{
	    return pattern_.height();
	}
	agg::rgba8 pixel(int x, int y) const
	{
	    unsigned c = pattern_(x,y);
	    return agg::rgba8(c & 0xff, (c >> 8) & 0xff, (c >> 16) & 0xff,(c >> 24) & 0xff);
	}
    private:
	ImageData32 const& pattern_;
    };

    template <typename T>
    agg_renderer<T>::agg_renderer(Map const& m, T & pixmap)
	: feature_style_processor<agg_renderer>(m),
	  pixmap_(pixmap),
	  t_(m.getWidth(),m.getHeight(),m.getCurrentExtent())
    {
	Color const& bg=m.getBackground();
	pixmap_.setBackground(bg);
	std::cout << "scale="<<m.scale()<<std::endl;
    }
    template <typename T>	
    void agg_renderer<T>::process(polygon_symbolizer const& sym,Feature const& feature)
    {
	typedef  coord_transform<CoordTransform,geometry_type> path_type;
	typedef agg::renderer_base<agg::pixfmt_rgba32> ren_base;    
	typedef agg::renderer_scanline_aa_solid<ren_base> renderer;
	    
	Color const& fill_  = sym.get_fill();
	    
	geometry_ptr const& geom=feature.get_geometry();
	if (geom && geom->num_points() > 2) 
	{
	    unsigned width = pixmap_.width();
	    unsigned height = pixmap_.height();
	    path_type path(t_,*geom);
	    agg::row_ptr_cache<agg::int8u> buf(pixmap_.raw_data(),width,height,width * 4);
	    agg::pixfmt_rgba32 pixf(buf);
	    ren_base renb(pixf);	    
		
	    unsigned r=fill_.red();
	    unsigned g=fill_.green();
	    unsigned b=fill_.blue();
	    unsigned a=fill_.alpha();
	    renderer ren(renb);
		
	    agg::rasterizer_scanline_aa<> ras;
	    agg::scanline_u8 sl;
	    ras.clip_box(0,0,width,height);
	    ras.add_path(path);
	    ren.color(agg::rgba8(r, g, b, a));
	    agg::render_scanlines(ras, sl, ren);
	}
    }

    template <typename T>
    void agg_renderer<T>::process(line_symbolizer const& sym,Feature const& feature)
    {   
	typedef agg::renderer_base<agg::pixfmt_rgba32> ren_base; 
	typedef coord_transform<CoordTransform,geometry_type> path_type;
	typedef agg::renderer_outline_aa<ren_base> renderer_oaa;
	typedef agg::rasterizer_outline_aa<renderer_oaa> rasterizer_outline_aa;
	typedef agg::renderer_scanline_aa_solid<ren_base> renderer;
	    
	geometry_ptr const& geom=feature.get_geometry();
	if (geom && geom->num_points() > 1)
	{
	    path_type path(t_,*geom);
	    agg::row_ptr_cache<agg::int8u> buf(pixmap_.raw_data(),pixmap_.width(),pixmap_.height(),
					       pixmap_.width()*4);
	    agg::pixfmt_rgba32 pixf(buf);
	    ren_base renb(pixf);	    
		
	    mapnik::stroke const&  stroke_ = sym.get_stroke();
		
	    Color const& col = stroke_.get_color();
	    unsigned r=col.red();
	    unsigned g=col.green();
	    unsigned b=col.blue();	    
		
	    if (stroke_.has_dash())
	    {
		renderer ren(renb);	
		agg::rasterizer_scanline_aa<> ras;
		agg::scanline_u8 sl;
		agg::conv_dash<path_type> dash(path);
		dash_array const& d = stroke_.get_dash_array();
		dash_array::const_iterator itr = d.begin();
		dash_array::const_iterator end = d.end();
		while (itr != end)
		{
		    dash.add_dash(itr->first, itr->second);
		    ++itr;
		}
		agg::conv_stroke<agg::conv_dash<path_type > > stroke(dash);
		    
		line_join_e join=stroke_.get_line_join();
		if ( join == MITER_JOIN)
		    stroke.generator().line_join(agg::miter_join);
		else if( join == MITER_REVERT_JOIN) 
		    stroke.generator().line_join(agg::miter_join);
		else if( join == ROUND_JOIN) 
		    stroke.generator().line_join(agg::round_join);
		else
		    stroke.generator().line_join(agg::bevel_join);
		    
		line_cap_e cap=stroke_.get_line_cap();
		if (cap == BUTT_CAP)    
		    stroke.generator().line_cap(agg::butt_cap);
		else if (cap == SQUARE_CAP)
		    stroke.generator().line_cap(agg::square_cap);
		else 
		    stroke.generator().line_cap(agg::round_cap);
		    
		stroke.generator().miter_limit(4.0);
		stroke.generator().width(stroke_.get_width());
		    
		ras.clip_box(0,0,pixmap_.width(),pixmap_.height());
		ras.add_path(stroke);
		ren.color(agg::rgba8(r, g, b, int(255*stroke_.get_opacity())));
		agg::render_scanlines(ras, sl, ren);
	    }
	    else if (stroke_.get_width() <= 1.0)
	    {
		agg::line_profile_aa prof;
		prof.width(stroke_.get_width());
		renderer_oaa ren_oaa(renb, prof);
		rasterizer_outline_aa ras_oaa(ren_oaa);
		    
		ren_oaa.color(agg::rgba8(r, g, b, int(255*stroke_.get_opacity())));
		ren_oaa.clip_box(0,0,pixmap_.width(),pixmap_.height());
		ras_oaa.add_path(path);		
		    
	    }
	    else 
	    {
		renderer ren(renb);	
		agg::rasterizer_scanline_aa<> ras;
		agg::scanline_p8 sl;
		agg::conv_stroke<path_type>  stroke(path);
		    
		line_join_e join=stroke_.get_line_join();
		if ( join == MITER_JOIN)
		    stroke.generator().line_join(agg::miter_join);
		else if( join == MITER_REVERT_JOIN) 
		    stroke.generator().line_join(agg::miter_join);
		else if( join == ROUND_JOIN) 
		    stroke.generator().line_join(agg::round_join);
		else
		    stroke.generator().line_join(agg::bevel_join);
		    
		line_cap_e cap=stroke_.get_line_cap();
		if (cap == BUTT_CAP)    
		    stroke.generator().line_cap(agg::butt_cap);
		else if (cap == SQUARE_CAP)
		    stroke.generator().line_cap(agg::square_cap);
		else 
		    stroke.generator().line_cap(agg::round_cap);
		
		stroke.generator().miter_limit(4.0);
		stroke.generator().width(stroke_.get_width());
		    
		ras.clip_box(0,0,pixmap_.width(),pixmap_.height());
		ras.add_path(stroke);
		ren.color(agg::rgba8(r, g, b, int(255*stroke_.get_opacity())));
		agg::render_scanlines(ras, sl, ren);
	    }
	}
    }

    template <typename T>
    void agg_renderer<T>::process(point_symbolizer const& sym,Feature const& feature)
    {
	geometry_ptr const& geom=feature.get_geometry();
	if (geom)
	{
	    double x;
	    double y;
	    ImageData32 const& data = sym.get_data();
	    geom->label_position(&x,&y);
	    t_.forward_x(&x);
	    t_.forward_y(&y);
	    int w=data.width();
	    int h=data.height();    
	    int px=int(ceil(x - 0.5 * w));
	    int py=int(ceil(y - 0.5 * h));
	    pixmap_.set_rectangle_alpha(px,py,data);
	}
    }
    
    template <typename T>
    void  agg_renderer<T>::process(line_pattern_symbolizer const& sym,Feature const& feature)
    {
	typedef  coord_transform<CoordTransform,geometry_type> path_type;
	typedef agg::line_image_pattern<agg::pattern_filter_bilinear_rgba8> pattern_type;
	typedef agg::renderer_base<agg::pixfmt_rgba32> renderer_base;
	typedef agg::renderer_outline_image<renderer_base, pattern_type> renderer_type;
	typedef agg::rasterizer_outline_aa<renderer_type> rasterizer_type;

	geometry_ptr const& geom=feature.get_geometry();
	if (geom)
	{
	    unsigned width = pixmap_.width();
	    unsigned height = pixmap_.height();
	    ImageData32 const& pat = sym.get_pattern();
	    path_type path(t_,*geom);
	    agg::row_ptr_cache<agg::int8u> buf(pixmap_.raw_data(), width, height,width*4);
	    agg::pixfmt_rgba32 pixf(buf);
	    renderer_base ren_base(pixf);  
	    agg::pattern_filter_bilinear_rgba8 filter; 
	    pattern_source source(pat);
	    pattern_type pattern (filter,source);
	    renderer_type ren(ren_base, pattern);
	    ren.clip_box(0,0,width,height);
	    rasterizer_type ras(ren);	    
	    ras.add_path(path);    
	}
    }
    
    template <typename T>
    void agg_renderer<T>::process(polygon_pattern_symbolizer const& sym,Feature const& feature)
    {
	typedef  coord_transform<CoordTransform,geometry_type> path_type;
	typedef agg::renderer_base<agg::pixfmt_rgba32> ren_base; 
	typedef agg::wrap_mode_repeat wrap_x_type;
	typedef agg::wrap_mode_repeat wrap_y_type;
	typedef agg::image_accessor_wrap<agg::pixfmt_rgba32, 
	    wrap_x_type,
	    wrap_y_type> img_source_type;
	
	typedef agg::span_pattern_rgba<img_source_type> span_gen_type;
	
	typedef agg::renderer_scanline_aa<ren_base, 
	    agg::span_allocator<agg::rgba8>,
	    span_gen_type> renderer_type;  
	geometry_ptr const& geom=feature.get_geometry();
	if (geom)
	{
	    ImageData32 const& pattern = sym.get_pattern();
	    
	    unsigned width = pixmap_.width();
	    unsigned height = pixmap_.height();
	    path_type path(t_,*geom);
	    
	    agg::row_ptr_cache<agg::int8u> buf(pixmap_.raw_data(),width,height,width * 4);
	    agg::pixfmt_rgba32 pixf(buf);
	    ren_base renb(pixf);
	
	    unsigned w=pattern.width();
	    unsigned h=pattern.height();
	    agg::row_ptr_cache<agg::int8u> pattern_rbuf((agg::int8u*)pattern.getBytes(),w,h,w*4);  
	    
	    double x0,y0;
	    path.vertex(&x0,&y0);
	    path.rewind(0);
	
	    unsigned offset_x = unsigned(width - x0);
	    unsigned offset_y = unsigned(height - y0);
	
	    agg::span_allocator<agg::rgba8> sa;
	    img_source_type img_src(pattern_rbuf);
	    span_gen_type sg(img_src, offset_x, offset_y);
	    renderer_type rp(renb,sa, sg);
	
	    agg::rasterizer_scanline_aa<> ras;
	    agg::scanline_u8 sl;
	    ras.clip_box(0,0,width,height);
	    ras.add_path(path);
	    agg::render_scanlines(ras, sl, rp);   
	}
    }

    template <typename T>
    void agg_renderer<T>::process(raster_symbolizer const& ,Feature const& feature)
    {
	// TODO -- at the moment raster_symbolizer is an empty class 
	// used for type dispatching, but we can have some fancy raster
	// processing in a future (filters??). Just copy raster into pixmap for now.
	raster_ptr const& raster=feature.get_raster();
	if (raster)
	{
	    pixmap_.set_rectangle(raster->x_,raster->y_,raster->data_);
	}
    }
    
    template <typename T>
    void agg_renderer<T>::process(text_symbolizer const& sym ,Feature const& feature)
    {
	//TODO - implement text
	//std::cout << feature << std::endl;
	std::cout << sym.get_name() <<":" << feature[sym.get_name()] << std::endl;
    }

    template class agg_renderer<Image32>;
}