draw_f020_gen_n.R

#!/usr/bin/env Rscript
# Tai Sakuma <sakuma@cern.ch>

##__________________________________________________________________||
argv <- commandArgs(trailingOnly = FALSE)
olddir <- setwd('mianRs')
source('drawFuncs.R')
source('drawThemes.R')
source('all_outputs_are_newer_than_any_input.R')
setwd(olddir)

##__________________________________________________________________||
scriptdir = dirname(substring(argv[grep("--file=", argv)], 8))

##__________________________________________________________________||
eval(readArgs)

##__________________________________________________________________||
library(tidyr, warn.conflicts = FALSE, quietly = TRUE)
library(dplyr, warn.conflicts = FALSE, quietly = TRUE)

##__________________________________________________________________||
theme.this <- function()
  {
    cols <- c("darkgreen", "orange", "#ff00ff", "#ff0000", "#0080ff", "#00ff00", "brown")
    col.regions <- colorRampPalette(brewer.pal(9, "Blues"))(100)
    theme <- list(
      add.text = list(cex = 0.8, lineheight = 2.0), # text in strip
      axis.text = list(cex = 0.8),
      axis.line = list(lwd = 0.2, col = 'gray30'),
      reference.line = list(col = '#eeeeee', lwd = 0.2),
      regions = list(col = col.regions),
      background = list(col = "transparent")
    )
    modifyList(theme.economist(), theme)
  }

##__________________________________________________________________||
arg.tbl.dir <- if(exists('arg.tbl.dir')) arg.tbl.dir else 'tbl_01'

##__________________________________________________________________||
main <- function()
{
  sub <- function(varname, xlim = NULL, adjust = NULL)
  {
    tblFileName <- paste('tbl_n_component.', varname, '-w', '.txt', sep = '')

    tblPath <- file.path(arg.tbl.dir, tblFileName)
    if(!(file.exists(tblPath))) return()

    tblCompPath <- file.path(scriptdir, 'tbl', 'tbl_component_src_particle_energy.txt')
    if(!(file.exists(tblCompPath))) return()

    fig.id <- mk.fig.id()

    figFileNameNoSuf <- paste(fig.id, varname, sep = '_')
    suffixes <- c('.pdf', '.png')
    figFileName <- outer(figFileNameNoSuf, suffixes, paste, sep = '')
    figPaths <- file.path(arg.outdir, figFileName)
    
    if(!arg.force)
    {
      outfile_paths <- figPaths
      infile_paths <- c(tblPath, tblCompPath)
      if(all_outputs_are_newer_than_any_input(outfile_paths, infile_paths)) return()
    }

    dir.create(arg.outdir, recursive = TRUE, showWarnings = FALSE)

    tbl <- read.table(tblPath, header = TRUE)
    colnames(tbl)[colnames(tbl) == varname] <- 'val'

    if(!is.null(adjust)) tbl$val <- adjust(tbl$val)

    ## to draw right side vertical line for the first entry
    tbl_ <- tbl %>% group_by(component) %>% summarise(val = min(val))
    tbl_$n <- 0
    tbl_$nvar <- 0
    tbl <- rbind(tbl_, tbl)

    ## add particle type and energy of the gun
    tbl_comp <- read.table(tblCompPath, header = TRUE)
    tbl <- merge(tbl, tbl_comp)
    tbl$src_energy <- factor(tbl$src_energy)

    theme <- theme.this()

    p <- draw_figure(tbl, varname, xlim)
    p <- useOuterStrips(p)
    
    print.figure(p, fig.id = figFileNameNoSuf, theme = theme, width = 4.5, height = 6)
  }

  sub('gen_eta')
  sub('gen_energy', xlim = c(0, 310))
  sub('gen_pdg')
  sub('gen_phi')
  invisible()
}

##__________________________________________________________________||
draw_figure <- function(tbl, varname, xlim = NULL)
{

  ##________________________________________________________________||
  prepanel <- function(x, y, type, subscripts, groups = NULL, ...)
    {
      ret = prepanel.default.xyplot(x, y, type, subscripts, groups, ...)
      if (!is.null(xlim)) ret[['xlim']] <- xlim
      ret
    }

  ##________________________________________________________________||
  panel <- function(x, y, subscripts, groups = NULL, ...)
    {
      lim <- current.panel.limits()
      if(length(x) == 0) return()
      panel.grid(h = -1, v = -1)
      panel.xyplot(x, y, subscripts = subscripts, groups = groups, type = 's', ...)
    }

  golden_ratio <- 1.61803398875
  ##________________________________________________________________||
  xyplot(n ~ val | src_particle*src_energy,
         data = tbl,
         xlab = varname,
         aspect = 1/golden_ratio,
         prepanel = prepanel,
         between = list(x = 0.2, y = 0.2),
         scales = list(
           x = list(alternating = '1'),
           y = list(alternating = '1')
           ),
         panel = panel
         )
}

##__________________________________________________________________||
main()