from matplotlib import pyplot
from bridges.Blennerhassett import BlennerhassettBridge
from plotting.adjustments import adjust_overview_plots
from plotting.plots import make_plots
bridge_optimized = BlennerhassettBridge()
bridge_optimized.plot_elements()
# bridge_optimized.internal_forces_table(all_uls=True)
# bridge_optimized.dc_ratio_table()
# bridge_optimized.cost_table()
fig = bridge_optimized.plot_all_effects('LL', label='New', c='blue')
adjust_overview_plots(fig)
# fig = bridge_optimized.plot_all_effects('Cable_Replacement', label='New', c='blue')
# fig = bridge_optimized.plot_all_effects('Cable_Replacement_1', fig=fig, label='Old', c='red')
# adjust_overview_plots(fig)
#
# fig, axs = pyplot.subplots(2, 2, figsize=(8, 4), dpi=240)
# axs = fig.get_axes()
# bridge_optimized.network_arch.tie.plot_effects(axs[0], 'Tie Fracture', 'Web')
# bridge_optimized.network_arch.tie.plot_effects(axs[1], 'Tie Fracture', 'Top')
# bridge_optimized.network_arch.tie.plot_effects(axs[2], 'Tie Fracture', 'Bottom')
#
# pyplot.show()
#
# # bridges_dict = {'Final design': bridge_ref, 'Thrust line arch': bridge_optimized,
# 'Continuous arch shape': bridge_continuous}
from matplotlib import pyplot
from bridges.Blennerhassett import BlennerhassettBridge
from plotting.adjustments import adjust_overview_plots, adjust_effects_plots
from plotting.general import colors
# Monitor memory usage
tracemalloc.start()
# Calculate the base case
folder = 'base case'
bridge_ref = BlennerhassettBridge(arch_optimisation=False, self_stress_state='Overall-optimisation')
# Plot permanent state and compare to design drawings
fig = bridge_ref.plot_all_effects('Permanent', label='Base case', c=colors[0])
axs = fig.get_axes()
axs[0].axhline(-44.5, c=colors[1], lw=1)
axs[0].axhline(-38, c=colors[1], lw=1)
axs[1].axhline(36, c=colors[1], lw=1)
axs[1].axhline(34.5, c=colors[1], lw=1)
axs[2].axhline(2.420/4.19, c=colors[1], lw=1)
axs[2].axhline(1.600/4.19, c=colors[1], lw=1)
axs[3].axhline(3.69, c=colors[1], lw=1)
axs[3].axhline(-3.69, c=colors[1], lw=1)
axs[4].axhline(2.7, c=colors[1], lw=1)
axs[4].plot([0, 270], [-3.17, -3.17], label='Design drawings', c=colors[1], lw=1)
adjust_overview_plots(fig)
fig.savefig('permanent state.png')
# Plot live loading range and compare to design drawings