from StructuresExplained.solutions.structure.reactions.assembler import Assembler
if __name__ == "__main__":
from anastruct import SystemElements
from sympy import sympify
ss = SystemElements()
ss.add_element([[0, 0], [1, 0]])
ss.add_element([[1, 0], [1, 1]])
ss.add_element([[1, 0], [2, 0]])
ss.add_element([[2, 0], [3, 0]])
ss.point_load(2, Fy=10)
ss.point_load(3, Fy=-20)
ss.point_load(4, Fy=-30)
ss.point_load(5, Fx=-40)
ss.moment_load(2, Ty=-9)
ss.moment_load(1, 7)
ss.moment_load(3, 3)
ss.q_load(element_id=3, q=(-10, -20))
ss.add_support_roll(4)
ss.add_support_hinged(5)
ss.solve()
ass = Assembler(ss)
ass.assemble_structure()
print(
f"{sympify(ass.res.point_sum_y, evaluate=False)}\n{sympify(ass.res.point_sum_x, evaluate=False)}\n{ass.res.moments_sum}\n")
class connections:
def __init__(self, main_window, main_window_functions):
self.mw = main_window
self.fn = main_window_functions
self.ss = SystemElements()
self.ss.color_scheme = "dark"
self.was_solved = False
self.states = []
def add_beam(self):
try:
self.workaround()
e = self.mw.elementtype.currentIndex()
if self.mw.utilizeinfo.isChecked():
EI = float(self.fn.filter(self.mw.beam_E.text())) * float(
self.fn.filter(self.mw.beam_I.text()))
EA = float(self.fn.filter(self.mw.beam_E.text())) * float(
self.fn.filter(self.mw.beam_A.text()))
element_types = ["beam", "truss"]
self.ss.add_element(location=[
[
float(self.fn.filter(self.mw.beam_x1.text())),
float(self.fn.filter(self.mw.beam_y1.text()))
],
[
float(self.fn.filter(self.mw.beam_x2.text())),
float(self.fn.filter(self.mw.beam_y2.text()))
]
],
EI=EI,
EA=EA,
element_type=element_types[e])
else:
self.ss.add_element(
location=[[
float(self.fn.filter(self.mw.beam_x1.text())),
float(self.fn.filter(self.mw.beam_y1.text()))
],
[
float(self.fn.filter(
self.mw.beam_x2.text())),
float(self.fn.filter(self.mw.beam_y2.text()))
]])
self.visualize_structure()
self.states.append(pickle.dumps(self.ss))
except:
self.fn.warning()
def beam_info(self):
if self.mw.utilizeinfo.isChecked():
self.mw.frame_4.setHidden(False)
else:
self.mw.frame_4.setHidden(True)
def element_type_list(self):
if self.mw.elementtype.currentIndex() == 1:
self.mw.beam_I.setEnabled(False)
elif self.mw.elementtype.currentIndex() == 0:
self.mw.beam_I.setEnabled(True)
def add_node(self):
try:
if int(self.mw.node_id.text()) in self.ss.node_map.keys():
self.workaround()
self.ss.insert_node(element_id=int(self.mw.node_id.text()),
location=[
self.fn.filter(self.mw.node_x.text()),
self.fn.filter(self.mw.node_y.text())
])
self.mw.last_figure.click()
self.states.append(pickle.dumps(self.ss))
else:
self.fn.invalid_id_warning()
except:
self.fn.warning()
def add_support(self):
try:
if int(self.mw.support_pos.text()) in self.ss.node_map.keys():
self.workaround()
if self.mw.support_hinged.isChecked():
self.ss.add_support_hinged(
node_id=int(self.mw.support_pos.text()))
elif self.mw.support_roll.isChecked():
self.ss.add_support_roll(
node_id=int(self.mw.support_pos.text()),
angle=float(
self.fn.filter(self.mw.support_angle.text())))
elif self.mw.support_fixed.isChecked():
self.ss.add_support_fixed(
node_id=int(self.mw.support_pos.text()))
elif self.mw.support_spring.isChecked():
self.ss.add_support_spring(
node_id=int(self.mw.support_pos.text()),
translation=self.mw.spring_translation.text(),
k=self.mw.spring_k.text())
elif self.mw.support_internal_hinge.isChecked():
pass
self.mw.last_figure.click()
self.states.append(pickle.dumps(self.ss))
self.fn.enable_buttons()
else:
self.fn.invalid_id_warning()
except:
self.fn.warning()
def show_support_stuff(self):
if self.mw.support_roll.isChecked():
self.mw.support_angle.setHidden(
True) # Always true due to anaStruct bug
self.mw.label_113.setHidden(
True) # Always true due to anaStruct bug
self.mw.label_27.setHidden(
True) # Always true due to anaStruct bug
self.mw.label_71.setHidden(True)
self.mw.label_73.setHidden(True)
self.mw.spring_k.setHidden(True)
self.mw.spring_translation.setHidden(True)
elif self.mw.support_spring.isChecked():
self.mw.label_71.setHidden(False)
self.mw.label_73.setHidden(False)
self.mw.spring_k.setHidden(False)
self.mw.spring_translation.setHidden(False)
self.mw.support_angle.setHidden(
True) # Always true due to anaStruct bug
self.mw.label_27.setHidden(True)
self.mw.label_127.setHidden(False)
else:
self.mw.support_angle.setHidden(
True) # Always true due to anaStruct bug
self.mw.label_27.setHidden(True)
self.mw.label_71.setHidden(True)
self.mw.label_73.setHidden(True)
self.mw.spring_k.setHidden(True)
self.mw.label_113.setHidden(True)
self.mw.label_127.setHidden(True)
self.mw.spring_translation.setHidden(True)
def add_point_load(self):
try:
if int(self.mw.load_pos.text()) in self.ss.node_map.keys():
self.workaround()
if self.mw.load_moment.text() != '' and float(
self.mw.load_moment.text()) != 0:
self.ss.moment_load(
node_id=int(self.mw.load_pos.text()),
Ty=float(self.fn.filter(self.mw.load_moment.text())))
if float(self.mw.load_y.text()) == 0 and float(
self.mw.load_x.text()) == 0 and float(
self.mw.load_angle.text()) == 0:
pass
elif self.mw.load_y.text() != '' and self.mw.load_x.text(
) != '' and self.mw.load_angle.text() != '':
self.ss.point_load(
node_id=int(self.mw.load_pos.text()),
Fy=float(self.fn.filter(self.mw.load_y.text())),
Fx=float(self.fn.filter(self.mw.load_x.text())),
rotation=float(
self.fn.filter(self.mw.load_angle.text())))
self.mw.last_figure.click()
self.states.append(pickle.dumps(self.ss))
self.fn.enable_buttons()
else:
self.fn.invalid_id_warning()
except:
self.fn.warning()
def add_q_load(self):
try:
if int(self.mw.qload_pos.text()) in self.ss.node_map.keys():
if float(self.mw.qload_initial.text()) >= 0 and float(self.mw.qload_final.text()) >= 0 or \
float(self.mw.qload_initial.text()) <= 0 and float(self.mw.qload_final.text()) <= 0:
self.workaround()
if self.mw.qload_initial.text() == '':
self.mw.qload_final.setText(
self.fn.filter(self.mw.qload_final.text()))
if self.mw.qload_final.text() == '':
self.mw.qload_final.setText(
self.fn.filter(self.mw.qload_initial.text()))
self.ss.q_load(
element_id=int(self.mw.qload_pos.text()),
q=(float(self.fn.filter(self.mw.qload_initial.text())),
float(self.fn.filter(self.mw.qload_final.text()))))
self.mw.last_figure.click()
self.states.append(pickle.dumps(self.ss))
self.fn.enable_buttons()
else:
msg = QMessageBox()
msg.setWindowTitle(self.mw.warning_title)
msg.setText(self.mw.qload_warning)
msg.setIcon(QMessageBox.Warning)
x = msg.exec_()
else:
self.fn.invalid_id_warning()
except:
self.fn.warning()
def visualize_structure(self):
if self.ss.element_map:
self.mw.MplWidget.canvas.figure.clear()
ax = self.mw.MplWidget.canvas.figure.add_subplot(111)
self.fn.visualize(
self.ss.show_structure(show=False,
figure=(self.mw.MplWidget.canvas.figure,
ax)))
ax.patch.set_alpha(0.2)
self.mw.last_figure = self.mw.show_structure
else:
self.mw.MplWidget.plot(has_grid=self.mw.gridBox.isChecked())
self.fn.figurefix()
self.mw.last_figure = None
def visualize_diagram(self):
self.solve()
self.mw.MplWidget.canvas.figure.clear()
ax = self.mw.MplWidget.canvas.figure.add_subplot(111)
ax.patch.set_alpha(0.2)
self.fn.visualize(
self.ss.show_structure(show=False,
free_body_diagram=1,
figure=(self.mw.MplWidget.canvas.figure,
ax)))
self.mw.last_figure = self.mw.show_diagram
def visualize_supports(self):
self.solve()
self.mw.MplWidget.canvas.figure.clear()
ax = self.mw.MplWidget.canvas.figure.add_subplot(111)
ax.patch.set_alpha(0.2)
self.fn.visualize(
self.ss.show_reaction_force(
show=False, figure=(self.mw.MplWidget.canvas.figure, ax)))
self.mw.last_figure = self.mw.show_supports
def visualize_normal(self):
self.solve()
self.mw.MplWidget.canvas.figure.clear()
ax = self.mw.MplWidget.canvas.figure.add_subplot(111)
ax.patch.set_alpha(0.2)
self.fn.visualize(
self.ss.show_axial_force(show=False,
figure=(self.mw.MplWidget.canvas.figure,
ax)))
self.mw.last_figure = self.mw.show_normal
def visualize_shear(self):
self.solve()
self.mw.MplWidget.canvas.figure.clear()
ax = self.mw.MplWidget.canvas.figure.add_subplot(111)
ax.patch.set_alpha(0.2)
self.fn.visualize(
self.ss.show_shear_force(show=False,
figure=(self.mw.MplWidget.canvas.figure,
ax)))
self.mw.last_figure = self.mw.show_shear
def visualize_moment(self):
self.solve()
self.mw.MplWidget.canvas.figure.clear()
ax = self.mw.MplWidget.canvas.figure.add_subplot(111)
ax.patch.set_alpha(0.2)
self.fn.visualize(
self.ss.show_bending_moment(
show=False, figure=(self.mw.MplWidget.canvas.figure, ax)))
self.mw.last_figure = self.mw.show_moment
def visualize_displacement(self):
self.solve()
self.mw.MplWidget.canvas.figure.clear()
ax = self.mw.MplWidget.canvas.figure.add_subplot(111)
ax.patch.set_alpha(0.2)
self.fn.visualize(
self.ss.show_displacement(show=False,
figure=(self.mw.MplWidget.canvas.figure,
ax)))
self.mw.last_figure = self.mw.show_displacement
def solve(self):
self.was_solved = True
self.ss.solve()
def static_solver(self, clean=True):
if find_executable('latex'):
if self.mw.show_moment.isEnabled():
if (len(self.ss.supports_roll) == 1 and len(self.ss.supports_hinged) == 1) \
or (len(self.ss.supports_fixed) == 1):
dialog = QDialog()
prompt = PathPrompt(self.mw.language, dialog)
dialog.exec_()
if not prompt.userTerminated:
solve_path = prompt.path
file, ok = QFileDialog.getSaveFileName(
self.mw, self.mw.pdf_title, self.mw.pdf_text,
"PDF (*.pdf)")
if ok:
try:
self.mw.toolBox.setCurrentIndex(0)
pdf_dir, filename = split_dir_filename(file)
make_pdf_folders(pdf_dir)
self.ss.color_scheme = "bright"
plt.style.use('default')
mn = Manager(self.ss)
pdf_generator_thread = PDFGeneratorThread(
mn.generate_pdf,
self.mw.language,
pdf_path=pdf_dir,
filename=filename,
solve_path=solve_path,
path_warning=self.fn.path_warning,
)
self.fn.setupLoading(pdf_generator_thread)
pdf_generator_thread.finished.connect(
self.on_finished)
pdf_generator_thread.start()
self.mw.loadingScreen.exec_()
if not self.mw.loadingUi.userTerminated:
self.fn.pdf_generated_prompt()
if clean:
delete_folder(pdf_dir)
self.ss.color_scheme = "dark"
plt.style.use('dark_background')
except:
self.fn.latex_packages_warning()
else:
self.fn.static_warning()
else:
self.fn.warning()
else:
self.fn.latex_warning()
def on_finished(self):
self.mw.loadingScreen.close()
def reset_struct_elems(self):
self.ss = SystemElements()
self.ss.color_scheme = "dark"
self.states.clear()
self.mw.MplWidget.plot(has_grid=self.mw.gridBox.isChecked())
self.mw.MplWidget.set_background_alpha()
self.mw.MplWidget.set_subplot_alpha()
self.fn.figurefix()
self.was_solved = False
self.fn.disable_buttons()
def load_structure_aux(self, file):
with open(f'{file}', 'rb') as f:
self.ss, _, _ = pickle.load(f)
self.mw.struct_loaded = True
def workaround(self):
if self.was_solved:
self.ss = pickle.loads(self.states[-1])
self.was_solved = False
def reset(self):
self.workaround()
self.ss.remove_loads()
self.mw.MplWidget.canvas.figure.clear()
ax = self.mw.MplWidget.canvas.figure.add_subplot(111)
ax.patch.set_alpha(0.2)
self.fn.visualize(
self.ss.show_structure(show=False,
figure=(self.mw.MplWidget.canvas.figure,
ax)))
self.states.append(pickle.dumps(self.ss))
self.fn.disable_buttons()
class Artist:
fig_counter = 0
def __init__(
self,
system_elements,
node_order=None,
assemble_order=None,
target_dir="tmp",
):
self.ss = system_elements
self.branch_ss = SystemElements()
self.assemble_order = assemble_order
self.node_order = node_order
self.target_dir = target_dir
def draw_structure(self,
show=False,
save_figure=True,
plotting_start_node=0,
element_id=0):
plot_iterations = False
node_index = 0
figure = plt.figure(figsize=(12, 8))
subplot = figure.add_subplot(111)
plt.tight_layout()
for branch in self.assemble_order:
self.draw_element(branch)
first_node = next(i for i in self.node_order
if i == branch[0] or i == branch[1])
for roll in self.ss.supports_roll:
if roll.id == first_node:
self.draw_support(first_node, subplot)
for hinged in self.ss.supports_hinged:
if hinged.id == first_node:
self.draw_support(first_node, subplot)
for fixed in self.ss.supports_fixed:
if fixed.id == first_node:
self.draw_support(first_node, subplot)
if self.ss.loads_point.get(first_node):
self.draw_point_load(first_node)
if self.ss.loads_moment.get(first_node):
self.draw_moment(first_node)
elements_node1 = self.ss.node_element_map.get(branch[0])
elements_node2 = self.ss.node_element_map.get(branch[1])
for element in elements_node1:
if element in elements_node2:
q_load = self.ss.loads_q.get(element.id)
if q_load:
if branch == self.assemble_order[-1]:
self.draw_q_load(
get_relative_element_by_coordinates(
self.ss, self.branch_ss, element.id),
q_load)
else:
xi = element.vertex_1.x
yi = element.vertex_1.y
xf = element.vertex_2.x
yf = element.vertex_2.y
x_average = (xi + xf) / 2
y_average = (yi + yf) / 2
base = ((xf - xi)**2 + (yf - yi)**2)**0.5
load = ((-q_load[0][0] + -q_load[1][0]) * base) / 2
Fz = load * math.cos(element.angle)
Fx = load * math.cos(element.angle -
(90 * math.pi / 180))
h = 0.2 * self.ss.plotter.max_val_structure
x, y, len_x, len_y, point_load = get_arrow_patch_values(
Fx, Fz, (x_average, y_average), h)
plot_arrow(subplot, point_load,
[h, h, [x, y, len_x, len_y]])
# needs further testing
break
node_index += 1
if plotting_start_node in branch:
plot_iterations = True
if show and plot_iterations:
self.branch_ss.show_structure(show=False,
figure=(figure, subplot))
figure.show()
if save_figure and plot_iterations:
fig = self.branch_ss.show_structure(show=False,
figure=(figure, subplot))
fig.savefig(fr'{self.target_dir}\figs\structure{element_id}')
def generate_figures_for_pdf(self):
fig = self.ss.show_structure(show=False)
fig.savefig(fr'{self.target_dir}\figs\structure')
fig = self.ss.show_structure(show=False, free_body_diagram=3)
fig.savefig(fr'{self.target_dir}\figs\diagram1')
fig = self.ss.show_structure(show=False, free_body_diagram=2)
fig.savefig(fr'{self.target_dir}\figs\diagram2')
fig = self.ss.show_reaction_force(show=False)
fig.savefig(fr'{self.target_dir}\figs\supports')
fig = self.ss.show_axial_force(show=False)
fig.savefig(fr'{self.target_dir}\figs\axial')
fig = self.ss.show_shear_force(show=False)
fig.savefig(fr'{self.target_dir}\figs\shear')
fig = self.ss.show_bending_moment(show=False)
fig.savefig(fr'{self.target_dir}\figs\moment')
def draw_support(self, node_id, subplot, roll_direction=None):
support_node = self.ss.reaction_forces.get(node_id)
if round(support_node.Fx, 2):
plot_arrow(subplot, support_node.Fx,
self.ss.reaction_vectors_data.get(f"{node_id}Fx"))
if round(support_node.Fz, 2):
plot_arrow(subplot, support_node.Fz,
self.ss.reaction_vectors_data.get(f"{node_id}Fz"))
if round(support_node.Ty, 2):
plot_moment(subplot, self.ss.node_map.get(node_id),
support_node.Ty,
self.ss.reaction_vectors_data.get(f"{node_id}Ty"))
def draw_element(self, branch):
self.add_element_to_plot(branch)
def add_element_to_plot(self, element):
for node in range(len(element) - 1):
self.branch_ss.add_element(
[[
self.ss.node_map.get(element[node]).vertex.x,
self.ss.node_map.get(element[node]).vertex.y
],
[
self.ss.node_map.get(element[node + 1]).vertex.x,
self.ss.node_map.get(element[node + 1]).vertex.y
]])
def draw_point_load(self, node_id):
point_load = self.ss.loads_point[node_id]
self.branch_ss.point_load(
node_id=self.get_relative_node_by_coordinates(node_id),
Fx=point_load[0],
Fy=-point_load[1])
def draw_q_load(self, element_id, q_load):
self.branch_ss.q_load(element_id=element_id,
q=(q_load[0][0], q_load[1][0]))
def draw_moment(self, node_id):
moment_load = self.ss.loads_moment.get(node_id)
self.branch_ss.moment_load(
node_id=self.get_relative_node_by_coordinates(node_id),
Ty=moment_load)
def get_relative_node_by_coordinates(self, node_id):
coords = (self.ss.node_map.get(node_id).vertex.x,
self.ss.node_map.get(node_id).vertex.y)
for node_key in self.branch_ss.node_map:
node = self.branch_ss.node_map.get(node_key)
if coords[0] == node.vertex.x and coords[1] == node.vertex.y:
return node.id
return None