"""
Plotting of 3D wedges
=====================

Plot the 3D geometry of wedges and end-spacers

Download Input files

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

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

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

# %%
# Creating plot object and plotting
# ---------------------------------

plotter = plots.plots3d._create_plot()
spacers = plots.plots3d.get_endspacer_geometry(plots.output.plots3D[0], 1, 1)
plotter.add_mesh(spacers)
plotter.show()

# %%
# Create output file
# ------------------
plotter = plots.plots3d._create_plot()
plotter.add_mesh(spacers)
plotter.export_html("export_wedges_3d.html")

# Save vtk file
spacers.save("spacers.vtm")
plotter.show()