#!/usr/bin/env python3

import io
import math
import copy
import os.path
import time
import filecmp

from . import testbase

from fract4d_compiler import fc, fracttypes
from fract4d import fractal, image, formsettings

g_testfile = '''gnofract4d parameter file
version=2.0
bailout=5.1
x=0.0891
y=-0.314159
z=0.14
w=0.21
size=4.1
xy=0.00000001
xz=0.1
xw=0.09
yz=-0.1
yw=0.4
zw=0.2
maxiter=259
antialias=1
bailfunc=0
inner=2
outer=1
[function]
function=Mandelbar
[endsection]
[transform]=1
function=XFlip
formulafile=gf4d.uxf
[endsection]
[transform]=0
function=Inverse
formulafile=gf4d.uxf
@radius=3.0
[endsection]
[colorizer]=0
colorizer=1
colordata=000000548878548878548878588c78588c7c588c7c58907c5890805c94805c94805c94805c98845c98845c9884609c88609c88609c8860a08860a08c60a08c64a48c64a49064a49064a89064a89464a89468ac9468ac9468b09868b09868b0986cb49c6cb49c6cb49c6cb89c6cb8a06cb8a070bca070bca470bca470c0a470c0a470c0a874c4a874c4a874c4ac74c8ac74c8ac78ccb0303c38303c38343c38343c383840383840383c40383c403840403840443c44443c44443c48443c48443c4c483c4c483c50483c50483c544840544c40584c40584c405c4c405c4c406050406050406450406450406850446854446c54446c54447054447054447058447458447458447858487858487c5c487c5c48805c48805c48845c4884604888604888604c8c604c8c604c90644c90644c94644c94644c98684c98684c9c684c9c6850a06850a06c50a46c50a46c50a86c50a86c50ac7050ac7050b07054b07054b47054b47454b47454b87454b87454bc7454bc7854c07858c07858c47858c47858c87c58c87c58cc7c58cc7c58d07c58d08058d4805cd4805cd8805cd8805cdc845cdc845ce0845ce0845ce4845ce48860e88860e88860ec8860ec8860f08c60f08c60f48c60f48c60f89064303c3830403c30403c304440344440344844344848344c48384c4c38504c3850503854503c54543c58583c58583c5c5c3c5c5c406060406064406464406468446868446c6c446c6c44707048707448747448747848787848787c4c7c7c4c7c804c80844c808450848850848850888c508890548c90548c9454909454909854949858949c5898a0589ca0589ca45ca0a45ca0a8303c385ca4a85ca4ac60a8b060a8b060acb460acb460b0b864b0bc64b4bc64b4c064b8c068b8c468bcc468bcc868c0cc6cc0cc6cc4d06cc4d06cc8d470ccd870c8d870c4d870c0d86cbcdc6cb8dc6cb4dc68b0dc68ace068a8e068a4e064a0e0649ce46498e46090e4608ce46088e86084e85c80e85c7ce85c78ec5874ec5870ec586cec5868f05464f05460f0545cf05054f45054f45054f05058ec5458ec5458e8
[endsection]
[colorizer]=1
colorizer=1
colordata=30303044fc0094949438f4149090902ce82c8c8c8c24d84488888818cc5c8484840cc07480808000b08c7c7c7c0088a8787878005cc07474740038d8707070000cf46c6c6c1800e46868683800c46464645800a46060607800845c5c5c9c0060585858bc0040545454dc0020505050fc00004c4c4cfc1800484848fc3800444444fc5800404040fc78003c3c3cfc9800383838fcb800343434fcd800303030fcf8002c2c2ce4fc00282828c4fc00242424a4fc0020202084fc001c1c1c64fc0018181844fc0014141438f4141010102ce82c0c0c0c24d84408080818cc5c0404040cc07400000000b08c009c9c0000000070b4040404004ccc0808080020e80c0c0c0800f41010102800d41414144800b41818186800941c1c1c8c0070202020ac0050242424cc0030282828ec00102c2c2cfc0800303030fc2800343434fc4800383838fc68003c3c3cfc8800404040fca800444444fcc800484848fce8004c4c4cf4fc00505050d4fc00545454b4fc0058585894fc005c5c5c74fc0060606054fc006464643cf80868686834ec206c6c6c28e0387070701cd45074747410c86878787808bc807c7c7c009c9c8080800070b4848484004ccc8888880020e88c8c8c0800f49090902800d49494944800b49898986800949c9c9c8c0070a0a0a0ac0050a4a4a4cc0030a8a8a8ec0010acacacfc0800b0b0b0fc2800b4b4b4fc4800b8b8b8fc6800bcbcbcfc8800c0c0c0fca800c4c4c4fcc800c8c8c8fce800ccccccf4fc00d0d0d0d4fc00d4d4d4b4fc00d8d8d894fc00dcdcdc74fc00e0e0e054fc00e4e4e43cf808e8e8e834ec20ececec28e038f0f0f01cd450f4f4f410c868fcfcfc08bc80fcfcfc009c9cf8f8f80070b4f4f4f40038d8f0f0f0000cf4ececec1800e4e8e8e83800c4e4e4e45800a4e0e0e0780084dcdcdc9c0060d8d8d8bc0040d4d4d4dc0020d0d0d0fc0000ccccccfc1800c8c8c8fc3800c4c4c4fc5800c0c0c0fc7800bcbcbcfc9800b8b8b8fcb800b4b4b4fcd800b0b0b0fcf800acacace4fc00a8a8a8c4fc00a4a4a4a4fc00a0a0a084fc009c9c9c64fc00989898
[endsection]
[colorizer]=2
colorizer=1
colordata=0000000000a80400ac0408ac040cac0410ac0814b00818b0081cb00c20b00c24b41028b8102cb81430b81434bc1838c0183cc01c40c01c44c42048c8204cc82450c82454cc2858d0285cd02c60d02c64d43068d8306cd83470d83474dc3878e03c7ce03c80e04084e44088e84484e84488e8448ce84490e84894ec4898f04c9cf04ca0f050a4f450a8f854acf854b0f8287c00287c002c7c002c7c002c7c042c7c042c80042c800430800430800430800830800830840830840834840834840838840c38840c3c840c3c840c3c880c3c880c3c88103c88103c8c103c8c10408c10408c10408c14408c1440901440901444901444901444901844901844941848941848941c48981c4898204c98204c9c20509c20509c2450a02450a02854a02854a42858a42858a42c58a82c58a8305ca8305cac3060ac3060ac3460b03464b03464b03864b43864b43868b43c68b83c68b8406cb8406cbc406cbc4470bc4470bc4870c04870c04c74c04c74c44c74c45078c45078c85078c8547cc8547ccc547ccc5880cc5880d05880d05c84d05c84d45c88d45c88d46088d86088d8648cd8648cdc648cdc6890dc6890e06890e06c94e06c94e46c98e46c98e46c9ce46c9ce46ca0e46ca4e46ca8e46ca8e46cace46cace46cb0e46cb0e46cb4e46cb4e46cb8e46cb8e46cbce46cbce46cc0e46cc4e46cc4e46cc8e46ccce46ccce46cd0e46cd0e46cd4e46cd4e46cd8e46cdce46cdce46ce0e46ce4e46ce4e46ce8e46ce8e46ce8e468e8e468e8e464e8e464e8e460e8e460e8e45ce8e05ce8e05ce8dc58e8d854e8d850e8d450e8d44ce8d04ce8cc48e8cc44e8c844e8c840e8c840e8c440e8c43ce8c03ce8c038e8bc38e8bc34e8b834e8b830e8b430e8b42ce8b42ce8b02ce8b02ce8b028e8b028e8ac28e8ac24e8a824e8a824e4a420e4a420e4a020e4a020e09c1ce09818e09818e09414dc9414dc8c10dc8c10dc8410d88410d87c08d87c08d87408808080808080fcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcc0c0c0c0c0c0fcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfc
[endsection]
[colorizer]=3
colorizer=1
colordata=0000000000a80400ac0408ac040cac0410ac0814b00818b0081cb00c20b00c24b41028b8102cb81430b81434bc1838c0183cc01c40c01c44c42048c8204cc82450c82454cc2858d0285cd02c60d02c64d43068d8306cd83470d83474dc3878e03c7ce03c80e04084e44088e84484e84488e8448ce84490e84894ec4898f04c9cf04ca0f050a4f450a8f854acf854b0f8287c00287c002c7c002c7c002c7c042c7c042c80042c800430800430800430800830800830840830840834840834840838840c38840c3c840c3c840c3c880c3c880c3c88103c88103c8c103c8c10408c10408c10408c14408c1440901440901444901444901444901844901844941848941848941c48981c4898204c98204c9c20509c20509c2450a02450a02854a02854a42858a42858a42c58a82c58a8305ca8305cac3060ac3060ac3460b03464b03464b03864b43864b43868b43c68b83c68b8406cb8406cbc406cbc4470bc4470bc4870c04870c04c74c04c74c44c74c45078c45078c85078c8547cc8547ccc547ccc5880cc5880d05880d05c84d05c84d45c88d45c88d46088d86088d8648cd8648cdc648cdc6890dc6890e06890e06c94e06c94e46c98e46c98e46c9ce46c9ce46ca0e46ca4e46ca8e46ca8e46cace46cace46cb0e46cb0e46cb4e46cb4e46cb8e46cb8e46cbce46cbce46cc0e46cc4e46cc4e46cc8e46ccce46ccce46cd0e46cd0e46cd4e46cd4e46cd8e46cdce46cdce46ce0e46ce4e46ce4e46ce8e46ce8e46ce8e468e8e468e8e464e8e464e8e460e8e460e8e45ce8e05ce8e05ce8dc58e8d854e8d850e8d450e8d44ce8d04ce8cc48e8cc44e8c844e8c840e8c840e8c440e8c43ce8c03ce8c038e8bc38e8bc34e8b834e8b830e8b430e8b42ce8b42ce8b02ce8b02ce8b028e8b028e8ac28e8ac24e8a824e8a824e4a420e4a420e4a020e4a020e09c1ce09818e09818e09414dc9414dc8c10dc8c10dc8410d88410d87c08d87c08d87408808080808080fcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcc0c0c0c0c0c0fcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfcfc
[endsection]
'''

g_test2file = '''gnofract4d parameter file
version=2.0
[function]
formulafile=gf4d.frm
function=Mandelbrot
@bailfunc=manhattanish
@bailout=1e20
[endsection]
[inner]
formulafile=test.cfrm
function=flat
@_density=2.0
@_offset=0.5
@b=0
@col=(0.09,0.08,0.07,0.06)
@ep=2
@i=78
@myfunc=sqrt
@val=3.3
@val2=(2.0,3.7,6.1,8.9)
[endsection]
[outer]
formulafile=test.cfrm
function=Triangle
@power=3.0
@bailout=1.0e12
[endsection]
[colors]
colorizer=1
solids=[
000000ff
000000ff
]
colorlist=[
0.000000=00000000
1.000000=ffffffff
]
'''

g_test3file = '''gnofract4d parameter file
version=3.9
x=0.00000000000000000
y=0.00000000000000000
z=0.00000000000000000
w=0.00000000000000000
size=4.00000000000000000
xy=0.00000000000000000
xz=0.00000000000000000
xw=0.00000000000000000
yz=0.00000000000000000
yw=0.00000000000000000
zw=0.00000000000000000
maxiter=64
yflip=False
periodicity=1
period_tolerance=0.00000010000000000
[function]
formulafile=__inline__1.frm
function=Mandelbrot
formula=[
Mandelbrot {
; The classic Mandelbrot set
init:
	z = #zwpixel
loop:
	z = z * z + #pixel
bailout:
	@bailfunc(z) < @bailout
default:
float param bailout
	default = 4.0
endparam
float func bailfunc
	default = cmag
endfunc
}
]
@bailfunc=cmag
@_gradient=[
GIMP Gradient
7
0.000000 0.071429 0.142857 0.200000 0.133333 0.066667 1.000000 0.000000 0.266667 0.400000 1.000000 0 0
+0.214286 0.285714 0.133333 0.400000 0.400000 1.000000 0 0
+0.357143 0.428571 0.666667 0.533333 0.000000 1.000000 0 0
+0.500000 0.571429 0.533333 0.266667 0.266667 1.000000 0 0
+0.642857 0.714286 0.333333 0.400000 0.466667 1.000000 0 0
+0.785714 0.857143 0.466667 0.466667 0.400000 1.000000 0 0
+0.928571 1.000000 0.466667 0.466667 0.400000 1.000000 0 0
]
@bailout=4.00000000000000000
[endsection]
[outer]
formulafile=__inline__2.cfrm
function=continuous_potential
@_transfer=ident
@_density=1.00000000000000000
@_offset=0.00000000000000000
@bailout=4.00000000000000000
formula=[
continuous_potential {
final:
float ed = @bailout/(|z| + 1.0e-9)
#index = (#numiter + ed) / 256.0
default:
float param bailout
	default = 4.0
endparam
}
]
[endsection]
[inner]
formulafile=__inline__3.cfrm
function=zero
@_transfer=ident
@_density=1.00000000000000000
@_offset=0.00000000000000000
formula=[
zero (BOTH) {
final:
#solid = true
}
]
[endsection]
[colors]
colorizer=1
solids=[
000000ff
000000ff
]
'''

g_testfilemultiframes = '''gnofract4d parameter file
version=3.10
x=0.00000000000000000
y=0.00000000000000000
z=0.00000000000000000
w=0.00000000000000000
size=4.00000000000000000
xy=0.00000000000000000
xz=0.00000000000000000
xw=0.00000000000000000
yz=0.00000000000000000
yw=0.00000000000000000
zw=0.00000000000000000
maxiter=64
yflip=False
periodicity=1
period_tolerance=0.00000010000000000
[function]
formulafile=__inline__1.frm
function=Mandelbrot
formula=[
Mandelbrot {
; The classic Mandelbrot set
init:
	z = #zwpixel
loop:
	z = z * z + #pixel
bailout:
	@bailfunc(z) < @bailout
default:
float param bailout
	default = 4.0
endparam
float func bailfunc
	default = cmag
endfunc
}
]
@bailfunc=cmag
@_gradient=[
GIMP Gradient
7
0.000000 0.071429 0.142857 0.200000 0.133333 0.066667 1.000000 0.000000 0.266667 0.400000 1.000000 0 0
+0.214286 0.285714 0.133333 0.400000 0.400000 1.000000 0 0
+0.357143 0.428571 0.666667 0.533333 0.000000 1.000000 0 0
+0.500000 0.571429 0.533333 0.266667 0.266667 1.000000 0 0
+0.642857 0.714286 0.333333 0.400000 0.466667 1.000000 0 0
+0.785714 0.857143 0.466667 0.466667 0.400000 1.000000 0 0
+0.928571 1.000000 0.466667 0.466667 0.400000 1.000000 0 0
]
@bailout=4.00000000000000000
[endsection]
[outer]
formulafile=__inline__2.cfrm
function=continuous_potential
@_transfer=ident
@_density=1.00000000000000000
@_offset=0.00000000000000000
@bailout=4.00000000000000000
formula=[
continuous_potential {
final:
float ed = @bailout/(|z| + 1.0e-9)
#index = (#numiter + ed) / 256.0
default:
float param bailout
	default = 4.0
endparam
}
]
[endsection]
[inner]
formulafile=__inline__3.cfrm
function=zero
@_transfer=ident
@_density=1.00000000000000000
@_offset=0.00000000000000000
formula=[
zero (BOTH) {
final:
#solid = true
}
]
[endsection]
[colors]
colorizer=1
solids=[
000000ff
000000ff
]
[endsection]
[frames]
frames=20
[frame]=10
x=10.00000000000000000
[endsection]
[endsection]
'''


class WarningCatcher:
    def __init__(self):
        self.warnings = []

    def warn(self, msg):
        self.warnings.append(msg)


class Test(testbase.ClassSetup):
    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        cls.g_comp.add_path("maps", fc.FormulaTypes.GRADIENT)
        cls.g_comp.load_formula_file("gf4d.frm")
        cls.g_comp.load_formula_file("gf4d.cfrm")
        cls.g_comp.load_formula_file("gf4d.cfrm")

    def setUp(self):
        self.default_flat_params = [
            0.0,  # _offset
            1.0,  # _density
            0.4,  # float val
            0.7, 0.8, 0.9, 1.0,  # hyper val2
            4,  # int i
            1,  # bool b
            1,  # enum ep
            0.01, 0.02, 0.03, 0.04,  # color col
        ]

    def tearDown(self):
        pass

    def testCreation(self):
        f = fractal.T(Test.g_comp)
        f.compile()

    def testRead(self):
        file = g_testfile

        f = fractal.T(Test.g_comp)
        f.loadFctFile(io.StringIO(file))
        self.assertExpectedValues(f)

    def testUpsideDown(self):
        file = g_testfile

        file = file.replace('version=2.0', 'version=1.9', 1)
        f = fractal.T(Test.g_comp)
        f.loadFctFile(io.StringIO(file))
        self.assertEqual(f.params[f.XYANGLE], 0.00000001)
        self.assertEqual(f.yflip, True)
        f.reset()
        self.assertEqual(f.yflip, False)

    def assertExpectedValues(self, f):
        self.assertEqual(f.params[f.XCENTER], 0.0891)
        self.assertEqual(f.params[f.YCENTER], -0.314159)
        self.assertEqual(f.params[f.ZCENTER], 0.14)
        self.assertEqual(f.params[f.WCENTER], 0.21)
        self.assertEqual(f.params[f.MAGNITUDE], 4.1)
        self.assertEqual(f.params[f.XYANGLE], 0.00000001)
        self.assertEqual(f.params[f.XZANGLE], 0.1)
        self.assertEqual(f.params[f.XWANGLE], 0.09)
        self.assertEqual(f.params[f.YZANGLE], -0.1)
        self.assertEqual(f.params[f.YWANGLE], 0.4)
        self.assertEqual(f.params[f.ZWANGLE], 0.2)
        self.assertEqual(5.1, f.forms[0].get_named_param_value("@bailout"))

        self.assertEqual(f.forms[0].funcName, "Mandelbar")
        self.assertEqual(f.forms[0].funcFile, "gf4d.frm")
        self.assertEqual(f.forms[1].funcName, "continuous_potential")
        self.assertEqual(f.forms[1].funcFile, "gf4d.cfrm")
        self.assertEqual(f.forms[2].funcName, "zero")
        self.assertEqual(f.forms[2].funcFile, "gf4d.cfrm")

        self.assertEqual(2, len(f.transforms))
        t = f.transforms[0]
        self.assertEqual("gf4d.uxf", t.funcFile)
        self.assertEqual("Inverse", t.funcName)
        self.assertEqual(3.0, t.get_named_param_value("@radius"))

        self.assertEqual(f.maxiter, 259)
        g = f.get_gradient()
        self.assertTrue(len(g.segments) > 1)
        self.assertEqual(g.segments[0].left, 0.0)
        self.assertEqual(g.segments[-1].right, 1.0)
        self.assertEqual(f.solids[0], (0, 0, 0, 255))
        self.assertEqual(f.yflip, False)

        self.assertEqual(True, f.periodicity)
        self.assertEqual(1.0E-9, f.period_tolerance)

        sofile = f.compile()
        im = image.T(40, 30)
        f.draw(im)
        Test.g_comp.leave_dirty = True

    def testLoadGradientFunc(self):
        f = fractal.T(Test.g_comp)
        file = open("testdata/gradient_func.fct")
        f.loadFctFile(file)
        file.close()

        f.compile()
        (w, h) = (40, 30)
        im = image.T(w, h)
        f.draw(im)

    def testRefresh(self):
        formula = '''
test_circle {
loop:
z = pixel
bailout:
|z| < @bailout
default:
float param bailout
	default = 4.0
endparam
}
'''
        fracttest_file = os.path.join(Test.tmpdir.name, "fracttest.frm")
        with open(fracttest_file, "w") as ff:
            ff.write(formula)

        f = fractal.T(Test.g_comp)
        f.set_formula(fracttest_file, "test_circle")

        self.assertEqual(f.get_initparam(1, 0), 4.0)
        time.sleep(1.0)
        formula = formula.replace('4.0', '6.0')
        with open(fracttest_file, "w") as ff:
            ff.write(formula)

        self.assertEqual(f.get_initparam(1, 0), 4.0)
        f.refresh()
        self.assertEqual(f.get_initparam(1, 0), 6.0)

    def testSetSolid(self):
        f = fractal.T(Test.g_comp)
        f.set_solid(0, (255, 127, 8, 190))
        self.assertEqual((255, 127, 8, 190), f.solids[0])

    def testLoadMaliciousFile(self):
        'Try to inject code into a file in a way which worked on 2.0 and 2.1'
        bad_testfile = '''gnofract4d parameter file
version(2.0,None)+open("evil.txt","w")=2.0
[function]
formulafile=test.frm
function=parse_error
[endsection]
[inner]
formulafile=test.cfrm
function=flat
@myfunc=sqrt
[endsection]
[outer]
formulafile=test.cfrm
function=Triangle
@power=3.0
@bailout=1.0e12
[endsection]
[colors]
colorizer=1
solids=[
000000ff
000000ff
]
colorlist=[
0.000000=00000000
1.000000=ffffffff
]
'''
        wc = WarningCatcher()
        f = fractal.T(Test.g_comp)
        f.warn = wc.warn
        self.assertRaises(ValueError, f.loadFctFile,
                          (io.StringIO(bad_testfile)))

        self.assertEqual(os.path.exists('evil.txt'), False)

    def testLoadBadColorizerType(self):
        bad_testfile = '''gnofract4d parameter file
version=2.0
[function]
formulafile=test.frm
function=test_noz
[endsection]
[inner]
formulafile=test.cfrm
function=flat
@myfunc=sqrt
[endsection]
[outer]
formulafile=test.cfrm
function=Triangle
@power=3.0
@bailout=1.0e12
[endsection]
[colors]
colorizer=7
solids=[
000000ff
000000ff
]
colorlist=[
0.000000=00000000
1.000000=ffffffff
]
'''
        f = fractal.T(Test.g_comp)
        self.assertRaises(ValueError, f.loadFctFile,
                          (io.StringIO(bad_testfile)))

    def testLoadColorFile(self):
        testfile = '''gnofract4d parameter file
version=2.0
[function]
formulafile=test.frm
function=test_noz
[endsection]
[inner]
formulafile=test.cfrm
function=flat
@myfunc=sqrt
[endsection]
[outer]
formulafile=test.cfrm
function=Triangle
@power=3.0
@bailout=1.0e12
[endsection]
[colors]
colorizer=0
file=maps/4zebbowx.map
solids=[
000000ff
000000ff
]
'''
        f = fractal.T(Test.g_comp)
        f.deserialize(testfile)
        self.assertEqual(len(f.get_gradient().segments), 255)

    def testLoadFileWithBadFormula(self):
        bad_testfile = '''gnofract4d parameter file
version=2.0
[function]
formulafile=test.frm
function=parse_error
[endsection]
[inner]
formulafile=test.cfrm
function=flat
@myfunc=sqrt
[endsection]
[outer]
formulafile=test.cfrm
function=Triangle
@power=3.0
@bailout=1.0e12
[endsection]
[colors]
colorizer=1
solids=[
000000ff
000000ff
]
colorlist=[
0.000000=00000000
1.000000=ffffffff
]
'''
        f = fractal.T(Test.g_comp)
        self.assertRaises(ValueError, f.loadFctFile,
                          (io.StringIO(bad_testfile)))

    def testLoadBadColor(self):
        bad_testfile = '''gnofract4d parameter file
version=2.0
[function]
formulafile=test.frm
function=test_noz
[endsection]
[inner]
formulafile=test.cfrm
function=flat
@myfunc=sqrt
[endsection]
[outer]
formulafile=test.cfrm
function=Triangle
@power=3.0
@bailout=1.0e12
[endsection]
[colors]
colorizer=0
solids=[
000000ff
000000ff
]
colorlist=[
0.000000
1.000000=ffffffff
]
'''
        f = fractal.T(Test.g_comp)
        self.assertRaises(ValueError, f.loadFctFile,
                          (io.StringIO(bad_testfile)))

    def testLoadBoolParamSavedByOlderVersion(self):
        '''Bug reported by Elaine Normandy: file saved by 2.7 containing
        boolean param can\'t be loaded by 2.8'''

        f = fractal.T(Test.g_comp)
        file = open("testdata/chainsoflight.fct")
        f.loadFctFile(file)
        file.close()
        self.assertEqual(f.periodicity, True)

    def testSaveFlag(self):
        'Test that we know when we\'re up-to-date on disk'
        f = fractal.T(Test.g_comp)
        self.assertEqual(f.saved, True)

        f.set_param(0, 7.3)
        self.assertEqual(f.saved, False)

        savefile = io.StringIO("")
        f.save(savefile)
        self.assertEqual(f.saved, True)

        c = copy.copy(f)
        self.assertEqual(c.saved, True)

        f.set_param(0, 7.3)
        self.assertEqual(f.saved, True)

        f.set_param(0, 7.7)
        self.assertEqual(f.saved, False)

        x = f.serialize()
        self.assertEqual(f.saved, False)

        f.loadFctFile(io.StringIO(g_testfile))
        self.assertEqual(f.saved, True)

    def testLoadBadFileRaises(self):
        'Test we throw an exception when loading an invalid file'
        f = fractal.T(Test.g_comp)
        not_a_file = io.StringIO("ceci n'est pas un file")
        self.assertRaises(Exception, f.loadFctFile, not_a_file)

    def testSetBadGradientFromFile(self):
        f = fractal.T(Test.g_comp)
        self.assertRaises(
            IOError,
            f.set_gradient_from_file,
            "fish.uxf",
            "moops")

    def testSetGradientFromFile(self):
        f = fractal.T(Test.g_comp)
        f.set_gradient_from_file("blatte1.ugr", "blatte10")
        self.assertEqual("blatte10", f.get_gradient().name)

    def testIntParams(self):
        f = fractal.T(Test.g_comp)
        f.set_formula("test.frm", "fn_with_intparam")

        p = f.forms[0].formula.symbols.parameters()
        op = f.forms[0].formula.symbols.order_of_params()

        self.assertEqual(len(p), 2)
        self.assertEqual(p["t__a_x"].type, fracttypes.Int)
        self.assertEqual(op["t__a_x"], 1)
        self.assertEqual(op["__SIZE__"], 2)

        tp = f.forms[0].formula.symbols.type_of_params()

        self.assertEqual(len(tp), 2)
        self.assertEqual(tp[1], fracttypes.Int)

        f.set_initparam(1, "17", 0)
        self.assertEqual(f.forms[0].params[1], 17)
        self.assertTrue(isinstance(f.forms[0].params[1], int))

    def testCFParams(self):
        f = fractal.T(Test.g_comp)

        # offset, density, bailout
        self.assertEqual([0.0, 1.0, 4.0], f.forms[1].params)

        f.set_outer("test.cfrm", "Triangle")

        # offset, density, power, bailout
        self.assertEqual([0.0, 1.0, 2.0, 1.0e20], f.forms[1].params)

        cf0p = f.forms[1].formula.symbols.parameters()
        self.assertEqual(cf0p["t__a_bailout"].cname, "t__a_cf0bailout")
        p = f.forms[0].formula.symbols.parameters()
        self.assertEqual(p["t__a_bailout"].cname, "t__a_fbailout")

        cg = Test.g_comp.compile(f.forms[0].formula)
        Test.g_comp.compile(f.forms[1].formula)
        Test.g_comp.compile(f.forms[2].formula)

        f.forms[0].formula.merge(f.forms[1].formula, "cf0_")
        f.forms[0].formula.merge(f.forms[2].formula, "cf1_")

        p2 = f.forms[0].formula.symbols.parameters()

        # 11 = (2 x bailout), bailfunc, size, gradient +
        #      2x (density, offset, transfer)
        self.assertEqual(len(p2), 11)
        self.assertEqual(p2["t__a_bailout"].cname, "t__a_fbailout")
        self.assertEqual(p2["t__a_cf0bailout"].cname, "t__a_cf0bailout")

        op2 = f.forms[0].formula.symbols.order_of_params()

        expected_order = {
            't__a__gradient': 0,
            't__a_bailout': 1,
            't__a_cf0_offset': 2,
            't__a_cf0_density': 3,
            't__a_cf0power': 4,
            't__a_cf0bailout': 5,
            't__a_cf1_offset': 6,
            't__a_cf1_density': 7,
            '__SIZE__': 8
        }
        self.assertEqual(expected_order, op2)

        expected_params = [
            # f
            f.get_gradient(),
            4.0,
            # cf0
            0.0,
            1.0,
            2.0,
            1.0e20,
            # cf1
            0.0,
            1.0]

        params = f.all_params()
        self.assertEqual(expected_params, params)

        # check for appropriate snippets in the code
        cg.output_decls(f.forms[0].formula)
        c_code = cg.output_c(f.forms[0].formula)

        self.assertTrue("double t__a_cf0bailout = t__pfo->p[5]" in c_code)

        self.assertNotEqual(  # use
            c_code.find("log(t__a_cf0bailout)"), -1)

    def testAllParams(self):
        f = fractal.T(Test.g_comp)
        self.assertEqual(f.forms[1].params, [0.0, 1.0, 4.0])

        f.set_outer("test.cfrm", "Triangle")
        f.append_transform("gf4d.uxf", "Inverse")

        f.compile()
        params = f.all_params()
        self.assertEqual(11, len(params))

        self.assertEqual(
            [
                # f
                f.get_gradient(), 4.0,
                # cf0
                0.0, 1.0, 2.0, 1.0e20,
                # cf1
                0.0, 1.0,
                # transform
                1.0, 0.0, 0.0],
            params)

    def assertNearlyEqual(self, a, b):
        # check that each element is within epsilon of expected value
        epsilon = 1.0e-12
        for (ra, rb) in zip(a, b):
            if isinstance(ra, float):
                d = abs(ra - rb)
                self.assertTrue(d < epsilon, "%f != %f (by %f)" % (ra, rb, d))
            else:
                self.assertEqual(ra, rb)

    def testLoadRGBColorizer(self):
        'load an rgb colorizer'
        file = '''gnofract4d parameter file
version=1.6
bailout=4
x=-0.13125000000000000555
y=-0.7562499999999999778
z=0
w=0
size=0.4000000000000000222
xy=0
xz=0
xw=0
yz=0
yw=0
zw=0
maxiter=1600
antialias=1
bailfunc=0
inner=1
outer=1
[function]
function=Mandelbrot
[endsection]
[colors]
colorizer=0
red=0.87
green=0.666
blue=0.3
[endsection]
'''
        f = fractal.T(Test.g_comp)
        rgb_file = io.StringIO(file)

        f.loadFctFile(rgb_file)
        self.assertEqual(f.forms[1].funcName, "rgb")

    def testLoadWithInlineFormula(self):
        f1 = fractal.T(Test.g_comp)
        file1 = io.StringIO(g_test3file)
        f1.loadFctFile(file1)

        self.assertTrue("__inline__" in f1.forms[0].funcFile)
        self.assertEqual(f1.forms[0].funcName, "Mandelbrot")

        self.assertTrue("__inline__" in f1.forms[1].funcFile)
        self.assertEqual(f1.forms[1].funcName, "continuous_potential")

        self.assertTrue("__inline__" in f1.forms[2].funcFile)
        self.assertEqual(f1.forms[2].funcName, "zero")

        outfile = io.StringIO()
        f1.save(outfile)

        self.assertTrue(outfile.getvalue().count("formula=[") == 3)

    def testSaveWithCFParams(self):
        'load and save a file with a colorfunc which has parameters'
        f1 = fractal.T(Test.g_comp)
        file1 = io.StringIO(g_test2file)
        f1.loadFctFile(file1)

        f1.compile()

        self.assertEqual(f1.forms[1].params, [0.0, 1.0, 3.0, 1.0e12])
        self.assertEqual(f1.forms[2].params, [
            0.5,  # _offset
            2.0,  # _density
            3.3,  # float val
            2.0, 3.7, 6.1, 8.9,  # hyper val2
            78,  # int i
            0,  # bool b
            2,  # enum ep
            0.09, 0.08, 0.07, 0.06,  # color col
        ])
        self.assertEqual(f1.forms[2].get_func_value("@myfunc"), "sqrt")

        # save again
        file2 = io.StringIO()
        f1.save(file2)
        saved = file2.getvalue()
        self.assertTrue(saved.startswith("gnofract4d parameter file"))
        self.assertNotEqual(saved.find("@power=3.0"), -1)

        # load it into another instance
        file3 = io.StringIO(saved)
        f3 = fractal.T(Test.g_comp)
        f3.loadFctFile(file3)

        self.assertFractalsEqual(f1, f3)
        self.assertEqual(f3.forms[2].get_func_value("@myfunc"), "sqrt")

    def testParseVersionString(self):
        f = fractal.T(Test.g_comp)
        self.assertEqual(2000.0, f.parse_version_string("2.0"))
        self.assertTrue(f.parse_version_string("2.14")
                        > f.parse_version_string("2.9"))

    def assertFuncsEqual(self, form1, form2):
        for name in form1.func_names():
            self.assertEqual(form1.get_func_value(name),
                             form2.get_func_value(name))

    def assertFormSettingsEqual(self, fs1, fs2):
        self.assertFuncsEqual(fs1, fs2)
        self.assertEqual(fs1.params, fs2.params)
        self.assertEqual(fs1.paramtypes, fs2.paramtypes)
        self.assertEqual(fs1.funcName, fs2.funcName)
        self.assertEqual(fs1.funcFile, fs2.funcFile)

    def assertFractalsEqual(self, f1, f2):
        # check that they are equivalent
        self.assertEqual(f1.maxiter, f2.maxiter)
        self.assertEqual(f1.params, f2.params)

        self.assertFormSettingsEqual(f1.forms[0], f2.forms[0])
        self.assertFormSettingsEqual(f1.forms[1], f2.forms[1])
        self.assertFormSettingsEqual(f1.forms[2], f2.forms[2])

        self.assertEqual(f1.yflip, f2.yflip)

        self.assertEqual(f1.get_gradient(), f2.get_gradient())
        self.assertEqual(f1.warp_param, f2.warp_param)

        self.assertEqual(f1.periodicity, f2.periodicity)
        self.assertEqual(f1.period_tolerance, f2.period_tolerance)

    def testSave(self):
        self.runSaveTest(False)
        self.runSaveTest(True)

    def runSaveTest(self, compressed):
        # load some settings
        f1 = fractal.T(Test.g_comp)
        file1 = io.StringIO(g_testfile)
        f1.loadFctFile(file1)

        # save again
        file2 = io.StringIO()
        f1.save(file2, compress=compressed)
        saved = file2.getvalue()
        self.assertTrue(saved.startswith("gnofract4d parameter file"))

        # load it into another instance
        file3 = io.StringIO(saved)
        f2 = fractal.T(Test.g_comp)
        f2.loadFctFile(file3)
        f2.auto_deepen = False
        f2.auto_tolerance = False

        self.assertExpectedValues(f2)

        # check that they are equivalent
        self.assertFractalsEqual(f1, f2)

    def testResetZoom(self):
        # mandelbrot has no specifier, picks up 4.0
        f = fractal.T(Test.g_comp)

        f.set_param(f.MAGNITUDE, 0.002)

        f.reset_zoom()
        self.assertEqual(4.0, f.get_param(f.MAGNITUDE))

        # a fractal which sets #magnitude
        f.set_formula("gf4d.frm", "Buffalo")
        f.set_param(f.MAGNITUDE, 0.002)

        f.reset_zoom()
        self.assertEqual(6.0, f.get_param(f.MAGNITUDE))

    def testRelocation(self):
        f = fractal.T(Test.g_comp)

        f.compile()

        # zoom
        f.relocate(0.0, 0.0, 2.0)
        tparams = [0.0] * 11
        tparams[f.MAGNITUDE] = 8.0
        self.assertNearlyEqual(f.params, tparams)

        # relocate
        f.relocate(1.0, 2.0, 1.0)
        tparams[f.XCENTER] = 8.0
        tparams[f.YCENTER] = -16.0
        self.assertNearlyEqual(f.params, tparams)

        # rotated relocation
        f.relocate(-1.0, -2.0, 1.0)
        f.params[f.XYANGLE] = -math.pi / 2.0
        f.relocate(1.0, 2.0, 1.0)
        tparams[f.XCENTER] = 16.0
        tparams[f.YCENTER] = 8.0
        tparams[f.XYANGLE] = -math.pi / 2.0

        self.assertNearlyEqual(f.params, tparams)

        # Julia relocation
        f.relocate(-1.0, -2.0, 1.0)
        f.params[f.XYANGLE] = 0
        f.params[f.XZANGLE] = f.params[f.YWANGLE] = math.pi / 2.0
        f.relocate(1.0, 2.0, 1.0)

        tparams = [0.0] * 11
        tparams[f.MAGNITUDE] = 8.0
        tparams[f.ZCENTER] = 8.0
        tparams[f.WCENTER] = -16.0
        tparams[f.XZANGLE] = tparams[f.YWANGLE] = math.pi / 2.0

        self.assertNearlyEqual(f.params, tparams)

        # equivalent Julia relocation using axis param
        f.reset()

        f.relocate(-1.0, -2.0, 1.0, 2)

        tparams = [0.0] * 11
        tparams[f.MAGNITUDE] = 4.0
        tparams[f.ZCENTER] = -4.0
        tparams[f.WCENTER] = 8.0

        self.assertNearlyEqual(f.params, tparams)

    def testNudges(self):
        f = fractal.T(Test.g_comp)

        f.set_formula("gf4d.frm", "Nova")
        f.compile()

        f.nudge(-1, 0)
        tparams = [0.0] * 11
        tparams[f.MAGNITUDE] = 4.0
        tparams[f.XCENTER] = -4.0 * 0.025

        self.assertNearlyEqual(f.params, tparams)

        f.nudge(0, 1)
        tparams[f.YCENTER] = -4.0 * 0.025

        self.assertNearlyEqual(f.params, tparams)

        f.nudge(-1, -1, 2)
        tparams[f.ZCENTER] = -4.0 * 0.025
        tparams[f.WCENTER] = 4.0 * 0.025

        op = f.forms[0].formula.symbols.order_of_params()
        k_a = op["t__a_a"]

        oldparams = copy.copy(f.forms[0].params)
        f.nudge_param(k_a, 0, 1, 2)

        oldparams[k_a] += 1.0 * 0.025
        oldparams[k_a + 1] += 2.0 * 0.025

        self.assertNearlyEqual(f.forms[0].params, oldparams)

    def testPeriodTolerance(self):
        f = fractal.T(Test.g_comp)
        f.compile()

        (w, h) = (40, 30)
        im = image.T(w, h)
        f.draw(im)

        f.set_period_tolerance(1.0E10)  # really big!
        im2 = image.T(w, h)
        f.draw(im2)

        # the image with loose tolerance should be inside everywhere the
        # tight one is, and some more places too

        for y in range(h):
            for x in range(w):
                (is_solid, fate) = im.get_fate(x, y)
                if fate == 32:
                    (is_solid2, fate2) = im2.get_fate(x, y)
                    self.assertEqual(
                        fate, fate2, "tolerance lost a pixel @ %d, %d" % (x, y))

    def testDefaultFractal(self):
        f = fractal.T(Test.g_comp)

        # check defaults
        self.assertEqual(f.params[f.XCENTER], 0.0)
        self.assertEqual(f.params[f.YCENTER], 0.0)
        self.assertEqual(f.params[f.ZCENTER], 0.0)
        self.assertEqual(f.params[f.WCENTER], 0.0)
        self.assertEqual(f.params[f.MAGNITUDE], 4.0)
        self.assertEqual(f.params[f.XYANGLE], 0.0)
        self.assertEqual(f.params[f.XZANGLE], 0.0)
        self.assertEqual(f.params[f.XWANGLE], 0.0)
        self.assertEqual(f.params[f.YZANGLE], 0.0)
        self.assertEqual(f.params[f.YWANGLE], 0.0)
        self.assertEqual(f.params[f.ZWANGLE], 0.0)
        self.assertEqual(f.forms[0].params, [f.get_gradient(), 4.0])

        image1_file = os.path.join(Test.tmpdir.name, "def.tga")
        f.compile()
        (w, h) = (40, 30)
        im = image.T(w, h)
        f.auto_deepen = False
        f.auto_tolerance = False
        f.draw(im)
        im.save(image1_file)

        buf = im.image_buffer(0, 0)

        # corners must be white
        self.assertWhite(buf, 0, 0, w)
        self.assertWhite(buf, w - 1, 0, w)
        self.assertWhite(buf, 0, h - 1, w)
        self.assertWhite(buf, w - 1, h - 1, w)

        # center is black
        self.assertBlack(buf, w // 2, h // 2, w)

        # and vertically symmetrical
        for x in range(w):
            for y in range(h // 2):
                apos = (y * w + x) * 3
                bpos = ((h - y - 1) * w + x) * 3
                a = buf[apos:apos + 3]
                b = buf[bpos:bpos + 3]
                self.assertEqual(a, b)

        # draw it again in fragments and check result is identical
        image2_file = os.path.join(Test.tmpdir.name, "def2.tga")
        im = image.T(40, 4, 40, 30)
        im.start_save(image2_file)
        f.draw(im)
        im.finish_save()

        self.assertEqual(True, filecmp.cmp(image1_file, image2_file, False))

    def testReset(self):
        # test that formula's defaults are applied
        f = fractal.T(Test.g_comp)

        f.params[f.XCENTER] = 777.0
        f.set_formula("test.frm", "test_defaults")
        f.forms[0].set_named_item("@bailout", 7.1)

        f.set_inner("test.cfrm", "flat")
        f.forms[2].set_named_item("@val", 0.2)

        f.reset()
        self.assertEqual(f.maxiter, 200)
        self.assertEqual(f.params[f.XCENTER], 1.0)
        self.assertEqual(f.params[f.YCENTER], 2.0)
        self.assertEqual(f.params[f.ZCENTER], 7.1)
        self.assertEqual(f.params[f.WCENTER], 2.9)
        self.assertEqual(f.params[f.MAGNITUDE], 8.0)

        self.assertEqual(f.params[f.XYANGLE], 0.001)
        self.assertEqual(f.params[f.XZANGLE], 0.789)
        self.assertEqual(f.title, "Hello World")
        self.assertEqual(f.forms[0].params, [f.get_gradient(), 8.0, 7.0, 1.0])
        self.assertEqual(f.periodicity, 0)

        self.assertEqual(
            self.default_flat_params,
            f.forms[2].params)

    def testResetAngles(self):
        f = fractal.T(Test.g_comp)
        f.params[f.XYANGLE] = 0.1
        f.params[f.XZANGLE] = 0.2
        f.params[f.XWANGLE] = 0.3
        f.params[f.YZANGLE] = 0.4
        f.params[f.YWANGLE] = 0.5
        f.params[f.ZWANGLE] = 0.6
        f.reset_angles()
        self.assertEqual(f.params[f.XYANGLE:f.ZWANGLE + 1], [0.0] * 6)

    def testFutureWarning(self):
        'load a file from the future and check we complain'
        file = '''gnofract4d parameter file
version=99.9
'''
        warning_catcher = WarningCatcher()
        f = fractal.T(Test.g_comp)
        future_file = io.StringIO(file)

        f.warn = warning_catcher.warn
        f.loadFctFile(future_file)
        self.assertEqual(len(warning_catcher.warnings), 1)
        self.assertEqual(warning_catcher.warnings[0],
                         '''This file was created by a newer version of Gnofract 4D.
The image may not display correctly. Please upgrade to version 99.9 or higher.''')

    def testNoPeriodIfNoZ(self):
        'if z isn\'t used in the fractal, disable periodicity'
        f = fractal.T(Test.g_comp)

        f.set_formula("test.frm", "test_noz")
        f.compile()  # previously, failed to compile

    def testEpsilonTolerance(self):
        f = fractal.T(Test.g_comp)
        self.assertNearlyEqual(
            [(4.0 / 640.0) * 0.05], [f.epsilon_tolerance(640, 480)])

        self.assertNearlyEqual(
            [(4.0 / 640.0) * 0.05], [f.epsilon_tolerance(480, 640)])

    def testWarpParameter(self):
        # test using a specific parameter for warping
        f = fractal.T(Test.g_comp)
        self.assertEqual(f.warp_param, None)
        f.set_formula("test.frm", "test_warp_param")
        f.compile()
        f.reset()

        im = image.T(40, 30)
        f.draw(im)
        # im.save("no_warp.png") # should be completely white

        # now set the parameter to be warped
        f.set_warp_param("@p1")

        # check we call it warped in the parameter file
        s = f.serialize()
        self.assertNotEqual(-1, s.find("@p1=warp"))

        f.draw(im)
        # im.save("yes_warp.png") # should look like a circle

    def round_trip(self, f):
        f2 = fractal.T(Test.g_comp)
        s = f.serialize()
        f2.deserialize(s)
        return f2

    def testCircle(self):
        f = fractal.T(Test.g_comp)

        f.set_formula("test.frm", "test_circle")
        f.set_outer("gf4d.cfrm", "continuous_potential")
        f.compile()
        f.reset()
        self.assertEqual(f.forms[0].params, [f.get_gradient(), 4.0])
        (w, h) = (40, 30)
        im = image.T(w, h)
        f.draw(im)

        im.save(os.path.join(Test.tmpdir.name, "foo.tga"))
        # check that result is horizontally symmetrical
        buf = im.image_buffer(0, 0)
        for y in range(h):
            line = buf[y * w * 3:(y * w + w) * 3].tolist()
            line.reverse()
            revline = line
            line = buf[y * w * 3:(y * w + w) * 3].tolist()
            for x in range(w):
                a = line[x * 3:(x + 1) * 3]
                b = revline[x * 3:(x + 1) * 3]

                if a != b:
                    fate_buf = im.fate_buffer(0, y)
                    print(fate_buf[0:w].tolist())
                    self.assertEqual(
                        a, b, "%s != %s, %d != %d" %
                        (a, b, x, w - x))

        # and vertically symmetrical
        for x in range(w):
            for y in range(h // 2):
                apos = (y * w + x) * 3
                bpos = ((h - y - 1) * w + x) * 3
                a = buf[apos:apos + 3]
                b = buf[bpos:bpos + 3]
                self.assertEqual(a, b)

    def testDiagonal(self):
        f = fractal.T(Test.g_comp)
        f.set_formula("test.frm", "test_simpleshape")
        f.set_outer("gf4d.cfrm", "default")
        f.compile()
        f.reset()
        self.assertEqual(f.forms[0].params, [f.get_gradient(), 0.0])
        self.assertEqual(f.antialias, 1)
        (w, h) = (30, 30)
        im = image.T(w, h)
        f.draw(im)

        buf = im.image_buffer(0, 0)
        for y in range(h):
            for x in range(w):
                if x > y:
                    self.assertWhite(buf, x, y, w)
                elif y > x:
                    self.assertBlack(buf, x, y, w)
                else:
                    # pixels on boundary should be antialiased to 25% grey
                    # because 3 subpixels are white and 1 black
                    self.assertColor(buf, x, y, w, (255 * 3) // 4)

    def testRecolor(self):
        f = fractal.T(Test.g_comp)
        f.set_formula("test.frm", "test_simpleshape")
        f.set_outer("gf4d.cfrm", "default")
        f.compile()
        f.reset()
        self.assertEqual(f.forms[0].params, [f.get_gradient(), 0.0])
        self.assertEqual(f.antialias, 1)
        (w, h) = (30, 30)
        im = image.T(w, h)
        f.draw(im)

        buf = im.image_buffer(0, 0)
        for y in range(h):
            for x in range(w):
                if x > y:
                    self.assertWhite(buf, x, y, w)
                elif y > x:
                    self.assertBlack(buf, x, y, w)
                else:
                    # pixels on boundary should be antialiased to 25% grey
                    # because 3 subpixels are white and 1 black
                    self.assertColor(buf, x, y, w, (255 * 3) // 4)

    def testDiagonalWithColorFuncs(self):
        f = fractal.T(Test.g_comp)
        f.set_formula("test.frm", "test_simpleshape")
        f.set_inner("test.cfrm", "flat")
        f.set_outer("test.cfrm", "flat")
        f.forms[1].set_named_item("@val", 0.7)
        f.forms[1].set_named_item("@myfunc", "sqrt")

        f.forms[2].set_named_item("@val", 0.2)
        f.forms[2].set_named_item("@myfunc", "sin")

        outgrey = int(math.sqrt(0.7) * 255)
        ingrey = int(math.sin(0.2) * 255)

        self.check_diagonal_image(f, ingrey, outgrey)

        # check all this stuff survives serialization
        saved = f.serialize()

        f2 = fractal.T(Test.g_comp)
        f2.loadFctFile(io.StringIO(saved))
        self.check_diagonal_image(f2, ingrey, outgrey)

    def check_diagonal_image(self, f, ingrey, outgrey):
        f.get_gradient().load_list(
            [(0.0, 0, 0, 0, 255), (1.0, 255, 255, 255, 255)])
        f.compile()
        (w, h) = (30, 30)
        im = image.T(w, h)
        f.antialias = False
        f.draw(im)

        buf = im.image_buffer(0, 0)

        for y in range(h):
            for x in range(w):
                if x >= y:
                    self.assertColor(buf, x, y, w, outgrey)
                else:
                    self.assertColor(buf, x, y, w, ingrey)

    def testCubicRead(self):
        file = '''gnofract4d parameter file
version=1.7
bailout=4
x=0.2488828125
y=-1.3533515625
z=0
w=0
size=0.3365625
xy=0
xz=0
xw=0
yz=0
yw=0
zw=0
maxiter=256
antialias=1
bailfunc=0
inner=2
outer=1
[function]
function=Cubic Mandelbrot
a=(0.34,-0.28)
[endsection]
[colorizer]=0
colorizer=1
colordata=00000044286c441c704c4c7850788058a8885cd48c58cc8858c48858bc8854b48854ac8854a88850a0885098885090845088844c80844c7c844c7484486c84486484485c8448548044508044488044408040388040308040288040248044287c482c7c4c307c4c347c50387c543c7c54407c58447c5c487c60487c604c7c64507c68547c68587c6c5c7c70607c70647c74687c78687c7c6c7c7c707c80747c84787c847c788880788c84788c8878908c78948c789890789894789c9878a09c78a0a078a4a478a8a878a8ac78acac78b0b078b4b478b4b878b8bc78bcc078bcc478c0c878c4cc78c4cc78c0c06cc0b464bcac58bca050b89444b88c3cb48030b47428b06c1cb06014b0580cac5414a85018a4501ca04c249c4c289c482c98483494443890443c8c4044884048883c4c843c548038587c385c78346474346874306c7030706c2c78682c7c64288060288860248c5c249058209854209c501ca04c1ca84c18ac4818b04414b84014bc3c10c03c10c44014bc4018b4401cac4420a444249c442894442c8c48308448347c483874483c6c4c40644c445c4c48544c4c4c50504450543c505834505c2c54602454641c546814546c105068184c6420486028485c3044583840543c3c50443c4c4c38485434445c304060303c682c3870283478243080242c8434387c4044744c4c6c585868646060746c588078548c804c988c44a494409c8c4894884c908050887c5484785c7c7060746c64706468686070645c745c547854507c50488448448844408c3c3890343498302c9c2828a02424a41c1cac1418b01010b4080cb80408bc0810b80c14b81018b8141cb81820b41c24b42028b4242cb42830b02c34b0303cb03440b03844ac3c48ac3c4cac4050ac4454ac4858a84c5ca85060a85468a8586ca45c70a46074a46478a4687ca06c80a07084a07488a0748ca07080a86c74ac6868b4645cb86050c05c44c45838cc582cd05428cc5024cc4c24c84c20c8481cc4441cc44018c04018c058e81c58d82454c82854b83050a83850983c5088444c784c4c6850485858484860443864
[endsection]
[colorizer]=1
colorizer=0
red=0.377255787461439
green=1
blue=0.5543108971162746
[endsection]
'''
        f = fractal.T(Test.g_comp)
        wc = WarningCatcher()
        f.warn = wc.warn
        f.loadFctFile(io.StringIO(file))

        self.assertEqual(
            [f.get_gradient(), 4.0, 0.34, -0.28],
            f.forms[0].params)

    def testNewGradientRead(self):
        file = '''gnofract4d parameter file
version=2.8
x=0.00000000000000000
y=0.00000000000000000
z=0.00000000000000000
w=0.00000000000000000
size=4.00000000000000000
xy=0.00000000000000000
xz=0.00000000000000000
xw=0.00000000000000000
yz=0.00000000000000000
yw=0.00000000000000000
zw=0.00000000000000000
maxiter=256
yflip=0
periodicity=1
[function]
formulafile=gf4d.frm
function=Mandelbrot
@bailfunc=cmag
@_gradient=[
GIMP Gradient
Name: /usr/share/gimp/1.2/gradients/Abstract_3
6
0.000000 0.050083 0.435726 0.000000 0.424242 0.070751 1.000000 1.000000 0.725647 0.428066 1.000000 0 0
0.435726 0.490818 0.590985 1.000000 0.725647 0.428066 1.000000 0.115248 0.249315 0.651515 1.000000 0 0
0.590985 0.660267 0.799666 0.115248 0.249315 0.651515 1.000000 0.552948 0.624658 0.550758 1.000000 0 0
0.799666 0.879800 0.943239 0.552948 0.624658 0.550758 1.000000 0.990647 1.000000 0.450000 1.000000 0 0
0.943239 0.961603 0.979967 0.990647 1.000000 0.450000 1.000000 0.317635 0.843781 1.000000 1.000000 0 0
0.979967 0.989983 1.000000 0.317635 0.843781 1.000000 1.000000 0.000000 1.000000 0.000000 1.000000 0 0
]
@bailout=4.00000000000000000
[endsection]
[inner]
formulafile=gf4d.cfrm
function=zero
@_transfer=ident
@_density=1.00000000000000000
@_offset=0.00000000000000000
[endsection]
[outer]
formulafile=gf4d.cfrm
function=continuous_potential
@_transfer=ident
@_density=1.00000000000000000
@_offset=0.00000000000000000
@bailout=4.00000000000000000
[endsection]
[colors]
colorizer=1
solids=[
000000ff
000000ff
]
'''
        f = fractal.T(Test.g_comp)
        f.loadFctFile(io.StringIO(file))

        g = f.get_gradient()
        self.assertEqual(len(g.segments), 6)

    def failBuf(self, buf):
        self.assertTrue(False)

    def assertWhite(self, buf, x, y, w):
        self.assertColor(buf, x, y, w, 255)

    def assertBlack(self, buf, x, y, w):
        self.assertColor(buf, x, y, w, 0)

    def assertColor(self, buf, x, y, w, c):
        off = (x + y * w) * 3
        r = buf[off]
        g = buf[off + 1]
        b = buf[off + 2]
        self.assertEqual(r, c)
        self.assertEqual(g, c)
        self.assertEqual(b, c)

    def testTransforms(self):
        f = fractal.T(Test.g_comp)
        self.assertEqual([], f.transforms)
        f.append_transform("gf4d.uxf", "Inverse")
        self.assertEqual(1, len(f.transforms))
        t = f.transforms[0]
        self.assertTrue(isinstance(t, formsettings.T))
        self.assertEqual("Inverse", t.funcName)

        f.remove_transform(0)
        self.assertEqual(0, len(f.transforms))

    def testSet(self):
        f = fractal.T(Test.g_comp)
        f.set_formula("gf4d.frm", "Mandelbar")
        f.set_inner("gf4d.cfrm", "zero")
        f.set_outer("gf4d.cfrm", "default")

        self.assertEqual(f.forms[1].funcFile, "gf4d.cfrm")
        self.assertEqual(f.forms[2].funcFile, "gf4d.cfrm")
        self.assertEqual(f.forms[1].funcName, "default")
        self.assertEqual(f.forms[2].funcName, "zero")

        f.compile()
        im = image.T(4, 3)
        f.draw(im)

    def testFct(self):
        file = open("testdata/test.fct")
        f = fractal.T(Test.g_comp)
        f.loadFctFile(file)
        file.close()
        f.compile()
        im = image.T(64, 48)
        f.draw(im)

    def testCopy(self):
        f = fractal.T(Test.g_comp)
        f.set_formula("gf4d.frm", "Barnsley Type 1")
        f.forms[0].set_named_item("@bailfunc", "manhattanish")
        f.set_outer("test.cfrm", "flat")
        f.forms[1].set_named_item("@ep", 2)
        f.forms[1].set_named_item("@i", 789)
        f.forms[1].set_named_item("@_transfer", "sqrt")
        f.set_warp_param(2)
        f.periodicity = False
        f.period_tolerance = 1.0e-5
        c = copy.copy(f)

        self.assertFractalsEqual(f, c)

        # some tests to ensure data is actually separate

        # test a parameter
        mag = c.get_param(c.MAGNITUDE)
        self.assertEqual(mag, f.get_param(f.MAGNITUDE))

        f.set_param(f.MAGNITUDE, 89.1)
        self.assertEqual(mag, c.get_param(c.MAGNITUDE))
        self.assertNotEqual(mag, f.get_param(f.MAGNITUDE))

        # test formula
        formName = c.forms[0].funcName
        self.assertEqual(formName, f.forms[0].funcName)

        f.set_formula("gf4d.frm", "Mandelbar")
        self.assertEqual(formName, c.forms[0].funcName)
        self.assertNotEqual(formName, f.forms[0].funcName)

        # test colors
        new_colors = c.get_gradient().segments
        old_colors = f.get_gradient().segments

        c0 = new_colors[0].left_color
        for i in range(len(new_colors)):
            self.assertEqual(new_colors[i].left_color,
                             old_colors[i].left_color)

        old_colors[0].left_color = [0.7, 0.3, 0.6, 0.5]
        self.assertEqual(c0, c.get_gradient().segments[0].left_color)
        self.assertNotEqual(c0, f.get_gradient().segments[0].left_color)

    def testCopy2(self):
        '''There was a bug where copy() would reset func values.
        Check for recurrence'''
        f = fractal.T(Test.g_comp)
        file = open("testdata/julfn.fct")
        f.loadFctFile(file)
        file.close()
        f.forms[0].set_named_item("@fn1", "sinh")

        self.assertEqual(f.forms[0].get_func_value("@fn1"), "sinh")

        c = copy.copy(f)

        self.assertEqual(f.forms[0].get_func_value("@fn1"), "sinh")

    def assertDirty(self, f):
        self.assertEqual(f.dirty, True)

    def assertClean(self, f):
        self.assertEqual(f.dirty, False)

    def testDirtyFlag(self):
        f = fractal.T(Test.g_comp)
        self.assertDirty(f)
        f.clean()
        self.assertClean(f)

        # set to existing value
        f.set_param(f.MAGNITUDE, f.params[f.MAGNITUDE])
        self.assertClean(f)

        # set to different value
        f.set_param(f.MAGNITUDE, f.params[f.MAGNITUDE] * 2.0)
        self.assertDirty(f)

        f.clean()
        f.forms[0].set_named_func("@bailfunc", "real2")
        self.assertDirty(f)

    def testLoadGivesCorrectParameters(self):
        f = fractal.T(Test.g_comp)
        self.assertEqual(len(f.forms[0].formula.symbols.parameters()), 3)
        file = open("testdata/elfglow.fct")
        f.loadFctFile(file)
        file.close()
        self.assertEqual(len(f.forms[0].formula.symbols.parameters()), 5)

    def testFractalBadness(self):
        f = fractal.T(Test.g_comp)
        self.assertRaises(ValueError, f.set_formula, "gf4d.frm", "xMandelbrot")
        self.assertRaises(ValueError, f.set_inner, "gf4d.cfrm", "xdefault")
        self.assertRaises(ValueError, f.set_outer, "gf4d.cfrm", "xzero")

        # none of these should have changed the fractal,which should still work
        self.assertEqual(f.forms[0].funcName, "Mandelbrot")
        f.compile()

    def testTumorCrash(self):
        f = fractal.T(Test.g_comp)
        file = open("testdata/tumor.fct")
        f.loadFctFile(file)
        file.close()
        f.compile()
        f.set_formula("gf4d.frm", "Buffalo")
        f.compile()
        im = image.T(40, 30)
        f.draw(im)

    def testBlend(self):
        f = fractal.T(Test.g_comp)
        f2 = fractal.T(Test.g_comp)
        f2.set_param(f.XCENTER, 4.0)
        f2.set_param(f.MAGNITUDE, 1.0)
        f2.forms[0].set_named_item("@bailout", 4000.0)

        blend = f.blend(f2, 0.0)
        self.assertFractalsEqual(blend, f)
        blend = f.blend(f2, 1.0)
        self.assertFractalsEqual(blend, f2)
        blend = f.blend(f2, 0.5)
        self.assertEqual(2.0, blend.get_param(f.XCENTER))  # linear blend
        self.assertTrue(
            2.5 > blend.get_param(
                f.MAGNITUDE))  # exponential blend
        self.assertEqual(
            2002.0, blend.forms[0].get_named_param_value("@bailout"))

    def testBadBlend(self):
        f1 = fractal.T(Test.g_comp)
        f2 = fractal.T(Test.g_comp)
        f2.set_formula("gf4d.frm", "Magnet")
        self.assertRaises(ValueError, f1.blend, f2, 0.5)

    def testBlendAngles(self):
        f = fractal.T(Test.g_comp)
        f2 = fractal.T(Test.g_comp)

        f2.set_param(f.XYANGLE, math.pi / 4.0)
        blend = f.blend(f2, 0.5)

        self.assertEqual(math.pi / 8.0, blend.get_param(f.XYANGLE))

    def testSetCompilerOptions(self):
        f = fractal.T(Test.g_comp)
        f.set_compiler_option("optimize", 1)
        self.assertEqual({"optimize": 1}, f.compiler_options)

    def testImage(self):
        f = fractal.T(Test.g_comp)
        f.set_formula("test.frm", "ident")
        f.set_inner("test.cfrm", "image")
        f.compile()
        im = image.T(30, 30)
        f.draw(im)
        im.save(os.path.join(Test.tmpdir.name, "foo.tga"))

    def disabled_testPeriodColorfunc(self):
        f = fractal.T(Test.g_comp)
        f.set_inner("gf4d.cfrm", "Periodicity")
        f.compile()
        im = image.T(30, 30)
        f.draw(im)

    def testDetermineDirection(self):
        f = fractal.T(Test.g_comp)

        self.tryDirections(
            f, fractal.BLEND_NEAREST, [
                True, False, True, False, True])
        self.tryDirections(
            f, fractal.BLEND_FURTHEST, [
                False, True, False, True, False])
        self.tryDirections(f, fractal.BLEND_CW, [True] * 5)
        self.tryDirections(f, fractal.BLEND_CCW, [False] * 5)

        self.assertRaises(ValueError, f.determine_direction, 0, math.pi, 77)

    def tryDirections(self, f, mode, expected):
        self.assertEqual(
            expected[0],
            f.determine_direction(0, math.pi / 2.0, mode))

        self.assertEqual(
            expected[1],
            f.determine_direction(0, -math.pi / 2.0, mode))

        self.assertEqual(
            expected[2],
            f.determine_direction(0, math.pi, mode))

        self.assertEqual(
            expected[3],
            f.determine_direction(0, math.pi * 1.5, mode))

        self.assertEqual(
            expected[4],
            f.determine_direction(0, -math.pi * 1.5, mode))

    def assertValidType(self, val):
        self.assertNotEqual(
            val.__class__,
            list,
            "%s shouldn't be a list" %
            val)

    def testMandelbrotMix4(self):
        # regression test
        f = fractal.T(Test.g_comp)
        f.set_formula("test.frm", "MandelbrotMix4")
        s = f.forms[0].formula.symbols
        # print s["t__a_p1"]
        # print s["t__a_p99"]

        for val in f.forms[0].formula.symbols.default_params():
            self.assertValidType(val)

    def testJm25(self):
        # regression test for a problem accidentally introduced in
        # private builds of gf4d 3.5
        f = fractal.T(Test.g_comp)
        f.set_formula("fractint-g4.frm", "Jm_25")
        f.compile()
        self.assertEqual(len(f.forms[0].params), len(f.forms[0].paramtypes))
