widgets.py

"""
======
Slider
======

In this example, sliders are used to control the frequency and amplitude of
a sine wave.

"""

importinspect

importnumpyasnp
importmatplotlib.pyplotasplt
frommatplotlib.widgetsimportSlider, Button

fromdata_prototype.artistimportCompatibilityArtistasCA
fromdata_prototype.lineimportLine
fromdata_prototype.containersimportFuncContainer
fromdata_prototype.descriptionimportDesc
fromdata_prototype.conversion_edgeimportFuncEdge


classSliderContainer(FuncContainer):
def__init__(self, xfuncs, /, **sliders):
self._sliders=sliders
forsliderinsliders.values():
slider.on_changed(
lambda_, sld=slider: sld.ax.figure.canvas.draw_idle(),
 )

defget_needed_keys(f, offset=1):
returntuple(inspect.signature(f).parameters)[offset:]

super().__init__(
 {
k: (
s,
# this line binds the correct sliders to the functions
# and makes lambdas that match the API FuncContainer needs
lambdax, keys=get_needed_keys(f), f=f: f(
x, *(sliders[k].valforkinkeys)
 ),
 )
fork, (s, f) inxfuncs.items()
 },
 )

def_query_hash(self, graph, parent_coordinates):
key=super()._query_hash(graph, parent_coordinates)
# inject the slider values into the hashing logic
returnhash((key, tuple(s.valforsinself._sliders.values())))


# Define initial parameters
init_amplitude=5
init_frequency=3

# Create the figure and the line that we will manipulate
fig, ax=plt.subplots()
ax.set_xlim(0, 1)
ax.set_ylim(-7, 7)

ax.set_xlabel("Time [s]")

# adjust the main plot to make room for the sliders
fig.subplots_adjust(left=0.25, bottom=0.25, right=0.75)

# Make a horizontal slider to control the frequency.
axfreq=fig.add_axes([0.25, 0.1, 0.65, 0.03])
freq_slider=Slider(
ax=axfreq,
label="Frequency [Hz]",
valmin=0.1,
valmax=30,
valinit=init_frequency,
)

# Make a vertically oriented slider to control the amplitude
axamp=fig.add_axes([0.1, 0.25, 0.0225, 0.63])
amp_slider=Slider(
ax=axamp,
label="Amplitude",
valmin=0,
valmax=10,
valinit=init_amplitude,
orientation="vertical",
)

# Make a vertically oriented slider to control the phase
axphase=fig.add_axes([0.85, 0.25, 0.0225, 0.63])
phase_slider=Slider(
ax=axphase,
label="Phase [rad]",
valmin=-2*np.pi,
valmax=2*np.pi,
valinit=0,
orientation="vertical",
)

# pick a cyclic color map
cmap=plt.get_cmap("twilight")

# set up the data container
fc=SliderContainer(
 {
# the x data does not need the sliders values
"x": (("N",), lambdat: t),
"y": (
 ("N",),
# the y data needs all three sliders
lambdat, amplitude, frequency, phase: amplitude
*np.sin(2*np.pi*frequency*t+phase),
 ),
# the color data has to take the x (because reasons), but just
# needs the phase
"color": ((1,), lambda_, phase: phase),
 },
# bind the sliders to the data container
amplitude=amp_slider,
frequency=freq_slider,
phase=phase_slider,
)
lw=Line(
fc,
# color map phase (scaled to 2pi and wrapped to [0, 1])
 [
FuncEdge.from_func(
"color",
lambdacolor: cmap((color/ (2*np.pi)) %1),
 {"color": Desc((1,))},
 {"color": Desc((), "display")},
 )
 ],
linewidth=5.0,
linestyle="-",
)
ax.add_artist(CA(lw))


# Create a `matplotlib.widgets.Button` to reset the sliders to initial values.
resetax=fig.add_axes([0.8, 0.025, 0.1, 0.04])
button=Button(resetax, "Reset", hovercolor="0.975")
button.on_clicked(
lambda_: [sld.reset() forsldin (freq_slider, amp_slider, phase_slider)]
)

plt.show()