def makeWorkerAndFunc(self, image, cmap):
siteobj = FractalSite()
site = fract4dc.site_create(siteobj)
file = self.compileColorDiagonal()
handle = fract4dc.pf_load(file)
pfunc = fract4dc.pf_create(handle)
fract4dc.pf_init(pfunc,pos_params, self.color_diagonal_params)
fw = fract4dc.fw_create(1,pfunc,cmap,image,site)
ff = fract4dc.ff_create(
[0.0, 0.0, 0.0, 0.0,
4.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
2,
100,
0,
1,
pfunc,
cmap,
0,
1,
0,
image,
site,
fw,
False,
1.0E-9)
return (fw,ff,site,handle,pfunc)
def setUp(self):
global g_comp
self.compiler = g_comp
self.f = fractal.T(self.compiler)
self.f.render_type = 2
self.f.set_formula("test.frm", "test_hypersphere")
self.f.compile()
handle = fract4dc.pf_load(self.f.outputfile)
self.pfunc = fract4dc.pf_create(handle)
self.cmap = fract4dc.cmap_create_gradient(self.f.get_gradient().segments)
(r,g,b,a) = self.f.solids[0]
fract4dc.cmap_set_solid(self.cmap,0,r,g,b,a)
(r,g,b,a) = self.f.solids[1]
fract4dc.cmap_set_solid(self.cmap,1,r,g,b,a)
initparams = self.f.all_params()
fract4dc.pf_init(self.pfunc,self.f.params,initparams)
self.im = image.T(40,30)
siteobj = FractalSite()
self.fw = fract4dc.fw_create(
1,self.pfunc,self.cmap,self.im._img,self.f.site)
self.ff = fract4dc.ff_create(
[0.0, 0.0, 0.0, 0.0,
4.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
2,
100,
0,
1,
self.pfunc,
self.cmap,
0,
1,
2, # 3D
self.im._img,
self.f.site,
self.fw,
False,
1.0E-9)
def setUp(self):
global g_comp
self.compiler = g_comp
self.f = fractal.T(self.compiler)
self.f.render_type = 2
self.f.set_formula("test.frm", "test_hypersphere")
self.f.compile()
handle = fract4dc.pf_load(self.f.outputfile)
self.pfunc = fract4dc.pf_create(handle)
self.cmap = fract4dc.cmap_create_gradient(
self.f.get_gradient().segments)
(r, g, b, a) = self.f.solids[0]
fract4dc.cmap_set_solid(self.cmap, 0, r, g, b, a)
(r, g, b, a) = self.f.solids[1]
fract4dc.cmap_set_solid(self.cmap, 1, r, g, b, a)
initparams = self.f.all_params()
fract4dc.pf_init(self.pfunc, self.f.params, initparams)
self.im = image.T(40, 30)
siteobj = FractalSite()
self.fw = fract4dc.fw_create(1, self.pfunc, self.cmap, self.im._img,
self.f.site)
self.ff = fract4dc.ff_create(
[0.0, 0.0, 0.0, 0.0, 4.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
2,
100,
0,
1,
self.pfunc,
self.cmap,
0,
1,
2, # 3D
self.im._img,
self.f.site,
self.fw,
False,
1.0E-9)
def testVectors(self):
siteobj = FractalSite()
site = fract4dc.site_create(siteobj)
file = self.compileColorDiagonal()
handle = fract4dc.pf_load(file)
pfunc = fract4dc.pf_create(handle)
fract4dc.pf_init(pfunc,pos_params,self.color_diagonal_params)
(w,h,tw,th) = (40,20,40,20)
im = image.T(w,h)
cmap = fract4dc.cmap_create([(1.0, 255, 255, 255, 255)])
fw = fract4dc.fw_create(1,pfunc,cmap,im._img,site)
ff = fract4dc.ff_create(
[0.0, 0.0, 0.0, 0.0,
4.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
2,
100,
0,
1,
pfunc,
cmap,
0,
1,
0,
im._img,
site,
fw,
False,
1.0E-9)
# check dx, dy and topleft
dx = fract4dc.ff_get_vector(ff, fract4dc.DELTA_X)
self.assertNearlyEqual(dx, [4.0/tw,0.0,0.0,0.0])
dy = fract4dc.ff_get_vector(ff, fract4dc.DELTA_Y);
self.assertNearlyEqual(dy, [0.0,-2.0/th,0.0,0.0])
topleft = fract4dc.ff_get_vector(ff, fract4dc.TOPLEFT);
self.assertNearlyEqual(topleft, [-2.0 + 4.0/(tw*2),1.0 - 2.0/(th*2),0.0,0.0])
# check they are updated if image is bigger
(w,h,tw,th) = (40,20,400,200)
im = image.T(w,h,tw,th)
fw = fract4dc.fw_create(1,pfunc,cmap,im._img,site)
ff = fract4dc.ff_create(
[0.0, 0.0, 0.0, 0.0,
4.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
2,
100,
0,
1,
pfunc,
cmap,
0,
1,
0,
im._img,
site,
fw,
False,
1.0E-9)
# check dx, dy and topleft
dx = fract4dc.ff_get_vector(ff, fract4dc.DELTA_X)
self.assertNearlyEqual(dx, [4.0/tw,0.0,0.0,0.0])
dy = fract4dc.ff_get_vector(ff, fract4dc.DELTA_Y);
self.assertNearlyEqual(dy, [0.0,-2.0/th,0.0,0.0])
topleft = fract4dc.ff_get_vector(ff, fract4dc.TOPLEFT);
self.assertNearlyEqual(topleft, [-2.0 + 4.0/(tw*2),1.0 - 2.0/(th*2),0.0,0.0])
offx = 40
offy = 10
im.set_offset(offx, offy)
fw = fract4dc.fw_create(1,pfunc,cmap,im._img,site)
ff = fract4dc.ff_create(
[0.0, 0.0, 0.0, 0.0,
4.0,
0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
2,
100,
0,
1,
pfunc,
cmap,
0,
1,
0,
im._img,
site,
fw,
False,
1.0E-9)
# check dx, dy and topleft
dx = fract4dc.ff_get_vector(ff, fract4dc.DELTA_X)
self.assertNearlyEqual(dx, [4.0/tw,0.0,0.0,0.0])
dy = fract4dc.ff_get_vector(ff, fract4dc.DELTA_Y);
self.assertNearlyEqual(dy, [0.0,-2.0/th,0.0,0.0])
topleft = fract4dc.ff_get_vector(ff, fract4dc.TOPLEFT);
self.assertNearlyEqual(topleft, [
-2.0 + dx[0] * (offx + 0.5),
1.0 + dy[1] * (offy + 0.5),
0.0,0.0])
