pyPDAF.PDAF.assimilate_pf

pyPDAF.PDAF.assimilate_pf()

It is recommended to use pyPDAF.PDAF.omi_assimilate_global() or pyPDAF.PDAF.omi_assimilate_nonlin_nondiagR().

PDAF-OMI modules require fewer user-supplied functions and improved efficiency.

This function will use particle filter for a single DA step. This is a fully nonlinear filter, and may require a high number of ensemble members. A review of particle filter can be found at [1]. The function should be called at each model step.

The function is a combination of pyPDAF.PDAF.put_state_pf() and pyPDAF.PDAF.get_state().

This function executes the user-supplied functions in the following sequence:

  1. py__collect_state_pdaf

  2. py__prepoststep_state_pdaf

  3. py__init_dim_obs_pdaf

  4. py__init_obs_pdaf

  5. py__obs_op_pdaf (for each ensemble member)

  6. py__likelihood_pdaf

  7. core DA algorithm

  8. py__prepoststep_state_pdaf

  9. py__distribute_state_pdaf

  10. py__next_observation_pdaf

Deprecated since version 1.0.0: This function is replaced by pyPDAF.PDAF.omi_assimilate_global() and pyPDAF.PDAF.omi_assimilate_nonlin_nondiagR()

References

Parameters:

  • py__collect_state_pdaf (Callable[dim_p:int, state_p : ndarray[tuple[dim_p], np.float64]]) –

    Routine to collect a state vector

    Callback Parameters
    • dim_pint

      • pe-local state dimension

    • state_pndarray[tuple[dim_p], np.float64]

      • local state vector

    Callback Returns
    • state_pndarray[tuple[dim_p], np.float64]

      • local state vector

  • py__distribute_state_pdaf (Callable[dim_p:int, state_p : ndarray[tuple[dim_p], np.float64]]) –

    Routine to distribute a state vector

    Callback Parameters
    • dim_pint

      • pe-local state dimension

    • state_pndarray[tuple[dim_p], np.float64]

      • local state vector

    Callback Returns
    • state_pndarray[tuple[dim_p], np.float64]

      • local state vector

  • py__init_dim_obs_pdaf (Callable[step:int, dim_obs_p:int]) –

    Initialize dimension of observation vector

    Callback Parameters
    • stepint

      • current time step

    • dim_obs_pint

      • dimension of observation vector

    Callback Returns
    • dim_obs_pint

      • dimension of observation vector

  • py__obs_op_pdaf (Callable[step:int, dim_p:int, dim_obs_p:int, state_p : ndarray[tuple[dim_p], np.float64], m_state_p : ndarray[tuple[dim_obs_p], np.float64]]) –

    Observation operator

    Callback Parameters
    • stepint

      • Current time step

    • dim_pint

      • Size of state vector (local part in case of parallel decomposed state)

    • dim_obs_pint

      • Size of observation vector

    • state_pndarray[tuple[dim_p], np.float64]

      • Model state vector

    • m_state_pndarray[tuple[dim_obs_p], np.float64]

      • Observed state vector (i.e. the result after applying the observation operator to state_p)

    Callback Returns
    • m_state_pndarray[tuple[dim_obs_p], np.float64]

      • Observed state vector (i.e. the result after applying the observation operator to state_p)

  • py__init_obs_pdaf (Callable[step:int, dim_obs_p:int, observation_p : ndarray[tuple[dim_obs_p], np.float64]]) –

    Initialize observation vector

    Callback Parameters
    • stepint

      • Current time step

    • dim_obs_pint

      • Size of the observation vector

    • observation_pndarray[tuple[dim_obs_p], np.float64]

      • Vector of observations

    Callback Returns
    • observation_pndarray[tuple[dim_obs_p], np.float64]

      • Vector of observations

  • py__prepoststep_pdaf (Callable[step:int, dim_p:int, dim_ens:int, dim_ens_p:int, dim_obs_p:int, state_p : ndarray[tuple[dim_p], np.float64], uinv : ndarray[tuple[dim_ens-1, dim_ens-1], np.float64], ens_p : ndarray[tuple[dim_p, dim_ens], np.float64], flag:int]) –

    User supplied pre/poststep routine

    Callback Parameters
    • stepint

      • current time step (negative for call after forecast)

    • dim_pint

      • pe-local state dimension

    • dim_ensint

      • size of state ensemble

    • dim_ens_pint

      • pe-local size of ensemble

    • dim_obs_pint

      • pe-local dimension of observation vector

    • state_pndarray[tuple[dim_p], np.float64]

      • pe-local forecast/analysis state (the array ‘state_p’ is not generally not initialized in the case of seik. it can be used freely here.)

    • uinvndarray[tuple[dim_ens-1, dim_ens-1], np.float64]

      • inverse of matrix u

    • ens_pndarray[tuple[dim_p, dim_ens], np.float64]

      • pe-local state ensemble

    • flagint

      • pdaf status flag

    Callback Returns
    • state_pndarray[tuple[dim_p], np.float64]

      • pe-local forecast/analysis state (the array ‘state_p’ is not generally not initialized in the case of seik. it can be used freely here.)

    • uinvndarray[tuple[dim_ens-1, dim_ens-1], np.float64]

      • inverse of matrix u

    • ens_pndarray[tuple[dim_p, dim_ens], np.float64]

      • pe-local state ensemble

  • py__likelihood_pdaf (Callable[step:int, dim_obs_p:int, obs_p : ndarray[tuple[dim_obs_p], np.float64], resid : ndarray[tuple[dim_obs_p], np.float64], likely:float]) –

    Compute observation likelihood for an ensemble member

    Callback Parameters
    • stepint

      • Current time step

    • dim_obs_pint

      • Number of observations at current time step (i.e. the size of the observation vector)

    • obs_pndarray[tuple[dim_obs_p], np.float64]

      • Vector of observations

    • residndarray[tuple[dim_obs_p], np.float64]

      • Input vector holding the residual

    • likelyfloat

      • Output value of the likelihood

    Callback Returns
    • likelyfloat

      • Output value of the likelihood

  • py__next_observation_pdaf (Callable[stepnow:int, nsteps:int, doexit:int, time:float]) –

    Provide time step and time of next observation

    Callback Parameters
    • stepnowint

      • number of the current time step

    • nstepsint

      • number of time steps until next obs

    • doexitint

      • whether to exit forecasting (1 for exit)

    • timefloat

      • current model (physical) time

    Callback Returns
    • nstepsint

      • number of time steps until next obs

    • doexitint

      • whether to exit forecasting (1 for exit)

    • timefloat

      • current model (physical) time

Returns:

flag – Status flag

Return type:

int