Skip to content
GitLab
Select Git revision
  • master default
Find file BlameHistoryPermalink
panel.R 8.30 KiB
1 
# Usefull library
2 
library(ggplot2)
3 
library(scales)
4 
library(qpdf)
5 
library(gridExtra)
6 
library(gridtext)
7 
library(dplyr)
8 
library(grid)
9
10
11 
# Time panel
12 
panel = function (df_data, df_meta, figdir, filedir_opt='', filename_opt='', variable='', df_trend=NULL, p_threshold=0.1, unit2day=365.25, type='', missRect=FALSE) {
13
14 
    if (all(class(df_data) != 'list')) {
15 
        df_data = list(df_data)
16 
    }
17
18 
    nbp = length(df_data)
19
20 
    if (all(class(df_trend) != 'list')) {
21 
        df_trend = list(df_trend)
22 
        if (length(df_trend) == 1) {
23 
            df_trend = replicate(nbp, df_trend)
24 
        }}
25
26 
    if (all(class(p_threshold) != 'list')) {
27 
        p_threshold = list(p_threshold)
28 
        if (length(p_threshold) == 1) {
29 
            p_threshold = replicate(nbp, p_threshold)
30 
        }}
31
32 
    if (all(class(unit2day) != 'list')) {
33 
        unit2day = list(unit2day)
34 
        if (length(unit2day) == 1) {
35 
            unit2day = replicate(nbp, unit2day)
36 
        }}
37
38 
    if (all(class(type) != 'list')) {
39 
        type = list(type)
40 
        if (length(type) == 1) {
41 
            type = replicate(nbp, type)
42 
        }}
43
44 
    if (all(class(missRect) != 'list')) {
45 
        missRect = list(missRect)
46 
        if (length(missRect) == 1) {
47 
            missRect = replicate(nbp, missRect)
48 
        }}
49
50 
    # print(df_data)
51 
    # print(df_trend)
52 
    # print(p_threshold)
53 
    # print(unit2day)
54 
    # print(missRect)
55
56 
    list_df2plot = vector(mode='list', length=nbp)
57
58 
    for (i in 1:nbp) {
59
60 
        df2plot = list(data=df_data[[i]],
61 
                       trend=df_trend[[i]],
62 
                       p_threshold=p_threshold[[i]],
63 
                       unit2day=unit2day[[i]],
64 
                       type=type[[i]],
65 
                       missRect=missRect[[i]])
66
67 
        list_df2plot[[i]] = df2plot
68 
    }
69
70 
    # print(list_df2plot)
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
outfile = "Panels" if (filename_opt != '') { outfile = paste(outfile, '_', filename_opt, sep='') } outfile = paste(outfile, '.pdf', sep='') # If there is not a dedicated figure directory it creats one outdir = file.path(figdir, filedir_opt, sep='') if (!(file.exists(outdir))) { dir.create(outdir) } outdirTmp = file.path(outdir, 'tmp') if (!(file.exists(outdirTmp))) { dir.create(outdirTmp) } # Get all different stations code Code = levels(factor(df_meta$code)) for (code in Code) { # Print code of the station for the current plotting print(paste("Plotting for sation :", code)) nbg = nbp+1 P = vector(mode='list', length=nbg) # P = as.list(rep(void, nbp)) # print(nbp) # print(P) gtext = text_panel(code, df_meta) P[[1]] = gtext for (i in 1:nbp) { df_data = list_df2plot[[i]]$data df_trend = list_df2plot[[i]]$trend p_threshold = list_df2plot[[i]]$p_threshold unit2day = list_df2plot[[i]]$unit2day missRect = list_df2plot[[i]]$missRect type = list_df2plot[[i]]$type p = time_panel(code, df_data, df_trend, missRect, p_threshold, unit2day, type) P[[i+1]] = p } # plot = grid.arrange(gtext, P[[1]], P[[2]], P[[3]], P[[4]], # heights=c(1/8, 1/8, 2/8, 2/8, 2/8), # ncol=1, nrow=5) plot = grid.arrange(grobs=P, heights=c(rep(1/((nbg-2)*2+2), 2), rep(2/((nbg-2)*2+2), nbg-2)), ncol=1, nrow=nbg) # Saving ggsave(plot=plot, path=outdirTmp, filename=paste(as.character(code), '.pdf', sep=''), width=21, height=29.7, units='cm', dpi=100) } pdf_combine(input=file.path(outdirTmp, list.files(outdirTmp)), output=file.path(outdir, outfile)) unlink(outdirTmp, recursive=TRUE)
141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210
} time_panel = function (code, df_data, df_trend, missRect, p_threshold, unit2day, type) { if (type == 'sqrt') { df_data[, 'Qm3s'] = apply(df_data[, 'Qm3s'], 1, sqrt) } df_data_code = df_data[df_data$code == code,] dDate = df_data_code$Date[length(df_data_code$Date)] - df_data_code$Date[1] datebreak = round(as.numeric(dDate) / unit2day / 11 , 0) p = ggplot() + theme_bw() + geom_line(aes(x=df_data_code$Date, y=df_data_code$Qm3s), color='black') if (missRect) { NAdate = df_data_code$Date[is.na(df_data_code$Qm3s)] dNAdate = diff(NAdate) NAdate_Down = NAdate[append(Inf, dNAdate) != 1] NAdate_Up = NAdate[append(dNAdate, Inf) != 1] p = p + geom_rect(aes(xmin=NAdate_Down, ymin=0, xmax=NAdate_Up, ymax=max(df_data_code$Qm3s, na.rm=TRUE)*1.1), linetype=0, fill='Wheat', alpha=0.3) } if (!is.null(df_trend)) { if (df_trend[df_trend$code == code,]$p < p_threshold) { abs = c(df_data_code$Date[1], df_data_code$Date[length(df_data_code$Date)]) abs_num = as.numeric(abs)/unit2day ord = abs_num * df_trend$trend[df_trend$code == code] + df_trend$intercept[df_trend$code == code] p = p + geom_line(aes(x=abs, y=ord), color='cornflowerblue') }} p = p + # ggtitle(paste(variable, 'station', # as.character(code), sep=' ')) + ylab(expression(paste('débit [', m^{3}, '.', s^{-1}, ']', sep=''))) + xlab('date') + scale_x_date(date_breaks=paste(as.character(datebreak), 'year', sep=' '), date_labels="%Y", limits=c(min(df_data_code$Date), max(df_data_code$Date)), expand=c(0, 0)) if (type == 'sqrt') { p = p + scale_y_continuous(breaks=seq(0, 100, 10), minor_breaks=seq(0, 100, 5),
211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274
limits=c(0, max(df_data_code$Qm3s, na.rm=TRUE)*1.1), expand=c(0, 0)) } else if (type == 'time') { p = p + scale_y_continuous(breaks=seq(0, 10000, 1000), minor_breaks=seq(0, 10000, 500), limits=c(0, max(df_data_code$Qm3s, na.rm=TRUE)*1.1), expand=c(0, 0)) } else { p = p + scale_y_continuous(limits=c(0, max(df_data_code$Qm3s, na.rm=TRUE)*1.1), expand=c(0, 0)) } p = p + theme( panel.background=element_rect(fill="white"), plot.margin=margin(0, 5, 0, 5, unit="mm")) return(p) } text_panel = function(code, df_meta) { df_meta_code = df_meta[df_meta$code == code,] text = paste( "<span style='font-size:18pt'> station <b>", code, "</b></span><br>", "nom : ", df_meta_code$nom, "<br>", "territoire : ", df_meta_code$territoire, "<br>", "position : (", df_meta_code$L93X, "; ", df_meta_code$L93Y, ")", "<br>", "surface : ", df_meta_code$surface_km2, " km<sup>2</sup>", sep='') gtext = richtext_grob(text, x=0, y=1, margin=unit(c(5, 5, 5, 5), "mm"), hjust=0, vjust=1, gp=gpar(col="grey20", fontsize=12)) return(gtext) } void = ggplot() + geom_blank(aes(1,1)) + theme( plot.background = element_blank(), panel.grid.major = element_blank(), panel.grid.minor = element_blank(), panel.border = element_blank(), panel.background = element_blank(), axis.title.x = element_blank(), axis.title.y = element_blank(), axis.text.x = element_blank(), axis.text.y = element_blank(), axis.ticks = element_blank(), axis.line = element_blank() )