test_3DSandbox_only.py

from imfractal import CSandbox3D
import matplotlib.pyplot as plt

import time
import numpy as np
import math

def checkerboard(w, h, c0, c1, blocksize):
        tile = np.array([[c0,c1],[c1,c0]]).repeat(blocksize, axis=0).repeat(blocksize, axis=1)
        grid = np.tile(tile,(int(math.ceil((h+0.0)/(2*blocksize))),int(math.ceil((w+0.0)/(2*blocksize)))))
        return grid[:h,:w]

def checker_3d(size, width):
    res = np.zeros((size, size, size))

    res[size/2] = checkerboard(size, size, 0, 1, width)

    return res

def do_test():

    # construct checkerboard
    checker = checker_3d(50, 25)

    params = {
        "zero": 1,
        "one": 0.75,
        "two": 3.7,
        "three": 1,
        "four": 15,
        "five": 0,
        "mask_filename": '',
        "seven": "no",
        "eight": 'S',
        "nine": 'M',
        "threshold": 200,
        "total_pixels":6000,
        "adaptive" : False, # adaptive threshold (only for not holder)
        "laplacian": False,
        "gradient" : False
    }

    aux = CSandbox3D(21)
    aux.setDef(40, 1.02, True, params)

    print "Computing Sandbox 3D Multifractal Spectrum... (checkerboard)"
    t =  time.clock()
    fds = aux.getFDs('', checker)
    t =  time.clock()-t
    print "Time 3D MFS: ", t
    print fds

    plt.title('Monofractal')
    plt.ylim((1.0, 3.5))
    plt.plot(fds, 'x', label = 'Mono')
    
    multif = np.load('exps/data/img3d.npy')
    print "Computing Sandbox 3D Multifractal Spectrum... (artificial multifractal)"
    thresh = 70
    t =  time.clock()
    fds = aux.getFDs('', multif > thresh)
    t =  time.clock()-t
    print "Time 3D MFS: ", t
    print fds

    plt.plot(fds,'-', label = 'Multi')
    plt.legend()
    plt.show()