GraphIF/plot_utilities.py at main · Giuseppe499/GraphIF · 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
import matplotlib


def get_font_size():
    return matplotlib.rcParams["font.size"]


def set_font_size(size):
    matplotlib.rcParams.update({"font.size": size})


class FontSize:
    def __init__(self, size):
        self.size = size
        self.original_size = get_font_size()

    def __enter__(self):
        set_font_size(self.size)

    def __exit__(self, exc_type, exc_val, exc_tb):
        set_font_size(self.original_size)


def get_marker_size():
    return matplotlib.rcParams["lines.markersize"]


def set_marker_size(size):
    matplotlib.rcParams.update({"lines.markersize": size})


class MarkerSize:
    def __init__(self, size):
        self.size = size
        self.original_size = get_marker_size()

    def __enter__(self):
        set_marker_size(self.size)

    def __exit__(self, exc_type, exc_val, exc_tb):
        set_marker_size(self.original_size)


def set_cmap(cmap):
    matplotlib.rcParams.update({"image.cmap": cmap})