-
Notifications
You must be signed in to change notification settings - Fork 1
/
utils.py
57 lines (46 loc) · 2.04 KB
/
utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
# Original Author: Prof. Nipun Batra
# nipunbatra.github.io
from math import sqrt
import matplotlib
SPINE_COLOR = 'gray'
def latexify(fig_width=None, fig_height=None, columns=1):
"""Set up matplotlib's RC params for LaTeX plotting.
Call this before plotting a figure.
Parameters
----------
fig_width : float, optional, inches
fig_height : float, optional, inches
columns : {1, 2}
"""
# code adapted from http://www.scipy.org/Cookbook/Matplotlib/LaTeX_Examples
# Width and max height in inches for IEEE journals taken from
# computer.org/cms/Computer.org/Journal%20templates/transactions_art_guide.pdf
# assert (columns in [1, 2])
if fig_width is None:
fig_width = 3.39 if columns == 1 else 6.9 # width in inches
if fig_height is None:
golden_mean = (sqrt(5) - 1.0)/2.0 # Aesthetic ratio
fig_height = fig_width*golden_mean # height in inches
# MAX_HEIGHT_INCHES = 8.0
# MAX_HEIGHT_INCHES = 10
# if fig_height > MAX_HEIGHT_INCHES:
# print("WARNING: fig_height too large:" + fig_height +
# "so will reduce to" + str(MAX_HEIGHT_INCHES) + "inches.")
# fig_height = MAX_HEIGHT_INCHES
# NB (bart): default font-size in latex is 11. This should exactly match
# the font size in the text if the figwidth is set appropriately.
# Note that this does not hold if you put two figures next to each other using
# minipage. You need to use subplots.
params = {'backend': 'ps',
# 'text.latex.preamble': ['\\usepackage{gensymb}'],
'axes.labelsize': 11, # fontsize for x and y labels (was 12 and before 10)
'axes.titlesize': 11,
'font.size': 11, # was 12 and before 10
'legend.fontsize': 11, # was 12 and before 10
'xtick.labelsize': 11,
'ytick.labelsize': 11,
# 'text.usetex': True,
'figure.figsize': [fig_width, fig_height],
'font.family': 'serif'
}
matplotlib.rcParams.update(params)