/[MITgcm]/MITgcm/model/src/the_model_main.F
ViewVC logotype

Diff of /MITgcm/model/src/the_model_main.F

Parent Directory Parent Directory | Revision Log Revision Log | View Revision Graph Revision Graph | View Patch Patch

revision 1.16 by cnh, Tue Jun 30 17:21:11 1998 UTC revision 1.79 by edhill, Fri Sep 10 12:19:30 2004 UTC
# Line 1  Line 1 
1  C $Header$  C $Header$
2    C $Name$
3    
4  #include "CPP_EEOPTIONS.h"  CBOI
5    C
6    C !TITLE: MITGCM KERNEL CODE SYNOPSIS
7    C !AUTHORS: mitgcm developers ( support@mitgcm.org )
8    C !AFFILIATION: Massachussetts Institute of Technology
9    C !DATE:
10    C !INTRODUCTION: Kernel dynamical routines
11    C This document summarises MITgcm code under the model/ subdirectory.
12    C The code under model/ ( src/ and inc/ ) contains most of
13    C the driver routines for the baseline forms of the kernel equations in the
14    C MITgcm algorithm. Numerical code for much of the baseline forms of
15    C these equations is also under the model/ directory. Other numerical code
16    C used for the kernel equations is contained in packages in the pkg/
17    C directory tree.
18    C Code for auxilliary equations and alternate dicretizations of the kernel
19    C equations and algorithm can also be found in the pkg/ directory tree.
20    C
21    C \subsection{Getting Help and Reporting Errors and Problems}
22    C If you have questions please e-mail support@mitgcm.org. We also welcome
23    C reports of errors and inconsistencies in the code or in the accompanying
24    C documentation. Please feel free to send these
25    C to support@mitgcm.org. For further information and to review
26    C problems reported to support@mitgcm.org please visit http://mitgcm.org.
27    C
28    C \subsection{MITgcm Kernel Code Calling Sequence}
29    C \bv
30    C
31    C Invocation from WRAPPER level...
32    C  :
33    C  :
34    C  |
35    C  |-THE_MODEL_MAIN :: Primary driver for the MITgcm algorithm
36    C    |              :: Called from WRAPPER level numerical
37    C    |              :: code innvocation routine. On entry
38    C    |              :: to THE_MODEL_MAIN separate thread and
39    C    |              :: separate processes will have been established.
40    C    |              :: Each thread and process will have a unique ID
41    C    |              :: but as yet it will not be associated with a
42    C    |              :: specific region in decomposed discrete space.
43    C    |
44    C    |-INITIALISE_FIXED :: Set fixed model arrays such as topography,
45    C    | |                :: grid, solver matrices etc..
46    C    | |              
47    C    | |-INI_PARMS :: Routine to set kernel model parameters.
48    C    | |           :: By default kernel parameters are read from file
49    C    | |           :: "data" in directory in which code executes.
50    C    | |
51    C    | |-MON_INIT :: Initialises monitor pacakge ( see pkg/monitor )
52    C    | |
53    C    | |-INI_GRID :: Control grid array (vert. and hori.) initialisation.
54    C    | | |        :: Grid arrays are held and described in GRID.h.
55    C    | | |
56    C    | | |-INI_VERTICAL_GRID        :: Initialise vertical grid arrays.
57    C    | | |
58    C    | | |-INI_CARTESIAN_GRID       :: Cartesian horiz. grid initialisation
59    C    | | |                          :: (calculate grid from kernel parameters).
60    C    | | |
61    C    | | |-INI_SPHERICAL_POLAR_GRID :: Spherical polar horiz. grid
62    C    | | |                          :: initialisation (calculate grid from
63    C    | | |                          :: kernel parameters).
64    C    | | |
65    C    | | |-INI_CURVILINEAR_GRID     :: General orthogonal, structured horiz.
66    C    | |                            :: grid initialisations. ( input from raw
67    C    | |                            :: grid files, LONC.bin, DXF.bin etc... )
68    C    | |
69    C    | |-INI_DEPTHS    :: Read (from "bathyFile") or set bathymetry/orgography.
70    C    | |
71    C    | |-INI_MASKS_ETC :: Derive horizontal and vertical cell fractions and
72    C    | |               :: land masking for solid-fluid boundaries.
73    C    | |
74    C    | |-INI_LINEAR_PHSURF :: Set ref. surface Bo_surf
75    C    | |
76    C    | |-INI_CORI          :: Set coriolis term. zero, f-plane, beta-plane,
77    C    | |                   :: sphere optins are coded.
78    C    | |
79    C    | |-PACAKGES_BOOT      :: Start up the optional package environment.
80    C    | |                    :: Runtime selection of active packages.
81    C    | |
82    C    | |-PACKAGES_READPARMS :: Call active package internal parameter load.
83    C    | | |
84    C    | | |-GMREDI_READPARMS    :: GM Package. see pkg/gmredi
85    C    | | |-KPP_READPARMS       :: KPP Package. see pkg/kpp
86    C    | | |-SHAP_FILT_READPARMS :: Shapiro filter package. see pkg/shap_filt
87    C    | | |-OBCS_READPARMS      :: Open bndy package. see pkg/obcs
88    C    | | |-AIM_READPARMS       :: Intermediate Atmos. pacakage. see pkg/aim
89    C    | | |-COST_READPARMS      :: Cost function package. see pkg/cost
90    C    | | |-CTRL_INIT           :: Control vector support package. see pkg/ctrl
91    C    | | |-OPTIM_READPARMS     :: Optimisation support package. see pkg/ctrl
92    C    | | |-GRDCHK_READPARMS    :: Gradient check package. see pkg/grdchk
93    C    | | |-ECCO_READPARMS      :: ECCO Support Package. see pkg/ecco
94    C    | | |-PTRACERS_READPARMS  :: multiple tracer package, see pkg/ptracers
95    C    | | |-GCHEM_READPARMS     :: tracer interface package, see pkg/gchem
96    C    | |
97    C    | |-PACKAGES_CHECK
98    C    | | |
99    C    | | |-KPP_CHECK           :: KPP Package. pkg/kpp
100    C    | | |-OBCS_CHECK          :: Open bndy Pacakge. pkg/obcs
101    C    | | |-GMREDI_CHECK        :: GM Package. pkg/gmredi
102    C    | |
103    C    | |-PACKAGES_INIT_FIXED
104    C    | | |-OBCS_INIT_FIXED     :: Open bndy Package. see pkg/obcs
105    C    | | |-FLT_INIT            :: Floats Package. see pkg/flt
106    C    | | |-GCHEM_INIT_FIXED    :: tracer interface pachage, see pkg/gchem
107    C    | |
108    C    | |-ZONAL_FILT_INIT       :: FFT filter Package. see pkg/zonal_filt
109    C    | |
110    C    | |-INI_CG2D              :: 2d con. grad solver initialisation.
111    C    | |
112    C    | |-INI_CG3D              :: 3d con. grad solver initialisation.
113    C    | |
114    C    | |-CONFIG_SUMMARY        :: Provide synopsis of kernel setup.
115    C    |                         :: Includes annotated table of kernel
116    C    |                         :: parameter settings.
117    C    |
118    C    |-CTRL_UNPACK :: Control vector support package. see pkg/ctrl
119    C    |
120    C    |-ADTHE_MAIN_LOOP :: Derivative evaluating form of main time stepping loop
121    C    !                 :: Auotmatically gerenrated by TAMC/TAF.
122    C    |
123    C    |-CTRL_PACK   :: Control vector support package. see pkg/ctrl
124    C    |
125    C    |-GRDCHK_MAIN :: Gradient check package. see pkg/grdchk
126    C    |
127    C    |-THE_MAIN_LOOP :: Main timestepping loop routine.
128    C    | |
129    C    | |-INITIALISE_VARIA :: Set the initial conditions for time evolving
130    C    | | |                :: variables
131    C    | | |
132    C    | | |-INI_LINEAR_PHISURF :: Set ref. surface Bo_surf
133    C    | | |
134    C    | | |-INI_CORI     :: Set coriolis term. zero, f-plane, beta-plane,
135    C    | | |              :: sphere optins are coded.
136    C    | | |
137    C    | | |-INI_CG2D     :: 2d con. grad solver initialisation.
138    C    | | |-INI_CG3D     :: 3d con. grad solver initialisation.
139    C    | | |-INI_MIXING   :: Initialise diapycnal diffusivity.
140    C    | | |-INI_DYNVARS  :: Initialise to zero all DYNVARS.h arrays (dynamical
141    C    | | |              :: fields).
142    C    | | |
143    C    | | |-INI_FIELDS   :: Control initialising model fields to non-zero
144    C    | | | |-INI_VEL    :: Initialize 3D flow field.
145    C    | | | |-INI_THETA  :: Set model initial temperature field.
146    C    | | | |-INI_SALT   :: Set model initial salinity field.
147    C    | | | |-INI_PSURF  :: Set model initial free-surface height/pressure.
148    C    | | | |-INI_PRESSURE :: Compute model initial hydrostatic pressure
149    C    | | | |-READ_CHECKPOINT :: Read the checkpoint
150    C    | | |
151    C    | | |-INI_TR1      :: Set initial tracer 1 distribution.
152    C    | | |
153    C    | | |-THE_CORRECTION_STEP :: Step forward to next time step.
154    C    | | | |                   :: Here applied to move restart conditions
155    C    | | | |                   :: (saved in mid timestep) to correct level in
156    C    | | | |                   :: time (only used for pre-c35).
157    C    | | | |
158    C    | | | |-CALC_GRAD_PHI_SURF :: Return DDx and DDy of surface pressure
159    C    | | | |-CORRECTION_STEP    :: Pressure correction to momentum
160    C    | | | |-CYCLE_TRACER       :: Move tracers forward in time.
161    C    | | | |-OBCS_APPLY         :: Open bndy package. see pkg/obcs
162    C    | | | |-SHAP_FILT_APPLY    :: Shapiro filter package. see pkg/shap_filt
163    C    | | | |-ZONAL_FILT_APPLY   :: FFT filter package. see pkg/zonal_filt
164    C    | | | |-CONVECTIVE_ADJUSTMENT :: Control static instability mixing.
165    C    | | | | |-FIND_RHO  :: Find adjacent densities.
166    C    | | | | |-CONVECT   :: Mix static instability.
167    C    | | | | |-TIMEAVE_CUMULATE :: Update convection statistics.
168    C    | | | |
169    C    | | | |-CALC_EXACT_ETA        :: Change SSH to flow divergence.    
170    C    | | |
171    C    | | |-CONVECTIVE_ADJUSTMENT_INI :: Control static instability mixing
172    C    | | | |                         :: Extra time history interactions.
173    C    | | | |                      
174    C    | | | |-FIND_RHO  :: Find adjacent densities.
175    C    | | | |-CONVECT   :: Mix static instability.
176    C    | | | |-TIMEAVE_CUMULATE :: Update convection statistics.
177    C    | | |
178    C    | | |-PACKAGES_INIT_VARIABLES :: Does initialisation of time evolving
179    C    | | | |                       :: package data.
180    C    | | | |
181    C    | | | |-GMREDI_INIT          :: GM package. ( see pkg/gmredi )
182    C    | | | |-KPP_INIT             :: KPP package. ( see pkg/kpp )
183    C    | | | |-KPP_OPEN_DIAGS    
184    C    | | | |-OBCS_INIT_VARIABLES  :: Open bndy. package. ( see pkg/obcs )
185    C    | | | |-PTRACERS_INIT        :: multi. tracer package,(see pkg/ptracers)
186    C    | | | |-GCHEM_INIT           : : tracer interface pkg (see pkh/gchem)
187    C    | | | |-AIM_INIT             :: Interm. atmos package. ( see pkg/aim )
188    C    | | | |-CTRL_MAP_INI         :: Control vector package.( see pkg/ctrl )
189    C    | | | |-COST_INIT            :: Cost function package. ( see pkg/cost )
190    C    | | | |-ECCO_INIT            :: ECCO support package. ( see pkg/ecco )
191    C    | | | |-INI_FORCING          :: Set model initial forcing fields.
192    C    | | |   |                    :: Either set in-line or from file as shown.
193    C    | | |   |-READ_FLD_XY_RS(zonalWindFile)
194    C    | | |   |-READ_FLD_XY_RS(meridWindFile)
195    C    | | |   |-READ_FLD_XY_RS(surfQFile)
196    C    | | |   |-READ_FLD_XY_RS(EmPmRfile)
197    C    | | |   |-READ_FLD_XY_RS(thetaClimFile)
198    C    | | |   |-READ_FLD_XY_RS(saltClimFile)
199    C    | | |   |-READ_FLD_XY_RS(surfQswFile)
200    C    | | |
201    C    | | |-CALC_SURF_DR   :: Calculate the new surface level thickness.
202    C    | | |-UPDATE_SURF_DR :: Update the surface-level thickness fraction.
203    C    | | |-UPDATE_CG2D    :: Update 2d conjugate grad. for Free-Surf.
204    C    | | |-STATE_SUMMARY    :: Summarize model prognostic variables.
205    C    | | |-TIMEAVE_STATVARS :: Time averaging package ( see pkg/timeave ).
206    C    | |
207    C    | |-WRITE_STATE      :: Controlling routine for IO to dump model state.
208    C    | | |-WRITE_REC_XYZ_RL :: Single file I/O
209    C    | | |-WRITE_FLD_XYZ_RL :: Multi-file I/O
210    C    | |
211    C    | |-MONITOR          :: Monitor state ( see pkg/monitor )
212    C    | |-CTRL_MAP_FORCING :: Control vector support package. ( see pkg/ctrl )
213    C====|>|
214    C====|>| ****************************
215    C====|>| BEGIN MAIN TIMESTEPPING LOOP
216    C====|>| ****************************
217    C====|>|
218    C/\  | |-FORWARD_STEP     :: Step forward a time-step ( AT LAST !!! )
219    C/\  | | |
220    C/\  | | |-DUMMY_IN_STEPPING :: autodiff package ( pkg/autoduff ).
221    C/\  | | |-CALC_EXACT_ETA :: Change SSH to flow divergence.
222    C/\  | | |-CALC_SURF_DR   :: Calculate the new surface level thickness.
223    C/\  | | |-EXF_GETFORCING :: External forcing package. ( pkg/exf )
224    C/\  | | |-EXTERNAL_FIELDS_LOAD :: Control loading time dep. external data.
225    C/\  | | | |                    :: Simple interpolcation between end-points
226    C/\  | | | |                    :: for forcing datasets.
227    C/\  | | | |                  
228    C/\  | | | |-EXCH :: Sync forcing. in overlap regions.
229    C/\  | | |-SEAICE_MODEL   :: Compute sea-ice terms. ( pkg/seaice )
230    C/\  | | |-FREEZE         :: Limit surface temperature.
231    C/\  | | |-GCHEM_FIELD_LOAD :: load tracer forcing fields (pkg/gchem)
232    C/\  | | |
233    C/\  | | |-THERMODYNAMICS :: theta, salt + tracer equations driver.
234    C/\  | | | |
235    C/\  | | | |-INTEGRATE_FOR_W :: Integrate for vertical velocity.
236    C/\  | | | |-OBCS_APPLY_W    :: Open bndy. package ( see pkg/obcs ).
237    C/\  | | | |-FIND_RHO        :: Calculates [rho(S,T,z)-RhoConst] of a slice
238    C/\  | | | |-GRAD_SIGMA      :: Calculate isoneutral gradients
239    C/\  | | | |-CALC_IVDC       :: Set Implicit Vertical Diffusivity for Convection
240    C/\  | | | |
241    C/\  | | | |-OBCS_CALC            :: Open bndy. package ( see pkg/obcs ).
242    C/\  | | | |-EXTERNAL_FORCING_SURF:: Accumulates appropriately dimensioned
243    C/\  | | | | |                    :: forcing terms.
244    C/\  | | | | |-PTRACERS_FORCING_SURF :: Tracer package ( see pkg/ptracers ).
245    C/\  | | | |
246    C/\  | | | |-GMREDI_CALC_TENSOR   :: GM package ( see pkg/gmredi ).
247    C/\  | | | |-GMREDI_CALC_TENSOR_DUMMY :: GM package ( see pkg/gmredi ).
248    C/\  | | | |-KPP_CALC             :: KPP package ( see pkg/kpp ).
249    C/\  | | | |-KPP_CALC_DUMMY       :: KPP package ( see pkg/kpp ).
250    C/\  | | | |-AIM_DO_ATMOS_PHYSICS :: Intermed. atmos package ( see pkg/aim ).
251    C/\  | | | |-GAD_ADVECTION        :: Generalised advection driver (multi-dim
252    C/\  | | | |                         advection case) (see pkg/gad).
253    C/\  | | | |-CALC_COMMON_FACTORS  :: Calculate common data (such as volume flux)
254    C/\  | | | |-CALC_DIFFUSIVITY     :: Calculate net vertical diffusivity
255    C/\  | | | | |
256    C/\  | | | | |-GMREDI_CALC_DIFF   :: GM package ( see pkg/gmredi ).
257    C/\  | | | | |-KPP_CALC_DIFF      :: KPP package ( see pkg/kpp ).
258    C/\  | | | |
259    C/\  | | | |-CALC_GT              :: Calculate the temperature tendency terms
260    C/\  | | | | |
261    C/\  | | | | |-GAD_CALC_RHS       :: Generalised advection package
262    C/\  | | | | | |                  :: ( see pkg/gad )
263    C/\  | | | | | |-KPP_TRANSPORT_T  :: KPP non-local transport ( see pkg/kpp ).
264    C/\  | | | | |
265    C/\  | | | | |-EXTERNAL_FORCING_T :: Problem specific forcing for temperature.
266    C/\  | | | | |-ADAMS_BASHFORTH2   :: Extrapolate tendencies forward in time.
267    C/\  | | | | |-FREESURF_RESCALE_G :: Re-scale Gt for free-surface height.
268    C/\  | | | |
269    C/\  | | | |-TIMESTEP_TRACER      :: Step tracer field forward in time
270    C/\  | | | |
271    C/\  | | | |-CALC_GS              :: Calculate the salinity tendency terms
272    C/\  | | | | |
273    C/\  | | | | |-GAD_CALC_RHS       :: Generalised advection package
274    C/\  | | | | | |                  :: ( see pkg/gad )
275    C/\  | | | | | |-KPP_TRANSPORT_S  :: KPP non-local transport ( see pkg/kpp ).
276    C/\  | | | | |
277    C/\  | | | | |-EXTERNAL_FORCING_S :: Problem specific forcing for salt.
278    C/\  | | | | |-ADAMS_BASHFORTH2   :: Extrapolate tendencies forward in time.
279    C/\  | | | | |-FREESURF_RESCALE_G :: Re-scale Gs for free-surface height.
280    C/\  | | | |
281    C/\  | | | |-TIMESTEP_TRACER      :: Step tracer field forward in time
282    C/\  | | | |
283    C/\  | | | |-CALC_GTR1            :: Calculate other tracer(s) tendency terms
284    C/\  | | | | |
285    C/\  | | | | |-GAD_CALC_RHS       :: Generalised advection package
286    C/\  | | | | |                    :: ( see pkg/gad )
287    C/\  | | | | |
288    C/\  | | | | |-EXTERNAL_FORCING_TR:: Problem specific forcing for tracer.
289    C/\  | | | | |-ADAMS_BASHFORTH2   :: Extrapolate tendencies forward in time.
290    C/\  | | | | |-FREESURF_RESCALE_G :: Re-scale Gs for free-surface height.
291    C/\  | | | |
292    C/\  | | | |-TIMESTEP_TRACER      :: Step tracer field forward in time
293    C/\  | | | |
294    C/\  | | | |-PTRACERS_INTEGRATE   :: Integrate other tracer(s) ( see pkg/ptracers ).
295    C/\  | | | | |
296    C/\  | | | | |-GAD_CALC_RHS       :: Generalised advection package
297    C/\  | | | | | |                  :: ( see pkg/gad )
298    C/\  | | | | | |-KPP_TRANSPORT_PTR:: KPP non-local transport ( see pkg/kpp ).
299    C/\  | | | | |
300    C/\  | | | | |-PTRACERS_FORCING   :: Problem specific forcing for tracer.
301    C/\  | | | | |-GCHEM_FORCING_INT  :: tracer forcing for gchem pkg (if all
302    C/\  | | | | |                       tendancy terms calcualted together)
303    C/\  | | | | |-ADAMS_BASHFORTH2   :: Extrapolate tendencies forward in time.
304    C/\  | | | | |-FREESURF_RESCALE_G :: Re-scale Gs for free-surface height.
305    C/\  | | | | |-TIMESTEP_TRACER    :: Step tracer field forward in time
306    C/\  | | | |
307    C/\  | | | |-OBCS_APPLY_TS        :: Open bndy. package (see pkg/obcs ).
308    C/\  | | | |
309    C/\  | | | |-IMPLDIFF             :: Solve vertical implicit diffusion equation.
310    C/\  | | | |-OBCS_APPLY_TS        :: Open bndy. package (see pkg/obcs ).
311    C/\  | | | |
312    C/\  | | | |-AIM_AIM2DYN_EXCHANGES :: Inetermed. atmos (see pkg/aim).
313    C/\  | | | |-EXCH                 :: Update overlaps
314    C/\  | | |
315    C/\  | | |-DYNAMICS       :: Momentum equations driver.
316    C/\  | | | |
317    C/\  | | | |-CALC_GRAD_PHI_SURF :: Calculate the gradient of the surface
318    C/\  | | | |                       Potential anomaly.
319    C/\  | | | |-CALC_VISCOSITY     :: Calculate net vertical viscosity
320    C/\  | | | | |-KPP_CALC_VISC    :: KPP package ( see pkg/kpp ).
321    C/\  | | | |                                                      
322    C/\  | | | |-CALC_PHI_HYD       :: Integrate the hydrostatic relation.
323    C/\  | | | |-MOM_FLUXFORM       :: Flux form mom eqn. package ( see
324    C/\  | | | |                       pkg/mom_fluxform ).
325    C/\  | | | |-MOM_VECINV         :: Vector invariant form mom eqn. package ( see
326    C/\  | | | |                       pkg/mom_vecinv   ).
327    C/\  | | | |-TIMESTEP           :: Step momentum fields forward in time
328    C/\  | | | |-OBCS_APPLY_UV      :: Open bndy. package (see pkg/obcs ).
329    C/\  | | | |
330    C/\  | | | |-IMPLDIFF           :: Solve vertical implicit diffusion equation.
331    C/\  | | | |-OBCS_APPLY_UV      :: Open bndy. package (see pkg/obcs ).
332    C/\  | | | |
333    C/\  | | | |-TIMEAVE_CUMUL_1T   :: Time averaging package ( see pkg/timeave ).
334    C/\  | | | |-TIMEAVE_CUMUATE    :: Time averaging package ( see pkg/timeave ).
335    C/\  | | | |-DEBUG_STATS_RL     :: Quick debug package ( see pkg/debug ).
336    C/\  | | |
337    C/\  | | |-CALC_GW        :: vert. momentum tendency terms ( NH, QH only ).
338    C/\  | | |
339    C/\  | | |-UPDATE_SURF_DR :: Update the surface-level thickness fraction.
340    C/\  | | |
341    C/\  | | |-UPDATE_CG2D    :: Update 2d conjugate grad. for Free-Surf.
342    C/\  | | |
343    C/\  | | |-SOLVE_FOR_PRESSURE           :: Find surface pressure.
344    C/\  | | | |-CALC_DIV_GHAT     :: Form the RHS of the surface pressure eqn.
345    C/\  | | | |-CG2D              :: Two-dim pre-con. conjugate-gradient.
346    C/\  | | | |-CG3D              :: Three-dim pre-con. conjugate-gradient solver.
347    C/\  | | |
348    C/\  | | |-THE_CORRECTION_STEP          :: Step forward to next time step.
349    C/\  | | | |
350    C/\  | | | |-CALC_GRAD_PHI_SURF :: Return DDx and DDy of surface pressure
351    C/\  | | | |-CORRECTION_STEP    :: Pressure correction to momentum
352    C/\  | | | |-CYCLE_TRACER       :: Move tracers forward in time.
353    C/\  | | | |-OBCS_APPLY         :: Open bndy package. see pkg/obcs
354    C/\  | | | |-SHAP_FILT_APPLY    :: Shapiro filter package. see pkg/shap_filt
355    C/\  | | | |-ZONAL_FILT_APPLY   :: FFT filter package. see pkg/zonal_filt
356    C/\  | | | |-CONVECTIVE_ADJUSTMENT :: Control static instability mixing.
357    C/\  | | | | |-FIND_RHO  :: Find adjacent densities.
358    C/\  | | | | |-CONVECT   :: Mix static instability.
359    C/\  | | | | |-TIMEAVE_CUMULATE :: Update convection statistics.
360    C/\  | | | |
361    C/\  | | | |-CALC_EXACT_ETA        :: Change SSH to flow divergence.    
362    C/\  | | |
363    C/\  | | |-DO_FIELDS_BLOCKING_EXCHANGES :: Sync up overlap regions.
364    C/\  | | | |-EXCH                                                  
365    C/\  | | |
366    C/\  | | |-GCHEM_FORCING_SEP :: tracer forcing for gchem pkg (if
367    C/\  | | |                      tracer dependent tendencies calculated
368    C/\  | | |                      separatly)
369    C/\  | | |
370    C/\  | | |-FLT_MAIN         :: Float package ( pkg/flt ).
371    C/\  | | |
372    C/\  | | |-MONITOR          :: Monitor package ( pkg/monitor ).
373    C/\  | | |
374    C/\  | | |-DO_THE_MODEL_IO  :: Standard diagnostic I/O.
375    C/\  | | | |-WRITE_STATE    :: Core state I/O
376    C/\  | | | |-TIMEAVE_STATV_WRITE :: Time averages. see pkg/timeave
377    C/\  | | | |-AIM_WRITE_DIAGS     :: Intermed. atmos diags. see pkg/aim
378    C/\  | | | |-GMREDI_DIAGS        :: GM diags. see pkg/gmredi
379    C/\  | | | |-KPP_DO_DIAGS        :: KPP diags. see pkg/kpp
380    C/\  | | | |-SBO_CALC            :: SBO diags. see pkg/sbo
381    C/\  | | | |-SBO_DIAGS           :: SBO diags. see pkg/sbo
382    C/\  | | | |-SEAICE_DO_DIAGS     :: SEAICE diags. see pkg/seaice
383    C/\  | | | |-GCHEM_DIAGS         :: gchem diags. see pkg/gchem
384    C/\  | | |
385    C/\  | | |-WRITE_CHECKPOINT :: Do I/O for restart files.
386    C/\  | |
387    C/\  | |-COST_TILE        :: Cost function package. ( see pkg/cost )
388    C<===|=|
389    C<===|=| **************************
390    C<===|=| END MAIN TIMESTEPPING LOOP
391    C<===|=| **************************
392    C<===|=|
393    C    | |-COST_FINAL       :: Cost function package. ( see pkg/cost )
394    C    |
395    C    |-WRITE_CHECKPOINT :: Final state storage, for restart.
396    C    |
397    C    |-TIMER_PRINTALL :: Computational timing summary
398    C    |
399    C    |-COMM_STATS     :: Summarise inter-proc and inter-thread communication
400    C                     :: events.
401    C \ev
402    C
403    C
404    CEOI
405    
406    
407    #include "AD_CONFIG.h"
408    #include "PACKAGES_CONFIG.h"
409    #include "CPP_OPTIONS.h"
410    
411    CBOP
412    
413    C     !ROUTINE: THE_MODEL_MAIN
414    
415    C     !INTERFACE:
416        SUBROUTINE THE_MODEL_MAIN(myThid)        SUBROUTINE THE_MODEL_MAIN(myThid)
417  C     /==========================================================\        IMPLICIT NONE
418  C     | SUBROUTINE THE_MODEL_MAIN                                |  
419  C     | o Master controlling routine for model using the MITgcm  |  C     !DESCRIPTION: \bv
420  C     |   UV parallel wrapper.                                   |  C     *==========================================================*
421  C     |==========================================================|  C     | SUBROUTINE THE_MODEL_MAIN                                
422  C     | THE_MODEL_MAIN is invoked by the MITgcm UV parallel      |  C     | o Master controlling routine for model using the MITgcm  
423  C     | wrapper with a single integer argument "myThid". This    |  C     |   UV parallel wrapper.                                    
424  C     | variable identifies the thread number of an instance of  |  C     *==========================================================*
425  C     | THE_MODEL_MAIN. Each instance of THE_MODEL_MAIN works    |  C     | THE_MODEL_MAIN is invoked by the MITgcm UV parallel      
426  C     | on a particular region of the models domain and          |  C     | wrapper with a single integer argument "myThid". This    
427  C     | synchronises with other instances as necessary. The      |  C     | variable identifies the thread number of an instance of  
428  C     | routine has to "understand" the MITgcm parallel          |  C     | THE_MODEL_MAIN. Each instance of THE_MODEL_MAIN works    
429  C     | environment and the numerical algorithm. Editing this    |  C     | on a particular region of the models domain and          
430  C     | routine is best done with some knowledge of both aspects.|  C     | synchronises with other instances as necessary. The      
431  C     | Notes                                                    |  C     | routine has to "understand" the MITgcm parallel          
432  C     | =====                                                    |  C     | environment and the numerical algorithm. Editing this    
433  C     | C*P* comments indicating place holders for which code is |  C     | routine is best done with some knowledge of both aspects.
434  C     |      presently being developed.                          |  C     | Notes                                                    
435  C     \==========================================================/  C     | =====                                                    
436  C  C     | C*P* comments indicating place holders for which code is  
437  C     Call Tree  C     |      presently being developed.                          
438  C     =========  C     *==========================================================*
439  C      C     \ev
440  C      main ( eesupp )  
441  C       |  C     !CALLING SEQUENCE:
442  C       .  C     THE_MODEL_MAIN()
443  C       .  C       |
444  C       . Generic environment initialisation ( see eesupp/src and  C       |
445  C       .                                      eesupp/inc )        C       |--INITIALISE_FIXED
446  C       . multiple threads and/or processes are created in here  C       |   o Set model configuration (fixed arrays)
447  C       .  C       |     Topography, hydrography, timestep, grid, etc..
448  C       .  C       |
449  C       .  C       |--CTRL_UNPACK      o Derivative mode. Unpack control vector.
450  C       |  C       |
451  C       |-THE_MODEL_MAIN - Begin specific model. One instance  C       |--ADTHE_MAIN_LOOP  o Main timestepping loop for combined
452  C       |  |               of this codes exists for each thread  C       |                     prognostic and reverse mode integration.
453  C       |  |               and/or instance. Each instance manages  C       |
454  C       |  |               a specifc set of tiles.                C       |--CTRL_PACK        o Derivative mode. Unpack control vector.
455  C       |  |  C       |
456  C       |  |--INITIALISE  C       |--GRDCHK_MAIN      o Gradient check control routine.
457  C       |  |   o Set initial conditions and model configuration  C       |
458  C       |  |     Topography, hydrography, timestep, grid, etc..  C       |--THE_MAIN_LOOP    o Main timestepping loop for pure prognostic
459  C       |  |  C       |                     integration.
460  C  ==>  |  | ** Time stepping loop starts here **  C       |
461  C  |    |  |  C       |--WRITE_CHECKPOINT o Write retsart information.
462  C /|\   |  |  C       |
463  C  |    |  |--LOAD_EXTERNAL_DATA  C       |--TIMER_PRINTALL   o Print out timing statistics.
464  C /|\   |  |   o Load and/or set time dependent forcing fields  C       |
465  C  |    |  |  C       |--COMM_STATS       o Print out communication statistics.
 C /|\   |  |--DYNAMICS  
 C  |    |  |   o Evaluate "forward" terms  
 C /|\   |  |  
 C  |    |  |--DO_THE_MODEL_IO  
 C /|\   |  |   o Write model state  
 C  |    |  |  
 C /|\   |  |--SOLVE_FOR_PRESSURE  
 C  |    |  |   o Find pressure field to keep flow non-divergent  
 C /|\   |  |  
 C  |    |  |--DO_GTERM_BLOCKING_EXCHANGES  
 C /|\   |  |   o Update overlap regions  
 C  |    |  |  
 C /|\   |  |--WRITE_CHECKPOINT  
 C  |    |  |   o Write restart file(s)  
 C /|\   |  |  
 C  |    |  |  
 C  |<== |  | ** Time stepping loop finishes here **  
 C       |  |  
 C       |  |--WRITE_CHECKPOINT  
 C       |  |--WRITE_STATE  
 C       |  
 C       .  
 C       .  
 C       . Generic environment termination ( see eesupp/src and  
 C       .                                      eesupp/inc )        
 C       .  
 C       .  
466    
467    C     !USES:
468  C     == Global variables ===  C     == Global variables ===
469  #include "SIZE.h"  #include "SIZE.h"
470  #include "EEPARAMS.h"  #include "EEPARAMS.h"
471  #include "PARAMS.h"  #include "PARAMS.h"
472  #include "CG2D.h"  #ifdef ALLOW_MNC
473    #include "MNC_PARAMS.h"
474    #endif
475  #include "DYNVARS.h"  #include "DYNVARS.h"
476  #include "AVER.h"  #ifdef ALLOW_NONHYDROSTATIC
477    # include "CG3D.h"
478    #endif
479    
480    #ifdef ALLOW_ECCO_OPTIMIZATION
481    # include "optim.h"
482    #endif
483    #ifdef ALLOW_AUTODIFF_TAMC
484    # include "tamc.h"
485    #endif
486    
487    C     !INPUT/OUTPUT PARAMETERS:
488  C     == Routine arguments ==  C     == Routine arguments ==
489  C     myThid - Thread number for this instance of the routine.  C     myThid - Thread number for this instance of the routine.
490        INTEGER myThid          INTEGER myThid  
491    
492    C     !LOCAL VARIABLES:
493  C     == Local variables ==  C     == Local variables ==
494  C     Note: Under the multi-threaded model myCurrentIter and  C     Note: Under the multi-threaded model myCurrentIter and
495  C           myCurrentTime are local variables passed around as routine  C           myCurrentTime are local variables passed around as routine
496  C           arguments. Although this is fiddly it saves the need to  C           arguments. Although this is fiddly it saves the need to
497  C           impose additional synchronisation points when they are  C           impose additional synchronisation points when they are
498  C           updated.  C           updated.
 C     myCurrentIter - Iteration counter for this thread  
499  C     myCurrentTime - Time counter for this thread  C     myCurrentTime - Time counter for this thread
500  C     I             - Loop counter  C     myCurrentIter - Iteration counter for this thread
501        INTEGER I, myCurrentIter        INTEGER myCurrentIter
502        REAL    myCurrentTime        _RL     myCurrentTime
503          logical  exst
504          logical  lastdiva
505    #ifndef ALLOW_ECCO_OPTIMIZATION
506          integer optimcycle
507    #endif
508    CEOP
509    c--   set default:
510    #ifndef ALLOW_ECCO_OPTIMIZATION
511          optimcycle = 0
512    #endif
513          lastdiva = .TRUE.
514    
515    #ifdef ALLOW_DEBUG
516          IF (debugMode) CALL DEBUG_ENTER('THE_MODEL_MAIN',myThid)
517    #endif
518    
519    C--   This timer encompasses the whole code
520          CALL TIMER_START('ALL                    [THE_MODEL_MAIN]',myThid)
521    
522    #ifdef ALLOW_DEBUG
523          IF (debugMode) CALL DEBUG_CALL('INITIALISE_FIXED',myThid)
524    #endif
525    
526    C--   Set model configuration (fixed arrays)
527          CALL TIMER_START('INITIALISE_FIXED       [THE_MODEL_MAIN]',myThid)
528          CALL INITIALISE_FIXED( myThid )
529          CALL TIMER_STOP ('INITIALISE_FIXED       [THE_MODEL_MAIN]',myThid)
530    
 C--   Set model initial conditions  
       CALL INITIALISE( myThid )  
531        myCurrentTime = startTime        myCurrentTime = startTime
532        myCurrentIter = nIter0        myCurrentIter = nIter0
533    
534  C--   Dump for start state  #if ( defined (ALLOW_ADMTLM) )
535        CALL WRITE_STATE( .TRUE., myCurrentTime, myCurrentIter, myThid )  c
536          CALL TIMER_START('ADMTLM_DRIVER          [THE_MODEL_MAIN]',mythid)
537          CALL ADMTLM_DRIVER ( myThid )
538          CALL TIMER_STOP ('ADMTLM_DRIVER          [THE_MODEL_MAIN]',mythid)
539    c
540    #elif ( defined (ALLOW_AUTODIFF))
541    
542    # ifndef EXCLUDE_CTRL_PACK
543          _BEGIN_MASTER( mythid )
544          IF (myProcId .eq. 0) THEN
545             inquire( file='costfinal', exist=exst )
546             IF ( .NOT. exst .AND. niter0 .EQ. 0 ) THEN
547                IF ( optimcycle.NE.0 ) THEN
548                   CALL TIMER_START('CTRL_UNPACK   [THE_MODEL_MAIN]',mythid)
549                   CALL CTRL_UNPACK( .TRUE. , mythid )
550                   CALL TIMER_STOP ('CTRL_UNPACK   [THE_MODEL_MAIN]',mythid)
551                ENDIF
552                IF ( optimcycle.EQ.0 .AND. mycurrentiter.EQ.niter0 ) THEN
553                   CALL TIMER_START('CTRL_PACK     [THE_MODEL_MAIN]',mythid)
554                   CALL CTRL_PACK( .TRUE. , mythid )
555                   CALL TIMER_STOP ('CTRL_PACK     [THE_MODEL_MAIN]',mythid)
556                ENDIF
557             ENDIF
558          ENDIF
559          _END_MASTER( mythid )
560          _BARRIER
561    # endif /* EXCLUDE_CTRL_PACK */
562    
563          CALL COST_DEPENDENT_INIT ( mythid )
564          _BARRIER
565    
566    # if ( defined (ALLOW_TANGENTLINEAR_RUN) )
567    c
568          CALL TIMER_START('G_THE_MAIN_LOOP           [TANGENT RUN]',mythid)
569          CALL G_THE_MAIN_LOOP ( myCurrentTime, myCurrentIter, myThid )
570          CALL TIMER_STOP ('G_THE_MAIN_LOOP           [TANGENT RUN]',mythid)
571    c
572    # elif ( defined (ALLOW_ADJOINT_RUN) || \
573             defined (ALLOW_ECCO_OPTIMIZATION) )
574    c
575    #  ifdef ALLOW_DIVIDED_ADJOINT
576    c-- The following assumes the TAF option '-pure'
577          inquire( file='costfinal', exist=exst )
578          IF ( .NOT. exst) THEN
579             CALL TIMER_START('MDTHE_MAIN_LOOP            [MD RUN]', mythid)
580             CALL MDTHE_MAIN_LOOP ( myCurrentTime, myCurrentIter, mythid )
581             CALL TIMER_STOP ('MDTHE_MAIN_LOOP            [MD RUN]', mythid)
582             CALL COST_FINAL_STORE ( mythid, lastdiva )
583          ELSE
584             CALL TIMER_START('ADTHE_MAIN_LOOP       [ADJOINT RUN]', mythid)
585             CALL ADTHE_MAIN_LOOP ( myCurrentTime, myCurrentIter, mythid )
586             CALL TIMER_STOP ('ADTHE_MAIN_LOOP       [ADJOINT RUN]', mythid)
587             CALL COST_FINAL_RESTORE ( mythid, lastdiva )
588          ENDIF
589    c--
590    #  else /* ALLOW_DIVIDED_ADJOINT undef */
591          CALL TIMER_START('ADTHE_MAIN_LOOP          [ADJOINT RUN]', mythid)
592          CALL ADTHE_MAIN_LOOP ( myCurrentTime, myCurrentIter, mythid )
593    cph      CALL ADTHE_MAIN_LOOP ( mythid )
594          CALL TIMER_STOP ('ADTHE_MAIN_LOOP          [ADJOINT RUN]', mythid)
595    #  endif /* ALLOW_DIVIDED_ADJOINT */
596    c
597    # endif
598    
599    # ifndef EXCLUDE_CTRL_PACK
600          _BEGIN_MASTER( mythid )
601          IF ( myProcId .eq. 0 .AND. lastdiva ) THEN
602             CALL TIMER_START('CTRL_PACK           [THE_MODEL_MAIN]',mythid)
603             call CTRL_PACK( .FALSE. , mythid )
604             CALL TIMER_STOP ('CTRL_PACK           [THE_MODEL_MAIN]',mythid)
605          ENDIF
606          _END_MASTER( mythid )
607          _BARRIER
608    # endif /* EXCLUDE_CTRL_PACK */
609    
 C--   Begin time stepping loop  
       DO I=1, nTimeSteps  
610    
611  C--    Load forcing/external data fields  # ifdef ALLOW_GRDCHK
612         CALL LOAD_EXTERNAL_FIELDS( myCurrentTime, myCurrentIter, myThid )        IF ( useGrdchk .AND. lastdiva ) THEN
613             CALL TIMER_START('GRDCHK_MAIN         [THE_MODEL_MAIN]',mythid)
614             CALL GRDCHK_MAIN( mythid )
615             CALL TIMER_STOP ('GRDCHK_MAIN         [THE_MODEL_MAIN]',mythid)
616             _BARRIER
617          ENDIF
618    # endif
619    
620    #else /* ALL AD-related undef */
621    
622  C--    Step forward fields and calculate time tendency terms  #ifdef ALLOW_DEBUG
623         CALL DYNAMICS( myCurrentTime, myCurrentIter, myThid )        IF (debugMode) CALL DEBUG_CALL('THE_MAIN_LOOP',myThid)
624    #endif
625  C--    Do time averages  
626         IF (taveFreq.GT.0.) THEN  C--   Call time stepping loop of full model
627           CALL DO_TIME_AVERAGES( myCurrentTime, myCurrentIter, myThid )        CALL TIMER_START('THE_MAIN_LOOP          [THE_MODEL_MAIN]',myThid)
628         ENDIF        CALL THE_MAIN_LOOP( myCurrentTime, myCurrentIter, myThid )
629          CALL TIMER_STOP ('THE_MAIN_LOOP          [THE_MODEL_MAIN]',myThid)
630  C--    Do IO if needed.  
631  C      Note:  #endif /* ALLOW_TANGENTLINEAR_RUN ALLOW_ADJOINT_RUN ALLOW_ADMTLM_RUN */
632  C      =====  
633  C      At this point model arrays hold U,V,T,S  at "time-level" N  C--   Final checkpoint (in case the in-loop checkpoint was missed)
634  C      and cg2d_x at "time-level" N-1/2 where N = I+timeLevBase-1.        CALL TIMER_START('WRITE_CHECKPOINT       [THE_MODEL_MAIN]',myThid)
635  C      By convention this is taken to be the model "state".        CALL PACKAGES_WRITE_PICKUP(
636         CALL DO_THE_MODEL_IO( myCurrentTime, myCurrentIter, myThid )       I        .TRUE., myCurrentTime, myCurrentIter, myThid )
637          CALL WRITE_CHECKPOINT(
638  C--    Solve elliptic equation(s).       I        .TRUE., myCurrentTime, myCurrentIter, myThid )
639  C      Two-dimensional only for conventional hydrostatic or        CALL TIMER_STOP ('WRITE_CHECKPOINT       [THE_MODEL_MAIN]',myThid)
640  C      three-dimensional for non-hydrostatic and/or IGW scheme.  
641         CALL SOLVE_FOR_PRESSURE( myThid )        CALL TIMER_STOP ('ALL                    [THE_MODEL_MAIN]',myThid)
642    
643  C--    Do "blocking" sends and receives for tendency "overlap" terms  #ifdef ALLOW_MNC
644         CALL DO_GTERM_BLOCKING_EXCHANGES( myThid )        IF (useMNC) THEN
645    C       Close all open NetCDF files
646         myCurrentIter = myCurrentIter + 1          _BEGIN_MASTER( mythid )
647         myCurrentTime = myCurrentTime + deltaTClock          CALL MNC_FILE_CLOSE_ALL( myThid )
648            _END_MASTER( mythid )
 C--    Save state for restarts  
 C      Note:  
 C      =====  
 C      Because of the ordering of the timestepping code and  
 C      tendency term code at end of loop model arrays hold  
 C      U,V,T,S  at "time-level" N but gu, gv, gs, gt, guNM1,...  
 C      at "time-level" N+1/2 (guNM1 at "time-level" N+1/2 is  
 C      gu at "time-level" N-1/2) and cg2d_x at "time-level" N+1/2.  
 C       where N = I+timeLevBase-1  
 C      Thus a checkpoint contains U.0000000000, GU.0000000001 and  
 C      cg2d_x.0000000001 in the indexing scheme used for the model  
 C      "state" files. This example is referred to as a checkpoint  
 C      at time level 1  
        CALL WRITE_CHECKPOINT( .FALSE., myCurrentTime, myCurrentIter, myThid )  
   
       ENDDO  
   
 C--   Final checkpoint (incase the in-loop checkpoint was missed)  
       CALL WRITE_CHECKPOINT( .TRUE., myCurrentTime, myCurrentIter, myThid )  
   
 C--   Step-forward U/V/Theta/Salt for purposes of final I/O dump  
       CALL DYNAMICS( myCurrentTime, myCurrentIter, myThid )  
   
 C--   Do time averages  
       IF (taveFreq.GT.0.) THEN  
         CALL DO_TIME_AVERAGES( myCurrentTime, myCurrentIter, myThid )  
649        ENDIF        ENDIF
650    #endif
651    
652  C--   Dump for end state  C--   Write timer statistics
653        CALL WRITE_STATE( .FALSE., myCurrentTime, myCurrentIter, myThid )        IF ( myThid .EQ. 1 ) THEN
654           CALL TIMER_PRINTALL( myThid )
655           CALL COMM_STATS
656          ENDIF
657    
658    #ifdef ALLOW_DEBUG
659          IF (debugMode) CALL DEBUG_LEAVE('THE_MODEL_MAIN',myThid)
660    #endif
661    
662        RETURN        RETURN
663        END        END

Legend:
Removed from v.1.16  
changed lines
  Added in v.1.79

  ViewVC Help
Powered by ViewVC 1.1.22