153 lines
4.4 KiB
Python
153 lines
4.4 KiB
Python
#!/usr/bin/env python3
|
|
################################################################################
|
|
# Define the prefered plotting defaults.
|
|
# These generally translate to how I want stuff to show up in IEEE papers.
|
|
# Note that when I do my debugging, I override figure.figsize in my testing
|
|
# enviornment.
|
|
################################################################################
|
|
|
|
import matplotlib
|
|
import matplotlib.font_manager as FM
|
|
import re
|
|
from matplotlib import rcParams, pyplot as pp
|
|
from cycler import cycler
|
|
|
|
POLAR_YLIM_CONST=(-18,-6)
|
|
POLAR_YLIM_CONST_MEAS=(-22,-10)
|
|
POLAR_YLIM_CONST_MEAS=(-15,-3)
|
|
POLAR_YLIM_CONST_ALT=(-32,-6)
|
|
GAIN_FIXED_YLIM1=(-20,-0)
|
|
GAIN_FIXED_YLIM1_REVISED=(-18,-8)
|
|
GAIN_FIXED_YLIM2=(-30,-0)
|
|
GAIN_FIXED_YLIM3=(-10,10)
|
|
GAIN_FIXED_YLIM4=(-0,20)
|
|
GAIN_FIXED_YLIM5=(-20,0)
|
|
|
|
fcFontList = FM.get_fontconfig_fonts()
|
|
# Search only for fonts that have name matches similar to this
|
|
# note this is ALSO a priority list
|
|
fontsDesired = ['Times', 'Helvetica', 'Arial']
|
|
fontsDesiredRe = re.compile('|'.join(fontsDesired), flags=re.IGNORECASE)
|
|
# Create a unique set of the fonts selected out of all of the system fonts
|
|
fontsAvailable = frozenset([FM.FontProperties(fname=fcFont).get_name()\
|
|
for fcFont in fcFontList if fontsDesiredRe.search(fcFont) != None])
|
|
|
|
fontSelected=None
|
|
for fontSearch in fontsDesired:
|
|
for fontFound in fontsAvailable:
|
|
if re.search(fontSearch, fontFound, flags=re.IGNORECASE) != None:
|
|
fontSelected = fontFound
|
|
break
|
|
if fontSelected != None:
|
|
break
|
|
if fontSelected == None:
|
|
print("WARN: None of the requested fonts found on this system.")
|
|
else:
|
|
print("INFO: Using font '%s'" % (fontSelected))
|
|
newFontPriority = [fontSelected]
|
|
newFontPriority.extend(rcParams['font.serif'])
|
|
rcParams['font.serif'] = newFontPriority
|
|
|
|
rcParams['grid.alpha'] = 0.5
|
|
rcParams['grid.linestyle'] = ':'
|
|
rcParams['font.family'] = ['serif']
|
|
rcParams['font.size'] = 7
|
|
#rcParams['figure.titlesize'] = "medium"
|
|
rcParams['axes.titlesize'] = 9
|
|
rcParams['figure.titlesize'] = rcParams['axes.titlesize']
|
|
#rcParams['mathtext.fontset'] = 'dejavuserif'
|
|
#rcParams['mathtext.fontset'] = 'custom'
|
|
rcParams['mathtext.it'] = 'serif:italic'
|
|
rcParams['mathtext.bf'] = 'serif:bold'
|
|
rcParams['mathtext.sf'] = 'serif'
|
|
rcParams['mathtext.tt'] = 'monospace'
|
|
rcParams['lines.linewidth'] = 1.0
|
|
#rcParams['axes.grid'] = True
|
|
|
|
# axes.prop_cycle
|
|
COLOR_CYCLE_LIST = [
|
|
[0, 0.4470, 0.7410],
|
|
[0.8500, 0.3250, 0.0980],
|
|
[0.4940, 0.1840, 0.5560],
|
|
[0.4660, 0.6740, 0.1880],
|
|
[0.3010, 0.7450, 0.9330],
|
|
[0.6350, 0.0780, 0.1840],
|
|
[0.9290, 0.6940, 0.1250],
|
|
[1, 0, 1]]#,
|
|
# [0, 1, 1],
|
|
# [1, 0, 0],
|
|
# [0, 1, 0]]
|
|
|
|
rcParams['axes.prop_cycle'] = (cycler('linestyle',['-','--'])*cycler(color=COLOR_CYCLE_LIST))
|
|
|
|
for tri in COLOR_CYCLE_LIST:
|
|
color = '0x' + ''.join([ "%02x" % int(255*x) for x in tri])
|
|
|
|
figures_directory='figures'
|
|
|
|
def figAnnotateCorner(hfig, msg, corner=1, ratio=0.01, clear=True):
|
|
if clear:
|
|
figAnnotateClear(hfig)
|
|
if corner in [1,2]:
|
|
loc_x = ratio
|
|
algn_h = 'left'
|
|
else:
|
|
loc_x = 1-ratio
|
|
algn_h = 'right'
|
|
if corner in [1,4]:
|
|
loc_y = ratio
|
|
algn_v = 'bottom'
|
|
else:
|
|
loc_y = 1-ratio
|
|
algn_v = 'top'
|
|
ax = hfig.get_axes()[0]
|
|
hfig.text(loc_x, loc_y, msg, transform=ax.transAxes,
|
|
va=algn_v, ha=algn_h)
|
|
|
|
def addHalfTicks(haxis, gridOn=True):
|
|
ticks = haxis.get_yticks()
|
|
minor_ticks = (ticks[1:]+ticks[:-1])/2
|
|
haxis.set_yticks(minor_ticks, minor=True)
|
|
if gridOn:
|
|
haxis.grid(gridOn, which='major')
|
|
haxis.grid(gridOn, which='minor')
|
|
|
|
def axAnnotateCorner(ha, msg, corner=1, ratio=0.01):
|
|
if corner in [1,2]:
|
|
loc_x = ratio
|
|
algn_h = 'left'
|
|
else:
|
|
loc_x = 1-ratio
|
|
algn_h = 'right'
|
|
if corner in [1,4]:
|
|
loc_y = ratio
|
|
algn_v = 'bottom'
|
|
else:
|
|
loc_y = 1-ratio
|
|
algn_v = 'top'
|
|
#ax = ha.get_axes()[0]
|
|
ha.text(loc_x, loc_y, msg, transform=ha.transAxes,
|
|
va=algn_v, ha=algn_h)
|
|
|
|
def figAnnotateClear(hobj):
|
|
if type(hobj == matplotlib.figure.Figure):
|
|
hfig = hobj
|
|
children = hfig.get_children()
|
|
for child in children:
|
|
if type(child) == matplotlib.text.Text:
|
|
child.remove()
|
|
|
|
def annotateArrow(ha, y, x_list, direction='right'):
|
|
x=[min(x_list), max(x_list)]
|
|
if direction == 'right':
|
|
x=[x[-1], x[0]]
|
|
|
|
ha.annotate("",
|
|
xy=(x[0], y), xycoords='data',
|
|
xytext=(x[1], y), textcoords='data',
|
|
arrowprops=dict(width=2, headwidth=8, headlength=6, facecolor='black'))
|
|
|
|
def annotateArrowStr(ha, msg, y, x):
|
|
ha.annotate(msg,
|
|
xy=(x, y), xycoords='data',
|
|
xytext=(x, y), textcoords='data')
|