MODFLOW 6  version 6.5.0.dev2
MODFLOW 6 Code Documentation
Disu.f90
Go to the documentation of this file.
1 module disumodule
2 
4  use kindmodule, only: dp, i4b, lgp
13  use basedismodule, only: disbasetype
17  use tdismodule, only: kstp, kper, pertim, totim, delt
18  use disvgeom, only: line_unit_vector
19 
20  implicit none
21 
22  private
23  public :: disutype
24  public :: disu_cr
25  public :: castasdisutype
26 
27  !> @brief Unstructured grid discretization
28  type, extends(disbasetype) :: disutype
29  integer(I4B), pointer :: njausr => null() ! user-specified nja size
30  integer(I4B), pointer :: nvert => null() ! number of x,y vertices
31  real(dp), pointer :: voffsettol => null() ! vertical offset tolerance
32  real(dp), dimension(:, :), pointer, contiguous :: vertices => null() ! cell vertices stored as 2d array of x and y
33  real(dp), dimension(:, :), pointer, contiguous :: cellxy => null() ! cell center stored as 2d array of x and y
34  real(dp), dimension(:), pointer, contiguous :: top1d => null() ! (size:nodesuser) cell top elevation
35  real(dp), dimension(:), pointer, contiguous :: bot1d => null() ! (size:nodesuser) cell bottom elevation
36  real(dp), dimension(:), pointer, contiguous :: area1d => null() ! (size:nodesuser) cell area, in plan view
37  integer(I4B), dimension(:), pointer, contiguous :: iainp => null() ! (size:nodesuser+1) user iac converted ia
38  integer(I4B), dimension(:), pointer, contiguous :: jainp => null() ! (size:njausr) user-input ja array
39  integer(I4B), dimension(:), pointer, contiguous :: ihcinp => null() ! (size:njausr) user-input ihc array
40  real(dp), dimension(:), pointer, contiguous :: cl12inp => null() ! (size:njausr) user-input cl12 array
41  real(dp), dimension(:), pointer, contiguous :: hwvainp => null() ! (size:njausr) user-input hwva array
42  real(dp), dimension(:), pointer, contiguous :: angldegxinp => null() ! (size:njausr) user-input angldegx array
43  integer(I4B), pointer :: iangledegx => null() ! =1 when angle information was present in input, 0 otherwise
44  integer(I4B), dimension(:), pointer, contiguous :: iavert => null() ! cell vertex pointer ia array
45  integer(I4B), dimension(:), pointer, contiguous :: javert => null() ! cell vertex pointer ja array
46  integer(I4B), dimension(:), pointer, contiguous :: idomain => null() ! idomain (nodes)
47  logical(LGP) :: readfromfile ! True, when DIS is read from file (almost always)
48 
49  contains
50 
51  procedure :: dis_df => disu_df
52  procedure :: disu_load
53  procedure :: dis_da => disu_da
54  procedure :: get_dis_type => get_dis_type
55  procedure :: get_dis_enum => get_dis_enum
56  procedure :: disu_ck
57  procedure :: grid_finalize
58  procedure :: get_nodenumber_idx1
59  procedure :: nodeu_to_string
60  procedure :: nodeu_to_array
61  procedure :: nodeu_from_string
62  procedure :: nodeu_from_cellid
63  procedure :: connection_normal
64  procedure :: connection_vector
65  procedure :: supports_layers
66  procedure :: get_ncpl
67  procedure, public :: record_array
68  procedure, public :: record_srcdst_list_header
69  ! -- private
70  procedure :: allocate_scalars
71  procedure :: allocate_arrays
72  procedure :: allocate_arrays_mem
73  procedure :: source_options
74  procedure :: source_dimensions
75  procedure :: source_griddata
76  procedure :: source_connectivity
77  procedure :: source_vertices
78  procedure :: source_cell2d
79  procedure :: log_options
80  procedure :: log_dimensions
81  procedure :: log_griddata
82  procedure :: log_connectivity
83  procedure :: define_cellverts
84  procedure :: write_grb
85  !
86  ! -- Read a node-sized model array (reduced or not)
87  procedure :: read_int_array
88  procedure :: read_dbl_array
89 
90  end type disutype
91 
93  logical :: length_units = .false.
94  logical :: nogrb = .false.
95  logical :: xorigin = .false.
96  logical :: yorigin = .false.
97  logical :: angrot = .false.
98  logical :: voffsettol = .false.
99  logical :: nodes = .false.
100  logical :: nja = .false.
101  logical :: nvert = .false.
102  logical :: top = .false.
103  logical :: bot = .false.
104  logical :: area = .false.
105  logical :: idomain = .false.
106  logical :: iac = .false.
107  logical :: ja = .false.
108  logical :: ihc = .false.
109  logical :: cl12 = .false.
110  logical :: hwva = .false.
111  logical :: angldegx = .false.
112  logical :: iv = .false.
113  logical :: xv = .false.
114  logical :: yv = .false.
115  logical :: icell2d = .false.
116  logical :: xc = .false.
117  logical :: yc = .false.
118  logical :: ncvert = .false.
119  logical :: icvert = .false.
120  end type disufoundtype
121 
122 contains
123 
124  !> @brief Create a new unstructured discretization object
125  !<
126  subroutine disu_cr(dis, name_model, input_mempath, inunit, iout)
127  ! -- dummy
128  class(disbasetype), pointer :: dis
129  character(len=*), intent(in) :: name_model
130  character(len=*), intent(in) :: input_mempath
131  integer(I4B), intent(in) :: inunit
132  integer(I4B), intent(in) :: iout
133  ! -- local
134  type(disutype), pointer :: disnew
135  character(len=*), parameter :: fmtheader = &
136  "(1X, /1X, 'DISU -- UNSTRUCTURED GRID DISCRETIZATION PACKAGE,', &
137  &' VERSION 2 : 3/27/2014 - INPUT READ FROM MEMPATH: ', A, //)"
138  !
139  ! -- Create a new discretization object
140  allocate (disnew)
141  dis => disnew
142  !
143  ! -- Allocate scalars and assign data
144  call dis%allocate_scalars(name_model, input_mempath)
145  dis%inunit = inunit
146  dis%iout = iout
147  !
148  ! -- If disu is enabled
149  if (inunit > 0) then
150  !
151  ! -- Identify package
152  if (iout > 0) then
153  write (iout, fmtheader) dis%input_mempath
154  end if
155  !
156  ! -- load disu
157  call disnew%disu_load()
158  end if
159  !
160  end subroutine disu_cr
161 
162  !> @brief Transfer IDM data into this discretization object
163  !<
164  subroutine disu_load(this)
165  ! -- dummy
166  class(disutype) :: this
167  !
168  ! -- source input data
169  call this%source_options()
170  call this%source_dimensions()
171  call this%source_griddata()
172  call this%source_connectivity()
173  !
174  ! -- If NVERT specified and greater than 0, then source VERTICES and CELL2D
175  if (this%nvert > 0) then
176  call this%source_vertices()
177  call this%source_cell2d()
178  else
179  ! -- connection direction information cannot be calculated
180  this%icondir = 0
181  end if
182  !
183  ! -- Make some final disu checks on the non-reduced user-provided
184  ! input
185  call this%disu_ck()
186  !
187  end subroutine disu_load
188 
189  !> @brief Define the discretization
190  !<
191  subroutine disu_df(this)
192  ! -- dummy
193  class(disutype) :: this
194  !
195  call this%grid_finalize()
196  !
197  end subroutine disu_df
198 
199  !> @brief Finalize the grid
200  !<
201  subroutine grid_finalize(this)
202  ! -- dummy
203  class(disutype) :: this
204  ! -- locals
205  integer(I4B) :: n
206  integer(I4B) :: node
207  integer(I4B) :: noder
208  integer(I4B) :: nrsize
209  ! -- formats
210  character(len=*), parameter :: fmtdz = &
211  "('CELL (',i0,',',i0,',',i0,') THICKNESS <= 0. ', &
212  &'TOP, BOT: ',2(1pg24.15))"
213  character(len=*), parameter :: fmtnr = &
214  "(/1x, 'The specified IDOMAIN results in a reduced number of cells.',&
215  &/1x, 'Number of user nodes: ',I0,&
216  &/1X, 'Number of nodes in solution: ', I0, //)"
217  !
218  ! -- count active cells
219  this%nodes = 0
220  do n = 1, this%nodesuser
221  if (this%idomain(n) > 0) this%nodes = this%nodes + 1
222  end do
223  !
224  ! -- Check to make sure nodes is a valid number
225  if (this%nodes == 0) then
226  call store_error('Model does not have any active nodes. &
227  &Ensure IDOMAIN array has some values greater &
228  &than zero.')
229  call store_error_filename(this%input_fname)
230  end if
231  !
232  ! -- Write message if reduced grid
233  if (this%nodes < this%nodesuser) then
234  write (this%iout, fmtnr) this%nodesuser, this%nodes
235  end if
236  !
237  ! -- Array size is now known, so allocate
238  call this%allocate_arrays()
239  !
240  ! -- Fill the nodereduced array with the reduced nodenumber, or
241  ! a negative number to indicate it is a pass-through cell, or
242  ! a zero to indicate that the cell is excluded from the
243  ! solution. (negative idomain not supported for disu)
244  if (this%nodes < this%nodesuser) then
245  noder = 1
246  do node = 1, this%nodesuser
247  if (this%idomain(node) > 0) then
248  this%nodereduced(node) = noder
249  noder = noder + 1
250  elseif (this%idomain(node) < 0) then
251  this%nodereduced(node) = -1
252  else
253  this%nodereduced(node) = 0
254  end if
255  end do
256  end if
257  !
258  ! -- Fill nodeuser if a reduced grid
259  if (this%nodes < this%nodesuser) then
260  noder = 1
261  do node = 1, this%nodesuser
262  if (this%idomain(node) > 0) then
263  this%nodeuser(noder) = node
264  noder = noder + 1
265  end if
266  end do
267  end if
268  !
269  ! -- Move top1d, bot1d, and area1d into top, bot, and area
270  do node = 1, this%nodesuser
271  noder = node
272  if (this%nodes < this%nodesuser) noder = this%nodereduced(node)
273  if (noder <= 0) cycle
274  this%top(noder) = this%top1d(node)
275  this%bot(noder) = this%bot1d(node)
276  this%area(noder) = this%area1d(node)
277  end do
278  !
279  ! -- fill cell center coordinates
280  if (this%nvert > 0) then
281  do node = 1, this%nodesuser
282  noder = node
283  if (this%nodes < this%nodesuser) noder = this%nodereduced(node)
284  if (noder <= 0) cycle
285  this%xc(noder) = this%cellxy(1, node)
286  this%yc(noder) = this%cellxy(2, node)
287  end do
288  else
289  call mem_reallocate(this%xc, 0, 'XC', this%memoryPath)
290  call mem_reallocate(this%yc, 0, 'YC', this%memoryPath)
291  end if
292  !
293  ! -- create and fill the connections object
294  nrsize = 0
295  if (this%nodes < this%nodesuser) nrsize = this%nodes
296  allocate (this%con)
297  call this%con%disuconnections(this%name_model, this%nodes, &
298  this%nodesuser, nrsize, &
299  this%nodereduced, this%nodeuser, &
300  this%iainp, this%jainp, &
301  this%ihcinp, this%cl12inp, &
302  this%hwvainp, this%angldegxinp, &
303  this%iangledegx)
304  this%nja = this%con%nja
305  this%njas = this%con%njas
306  !
307  end subroutine grid_finalize
308 
309  !> @brief Check discretization info
310  !<
311  subroutine disu_ck(this)
312  ! -- dummy
313  class(disutype) :: this
314  ! -- local
315  integer(I4B) :: n, m
316  integer(I4B) :: ipos
317  integer(I4B) :: ihc
318  real(DP) :: dz
319  ! -- formats
320  character(len=*), parameter :: fmtidm = &
321  &"('Invalid idomain value ', i0, ' specified for node ', i0)"
322  character(len=*), parameter :: fmtdz = &
323  &"('Cell ', i0, ' with thickness <= 0. Top, bot: ', 2(1pg24.15))"
324  character(len=*), parameter :: fmtarea = &
325  &"('Cell ', i0, ' with area <= 0. Area: ', 1(1pg24.15))"
326  character(len=*), parameter :: fmtjan = &
327  &"('Cell ', i0, ' must have its first connection be itself. Found: ', i0)"
328  character(len=*), parameter :: fmtjam = &
329  &"('Cell ', i0, ' has invalid connection in JA. Found: ', i0)"
330  character(len=*), parameter :: fmterrmsg = &
331  "('Top elevation (', 1pg15.6, ') for cell ', i0, ' is above bottom &
332  &elevation (', 1pg15.6, ') for cell ', i0, '. Based on node numbering &
333  &rules cell ', i0, ' must be below cell ', i0, '.')"
334  !
335  ! -- Check connectivity
336  do n = 1, this%nodesuser
337  !
338  ! -- Ensure first connection is to itself, and
339  ! that ja(ia(n)) is positive
340  ipos = this%iainp(n)
341  m = this%jainp(ipos)
342  if (m < 0) then
343  m = abs(m)
344  this%jainp(ipos) = m
345  end if
346  if (n /= m) then
347  write (errmsg, fmtjan) n, m
348  call store_error(errmsg)
349  end if
350  !
351  ! -- Check for valid node numbers in connected cells
352  do ipos = this%iainp(n) + 1, this%iainp(n + 1) - 1
353  m = this%jainp(ipos)
354  if (m < 0 .or. m > this%nodesuser) then
355  ! -- make sure first connection is to itself
356  write (errmsg, fmtjam) n, m
357  call store_error(errmsg)
358  end if
359  end do
360  end do
361  !
362  ! -- terminate if errors found
363  if (count_errors() > 0) then
364  if (this%inunit > 0) then
365  call store_error_filename(this%input_fname)
366  end if
367  end if
368  !
369  ! -- Ensure idomain values are valid
370  do n = 1, this%nodesuser
371  if (this%idomain(n) > 1 .or. this%idomain(n) < 0) then
372  write (errmsg, fmtidm) this%idomain(n), n
373  call store_error(errmsg)
374  end if
375  end do
376  !
377  ! -- Check for zero and negative thickness and zero or negative areas
378  ! for cells with idomain == 1
379  do n = 1, this%nodesuser
380  if (this%idomain(n) == 1) then
381  dz = this%top1d(n) - this%bot1d(n)
382  if (dz <= dzero) then
383  write (errmsg, fmt=fmtdz) n, this%top1d(n), this%bot1d(n)
384  call store_error(errmsg)
385  end if
386  if (this%area1d(n) <= dzero) then
387  write (errmsg, fmt=fmtarea) n, this%area1d(n)
388  call store_error(errmsg)
389  end if
390  end if
391  end do
392  !
393  ! -- check to make sure voffsettol is >= 0
394  if (this%voffsettol < dzero) then
395  write (errmsg, '(a, 1pg15.6)') &
396  'Vertical offset tolerance must be greater than zero. Found ', &
397  this%voffsettol
398  call store_error(errmsg)
399  if (this%inunit > 0) then
400  call store_error_filename(this%input_fname)
401  end if
402  end if
403  !
404  ! -- For cell n, ensure that underlying cells have tops less than
405  ! or equal to the bottom of cell n
406  do n = 1, this%nodesuser
407  do ipos = this%iainp(n) + 1, this%iainp(n + 1) - 1
408  m = this%jainp(ipos)
409  ihc = this%ihcinp(ipos)
410  if (ihc == 0 .and. m > n) then
411  dz = this%top1d(m) - this%bot1d(n)
412  if (dz > this%voffsettol) then
413  write (errmsg, fmterrmsg) this%top1d(m), m, this%bot1d(n), n, m, n
414  call store_error(errmsg)
415  end if
416  end if
417  end do
418  end do
419  !
420  ! -- terminate if errors found
421  if (count_errors() > 0) then
422  if (this%inunit > 0) then
423  call store_error_filename(this%input_fname)
424  end if
425  end if
426  !
427  end subroutine disu_ck
428 
429  !> @brief Deallocate variables
430  !<
431  subroutine disu_da(this)
432  ! -- dummy
433  class(disutype) :: this
434  !
435  ! -- Deallocate idm memory
436  call memorylist_remove(this%name_model, 'DISU', idm_context)
437  call memorylist_remove(component=this%name_model, &
438  context=idm_context)
439  !
440  ! -- scalars
441  call mem_deallocate(this%njausr)
442  call mem_deallocate(this%nvert)
443  call mem_deallocate(this%voffsettol)
444  call mem_deallocate(this%iangledegx)
445  !
446  ! -- arrays
447  if (this%readFromFile) then
448  call mem_deallocate(this%top1d)
449  call mem_deallocate(this%bot1d)
450  call mem_deallocate(this%area1d)
451  if (associated(this%iavert)) then
452  call mem_deallocate(this%iavert)
453  call mem_deallocate(this%javert)
454  end if
455  call mem_deallocate(this%vertices)
456  call mem_deallocate(this%iainp)
457  call mem_deallocate(this%jainp)
458  call mem_deallocate(this%ihcinp)
459  call mem_deallocate(this%cl12inp)
460  call mem_deallocate(this%hwvainp)
461  call mem_deallocate(this%angldegxinp)
462  end if
463  !
464  call mem_deallocate(this%idomain)
465  call mem_deallocate(this%cellxy)
466  !
467  call mem_deallocate(this%nodeuser)
468  call mem_deallocate(this%nodereduced)
469  !
470  ! -- DisBaseType deallocate
471  call this%DisBaseType%dis_da()
472  !
473  end subroutine disu_da
474 
475  !> @brief Convert a user nodenumber to a string (nodenumber)
476  !<
477  subroutine nodeu_to_string(this, nodeu, str)
478  ! -- dummy
479  class(disutype) :: this
480  integer(I4B), intent(in) :: nodeu
481  character(len=*), intent(inout) :: str
482  ! -- local
483  character(len=10) :: nstr
484  !
485  write (nstr, '(i0)') nodeu
486  str = '('//trim(adjustl(nstr))//')'
487  !
488  end subroutine nodeu_to_string
489 
490  !> @brief Convert a user nodenumber to an array (nodenumber)
491  !<
492  subroutine nodeu_to_array(this, nodeu, arr)
493  class(disutype) :: this
494  integer(I4B), intent(in) :: nodeu
495  integer(I4B), dimension(:), intent(inout) :: arr
496  ! -- local
497  integer(I4B) :: isize
498  !
499  ! -- check the size of arr
500  isize = size(arr)
501  if (isize /= this%ndim) then
502  write (errmsg, '(a,i0,a,i0,a)') &
503  'Program error: nodeu_to_array size of array (', isize, &
504  ') is not equal to the discretization dimension (', this%ndim, ')'
505  call store_error(errmsg, terminate=.true.)
506  end if
507  !
508  ! -- fill array
509  arr(1) = nodeu
510  !
511  end subroutine nodeu_to_array
512 
513  !> @brief Copy options from IDM into package
514  !<
515  subroutine source_options(this)
516  ! -- dummy
517  class(disutype) :: this
518  ! -- locals
519  character(len=LENVARNAME), dimension(3) :: lenunits = &
520  &[character(len=LENVARNAME) :: 'FEET', 'METERS', 'CENTIMETERS']
521  type(disufoundtype) :: found
522  !
523  ! -- update defaults with idm sourced values
524  call mem_set_value(this%lenuni, 'LENGTH_UNITS', this%input_mempath, &
525  lenunits, found%length_units)
526  call mem_set_value(this%nogrb, 'NOGRB', this%input_mempath, found%nogrb)
527  call mem_set_value(this%xorigin, 'XORIGIN', this%input_mempath, found%xorigin)
528  call mem_set_value(this%yorigin, 'YORIGIN', this%input_mempath, found%yorigin)
529  call mem_set_value(this%angrot, 'ANGROT', this%input_mempath, found%angrot)
530  call mem_set_value(this%voffsettol, 'VOFFSETTOL', this%input_mempath, &
531  found%voffsettol)
532  !
533  ! -- log values to list file
534  if (this%iout > 0) then
535  call this%log_options(found)
536  end if
537  !
538  end subroutine source_options
539 
540  !> @brief Write user options to list file
541  !<
542  subroutine log_options(this, found)
543  ! -- dummy
544  class(disutype) :: this
545  type(disufoundtype), intent(in) :: found
546  !
547  write (this%iout, '(1x,a)') 'Setting Discretization Options'
548  !
549  if (found%length_units) then
550  write (this%iout, '(4x,a,i0)') 'Model length unit [0=UND, 1=FEET, &
551  &2=METERS, 3=CENTIMETERS] set as ', this%lenuni
552  end if
553  !
554  if (found%nogrb) then
555  write (this%iout, '(4x,a,i0)') 'Binary grid file [0=GRB, 1=NOGRB] &
556  &set as ', this%nogrb
557  end if
558  !
559  if (found%xorigin) then
560  write (this%iout, '(4x,a,G0)') 'XORIGIN = ', this%xorigin
561  end if
562  !
563  if (found%yorigin) then
564  write (this%iout, '(4x,a,G0)') 'YORIGIN = ', this%yorigin
565  end if
566  !
567  if (found%angrot) then
568  write (this%iout, '(4x,a,G0)') 'ANGROT = ', this%angrot
569  end if
570  !
571  if (found%voffsettol) then
572  write (this%iout, '(4x,a,G0)') 'VERTICAL_OFFSET_TOLERANCE = ', &
573  this%voffsettol
574  end if
575  !
576  write (this%iout, '(1x,a,/)') 'End Setting Discretization Options'
577  !
578  end subroutine log_options
579 
580  !> @brief Copy dimensions from IDM into package
581  !<
582  subroutine source_dimensions(this)
583  ! -- dummy
584  class(disutype) :: this
585  ! -- locals
586  integer(I4B) :: n
587  type(disufoundtype) :: found
588  !
589  ! -- update defaults with idm sourced values
590  call mem_set_value(this%nodesuser, 'NODES', this%input_mempath, found%nodes)
591  call mem_set_value(this%njausr, 'NJA', this%input_mempath, found%nja)
592  call mem_set_value(this%nvert, 'NVERT', this%input_mempath, found%nvert)
593  !
594  ! -- log simulation values
595  if (this%iout > 0) then
596  call this%log_dimensions(found)
597  end if
598  !
599  ! -- verify dimensions were set
600  if (this%nodesuser < 1) then
601  call store_error( &
602  'NODES was not specified or was specified incorrectly.')
603  end if
604  if (this%njausr < 1) then
605  call store_error( &
606  'NJA was not specified or was specified incorrectly.')
607  end if
608  !
609  ! -- terminate if errors were detected
610  if (count_errors() > 0) then
611  call store_error_filename(this%input_fname)
612  end if
613  !
614  ! -- allocate vectors that are the size of nodesuser
615  this%readFromFile = .true.
616  call mem_allocate(this%top1d, this%nodesuser, 'TOP1D', this%memoryPath)
617  call mem_allocate(this%bot1d, this%nodesuser, 'BOT1D', this%memoryPath)
618  call mem_allocate(this%area1d, this%nodesuser, 'AREA1D', this%memoryPath)
619  call mem_allocate(this%idomain, this%nodesuser, 'IDOMAIN', this%memoryPath)
620  call mem_allocate(this%vertices, 2, this%nvert, 'VERTICES', this%memoryPath)
621  call mem_allocate(this%iainp, this%nodesuser + 1, 'IAINP', this%memoryPath)
622  call mem_allocate(this%jainp, this%njausr, 'JAINP', this%memoryPath)
623  call mem_allocate(this%ihcinp, this%njausr, 'IHCINP', this%memoryPath)
624  call mem_allocate(this%cl12inp, this%njausr, 'CL12INP', this%memoryPath)
625  call mem_allocate(this%hwvainp, this%njausr, 'HWVAINP', this%memoryPath)
626  call mem_allocate(this%angldegxinp, this%njausr, 'ANGLDEGXINP', &
627  this%memoryPath)
628  if (this%nvert > 0) then
629  call mem_allocate(this%cellxy, 2, this%nodesuser, 'CELLXY', this%memoryPath)
630  else
631  call mem_allocate(this%cellxy, 2, 0, 'CELLXY', this%memoryPath)
632  end if
633  !
634  ! -- initialize all cells to be active (idomain = 1)
635  do n = 1, this%nodesuser
636  this%idomain(n) = 1
637  end do
638  !
639  end subroutine source_dimensions
640 
641  !> @brief Write dimensions to list file
642  !<
643  subroutine log_dimensions(this, found)
644  class(disutype) :: this
645  type(disufoundtype), intent(in) :: found
646  !
647  write (this%iout, '(1x,a)') 'Setting Discretization Dimensions'
648  !
649  if (found%nodes) then
650  write (this%iout, '(4x,a,i0)') 'NODES = ', this%nodesuser
651  end if
652  !
653  if (found%nja) then
654  write (this%iout, '(4x,a,i0)') 'NJA = ', this%njausr
655  end if
656  !
657  if (found%nvert) then
658  write (this%iout, '(4x,a,i0)') 'NVERT = ', this%nvert
659  end if
660  !
661  write (this%iout, '(1x,a,/)') 'End Setting Discretization Dimensions'
662  !
663  end subroutine log_dimensions
664 
665  !> @brief Copy grid data from IDM into package
666  !<
667  subroutine source_griddata(this)
668  ! -- dummy
669  class(disutype) :: this
670  ! -- locals
671  type(disufoundtype) :: found
672  !
673  ! -- update defaults with idm sourced values
674  call mem_set_value(this%top1d, 'TOP', this%input_mempath, found%top)
675  call mem_set_value(this%bot1d, 'BOT', this%input_mempath, found%bot)
676  call mem_set_value(this%area1d, 'AREA', this%input_mempath, found%area)
677  call mem_set_value(this%idomain, 'IDOMAIN', this%input_mempath, found%idomain)
678  !
679  ! -- log simulation values
680  if (this%iout > 0) then
681  call this%log_griddata(found)
682  end if
683  !
684  end subroutine source_griddata
685 
686  !> @brief Write griddata found to list file
687  !<
688  subroutine log_griddata(this, found)
689  ! -- dummy
690  class(disutype) :: this
691  type(disufoundtype), intent(in) :: found
692  !
693  write (this%iout, '(1x,a)') 'Setting Discretization Griddata'
694  !
695  if (found%top) then
696  write (this%iout, '(4x,a)') 'TOP set from input file'
697  end if
698  !
699  if (found%bot) then
700  write (this%iout, '(4x,a)') 'BOT set from input file'
701  end if
702  !
703  if (found%area) then
704  write (this%iout, '(4x,a)') 'AREA set from input file'
705  end if
706  !
707  if (found%idomain) then
708  write (this%iout, '(4x,a)') 'IDOMAIN set from input file'
709  end if
710  !
711  write (this%iout, '(1x,a,/)') 'End Setting Discretization Griddata'
712  !
713  end subroutine log_griddata
714 
715  !> @brief Copy grid connectivity info from IDM into package
716  !<
717  subroutine source_connectivity(this)
718  ! -- dummy
719  class(disutype) :: this
720  ! -- locals
721  type(disufoundtype) :: found
722  integer(I4B), dimension(:), contiguous, pointer :: iac => null()
723  ! -- formats
724  !
725  ! -- update defaults with idm sourced values
726  call mem_set_value(this%jainp, 'JA', this%input_mempath, found%ja)
727  call mem_set_value(this%ihcinp, 'IHC', this%input_mempath, found%ihc)
728  call mem_set_value(this%cl12inp, 'CL12', this%input_mempath, found%cl12)
729  call mem_set_value(this%hwvainp, 'HWVA', this%input_mempath, found%hwva)
730  call mem_set_value(this%angldegxinp, 'ANGLDEGX', this%input_mempath, &
731  found%angldegx)
732  !
733  ! -- set pointer to iac input array
734  call mem_setptr(iac, 'IAC', this%input_mempath)
735  !
736  ! -- Convert iac to ia
737  if (associated(iac)) call iac_to_ia(iac, this%iainp)
738  !
739  ! -- Set angldegx flag if found
740  if (found%angldegx) this%iangledegx = 1
741  !
742  ! -- log simulation values
743  if (this%iout > 0) then
744  call this%log_connectivity(found, iac)
745  end if
746  !
747  end subroutine source_connectivity
748 
749  !> @brief Write griddata found to list file
750  !<
751  subroutine log_connectivity(this, found, iac)
752  class(disutype) :: this
753  type(disufoundtype), intent(in) :: found
754  integer(I4B), dimension(:), contiguous, pointer, intent(in) :: iac
755  !
756  write (this%iout, '(1x,a)') 'Setting Discretization Connectivity'
757  !
758  if (associated(iac)) then
759  write (this%iout, '(4x,a)') 'IAC set from input file'
760  end if
761  !
762  if (found%ja) then
763  write (this%iout, '(4x,a)') 'JA set from input file'
764  end if
765  !
766  if (found%ihc) then
767  write (this%iout, '(4x,a)') 'IHC set from input file'
768  end if
769  !
770  if (found%cl12) then
771  write (this%iout, '(4x,a)') 'CL12 set from input file'
772  end if
773  !
774  if (found%hwva) then
775  write (this%iout, '(4x,a)') 'HWVA set from input file'
776  end if
777  !
778  if (found%angldegx) then
779  write (this%iout, '(4x,a)') 'ANGLDEGX set from input file'
780  end if
781  !
782  write (this%iout, '(1x,a,/)') 'End Setting Discretization Connectivity'
783  !
784  end subroutine log_connectivity
785 
786  !> @brief Copy grid vertex data from IDM into package
787  !<
788  subroutine source_vertices(this)
789  ! -- dummy
790  class(disutype) :: this
791  ! -- local
792  integer(I4B) :: i
793  real(DP), dimension(:), contiguous, pointer :: vert_x => null()
794  real(DP), dimension(:), contiguous, pointer :: vert_y => null()
795  ! -- formats
796  !
797  ! -- set pointers to memory manager input arrays
798  call mem_setptr(vert_x, 'XV', this%input_mempath)
799  call mem_setptr(vert_y, 'YV', this%input_mempath)
800  !
801  ! -- set vertices 2d array
802  if (associated(vert_x) .and. associated(vert_y)) then
803  do i = 1, this%nvert
804  this%vertices(1, i) = vert_x(i)
805  this%vertices(2, i) = vert_y(i)
806  end do
807  else
808  call store_error('Required Vertex arrays not found.')
809  end if
810  !
811  ! -- log
812  if (this%iout > 0) then
813  write (this%iout, '(1x,a)') 'Discretization Vertex data loaded'
814  end if
815  !
816  end subroutine source_vertices
817 
818  !> @brief Build data structures to hold cell vertex info
819  !<
820  subroutine define_cellverts(this, icell2d, ncvert, icvert)
821  ! -- modules
822  use sparsemodule, only: sparsematrix
823  ! -- dummy
824  class(disutype) :: this
825  integer(I4B), dimension(:), contiguous, pointer, intent(in) :: icell2d
826  integer(I4B), dimension(:), contiguous, pointer, intent(in) :: ncvert
827  integer(I4B), dimension(:), contiguous, pointer, intent(in) :: icvert
828  ! -- locals
829  type(sparsematrix) :: vert_spm
830  integer(I4B) :: i, j, ierr
831  integer(I4B) :: icv_idx, startvert, maxnnz = 5
832  !
833  ! -- initialize sparse matrix
834  call vert_spm%init(this%nodesuser, this%nvert, maxnnz)
835  !
836  ! -- add sparse matrix connections from input memory paths
837  icv_idx = 1
838  do i = 1, this%nodesuser
839  if (icell2d(i) /= i) call store_error('ICELL2D input sequence violation.')
840  do j = 1, ncvert(i)
841  call vert_spm%addconnection(i, icvert(icv_idx), 0)
842  if (j == 1) then
843  startvert = icvert(icv_idx)
844  elseif (j == ncvert(i) .and. (icvert(icv_idx) /= startvert)) then
845  call vert_spm%addconnection(i, startvert, 0)
846  end if
847  icv_idx = icv_idx + 1
848  end do
849  end do
850  !
851  ! -- allocate and fill iavert and javert
852  call mem_allocate(this%iavert, this%nodesuser + 1, 'IAVERT', this%memoryPath)
853  call mem_allocate(this%javert, vert_spm%nnz, 'JAVERT', this%memoryPath)
854  call vert_spm%filliaja(this%iavert, this%javert, ierr)
855  call vert_spm%destroy()
856  !
857  end subroutine define_cellverts
858 
859  !> @brief Copy cell2d data from IDM into package
860  !<
861  subroutine source_cell2d(this)
862  ! -- dummy
863  class(disutype) :: this
864  ! -- locals
865  integer(I4B), dimension(:), contiguous, pointer :: icell2d => null()
866  integer(I4B), dimension(:), contiguous, pointer :: ncvert => null()
867  integer(I4B), dimension(:), contiguous, pointer :: icvert => null()
868  real(DP), dimension(:), contiguous, pointer :: cell_x => null()
869  real(DP), dimension(:), contiguous, pointer :: cell_y => null()
870  integer(I4B) :: i
871  !
872  ! -- set pointers to input path ncvert and icvert
873  call mem_setptr(icell2d, 'ICELL2D', this%input_mempath)
874  call mem_setptr(ncvert, 'NCVERT', this%input_mempath)
875  call mem_setptr(icvert, 'ICVERT', this%input_mempath)
876  !
877  ! --
878  if (associated(icell2d) .and. associated(ncvert) &
879  .and. associated(icvert)) then
880  call this%define_cellverts(icell2d, ncvert, icvert)
881  else
882  call store_error('Required cell vertex arrays not found.')
883  end if
884  !
885  ! -- set pointers to cell center arrays
886  call mem_setptr(cell_x, 'XC', this%input_mempath)
887  call mem_setptr(cell_y, 'YC', this%input_mempath)
888  !
889  ! -- set cell centers
890  if (associated(cell_x) .and. associated(cell_y)) then
891  do i = 1, this%nodesuser
892  this%cellxy(1, i) = cell_x(i)
893  this%cellxy(2, i) = cell_y(i)
894  end do
895  else
896  call store_error('Required cell center arrays not found.')
897  end if
898  !
899  ! -- log
900  if (this%iout > 0) then
901  write (this%iout, '(1x,a)') 'Discretization Cell2d data loaded'
902  end if
903  !
904  end subroutine source_cell2d
905 
906  !> @brief Write a binary grid file
907  !<
908  subroutine write_grb(this, icelltype)
909  ! -- modules
910  use openspecmodule, only: access, form
911  ! -- dummy
912  class(disutype) :: this
913  integer(I4B), dimension(:), intent(in) :: icelltype
914  ! -- local
915  integer(I4B) :: i, iunit, ntxt
916  integer(I4B), parameter :: lentxt = 100
917  character(len=50) :: txthdr
918  character(len=lentxt) :: txt
919  character(len=LINELENGTH) :: fname
920  ! -- formats
921  character(len=*), parameter :: fmtgrdsave = &
922  "(4X,'BINARY GRID INFORMATION WILL BE WRITTEN TO:', &
923  &/,6X,'UNIT NUMBER: ', I0,/,6X, 'FILE NAME: ', A)"
924  !
925  ! -- Initialize
926  ntxt = 10
927  if (this%nvert > 0) ntxt = ntxt + 5
928  !
929  ! -- Open the file
930  fname = trim(this%input_fname)//'.grb'
931  iunit = getunit()
932  write (this%iout, fmtgrdsave) iunit, trim(adjustl(fname))
933  call openfile(iunit, this%iout, trim(adjustl(fname)), 'DATA(BINARY)', &
934  form, access, 'REPLACE')
935  !
936  ! -- write header information
937  write (txthdr, '(a)') 'GRID DISU'
938  txthdr(50:50) = new_line('a')
939  write (iunit) txthdr
940  write (txthdr, '(a)') 'VERSION 1'
941  txthdr(50:50) = new_line('a')
942  write (iunit) txthdr
943  write (txthdr, '(a, i0)') 'NTXT ', ntxt
944  txthdr(50:50) = new_line('a')
945  write (iunit) txthdr
946  write (txthdr, '(a, i0)') 'LENTXT ', lentxt
947  txthdr(50:50) = new_line('a')
948  write (iunit) txthdr
949  !
950  ! -- write variable definitions
951  write (txt, '(3a, i0)') 'NODES ', 'INTEGER ', 'NDIM 0 # ', this%nodesuser
952  txt(lentxt:lentxt) = new_line('a')
953  write (iunit) txt
954  write (txt, '(3a, i0)') 'NJA ', 'INTEGER ', 'NDIM 0 # ', this%con%nja
955  txt(lentxt:lentxt) = new_line('a')
956  write (iunit) txt
957  write (txt, '(3a, 1pg24.15)') 'XORIGIN ', 'DOUBLE ', 'NDIM 0 # ', this%xorigin
958  txt(lentxt:lentxt) = new_line('a')
959  write (iunit) txt
960  write (txt, '(3a, 1pg24.15)') 'YORIGIN ', 'DOUBLE ', 'NDIM 0 # ', this%yorigin
961  txt(lentxt:lentxt) = new_line('a')
962  write (iunit) txt
963  write (txt, '(3a, 1pg24.15)') 'ANGROT ', 'DOUBLE ', 'NDIM 0 # ', this%angrot
964  txt(lentxt:lentxt) = new_line('a')
965  write (iunit) txt
966  write (txt, '(3a, i0)') 'TOP ', 'DOUBLE ', 'NDIM 1 ', this%nodesuser
967  txt(lentxt:lentxt) = new_line('a')
968  write (iunit) txt
969  write (txt, '(3a, i0)') 'BOT ', 'DOUBLE ', 'NDIM 1 ', this%nodesuser
970  txt(lentxt:lentxt) = new_line('a')
971  write (iunit) txt
972  write (txt, '(3a, i0)') 'IA ', 'INTEGER ', 'NDIM 1 ', this%nodesuser + 1
973  txt(lentxt:lentxt) = new_line('a')
974  write (iunit) txt
975  write (txt, '(3a, i0)') 'JA ', 'INTEGER ', 'NDIM 1 ', this%con%nja
976  txt(lentxt:lentxt) = new_line('a')
977  write (iunit) txt
978  write (txt, '(3a, i0)') 'ICELLTYPE ', 'INTEGER ', 'NDIM 1 ', this%nodesuser
979  txt(lentxt:lentxt) = new_line('a')
980  write (iunit) txt
981  !
982  ! -- if vertices have been read then write additional header information
983  if (this%nvert > 0) then
984  write (txt, '(3a, i0)') 'VERTICES ', 'DOUBLE ', 'NDIM 2 2 ', this%nvert
985  txt(lentxt:lentxt) = new_line('a')
986  write (iunit) txt
987  write (txt, '(3a, i0)') 'CELLX ', 'DOUBLE ', 'NDIM 1 ', this%nodesuser
988  txt(lentxt:lentxt) = new_line('a')
989  write (iunit) txt
990  write (txt, '(3a, i0)') 'CELLY ', 'DOUBLE ', 'NDIM 1 ', this%nodesuser
991  txt(lentxt:lentxt) = new_line('a')
992  write (iunit) txt
993  write (txt, '(3a, i0)') 'IAVERT ', 'INTEGER ', 'NDIM 1 ', this%nodesuser + 1
994  txt(lentxt:lentxt) = new_line('a')
995  write (iunit) txt
996  write (txt, '(3a, i0)') 'JAVERT ', 'INTEGER ', 'NDIM 1 ', size(this%javert)
997  txt(lentxt:lentxt) = new_line('a')
998  write (iunit) txt
999  end if
1000  !
1001  ! -- write data
1002  write (iunit) this%nodesuser ! nodes
1003  write (iunit) this%nja ! nja
1004  write (iunit) this%xorigin ! xorigin
1005  write (iunit) this%yorigin ! yorigin
1006  write (iunit) this%angrot ! angrot
1007  write (iunit) this%top1d ! top
1008  write (iunit) this%bot1d ! bot
1009  write (iunit) this%con%iausr ! ia
1010  write (iunit) this%con%jausr ! ja
1011  write (iunit) icelltype ! icelltype
1012  !
1013  ! -- if vertices have been read then write additional data
1014  if (this%nvert > 0) then
1015  write (iunit) this%vertices ! vertices
1016  write (iunit) (this%cellxy(1, i), i=1, this%nodesuser) ! cellx
1017  write (iunit) (this%cellxy(2, i), i=1, this%nodesuser) ! celly
1018  write (iunit) this%iavert ! iavert
1019  write (iunit) this%javert ! javert
1020  end if
1021  !
1022  ! -- Close the file
1023  close (iunit)
1024  !
1025  end subroutine write_grb
1026 
1027  !> @brief Get reduced node number from user node number
1028  !<
1029  function get_nodenumber_idx1(this, nodeu, icheck) result(nodenumber)
1030  class(disutype), intent(in) :: this
1031  integer(I4B), intent(in) :: nodeu
1032  integer(I4B), intent(in) :: icheck
1033  integer(I4B) :: nodenumber
1034  !
1035  if (icheck /= 0) then
1036  if (nodeu < 1 .or. nodeu > this%nodes) then
1037  write (errmsg, '(a,i0,a,i0,a)') &
1038  'Node number (', nodeu, ') is less than 1 or greater than nodes (', &
1039  this%nodes, ').'
1040  call store_error(errmsg)
1041  end if
1042  end if
1043  !
1044  ! -- set node number to passed in nodenumber since there is a one to one
1045  ! mapping for an unstructured grid
1046  if (this%nodes == this%nodesuser) then
1047  nodenumber = nodeu
1048  else
1049  nodenumber = this%nodereduced(nodeu)
1050  end if
1051  !
1052  end function get_nodenumber_idx1
1053 
1054  !> @brief Get normal vector components between the cell and a given neighbor
1055  !!
1056  !! The normal points outward from the shared face between noden and nodem.
1057  !<
1058  subroutine connection_normal(this, noden, nodem, ihc, xcomp, ycomp, zcomp, &
1059  ipos)
1060  ! -- dummy
1061  class(disutype) :: this
1062  integer(I4B), intent(in) :: noden !< cell (reduced nn)
1063  integer(I4B), intent(in) :: nodem !< neighbor (reduced nn)
1064  integer(I4B), intent(in) :: ihc !< horizontal connection flag
1065  real(DP), intent(inout) :: xcomp
1066  real(DP), intent(inout) :: ycomp
1067  real(DP), intent(inout) :: zcomp
1068  integer(I4B), intent(in) :: ipos
1069  ! -- local
1070  real(DP) :: angle, dmult
1071  !
1072  ! -- Set vector components based on ihc
1073  if (ihc == 0) then
1074  !
1075  ! -- connection is vertical
1076  xcomp = dzero
1077  ycomp = dzero
1078  if (nodem < noden) then
1079  !
1080  ! -- nodem must be above noden, so upward connection
1081  zcomp = done
1082  else
1083  !
1084  ! -- nodem must be below noden, so downward connection
1085  zcomp = -done
1086  end if
1087  else
1088  ! -- find from anglex, since anglex is symmetric, need to flip vector
1089  ! for lower triangle (nodem < noden)
1090  angle = this%con%anglex(this%con%jas(ipos))
1091  dmult = done
1092  if (nodem < noden) dmult = -done
1093  xcomp = cos(angle) * dmult
1094  ycomp = sin(angle) * dmult
1095  zcomp = dzero
1096  end if
1097  !
1098  end subroutine connection_normal
1099 
1100  !> @brief Get unit vector components between the cell and a given neighbor
1101  !!
1102  !! Saturation must be provided to compute cell center vertical coordinates.
1103  !! Also return the straight-line connection length.
1104  !<
1105  subroutine connection_vector(this, noden, nodem, nozee, satn, satm, ihc, &
1106  xcomp, ycomp, zcomp, conlen)
1107  ! -- dummy
1108  class(disutype) :: this
1109  integer(I4B), intent(in) :: noden
1110  integer(I4B), intent(in) :: nodem
1111  logical, intent(in) :: nozee
1112  real(DP), intent(in) :: satn
1113  real(DP), intent(in) :: satm
1114  integer(I4B), intent(in) :: ihc
1115  real(DP), intent(inout) :: xcomp
1116  real(DP), intent(inout) :: ycomp
1117  real(DP), intent(inout) :: zcomp
1118  real(DP), intent(inout) :: conlen
1119  ! -- local
1120  real(DP) :: xn, xm, yn, ym, zn, zm
1121  !
1122  ! -- Terminate with error if requesting unit vector components for problems
1123  ! without cell data
1124  if (size(this%cellxy, 2) < 1) then
1125  write (errmsg, '(a)') &
1126  'Cannot calculate unit vector components for DISU grid if VERTEX '// &
1127  'data are not specified'
1128  call store_error(errmsg, terminate=.true.)
1129  end if
1130  !
1131  ! -- get xy center coords
1132  xn = this%xc(noden)
1133  yn = this%yc(noden)
1134  xm = this%xc(nodem)
1135  ym = this%yc(nodem)
1136  !
1137  ! -- Set vector components based on ihc
1138  if (ihc == 0) then
1139  !
1140  ! -- vertical connection, calculate z as cell center elevation
1141  zn = this%bot(noden) + dhalf * (this%top(noden) - this%bot(noden))
1142  zm = this%bot(nodem) + dhalf * (this%top(nodem) - this%bot(nodem))
1143  else
1144  !
1145  ! -- horizontal connection, with possible z component due to cell offsets
1146  ! and/or water table conditions
1147  if (nozee) then
1148  zn = dzero
1149  zm = dzero
1150  else
1151  zn = this%bot(noden) + dhalf * satn * (this%top(noden) - this%bot(noden))
1152  zm = this%bot(nodem) + dhalf * satm * (this%top(nodem) - this%bot(nodem))
1153  end if
1154  end if
1155  !
1156  ! -- Use coords to find vector components and connection length
1157  call line_unit_vector(xn, yn, zn, xm, ym, zm, xcomp, ycomp, zcomp, &
1158  conlen)
1159  !
1160  end subroutine connection_vector
1161 
1162  !> @brief Get the discretization type
1163  !<
1164  subroutine get_dis_type(this, dis_type)
1165  ! -- dummy
1166  class(disutype), intent(in) :: this
1167  character(len=*), intent(out) :: dis_type
1168  !
1169  dis_type = "DISU"
1170  !
1171  end subroutine get_dis_type
1172 
1173  !> @brief Get the discretization type enumeration
1174  function get_dis_enum(this) result(dis_enum)
1175  use constantsmodule, only: disu
1176  class(disutype), intent(in) :: this
1177  integer(I4B) :: dis_enum
1178  dis_enum = disu
1179  end function get_dis_enum
1180 
1181  !> @brief Allocate and initialize scalar variables
1182  !<
1183  subroutine allocate_scalars(this, name_model, input_mempath)
1184  ! -- dummy
1185  class(disutype) :: this
1186  character(len=*), intent(in) :: name_model
1187  character(len=*), intent(in) :: input_mempath
1188  !
1189  ! -- Allocate parent scalars
1190  call this%DisBaseType%allocate_scalars(name_model, input_mempath)
1191  !
1192  ! -- Allocate variables for DISU
1193  call mem_allocate(this%njausr, 'NJAUSR', this%memoryPath)
1194  call mem_allocate(this%nvert, 'NVERT', this%memoryPath)
1195  call mem_allocate(this%voffsettol, 'VOFFSETTOL', this%memoryPath)
1196  call mem_allocate(this%iangledegx, 'IANGLEDEGX', this%memoryPath)
1197  !
1198  ! -- Set values
1199  this%ndim = 1
1200  this%njausr = 0
1201  this%nvert = 0
1202  this%voffsettol = dzero
1203  this%iangledegx = 0
1204  this%readFromFile = .false.
1205  !
1206  end subroutine allocate_scalars
1207 
1208  !> @brief Allocate and initialize arrays
1209  !<
1210  subroutine allocate_arrays(this)
1211  ! -- dummy
1212  class(disutype) :: this
1213  !
1214  ! -- Allocate arrays in DisBaseType (mshape, top, bot, area)
1215  call this%DisBaseType%allocate_arrays()
1216  !
1217  ! -- Allocate arrays in DISU
1218  if (this%nodes < this%nodesuser) then
1219  call mem_allocate(this%nodeuser, this%nodes, 'NODEUSER', this%memoryPath)
1220  call mem_allocate(this%nodereduced, this%nodesuser, 'NODEREDUCED', &
1221  this%memoryPath)
1222  else
1223  call mem_allocate(this%nodeuser, 1, 'NODEUSER', this%memoryPath)
1224  call mem_allocate(this%nodereduced, 1, 'NODEREDUCED', this%memoryPath)
1225  end if
1226  !
1227  ! -- Initialize
1228  this%mshape(1) = this%nodesuser
1229  !
1230  end subroutine allocate_arrays
1231 
1232  !> @brief Allocate arrays in memory manager
1233  !<
1234  subroutine allocate_arrays_mem(this)
1235  ! -- modules
1237  ! -- dummy
1238  class(disutype) :: this
1239  !
1240  call mem_allocate(this%idomain, this%nodes, 'IDOMAIN', this%memoryPath)
1241  call mem_allocate(this%vertices, 2, this%nvert, 'VERTICES', this%memoryPath)
1242  call mem_allocate(this%cellxy, 2, this%nodes, 'CELLXY', this%memoryPath)
1243  !
1244  end subroutine allocate_arrays_mem
1245 
1246  !> @brief Convert a string to a user nodenumber
1247  !!
1248  !! Parse and return user nodenumber.
1249  !! If flag_string is present and true, the first token may be
1250  !! non-numeric (e.g. boundary name). In this case, return -2.
1251  !<
1252  function nodeu_from_string(this, lloc, istart, istop, in, iout, line, &
1253  flag_string, allow_zero) result(nodeu)
1254  ! -- dummy
1255  class(disutype) :: this
1256  integer(I4B), intent(inout) :: lloc
1257  integer(I4B), intent(inout) :: istart
1258  integer(I4B), intent(inout) :: istop
1259  integer(I4B), intent(in) :: in
1260  integer(I4B), intent(in) :: iout
1261  character(len=*), intent(inout) :: line
1262  logical, optional, intent(in) :: flag_string
1263  logical, optional, intent(in) :: allow_zero
1264  integer(I4B) :: nodeu
1265  ! -- local
1266  integer(I4B) :: lloclocal, ndum, istat, n
1267  real(dp) :: r
1268  !
1269  if (present(flag_string)) then
1270  if (flag_string) then
1271  ! Check to see if first token in line can be read as an integer.
1272  lloclocal = lloc
1273  call urword(line, lloclocal, istart, istop, 1, ndum, r, iout, in)
1274  read (line(istart:istop), *, iostat=istat) n
1275  if (istat /= 0) then
1276  ! First token in line is not an integer; return flag to this effect.
1277  nodeu = -2
1278  return
1279  end if
1280  end if
1281  end if
1282  !
1283  call urword(line, lloc, istart, istop, 2, nodeu, r, iout, in)
1284  !
1285  if (nodeu == 0) then
1286  if (present(allow_zero)) then
1287  if (allow_zero) then
1288  return
1289  end if
1290  end if
1291  end if
1292  !
1293  if (nodeu < 1 .or. nodeu > this%nodesuser) then
1294  write (errmsg, '(a,i0,a)') &
1295  "Node number in list (", nodeu, ") is outside of the grid. "// &
1296  "Cell number cannot be determined in line '"// &
1297  trim(adjustl(line))//"'."
1298  call store_error(errmsg)
1299  call store_error_unit(in)
1300  end if
1301  !
1302  end function nodeu_from_string
1303 
1304  !> @brief Convert a cellid string to a user nodenumber
1305  !!
1306  !! If flag_string is present and true, the first token may be
1307  !! non-numeric (e.g. boundary name). In this case, return -2.
1308  !!
1309  !! If allow_zero is present and true, and all indices are zero, the
1310  !! result can be zero. If allow_zero is false, a zero in any index is an error.
1311  !<
1312  function nodeu_from_cellid(this, cellid, inunit, iout, flag_string, &
1313  allow_zero) result(nodeu)
1314  ! -- return
1315  integer(I4B) :: nodeu
1316  ! -- dummy
1317  class(disutype) :: this
1318  character(len=*), intent(inout) :: cellid
1319  integer(I4B), intent(in) :: inunit
1320  integer(I4B), intent(in) :: iout
1321  logical, optional, intent(in) :: flag_string
1322  logical, optional, intent(in) :: allow_zero
1323  ! -- local
1324  integer(I4B) :: lloclocal, istart, istop, ndum, n
1325  integer(I4B) :: istat
1326  real(dp) :: r
1327  !
1328  if (present(flag_string)) then
1329  if (flag_string) then
1330  ! Check to see if first token in cellid can be read as an integer.
1331  lloclocal = 1
1332  call urword(cellid, lloclocal, istart, istop, 1, ndum, r, iout, inunit)
1333  read (cellid(istart:istop), *, iostat=istat) n
1334  if (istat /= 0) then
1335  ! First token in cellid is not an integer; return flag to this effect.
1336  nodeu = -2
1337  return
1338  end if
1339  end if
1340  end if
1341  !
1342  lloclocal = 1
1343  call urword(cellid, lloclocal, istart, istop, 2, nodeu, r, iout, inunit)
1344  !
1345  if (nodeu == 0) then
1346  if (present(allow_zero)) then
1347  if (allow_zero) then
1348  return
1349  end if
1350  end if
1351  end if
1352  !
1353  if (nodeu < 1 .or. nodeu > this%nodesuser) then
1354  write (errmsg, '(a,i0,a)') &
1355  "Cell number cannot be determined for cellid ("// &
1356  trim(adjustl(cellid))//") and results in a user "// &
1357  "node number (", nodeu, ") that is outside of the grid."
1358  call store_error(errmsg)
1359  call store_error_unit(inunit)
1360  end if
1361  !
1362  end function nodeu_from_cellid
1363 
1364  !> @brief Indicates whether the grid discretization supports layers
1365  !<
1366  logical function supports_layers(this)
1367  ! -- dummy
1368  class(disutype) :: this
1369  !
1370  supports_layers = .false.
1371  !
1372  end function supports_layers
1373 
1374  !> @brief Get number of cells per layer (total nodes since DISU isn't layered)
1375  !<
1376  function get_ncpl(this)
1377  ! -- return
1378  integer(I4B) :: get_ncpl
1379  ! -- dummy
1380  class(disutype) :: this
1381  !
1382  get_ncpl = this%nodesuser
1383  !
1384  end function get_ncpl
1385 
1386  !> @brief Read an integer array
1387  !<
1388  subroutine read_int_array(this, line, lloc, istart, istop, iout, in, &
1389  iarray, aname)
1390  ! -- dummy
1391  class(disutype), intent(inout) :: this
1392  character(len=*), intent(inout) :: line
1393  integer(I4B), intent(inout) :: lloc
1394  integer(I4B), intent(inout) :: istart
1395  integer(I4B), intent(inout) :: istop
1396  integer(I4B), intent(in) :: in
1397  integer(I4B), intent(in) :: iout
1398  integer(I4B), dimension(:), pointer, contiguous, intent(inout) :: iarray
1399  character(len=*), intent(in) :: aname
1400  ! -- local
1401  integer(I4B) :: nval
1402  integer(I4B), dimension(:), pointer, contiguous :: itemp
1403  !
1404  ! -- Point the temporary pointer array, which is passed to the reading
1405  ! subroutine. The temporary array will point to ibuff if it is a
1406  ! reduced structured system, or to iarray if it is an unstructured
1407  ! model.
1408  if (this%nodes < this%nodesuser) then
1409  nval = this%nodesuser
1410  itemp => this%ibuff
1411  else
1412  nval = this%nodes
1413  itemp => iarray
1414  end if
1415  !
1416  ! -- Read the array
1417  ! -- Read unstructured input
1418  call readarray(in, itemp, aname, this%ndim, nval, iout, 0)
1419  !
1420  ! -- If reduced model, then need to copy from itemp(=>ibuff) to iarray
1421  if (this%nodes < this%nodesuser) then
1422  call this%fill_grid_array(itemp, iarray)
1423  end if
1424  !
1425  end subroutine read_int_array
1426 
1427  !> @brief Read a double precision array
1428  !<
1429  subroutine read_dbl_array(this, line, lloc, istart, istop, iout, in, &
1430  darray, aname)
1431  ! -- dummy
1432  class(disutype), intent(inout) :: this
1433  character(len=*), intent(inout) :: line
1434  integer(I4B), intent(inout) :: lloc
1435  integer(I4B), intent(inout) :: istart
1436  integer(I4B), intent(inout) :: istop
1437  integer(I4B), intent(in) :: in
1438  integer(I4B), intent(in) :: iout
1439  real(DP), dimension(:), pointer, contiguous, intent(inout) :: darray
1440  character(len=*), intent(in) :: aname
1441  ! -- local
1442  integer(I4B) :: nval
1443  real(DP), dimension(:), pointer, contiguous :: dtemp
1444  !
1445  ! -- Point the temporary pointer array, which is passed to the reading
1446  ! subroutine. The temporary array will point to dbuff if it is a
1447  ! reduced structured system, or to darray if it is an unstructured
1448  ! model.
1449  if (this%nodes < this%nodesuser) then
1450  nval = this%nodesuser
1451  dtemp => this%dbuff
1452  else
1453  nval = this%nodes
1454  dtemp => darray
1455  end if
1456  !
1457  ! -- Read the array
1458  call readarray(in, dtemp, aname, this%ndim, nval, iout, 0)
1459  !
1460  ! -- If reduced model, then need to copy from dtemp(=>dbuff) to darray
1461  if (this%nodes < this%nodesuser) then
1462  call this%fill_grid_array(dtemp, darray)
1463  end if
1464  !
1465  end subroutine read_dbl_array
1466 
1467  !> @brief Record a double precision array
1468  !!
1469  !! The array is written to a formatted or unformatted external file
1470  !! depending on the arguments.
1471  !<
1472  subroutine record_array(this, darray, iout, iprint, idataun, aname, &
1473  cdatafmp, nvaluesp, nwidthp, editdesc, dinact)
1474  ! -- dummy
1475  class(disutype), intent(inout) :: this
1476  real(DP), dimension(:), pointer, contiguous, intent(inout) :: darray !< double precision array to record
1477  integer(I4B), intent(in) :: iout !< ascii output unit number
1478  integer(I4B), intent(in) :: iprint !< whether to print the array
1479  integer(I4B), intent(in) :: idataun !< binary output unit number
1480  character(len=*), intent(in) :: aname !< text descriptor
1481  character(len=*), intent(in) :: cdatafmp ! write format
1482  integer(I4B), intent(in) :: nvaluesp !< values per line
1483  integer(I4B), intent(in) :: nwidthp !< number width
1484  character(len=*), intent(in) :: editdesc !< format type (I, G, F, S, E)
1485  real(DP), intent(in) :: dinact !< double precision value for cells excluded from model domain
1486  ! -- local
1487  integer(I4B) :: k, ifirst
1488  integer(I4B) :: nlay
1489  integer(I4B) :: nrow
1490  integer(I4B) :: ncol
1491  integer(I4B) :: nval
1492  integer(I4B) :: nodeu, noder
1493  integer(I4B) :: istart, istop
1494  real(DP), dimension(:), pointer, contiguous :: dtemp
1495  ! -- formats
1496  character(len=*), parameter :: fmthsv = &
1497  "(1X,/1X,a,' WILL BE SAVED ON UNIT ',I4, &
1498  &' AT END OF TIME STEP',I5,', STRESS PERIOD ',I4)"
1499  !
1500  ! -- set variables
1501  nlay = 1
1502  nrow = 1
1503  ncol = this%mshape(1)
1504  !
1505  ! -- If this is a reduced model, then copy the values from darray into
1506  ! dtemp.
1507  if (this%nodes < this%nodesuser) then
1508  nval = this%nodes
1509  dtemp => this%dbuff
1510  do nodeu = 1, this%nodesuser
1511  noder = this%get_nodenumber(nodeu, 0)
1512  if (noder <= 0) then
1513  dtemp(nodeu) = dinact
1514  cycle
1515  end if
1516  dtemp(nodeu) = darray(noder)
1517  end do
1518  else
1519  nval = this%nodes
1520  dtemp => darray
1521  end if
1522  !
1523  ! -- Print to iout if iprint /= 0
1524  if (iprint /= 0) then
1525  istart = 1
1526  do k = 1, nlay
1527  istop = istart + nrow * ncol - 1
1528  call ulaprufw(ncol, nrow, kstp, kper, k, iout, dtemp(istart:istop), &
1529  aname, cdatafmp, nvaluesp, nwidthp, editdesc)
1530  istart = istop + 1
1531  end do
1532  end if
1533  !
1534  ! -- Save array to an external file.
1535  if (idataun > 0) then
1536  ! -- write to binary file by layer
1537  ifirst = 1
1538  istart = 1
1539  do k = 1, nlay
1540  istop = istart + nrow * ncol - 1
1541  if (ifirst == 1) write (iout, fmthsv) &
1542  trim(adjustl(aname)), idataun, &
1543  kstp, kper
1544  ifirst = 0
1545  call ulasav(dtemp(istart:istop), aname, kstp, kper, &
1546  pertim, totim, ncol, nrow, k, idataun)
1547  istart = istop + 1
1548  end do
1549  elseif (idataun < 0) then
1550  !
1551  ! -- write entire array as one record
1552  call ubdsv1(kstp, kper, aname, -idataun, dtemp, ncol, nrow, nlay, &
1553  iout, delt, pertim, totim)
1554  end if
1555  !
1556  end subroutine record_array
1557 
1558  !> @brief Record list header for imeth=6
1559  !<
1560  subroutine record_srcdst_list_header(this, text, textmodel, textpackage, &
1561  dstmodel, dstpackage, naux, auxtxt, &
1562  ibdchn, nlist, iout)
1563  ! -- dummy
1564  class(disutype) :: this
1565  character(len=16), intent(in) :: text
1566  character(len=16), intent(in) :: textmodel
1567  character(len=16), intent(in) :: textpackage
1568  character(len=16), intent(in) :: dstmodel
1569  character(len=16), intent(in) :: dstpackage
1570  integer(I4B), intent(in) :: naux
1571  character(len=16), dimension(:), intent(in) :: auxtxt
1572  integer(I4B), intent(in) :: ibdchn
1573  integer(I4B), intent(in) :: nlist
1574  integer(I4B), intent(in) :: iout
1575  ! -- local
1576  integer(I4B) :: nlay, nrow, ncol
1577  !
1578  nlay = 1
1579  nrow = 1
1580  ncol = this%mshape(1)
1581  !
1582  ! -- Use ubdsv06 to write list header
1583  call ubdsv06(kstp, kper, text, textmodel, textpackage, dstmodel, dstpackage, &
1584  ibdchn, naux, auxtxt, ncol, nrow, nlay, &
1585  nlist, iout, delt, pertim, totim)
1586  !
1587  end subroutine record_srcdst_list_header
1588 
1589  !> @brief Cast base to DISU
1590  !<
1591  function castasdisutype(dis) result(disu)
1592  ! -- dummy
1593  class(*), pointer :: dis !< base pointer to DISU object
1594  ! -- return
1595  class(disutype), pointer :: disu !< the resulting DISU pointer
1596  !
1597  disu => null()
1598  select type (dis)
1599  class is (disutype)
1600  disu => dis
1601  end select
1602  !
1603  end function castasdisutype
1604 
1605 end module disumodule
subroutine, public iac_to_ia(iac, ia)
Convert an iac array into an ia array.
This module contains simulation constants.
Definition: Constants.f90:9
integer(i4b), parameter linelength
maximum length of a standard line
Definition: Constants.f90:44
@ disu
DISV6 discretization.
Definition: Constants.f90:156
integer(i4b), parameter lenvarname
maximum length of a variable name
Definition: Constants.f90:17
real(dp), parameter dhalf
real constant 1/2
Definition: Constants.f90:67
real(dp), parameter dzero
real constant zero
Definition: Constants.f90:64
integer(i4b), parameter lenmempath
maximum length of the memory path
Definition: Constants.f90:26
real(dp), parameter done
real constant 1
Definition: Constants.f90:75
subroutine allocate_scalars(this, name_model, input_mempath)
Allocate and initialize scalar variables.
Definition: Disu.f90:1184
subroutine write_grb(this, icelltype)
Write a binary grid file.
Definition: Disu.f90:909
subroutine allocate_arrays(this)
Allocate and initialize arrays.
Definition: Disu.f90:1211
subroutine disu_load(this)
Transfer IDM data into this discretization object.
Definition: Disu.f90:165
subroutine source_connectivity(this)
Copy grid connectivity info from IDM into package.
Definition: Disu.f90:718
subroutine, public disu_cr(dis, name_model, input_mempath, inunit, iout)
Create a new unstructured discretization object.
Definition: Disu.f90:127
subroutine source_dimensions(this)
Copy dimensions from IDM into package.
Definition: Disu.f90:583
subroutine source_options(this)
Copy options from IDM into package.
Definition: Disu.f90:516
subroutine log_dimensions(this, found)
Write dimensions to list file.
Definition: Disu.f90:644
subroutine disu_da(this)
Deallocate variables.
Definition: Disu.f90:432
class(disutype) function, pointer, public castasdisutype(dis)
Cast base to DISU.
Definition: Disu.f90:1592
integer(i4b) function get_ncpl(this)
Get number of cells per layer (total nodes since DISU isn't layered)
Definition: Disu.f90:1377
subroutine define_cellverts(this, icell2d, ncvert, icvert)
Build data structures to hold cell vertex info.
Definition: Disu.f90:821
subroutine read_dbl_array(this, line, lloc, istart, istop, iout, in, darray, aname)
Read a double precision array.
Definition: Disu.f90:1431
integer(i4b) function nodeu_from_cellid(this, cellid, inunit, iout, flag_string, allow_zero)
Convert a cellid string to a user nodenumber.
Definition: Disu.f90:1314
subroutine log_options(this, found)
Write user options to list file.
Definition: Disu.f90:543
subroutine record_srcdst_list_header(this, text, textmodel, textpackage, dstmodel, dstpackage, naux, auxtxt, ibdchn, nlist, iout)
Record list header for imeth=6.
Definition: Disu.f90:1563
subroutine get_dis_type(this, dis_type)
Get the discretization type.
Definition: Disu.f90:1165
subroutine source_vertices(this)
Copy grid vertex data from IDM into package.
Definition: Disu.f90:789
subroutine source_cell2d(this)
Copy cell2d data from IDM into package.
Definition: Disu.f90:862
subroutine log_griddata(this, found)
Write griddata found to list file.
Definition: Disu.f90:689
subroutine allocate_arrays_mem(this)
Allocate arrays in memory manager.
Definition: Disu.f90:1235
subroutine source_griddata(this)
Copy grid data from IDM into package.
Definition: Disu.f90:668
subroutine nodeu_to_array(this, nodeu, arr)
Convert a user nodenumber to an array (nodenumber)
Definition: Disu.f90:493
subroutine connection_vector(this, noden, nodem, nozee, satn, satm, ihc, xcomp, ycomp, zcomp, conlen)
Get unit vector components between the cell and a given neighbor.
Definition: Disu.f90:1107
subroutine record_array(this, darray, iout, iprint, idataun, aname, cdatafmp, nvaluesp, nwidthp, editdesc, dinact)
Record a double precision array.
Definition: Disu.f90:1474
subroutine disu_df(this)
Define the discretization.
Definition: Disu.f90:192
logical function supports_layers(this)
Indicates whether the grid discretization supports layers.
Definition: Disu.f90:1367
subroutine read_int_array(this, line, lloc, istart, istop, iout, in, iarray, aname)
Read an integer array.
Definition: Disu.f90:1390
subroutine grid_finalize(this)
Finalize the grid.
Definition: Disu.f90:202
integer(i4b) function get_nodenumber_idx1(this, nodeu, icheck)
Get reduced node number from user node number.
Definition: Disu.f90:1030
subroutine nodeu_to_string(this, nodeu, str)
Convert a user nodenumber to a string (nodenumber)
Definition: Disu.f90:478
integer(i4b) function nodeu_from_string(this, lloc, istart, istop, in, iout, line, flag_string, allow_zero)
Convert a string to a user nodenumber.
Definition: Disu.f90:1254
subroutine disu_ck(this)
Check discretization info.
Definition: Disu.f90:312
integer(i4b) function get_dis_enum(this)
Get the discretization type enumeration.
Definition: Disu.f90:1175
subroutine log_connectivity(this, found, iac)
Write griddata found to list file.
Definition: Disu.f90:752
subroutine connection_normal(this, noden, nodem, ihc, xcomp, ycomp, zcomp, ipos)
Get normal vector components between the cell and a given neighbor.
Definition: Disu.f90:1060
subroutine, public line_unit_vector(x0, y0, z0, x1, y1, z1, xcomp, ycomp, zcomp, vmag)
Calculate the vector components (xcomp, ycomp, and zcomp) for a line defined by two points,...
Definition: DisvGeom.f90:508
subroutine, public ubdsv1(kstp, kper, text, ibdchn, buff, ncol, nrow, nlay, iout, delt, pertim, totim)
Record cell-by-cell flow terms for one component of flow as a 3-D array with extra record to indicate...
integer(i4b) function, public getunit()
Get a free unit number.
subroutine, public ulaprufw(ncol, nrow, kstp, kper, ilay, iout, buf, text, userfmt, nvalues, nwidth, editdesc)
Print 1 layer array with user formatting in wrap format.
subroutine, public ubdsv06(kstp, kper, text, modelnam1, paknam1, modelnam2, paknam2, ibdchn, naux, auxtxt, ncol, nrow, nlay, nlist, iout, delt, pertim, totim)
Write header records for cell-by-cell flow terms for one component of flow.
subroutine, public ulasav(buf, text, kstp, kper, pertim, totim, ncol, nrow, ilay, ichn)
Save 1 layer array on disk.
subroutine, public openfile(iu, iout, fname, ftype, fmtarg_opt, accarg_opt, filstat_opt, mode_opt)
Open a file.
Definition: InputOutput.f90:30
subroutine, public urword(line, icol, istart, istop, ncode, n, r, iout, in)
Extract a word from a string.
This module defines variable data types.
Definition: kind.f90:8
subroutine, public memorylist_remove(component, subcomponent, context)
character(len=20) access
Definition: OpenSpec.f90:7
character(len=20) form
Definition: OpenSpec.f90:7
This module contains simulation methods.
Definition: Sim.f90:10
subroutine, public store_error(msg, terminate)
Store an error message.
Definition: Sim.f90:92
integer(i4b) function, public count_errors()
Return number of errors.
Definition: Sim.f90:59
subroutine, public store_error_filename(filename, terminate)
Store the erroring file name.
Definition: Sim.f90:203
subroutine, public store_error_unit(iunit, terminate)
Store the file unit number.
Definition: Sim.f90:168
This module contains simulation variables.
Definition: SimVariables.f90:9
character(len=maxcharlen) errmsg
error message string
character(len=linelength) idm_context
real(dp), pointer, public pertim
time relative to start of stress period
Definition: tdis.f90:30
real(dp), pointer, public totim
time relative to start of simulation
Definition: tdis.f90:32
integer(i4b), pointer, public kstp
current time step number
Definition: tdis.f90:24
integer(i4b), pointer, public kper
current stress period number
Definition: tdis.f90:23
real(dp), pointer, public delt
length of the current time step
Definition: tdis.f90:29
Unstructured grid discretization.
Definition: Disu.f90:28