"""
Show Objectives and Design variables
=====================

Plot the 3D geometry of wedges and end-spacers

Download Input files

* :download:`dipole_2d.data <../../../../../examples/input_files/dipole_iron_trans_opt.data>`
* :download:`dipole_2d.post.xml <../../../../../examples/input_files/dipole_iron_trans_opt.post.xml>`
"""

# %%
# Setting up
# ----------
# Standard Loading of input, Parsing and Initializing RoxiePlotOutputs object
#
from roxieapi.output.plots import RoxiePlotOutputs

plots = RoxiePlotOutputs(
    "../input_files/dipole_iron_trans_opt.post.xml",
    "../input_files/dipole_iron_trans_opt.data",
)

# %%
# Show Design variables
# ---------------------

for idx, opt in plots.output.opt.items():
    print(f"Optimization step nr {idx}:")
    plots.output_design_variables(1)


# %%
# Show Objectives
# ---------------

plots.output_objective_results(1)

# %%
# Retreive tables as pandas Dataframe
# -----------------------------------
output = plots.output
input = plots.input

dv = plots.tables.print_design_variables(output.opt[1].designVariables, input.design)
obj = plots.tables.print_objective_results(
    output.opt[1].objectiveResults, input.objective
)

# %%
# Design variable table
# ^^^^^^^^^^^^^^^^^^^^^
dv

# %%
# Objectives table
# ^^^^^^^^^^^^^^^^
obj