PageRenderTime 59ms CodeModel.GetById 20ms RepoModel.GetById 0ms app.codeStats 0ms

/WPS/geogrid/src/process_tile_module.F

http://github.com/jbeezley/wrf-fire
FORTRAN Legacy | 2084 lines | 1265 code | 310 blank | 509 comment | 366 complexity | 039d3d2e0118ec2527dcd4f13fac8fbe MD5 | raw file
Possible License(s): AGPL-1.0

Large files files are truncated, but you can click here to view the full file

  1. !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  2. ! Module: process_tile
  3. !
  4. ! Description:
  5. !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  6. module process_tile_module
  7. use module_debug
  8. contains
  9. !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  10. ! Name: process_tile
  11. !
  12. ! Purpose: To process a tile, whose lower-left corner is at
  13. ! (tile_i_min, tile_j_min) and whose upper-right corner is at
  14. ! (tile_i_max, tile_j_max), of the model grid given by which_domain
  15. !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  16. subroutine process_tile(which_domain, grid_type, dynopt, &
  17. dummy_start_dom_i, dummy_end_dom_i, &
  18. dummy_start_dom_j, dummy_end_dom_j, &
  19. dummy_start_patch_i, dummy_end_patch_i, &
  20. dummy_start_patch_j, dummy_end_patch_j, &
  21. extra_col, extra_row)
  22. use bitarray_module
  23. use hash_module
  24. use llxy_module
  25. use misc_definitions_module
  26. use output_module
  27. use smooth_module
  28. use source_data_module
  29. use gridinfo_module, only : subgrid_ratio_x, subgrid_ratio_y
  30. implicit none
  31. ! Arguments
  32. integer, intent(in) :: which_domain, dynopt, &
  33. dummy_start_dom_i, dummy_end_dom_i, dummy_start_dom_j, dummy_end_dom_j, &
  34. dummy_start_patch_i, dummy_end_patch_i, dummy_start_patch_j, dummy_end_patch_j
  35. logical, intent(in) :: extra_col, extra_row
  36. character (len=1), intent(in) :: grid_type
  37. ! Local variables
  38. integer :: i, j, k, kk, istatus, ifieldstatus, idomcatstatus, field_count
  39. integer :: min_category, max_category, min_level, max_level, &
  40. smth_opt, smth_passes, num_landmask_categories
  41. integer :: start_dom_i, end_dom_i, start_dom_j, end_dom_j, end_dom_stag_i, end_dom_stag_j
  42. integer :: start_patch_i, end_patch_i, start_patch_j, end_patch_j, end_patch_stag_i, end_patch_stag_j
  43. integer :: start_mem_i, end_mem_i, start_mem_j, end_mem_j, end_mem_stag_i, end_mem_stag_j
  44. integer :: start_mem_sub_i, end_mem_sub_i, start_mem_sub_j, end_mem_sub_j
  45. integer :: sm1, em1, sm2, em2
  46. integer :: istagger
  47. integer, dimension(MAX_LANDMASK_CATEGORIES) :: landmask_value
  48. real :: sum, dominant, msg_fill_val, topo_flag_val, mass_flag, land_total, water_total
  49. real, dimension(16) :: corner_lats, corner_lons
  50. real, pointer, dimension(:,:) :: xlat_array, xlon_array, &
  51. xlat_array_u, xlon_array_u, &
  52. xlat_array_v, xlon_array_v, &
  53. clat_array, clon_array, &
  54. xlat_array_subgrid, xlon_array_subgrid, &
  55. f_array, e_array, &
  56. mapfac_array_m_x, mapfac_array_u_x, mapfac_array_v_x, &
  57. mapfac_array_m_y, mapfac_array_u_y, mapfac_array_v_y, &
  58. mapfac_array_x_subgrid, mapfac_array_y_subgrid, &
  59. sina_array, cosa_array
  60. real, pointer, dimension(:,:) :: xlat_ptr, xlon_ptr, mapfac_ptr_x, mapfac_ptr_y, landmask, dominant_field
  61. real, pointer, dimension(:,:,:) :: field, slp_field
  62. logical :: is_water_mask, only_save_dominant, halt_on_missing
  63. character (len=19) :: datestr
  64. character (len=128) :: fieldname, gradname, domname, landmask_name
  65. character (len=256) :: temp_string
  66. type (bitarray) :: processed_domain
  67. type (hashtable) :: processed_fieldnames
  68. character (len=128), dimension(2) :: dimnames
  69. integer :: sr_x, sr_y
  70. logical :: subgrid_var
  71. sr_x=subgrid_ratio_x(which_domain)
  72. sr_y=subgrid_ratio_y(which_domain)
  73. datestr = '0000-00-00_00:00:00'
  74. field_count = 0
  75. mass_flag=1.0
  76. ! The following pertains primarily to the C grid
  77. ! Determine whether only (n-1)th rows/columns should be computed for variables
  78. ! on staggered grid. In a distributed memory situation, not every tile should
  79. ! have only (n-1)th rows/columns computed, or we end up with (n-k)
  80. ! rows/columns when there are k patches in the y/x direction
  81. if (extra_col) then
  82. start_patch_i = dummy_start_patch_i ! The seemingly pointless renaming of start
  83. end_patch_i = dummy_end_patch_i - 1 ! naming convention with modified end_patch variables,
  84. end_patch_stag_i = dummy_end_patch_i ! variables is so that we can maintain consistent
  85. ! which are marked as intent(in)
  86. start_mem_i = start_patch_i - HALO_WIDTH
  87. end_mem_i = end_patch_i + HALO_WIDTH
  88. end_mem_stag_i = end_patch_stag_i + HALO_WIDTH
  89. else
  90. start_patch_i = dummy_start_patch_i
  91. end_patch_i = dummy_end_patch_i
  92. end_patch_stag_i = dummy_end_patch_i
  93. start_mem_i = start_patch_i - HALO_WIDTH
  94. end_mem_i = end_patch_i + HALO_WIDTH
  95. end_mem_stag_i = end_patch_stag_i + HALO_WIDTH
  96. end if
  97. if (extra_row) then
  98. start_patch_j = dummy_start_patch_j
  99. end_patch_j = dummy_end_patch_j - 1
  100. end_patch_stag_j = dummy_end_patch_j
  101. start_mem_j = start_patch_j - HALO_WIDTH
  102. end_mem_j = end_patch_j + HALO_WIDTH
  103. end_mem_stag_j = end_patch_stag_j + HALO_WIDTH
  104. else
  105. start_patch_j = dummy_start_patch_j
  106. end_patch_j = dummy_end_patch_j
  107. end_patch_stag_j = dummy_end_patch_j
  108. start_mem_j = start_patch_j - HALO_WIDTH
  109. end_mem_j = end_patch_j + HALO_WIDTH
  110. end_mem_stag_j = end_patch_stag_j + HALO_WIDTH
  111. end if
  112. start_dom_i = dummy_start_dom_i
  113. if (grid_type == 'C') then
  114. end_dom_i = dummy_end_dom_i - 1
  115. end_dom_stag_i = dummy_end_dom_i
  116. else if (grid_type == 'E') then
  117. end_dom_i = dummy_end_dom_i
  118. end_dom_stag_i = dummy_end_dom_i
  119. end if
  120. start_dom_j = dummy_start_dom_j
  121. if (grid_type == 'C') then
  122. end_dom_j = dummy_end_dom_j - 1
  123. end_dom_stag_j = dummy_end_dom_j
  124. else if (grid_type == 'E') then
  125. end_dom_j = dummy_end_dom_j
  126. end_dom_stag_j = dummy_end_dom_j
  127. end if
  128. start_mem_sub_i = (start_mem_i - 1 ) * sr_x + 1
  129. if(.not.extra_col)then
  130. end_mem_sub_i = (end_mem_i) * sr_x
  131. else
  132. end_mem_sub_i = (end_mem_i + 1) * sr_x
  133. end if
  134. start_mem_sub_j = (start_mem_j -1 ) * sr_y + 1
  135. if(.not.extra_row)then
  136. end_mem_sub_j = (end_mem_j) * sr_y
  137. else
  138. end_mem_sub_j = (end_mem_j + 1) * sr_y
  139. end if
  140. ! Allocate arrays to hold all lat/lon fields; these will persist for the duration of
  141. ! the process_tile routine
  142. ! For C grid, we have M, U, and V points
  143. ! For E grid, we have only M and V points
  144. allocate(xlat_array(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  145. allocate(xlon_array(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  146. allocate(xlat_array_v(start_mem_i:end_mem_i, start_mem_j:end_mem_stag_j))
  147. allocate(xlon_array_v(start_mem_i:end_mem_i, start_mem_j:end_mem_stag_j))
  148. if (grid_type == 'C') then
  149. allocate(xlat_array_u(start_mem_i:end_mem_stag_i, start_mem_j:end_mem_j))
  150. allocate(xlon_array_u(start_mem_i:end_mem_stag_i, start_mem_j:end_mem_j))
  151. allocate(clat_array(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  152. allocate(clon_array(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  153. allocate(xlat_array_subgrid(start_mem_sub_i:end_mem_sub_i,start_mem_sub_j:end_mem_sub_j))
  154. allocate(xlon_array_subgrid(start_mem_sub_i:end_mem_sub_i,start_mem_sub_j:end_mem_sub_j))
  155. allocate(mapfac_array_x_subgrid(start_mem_sub_i:end_mem_sub_i,start_mem_sub_j:end_mem_sub_j))
  156. allocate(mapfac_array_y_subgrid(start_mem_sub_i:end_mem_sub_i,start_mem_sub_j:end_mem_sub_j))
  157. end if
  158. ! Initialize hash table to track which fields have been processed
  159. call hash_init(processed_fieldnames)
  160. !
  161. ! Calculate lat/lon for every point in the tile (XLAT and XLON)
  162. ! The xlat_array and xlon_array arrays will be used in processing other fields
  163. !
  164. call mprintf(.true.,STDOUT,' Processing XLAT and XLONG')
  165. if (grid_type == 'C') then
  166. call get_lat_lon_fields(xlat_array, xlon_array, start_mem_i, &
  167. start_mem_j, end_mem_i, end_mem_j, M)
  168. call get_lat_lon_fields(xlat_array_v, xlon_array_v, start_mem_i, &
  169. start_mem_j, end_mem_i, end_mem_stag_j, V)
  170. call get_lat_lon_fields(xlat_array_u, xlon_array_u, start_mem_i, &
  171. start_mem_j, end_mem_stag_i, end_mem_j, U)
  172. call get_lat_lon_fields(clat_array, clon_array, start_mem_i, &
  173. start_mem_j, end_mem_i, end_mem_j, M, comp_ll=.true.)
  174. call get_lat_lon_fields(xlat_array_subgrid, xlon_array_subgrid, &
  175. start_mem_sub_i, start_mem_sub_j, end_mem_sub_i, end_mem_sub_j, M, sub_x=sr_x, sub_y=sr_y)
  176. call get_map_factor(xlat_array_subgrid, xlon_array_subgrid, mapfac_array_x_subgrid, &
  177. mapfac_array_y_subgrid, start_mem_sub_i, start_mem_sub_j, end_mem_sub_i, end_mem_sub_j)
  178. corner_lats(1) = xlat_array(start_patch_i,start_patch_j)
  179. corner_lats(2) = xlat_array(start_patch_i,end_patch_j)
  180. corner_lats(3) = xlat_array(end_patch_i,end_patch_j)
  181. corner_lats(4) = xlat_array(end_patch_i,start_patch_j)
  182. corner_lats(5) = xlat_array_u(start_patch_i,start_patch_j)
  183. corner_lats(6) = xlat_array_u(start_patch_i,end_patch_j)
  184. corner_lats(7) = xlat_array_u(end_patch_stag_i,end_patch_j)
  185. corner_lats(8) = xlat_array_u(end_patch_stag_i,start_patch_j)
  186. corner_lats(9) = xlat_array_v(start_patch_i,start_patch_j)
  187. corner_lats(10) = xlat_array_v(start_patch_i,end_patch_stag_j)
  188. corner_lats(11) = xlat_array_v(end_patch_i,end_patch_stag_j)
  189. corner_lats(12) = xlat_array_v(end_patch_i,start_patch_j)
  190. call xytoll(real(start_patch_i)-0.5, real(start_patch_j)-0.5, corner_lats(13), corner_lons(13), M)
  191. call xytoll(real(start_patch_i)-0.5, real(end_patch_j)+0.5, corner_lats(14), corner_lons(14), M)
  192. call xytoll(real(end_patch_i)+0.5, real(end_patch_j)+0.5, corner_lats(15), corner_lons(15), M)
  193. call xytoll(real(end_patch_i)+0.5, real(start_patch_j)-0.5, corner_lats(16), corner_lons(16), M)
  194. corner_lons(1) = xlon_array(start_patch_i,start_patch_j)
  195. corner_lons(2) = xlon_array(start_patch_i,end_patch_j)
  196. corner_lons(3) = xlon_array(end_patch_i,end_patch_j)
  197. corner_lons(4) = xlon_array(end_patch_i,start_patch_j)
  198. corner_lons(5) = xlon_array_u(start_patch_i,start_patch_j)
  199. corner_lons(6) = xlon_array_u(start_patch_i,end_patch_j)
  200. corner_lons(7) = xlon_array_u(end_patch_stag_i,end_patch_j)
  201. corner_lons(8) = xlon_array_u(end_patch_stag_i,start_patch_j)
  202. corner_lons(9) = xlon_array_v(start_patch_i,start_patch_j)
  203. corner_lons(10) = xlon_array_v(start_patch_i,end_patch_stag_j)
  204. corner_lons(11) = xlon_array_v(end_patch_i,end_patch_stag_j)
  205. corner_lons(12) = xlon_array_v(end_patch_i,start_patch_j)
  206. else if (grid_type == 'E') then
  207. call get_lat_lon_fields(xlat_array, xlon_array, start_mem_i, &
  208. start_mem_j, end_mem_i, end_mem_j, HH)
  209. call get_lat_lon_fields(xlat_array_v, xlon_array_v, start_mem_i, &
  210. start_mem_j, end_mem_i, end_mem_stag_j, VV)
  211. corner_lats(1) = xlat_array(start_patch_i,start_patch_j)
  212. corner_lats(2) = xlat_array(start_patch_i,end_patch_j)
  213. corner_lats(3) = xlat_array(end_patch_i,end_patch_j)
  214. corner_lats(4) = xlat_array(end_patch_i,start_patch_j)
  215. corner_lats(5) = xlat_array_v(start_patch_i,start_patch_j)
  216. corner_lats(6) = xlat_array_v(start_patch_i,end_patch_stag_j)
  217. corner_lats(7) = xlat_array_v(end_patch_i,end_patch_stag_j)
  218. corner_lats(8) = xlat_array_v(end_patch_i,start_patch_j)
  219. corner_lats(9) = 0.0
  220. corner_lats(10) = 0.0
  221. corner_lats(11) = 0.0
  222. corner_lats(12) = 0.0
  223. corner_lats(13) = 0.0
  224. corner_lats(14) = 0.0
  225. corner_lats(15) = 0.0
  226. corner_lats(16) = 0.0
  227. corner_lons(1) = xlon_array(start_patch_i,start_patch_j)
  228. corner_lons(2) = xlon_array(start_patch_i,end_patch_j)
  229. corner_lons(3) = xlon_array(end_patch_i,end_patch_j)
  230. corner_lons(4) = xlon_array(end_patch_i,start_patch_j)
  231. corner_lons(5) = xlon_array_v(start_patch_i,start_patch_j)
  232. corner_lons(6) = xlon_array_v(start_patch_i,end_patch_stag_j)
  233. corner_lons(7) = xlon_array_v(end_patch_i,end_patch_stag_j)
  234. corner_lons(8) = xlon_array_v(end_patch_i,start_patch_j)
  235. corner_lons(9) = 0.0
  236. corner_lons(10) = 0.0
  237. corner_lons(11) = 0.0
  238. corner_lons(12) = 0.0
  239. corner_lons(13) = 0.0
  240. corner_lons(14) = 0.0
  241. corner_lons(15) = 0.0
  242. corner_lons(16) = 0.0
  243. end if
  244. ! Initialize the output module now that we have the corner point lats/lons
  245. call output_init(which_domain, 'OUTPUT FROM GEOGRID V3.3', '0000-00-00_00:00:00', grid_type, dynopt, &
  246. corner_lats, corner_lons, &
  247. start_dom_i, end_dom_i, start_dom_j, end_dom_j, &
  248. start_patch_i, end_patch_i, start_patch_j, end_patch_j, &
  249. start_mem_i, end_mem_i, start_mem_j, end_mem_j, &
  250. extra_col, extra_row)
  251. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, &
  252. 'XLAT_M', datestr, real_array = xlat_array)
  253. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, &
  254. 'XLONG_M', datestr, real_array = xlon_array)
  255. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_stag_j, 1, 1, &
  256. 'XLAT_V', datestr, real_array = xlat_array_v)
  257. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_stag_j, 1, 1, &
  258. 'XLONG_V', datestr, real_array = xlon_array_v)
  259. if (grid_type == 'C') then
  260. call write_field(start_mem_i, end_mem_stag_i, start_mem_j, end_mem_j, 1, 1, &
  261. 'XLAT_U', datestr, real_array = xlat_array_u)
  262. call write_field(start_mem_i, end_mem_stag_i, start_mem_j, end_mem_j, 1, 1, &
  263. 'XLONG_U', datestr, real_array = xlon_array_u)
  264. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, &
  265. 'CLAT', datestr, real_array = clat_array)
  266. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, &
  267. 'CLONG', datestr, real_array = clon_array)
  268. call write_field(start_mem_sub_i, end_mem_sub_i, start_mem_sub_j, end_mem_sub_j, 1, 1, &
  269. 'FXLAT', datestr, real_array = xlat_array_subgrid)
  270. call write_field(start_mem_sub_i, end_mem_sub_i, start_mem_sub_j, end_mem_sub_j, 1, 1, &
  271. 'FXLONG', datestr, real_array = xlon_array_subgrid)
  272. if (associated(clat_array)) deallocate(clat_array)
  273. if (associated(clon_array)) deallocate(clon_array)
  274. end if
  275. !
  276. ! Calculate map factor for current domain
  277. !
  278. if (grid_type == 'C') then
  279. call mprintf(.true.,STDOUT,' Processing MAPFAC')
  280. allocate(mapfac_array_m_x(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  281. allocate(mapfac_array_m_y(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  282. call get_map_factor(xlat_array, xlon_array, mapfac_array_m_x, mapfac_array_m_y, start_mem_i, &
  283. start_mem_j, end_mem_i, end_mem_j)
  284. ! Global WRF uses map scale factors in X and Y directions, but "regular" WRF uses a single MSF
  285. ! on each staggering. In the case of regular WRF, we can assume that MAPFAC_MX = MAPFAC_MY = MAPFAC_M,
  286. ! and so we can simply write MAPFAC_MX as the MAPFAC_M field. Ultimately, when global WRF is
  287. ! merged into the WRF trunk, we will need only two map scale factor fields for each staggering,
  288. ! in the x and y directions, and these will be the same in the case of non-Cassini projections
  289. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, 'MAPFAC_M', &
  290. datestr, real_array = mapfac_array_m_x)
  291. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, 'MAPFAC_MX', &
  292. datestr, real_array = mapfac_array_m_x)
  293. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, 'MAPFAC_MY', &
  294. datestr, real_array = mapfac_array_m_y)
  295. allocate(mapfac_array_v_x(start_mem_i:end_mem_i, start_mem_j:end_mem_stag_j))
  296. allocate(mapfac_array_v_y(start_mem_i:end_mem_i, start_mem_j:end_mem_stag_j))
  297. call get_map_factor(xlat_array_v, xlon_array_v, mapfac_array_v_x, mapfac_array_v_y, start_mem_i, &
  298. start_mem_j, end_mem_i, end_mem_stag_j)
  299. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_stag_j, 1, 1, 'MAPFAC_V', &
  300. datestr, real_array = mapfac_array_v_x)
  301. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_stag_j, 1, 1, 'MAPFAC_VX', &
  302. datestr, real_array = mapfac_array_v_x)
  303. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_stag_j, 1, 1, 'MAPFAC_VY', &
  304. datestr, real_array = mapfac_array_v_y)
  305. allocate(mapfac_array_u_x(start_mem_i:end_mem_stag_i, start_mem_j:end_mem_j))
  306. allocate(mapfac_array_u_y(start_mem_i:end_mem_stag_i, start_mem_j:end_mem_j))
  307. call get_map_factor(xlat_array_u, xlon_array_u, mapfac_array_u_x, mapfac_array_u_y, start_mem_i, &
  308. start_mem_j, end_mem_stag_i, end_mem_j)
  309. call write_field(start_mem_i, end_mem_stag_i, start_mem_j, end_mem_j, 1, 1, 'MAPFAC_U', &
  310. datestr, real_array = mapfac_array_u_x)
  311. call write_field(start_mem_i, end_mem_stag_i, start_mem_j, end_mem_j, 1, 1, 'MAPFAC_UX', &
  312. datestr, real_array = mapfac_array_u_x)
  313. call write_field(start_mem_i, end_mem_stag_i, start_mem_j, end_mem_j, 1, 1, 'MAPFAC_UY', &
  314. datestr, real_array = mapfac_array_u_y)
  315. end if
  316. !
  317. ! Coriolis parameters (E and F)
  318. !
  319. call mprintf(.true.,STDOUT,' Processing F and E')
  320. allocate(f_array(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  321. allocate(e_array(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  322. call get_coriolis_parameters(xlat_array, f_array, e_array, &
  323. start_mem_i, start_mem_j, end_mem_i, end_mem_j)
  324. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, 'E', &
  325. datestr, real_array = e_array)
  326. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, 'F', &
  327. datestr, real_array = f_array)
  328. if (associated(f_array)) deallocate(f_array)
  329. if (associated(e_array)) deallocate(e_array)
  330. !
  331. ! Rotation angle (SINALPHA and COSALPHA)
  332. !
  333. if (grid_type == 'C') then
  334. call mprintf(.true.,STDOUT,' Processing ROTANG')
  335. allocate(sina_array(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  336. allocate(cosa_array(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  337. call get_rotang(xlat_array, xlon_array, cosa_array, sina_array, &
  338. start_mem_i, start_mem_j, end_mem_i, end_mem_j)
  339. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, 'SINALPHA', &
  340. datestr, real_array = sina_array)
  341. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, 'COSALPHA', &
  342. datestr, real_array = cosa_array)
  343. if (associated(sina_array)) deallocate(sina_array)
  344. if (associated(cosa_array)) deallocate(cosa_array)
  345. end if
  346. ! Every field up until now should probably just be processed regardless of what the user
  347. ! has specified for fields to be processed.
  348. ! Hereafter, we process user-specified fields
  349. !
  350. ! First process the field that we will derive a landmask from
  351. !
  352. call get_landmask_field(geog_data_res(which_domain), landmask_name, is_water_mask, landmask_value, istatus)
  353. do kk=1,MAX_LANDMASK_CATEGORIES
  354. if (landmask_value(kk) == INVALID) then
  355. num_landmask_categories = kk-1
  356. exit
  357. end if
  358. end do
  359. if (kk > MAX_LANDMASK_CATEGORIES) num_landmask_categories = MAX_LANDMASK_CATEGORIES
  360. if (istatus /= 0) then
  361. call mprintf(.true.,WARN,'No field specified for landmask calculation. Will set landmask=1 at every grid point.')
  362. allocate(landmask(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  363. landmask = 1.
  364. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, 'LANDMASK', &
  365. datestr, landmask)
  366. else
  367. allocate(landmask(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  368. landmask = 1.
  369. call mprintf(.true.,STDOUT,' Processing %s', s1=trim(landmask_name))
  370. call get_missing_fill_value(landmask_name, msg_fill_val, istatus)
  371. if (istatus /= 0) msg_fill_val = NAN
  372. call get_halt_on_missing(landmask_name, halt_on_missing, istatus)
  373. if (istatus /= 0) halt_on_missing = .false.
  374. ! Do we calculate a dominant category for this field?
  375. call get_domcategory_name(landmask_name, domname, only_save_dominant, idomcatstatus)
  376. temp_string = ' '
  377. temp_string(1:128) = landmask_name
  378. call hash_insert(processed_fieldnames, temp_string)
  379. call get_max_categories(landmask_name, min_category, max_category, istatus)
  380. allocate(field(start_mem_i:end_mem_i, start_mem_j:end_mem_j, min_category:max_category))
  381. if (.not. only_save_dominant) then
  382. field_count = field_count + 1
  383. call mprintf(.true.,LOGFILE,'Processing field %i of %i (%s)', &
  384. i1=field_count,i2=NUM_FIELDS-NUM_AUTOMATIC_FIELDS,s1=landmask_name)
  385. else
  386. field_count = field_count + 1
  387. call mprintf(.true.,LOGFILE,'Processing field %i of %i (%s)', &
  388. i1=field_count,i2=NUM_FIELDS-NUM_AUTOMATIC_FIELDS,s1=domname)
  389. end if
  390. if (grid_type == 'C') then
  391. call calc_field(landmask_name, field, xlat_array, xlon_array, M, &
  392. start_mem_i, end_mem_i, start_mem_j, end_mem_j, &
  393. min_category, max_category, processed_domain, 1, landmask=landmask, sr_x=1, sr_y=1)
  394. else if (grid_type == 'E') then
  395. call calc_field(landmask_name, field, xlat_array, xlon_array, HH, &
  396. start_mem_i, end_mem_i, start_mem_j, end_mem_j, &
  397. min_category, max_category, processed_domain, 1, landmask=landmask, sr_x=1, sr_y=1)
  398. end if
  399. ! Find fractions
  400. do i=start_mem_i, end_mem_i
  401. do j=start_mem_j, end_mem_j
  402. sum = 0.
  403. do k=min_category,max_category
  404. sum = sum + field(i,j,k)
  405. end do
  406. if (sum > 0.0) then
  407. do k=min_category,max_category
  408. field(i,j,k) = field(i,j,k) / sum
  409. end do
  410. else
  411. do k=min_category,max_category
  412. field(i,j,k) = msg_fill_val
  413. end do
  414. end if
  415. end do
  416. end do
  417. ! If user wants to halt when a missing value is found in output field, check now
  418. if (halt_on_missing) then
  419. do i=start_mem_i, end_mem_i
  420. do j=start_mem_j, end_mem_j
  421. ! Only need to examine k=1
  422. if (field(i,j,1) == msg_fill_val) then
  423. call mprintf(.true.,ERROR,' Missing value encountered in output field. Quitting.')
  424. end if
  425. end do
  426. end do
  427. end if
  428. if (is_water_mask) then
  429. call mprintf(.true.,STDOUT,' Calculating landmask from %s ( WATER =', &
  430. newline=.false.,s1=trim(landmask_name))
  431. else
  432. call mprintf(.true.,STDOUT,' Calculating landmask from %s ( LAND =', &
  433. newline=.false.,s1=trim(landmask_name))
  434. end if
  435. do k = 1, num_landmask_categories
  436. call mprintf(.true.,STDOUT,' %i',newline=.false.,i1=landmask_value(k))
  437. if (k == num_landmask_categories) call mprintf(.true.,STDOUT,')')
  438. end do
  439. ! Calculate landmask
  440. if (is_water_mask) then
  441. do i=start_mem_i, end_mem_i
  442. do j=start_mem_j, end_mem_j
  443. water_total = -1.
  444. do k=1,num_landmask_categories
  445. if (landmask_value(k) >= min_category .and. landmask_value(k) <= max_category) then
  446. if (field(i,j,landmask_value(k)) /= msg_fill_val) then
  447. if (water_total < 0.) water_total = 0.
  448. water_total = water_total + field(i,j,landmask_value(k))
  449. else
  450. water_total = -1.
  451. exit
  452. end if
  453. end if
  454. end do
  455. if (water_total >= 0.0) then
  456. if (water_total < 0.50) then
  457. landmask(i,j) = 1.
  458. else
  459. landmask(i,j) = 0.
  460. end if
  461. else
  462. landmask(i,j) = -1.
  463. end if
  464. end do
  465. end do
  466. else
  467. do i=start_mem_i, end_mem_i
  468. do j=start_mem_j, end_mem_j
  469. land_total = -1.
  470. do k=1,num_landmask_categories
  471. if (landmask_value(k) >= min_category .and. landmask_value(k) <= max_category) then
  472. if (field(i,j,landmask_value(k)) /= msg_fill_val) then
  473. if (land_total < 0.) land_total = 0.
  474. land_total = land_total + field(i,j,landmask_value(k))
  475. else
  476. land_total = -1.
  477. exit
  478. end if
  479. end if
  480. end do
  481. if (land_total >= 0.0) then
  482. if (land_total > 0.50) then
  483. landmask(i,j) = 1.
  484. else
  485. landmask(i,j) = 0.
  486. end if
  487. else
  488. landmask(i,j) = -1.
  489. end if
  490. end do
  491. end do
  492. end if
  493. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, 'LANDMASK', &
  494. datestr, landmask)
  495. ! If we should only save the dominant category, then no need to write out fractional field
  496. if (.not.only_save_dominant .or. (idomcatstatus /= 0)) then
  497. ! Finally, we may be asked to smooth the fractional field
  498. call get_smooth_option(landmask_name, smth_opt, smth_passes, istatus)
  499. if (istatus == 0) then
  500. if (grid_type == 'C') then
  501. if (smth_opt == ONETWOONE) then
  502. call one_two_one(field, &
  503. start_patch_i, end_patch_i, &
  504. start_patch_j, end_patch_j, &
  505. start_mem_i, end_mem_i, &
  506. start_mem_j, end_mem_j, &
  507. min_category, max_category, &
  508. smth_passes, msg_fill_val)
  509. else if (smth_opt == SMTHDESMTH) then
  510. call smth_desmth(field, &
  511. start_patch_i, end_patch_i, &
  512. start_patch_j, end_patch_j, &
  513. start_mem_i, end_mem_i, &
  514. start_mem_j, end_mem_j, &
  515. min_category, max_category, &
  516. smth_passes, msg_fill_val)
  517. else if (smth_opt == SMTHDESMTH_SPECIAL) then
  518. call smth_desmth_special(field, &
  519. start_patch_i, end_patch_i, &
  520. start_patch_j, end_patch_j, &
  521. start_mem_i, end_mem_i, &
  522. start_mem_j, end_mem_j, &
  523. min_category, max_category, &
  524. smth_passes, msg_fill_val)
  525. end if
  526. else if (grid_type == 'E') then
  527. if (smth_opt == ONETWOONE) then
  528. call one_two_one_egrid(field, &
  529. start_patch_i, end_patch_i, &
  530. start_patch_j, end_patch_j, &
  531. start_mem_i, end_mem_i, &
  532. start_mem_j, end_mem_j, &
  533. min_category, max_category, &
  534. smth_passes, msg_fill_val, 1.0)
  535. else if (smth_opt == SMTHDESMTH) then
  536. call smth_desmth_egrid(field, &
  537. start_patch_i, end_patch_i, &
  538. start_patch_j, end_patch_j, &
  539. start_mem_i, end_mem_i, &
  540. start_mem_j, end_mem_j, &
  541. min_category, max_category, &
  542. smth_passes, msg_fill_val, 1.0)
  543. else if (smth_opt == SMTHDESMTH_SPECIAL) then
  544. call mprintf(.true.,WARN,'smth-desmth_special is not currently implemented for NMM. '// &
  545. 'No smoothing will be done.')
  546. end if
  547. end if
  548. end if
  549. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, &
  550. min_category, max_category, trim(landmask_name), &
  551. datestr, real_array=field)
  552. end if
  553. if (idomcatstatus == 0) then
  554. allocate(dominant_field(start_mem_i:end_mem_i, start_mem_j:end_mem_j))
  555. if (.not. only_save_dominant) then
  556. field_count = field_count + 1
  557. call mprintf(.true.,LOGFILE,'Processing field %i of %i (%s)', &
  558. i1=field_count,i2=NUM_FIELDS-NUM_AUTOMATIC_FIELDS,s1=domname)
  559. end if
  560. do i=start_mem_i, end_mem_i
  561. do j=start_mem_j, end_mem_j
  562. if ((landmask(i,j) == 1. .and. is_water_mask) .or. &
  563. (landmask(i,j) == 0. .and. .not.is_water_mask)) then
  564. dominant = 0.
  565. dominant_field(i,j) = real(min_category-1)
  566. do k=min_category,max_category
  567. do kk=1,num_landmask_categories
  568. if (k == landmask_value(kk)) exit
  569. end do
  570. if (field(i,j,k) > dominant .and. kk > num_landmask_categories) then
  571. dominant_field(i,j) = real(k)
  572. dominant = field(i,j,k)
  573. end if
  574. end do
  575. else
  576. dominant = 0.
  577. dominant_field(i,j) = real(min_category-1)
  578. do k=min_category,max_category
  579. do kk=1,num_landmask_categories
  580. if (field(i,j,k) > dominant .and. k == landmask_value(kk)) then
  581. dominant_field(i,j) = real(k)
  582. dominant = field(i,j,k)
  583. end if
  584. end do
  585. end do
  586. end if
  587. end do
  588. end do
  589. call write_field(start_mem_i, end_mem_i, start_mem_j, end_mem_j, 1, 1, trim(domname), &
  590. datestr, dominant_field)
  591. deallocate(dominant_field)
  592. end if
  593. deallocate(field)
  594. end if
  595. !
  596. ! Now process all other fields specified by the user
  597. !
  598. call reset_next_field()
  599. ifieldstatus = 0
  600. do while (ifieldstatus == 0)
  601. call get_next_fieldname(fieldname, ifieldstatus)
  602. ! There is another field in the GEOGRID.TBL file
  603. if (ifieldstatus == 0) then
  604. temp_string(1:128) = fieldname
  605. ! If this field is still to be processed
  606. if (.not. hash_search(processed_fieldnames, temp_string)) then
  607. call hash_insert(processed_fieldnames, temp_string)
  608. call mprintf(.true.,STDOUT,' Processing %s', s1=trim(fieldname))
  609. call get_output_stagger(fieldname, istagger, istatus)
  610. dimnames(:) = 'null'
  611. subgrid_var=is_subgrid_var(fieldname)
  612. if( subgrid_var ) call get_subgrid_dim_name(dimnames)
  613. if (istagger == M .and. .not. subgrid_var ) then
  614. sm1 = start_mem_i
  615. em1 = end_mem_i
  616. sm2 = start_mem_j
  617. em2 = end_mem_j
  618. xlat_ptr => xlat_array
  619. xlon_ptr => xlon_array
  620. mapfac_ptr_x => mapfac_array_m_x
  621. mapfac_ptr_y => mapfac_array_m_y
  622. else if ( subgrid_var ) then
  623. sm1 = start_mem_sub_i
  624. em1 = end_mem_sub_i
  625. sm2 = start_mem_sub_j
  626. em2 = end_mem_sub_j
  627. xlat_ptr => xlat_array_subgrid
  628. xlon_ptr => xlon_array_subgrid
  629. mapfac_ptr_x => mapfac_array_x_subgrid
  630. mapfac_ptr_y => mapfac_array_y_subgrid
  631. else if (istagger == U) then ! In the case that extra_cols = .false.
  632. sm1 = start_mem_i ! we should have that end_mem_stag is
  633. em1 = end_mem_stag_i ! the same as end_mem, so we do not need
  634. sm2 = start_mem_j ! to check extra_cols or extra rows here
  635. em2 = end_mem_j
  636. xlat_ptr => xlat_array_u
  637. xlon_ptr => xlon_array_u
  638. mapfac_ptr_x => mapfac_array_u_x
  639. mapfac_ptr_y => mapfac_array_u_y
  640. else if (istagger == V) then
  641. sm1 = start_mem_i
  642. em1 = end_mem_i
  643. sm2 = start_mem_j
  644. em2 = end_mem_stag_j
  645. xlat_ptr => xlat_array_v
  646. xlon_ptr => xlon_array_v
  647. mapfac_ptr_x => mapfac_array_v_x
  648. mapfac_ptr_y => mapfac_array_v_y
  649. else if (istagger == HH) then ! E grid
  650. sm1 = start_mem_i
  651. em1 = end_mem_i
  652. sm2 = start_mem_j
  653. em2 = end_mem_j
  654. xlat_ptr => xlat_array
  655. xlon_ptr => xlon_array
  656. mapfac_ptr_x => mapfac_array_m_x
  657. mapfac_ptr_y => mapfac_array_m_y
  658. else if (istagger == VV) then ! E grid
  659. sm1 = start_mem_i
  660. em1 = end_mem_i
  661. sm2 = start_mem_j
  662. em2 = end_mem_stag_j
  663. xlat_ptr => xlat_array_v
  664. xlon_ptr => xlon_array_v
  665. mapfac_ptr_x => mapfac_array_v_x
  666. mapfac_ptr_y => mapfac_array_v_y
  667. end if
  668. call get_missing_fill_value(fieldname, msg_fill_val, istatus)
  669. if (istatus /= 0) msg_fill_val = NAN
  670. call get_halt_on_missing(fieldname, halt_on_missing, istatus)
  671. if (istatus /= 0) halt_on_missing = .false.
  672. ! Destination field type is CONTINUOUS
  673. if (iget_fieldtype(fieldname,istatus) == CONTINUOUS) then
  674. call get_max_levels(fieldname, min_level, max_level, istatus)
  675. allocate(field(sm1:em1, sm2:em2, min_level:max_level))
  676. field_count = field_count + 1
  677. call mprintf(.true.,LOGFILE,'Processing field %i of %i (%s)', &
  678. i1=field_count,i2=NUM_FIELDS-NUM_AUTOMATIC_FIELDS,s1=fieldname)
  679. if ( subgrid_var ) then
  680. call calc_field(fieldname, field, xlat_ptr, xlon_ptr, istagger, &
  681. sm1, em1, sm2, em2, min_level, max_level, &
  682. processed_domain, 1, sr_x=sr_x, sr_y=sr_y)
  683. else
  684. call calc_field(fieldname, field, xlat_ptr, xlon_ptr, istagger, &
  685. sm1, em1, sm2, em2, min_level, max_level, &
  686. processed_domain, 1, landmask=landmask)
  687. end if
  688. ! If user wants to halt when a missing value is found in output field, check now
  689. if (halt_on_missing) then
  690. do i=sm1, em1
  691. do j=sm2, em2
  692. ! Only need to examine k=1
  693. if (field(i,j,1) == msg_fill_val) then
  694. call mprintf(.true.,ERROR,' Missing value encountered in output field. Quitting.')
  695. end if
  696. end do
  697. end do
  698. end if
  699. ! We may be asked to smooth the fractional field
  700. call get_smooth_option(fieldname, smth_opt, smth_passes, istatus)
  701. if (istatus == 0) then
  702. if (grid_type == 'C') then
  703. if (smth_opt == ONETWOONE) then
  704. call one_two_one(field, &
  705. start_patch_i, end_patch_i, &
  706. start_patch_j, end_patch_j, &
  707. sm1, em1, &
  708. sm2, em2, &
  709. min_level, max_level, &
  710. smth_passes, msg_fill_val)
  711. else if (smth_opt == SMTHDESMTH) then
  712. call smth_desmth(field, &
  713. start_patch_i, end_patch_i, &
  714. start_patch_j, end_patch_j, &
  715. sm1, em1, &
  716. sm2, em2, &
  717. min_level, max_level, &
  718. smth_passes, msg_fill_val)
  719. else if (smth_opt == SMTHDESMTH_SPECIAL) then
  720. call smth_desmth_special(field, &
  721. start_patch_i, end_patch_i, &
  722. start_patch_j, end_patch_j, &
  723. sm1, em1, &
  724. sm2, em2, &
  725. min_level, max_level, &
  726. smth_passes, msg_fill_val)
  727. end if
  728. else if (grid_type == 'E') then
  729. if (trim(fieldname) == 'HGT_M' ) then
  730. topo_flag_val=1.0
  731. mass_flag=1.0
  732. else if (trim(fieldname) == 'HGT_V') then
  733. topo_flag_val=1.0
  734. mass_flag=0.0
  735. else
  736. topo_flag_val=0.0
  737. end if
  738. if (smth_opt == ONETWOONE) then
  739. call one_two_one_egrid(field, &
  740. start_patch_i, end_patch_i, &
  741. start_patch_j, end_patch_j, &
  742. sm1, em1, &
  743. sm2, em2, &
  744. min_level, max_level, &
  745. smth_passes, topo_flag_val, mass_flag)
  746. else if (smth_opt == SMTHDESMTH) then
  747. call smth_desmth_egrid(field, &
  748. start_patch_i, end_patch_i, &
  749. start_patch_j, end_patch_j, &
  750. sm1, em1, &
  751. sm2, em2, &
  752. min_level, max_level, &
  753. smth_passes, topo_flag_val, mass_flag)
  754. else if (smth_opt == SMTHDESMTH_SPECIAL) then
  755. call mprintf(.true.,WARN,'smth-desmth_special is not currently implemented for NMM. '// &
  756. 'No smoothing will be done.')
  757. end if
  758. end if
  759. end if
  760. call write_field(sm1, em1, sm2, em2, &
  761. min_level, max_level, trim(fieldname), datestr, real_array=field)
  762. ! Do we calculate directional derivatives from this field?
  763. call get_dfdx_name(fieldname, gradname, istatus)
  764. if (istatus == 0) then
  765. allocate(slp_field(sm1:em1,sm2:em2,min_level:max_level))
  766. field_count = field_count + 1
  767. call mprintf(.true.,LOGFILE,'Processing field %i of %i (%s)', &
  768. i1=field_count,i2=NUM_FIELDS-NUM_AUTOMATIC_FIELDS,s1=gradname)
  769. if (grid_type == 'C') then
  770. call calc_dfdx(field, slp_field, sm1, sm2, min_level, em1, em2, max_level, &
  771. which_domain, mapfac_ptr_x, subgrid_var)
  772. else if (grid_type == 'E') then
  773. call calc_dfdx(field, slp_field, sm1, sm2, min_level, em1, em2, max_level, &
  774. which_domain, subgrid_var=subgrid_var)
  775. end if
  776. call write_field(sm1, em1, sm2, em2, &
  777. min_level, max_level, trim(gradname), datestr, real_array=slp_field)
  778. deallocate(slp_field)
  779. end if
  780. call get_dfdy_name(fieldname, gradname, istatus)
  781. if (istatus == 0) then
  782. allocate(slp_field(sm1:em1,sm2:em2,min_level:max_level))
  783. field_count = field_count + 1
  784. call mprintf(.true.,LOGFILE,'Processing field %i of %i (%s)', &
  785. i1=field_count,i2=NUM_FIELDS-NUM_AUTOMATIC_FIELDS,s1=gradname)
  786. if (grid_type == 'C') then
  787. call calc_dfdy(field, slp_field, sm1, sm2, min_level, em1, em2, max_level, &
  788. which_domain, mapfac_ptr_y, subgrid_var)
  789. else if (grid_type == 'E') then
  790. call calc_dfdy(field, slp_field, sm1, sm2, min_level, em1, em2, max_level, &
  791. which_domain, subgrid_var=subgrid_var)
  792. end if
  793. call write_field(sm1, em1, sm2, em2, &
  794. min_level, max_level, trim(gradname), datestr, real_array=slp_field)
  795. deallocate(slp_field)
  796. end if
  797. deallocate(field)
  798. ! Destination field type is CATEGORICAL
  799. else
  800. call get_max_categories(fieldname, min_category, max_category, istatus)
  801. allocate(field(sm1:em1, sm2:em2, min_category:max_category))
  802. ! Do we calculate a dominant category for this field?
  803. call get_domcategory_name(fieldname, domname, only_save_dominant, idomcatstatus)
  804. if (.not. only_save_dominant) then
  805. field_count = field_count + 1
  806. call mprintf(.true.,LOGFILE,'Processing field %i of %i (%s)', &
  807. i1=field_count,i2=NUM_FIELDS-NUM_AUTOMATIC_FIELDS,s1=fieldname)
  808. else
  809. field_count = field_count + 1
  810. call mprintf(.true.,LOGFILE,'Processing field %i of %i (%s)', &
  811. i1=field_count,i2=NUM_FIELDS-NUM_AUTOMATIC_FIELDS,s1=domname)
  812. end if
  813. if ( subgrid_var ) then
  814. call calc_field(fieldname, field, xlat_ptr, xlon_ptr, istagger, &
  815. sm1, em1, sm2, em2, min_category, max_category, &
  816. processed_domain, 1, sr_x=sr_x, sr_y=sr_y)
  817. else
  818. call calc_field(fieldname, field, xlat_ptr, xlon_ptr, istagger, &
  819. sm1, em1, sm2, em2, min_category, max_category, &
  820. processed_domain, 1, landmask=landmask)
  821. end if
  822. ! Find fractions
  823. do i=sm1, em1
  824. do j=sm2, em2
  825. sum = 0.
  826. do k=min_category,max_category
  827. sum = sum + field(i,j,k)
  828. end do
  829. if (sum > 0.0) then
  830. do k=min_category,max_category
  831. field(i,j,k) = field(i,j,k) / sum
  832. end do
  833. else
  834. do k=min_category,max_category
  835. field(i,j,k) = msg_fill_val
  836. end do
  837. end if
  838. end do
  839. end do
  840. ! If user wants to halt when a missing value is found in output field, check now
  841. if (halt_on_missing) then
  842. do i=sm1, em1
  843. do j=sm2, em2
  844. ! Only need to examine k=1
  845. if (field(i,j,1) == msg_fill_val) then
  846. call mprintf(.true.,ERROR,' Missing value encountered in output field. Quitting.')
  847. end if
  848. end do
  849. end do
  850. end if
  851. ! If we should only save the dominant category, then no need to write out fractional field
  852. if (.not.only_save_dominant .or. (idomcatstatus /= 0)) then
  853. ! Finally, we may be asked to smooth the fractional field
  854. call get_smooth_option(fieldname, smth_opt, smth_passes, istatus)
  855. if (istatus == 0) then
  856. if (grid_type == 'C') then
  857. if (smth_opt == ONETWOONE) then
  858. call one_two_one(field, &
  859. start_patch_i, end_patch_i, &
  860. start_patch_j, end_patch_j, &
  861. sm1, em1, &
  862. sm2, em2, &
  863. min_category, max_category, &
  864. smth_passes, msg_fill_val)
  865. else if (smth_opt

Large files files are truncated, but you can click here to view the full file