%%
%% %CopyrightBegin%
%% 
%% Copyright Ericsson AB 1996-2009. All Rights Reserved.
%% 
%% The contents of this file are subject to the Erlang Public License,
%% Version 1.1, (the "License"); you may not use this file except in
%% compliance with the License. You should have received a copy of the
%% Erlang Public License along with this software. If not, it can be
%% retrieved online at http://www.erlang.org/.
%% 
%% Software distributed under the License is distributed on an "AS IS"
%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
%% the License for the specific language governing rights and limitations
%% under the License.
%% 
%% %CopyrightEnd%
%%

%%
%% Purpose : Fractal trees

-module(frac).

-export([start/0, go/0, test/0, grow/2, expand/3, subst/2]).

%% 0L - grammer -- context insensitive lindenmayer grammer

start() ->
    spawn(frac,go,[]).

go() ->
    draw(),
    receive
	_X -> true
    end.

draw() ->
    S=gs:start(),
    Width = 800,
    Ht = 520,
    Title="Context Insensitive Lindenmayer Grammer (L0) Trees",
    Win=gs:create(window,S,[{title,Title},{width,Width},{height,Ht}]),
    Canvas=gs:create(canvas,Win,[{width,Width},{height,Ht},{bg,{237,224,189}}]),
    gs:config(Win,[{iconname,"Plants"},{map,true}]),
    draw(Canvas, 1, Width, Ht),
    draw(Canvas, 2, Width, Ht),
    draw(Canvas, 3, Width, Ht),
    draw(Canvas, 4, Width, Ht).

draw(Graph, Index, Width, Ht) ->
    draw_frac(Graph, Index, Width, Ht).
    
test() ->
    grow(3,1).

grow(NGens, RuleNumber) ->
    lists:flatten(expand(NGens, RuleNumber, [a])).

rule(1,a) -> [b,'[',a,']',b,'(',a,')',a];
rule(1,b) -> [b,b];

rule(2,a) -> [b,'[',a,'[',b,a,']',']'];
rule(2,b) -> [b,'(','(',b,')',a,')',c];
rule(2,c) -> [c,d];

rule(3,a) -> [d,'[',d,b,e,']','(',d,c,e,')'];
rule(3,b) -> [d,'[',d,a,f,']','(',d,c,f,')',f];
rule(3,c) -> [d,'[',d,b,g,']','(',d,a,g,')',g];

rule(4,a) -> [c,'(',b,a,'(',b,')',')',c,'[',b,a,'[',b,']',']'];
rule(4,b) -> [c,'(',b,e,')',c,'[',b,f,']'];
rule(4,c) -> [g,c,c];

rule(_,X) -> X.


step(a) -> 1.0;
step(b) -> 0.8;
step(c) -> 0.6;
step(d) -> 0.7;
step(e) -> 0.6;
step(f) -> 0.65;
step(g) -> 0.75;
step(_) -> 1.0.

start_coords(1) -> {0.8,0.8};
start_coords(2) -> {0.6,0.8};
start_coords(3) -> {0.4,0.8};
start_coords(4) -> {0.2,0.8};
start_coords(_) -> {0.5, 0.5}.

gens(1) -> 5;
gens(_) -> 5.

scale(1) -> 5;
scale(2) -> 40;
scale(3) -> 40;
scale(4) -> 4;
scale(_) -> 5.

expand(0,_,X) ->
    X;
expand(N,Index,X) ->
    expand(N - 1, Index, lists:flatten(subst(X, Index))).


subst([],_) -> [];
subst([H|T],Index) ->
    [rule(Index,H)|subst(T,Index)].


draw_frac(Id, Index, Width, Ht) ->
    X0 = 100,
    Y0 = 100,
    {XScale,YScale} = start_coords(Index),
    Xstart = trunc(X0 + Width*XScale),
    Ystart = trunc(Y0 + Ht*YScale),
    Angle = 270.0 * 3.14159 / 180.0,
    Scale = scale(Index),
    N = gens(Index),
    Tree = grow(N,Index),
    drawit(Tree, Id, Xstart, Ystart, Angle, Scale, []).

% drawit(Tree,S0,S).

drawit([],_,_,_,_,_,_) ->
    true;
drawit(['('|T],Id,X0,Y0,Ang,Scale,Stack) ->
    Ang1 =  Ang + (20.0 * 3.14159 / 180.0),
    Scale1 = Scale * 0.8,
    drawit(T,Id,X0,Y0,Ang1,Scale1,[{X0,Y0,Ang,Scale}|Stack]);
drawit(['['|T],Id,X0,Y0,Ang,Scale,Stack) ->
    Ang1 = Ang - (40.0 * 3.14159 / 180.0),
    Scale1 = Scale * 0.8,
    drawit(T,Id,X0,Y0,Ang1,Scale1,[{X0,Y0,Ang,Scale}|Stack]);
drawit([')'|T],Id,_,_,_,_,[{X1,Y1,Ang1,Scale1}|Stack]) ->
    drawit(T,Id,X1,Y1,Ang1,Scale1,Stack);
drawit([']'|T],Id,_,_,_,_,[{X1,Y1,Ang1,Scale1}|Stack]) ->
    drawit(T,Id,X1,Y1,Ang1,Scale1,Stack);
drawit([Symbol|T],Id,X0,Y0,Ang,Scale,Stack) ->
    Size = step(Symbol),
    L = Size * Scale,
    {X1, Y1} = plotit(Id,X0,Y0,L,Ang),
    drawit(T,Id,X1,Y1,Ang,Scale,Stack).

plotit(Id,X0,Y0,L,A) ->
    CosA = math:cos(A),
    SinA = math:sin(A),
    X = trunc(X0 + L*CosA),
    Y = trunc(Y0 + L*SinA),
    gs:create(line,Id,[{coords,[{X0,Y0},{X,Y}]}]),
    {X,Y}.