-
Notifications
You must be signed in to change notification settings - Fork 31
/
theme.R
103 lines (98 loc) · 4.37 KB
/
theme.R
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#' A custom theme specifically for use with ridgeline plots
#'
#' This theme has some special modifications that make ridgeline plots look better, such as properly aligned y axis labels.
#' It can draw plots with and without background grids (see examples).
#'
#' @param font_size Overall font size. Default is 14.
#' @param font_family Default font family.
#' @param line_size Default line size.
#' @param grid If `TRUE` (default), a background grid is drawn. If `FALSE`, background is left empty.
#' @param center_axis_labels If `TRUE`, axis labels are drawn centered. If `FALSE` (default), axis lables are
#' drawn right/top-aligned.
#' @return The theme.
#' @examples
#' library(ggplot2)
#'
#' # Example with background grid
#' ggplot(iris, aes(x = Sepal.Length, y = Species, group = Species)) +
#' geom_density_ridges(rel_min_height = 0.005) +
#' scale_y_discrete(expand = c(0.01, 0)) +
#' scale_x_continuous(expand = c(0.01, 0)) +
#' theme_ridges()
#'
#' # Example without background grid
#' ggplot(iris, aes(x = Sepal.Length, y = Species, group = Species)) +
#' geom_density_ridges() +
#' scale_y_discrete(expand = c(0.01, 0)) +
#' scale_x_continuous(expand = c(0.01, 0)) +
#' theme_ridges(grid = FALSE)
#'
#' @export
theme_ridges <- function(font_size = 14, font_family = "", line_size = .5, grid = TRUE, center_axis_labels = FALSE) {
half_line <- font_size / 2
small_rel <- 0.857
small_size <- small_rel * font_size
color <- "grey90"
if (grid) {
panel.grid.major <- element_line(colour = color, linewidth = line_size)
axis.ticks <- element_line(colour = color, linewidth = line_size)
axis.ticks.y <- axis.ticks
}
else {
panel.grid.major <- element_blank()
axis.ticks <- element_line(colour = "black", linewidth = line_size)
axis.ticks.y <- element_blank()
}
if (center_axis_labels) {
axis_just <- 0.5
}
else {
axis_just <- 1.0
}
theme_grey(base_size = font_size, base_family = font_family) %+replace%
theme(
rect = element_rect(fill = "transparent", colour = NA, color = NA, linewidth = 0, linetype = 0),
text = element_text(family = font_family, face = "plain", colour = "black",
size = font_size, hjust = 0.5, vjust = 0.5, angle = 0, lineheight = .9,
margin = margin(), debug = FALSE),
axis.text = element_text(colour = "black", size = small_size),
#axis.title = element_text(face = "bold"),
axis.text.x = element_text(margin = margin(t = small_size / 4), vjust = 1),
axis.text.y = element_text(margin = margin(r = small_size / 4), hjust = 1, vjust = 0),
axis.title.x = element_text(
margin = margin(t = small_size / 2, b = small_size / 4),
hjust = axis_just
),
axis.title.y = element_text(
angle = 90,
margin = margin(r = small_size / 2, l = small_size / 4),
hjust = axis_just
),
axis.ticks = axis.ticks,
axis.ticks.y = axis.ticks.y,
axis.line = element_blank(),
legend.key = element_blank(),
legend.key.size = grid::unit(1, "lines"),
legend.text = element_text(size = rel(small_rel)),
legend.justification = c("left", "center"),
panel.background = element_blank(),
panel.border = element_blank(),
# make grid lines
panel.grid.major = panel.grid.major,
panel.grid.minor = element_blank(),
strip.text = element_text(size = rel(small_rel)),
strip.background = element_rect(fill = "grey80", colour = "grey50", linewidth = 0),
plot.background = element_blank(),
plot.title = element_text(face = "bold",
size = font_size,
margin = margin(b = half_line), hjust = 0),
plot.subtitle = element_text(size = rel(small_rel),
hjust = 0, vjust = 1,
margin = margin(b = half_line * small_rel)),
plot.caption = element_text(size = rel(small_rel),
hjust = 1, vjust = 1,
margin = margin(t = half_line * small_rel)),
plot.margin = margin(half_line, font_size, half_line, half_line),
complete = TRUE
)
}