import anon
#anon.conf.config.update('jax_disable_jit', True)
import elle.sections
import emme as em

# column_section = em.aisc.load("w14x159")
# girder_section = em.aisc.load("w33x130")
column_section = {"A": 576.0, "I": 27_648.0}
girder_section = {"A": 684.0, "I": 34_383.8}

# Create a model Assembler
model = em.SkeletalModel(ndm=2,ndf=3)

# Set up nodes
E = 3.6e3
ft = 12
B, H = 30.*ft, 13.*ft
model.node("1",  0.,  0.)
model.node("2",  0.,  H )
model.node("3", B/2,  H )
model.node("4",  B ,  H )
model.node("5",  B ,  0.)

model.beam("a", "1", "2", **column_section,E=E)
model.beam("b", "2", "3", **girder_section,E=E)
model.beam("c", "3", "4", **girder_section,E=E)
model.beam("d", "4", "5", **column_section,E=E)

model.boun("1", [1,1,1]) 
model.boun("5", [1,1,1])

model.load("2", 20.0, dof="x")

em.plot_structure(model);

f = model.compose()

U = f()
U

fig, ax = em.plot_displ(model, U, scale=1e3)

model.delems["a"].ke_matrix()

em.K_matrix(model)