def test_two_d_mesh_w_1_profile():
rho = 1.8
sl1 = sm.Soil(g_mod=50, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sl2 = sm.Soil(g_mod=100, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sl3 = sm.Soil(g_mod=400, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sp = sm.SoilProfile()
sp.add_layer(0, sl1)
sp.add_layer(5, sl2)
sp.add_layer(12, sl3)
sp.x_angles = [0.0, 0.0, 0.0]
sp.height = 18
sp.x = 0
tds = sm.TwoDSystem(4, 15)
tds.add_sp(sp, x=0)
tds.x_surf = np.array([0])
tds.y_surf = np.array([0])
x_scale_pos = np.array([0])
x_scale_vals = np.array([2.])
fc = sm.num.mesh.FiniteElementOrth2DMeshConstructor(
tds, 0.3, x_scale_pos=x_scale_pos, x_scale_vals=x_scale_vals)
femesh = fc.femesh
ind = np.argmin(abs(femesh.y_nodes + 5.0))
assert np.isclose(-5.0, femesh.y_nodes[ind])
ind = np.argmin(abs(femesh.y_nodes + 12.0))
assert np.isclose(-12.0, femesh.y_nodes[ind])
ind = np.argmin(abs(femesh.y_nodes + 15.0))
assert np.isclose(-15.0, femesh.y_nodes[ind])
def run():
rho = 1.8
sl1 = sm.Soil(g_mod=50, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sl2 = sm.Soil(g_mod=100, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sl3 = sm.Soil(g_mod=400, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sl4 = sm.Soil(g_mod=600, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sp = sm.SoilProfile()
sp.add_layer(0, sl1)
sp.add_layer(5, sl2)
sp.add_layer(12, sl3)
sp.height = 18
sp.x = 0
sp.x_angles = [0.0, 0.0, 0.0]
fd = sm.RaftFoundation()
fd.width = 4
fd.height = 1.0
fd.depth = 0.5
fd.length = 100
tds = sm.TwoDSystem(width=10, height=15)
tds.add_sp(sp, x=0)
tds.x_surf = np.array([0, 40])
tds.y_surf = np.array([0, 0])
bd = sm.NullBuilding()
bd.set_foundation(fd, x=0)
tds.add_bd(bd, x=5)
x_scale_pos = np.array([0, 5, 15, 30])
x_scale_vals = np.array([2., 1.0, 2.0, 3.0])
femesh = sm.num.mesh.FiniteElement2DMesh(tds,
0.3,
x_scale_pos=x_scale_pos,
x_scale_vals=x_scale_vals)
import sys
import pyqtgraph as pg
from pyqtgraph.Qt import QtGui, QtCore
import o3plot
win = pg.plot()
win.setWindowTitle('ECP definition')
win.setXRange(0, femesh.tds.width)
win.setYRange(-femesh.tds.height, max(femesh.tds.y_surf))
o3plot.plot_finite_element_mesh(win, femesh)
# o3_plot.plot_two_d_system(win, tds)
if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
QtGui.QApplication.instance().exec_()
def test_save_and_load_2d_system():
vs = 250.0
sl = sm.Soil()
unit_mass = 1700.0
sl.g_mod = vs**2 * unit_mass
sl.poissons_ratio = 0.0
# sl.cohesion = sl.g_mod / 1000
sl.cohesion = 30.0e3
sl.id = 1
sl.phi = 0.0
sl.unit_dry_weight = unit_mass * 9.8
sl.specific_gravity = 2.65
sl.xi = 0.03 # for linear analysis
sl.sra_type = 'hyperbolic'
sl.inputs += ['sra_type', 'xi']
sp = sm.SoilProfile()
sp.add_layer(0, sl)
sp.height = 1.0e3
sp.id = 1
sp.x_angles = [0.0, 0.0]
fd = sm.RaftFoundation()
fd.width = 2
fd.depth = 0.8
fd.height = 1
fd.length = 100
fd.ip_axis = 'width'
bd = sm.SDOFBuilding()
bd.h_eff = 8.
bd.set_foundation(fd, x=0.0)
sys_width = 6 * fd.width
tds = sm.TwoDSystem(width=sys_width, height=10)
tds.add_sp(sp, 0)
tds.add_bd(bd, sys_width / 2)
tds.id = 1
ecp_out = sm.Output()
ecp_out.add_to_dict(tds)
# ecp_out.add_to_dict(sp)
p_str = json.dumps(ecp_out.to_dict(), skipkeys=["__repr__"], indent=4)
objs = sm.loads_json(p_str)
building = objs["building"][1]
system = objs["system"][1]
assert np.isclose(building.h_eff, 8)
assert np.isclose(building.fd.height, 1)
assert isinstance(system, sm.TwoDSystem)
assert system.sps[0].layer(1).xi == 0.03
sl5 = sm.Soil(g_mod=g_mod, unit_dry_weight=rho * 9.8, poissons_ratio=0.36) sp = sm.SoilProfile() h_face = 1.8 sp.add_layer(0, sl1) sp.add_layer(3.4, sl2) sp.add_layer(5.7, sl3) sp2 = sm.SoilProfile() sp2.add_layer(0, sl4) sp2.add_layer(3.9, sl2) sp2.add_layer(6.5, sl5) sp2.height = 20 sp.x_angles = [None, 0.07, 0.0] sp2.x_angles = [None, 0.00, 0.0] tds = sm.TwoDSystem(width=45, height=7.5) tds.add_sp(sp, x=0) tds.add_sp(sp2, x=17) tds.x_surf = np.array([0, 10, 12, 15, 17, 35, tds.width]) tds.y_surf = np.array([h_face, h_face, 0, 0, h_face, h_face+0.6, h_face + 0.6]) # tds.x_surf = np.array([0, 20, 21, tds.width]) # tds.y_surf = np.array([h_face-0.9, h_face, 0, 0.]) x_scale_pos = np.array([0, 5, 30]) x_scale_vals = np.array([2., 1.5, 2]) show_set_init_y_blocks = 0 show_ecp_definition = 0 show_get_special_coords_and_slopes = 0 show_adjust_blocks_to_be_consistent_with_slopes = 0 show_trim_grid_to_target_dh = 0
def test_two_d_mesh():
rho = 1.8
sl1 = sm.Soil(g_mod=50, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sl2 = sm.Soil(g_mod=100, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sl3 = sm.Soil(g_mod=400, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sl4 = sm.Soil(g_mod=600, unit_dry_weight=rho * 9.8, poissons_ratio=0.3)
sp = sm.SoilProfile()
sp.add_layer(0, sl1)
sp.add_layer(5, sl2)
sp.add_layer(12, sl3)
sp.height = 18
sp.x = 0
sp2 = sm.SoilProfile()
sp2.add_layer(0, sl1)
sp2.add_layer(7, sl4)
sp2.add_layer(12, sl3)
sp2.height = 20
sp.x_angles = [0.0, 0.05, 0.0]
sp2.x_angles = [0.0, 0.00, 0.0]
fd = sm.RaftFoundation()
fd.width = 2
fd.depth = 0
fd.height = 0
fd.length = 100
fd.ip_axis = 'width'
tds = sm.TwoDSystem(40, 15)
tds.add_sp(sp, x=0)
tds.add_sp(sp2, x=14)
tds.x_surf = np.array([0, 10, 12, 40])
tds.y_surf = np.array([0, 0, 2, 2])
bd = sm.NullBuilding()
bd.set_foundation(fd, x=0.0)
bd_x = 8.
tds.add_bd(bd, x=bd_x)
x_scale_pos = np.array([0, 5, 15, 30])
x_scale_vals = np.array([2., 1.0, 2.0, 3.0])
fc = sm.num.mesh.FiniteElementOrth2DMeshConstructor(
tds, 0.3, x_scale_pos=x_scale_pos, x_scale_vals=x_scale_vals)
femesh = fc.femesh
x_ind = femesh.get_indexes_at_xs([4.])[0]
y_ind = femesh.get_indexes_at_depths([1.99])[0]
sl_ind = femesh.soil_grid[x_ind][y_ind]
assert sl_ind == 1000000
assert femesh.get_active_nodes()[x_ind][y_ind] == 0
p_ind = fc.x_index_to_sp_index[x_ind]
assert p_ind == 0
y_ind = femesh.get_indexes_at_depths([-3])[0]
sl_ind = femesh.soil_grid[x_ind][y_ind]
assert femesh.soils[sl_ind].g_mod == 50
y_ind = femesh.get_indexes_at_depths([-8])[0]
sl_ind = femesh.soil_grid[x_ind][y_ind]
assert femesh.soils[sl_ind].g_mod == 100
y_ind = femesh.get_indexes_at_depths([-12.5])[0]
sl_ind = femesh.soil_grid[x_ind][y_ind]
assert femesh.soils[sl_ind].g_mod == 400
assert femesh.get_active_nodes()[x_ind][y_ind] == 1
x_ind = femesh.get_indexes_at_xs([16.])[0]
p_ind = fc.x_index_to_sp_index[x_ind]
assert p_ind == 1
y_ind = femesh.get_indexes_at_depths([1.99])[0]
sl_ind = femesh.soil_grid[x_ind][y_ind]
assert femesh.soils[sl_ind].g_mod == 50
y_ind = femesh.get_indexes_at_depths([-3])[0]
sl_ind = femesh.soil_grid[x_ind][y_ind]
assert femesh.soils[sl_ind].g_mod == 50
y_ind = femesh.get_indexes_at_depths([-8])[0]
sl_ind = femesh.soil_grid[x_ind][y_ind]
assert femesh.soils[sl_ind].g_mod == 600
y_ind = femesh.get_indexes_at_depths([-12.5])[0]
sl_ind = femesh.soil_grid[x_ind][y_ind]
assert femesh.soils[sl_ind].g_mod == 400
# check foundation
lhs_ind = np.argmin(abs(femesh.x_nodes - (bd_x - fd.width / 2)))
assert np.isclose(bd_x - fd.width / 2,
femesh.x_nodes[lhs_ind]), (bd_x - fd.width / 2,
femesh.x_nodes[lhs_ind])
rhs_ind = np.argmin(abs(femesh.x_nodes - (bd_x + fd.width / 2)))
assert np.isclose(bd_x + fd.width / 2, femesh.x_nodes[rhs_ind])
def test_mesh_vary_y():
vs = 150.0
rho = 1.8
g_mod = vs**2 * rho
sl0 = sm.Soil(g_mod=g_mod, unit_dry_weight=rho * 9.8, poissons_ratio=0.31)
sl1 = sm.Soil(g_mod=g_mod, unit_dry_weight=rho * 9.8, poissons_ratio=0.32)
sl2 = sm.Soil(g_mod=g_mod, unit_dry_weight=rho * 9.8, poissons_ratio=0.33)
sl3 = sm.Soil(g_mod=g_mod, unit_dry_weight=rho * 9.8, poissons_ratio=0.34)
sl4 = sm.Soil(g_mod=g_mod, unit_dry_weight=rho * 9.8, poissons_ratio=0.35)
sl5 = sm.Soil(g_mod=g_mod, unit_dry_weight=rho * 9.8, poissons_ratio=0.36)
sp = sm.SoilProfile()
sp.add_layer(0, sl1)
sp.add_layer(5, sl2)
sp.add_layer(12, sl3)
sp.height = 18
sp.x = 0
sp2 = sm.SoilProfile()
sp2.add_layer(0, sl4)
sp2.add_layer(7, sl5)
sp2.add_layer(12, sl0)
sp2.height = 20
sp.x_angles = [None, 0.01, 0.05]
sp2.x_angles = [None, 0.00, 0.0]
fd = sm.RaftFoundation()
fd.width = 2
fd.depth = 0
fd.ip_axis = 'width'
fd.height = 1
fd.length = 100
tds = sm.TwoDSystem(width=40, height=15)
tds.add_sp(sp, x=0)
tds.add_sp(sp2, x=14)
tds.x_surf = np.array([0, 10, 12, 25, 40])
tds.y_surf = np.array([0, 0, 2, 2.5, 2])
bd = sm.NullBuilding()
bd.set_foundation(fd, x=0.0)
tds.add_bd(bd, x=8)
x_scale_pos = np.array([0, 5, 15, 30])
x_scale_vals = np.array([2., 1.0, 2.0, 3.0])
fc = mesh2d_vary_y.FiniteElementVary2DMeshConstructor(
tds, 0.3, x_scale_pos=x_scale_pos, x_scale_vals=x_scale_vals)
show = 0
femesh = fc.femesh
if show:
import sys
import pyqtgraph as pg
from pyqtgraph.Qt import QtGui, QtCore
import o3plot
win = pg.plot()
win.setWindowTitle('ECP definition')
win.setXRange(0, tds.width)
win.setYRange(-tds.height, max(tds.y_surf))
o3plot.plot_finite_element_mesh_onto_win(win, femesh)
xcs = list(fc.yd)
xcs.sort()
xcs = np.array(xcs)
for i in range(len(xcs)):
win.addItem(pg.InfiniteLine(xcs[i], angle=90,
pen=(0, 255, 0, 100)))
o3plot.plot_two_d_system(tds, win)
for i, sd in enumerate(fc.sds):
x0 = sd[0][0]
x1 = sd[0][1]
y0 = sd[1][0]
y1 = sd[1][1]
win.plot(sd[0], sd[1], pen='r')
x0_ind = femesh.get_nearest_node_index_at_x(x0)
x1_ind = femesh.get_nearest_node_index_at_x(x1)
y0_ind = femesh.get_nearest_node_index_at_depth(y0, x0)
y1_ind = femesh.get_nearest_node_index_at_depth(y1, x1)
win.plot([femesh.x_nodes[x0_ind], femesh.x_nodes[x1_ind]], [
femesh.y_nodes[x0_ind][y0_ind], femesh.y_nodes[x1_ind][y1_ind]
],
pen='b')
if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
QtGui.QApplication.instance().exec_()
dhs = np.diff(femesh.y_nodes, axis=1)
assert np.min(-dhs) > 0.3 * 0.5, np.min(-dhs)
assert np.max(-dhs) < 0.3 * 2, np.min(-dhs)
sds = fc.sds
for sd in sds:
x0 = sd[0][0]
x1 = sd[0][1]
y0 = sd[1][0]
y1 = sd[1][1]
y0_ind = femesh.get_nearest_node_index_at_depth(y0, x0)
y1_ind = femesh.get_nearest_node_index_at_depth(y1, x1)
if y0 == 0 and y1 == 2:
assert y0_ind - 5 == y1_ind, (sd, y0_ind - 5, y1_ind)
else:
assert y0_ind == y1_ind, (sd, y0_ind, y1_ind)