3igen/make_test.py at main · dirkmunro89/3igen · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import json
import numpy as np

out = {}

ex = 50;
ey = 50;
ez = 50;
e = ex*ey*ez
gss = 2;
mat = [1.0, .3, "plane_stress"]

lx=1.
ly=1.
lz=1.

nx = ex+1;
ny = ey+1;
nz = ez+1;
n = nx*ny*nz;

nds = []
els = []
bds = []
lds = []

n=0
for k in range(nz):
    for j in range(ny):
        for i in range(nx):
            nds.append([float(i)*lx/ex,float(j)*ly/ey,float(k)*lz/ez] )
            for d in range(3):
                if k == 0:
                    bds.append([n,0.])
                if k == nz-2 and j == ny-2 and d == 1:
                    if i == 1:
                        lds.append([n,-1.])
                    elif i == nx-2:
                        lds.append([n,1.])
                n=n+1

n1=0
n2=1
n3=nx+1
n4=nx
n5=nx*ny
n6=nx*ny+1
n7=nx*ny+nx+1
n8=nx*ny+nx
for k in range (ez):
    for j in range(ey):
        for i in range(ex):
            els.append([n1,n2,n3,n4,n5,n6,n7,n8])
            n1=n1+1; n2=n2+1; n3=n3+1; n4=n4+1
            n5=n5+1; n6=n6+1; n7=n7+1; n8=n8+1
        n1=n1+1; n2=n2+1; n3=n3+1; n4=n4+1
        n5=n5+1; n6=n6+1; n7=n7+1; n8=n8+1
    n1=n1+nx; n2=n2+nx; n3=n3+nx; n4=n4+nx
    n5=n5+nx; n6=n6+nx; n7=n7+nx; n8=n8+nx

out["nodes"] = nds
out["elements"] = els
out["gauss"] = gss
out["material"] = mat
out["boundary"] = bds
out["load"] = lds

json_object = json.dumps(out, indent=4)

with open("sample.json", "w") as outfile:
    outfile.write(json_object)