import numpy        # type: ignore
from gridlock import Grid


if __name__ == '__main__':
    # xyz = [numpy.arange(-5.0, 6.0), numpy.arange(-4.0, 5.0), [-1.0, 1.0]]
    # eg = Grid(xyz)
    # egc = Grid.allocate(0.0)
    # # eg.draw_slab(egc, surface_normal=2, center=0, thickness=10, foreground=2)
    # eg.draw_cylinder(egc, surface_normal=2, center=[0, 0, 0], radius=4,
    #       thickness=10, num_points=1000, foreground=1)
    # eg.visualize_slice(egc, surface_normal=2, center=0, which_shifts=2)

    # xyz2 = [numpy.arange(-5.0, 6.0), [-1.0, 1.0], numpy.arange(-4.0, 5.0)]
    # eg2 = Grid(xyz2)
    # eg2c = Grid.allocate(0.0)
    # # eg2.draw_slab(eg2c, surface_normal=2, center=0, thickness=10, foreground=2)
    # eg2.draw_cylinder(eg2c, surface_normal=1, center=[0, 0, 0],
    #                   radius=4, thickness=10, num_points=1000, foreground=1.0)
    # eg2.visualize_slice(eg2c, surface_normal=1, center=0, which_shifts=1)

    # n = 20
    # m = 3
    # r1 = numpy.fromfunction(lambda x: numpy.sign(x - n) * 2 ** (abs(x - n)/m), (2*n, ))
    # print(r1)
    # xyz3 = [r1, numpy.linspace(-5.5, 5.5, 30), numpy.linspace(-5.5, 5.5, 10)]
    # xyz3 = [numpy.linspace(-5.5, 5.5, 10),
    #         numpy.linspace(-5.5, 5.5, 10),
    #         numpy.linspace(-5.5, 5.5, 10)]

    half_x = [.25, .5, 0.75, 1, 1.25, 1.5, 2, 2.5, 3, 3.5]
    xyz3 = [[-x for x in half_x[::-1]] + [0] + half_x,
            numpy.linspace(-5.5, 5.5, 10),
            numpy.linspace(-5.5, 5.5, 10)]
    eg = Grid(xyz3)
    egc = eg.allocate(0)
    # eg.draw_slab(Direction.z, 0, 10, 2)
    eg.save('/home/jan/Desktop/test.pickle')
    eg.draw_cylinder(egc, surface_normal=2, center=[0, 0, 0], radius=2.0,
                     thickness=10, num_poitns=1000, foreground=1)
    eg.draw_extrude_rectangle(egc, rectangle=[[-2, 1, -1], [0, 1, 1]],
                              direction=1, poalarity=+1, distance=5)
    eg.visualize_slice(egc, surface_normal=2, center=0, which_shifts=2)
    eg.visualize_isosurface(egc, which_shifts=2)