Newer
Older
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
#----------------------------------------------------------------------------
# Author: An-Chi Ho
# Date: 8th Oct 2021
#
# This script shows how to load irregular grid data by Start(), then regrid it
# by s2dv::CDORemap in the workflow. It is a solution before Start() can deal
# with irregular regridding directly.
#----------------------------------------------------------------------------
library(startR)
path <- paste0('/esarchive/exp/CMIP6/dcppA-hindcast/cmcc-cm2-sr5/cmip6-dcppA-hindcast_i1p1/',
'DCPP/CMCC/CMCC-CM2-SR5/dcppA-hindcast/$member$/Omon/$var$/gn/v20210312/',
'$var$_*_s$sdate$-$member$_gn_$aux$.nc')
data <- Start(dataset = path,
var = 'tos',
sdate = c('1960', '1961'),
aux = 'all',
aux_depends = 'sdate',
j = indices(2:361), # remove two indices to avoid white strips
i = indices(2:291), # remove two indices to avoid white strips
time = indices(1:12),
member = 'r1i1p1f1',
return_vars = list(j = NULL, i = NULL,
latitude = NULL, longitude = NULL),
retrieve = F)
func_regrid <- function(data) {
lons <- attr(data, 'Variables')$common$longitude
lats <- attr(data, 'Variables')$common$latitude
data <- s2dv::CDORemap(data, lons, lats, grid = 'r360x180',
method = 'bil', crop = FALSE)
lons_reg <- data[['lons']]
lats_reg <- data[['lats']]
return(list(data = data[[1]], lats = lats_reg, lons = lons_reg))
}
step <- Step(fun = func_regrid,
target_dims = list(data = c('j', 'i')),
output_dims = list(data = c('lon', 'lat'),
lats = 'lat', lons = 'lon'),
use_attributes = list(data = "Variables"))
wf <- AddStep(data, step)
res <- Compute(workflow = wf$data,
chunks = list(sdate = 2, time = 2))
names(res)
#[1] "data" "lats" "lons"
dim(res$data)
# lat lon dataset var sdate aux time member
# 180 360 1 1 2 1 12 1
dim(res$lons)
# lon dataset var sdate aux time member
# 360 1 1 2 1 12 1
dim(res$lats)
# lat dataset var sdate aux time member
# 180 1 1 2 1 12 1
s2dv::PlotEquiMap(drop(res$data)[ , , 1, 1],
lon = drop(res$lons)[, 1, 1],
lat = drop(res$lats)[, 1, 1])
# Plot Layout for sdate = 1 all the time steps
var <- s2dv:::.aperm2(drop(res$data)[, , 1, ], c(3, 1, 2))
s2dv::PlotLayout(PlotEquiMap, c('lon', 'lat'), var = var,
lon = drop(res$lons)[, 1, 1], lat = drop(res$lats)[, 1, 1])