!---------------------------------------------------------------------- ! NEMO system team, System and Interface for oceanic RElocable Nesting !---------------------------------------------------------------------- ! ! ! PROGRAM: create_restart ! ! DESCRIPTION: !> @brief !> This program create restart file. !> !> @details !> Variables are read from restart file, or standard output. !> Then theses variables are interpolated on fine grid. !> Finally table are split over new decomposition. !> !> @author !> J.Paul ! REVISION HISTORY: !> @date Nov, 2013 - Initial Version ! !> @note Software governed by the CeCILL licence (NEMOGCM/NEMO_CeCILL.txt) !> !> @todo !> - add attributes indices and refinement in output file !> - check fileout exist at the beginning !---------------------------------------------------------------------- !> @code PROGRAM create_restart ! USE netcdf ! nf90 library USE global ! global variable USE kind ! F90 kind parameter USE logger ! log file manager USE fct ! basic useful function USE date ! date manager USE att ! attribute manager USE dim ! dimension manager USE var ! variable manager USE file ! file manager USE multi ! multi file manager USE iom ! I/O manager USE dom ! domain manager USE grid ! grid manager USE vgrid ! vertical grid manager USE extrap ! extrapolation manager USE interp ! interpolation manager USE filter ! filter manager USE mpp ! MPP manager USE iom_mpp ! MPP I/O manager IMPLICIT NONE ! local variable CHARACTER(LEN=lc) :: cl_namelist CHARACTER(LEN=lc) :: cl_date CHARACTER(LEN=lc) :: cl_name CHARACTER(LEN=lc) :: cl_data INTEGER(i4) :: il_narg INTEGER(i4) :: il_status INTEGER(i4) :: il_fileid INTEGER(i4) :: il_attid INTEGER(i4) :: il_imin0 INTEGER(i4) :: il_imax0 INTEGER(i4) :: il_jmin0 INTEGER(i4) :: il_jmax0 INTEGER(i4) , DIMENSION(ip_maxdim) :: il_rho INTEGER(i4) , DIMENSION(2) :: il_xghost INTEGER(i4) , DIMENSION(2,2) :: il_offset INTEGER(i4) , DIMENSION(2,2,2) :: il_ind LOGICAL :: ll_exist TYPE(TFILE) :: tl_coord0 TYPE(TFILE) :: tl_coord1 TYPE(TFILE) :: tl_bathy1 TYPE(TFILE) :: tl_file TYPE(TDOM) :: tl_dom1 TYPE(TDOM) :: tl_dom0 TYPE(TATT) :: tl_att TYPE(TVAR) :: tl_depth TYPE(TVAR) :: tl_time TYPE(TVAR) :: tl_lon TYPE(TVAR) :: tl_lat TYPE(TVAR) :: tl_tmp TYPE(TVAR) , DIMENSION(:) , ALLOCATABLE :: tl_var TYPE(TVAR) , DIMENSION(:) , ALLOCATABLE :: tl_level TYPE(TDIM) , DIMENSION(ip_maxdim) :: tl_dim TYPE(TMPP) :: tl_mpp TYPE(TMPP) :: tl_mppout TYPE(TMULTI) :: tl_multi ! loop indices INTEGER(i4) :: ji INTEGER(i4) :: jj INTEGER(i4) :: jvar ! namelist variable CHARACTER(LEN=lc) :: cn_logfile = 'create_restart.log' CHARACTER(LEN=lc) :: cn_verbosity = 'warning' CHARACTER(LEN=lc) :: cn_varcfg = 'variable.cfg' CHARACTER(LEN=lc) :: cn_coord0 = '' INTEGER(i4) :: in_perio0 = -1 CHARACTER(LEN=lc) :: cn_coord1 = '' CHARACTER(LEN=lc) :: cn_bathy1 = '' INTEGER(i4) :: in_perio1 = -1 INTEGER(i4) :: in_extrap = 0 LOGICAL :: ln_fillclosed = .TRUE. CHARACTER(LEN=lc), DIMENSION(ig_maxvar) :: cn_varinfo = '' CHARACTER(LEN=lc), DIMENSION(ig_maxvar) :: cn_varfile = '' INTEGER(i4) :: in_imin0 = 0 INTEGER(i4) :: in_imax0 = 0 INTEGER(i4) :: in_jmin0 = 0 INTEGER(i4) :: in_jmax0 = 0 INTEGER(i4) :: in_rhoi = 0 INTEGER(i4) :: in_rhoj = 0 CHARACTER(LEN=lc) :: cn_fileout = 'restart' INTEGER(i4) :: in_nproc = 0 INTEGER(i4) :: in_niproc = 0 INTEGER(i4) :: in_njproc = 0 CHARACTER(LEN=lc) :: cn_type = '' !------------------------------------------------------------------- NAMELIST /namlog/ & !< logger namelist & cn_logfile, & !< log file & cn_verbosity !< log verbosity NAMELIST /namcfg/ & !< configuration namelist & cn_varcfg !< variable configuration file NAMELIST /namcrs/ & !< coarse grid namelist & cn_coord0, & !< coordinate file & in_perio0 !< periodicity index NAMELIST /namfin/ & !< fine grid namelist & cn_coord1, & !< coordinate file & cn_bathy1, & !< bathymetry file & in_perio1, & !< periodicity index & in_extrap, & !< & ln_fillclosed !< fill closed sea NAMELIST /namvar/ & !< variable namelist & cn_varinfo, & !< list of variable and interpolation method to be used. (ex: 'votemper:linear','vosaline:cubic' ) & cn_varfile !< list of variable file NAMELIST /namnst/ & !< nesting namelist & in_imin0, & !< i-direction lower left point indice & in_imax0, & !< i-direction upper right point indice & in_jmin0, & !< j-direction lower left point indice & in_jmax0, & !< j-direction upper right point indice & in_rhoi, & !< refinement factor in i-direction & in_rhoj !< refinement factor in j-direction NAMELIST /namout/ & !< output namlist & cn_fileout, & !< fine grid bathymetry file & in_nproc, & !< number of processor to be used & in_niproc, & !< i-direction number of processor & in_njproc, & !< j-direction numebr of processor & cn_type !< output type format (dimg, cdf) !------------------------------------------------------------------- !1- namelist !1-1 get namelist il_narg=COMMAND_ARGUMENT_COUNT() !f03 intrinsec IF( il_narg/=1 )THEN PRINT *,"ERROR in create_restart: need a namelist" STOP ELSE CALL GET_COMMAND_ARGUMENT(1,cl_namelist) !f03 intrinsec ENDIF !1-2 read namelist INQUIRE(FILE=TRIM(cl_namelist), EXIST=ll_exist) IF( ll_exist )THEN il_fileid=fct_getunit() OPEN( il_fileid, FILE=TRIM(cl_namelist), & & FORM='FORMATTED', & & ACCESS='SEQUENTIAL', & & STATUS='OLD', & & ACTION='READ', & & IOSTAT=il_status) CALL fct_err(il_status) IF( il_status /= 0 )THEN PRINT *,"ERROR in create_restart: error opening "//TRIM(cl_namelist) STOP ENDIF READ( il_fileid, NML = namlog ) !1-2-1 define log file CALL logger_open(TRIM(cn_logfile),TRIM(cn_verbosity)) CALL logger_header() READ( il_fileid, NML = namcfg ) !1-2-2 get variable extra information CALL var_def_extra(TRIM(cn_varcfg)) READ( il_fileid, NML = namcrs ) READ( il_fileid, NML = namfin ) READ( il_fileid, NML = namvar ) !1-2-3 add user change in extra information CALL var_chg_extra(cn_varinfo) !1-2-4 match variable with file tl_multi=multi_init(cn_varfile) READ( il_fileid, NML = namnst ) READ( il_fileid, NML = namout ) CLOSE( il_fileid, IOSTAT=il_status ) CALL fct_err(il_status) IF( il_status /= 0 )THEN CALL logger_error("CREATE RESTART: closing "//TRIM(cl_namelist)) ENDIF ELSE PRINT *,"ERROR in create_restart: can't find "//TRIM(cl_namelist) ENDIF !2- open files IF( cn_coord0 /= '' )THEN tl_coord0=file_init(TRIM(cn_coord0),id_perio=in_perio0) CALL iom_open(tl_coord0) ELSE CALL logger_fatal("CREATE RESTART: no coarse grid coordinate found. "//& & "check namelist") ENDIF IF( TRIM(cn_coord1) /= '' )THEN tl_coord1=file_init(TRIM(cn_coord1),id_perio=in_perio1) CALL iom_open(tl_coord1) ELSE CALL logger_fatal("CREATE RESTART: no fine grid coordinate found. "//& & "check namelist") ENDIF IF( TRIM(cn_bathy1) /= '' )THEN tl_bathy1=file_init(TRIM(cn_bathy1),id_perio=in_perio1) CALL iom_open(tl_bathy1) ELSE CALL logger_fatal("CREATE RESTART: no fine grid bathymetry found. "//& & "check namelist") ENDIF !3- check !3-2-1 check refinement factor il_rho(:)=1 IF( in_rhoi < 1 .OR. in_rhoj < 1 )THEN CALL logger_error("CREATE RESTART: invalid refinement factor."//& & " check namelist "//TRIM(cl_namelist)) ELSE il_rho(jp_I)=in_rhoi il_rho(jp_J)=in_rhoj ENDIF IF( cn_coord0 /= '' )THEN !.OR. cn_bathy0 /= '' )THEN !3-1 check namelist IF( in_imin0 < 1 .OR. in_imax0 < 1 .OR. in_jmin0 < 1 .OR. in_jmax0 < 1)THEN ! compute coarse grid indices around fine grid IF( cn_coord0 /= '' )THEN il_ind(:,:,:)=grid_get_coarse_index(tl_coord0, tl_coord1 ) ENDIF il_imin0=il_ind(1,1,1) ; il_imax0=il_ind(1,2,1) il_jmin0=il_ind(2,1,1) ; il_jmax0=il_ind(2,2,1) il_offset(:,:)=il_ind(:,:,2) ELSE il_imin0=in_imin0 ; il_imax0=in_imax0 il_jmin0=in_jmin0 ; il_jmax0=in_jmax0 il_offset(1,:)=FLOOR(REAL(il_rho(jp_I)-1,dp)*0.5) il_offset(2,:)=FLOOR(REAL(il_rho(jp_J)-1,dp)*0.5) ENDIF !3-2 check domain validity IF( cn_coord0 /= '' )THEN CALL grid_check_dom(tl_coord0, il_imin0, il_imax0, il_jmin0, il_jmax0) ENDIF !3-3 check coordinate file IF( cn_coord0 /= '' )THEN CALL grid_check_coincidence( tl_coord0, tl_coord1, & & il_imin0, il_imax0, & & il_jmin0, il_jmax0, & & il_rho(:) ) ENDIF ENDIF ! compute level ALLOCATE(tl_level(ig_npoint)) tl_level(:)=vgrid_get_level(tl_bathy1, cl_namelist ) ! remove ghost cell il_xghost(:)=grid_get_ghost(tl_bathy1) DO ji=1,ig_npoint CALL grid_del_ghost(tl_level(ji), il_xghost(1), il_xghost(2)) ENDDO ! close CALL iom_close(tl_bathy1) !4- work on variables IF( .NOT. ASSOCIATED(tl_multi%t_file) )THEN CALL logger_error("CREATE RESTART: no file to work on. "//& & "check cn_varfile in namelist.") ELSE ALLOCATE( tl_var( tl_multi%i_nvar ) ) jvar=0 ! for each file DO ji=1,tl_multi%i_nfile WRITE(cl_data,'(a,i2.2)') 'data_',jvar+1 IF( .NOT. ASSOCIATED(tl_multi%t_file(ji)%t_var) )THEN CALL logger_error("CREATE RESTART: no variable to work on for "//& & "file "//TRIM(tl_multi%t_file(ji)%c_name)//& & ". check cn_varfile in namelist.") ELSEIF( TRIM(tl_multi%t_file(ji)%c_name) == TRIM(cl_data) )THEN !4-1 use input matrix to fill variable ! for each variable initialise from matrix DO jj=1,tl_multi%t_file(ji)%i_nvar jvar=jvar+1 tl_tmp=tl_multi%t_file(ji)%t_var(jj) !4-1-1 fill value with matrix data ! pb voir comment gerer nb de dimension tl_var(jvar)=create_restart_matrix(tl_tmp, tl_coord1) !4-1-2 use mask CALL create_restart_mask(tl_var(jvar), tl_level(:)) ENDDO ELSE !4-2 use file to fill variable ! open file tl_file=file_init(TRIM(tl_multi%t_file(ji)%c_name)) CALL iom_open(tl_file) ! get or check depth value IF( tl_file%i_depthid /= 0 )THEN IF( ASSOCIATED(tl_depth%d_value) )THEN IF( ANY( tl_depth%d_value(:,:,:,:) /= & & tl_tmp%d_value(:,:,:,:) ) )THEN CALL logger_fatal("CREATE RESTART: depth value from "//& & TRIM(tl_multi%t_file(ji)%c_name)//" not conform "//& & " to those from former file(s).") ENDIF ELSE tl_depth=iom_read_var(tl_file,tl_file%i_depthid) ENDIF ENDIF ! get or check time value IF( tl_file%i_timeid /= 0 )THEN IF( ASSOCIATED(tl_time%d_value) )THEN IF( ANY( tl_time%d_value(:,:,:,:) /= & & tl_tmp%d_value(:,:,:,:) ) )THEN CALL logger_fatal("CREATE RESTART: time value from "//& & TRIM(tl_multi%t_file(ji)%c_name)//" not conform "//& & " to those from former file(s).") ENDIF ELSE tl_time=iom_read_var(tl_file,tl_file%i_timeid) ENDIF ENDIF IF( ANY( tl_file%t_dim(1:2)%i_len /= & & tl_coord0%t_dim(1:2)%i_len) )THEN !4-2-1 extract value from fine grid !4-2-1-1 compute domain on fine grid tl_dom1=create__restart_get_dom_coord(tl_file, tl_coord1) ! open mpp file on domain !4-2-1-2 init mpp structure tl_mpp=mpp_init(tl_file) !4-2-1-3 get processor to be used CALL mpp_get_use( tl_mpp, tl_dom1 ) !4-2-1-4 open mpp files CALL iom_mpp_open(tl_mpp) ! for each variable of this file DO jj=1,tl_multi%t_file(ji)%i_nvar jvar=jvar+1 cl_name=tl_multi%t_file(ji)%t_var(jj)%c_name !4-2-1-5 read variable over domain tl_var(jvar)=iom_mpp_read_var( tl_mpp, TRIM(cl_name), & & td_dom=tl_dom1 ) !4-2-1-7 add attribute to variable tl_att=att_init('src_file',TRIM(fct_basename(tl_mpp%c_name))) CALL var_move_att(tl_var(jvar), tl_att) ! clean structure CALL att_clean(tl_att) !4-2-1-8 use mask CALL create_restart_mask(tl_var(jvar), tl_level(:)) !4-2-1-9 add ghost cell CALL grid_add_ghost( tl_var(jvar), & & tl_dom1%i_ighost,tl_dom1%i_jghost ) ENDDO !4-2-1-2 close mpp file CALL iom_mpp_close(tl_mpp) ! clean structure CALL mpp_clean(tl_mpp) CALL dom_clean(tl_dom1) ELSE !4-2-2 get value from coarse grid !4-2-2-1 compute domain on coarse grid tl_dom0=create__restart_get_dom_index(tl_file, il_imin0, il_jmin0, & & il_imax0, il_jmax0) !4-2-2-2 add extra band (if possible) to compute interpolation CALL dom_add_extra(tl_dom0) ! open mpp file on domain !4-2-2-3 init mpp structure tl_mpp=mpp_init(tl_file) !4-2-2-4 get processor to be used CALL mpp_get_use( tl_mpp, tl_dom0 ) !4-2-2-5 open mpp files CALL iom_mpp_open(tl_mpp) ! for each variable of this file DO jj=1,tl_multi%t_file(ji)%i_nvar jvar=jvar+1 cl_name=tl_multi%t_file(ji)%t_var(jj)%c_name print *,'work on ',trim(cl_name) !4-2-2-6 read variable over domain tl_var(jvar)=iom_mpp_read_var( tl_mpp, TRIM(cl_name), & & td_dom=tl_dom0 ) !4-2-2-7 interpolate variable CALL create_restart_interp(tl_var(jvar), tl_level(:), & & il_rho(:), & & id_offset=il_offset(:,:)) !tl_att=att_init('add_offset',0.) !CALL var_move_att(tl_var(jvar), tl_att) !tl_att=att_init('scale_factor',1.) !CALL var_move_att(tl_var(jvar), tl_att) !4-2-2-8 remove extraband added to domain CALL dom_del_extra( tl_var(jvar), tl_dom0, il_rho(:) ) !4-2-2-10 add attribute to variable tl_att=att_init('src_file',TRIM(fct_basename(tl_mpp%c_name))) CALL var_move_att(tl_var(jvar), tl_att) ! clean structure CALL att_clean(tl_att) !4-2-2-11 use mask CALL create_restart_mask(tl_var(jvar), tl_level(:)) !4-2-2-12 add ghost cell CALL grid_add_ghost( tl_var(jvar), & & tl_dom0%i_ighost,tl_dom0%i_jghost ) ENDDO !4-2-2-2 close mpp file CALL iom_mpp_close(tl_mpp) ! clean structure CALL mpp_clean(tl_mpp) CALL dom_clean(tl_dom0) ENDIF ! close file CALL iom_close(tl_file) ! clean structure CALL file_clean(tl_file) ENDIF ENDDO ENDIF !5- use additional request DO jvar=1,tl_multi%i_nvar !5-1 forced min and max value CALL var_limit_value(tl_var(jvar)) !5-2 filter CALL filter_fill_value(tl_var(jvar)) !5-3 extrapolate CALL extrap_fill_value(tl_var(jvar), id_iext=in_extrap, & & id_jext=in_extrap, & & id_kext=in_extrap) ENDDO !6- create file IF( in_niproc == 0 .AND. & & in_njproc == 0 .AND. & & in_nproc == 0 )THEN in_niproc = 1 in_njproc = 1 in_nproc = 1 ENDIF tl_mppout=mpp_init( TRIM(cn_fileout), tl_var(1), & & in_niproc, in_njproc, in_nproc, & & cd_type=cn_type) !6-1 add dimension tl_dim(:)=var_max_dim(tl_var(:)) DO ji=1,ip_maxdim IF( tl_dim(ji)%l_use )THEN CALL mpp_move_dim(tl_mppout, tl_dim(ji)) SELECT CASE(TRIM(tl_dim(ji)%c_sname)) CASE('z','t') DO jj=1,tl_mppout%i_nproc CALL file_add_dim(tl_mppout%t_proc(jj), tl_dim(ji)) ENDDO END SELECT ENDIF ENDDO !6-2 add variables !IF( ALL( tl_dim(1:2)%l_use ) )THEN ! ! add longitude ! tl_lon=iom_read_var(tl_coord1,'longitude') ! CALL mpp_add_var(tl_mppout, tl_lon) ! CALL var_clean(tl_lon) ! ! add latitude ! tl_lat=iom_read_var(tl_coord1,'latitude') ! CALL mpp_add_var(tl_mppout, tl_lat) ! CALL var_clean(tl_lat) !ENDIF !IF( tl_dim(3)%l_use )THEN ! ! add depth ! CALL mpp_add_var(tl_mppout, tl_depth) ! CALL var_clean(tl_depth) !ENDIF !IF( tl_dim(4)%l_use )THEN ! ! add time ! CALL mpp_add_var(tl_mppout, tl_time) ! CALL var_clean(tl_time) !ENDIF ! add other variable DO jvar=1,tl_multi%i_nvar CALL mpp_add_var(tl_mppout, tl_var(jvar)) CALL var_clean(tl_var(jvar)) ENDDO DO ji=1,4 CALL mpp_add_var(tl_mppout,tl_level(ji)) ENDDO !6-3 add some attribute tl_att=att_init("Created_by","SIREN create_restart") CALL mpp_add_att(tl_mppout, tl_att) cl_date=date_print(date_now()) tl_att=att_init("Creation_date",TRIM(cl_date)) CALL mpp_add_att(tl_mppout, tl_att) ! add attribute periodicity il_attid=0 IF( ASSOCIATED(tl_mppout%t_proc(1)%t_att) )THEN il_attid=att_get_id(tl_mppout%t_proc(1)%t_att(:),'periodicity') ENDIF IF( tl_coord1%i_perio >= 0 .AND. il_attid == 0 )THEN tl_att=att_init('periodicity',tl_coord1%i_perio) CALL mpp_add_att(tl_mppout,tl_att) ENDIF il_attid=0 IF( ASSOCIATED(tl_mppout%t_proc(1)%t_att) )THEN il_attid=att_get_id(tl_mppout%t_proc(1)%t_att(:),'ew_overlap') ENDIF IF( tl_coord1%i_ew >= 0 .AND. il_attid == 0 )THEN tl_att=att_init('ew_overlap',tl_coord1%i_ew) CALL mpp_add_att(tl_mppout,tl_att) ENDIF !6-4 create file CALL iom_mpp_create(tl_mppout) !6-5 write file CALL iom_mpp_write_file(tl_mppout) !6-6 close file CALL iom_mpp_close(tl_mppout) IF( cn_coord0 /= '' ) CALL iom_close(tl_coord0) !7- clean DEALLOCATE(tl_var) CALL mpp_clean(tl_mppout) CALL file_clean(tl_coord1) CALL file_clean(tl_coord0) ! close log file CALL logger_footer() CALL logger_close() !> @endcode CONTAINS !------------------------------------------------------------------- !> @brief !> !> @details !> !> @author J.Paul !> - Nov, 2013- Initial Version !> !------------------------------------------------------------------- !> @code FUNCTION create_restart_level(td_level1) IMPLICIT NONE ! Argument TYPE(TFILE), INTENT(IN) :: td_level1 ! function TYPE(TVAR), DIMENSION(4) :: create_restart_level ! local variable TYPE(TFILE) :: tl_level1 TYPE(TVAR), DIMENSION(4) :: tl_var TYPE(TMPP) :: tl_mpplevel1 ! loop indices !---------------------------------------------------------------- !0- compute domain tl_dom1=dom_init(td_level1) !1 init mpp structure tl_level1=td_level1 tl_mpplevel1=mpp_init(tl_level1) CALL file_clean(tl_level1) !2 get processor to be used CALL mpp_get_use( tl_mpplevel1, tl_dom1 ) !3 open mpp files CALL iom_mpp_open(tl_mpplevel1) tl_var(jp_T)=iom_mpp_read_var(tl_mpplevel1,'tlevel',td_dom=tl_dom1) tl_var(jp_U)=iom_mpp_read_var(tl_mpplevel1,'ulevel',td_dom=tl_dom1) tl_var(jp_V)=iom_mpp_read_var(tl_mpplevel1,'vlevel',td_dom=tl_dom1) tl_var(jp_F)=iom_mpp_read_var(tl_mpplevel1,'flevel',td_dom=tl_dom1) !4 save result create_restart_level(:)=tl_var(:) !5 clean CALL iom_mpp_close(tl_mpplevel1) CALL mpp_clean(tl_mpplevel1) END FUNCTION create_restart_level !> @endcode !------------------------------------------------------------------- !> @brief !> This function create variable, filled with matrix value !> !> @details !> A variable is create with the same name that the input variable, !> and with dimension of the coordinate file. !> Then the variable table of value is split into equal subdomain. !> Each subdomain is fill with the linked value of the matrix. !> !> @author J.Paul !> - Nov, 2013- Initial Version !> !> @param[in] td_var : variable structure !> @param[in] td_coord : coordinate !> @return variable structure !------------------------------------------------------------------- !> @code FUNCTION create_restart_matrix(td_var, td_coord) IMPLICIT NONE ! Argument TYPE(TVAR) , INTENT(IN) :: td_var TYPE(TFILE), INTENT(IN) :: td_coord ! function TYPE(TVAR) :: create_restart_matrix ! local variable INTEGER(i4) :: il_ighost INTEGER(i4) :: il_jghost INTEGER(i4) , DIMENSION(2) :: il_xghost INTEGER(i4) , DIMENSION(3) :: il_dim INTEGER(i4) , DIMENSION(3) :: il_size INTEGER(i4) , DIMENSION(3) :: il_rest INTEGER(i4) , DIMENSION(:) , ALLOCATABLE :: il_ishape INTEGER(i4) , DIMENSION(:) , ALLOCATABLE :: il_jshape INTEGER(i4) , DIMENSION(:) , ALLOCATABLE :: il_kshape REAL(dp) , DIMENSION(:,:,:,:), ALLOCATABLE :: dl_value TYPE(TVAR) :: tl_lon TYPE(TVAR) :: tl_lat TYPE(TVAR) :: tl_var TYPE(TDIM) , DIMENSION(ip_maxdim) :: tl_dim ! loop indices INTEGER(i4) :: ji INTEGER(i4) :: jj INTEGER(i4) :: jk !---------------------------------------------------------------- !1- read output grid tl_lon=iom_read_var(td_coord,'longitude') tl_lat=iom_read_var(td_coord,'latitude') !2- look for ghost cell il_xghost(:)=grid_get_ghost( tl_lon, tl_lat ) il_ighost=il_xghost(1)*ig_ghost il_jghost=il_xghost(2)*ig_ghost !3- write value on grid !3-1 get matrix dimension il_dim(:)=td_var%t_dim(1:3)%i_len !3-2 output dimension tl_dim(:)=tl_lon%t_dim(:) ! remove ghost cell tl_dim(1)%i_len=tl_dim(1)%i_len - 2*il_xghost(1)*ig_ghost tl_dim(2)%i_len=tl_dim(2)%i_len - 2*il_xghost(2)*ig_ghost !3-3 split output domain in N subdomain depending of matrix dimension il_size(:) = tl_dim(1:3)%i_len / il_dim(:) il_rest(:) = MOD(tl_dim(1:3)%i_len, il_dim(:)) ALLOCATE( il_ishape(il_dim(1)+1) ) il_ishape(:)=0 DO ji=2,il_dim(1)+1 il_ishape(ji)=il_ishape(ji-1)+il_size(1) ENDDO ! add rest to last cell il_ishape(il_dim(1)+1)=il_ishape(il_dim(1)+1)+il_rest(1) ALLOCATE( il_jshape(il_dim(2)+1) ) il_jshape(:)=0 DO jj=2,il_dim(2)+1 il_jshape(jj)=il_jshape(jj-1)+il_size(2) ENDDO ! add rest to last cell il_jshape(il_dim(2)+1)=il_jshape(il_dim(2)+1)+il_rest(2) ALLOCATE( il_kshape(il_dim(3)+1) ) il_kshape(:)=0 DO jk=2,il_dim(3)+1 il_kshape(jk)=il_kshape(jk-1)+il_size(3) ENDDO ! add rest to last cell il_kshape(il_dim(3)+1)=il_kshape(il_dim(3)+1)+il_rest(3) !3-3 write ouput table of value ALLOCATE(dl_value( tl_dim(1)%i_len, & & tl_dim(2)%i_len, & & tl_dim(3)%i_len, & & tl_dim(4)%i_len) ) dl_value(:,:,:,:)=0 DO jk=2,il_dim(3)+1 DO jj=2,il_dim(2)+1 DO ji=2,il_dim(1)+1 dl_value( 1+il_ishape(ji-1):il_ishape(ji), & & 1+il_jshape(jj-1):il_jshape(jj), & & 1+il_kshape(jk-1):il_kshape(jk), & & 1 ) = td_var%d_value(ji-1,jj-1,jk-1,1) ENDDO ENDDO ENDDO !3-4 initialise variable with value tl_var=var_init(TRIM(td_var%c_name),dl_value(:,:,:,:)) DEALLOCATE(dl_value) !4- add ghost cell CALL grid_add_ghost(tl_var,il_ighost,il_jghost) !5- save result create_restart_matrix=tl_var END FUNCTION create_restart_matrix !> @endcode !------------------------------------------------------------------- !> @brief !> !> @details !> !> @author J.Paul !> - Nov, 2013- Initial Version !> !------------------------------------------------------------------- !> @code FUNCTION create__restart_get_dom_coord( td_file, td_coord, & & id_rho ) IMPLICIT NONE ! Argument TYPE(TFILE), INTENT(IN) :: td_file TYPE(TFILE), INTENT(IN) :: td_coord INTEGER(i4), DIMENSION(:), INTENT(IN), OPTIONAL :: id_rho ! function TYPE(TDOM) :: create__restart_get_dom_coord ! local variable INTEGER(i4) :: il_pivot INTEGER(i4) :: il_perio INTEGER(i4) :: il_imin INTEGER(i4) :: il_imax INTEGER(i4) :: il_jmin INTEGER(i4) :: il_jmax INTEGER(i4), DIMENSION(2,2,2) :: il_ind TYPE(TFILE) :: tl_file TYPE(TDOM) :: tl_dom ! loop indices !---------------------------------------------------------------- tl_file=td_file !1- open file CALL iom_open(tl_file) ! get periodicity il_pivot=grid_get_pivot(tl_file) il_perio=grid_get_perio(tl_file,il_pivot) tl_file%i_perio=il_perio !2- compute file grid indices around coord grid il_ind(:,:,:)=grid_get_coarse_index(tl_file, td_coord ) il_imin=il_ind(1,1,1) ; il_imax=il_ind(1,2,1) il_jmin=il_ind(2,1,1) ; il_jmax=il_ind(2,2,1) !3- check grid coincidence CALL grid_check_coincidence( tl_file, td_coord, & & il_imin, il_imax, & & il_jmin, il_jmax, & & id_rho(:) ) !4- compute domain tl_dom=dom_init(tl_file, & & il_imin, il_imax, & & il_jmin, il_jmax) ! close file CALL iom_close(tl_file) ! save result create__restart_get_dom_coord=tl_dom END FUNCTION create__restart_get_dom_coord !> @endcode !------------------------------------------------------------------- !> @brief !> !> @details !> !> @author J.Paul !> - Nov, 2013- Initial Version !> !------------------------------------------------------------------- !> @code FUNCTION create__restart_get_dom_index( td_file, id_imin, id_jmin, & & id_imax, id_jmax ) IMPLICIT NONE ! Argument TYPE(TFILE), INTENT(IN) :: td_file INTEGER(i4), INTENT(IN) :: id_imin INTEGER(i4), INTENT(IN) :: id_imax INTEGER(i4), INTENT(IN) :: id_jmin INTEGER(i4), INTENT(IN) :: id_jmax ! function TYPE(TDOM) :: create__restart_get_dom_index ! local variable INTEGER(i4) :: il_pivot INTEGER(i4) :: il_perio TYPE(TFILE) :: tl_file TYPE(TDOM) :: tl_dom ! loop indices !---------------------------------------------------------------- ! init tl_file=td_file !1- open file CALL iom_open(tl_file) ! get periodicity il_pivot=grid_get_pivot(tl_file) il_perio=grid_get_perio(tl_file,il_pivot) tl_file%i_perio=il_perio !2- compute domain tl_dom=dom_init(tl_file, & & id_imin, id_imax, & & id_jmin, id_jmax) ! close file CALL iom_close(tl_file) ! save result create__restart_get_dom_index=tl_dom END FUNCTION create__restart_get_dom_index !> @endcode !------------------------------------------------------------------- !> @brief !> This subroutine !> !> @details !> !> @author J.Paul !> - Nov, 2013- Initial Version !> !> @param[in] !> @todo !------------------------------------------------------------------- !> @code SUBROUTINE create_restart_mask( td_var, td_mask ) IMPLICIT NONE ! Argument TYPE(TVAR) , INTENT(INOUT) :: td_var TYPE(TVAR), DIMENSION(:), INTENT(IN ) :: td_mask ! local variable INTEGER(i4), DIMENSION(:,:), ALLOCATABLE :: il_mask ! loop indices INTEGER(i4) :: jl INTEGER(i4) :: jk !---------------------------------------------------------------- IF( ANY(td_var%t_dim(1:2)%i_len /= td_mask(1)%t_dim(1:2)%i_len) )THEN CALL logger_error("CREATE RESTART MASK: dimension differ between "//& & "variable ("//& & TRIM(fct_str(td_var%t_dim(1)%i_len))//","//& & TRIM(fct_str(td_var%t_dim(2)%i_len))//& & ") and level ("//& & TRIM(fct_str(td_mask(1)%t_dim(1)%i_len))//","//& & TRIM(fct_str(td_mask(1)%t_dim(2)%i_len))//")") ELSE ALLOCATE( il_mask(td_var%t_dim(1)%i_len, & & td_var%t_dim(2)%i_len) ) SELECT CASE(TRIM(td_var%c_point)) CASE('T') il_mask(:,:)=INT(td_mask(jp_T)%d_value(:,:,1,1)) CASE('U') il_mask(:,:)=INT(td_mask(jp_U)%d_value(:,:,1,1)) CASE('V') il_mask(:,:)=INT(td_mask(jp_V)%d_value(:,:,1,1)) CASE('F') il_mask(:,:)=INT(td_mask(jp_F)%d_value(:,:,1,1)) END SELECT DO jl=1,td_var%t_dim(4)%i_len DO jk=1,td_var%t_dim(3)%i_len WHERE( il_mask(:,:) < jk ) td_var%d_value(:,:,jk,jl)=99 !td_var%d_fill !.AND. & !& td_var%d_value(:,:,jk,jl) == td_var%d_fill .OR. & !& il_mask(:,:) < jk .AND. & !& td_var%d_value(:,:,jk,jl) == 1 ) td_var%d_value(:,:,jk,jl)=99 !td_var%d_fill ENDDO ENDDO DEALLOCATE( il_mask ) ENDIF END SUBROUTINE create_restart_mask !> @endcode !------------------------------------------------------------------- !> @brief !> This subroutine !> !> @details !> !> @author J.Paul !> - Nov, 2013- Initial Version !> !> @param[in] !------------------------------------------------------------------- !> @code FUNCTION create_restart_extract(td_var, td_file, & & td_coord) IMPLICIT NONE ! Argument TYPE(TVAR) , INTENT(IN) :: td_var TYPE(TFILE), INTENT(IN) :: td_file TYPE(TFILE), INTENT(IN) :: td_coord ! function TYPE(TVAR) :: create_restart_extract ! local variable INTEGER(i4), DIMENSION(2,2,2) :: il_ind INTEGER(i4) :: il_pivot INTEGER(i4) :: il_perio INTEGER(i4) :: il_imin INTEGER(i4) :: il_jmin INTEGER(i4) :: il_imax INTEGER(i4) :: il_jmax TYPE(TFILE) :: tl_file TYPE(TMPP) :: tl_mpp TYPE(TATT) :: tl_att TYPE(TVAR) :: tl_var TYPE(TDOM) :: tl_dom ! loop indices !---------------------------------------------------------------- IF( td_file%i_id == 0 )THEN CALL logger_error("CREATE RESTART EXTRACT: file "//& & TRIM(td_file%c_name)//" not opened ") ELSE !init tl_file=td_file !1- open file CALL iom_open(tl_file) ! get periodicity il_pivot=grid_get_pivot(tl_file) il_perio=grid_get_perio(tl_file,il_pivot) tl_file%i_perio=il_perio !2- compute file grid indices around coord grid il_ind(:,:,:)=grid_get_coarse_index(tl_file, td_coord ) il_imin=il_ind(1,1,1) ; il_imax=il_ind(1,2,1) il_jmin=il_ind(2,1,1) ; il_jmax=il_ind(2,2,1) !3- check grid coincidence CALL grid_check_coincidence( tl_file, td_coord, & & il_imin, il_imax, & & il_jmin, il_jmax, & & (/1, 1, 1/) ) !4- compute domain tl_dom=dom_init(tl_file, & & il_imin, il_imax, & & il_jmin, il_jmax) ! close file CALL iom_close(tl_file) !5- read bathymetry on domain (ugly way to do it, have to work on it) !5-1 init mpp structure tl_mpp=mpp_init(tl_file) CALL file_clean(tl_file) !5-2 get processor to be used CALL mpp_get_use( tl_mpp, tl_dom ) !5-3 open mpp files CALL iom_mpp_open(tl_mpp) !5-4 read variable on domain tl_var=iom_mpp_read_var(tl_mpp,TRIM(td_var%c_name),td_dom=tl_dom) !5-5 close mpp file CALL iom_mpp_close(tl_mpp) !6- add ghost cell CALL grid_add_ghost(tl_var,tl_dom%i_ighost,tl_dom%i_jghost) !7- check result IF( ANY( tl_var%t_dim(:)%l_use .AND. & & tl_var%t_dim(:)%i_len /= td_coord%t_dim(:)%i_len) )THEN CALL logger_debug("CREATE BATHY EXTRACT: "//& & "dimensoin of variable "//TRIM(td_var%c_name)//" "//& & TRIM(fct_str(tl_var%t_dim(1)%i_len))//","//& & TRIM(fct_str(tl_var%t_dim(2)%i_len))//","//& & TRIM(fct_str(tl_var%t_dim(3)%i_len))//","//& & TRIM(fct_str(tl_var%t_dim(4)%i_len)) ) CALL logger_debug("CREATE BATHY EXTRACT: "//& & "dimensoin of coordinate file "//& & TRIM(fct_str(td_coord%t_dim(1)%i_len))//","//& & TRIM(fct_str(td_coord%t_dim(2)%i_len))//","//& & TRIM(fct_str(td_coord%t_dim(3)%i_len))//","//& & TRIM(fct_str(td_coord%t_dim(4)%i_len)) ) CALL logger_fatal("CREATE BATHY EXTRACT: "//& & "dimensoin of extracted "//& & "variable and coordinate file dimension differ") ENDIF !8- add attribute to variable tl_att=att_init('src_file',TRIM(fct_basename(tl_mpp%c_name))) CALL var_move_att(tl_var, tl_att) !9- save result create_restart_extract=tl_var ! clean structure CALL var_clean(tl_var) CALL mpp_clean(tl_mpp) ENDIF END FUNCTION create_restart_extract !> @endcode !------------------------------------------------------------------- !> @brief !> This subroutine !> !> @details !> !> @author J.Paul !> - Nov, 2013- Initial Version !> !> @param[in] !> @todo !------------------------------------------------------------------- !> @code SUBROUTINE create_restart_interp( td_var, td_level,& & id_rho, & & id_offset, & & id_iext, id_jext) IMPLICIT NONE ! Argument TYPE(TVAR) , INTENT(INOUT) :: td_var TYPE(TVAR) , DIMENSION(:) , INTENT(INOUT) :: td_level INTEGER(i4), DIMENSION(:) , INTENT(IN ) :: id_rho INTEGER(i4), DIMENSION(:,:), INTENT(IN ) :: id_offset INTEGER(i4), INTENT(IN ), OPTIONAL :: id_iext INTEGER(i4), INTENT(IN ), OPTIONAL :: id_jext ! local variable TYPE(TVAR) :: tl_var INTEGER(i4) :: il_iext INTEGER(i4) :: il_jext ! loop indices !---------------------------------------------------------------- ! copy variable tl_var=td_var il_iext=3 IF( PRESENT(id_iext) ) il_iext=id_iext il_jext=3 IF( PRESENT(id_jext) ) il_jext=id_jext IF( il_iext < 2 .AND. td_var%c_interp(1) == 'cubic' )THEN CALL logger_warn("CREATE RESTART INTERP: at least extrapolation "//& & "on two points are required with cubic interpolation ") il_iext=2 ENDIF IF( il_jext < 2 .AND. td_var%c_interp(1) == 'cubic' )THEN CALL logger_warn("CREATE RESTART INTERP: at least extrapolation "//& & "on two points are required with cubic interpolation ") il_jext=2 ENDIF il_iext=0 il_jext=0 ! work on variable !1 add extraband CALL extrap_add_extrabands(tl_var, il_iext, il_jext) !2 extrapolate variable CALL extrap_fill_value( tl_var, td_level(:), & & id_offset(:,:), & & id_rho(:), & & id_iext=il_iext, id_jext=il_jext ) !3 interpolate variable CALL interp_fill_value( tl_var, id_rho(:), & & id_offset=id_offset(:,:) ) !4 remove extraband CALL extrap_del_extrabands(tl_var, il_iext*id_rho(jp_I), il_jext*id_rho(jp_J)) !5- save result td_var=tl_var ! clean variable structure CALL var_clean(tl_var) END SUBROUTINE create_restart_interp !> @endcode END PROGRAM create_restart