Skip to content

Arrow head length and head width option is added in nyquist_plot function #379

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 7 commits into from
Mar 18, 2020
Merged

Arrow head length and head width option is added in nyquist_plot function #379

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
b28e9ab
Nyquist_plot changed to accommodate arrow size
geekonloose Mar 4, 2020
a4f2d25
color option is added
geekonloose Mar 4, 2020
f5f05bd
formatting
geekonloose Mar 4, 2020
5de0899
formatting is retained
geekonloose Mar 5, 2020
8f8565b
line formatting corrected
geekonloose Mar 5, 2020
1f4885d
again line formatting
geekonloose Mar 5, 2020
7c96781
hl changed to arrowhead_width and hw changed to arrowhead_length
geekonloose Mar 5, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 9 additions & 4 deletions control/freqplot.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -433,8 +433,9 @@ def gen_zero_centered_series(val_min, val_max, period):
# Nyquist plot
#

def nyquist_plot(syslist, omega=None, Plot=True, color=None,
labelFreq=0, *args, **kwargs):
def nyquist_plot(syslist, omega=None, Plot=True,
labelFreq=0, arrowhead_length=0.1, arrowhead_width=0.1,
color=None, *args, **kwargs):
"""
Nyquist plot for a system

Expand All @@ -452,6 +453,8 @@ def nyquist_plot(syslist, omega=None, Plot=True, color=None,
Used to specify the color of the plot
labelFreq : int
Label every nth frequency on the plot
arrowhead_width : arrow head width
arrowhead_length : arrow head length
*args
Additional arguments for :func:`matplotlib.plot` (color, linestyle, etc)
**kwargs:
Expand Down Expand Up @@ -511,12 +514,14 @@ def nyquist_plot(syslist, omega=None, Plot=True, color=None,
ax = plt.gca()
# Plot arrow to indicate Nyquist encirclement orientation
ax.arrow(x[0], y[0], (x[1]-x[0])/2, (y[1]-y[0])/2, fc=c, ec=c,
head_width=0.2, head_length=0.2)
head_width=arrowhead_width,
head_length=arrowhead_length)

plt.plot(x, -y, '-', color=c, *args, **kwargs)
ax.arrow(
x[-1], -y[-1], (x[-1]-x[-2])/2, (y[-1]-y[-2])/2,
fc=c, ec=c, head_width=0.2, head_length=0.2)
fc=c, ec=c, head_width=arrowhead_width,
head_length=arrowhead_length)

# Mark the -1 point
plt.plot([-1], [0], 'r+')
Expand Down