%Assignment 3
type TComplexNumber :
    record
        a, b : real
    end record
%%%%%%%%%%%%%%%%%%%%%%%PROCEDURES AND FUNCTION%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

procedure ChangePlaneRtoI (var Msex, Msey, Rightx, Topy : int, var c, t : TComplexNumber)

    var oldca : real := c.a
    var oldcb : real := c.b

    c.a := Msex * ((t.a - c.a) / (maxx + 1)) - t.a
    c.b := Msey * ((t.b - c.b) / (maxy + 1)) - t.b

    t.a := Rightx * ((t.a - oldca) / (maxx + 1)) - t.a
    t.b := Topy * ((t.b - oldcb) / (maxy + 1)) - t.b
   
end ChangePlaneRtoI


procedure DrawMandelbrot (var p, t : TComplexNumber)


    %CONSTANSTS%
    const NumIterations := 255


    %VARIABLE DECLARATIONS%

    var z : TComplexNumber
    var SumZ : TComplexNumber
    var ProductZ : TComplexNumber
    var Oldza : real
    var c : TComplexNumber
    var Xscale : real := (t.a - p.a) / (maxx + 1)
    var Yscale : real := (t.b - p.b) / (maxy + 1)
    var TimeRunning : int := 0

    %counter and color
    var n : int := 0
    var xi, yi : int
    var OutputWindow : int



    c := p

    %Y hits max
    for y : 0 .. maxy + 1
        %X hits max
        for x : 0 .. maxx + 1

            %reset variables
            z.a := 0
            z.b := 0
            n := 0

            %iteration
            loop
                %z = z^2 + c
                Oldza := z.a
                z.a := ((z.a * z.a - z.b * z.b) + c.a)
                z.b := ((Oldza * z.b * 2) + c.b)
                %color and iteration counter
                n := n + 1

                exit when (sqrt (z.a ** 2 + z.b ** 2)) >= 2 or n >= NumIterations

            end loop

            if n > 255 then
                n := black
            end if
            %color each pixel
            drawdot (x, y, n)
           
            c.a += Xscale

        end for

        %reset variables
        c.a := p.a

        c.b += Yscale

    end for
   

end DrawMandelbrot

%%%%%%%%%%%%%%%%%%%%%%%%%%MAIN PROGRAM%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
setscreen ("graphics:800,600")

var c : TComplexNumber
var Msex, Msey : int
var Topy, Rightx : int
var button : int
var t : TComplexNumber
var key : string (1)

const escape := chr (27)


Mouse.ButtonChoose ("singlebutton")

%Initilise Variables
c.a := -2
c.b := -1.5
t.a := 2
t.b := 1.5
DrawMandelbrot (c, t)

loop
    %find mouse location and button status
    Mouse.Where (Msex, Msey, button)
   
    %if a button is pressed save x,y of cursor
    if button = 1 then

        Topy := Msey + round (maxy * 0.05)
        Rightx := Msex + round (maxx * 0.05)

        drawbox (Msex, Msey, Rightx, Topy, yellow)
        % View.Update

        %convert x,y coor to argan plane coor
        ChangePlaneRtoI (Msex, Msey, Rightx, Topy, c, t)


        DrawMandelbrot (c, t)


    end if

    %exit when escape pressed
    if hasch then
        getch (key)
        if key = escape then
            exit
        end if
    end if

end loop