import matplotlib.pyplot as plt
from matproplib.conditions import STPConditions

from bluemira.materials.basic import SS316
from bluemira.structural.crosssection import RectangularBeam
from bluemira.structural.model import FiniteElementModel

plt.close("all")

model = FiniteElementModel()
ss316 = SS316()
op_cond = STPConditions()

Structural Example

def make_platform(width, elevation, cross_section, *, elements=True, internodes=False):
    """
    Make a square platform at a certain elevation.
    """
    if not internodes:
        model.add_node(-width / 2, -width / 2, elevation)
        i = model.geometry.nodes[-1].id_number
        model.add_node(width / 2, -width / 2, elevation)
        model.add_node(width / 2, width / 2, elevation)
        model.add_node(-width / 2, width / 2, elevation)

        if elements:
            for j in range(3):
                model.add_element(i + j, i + j + 1, cross_section, ss316, op_cond)
            model.add_element(i + j + 1, i, cross_section, ss316, op_cond)
    if internodes:
        model.add_node(-width / 2, -width / 2, elevation)
        i = model.geometry.nodes[-1].id_number
        model.add_node(0, -width / 2, elevation)
        model.add_node(width / 2, -width / 2, elevation)
        model.add_node(width / 2, 0, elevation)
        model.add_node(width / 2, width / 2, elevation)
        model.add_node(0, width / 2, elevation)
        model.add_node(-width / 2, width / 2, elevation)
        model.add_node(-width / 2, 0, elevation)
        if elements:
            for j in range(7):
                model.add_element(i + j, i + j + 1, cross_section, ss316, op_cond)
            model.add_element(i + j + 1, i, cross_section, ss316, op_cond)

cs1 = RectangularBeam(5, 5)
cs2 = RectangularBeam(4, 4)
cs3 = RectangularBeam(2, 2)
cs4 = RectangularBeam(1, 1)

make_platform(100, 0, cs1, elements=False)
for i in range(4):
    model.add_support(i, dx=True, dy=True, dz=True, rx=True, ry=True, rz=True)

make_platform(60, 56, cs2, internodes=True)
model.add_element(0, 4, cs1, ss316, op_cond)
model.add_element(0, 5, cs1, ss316, op_cond)
model.add_element(0, 11, cs1, ss316, op_cond)
model.add_element(1, 5, cs1, ss316, op_cond)
model.add_element(1, 6, cs1, ss316, op_cond)
model.add_element(1, 7, cs1, ss316, op_cond)
model.add_element(2, 7, cs1, ss316, op_cond)
model.add_element(2, 8, cs1, ss316, op_cond)
model.add_element(2, 9, cs1, ss316, op_cond)
model.add_element(3, 9, cs1, ss316, op_cond)
model.add_element(3, 10, cs1, ss316, op_cond)
model.add_element(3, 11, cs1, ss316, op_cond)

make_platform(37, 116, cs3)

model.add_element(4, 12, cs2, ss316, op_cond)
model.add_element(5, 12, cs2, ss316, op_cond)
model.add_element(11, 12, cs2, ss316, op_cond)
model.add_element(5, 13, cs2, ss316, op_cond)
model.add_element(6, 13, cs2, ss316, op_cond)
model.add_element(7, 13, cs2, ss316, op_cond)
model.add_element(7, 14, cs2, ss316, op_cond)
model.add_element(8, 14, cs2, ss316, op_cond)
model.add_element(9, 14, cs2, ss316, op_cond)
model.add_element(9, 15, cs2, ss316, op_cond)
model.add_element(10, 15, cs2, ss316, op_cond)
model.add_element(11, 15, cs2, ss316, op_cond)

make_platform(12, 196, cs3)

model.add_element(12, 16, cs3, ss316, op_cond)
model.add_element(13, 17, cs3, ss316, op_cond)
model.add_element(14, 18, cs3, ss316, op_cond)
model.add_element(15, 19, cs3, ss316, op_cond)

make_platform(6, 276, cs3)

model.add_element(16, 20, cs4, ss316, op_cond)
model.add_element(17, 21, cs4, ss316, op_cond)
model.add_element(18, 22, cs4, ss316, op_cond)
model.add_element(19, 23, cs4, ss316, op_cond)

model.add_node(0, 0, 316)

for i in range(4):
    model.add_element(20 + i, 24, cs4, ss316, op_cond)

model.add_node(0, 0, 324)
model.add_element(24, 25, cs4, ss316, op_cond)

model.plot(show_cross_sections=True)

model.add_gravity_loads()
model.add_node_load(24, 1e6, "Fy")

result = model.solve()

result.plot(stress=True)

plt.show()