/* This file is part of Mapnik (c++ mapping toolkit)
 * Copyright (C) 2005 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$

#ifndef FILTER_PARSER_HPP
#define FILTER_PARSER_HPP

#include <boost/spirit/core.hpp>
#include <boost/spirit/symbols.hpp>
#include <boost/spirit/utility/confix.hpp>
#include <boost/spirit/utility/escape_char.hpp>
#include <boost/spirit/utility/chset.hpp> 

#include "value.hpp"
#include "comparison.hpp"
#include "expression.hpp"
#include "filter.hpp"
#include "regex_filter.hpp"

#include <stack>
#include <iostream>

using namespace boost::spirit;
using std::string;
using std::cerr;
using std::stack;

namespace mapnik
{    
    template <typename FeatureT>
    struct push_integer
    {
	push_integer(stack<ref_ptr<expression<FeatureT> > >& exprs)
	    : exprs_(exprs) {}
	
	void operator() (int val) const
	{
	    exprs_.push(ref_ptr<expression<FeatureT> >(new literal<FeatureT>(val)));
	}
	stack<ref_ptr<expression<FeatureT> > >& exprs_;
    };
   
    template <typename FeatureT>
    struct push_real
    {
	push_real(stack<ref_ptr<expression<FeatureT> > >& exprs)
	    : exprs_(exprs) {}
	void operator() (double val) const
	{
	    exprs_.push(ref_ptr<expression<FeatureT> >(new literal<FeatureT>(val)));
	}
	stack<ref_ptr<expression<FeatureT> > >& exprs_;
    };
    
    template <typename FeatureT>
    struct push_string
    {
	push_string(stack<ref_ptr<expression<FeatureT> > >& exprs)
	    : exprs_(exprs) {}
	
	template <typename Iter>
	void operator() (Iter start,Iter end) const
	{
	    string str(start,end);
	    char quote='\\';
	    string::size_type idx;
	    idx = str.find(quote);
	    while (idx != string::npos)
	    {	    		
		str.erase(idx,1);
		idx = str.find(quote);
	    }
	    exprs_.push(ref_ptr<expression<FeatureT> >(new literal<FeatureT>(str)));
	}
	stack<ref_ptr<expression<FeatureT> > >& exprs_;
    };
    
    template <typename FeatureT>
    struct push_property
    {
	push_property(stack<ref_ptr<expression<FeatureT> > >& exprs)
	    : exprs_(exprs) {}
	
	template <typename Iter>
	void operator() (Iter start,Iter end) const
	{
	    string str(start,end);
	    exprs_.push(ref_ptr<expression<FeatureT> >(new property<FeatureT>(str)));
	}
	stack<ref_ptr<expression<FeatureT> > >& exprs_;
    };

    template <typename FeatureT,typename Op>
    struct compose_expression
    {
	compose_expression(stack<ref_ptr<expression<FeatureT> > >& exprs)
	    : exprs_(exprs) {}

	template <typename Iter>
	void operator() (Iter,Iter) const
	{
	    if (exprs_.size()>=2)
	    {
		ref_ptr<expression<FeatureT> > right = exprs_.top();
		exprs_.pop();
		ref_ptr<expression<FeatureT> > left = exprs_.top();
		exprs_.pop();
		if (left && right)
		{
		    exprs_.push(ref_ptr<expression<FeatureT> >(new math_expr_b<FeatureT,Op>(*left,*right)));
		}
	    }
	}
	stack<ref_ptr<expression<FeatureT> > >& exprs_;
    };
    
    template <typename FeatureT>
    struct compose_regex
    {
	compose_regex(stack<ref_ptr<filter<FeatureT> > >& filters,
		       stack<ref_ptr<expression<FeatureT> > >& exprs)
	    : filters_(filters),exprs_(exprs) {}

	template <typename Iter>
	void operator() (Iter start,Iter end) const
	{
	    if (exprs_.size()>=1)
	    {
		ref_ptr<expression<FeatureT> > exp = exprs_.top();
		exprs_.pop();
		if (exp)
		{		    
		    std::string pattern(start,end);
		    try 
		    {
			filters_.push(ref_ptr<filter<FeatureT> >(new regex_filter<FeatureT>(*exp,pattern)));
		    }
		    catch (...)//boost::regex_error& ex)
		    {
		      cerr<<"error\n";//ex.what()<<"\n";
		    }
		    
		}
	    }
	}
	stack<ref_ptr<filter<FeatureT> > >& filters_;
	stack<ref_ptr<expression<FeatureT> > >& exprs_;
    };

  
    template <typename FeatureT,typename Op>
    struct compose_filter
    {
	compose_filter(stack<ref_ptr<filter<FeatureT> > >& filters,
		       stack<ref_ptr<expression<FeatureT> > >& exprs)
	    : filters_(filters),exprs_(exprs) {}

	template <typename Iter>
	void operator() (Iter,Iter) const
	{
	    if (exprs_.size()>=2)
	    {
		ref_ptr<expression<FeatureT> > right = exprs_.top();
		exprs_.pop();
		ref_ptr<expression<FeatureT> > left = exprs_.top();
		exprs_.pop();
		if (left && right)
		{
		    filters_.push(ref_ptr<filter<FeatureT> >(new compare_filter<FeatureT,Op>(*left,*right)));
		}
	    }
	}
	stack<ref_ptr<filter<FeatureT> > >& filters_;
	stack<ref_ptr<expression<FeatureT> > >& exprs_;
    };
    
    template <typename FeatureT>
    struct compose_and_filter
    {
	compose_and_filter(stack<ref_ptr<filter<FeatureT> > >& filters)
	    : filters_(filters) {}

	template <typename Iter>
	void operator() (Iter,Iter) const
	{
	    if (filters_.size()>=2)
	    {
		ref_ptr<filter<FeatureT> > right = filters_.top();
		filters_.pop();
		ref_ptr<filter<FeatureT> > left = filters_.top();
		filters_.pop();
		if (left && right)
		{
		    filters_.push(ref_ptr<filter<FeatureT> >(new logical_and<FeatureT>(*left,*right)));
		}
	    }
	}
	stack<ref_ptr<filter<FeatureT> > >& filters_;
    };
    
    template <typename FeatureT>
    struct compose_or_filter
    {
	compose_or_filter(stack<ref_ptr<filter<FeatureT> > >& filters)
	    : filters_(filters) {}

	template <typename Iter>
	void operator() (Iter,Iter) const
	{
	    if (filters_.size()>=2)
	    {
		ref_ptr<filter<FeatureT> > right = filters_.top();
		filters_.pop();
		ref_ptr<filter<FeatureT> > left = filters_.top();
		filters_.pop();
		if (left && right)
		{
		    filters_.push(ref_ptr<filter<FeatureT> >(new logical_or<FeatureT>(*left,*right)));
		}
	    }
	}
	stack<ref_ptr<filter<FeatureT> > >& filters_;
    };
    
    template <typename FeatureT>
    struct compose_not_filter
    {
	compose_not_filter(stack<ref_ptr<filter<FeatureT> > >& filters)
	    : filters_(filters) {}

	template <typename Iter>
	void operator() (Iter,Iter) const
	{
	    if (filters_.size()>=1)
	    {
		ref_ptr<filter<FeatureT> > filter_ = filters_.top();
		filters_.pop();
		if (filter_)
		{
		    filters_.push(ref_ptr<filter<FeatureT> >(new logical_not<FeatureT>(*filter_)));
		}
	    }
	}
	stack<ref_ptr<filter<FeatureT> > >& filters_;
    };
    
    template <typename FeatureT>
    struct filter_grammar : public grammar<filter_grammar<FeatureT> >
    {
	filter_grammar(stack<ref_ptr<filter<FeatureT> > >& filters_,
		       stack<ref_ptr<expression<FeatureT> > >& exprs_)
	    : filters(filters_),exprs(exprs_) {}
	
	template <typename ScannerT>
	struct definition
	{
	    definition(filter_grammar const& self)
	    {	
		typedef boost::spirit::chset<wchar_t> chset_t;

		func1_op = "sqrt","sin","cos";
		func2_op = "min","max";
		spatial_op = "Equals","Disjoint","Touches","Within","Overlaps",
		    "Crosses","Intersects","Contains","DWithin","Beyond","BBOX";
		
		chset_t BaseChar (L"\x41-\x5A\x61-\x7A\xC0-\xD6\xD8-\xF6\xF8-\xFF\x100-\x131\x134-\x13E"
				  L"\x141-\x148\x14A-\x17E\x180-\x1C3\x1CD-\x1F0\x1F4-\x1F5\x1FA-\x217"
				  L"\x250-\x2A8\x2BB-\x2C1\x386\x388-\x38A\x38C\x38E-\x3A1\x3A3-\x3CE"
				  L"\x3D0-\x3D6\x3DA\x3DC\x3DE\x3E0\x3E2-\x3F3\x401-\x40C\x40E-\x44F"
				  L"\x451-\x45C\x45E-\x481\x490-\x4C4\x4C7-\x4C8\x4CB-\x4CC\x4D0-\x4EB"
				  L"\x4EE-\x4F5\x4F8-\x4F9\x531-\x556\x559\x561-\x586\x5D0-\x5EA"
				  L"\x5F0-\x5F2\x621-\x63A\x641-\x64A\x671-\x6B7\x6BA-\x6BE\x6C0-\x6CE"
				  L"\x6D0-\x6D3\x6D5\x6E5-\x6E6\x905-\x939\x93D\x958-\x961\x985-\x98C"
				  L"\x98F-\x990\x993-\x9A8\x9AA-\x9B0\x9B2\x9B6-\x9B9\x9DC-\x9DD"
				  L"\x9DF-\x9E1\x9F0-\x9F1\xA05-\xA0A\xA0F-\xA10\xA13-\xA28\xA2A-\xA30"
				  L"\xA32-\xA33\xA35-\xA36\xA38-\xA39\xA59-\xA5C\xA5E\xA72-\xA74"
				  L"\xA85-\xA8B\xA8D\xA8F-\xA91\xA93-\xAA8\xAAA-\xAB0\xAB2-\xAB3"
				  L"\xAB5-\xAB9\xABD\xAE0\xB05-\xB0C\xB0F-\xB10\xB13-\xB28\xB2A-\xB30"
				  L"\xB32-\xB33\xB36-\xB39\xB3D\xB5C-\xB5D\xB5F-\xB61\xB85-\xB8A"
				  L"\xB8E-\xB90\xB92-\xB95\xB99-\xB9A\xB9C\xB9E-\xB9F\xBA3-\xBA4"
				  L"\xBA8-\xBAA\xBAE-\xBB5\xBB7-\xBB9\xC05-\xC0C\xC0E-\xC10\xC12-\xC28"
				  L"\xC2A-\xC33\xC35-\xC39\xC60-\xC61\xC85-\xC8C\xC8E-\xC90\xC92-\xCA8"
				  L"\xCAA-\xCB3\xCB5-\xCB9\xCDE\xCE0-\xCE1\xD05-\xD0C\xD0E-\xD10"
				  L"\xD12-\xD28\xD2A-\xD39\xD60-\xD61\xE01-\xE2E\xE30\xE32-\xE33"
				  L"\xE40-\xE45\xE81-\xE82\xE84\xE87-\xE88\xE8A\xE8D\xE94-\xE97"
				  L"\xE99-\xE9F\xEA1-\xEA3\xEA5\xEA7\xEAA-\xEAB\xEAD-\xEAE\xEB0"
				  L"\xEB2-\xEB3\xEBD\xEC0-\xEC4\xF40-\xF47\xF49-\xF69\x10A0-\x10C5"
				  L"\x10D0-\x10F6\x1100\x1102-\x1103\x1105-\x1107\x1109\x110B-\x110C"
				  L"\x110E-\x1112\x113C\x113E\x1140\x114C\x114E\x1150\x1154-\x1155"
				  L"\x1159\x115F-\x1161\x1163\x1165\x1167\x1169\x116D-\x116E"
				  L"\x1172-\x1173\x1175\x119E\x11A8\x11AB\x11AE-\x11AF\x11B7-\x11B8"
				  L"\x11BA\x11BC-\x11C2\x11EB\x11F0\x11F9\x1E00-\x1E9B\x1EA0-\x1EF9"
				  L"\x1F00-\x1F15\x1F18-\x1F1D\x1F20-\x1F45\x1F48-\x1F4D\x1F50-\x1F57"
				  L"\x1F59\x1F5B\x1F5D\x1F5F-\x1F7D\x1F80-\x1FB4\x1FB6-\x1FBC\x1FBE"
				  L"\x1FC2-\x1FC4\x1FC6-\x1FCC\x1FD0-\x1FD3\x1FD6-\x1FDB\x1FE0-\x1FEC"
				  L"\x1FF2-\x1FF4\x1FF6-\x1FFC\x2126\x212A-\x212B\x212E\x2180-\x2182"
				  L"\x3041-\x3094\x30A1-\x30FA\x3105-\x312C\xAC00-\xD7A3");
            
		chset_t Ideographic(L"\x4E00-\x9FA5\x3007\x3021-\x3029");
		chset_t Letter = BaseChar | Ideographic;
	    
		chset_t CombiningChar(L"\x0300-\x0345\x0360-\x0361\x0483-\x0486\x0591-\x05A1\x05A3-\x05B9"
				      L"\x05BB-\x05BD\x05BF\x05C1-\x05C2\x05C4\x064B-\x0652\x0670"
				      L"\x06D6-\x06DC\x06DD-\x06DF\x06E0-\x06E4\x06E7-\x06E8\x06EA-\x06ED"
				      L"\x0901-\x0903\x093C\x093E-\x094C\x094D\x0951-\x0954\x0962-\x0963"
				      L"\x0981-\x0983\x09BC\x09BE\x09BF\x09C0-\x09C4\x09C7-\x09C8"
				      L"\x09CB-\x09CD\x09D7\x09E2-\x09E3\x0A02\x0A3C\x0A3E\x0A3F"
				      L"\x0A40-\x0A42\x0A47-\x0A48\x0A4B-\x0A4D\x0A70-\x0A71\x0A81-\x0A83"
				      L"\x0ABC\x0ABE-\x0AC5\x0AC7-\x0AC9\x0ACB-\x0ACD\x0B01-\x0B03\x0B3C"
				      L"\x0B3E-\x0B43\x0B47-\x0B48\x0B4B-\x0B4D\x0B56-\x0B57\x0B82-\x0B83"
				      L"\x0BBE-\x0BC2\x0BC6-\x0BC8\x0BCA-\x0BCD\x0BD7\x0C01-\x0C03"
				      L"\x0C3E-\x0C44\x0C46-\x0C48\x0C4A-\x0C4D\x0C55-\x0C56\x0C82-\x0C83"
				      L"\x0CBE-\x0CC4\x0CC6-\x0CC8\x0CCA-\x0CCD\x0CD5-\x0CD6\x0D02-\x0D03"
				      L"\x0D3E-\x0D43\x0D46-\x0D48\x0D4A-\x0D4D\x0D57\x0E31\x0E34-\x0E3A"
				      L"\x0E47-\x0E4E\x0EB1\x0EB4-\x0EB9\x0EBB-\x0EBC\x0EC8-\x0ECD"
				      L"\x0F18-\x0F19\x0F35\x0F37\x0F39\x0F3E\x0F3F\x0F71-\x0F84"
				      L"\x0F86-\x0F8B\x0F90-\x0F95\x0F97\x0F99-\x0FAD\x0FB1-\x0FB7\x0FB9"
				      L"\x20D0-\x20DC\x20E1\x302A-\x302F\x3099\x309A");
	    
		chset_t Digit(L"\x0030-\x0039\x0660-\x0669\x06F0-\x06F9\x0966-\x096F\x09E6-\x09EF"
			      L"\x0A66-\x0A6F\x0AE6-\x0AEF\x0B66-\x0B6F\x0BE7-\x0BEF\x0C66-\x0C6F"
			      L"\x0CE6-\x0CEF\x0D66-\x0D6F\x0E50-\x0E59\x0ED0-\x0ED9\x0F20-\x0F29");
		
		chset_t Extender(L"\x00B7\x02D0\x02D1\x0387\x0640\x0E46\x0EC6\x3005\x3031-\x3035"
				 L"\x309D-\x309E\x30FC-\x30FE");
		
		chset_t NameChar =
		    Letter 
		    | Digit 
		    | L'.'
		    | L'-'
		    | L'_'
		    | L':'
		    | CombiningChar 
		    | Extender;
				
		number = strict_real_p [push_real<FeatureT>(self.exprs)] 
		    | int_p [push_integer<FeatureT>(self.exprs)];
		
		string_ = confix_p(L'\'',(*lex_escape_ch_p)
			 [push_string<FeatureT>(self.exprs)],
				   L'\'');
		
		property = L'[' >> ( (Letter | L'_' | L':') 
				    >> *NameChar )[push_property<FeatureT>(self.exprs)] >> L']';
		
		literal = number | string_ | property;
		
		function = literal | ( func1_op >> L'('>> literal >> L')') | 
		    (func2_op >> L'(' >> literal >> L','>> literal >> L')');
		
		factor = function 
		    | L'(' >> or_expr >> L')'
		    | ( L'-' >> factor) 
		    ;
		term = factor
		    >> *((L'*' >> factor) [compose_expression<FeatureT,mapnik::mult<value> >(self.exprs)] 
			 | (L'/' >> factor) [compose_expression<FeatureT,mapnik::div<value> >(self.exprs)]);
		
		expression = term >> *((L'+' >> term) [compose_expression<FeatureT,mapnik::add<value> >(self.exprs)] 
				       | (L'-' >> term) [compose_expression<FeatureT,mapnik::sub<value> >(self.exprs)]);

		regex = str_p(L".match")>>L'('>>confix_p(L'\'',(*lex_escape_ch_p)
						       [compose_regex<FeatureT>(self.filters,self.exprs)],
						       L'\'') >>L')';

		relation   = expression 
		    >> *((L">=" >> expression) 
			 [compose_filter<FeatureT,greater_than_or_equal<value> >(self.filters,self.exprs)]
			 | (L'>' >> expression)
			 [compose_filter<FeatureT,mapnik::greater_than<value> >(self.filters,self.exprs)]
			 | (L'<' >> expression)
			 [compose_filter<FeatureT,mapnik::less_than<value> >(self.filters,self.exprs)]
			 | (L"<=" >> expression)
			 [compose_filter<FeatureT,less_than_or_equal<value> >(self.filters,self.exprs)]
			 | regex );

		equation = relation >> *( ( L'=' >> relation)
		     [compose_filter<FeatureT,mapnik::equals<value> >(self.filters,self.exprs)]
		     | ( L"<>" >> relation)
		     [compose_filter<FeatureT,not_equals<value> >(self.filters,self.exprs)]);

		not_expr = equation | *(str_p(L"not") >> equation)[compose_not_filter<FeatureT>(self.filters)];

		and_expr = not_expr >> *(L"and" >> not_expr)[compose_and_filter<FeatureT>(self.filters)];

		or_expr  = and_expr >> *(L"or" >> and_expr)[compose_or_filter<FeatureT>(self.filters)];

		filter_statement = or_expr;	
	    }
	    
	    boost::spirit::rule<ScannerT> const& start() const
	    {
		return filter_statement;
	    }
	    	    
	    boost::spirit::rule<ScannerT> factor; 
	    boost::spirit::rule<ScannerT> term;
	    boost::spirit::rule<ScannerT> expression;
	    boost::spirit::rule<ScannerT> relation;
	    boost::spirit::rule<ScannerT> equation;
	    boost::spirit::rule<ScannerT> not_expr;
	    boost::spirit::rule<ScannerT> and_expr;
	    boost::spirit::rule<ScannerT> or_expr;
	    
	    boost::spirit::rule<ScannerT> filter_statement;   
	    boost::spirit::rule<ScannerT> literal;
	    boost::spirit::rule<ScannerT> number;
	    boost::spirit::rule<ScannerT> string_;
	    boost::spirit::rule<ScannerT> property;
	    boost::spirit::rule<ScannerT> function;
	    boost::spirit::rule<ScannerT> regex;
	    symbols<string> func1_op;
	    symbols<string> func2_op;
	    symbols<string> spatial_op;
	};
	stack<ref_ptr<filter<FeatureT> > >& filters;
	stack<ref_ptr<expression<FeatureT> > >& exprs;
    };    
}

#endif //FILTER_PARSER_HPP