From dac0813cc989330f9a3927ee4cc175b3f97dd361 Mon Sep 17 00:00:00 2001 From: Jan Petykiewicz Date: Sun, 24 Oct 2021 19:06:33 -0700 Subject: [PATCH] add example --- gridlock/examples/ex0.py | 44 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 gridlock/examples/ex0.py diff --git a/gridlock/examples/ex0.py b/gridlock/examples/ex0.py new file mode 100644 index 0000000..0e1cd86 --- /dev/null +++ b/gridlock/examples/ex0.py @@ -0,0 +1,44 @@ +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, eps=2) + # eg.draw_cylinder(egc, surface_normal=2, center=[0, 0, 0], radius=4, + # thickness=10, num_points=1000, eps=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, eps=2) + # eg2.draw_cylinder(eg2c, surface_normal=1, center=[0, 0, 0], + # radius=4, thickness=10, num_points=1000, eps=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, eps=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)