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