plotting_utils/acm_fig.py at main · ucsdwcsng/plotting_utils · GitHub

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
58
59
60
61
62
63
64
65
66
67
68
69
70
import matplotlib.pyplot as plt

import scienceplots
plt.style.use('science')


TEXTWIDTH = 506.295 / 72.27  # 7.006 inches (full two-column width)
COLWIDTH = 241.1474 / 72.27   # 3.337 inches (single column width)

def set_acm():
    print("Using ACM template page widths")
    TEXTWIDTH = 506.295 / 72.27  # 7.006 inches (full two-column width)
    COLWIDTH = 241.1474 / 72.27   # 3.337 inches (single column width)


def set_ieee():
    print("Using IEEE template page widths")
    TEXTWIDTH = 506.295 / 72.27  # 7.006 inches (full two-column width)
    COLWIDTH = 241.1474 / 72.27   # 3.337 inches (single column width)

set_acm()

# Font sizes
FONT_SIZES = {
    'font.size': 8,
    'axes.labelsize': 8,
    'xtick.labelsize': 7,
    'ytick.labelsize': 7,
    'legend.fontsize': 6,
    'axes.xmargin': 0.0
}

plt.rcParams.update(FONT_SIZES)
plt.rcParams['lines.linewidth'] = 1.0
plt.rcParams['axes.grid'] = True

def create_figure(width_fraction=1.0, aspect_ratio=0.75, use_textwidth=False, subplots=None):
    """Create figure at final display size"""
    base_width = TEXTWIDTH if use_textwidth else COLWIDTH
    width = base_width * width_fraction
    height = width * aspect_ratio
    fig = plt.figure(figsize=(width,height),layout='constrained')
    if subplots is not None:
        axs = [fig.add_subplot(sp[0],sp[1],sp[2]) for sp in subplots]
        return fig, axs
    return fig

def create_figure_wh(width_fraction=1.0, height_fraction=1.0, use_textwidth=False, subplots=None):
    """Create figure at final display size"""
    base_width = TEXTWIDTH if use_textwidth else COLWIDTH
    width = base_width * width_fraction
    height = base_width * height_fraction
    fig = plt.figure(figsize=(width,height),layout='constrained')
    if subplots is not None:
        axs = [fig.add_subplot(sp[0],sp[1],sp[2]) for sp in subplots]
        return fig, axs
    return fig


def add_legend(ax, loc='lower right', in_layout=True):
    leg = ax.legend(loc=loc,
               frameon=True,
               facecolor='white',
               framealpha=1.0,
               edgecolor='black',
               borderpad=0.3,
               labelspacing=0.3,
               handletextpad=0.5)
    leg.set_in_layout(in_layout)
    return leg