Changes between Version 9 and Version 10 of ImplementAnalysislknetf


Ignore:
Timestamp:
Jun 4, 2025, 12:40:20 PM (2 days ago)
Author:
lnerger
Comment:

--

Legend:

Unmodified
Added
Removed
Modified
  • ImplementAnalysislknetf

    v9 v10  
    5252== `PDAF_put_state_lknetf` ==
    5353
    54 The general espects of the filter-specific routines `PDAF_assimilate_*` have been described on the page [ModifyModelforEnsembleIntegration Modification of the model code for the ensemble integration].
    55 The interface for the routine `PDAF_assimilate_lknetf` contains several routine names for routines that operate on the local analysis domains (marked by `_l` at the end of the routine name). In addition, there are names for routines that consider all available observations required to perform local analyses with LKNETF within some sub-domain of a domain-decomposed model (marked by `_f` at the end of the routine name). In case of a serial execution of the assimilation program, these will be all globally available observations. However, if the program is executed with parallelization, this might be a smaller set of observations.
    56 
    57 To explain the  difference, it is assumed, for simplicity, that a local analysis domain consists of a single vertical column of the model grid. In addition, we assume that the domain decomposition splits the global model domain by vertical boundaries as is typical for ocean models and that the observations are spatially distributed observations of model fields that are part of the state vector.  Under these assumptions, the situation is the following: When a model uses domain decomposition, the LKNETF algorithm is executed such that for each model sub-domain a loop over all local analysis domains (e.g. vertical columns) that belong to the model sub-domain is performed. As each model sub-domain is treated by a different process, all loops are executed in parallel to each other.
    58 
    59 For the update of each single vertical column, observations from some larger domain surrounding the vertical column are required. If the influence radius for the observations is sufficiently small there will be vertical columns for which the relevant observations reside completely inside the model sub-domain of the process. However, if a vertical column is considered that is located close to the boundary of the model sub-domain, there will be some observations that don't belong spatially to the local model sub-domain, but to a neighboring model sub-domain. Nonetheless, these observations are required on the local model sub-domain. A simple way to handle this situation is to initialize for each process all globally available observations, together with their coordinates. While this method is simple, it can also become inefficient if the assimilation program is executed using a large number of processes. As for an initial implementation it is usually not the concern to obtain the highest parallel efficiency, the description below assumes that 'full' refers to the global model domain.
    60 
    61 The interface when using the LKNETF algorithm is the following:
     54This routine is used both in the ''fully-parallel'' and the ''flexible'' implementation variants of the data assimilation system. (See the page [wiki:OnlineModifyModelforEnsembleIntegration_PDAF3 Modification of the model code for the ensemble integration] for these variants)
     55
     56The interface for the routine `PDAF_assimilate_lestkf` contains several routine names for routines that operate on the local analysis domains (marked by `_l` at the end of the routine name). In addition, there are names for routines that consider all available observations required to perform local analyses with LKNETKF within some sub-domain of a domain-decomposed model (we refer to these as 'full' observations, marked by `_f` at the end of the routine name). In case of a serial execution of the assimilation program, these will be all globally available observations. However, if the program is executed with parallelization, one might choose a smaller set of observations. We will explain this is some detail below.
     57
     58Here, we list the full interface of the routine. Subsequently, the user-supplied routines specified in the call are explained.
     59
     60The interface is :
    6261{{{
    6362  SUBROUTINE PDAF_assimilate_lknetf(U_collect_state, U_distribute_state, &
    64        U_init_dim_obs_f, U_obs_op_f, U_init_obs_f, U_init_obs_l, U_prepoststep, &
    65        U_prodRinvA_l, U_prodRinvA_hyb_l, U_init_n_domains, U_init_dim_l, U_init_dim_obs_l, &
     63       U_init_dim_obs_f, U_obs_op_f, U_init_obs_f, &
     64       U_init_obs_l, U_prepoststep, U_prodRinvA_l, U_prodRinvA_hyb_l, &
     65       U_init_n_domains, U_init_dim_l, U_init_dim_obs_l, &
    6666       U_g2l_state, U_l2g_state, U_g2l_obs, U_init_obsvar, U_init_obsvar_l, &
    6767       U_likelihood_l, U_likelihood_hyb_l, U_next_observation, outflag)
     
    9494
    9595
    96 == `PDAF_put_state_lknetf` ==
    97 
    98 When the 'flexible' implementation variant is chosen for the assimilation system, the routine `PDAF_put_state_lknetf` has to be used instead of `PDAF_assimilate_lknetf`. The general aspects of the filter specific routines `PDAF_put_state_*` have been described on the page [ModifyModelforEnsembleIntegration Modification of the model code for the ensemble integration]. The interface of the routine is identical with that of `PDAF_assimilate_lknetf` with the exception the specification of the user-supplied routines `U_distribute_state` and `U_next_observation` are missing.
    99 
    100 The interface when using the LKNETF algorithm is the following:
    101 {{{
    102   SUBROUTINE PDAF_put_state_lknetf(U_collect_state, &
    103        U_init_dim_obs_f, U_obs_op_f, U_init_obs_f, U_init_obs_l, U_prepoststep, &
    104        U_prodRinvA_l, U_prodRinvA_hyb_l, U_init_n_domains, U_init_dim_l, U_init_dim_obs_l, &
     96== `PDAF_assim_offline_lknetf ` ==
     97
     98This routine is used to perform the analysis step for the offline mode of PDAF.
     99The interface of the routine is identical with that of the 'assimilate'-routine, except that the user-supplied routines `U_distribute_state`, `U_collect_state` and `U_next_observation` are missing.
     100
     101The 'assim_offline' routines were introduced with PDAF V3.0 to simplify the [wiki:OfflineImplementationGuide_PDAF3 implementation of the offline mode].
     102
     103The interface is :
     104{{{
     105  SUBROUTINE PDAF_assim_offline_lknetf( &
     106       U_init_dim_obs_f, U_obs_op_f, U_init_obs_f, &
     107       U_init_obs_l, U_prepoststep, U_prodRinvA_l, U_prodRinvA_hyb_l, &
     108       U_init_n_domains, U_init_dim_l, U_init_dim_obs_l, &
    105109       U_g2l_state, U_l2g_state, U_g2l_obs, U_init_obsvar, U_init_obsvar_l, &
    106110       U_likelihood_l, U_likelihood_hyb_l, outflag)
    107111}}}
    108112
     113== `PDAF_put_state_lknetf` ==
     114
     115This routine exists for backward-compatibility. In implementations that were done for PDAF V2.3.1 and before, a 'put_state' routine was used for the [wiki:OnlineFlexible_PDAF3 'flexible' parallelization variant] and for the [wiki:OfflineImplementationGuide_PDAF3 offline mode].  This routine allows to continue using the previous implementation structure.
     116The interface of the routine is identical with that of the 'assimilate'-routine, except that the user-supplied routines `U_distribute_state` and `U_next_observation` are missing.
     117
     118The interface is:
     119{{{
     120  SUBROUTINE PDAF_put_state_lknetf(U_collect_state, &
     121       U_init_dim_obs_f, U_obs_op_f, U_init_obs_f, &
     122       U_init_obs_l, U_prepoststep, U_prodRinvA_l, U_prodRinvA_hyb_l, &
     123       U_init_n_domains, U_init_dim_l, U_init_dim_obs_l, &
     124       U_g2l_state, U_l2g_state, U_g2l_obs, U_init_obsvar, U_init_obsvar_l, &
     125       U_likelihood_l, U_likelihood_hyb_l, outflag)
     126}}}
     127
     128
     129== Explanation of 'full observations' ==
     130
     131Above we mention the concept of 'full' observations. We distinguish them from the globally available observations for efficiency.
     132
     133Note: For an initial implementation, one might not needd to worry about high efficiency, so that 'full' can refer to all available observations.
     134
     135To explain why 'full' observations can be different from globally available observations, we assume, for simplicity, that  we have a 2-dimensional domain and that a local analysis domain consists of a single grid point of the model grid. In addition, we assume that the domain decomposition splits the global model domain in compact sub-domains and that the observations are spatially distributed observations of model fields that are part of the state vector. 
     136
     137The LESTKF performs a loop over all local analysis domains, i.e. grid points.  When a model uses domain decomposition, the loop is over all grid points that belong to a process sub-domain. As each model sub-domain is treated by a different process, all loops are executed parallel to each other.
     138
     139For the update of each local analysis domain (grid points), observations within the localization radius around its location are required. If the influence radius for the observations is sufficiently small, there will be grid points a for which the relevant observations reside completely inside the model sub-domain of the process. However, if a grid point is located close to the boundary of the model sub-domain, there will be some observations that reside on a neighboring process sub-domain, but are within the localization radius. One needs to assimilate these observations as otherwise, there could be unrealistic steps in the analysis field. However, there will also be observations that reside far away from the process sub-domain and will never influence the analysis result in this domain.
     140
     141A simple way to handle this situation is to initialize for each process all globally available observations, together with their coordinates. The observation operator would be applied on each sub-domain and then the observed ensemble would be collect using parallel communication with MPI. While this method is simple, it can also become inefficient if the assimilation program is executed using a large number of processes. In particular, all observation would need to be checked even if they are far away from the process sub-domain.
     142
     143More efficient is hence to select as 'full' observations only those observations that can have an effect on the local analyses of a process sub-domain. These are the observations that reside within the sub-domain, plus observations in neighboring sub-domains that reside within the localization radius. Setting up 'full' observations in this way leads to a smaller number of observations whose distance need to be checked for each local analysis domain. Howeever, one would need to find an implementation that provides the 'full' observations.
     144
     145|| Note: The handling of 'full' observations is one of the aspects that motivated the development of PDAF-OMI and the relaed avanced interface (now the PDAF3 interface). Here, PDAF-OMI does take case of the 'full' observations. See the [wiki:ImplementationofAnalysisStep_PDAF3 Implementation Guide for the Analysis Step for the advanced interface using PDAF-OMI]. ||
     146
    109147
    110148== User-supplied routines ==
    111149
    112 Here, all user-supplied routines are described that are required in the call to `PDAF_assimilate_lknetf`. For some of the generic routines, we link to the page on [ModifyModelforEnsembleIntegration modifying the model code for the ensemble integration].
     150Here, all user-supplied routines are described that are required in the calls to the analysis routines. For some of the generic routines, we link to the page on [wiki:OnlineModifyModelforEnsembleIntegration_PDAF3 modifying the model code for the ensemble integration].
    113151
    114152To indicate user-supplied routines we use the prefix `U_`. In the tutorials in `tutorial/` and in the template directory `templates/` these routines exist without the prefix, but with the extension `_pdaf`. The files are named correspondingly. In the section titles below we provide the name of the template file in parentheses.
     
    118156=== `U_collect_state` (collect_state_pdaf.F90) ===
    119157
    120 This routine is independent from the filter algorithm used.
    121 See the page on [InsertAnalysisStep#U_collect_statecollect_state_pdaf.F90 inserting the analysis step] for the description of this routine.
     158This routine is independent of the filter algorithm used.
     159See the page on [wiki:OnlineModifyModelforEnsembleIntegration_PDAF3#collect_state_pdafcollect_state_pdaf.F90 modifying the model code for the ensemble integration] for the description of this routine.
     160
    122161
    123162=== `U_distribute_state` (distribute_state_pdaf.F90) ===
    124163
    125164This routine is independent of the filter algorithm used.
    126 See the page on [InsertAnalysisStep#U_distribute_statedistribute_state_pdaf.F90 inserting the analysis step] for the description of this routine.
     165See the page on [wiki:OnlineModifyModelforEnsembleIntegration_PDAF3#distribute_state_pdafdistribute_state_pdaf.F90 modifying the model code for the ensemble integration] for the description of this routine.
    127166
    128167
     
    210249=== `U_prepoststep` (prepoststep_ens_pdaf.F90) ===
    211250
    212 This routine can generally be identical to that used for the global SEIK filter, which has already been described on the [ModifyModelforEnsembleIntegration#U_prepoststepprepoststep_ens_pdaf.F90 page on modifying the model code for the ensemble integration]. For completeness, the description is repeated:
    213 
    214 The interface of the routine is identical for all filters. However, the particular operations that are performed in the routine can be specific for each filter algorithm. Here, we exemplify the interface on the example of the ETKF.
    215 
    216 The interface for this routine is
    217 {{{
    218 SUBROUTINE prepoststep(step, dim_p, dim_ens, dim_ens_p, dim_obs_p, &
    219                        state_p, Uinv, ens_p, flag)
    220 
    221   INTEGER, INTENT(in) :: step        ! Current time step
    222                          ! (When the routine is called before the analysis -step is provided.)
    223   INTEGER, INTENT(in) :: dim_p       ! PE-local state dimension
    224   INTEGER, INTENT(in) :: dim_ens     ! Size of state ensemble
    225   INTEGER, INTENT(in) :: dim_ens_p   ! PE-local size of ensemble
    226   INTEGER, INTENT(in) :: dim_obs_p   ! PE-local dimension of observation vector
    227   REAL, INTENT(inout) :: state_p(dim_p) ! PE-local forecast/analysis state
    228                                      ! The array 'state_p' is not generally not initialized in the case of SEIK/EnKF/ETKF.
    229                                      ! It can be used freely in this routine.
    230   REAL, INTENT(inout) :: Uinv(dim_ens, dim_ens)  ! Inverse of matrix U
    231   REAL, INTENT(inout) :: ens_p(dim_p, dim_ens)   ! PE-local state ensemble
    232   INTEGER, INTENT(in) :: flag        ! PDAF status flag
    233 }}}
    234 
    235 The routine `U_prepoststep` is called once at the beginning of the assimilation process. In addition, it is called during the assimilation cycles before the analysis step and after the ensemble transformation. The routine is called by all filter processes (that is `filterpe=1`).
    236 
    237 The routine provides for the user the full access to the ensemble of model states. Thus, user-controlled pre- and post-step operations can be performed.  For example the forecast and the analysis states and ensemble covariance matrix can be analyzed, e.g. by computing the estimated variances. In addition, the estimates can be written to disk.
    238 
    239 Hint:
    240  * If a user considers to perform adjustments to the estimates (e.g. for balances), this routine is the right place for it.
    241  * Only for the SEEK filter the state vector (`state_p`) is initialized. For all other filters, the array is allocated, but it can be used freely during the execution of `U_prepoststep`.
    242  * The interface has a difference for ETKF and SEIK: For the ETKF, the array `Uinv` has size `dim_ens` x `dim_ens`. In contrast it has size `dim_ens-1` x `dim_ens-1` for the SEIK filter.
    243  * The interface through which `U_prepoststep` is called does not include the array of smoothed ensembles. In order to access the smoother ensemble array one has to set a pointer to it using a call to the routine `PDAF_get_smootherens` (see page on [AuxiliaryRoutines auxiliary routines])
     251The routine has already been described for modifying the model for the ensemble integration and for inserting the analysis step.
     252
     253See the page on [wiki:OnlineModifyModelforEnsembleIntegration_PDAF3#prepoststep_pdafprepoststep_ens_pdaf.F90 inserting the analysis step] for the description of this routine.
    244254
    245255
     
    537547
    538548This routine is independent of the filter algorithm used.
    539 See the page on [InsertAnalysisStep#U_next_observationnext_observation_pdaf.F90 inserting the analysis step] for the description of this routine.
     549See the page on [wiki:OnlineModifyModelforEnsembleIntegration_PDAF3#next_observation_pdafnext_observation_pdaf.F90 modifying the model code for the ensemble integration] for the description of this routine.
    540550
    541551== Execution order of user-supplied routines ==